mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
507 lines
12 KiB
507 lines
12 KiB
//---------------------------------------------------------------------------
|
|
//
|
|
// Microsoft Windows
|
|
// Copyright (C) Microsoft Corporation, 1992 - 1996
|
|
//
|
|
// File: cenumsch.cxx
|
|
//
|
|
// Contents: LDAP Schema Enumeration Code
|
|
//
|
|
// CLDAPSchemaEnum::CLDAPSchemaEnum()
|
|
// CLDAPSchemaEnum::CLDAPSchemaEnum
|
|
// CLDAPSchemaEnum::EnumObjects
|
|
// CLDAPSchemaEnum::EnumObjects
|
|
//
|
|
// History:
|
|
//----------------------------------------------------------------------------
|
|
#include "ldap.hxx"
|
|
#pragma hdrstop
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: CLDAPSchemaEnum::Create
|
|
//
|
|
// Synopsis:
|
|
//
|
|
// Arguments: [pCollection]
|
|
// [ppEnumVariant]
|
|
//
|
|
// Returns: HRESULT
|
|
//
|
|
// Modifies:
|
|
//
|
|
// History: 01-30-95 yihsins Created.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
HRESULT
|
|
CLDAPSchemaEnum::Create(
|
|
CLDAPSchemaEnum FAR* FAR* ppenumvariant,
|
|
BSTR bstrADsPath,
|
|
BSTR bstrServerPath,
|
|
VARIANT vFilter,
|
|
CCredentials& Credentials
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
CLDAPSchemaEnum FAR* penumvariant = NULL;
|
|
|
|
OBJECTINFO ObjectInfo;
|
|
POBJECTINFO pObjectInfo = &ObjectInfo;
|
|
|
|
memset(pObjectInfo, 0, sizeof(OBJECTINFO));
|
|
|
|
*ppenumvariant = NULL;
|
|
|
|
penumvariant = new CLDAPSchemaEnum();
|
|
if (!penumvariant)
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
BAIL_ON_FAILURE(hr);
|
|
}
|
|
|
|
hr = ADsAllocString( bstrADsPath, &penumvariant->_bstrADsPath);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
hr = ADsAllocString( bstrServerPath, &penumvariant->_bstrServerPath);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
hr = ADsObject(bstrADsPath, pObjectInfo);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
hr = SchemaOpen(
|
|
bstrServerPath,
|
|
&(penumvariant->_hSchema),
|
|
Credentials,
|
|
pObjectInfo->PortNumber
|
|
);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
hr = SchemaGetObjectCount(
|
|
penumvariant->_hSchema,
|
|
&(penumvariant->_nNumOfClasses),
|
|
&(penumvariant->_nNumOfProperties) );
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
hr = ObjectTypeList::CreateObjectTypeList(
|
|
vFilter,
|
|
&penumvariant->_pObjList );
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
penumvariant->_Credentials = Credentials;
|
|
|
|
*ppenumvariant = penumvariant;
|
|
|
|
FreeObjectInfo(pObjectInfo);
|
|
|
|
RRETURN(hr);
|
|
|
|
error:
|
|
|
|
FreeObjectInfo(pObjectInfo);
|
|
|
|
delete penumvariant;
|
|
|
|
RRETURN_EXP_IF_ERR(hr);
|
|
}
|
|
|
|
CLDAPSchemaEnum::CLDAPSchemaEnum()
|
|
: _bstrADsPath( NULL ),
|
|
_bstrServerPath( NULL ),
|
|
_hSchema( NULL ),
|
|
_pObjList( NULL ),
|
|
_dwCurrentEntry( 0 ),
|
|
_nNumOfClasses( 0 ),
|
|
_nNumOfProperties( 0 )
|
|
{
|
|
}
|
|
|
|
CLDAPSchemaEnum::~CLDAPSchemaEnum()
|
|
{
|
|
ADsFreeString( _bstrADsPath );
|
|
ADsFreeString( _bstrServerPath );
|
|
|
|
if ( _hSchema )
|
|
{
|
|
SchemaClose( &_hSchema );
|
|
}
|
|
|
|
if ( _pObjList != NULL )
|
|
{
|
|
delete _pObjList;
|
|
_pObjList = NULL;
|
|
}
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: CLDAPSchemaEnum::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 yihsins Created.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
STDMETHODIMP
|
|
CLDAPSchemaEnum::Next(
|
|
ULONG cElements,
|
|
VARIANT FAR* pvar,
|
|
ULONG FAR* pcElementFetched
|
|
)
|
|
{
|
|
ULONG cElementFetched = 0;
|
|
HRESULT hr = S_OK;
|
|
|
|
hr = EnumObjects( cElements,
|
|
pvar,
|
|
&cElementFetched );
|
|
|
|
if ( pcElementFetched )
|
|
*pcElementFetched = cElementFetched;
|
|
|
|
RRETURN_EXP_IF_ERR(hr);
|
|
}
|
|
|
|
HRESULT
|
|
CLDAPSchemaEnum::EnumObjects(
|
|
DWORD ObjectType,
|
|
ULONG cElements,
|
|
VARIANT FAR * pvar,
|
|
ULONG FAR * pcElementFetched
|
|
)
|
|
{
|
|
HRESULT hr;
|
|
switch (ObjectType)
|
|
{
|
|
case LDAP_CLASS_ID:
|
|
hr = EnumClasses(cElements, pvar, pcElementFetched);
|
|
break;
|
|
|
|
case LDAP_PROPERTY_ID:
|
|
hr = EnumProperties(cElements, pvar, pcElementFetched);
|
|
break;
|
|
|
|
case LDAP_SYNTAX_ID:
|
|
hr = EnumSyntaxObjects(cElements, pvar, pcElementFetched);
|
|
break;
|
|
|
|
default:
|
|
hr = S_FALSE;
|
|
|
|
}
|
|
RRETURN(hr);
|
|
}
|
|
|
|
HRESULT
|
|
CLDAPSchemaEnum::EnumObjects(
|
|
ULONG cElements,
|
|
VARIANT FAR* pvar,
|
|
ULONG FAR* pcElementFetched
|
|
)
|
|
{
|
|
DWORD i;
|
|
ULONG cRequested = 0;
|
|
ULONG cFetchedByPath = 0;
|
|
ULONG cTotalFetched = 0;
|
|
VARIANT FAR* pPathvar = pvar;
|
|
HRESULT hr = S_FALSE;
|
|
DWORD ObjectType;
|
|
|
|
for (i = 0; i < cElements; i++)
|
|
VariantInit(&pvar[i]);
|
|
|
|
cRequested = cElements;
|
|
|
|
while ( SUCCEEDED( _pObjList->GetCurrentObject(&ObjectType))
|
|
&& ((hr = EnumObjects( ObjectType,
|
|
cRequested,
|
|
pPathvar,
|
|
&cFetchedByPath)) == S_FALSE )
|
|
)
|
|
{
|
|
pPathvar += cFetchedByPath;
|
|
cRequested -= cFetchedByPath;
|
|
cTotalFetched += cFetchedByPath;
|
|
|
|
cFetchedByPath = 0;
|
|
|
|
if ( FAILED(_pObjList->Next()) )
|
|
{
|
|
if ( pcElementFetched )
|
|
*pcElementFetched = cTotalFetched;
|
|
RRETURN(S_FALSE);
|
|
}
|
|
|
|
_dwCurrentEntry = 0;
|
|
}
|
|
|
|
if ( pcElementFetched )
|
|
*pcElementFetched = cTotalFetched + cFetchedByPath;
|
|
|
|
RRETURN_EXP_IF_ERR(hr);
|
|
}
|
|
|
|
HRESULT
|
|
CLDAPSchemaEnum::EnumClasses(
|
|
ULONG cElements,
|
|
VARIANT FAR* pvar,
|
|
ULONG FAR* pcElementFetched
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
DWORD i = 0;
|
|
IDispatch *pDispatch = NULL;
|
|
|
|
while ( i < cElements )
|
|
{
|
|
hr = GetClassObject(&pDispatch);
|
|
if ( hr == S_FALSE )
|
|
break;
|
|
|
|
VariantInit( &pvar[i] );
|
|
pvar[i].vt = VT_DISPATCH;
|
|
pvar[i].pdispVal = pDispatch;
|
|
(*pcElementFetched)++;
|
|
i++;
|
|
}
|
|
|
|
RRETURN(hr);
|
|
}
|
|
|
|
HRESULT
|
|
CLDAPSchemaEnum::GetClassObject(
|
|
IDispatch ** ppDispatch
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
CLASSINFO *pClassInfo = NULL;
|
|
|
|
//
|
|
// Now send back the current object
|
|
//
|
|
if ( _dwCurrentEntry >= _nNumOfClasses )
|
|
goto error;
|
|
|
|
hr = SchemaGetClassInfoByIndex(
|
|
_hSchema,
|
|
_dwCurrentEntry,
|
|
&pClassInfo );
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
if (_Credentials.GetAuthFlags() & ADS_AUTH_RESERVED) {
|
|
|
|
//
|
|
// If we are in Umi land, then we cannot ask for dispatch.
|
|
//
|
|
hr = CLDAPClass::CreateClass(
|
|
_bstrADsPath,
|
|
_hSchema,
|
|
pClassInfo->pszName,
|
|
pClassInfo,
|
|
_Credentials,
|
|
ADS_OBJECT_BOUND,
|
|
IID_IUnknown,
|
|
(void **) ppDispatch
|
|
);
|
|
}
|
|
else {
|
|
|
|
hr = CLDAPClass::CreateClass(
|
|
_bstrADsPath,
|
|
_hSchema,
|
|
pClassInfo->pszName,
|
|
pClassInfo,
|
|
_Credentials,
|
|
ADS_OBJECT_BOUND,
|
|
IID_IDispatch,
|
|
(void **)ppDispatch
|
|
);
|
|
}
|
|
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
_dwCurrentEntry++;
|
|
|
|
|
|
RRETURN(S_OK);
|
|
|
|
error:
|
|
|
|
|
|
*ppDispatch = NULL;
|
|
RRETURN(S_FALSE);
|
|
}
|
|
|
|
HRESULT
|
|
CLDAPSchemaEnum::EnumProperties(
|
|
ULONG cElements,
|
|
VARIANT FAR* pvar,
|
|
ULONG FAR* pcElementFetched
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
DWORD i = 0;
|
|
IDispatch *pDispatch = NULL;
|
|
|
|
while ( i < cElements )
|
|
{
|
|
hr = GetPropertyObject(&pDispatch);
|
|
if ( hr == S_FALSE )
|
|
break;
|
|
|
|
VariantInit( &pvar[i] );
|
|
pvar[i].vt = VT_DISPATCH;
|
|
pvar[i].pdispVal = pDispatch;
|
|
(*pcElementFetched)++;
|
|
i++;
|
|
}
|
|
|
|
RRETURN(hr);
|
|
}
|
|
|
|
HRESULT
|
|
CLDAPSchemaEnum::GetPropertyObject(
|
|
IDispatch ** ppDispatch
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
PROPERTYINFO *pPropertyInfo = NULL;
|
|
|
|
//
|
|
// Now send back the current object
|
|
//
|
|
if ( _dwCurrentEntry >= _nNumOfProperties )
|
|
goto error;
|
|
|
|
hr = SchemaGetPropertyInfoByIndex(
|
|
_hSchema,
|
|
_dwCurrentEntry,
|
|
&pPropertyInfo );
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
if (_Credentials.GetAuthFlags() & ADS_AUTH_RESERVED) {
|
|
//
|
|
// In Umi land ask for IID_IUnknown.
|
|
//
|
|
hr = CLDAPProperty::CreateProperty(
|
|
_bstrADsPath,
|
|
_hSchema,
|
|
pPropertyInfo->pszPropertyName,
|
|
pPropertyInfo,
|
|
_Credentials,
|
|
ADS_OBJECT_BOUND,
|
|
IID_IUnknown,
|
|
(void **)ppDispatch
|
|
);
|
|
|
|
}
|
|
else {
|
|
hr = CLDAPProperty::CreateProperty(
|
|
_bstrADsPath,
|
|
_hSchema,
|
|
pPropertyInfo->pszPropertyName,
|
|
pPropertyInfo,
|
|
_Credentials,
|
|
ADS_OBJECT_BOUND,
|
|
IID_IDispatch,
|
|
(void **)ppDispatch
|
|
);
|
|
}
|
|
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
_dwCurrentEntry++;
|
|
|
|
RRETURN(S_OK);
|
|
|
|
error:
|
|
|
|
*ppDispatch = NULL;
|
|
RRETURN(S_FALSE);
|
|
}
|
|
|
|
HRESULT
|
|
CLDAPSchemaEnum::EnumSyntaxObjects(
|
|
ULONG cElements,
|
|
VARIANT FAR* pvar,
|
|
ULONG FAR* pcElementFetched
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
DWORD i = 0;
|
|
IDispatch *pDispatch = NULL;
|
|
|
|
while ( i < cElements )
|
|
{
|
|
hr = GetSyntaxObject(&pDispatch);
|
|
if ( hr == S_FALSE )
|
|
break;
|
|
|
|
VariantInit( &pvar[i] );
|
|
pvar[i].vt = VT_DISPATCH;
|
|
pvar[i].pdispVal = pDispatch;
|
|
(*pcElementFetched)++;
|
|
i++;
|
|
}
|
|
|
|
RRETURN(hr);
|
|
}
|
|
|
|
HRESULT
|
|
CLDAPSchemaEnum::GetSyntaxObject(
|
|
IDispatch ** ppDispatch
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
//
|
|
// Now send back the current object
|
|
//
|
|
if ( _dwCurrentEntry >= g_cLDAPSyntax )
|
|
goto error;
|
|
|
|
if (_Credentials.GetAuthFlags() & ADS_AUTH_RESERVED) {
|
|
//
|
|
// In Umi land ask for IID_IUnknown not dispatch.
|
|
//
|
|
hr = CLDAPSyntax::CreateSyntax(
|
|
_bstrADsPath,
|
|
&g_aLDAPSyntax[_dwCurrentEntry],
|
|
_Credentials,
|
|
ADS_OBJECT_BOUND,
|
|
IID_IUnknown,
|
|
(void **)ppDispatch
|
|
);
|
|
}
|
|
else {
|
|
hr = CLDAPSyntax::CreateSyntax(
|
|
_bstrADsPath,
|
|
&g_aLDAPSyntax[_dwCurrentEntry],
|
|
_Credentials,
|
|
ADS_OBJECT_BOUND,
|
|
IID_IDispatch,
|
|
(void **)ppDispatch
|
|
);
|
|
}
|
|
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
_dwCurrentEntry++;
|
|
|
|
RRETURN(S_OK);
|
|
|
|
error:
|
|
|
|
*ppDispatch = NULL;
|
|
RRETURN(S_FALSE);
|
|
}
|