377 lines
8.4 KiB
C++
377 lines
8.4 KiB
C++
|
//---------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Microsoft Windows
|
||
|
|
// Copyright (C) Microsoft Corporation, 1992 - 1995
|
||
|
|
//
|
||
|
|
// File: cenumdom.cxx
|
||
|
|
//
|
||
|
|
// Contents: NDS Object Enumeration Code
|
||
|
|
//
|
||
|
|
// CNDSTreeEnum::CNDSTreeEnum()
|
||
|
|
// CNDSTreeEnum::CNDSTreeEnum
|
||
|
|
// CNDSTreeEnum::EnumObjects
|
||
|
|
// CNDSTreeEnum::EnumObjects
|
||
|
|
//
|
||
|
|
// History:
|
||
|
|
//----------------------------------------------------------------------------
|
||
|
|
#include "NDS.hxx"
|
||
|
|
#pragma hdrstop
|
||
|
|
|
||
|
|
|
||
|
|
//+---------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: CNDSEnumVariant::Create
|
||
|
|
//
|
||
|
|
// Synopsis:
|
||
|
|
//
|
||
|
|
// Arguments: [pCollection]
|
||
|
|
// [ppEnumVariant]
|
||
|
|
//
|
||
|
|
// Returns: HRESULT
|
||
|
|
//
|
||
|
|
// Modifies:
|
||
|
|
//
|
||
|
|
// History: 01-30-95 krishnag Created.
|
||
|
|
//
|
||
|
|
//----------------------------------------------------------------------------
|
||
|
|
HRESULT
|
||
|
|
CNDSTreeEnum::Create(
|
||
|
|
CNDSTreeEnum FAR* FAR* ppenumvariant,
|
||
|
|
BSTR ADsPath,
|
||
|
|
VARIANT var,
|
||
|
|
CCredentials& Credentials
|
||
|
|
)
|
||
|
|
{
|
||
|
|
HRESULT hr = NOERROR;
|
||
|
|
CNDSTreeEnum FAR* penumvariant = NULL;
|
||
|
|
WCHAR szObjectFullName[MAX_PATH];
|
||
|
|
WCHAR szObjectClassName[MAX_PATH];
|
||
|
|
LPWSTR pszNDSPath = NULL;
|
||
|
|
DWORD dwModificationTime = 0L;
|
||
|
|
DWORD dwNumberOfEntries = 0L;
|
||
|
|
DWORD dwStatus = 0L;
|
||
|
|
|
||
|
|
*ppenumvariant = NULL;
|
||
|
|
|
||
|
|
penumvariant = new CNDSTreeEnum();
|
||
|
|
if (!penumvariant) {
|
||
|
|
hr = E_OUTOFMEMORY;
|
||
|
|
BAIL_ON_FAILURE(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
hr = ADsAllocString( ADsPath, &penumvariant->_ADsPath);
|
||
|
|
BAIL_ON_FAILURE(hr);
|
||
|
|
|
||
|
|
hr = BuildNDSFilterArray(
|
||
|
|
var,
|
||
|
|
(LPBYTE *)&penumvariant->_pNdsFilterList
|
||
|
|
);
|
||
|
|
if (FAILED(hr)) {
|
||
|
|
penumvariant->_pNdsFilterList = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
hr = ObjectTypeList::CreateObjectTypeList(
|
||
|
|
var,
|
||
|
|
&penumvariant->_pObjList
|
||
|
|
);
|
||
|
|
BAIL_ON_FAILURE(hr);
|
||
|
|
*/
|
||
|
|
penumvariant->_Credentials = Credentials;
|
||
|
|
|
||
|
|
*ppenumvariant = penumvariant;
|
||
|
|
|
||
|
|
hr = BuildNDSPathFromADsPath(
|
||
|
|
ADsPath,
|
||
|
|
&pszNDSPath
|
||
|
|
);
|
||
|
|
BAIL_ON_FAILURE(hr);
|
||
|
|
|
||
|
|
dwStatus = ADsNwNdsOpenObject(
|
||
|
|
pszNDSPath,
|
||
|
|
Credentials,
|
||
|
|
&penumvariant->_hObject,
|
||
|
|
szObjectFullName,
|
||
|
|
szObjectClassName,
|
||
|
|
&dwModificationTime,
|
||
|
|
&dwNumberOfEntries
|
||
|
|
);
|
||
|
|
|
||
|
|
if (dwStatus) {
|
||
|
|
hr = HRESULT_FROM_WIN32(GetLastError());
|
||
|
|
BAIL_ON_FAILURE(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (pszNDSPath) {
|
||
|
|
FreeADsStr(pszNDSPath);
|
||
|
|
}
|
||
|
|
|
||
|
|
RRETURN(hr);
|
||
|
|
|
||
|
|
error:
|
||
|
|
delete penumvariant;
|
||
|
|
*ppenumvariant = NULL;
|
||
|
|
|
||
|
|
if (pszNDSPath) {
|
||
|
|
|
||
|
|
FreeADsStr(pszNDSPath);
|
||
|
|
}
|
||
|
|
|
||
|
|
RRETURN_EXP_IF_ERR(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
CNDSTreeEnum::CNDSTreeEnum():
|
||
|
|
_ADsPath(NULL)
|
||
|
|
{
|
||
|
|
_pObjList = NULL;
|
||
|
|
_dwObjectReturned = 0;
|
||
|
|
_dwObjectCurrentEntry = 0;
|
||
|
|
_dwObjectTotal = 0;
|
||
|
|
_hObject = NULL;
|
||
|
|
_hOperationData = NULL;
|
||
|
|
_lpObjects = NULL;
|
||
|
|
_pNdsFilterList = NULL;
|
||
|
|
|
||
|
|
_fSchemaReturned = NULL;
|
||
|
|
_bNoMore = FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
CNDSTreeEnum::~CNDSTreeEnum()
|
||
|
|
{
|
||
|
|
if (_pNdsFilterList) {
|
||
|
|
FreeFilterList((LPBYTE)_pNdsFilterList);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
CNDSTreeEnum::EnumGenericObjects(
|
||
|
|
ULONG cElements,
|
||
|
|
VARIANT FAR* pvar,
|
||
|
|
ULONG FAR* pcElementFetched
|
||
|
|
)
|
||
|
|
{
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
IDispatch *pDispatch = NULL;
|
||
|
|
DWORD i = 0;
|
||
|
|
|
||
|
|
while (i < cElements) {
|
||
|
|
|
||
|
|
hr = GetGenObject(&pDispatch);
|
||
|
|
if (hr == S_FALSE) {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
VariantInit(&pvar[i]);
|
||
|
|
pvar[i].vt = VT_DISPATCH;
|
||
|
|
pvar[i].pdispVal = pDispatch;
|
||
|
|
(*pcElementFetched)++;
|
||
|
|
i++;
|
||
|
|
}
|
||
|
|
return(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
CNDSTreeEnum::GetGenObject(
|
||
|
|
IDispatch ** ppDispatch
|
||
|
|
)
|
||
|
|
{
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
DWORD dwStatus = 0L;
|
||
|
|
LPNDS_OBJECT_INFO lpCurrentObject = NULL;
|
||
|
|
IADs * pADs = NULL;
|
||
|
|
|
||
|
|
*ppDispatch = NULL;
|
||
|
|
|
||
|
|
if (!_hOperationData || (_dwObjectCurrentEntry == _dwObjectReturned)) {
|
||
|
|
|
||
|
|
if (_hOperationData) {
|
||
|
|
dwStatus = NwNdsFreeBuffer(_hOperationData);
|
||
|
|
_hOperationData = NULL;
|
||
|
|
_lpObjects = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
_dwObjectCurrentEntry = 0;
|
||
|
|
_dwObjectReturned = 0;
|
||
|
|
|
||
|
|
//
|
||
|
|
// Insert NDS code in here
|
||
|
|
//
|
||
|
|
if (_bNoMore) {
|
||
|
|
RRETURN(S_FALSE);
|
||
|
|
}
|
||
|
|
|
||
|
|
dwStatus = NwNdsListSubObjects(
|
||
|
|
_hObject,
|
||
|
|
MAX_CACHE_SIZE,
|
||
|
|
&_dwObjectReturned,
|
||
|
|
_pNdsFilterList,
|
||
|
|
&_hOperationData
|
||
|
|
);
|
||
|
|
if ((dwStatus != ERROR_SUCCESS) && (dwStatus != WN_NO_MORE_ENTRIES)) {
|
||
|
|
hr = HRESULT_FROM_WIN32(GetLastError());
|
||
|
|
BAIL_ON_FAILURE(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (dwStatus == WN_NO_MORE_ENTRIES) {
|
||
|
|
_bNoMore = TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
dwStatus = NwNdsGetObjectListFromBuffer(
|
||
|
|
_hOperationData,
|
||
|
|
&_dwObjectReturned,
|
||
|
|
NULL,
|
||
|
|
&_lpObjects
|
||
|
|
);
|
||
|
|
if (dwStatus != ERROR_SUCCESS) {
|
||
|
|
hr = HRESULT_FROM_WIN32(GetLastError());
|
||
|
|
}
|
||
|
|
BAIL_ON_FAILURE(hr);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Now send back the current object
|
||
|
|
//
|
||
|
|
|
||
|
|
lpCurrentObject = _lpObjects + _dwObjectCurrentEntry;
|
||
|
|
|
||
|
|
hr = CNDSGenObject::CreateGenericObject(
|
||
|
|
_ADsPath,
|
||
|
|
lpCurrentObject->szObjectName,
|
||
|
|
lpCurrentObject->szObjectClass,
|
||
|
|
_Credentials,
|
||
|
|
ADS_OBJECT_BOUND,
|
||
|
|
IID_IADs,
|
||
|
|
(void **)&pADs
|
||
|
|
);
|
||
|
|
BAIL_ON_FAILURE(hr);
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// InstantiateDerivedObject should addref this pointer for us.
|
||
|
|
//
|
||
|
|
|
||
|
|
hr = InstantiateDerivedObject(
|
||
|
|
pADs,
|
||
|
|
_Credentials,
|
||
|
|
IID_IDispatch,
|
||
|
|
(void **)ppDispatch
|
||
|
|
);
|
||
|
|
|
||
|
|
if (FAILED(hr)) {
|
||
|
|
hr = pADs->QueryInterface(
|
||
|
|
IID_IDispatch,
|
||
|
|
(void **)ppDispatch
|
||
|
|
);
|
||
|
|
BAIL_ON_FAILURE(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
_dwObjectCurrentEntry++;
|
||
|
|
|
||
|
|
|
||
|
|
error:
|
||
|
|
|
||
|
|
//
|
||
|
|
// GetGenObject returns only S_FALSE
|
||
|
|
//
|
||
|
|
|
||
|
|
if (FAILED(hr)) {
|
||
|
|
hr = S_FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Free the intermediate pADs pointer.
|
||
|
|
//
|
||
|
|
if (pADs) {
|
||
|
|
pADs->Release();
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
RRETURN_EXP_IF_ERR(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//+---------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: CNDSTreeEnum::Next
|
||
|
|
//
|
||
|
|
// Synopsis: Returns cElements number of requested NetOle objects in the
|
||
|
|
// array supplied in pvar.
|
||
|
|
//
|
||
|
|
// Arguments: [cElements] -- The number of elements requested by client
|
||
|
|
// [pvar] -- ptr to array of VARIANTs to for return objects
|
||
|
|
// [pcElementFetched] -- if non-NULL, then number of elements
|
||
|
|
// -- actually returned is placed here
|
||
|
|
//
|
||
|
|
// Returns: HRESULT -- S_OK if number of elements requested are returned
|
||
|
|
// -- S_FALSE if number of elements is < requested
|
||
|
|
//
|
||
|
|
// Modifies:
|
||
|
|
//
|
||
|
|
// History: 11-3-95 krishnag Created.
|
||
|
|
//
|
||
|
|
//----------------------------------------------------------------------------
|
||
|
|
STDMETHODIMP
|
||
|
|
CNDSTreeEnum::Next(
|
||
|
|
ULONG cElements,
|
||
|
|
VARIANT FAR* pvar,
|
||
|
|
ULONG FAR* pcElementFetched
|
||
|
|
)
|
||
|
|
{
|
||
|
|
ULONG cElementFetched = 0;
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
|
||
|
|
hr = EnumGenericObjects(
|
||
|
|
cElements,
|
||
|
|
pvar,
|
||
|
|
&cElementFetched
|
||
|
|
);
|
||
|
|
|
||
|
|
|
||
|
|
if (pcElementFetched) {
|
||
|
|
*pcElementFetched = cElementFetched;
|
||
|
|
}
|
||
|
|
RRETURN_EXP_IF_ERR(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
CNDSTreeEnum::EnumSchema(
|
||
|
|
ULONG cElements,
|
||
|
|
VARIANT FAR* pvar,
|
||
|
|
ULONG FAR* pcElementFetched
|
||
|
|
)
|
||
|
|
{
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
IDispatch *pDispatch = NULL;
|
||
|
|
|
||
|
|
if ( _fSchemaReturned )
|
||
|
|
RRETURN(S_FALSE);
|
||
|
|
|
||
|
|
if ( cElements > 0 )
|
||
|
|
{
|
||
|
|
hr = CNDSSchema::CreateSchema(
|
||
|
|
_ADsPath,
|
||
|
|
TEXT("Schema"),
|
||
|
|
_Credentials,
|
||
|
|
ADS_OBJECT_BOUND,
|
||
|
|
IID_IDispatch,
|
||
|
|
(void **)&pDispatch
|
||
|
|
);
|
||
|
|
|
||
|
|
if ( hr == S_OK )
|
||
|
|
{
|
||
|
|
VariantInit(&pvar[0]);
|
||
|
|
pvar[0].vt = VT_DISPATCH;
|
||
|
|
pvar[0].pdispVal = pDispatch;
|
||
|
|
(*pcElementFetched)++;
|
||
|
|
_fSchemaReturned = TRUE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
RRETURN(hr);
|
||
|
|
}
|
||
|
|
|