Leaked source code of windows server 2003
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.
 
 
 
 
 
 

464 lines
10 KiB

//---------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1992 - 1996
//
// File: cenumns.cxx
//
// Contents: LDAP Enumerator Code
//
// History:
//----------------------------------------------------------------------------
#include "ldap.hxx"
#pragma hdrstop
DWORD
GetDefaultServer(
DWORD dwPort,
BOOL fVerify,
LPWSTR szDomainDnsName,
LPWSTR szServerName,
BOOL fWriteable
);
DWORD
GetDefaultLdapServer(
LPWSTR Addresses[],
LPDWORD Count,
BOOL Verify,
DWORD dwPort
) ;
//+---------------------------------------------------------------------------
//
// Function: CLDAPNamespaceEnum::Create
//
// Synopsis:
//
// Arguments: [pCollection]
// [ppEnumVariant]
//
// Returns: HRESULT
//
// Modifies:
//
// History: 01-30-95 krishnag Created.
//
//----------------------------------------------------------------------------
HRESULT
CLDAPNamespaceEnum::Create(
CLDAPNamespaceEnum FAR* FAR* ppenumvariant,
VARIANT var,
CCredentials& Credentials,
LPTSTR pszNamespace
)
{
HRESULT hr = S_OK;
CLDAPNamespaceEnum FAR* penumvariant = NULL;
penumvariant = new CLDAPNamespaceEnum();
if (penumvariant == NULL){
hr = E_OUTOFMEMORY;
BAIL_ON_FAILURE(hr);
}
hr = ObjectTypeList::CreateObjectTypeList(
var,
&penumvariant->_pObjList
);
BAIL_ON_FAILURE(hr);
penumvariant->_Credentials = Credentials;
penumvariant->_pszNamespace = AllocADsStr(pszNamespace);
if (!(penumvariant->_pszNamespace)) {
hr = E_OUTOFMEMORY;
BAIL_ON_FAILURE(hr);
}
if( IsGCNamespace(pszNamespace) )
penumvariant->_dwPort = (DWORD) USE_DEFAULT_GC_PORT;
else
penumvariant->_dwPort = (DWORD) USE_DEFAULT_LDAP_PORT;
*ppenumvariant = penumvariant;
RRETURN(hr);
error:
if (penumvariant) {
delete penumvariant;
}
RRETURN_EXP_IF_ERR(hr);
}
//+---------------------------------------------------------------------------
//
// Function: CLDAPNamespaceEnum::CLDAPNamespaceEnum
//
// Synopsis:
//
//
// Arguments:
//
//
// Returns:
//
// Modifies:
//
// History: 01-30-95 krishnag Created.
//
//----------------------------------------------------------------------------
CLDAPNamespaceEnum::CLDAPNamespaceEnum()
{
_dwIndex = 0;
_pObjList = NULL;
}
//+---------------------------------------------------------------------------
//
// Function: CLDAPNamespaceEnum::~CLDAPNamespaceEnum
//
// Synopsis:
//
//
// Arguments:
//
// Returns:
//
// Modifies:
//
// History: 01-30-95 krishnag Created.
//
//----------------------------------------------------------------------------
CLDAPNamespaceEnum::~CLDAPNamespaceEnum()
{
if (_pszNamespace)
FreeADsMem(_pszNamespace);
if ( _pObjList )
delete _pObjList;
}
//+---------------------------------------------------------------------------
//
// Function: CLDAPNamespaceEnum::Next
//
// Synopsis: Returns cElements number of requested ADs 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:
//
//----------------------------------------------------------------------------
STDMETHODIMP
CLDAPNamespaceEnum::Next(
ULONG cElements,
VARIANT FAR* pvar,
ULONG FAR* pcElementFetched
)
{
ULONG cElementFetched = 0;
HRESULT hr = S_OK;
if ( _dwIndex > 0 ) // only your default ds will be returned, one element
{
if (pcElementFetched)
*pcElementFetched = 0;
RRETURN(S_FALSE);
}
hr = EnumObjects(
cElements,
pvar,
&cElementFetched
);
if (pcElementFetched) {
*pcElementFetched = cElementFetched;
}
RRETURN_EXP_IF_ERR(hr);
}
HRESULT
CLDAPNamespaceEnum::EnumObjects(
ULONG cElements,
VARIANT FAR* pvar,
ULONG FAR* pcElementFetched
)
{
HRESULT hr = S_FALSE;
IDispatch *pDispatch = NULL;
DWORD i = 0;
while (i < cElements) {
if ( _dwIndex > 0 ) // only your default ds will be returned,
// i.e. one element only
{
hr = S_FALSE;
break;
}
_dwIndex++;
hr = GetTreeObject(&pDispatch);
if (hr == S_FALSE) {
break;
}
VariantInit(&pvar[i]);
pvar[i].vt = VT_DISPATCH;
pvar[i].pdispVal = pDispatch;
(*pcElementFetched)++;
i++;
}
RRETURN_EXP_IF_ERR(hr);
}
HRESULT
CLDAPNamespaceEnum::GetTreeObject(
IDispatch ** ppDispatch
)
{
DWORD err = NO_ERROR;
HRESULT hr = S_OK;
LPTSTR *aValuesNamingContext = NULL;
LPTSTR *aValuesObjectClass = NULL;
int nCountValues = 0;
TCHAR *pszLDAPPathName = NULL;
TCHAR szADsClassName[64];
WCHAR szDomainName[MAX_PATH];
WCHAR szServerName[MAX_PATH];
TCHAR *pszNewADsPath = NULL;
TCHAR *pszLast = NULL;
LPWSTR pszNamingContext = NULL;
LPWSTR pszNewADsParent = NULL;
LPWSTR pszNewADsCommonName = NULL;
DWORD fVerify = FALSE;
*ppDispatch = NULL;
ADS_LDP *ld = NULL;
BOOL fGCDefaulted = FALSE;
//
// Now send back the current object
//
RetryGetDefaultServer:
if ( err = GetDefaultServer(
_dwPort,
fVerify,
szDomainName,
szServerName,
TRUE) ){
hr = HRESULT_FROM_WIN32(err);
BAIL_ON_FAILURE(hr);
}
//
// Read the naming contexts
//
if (_dwPort == USE_DEFAULT_GC_PORT) {
fGCDefaulted = TRUE;
pszNamingContext = NULL;
} else {
pszNamingContext = TEXT(LDAP_OPATT_DEFAULT_NAMING_CONTEXT);
}
hr = LdapOpenObject2(
szDomainName,
szServerName,
NULL,
&ld,
_Credentials,
_dwPort
);
if (((hr == HRESULT_FROM_WIN32(ERROR_BAD_NETPATH)) ||
(hr == HRESULT_FROM_WIN32(ERROR_DS_SERVER_DOWN)))
&& !fVerify) {
fVerify = TRUE;
goto RetryGetDefaultServer ;
}
BAIL_ON_FAILURE(hr);
if (!fGCDefaulted) {
hr = LdapReadAttributeFast(
ld,
NULL,
pszNamingContext,
&aValuesNamingContext,
&nCountValues
);
BAIL_ON_FAILURE(hr);
if ( nCountValues == 0 ) {
//
// The hr will be modified at the end of the function to S_FALSE
// in case of error
BAIL_ON_FAILURE(hr = E_FAIL);
}
hr = BuildADsPathFromLDAPPath2(
FALSE, //Server is present till DSSnapin Works
_pszNamespace,
szDomainName,
_dwPort,
fGCDefaulted ?
TEXT("") :
aValuesNamingContext[0],
&pszNewADsPath
);
} else {
//
// In this case we want to force it to be GC://yourDomain
// so that all searches will be truly global
//
hr = BuildADsPathFromLDAPPath2(
TRUE, // server is present
_pszNamespace,
szDomainName,
_dwPort,
TEXT(""),
&pszNewADsPath
);
}
BAIL_ON_FAILURE(hr);
// this part is common to both code paths
hr = BuildADsParentPath(
pszNewADsPath,
&pszNewADsParent,
&pszNewADsCommonName
);
BAIL_ON_FAILURE(hr);
nCountValues = 0;
//
// Read the object class of the path if necessary
//
if (!fGCDefaulted) {
hr = LdapReadAttributeFast(
ld,
aValuesNamingContext[0],
L"objectClass",
&aValuesObjectClass,
&nCountValues
);
BAIL_ON_FAILURE(hr);
}
if ( nCountValues == 0 && !fGCDefaulted)
{
// This object exists but does not contain objectClass attribute
// which is required for all DS objects. Hence, ignore the object
// and return error.
hr = E_ADS_BAD_PATHNAME;
BAIL_ON_FAILURE(hr);
}
//
// Create the object
//
if (fGCDefaulted) {
hr = CLDAPGenObject::CreateGenericObject(
pszNewADsParent,
pszNewADsCommonName,
L"top",
_Credentials,
ADS_OBJECT_BOUND,
IID_IUnknown,
(void **) ppDispatch
);
} else {
//
// Send all the classes so we can load all extensions
//
hr = CLDAPGenObject::CreateGenericObject(
pszNewADsParent,
pszNewADsCommonName,
aValuesObjectClass,
nCountValues,
_Credentials,
ADS_OBJECT_BOUND,
IID_IDispatch,
(void **) ppDispatch
);
}
BAIL_ON_FAILURE(hr);
error:
if ( aValuesNamingContext )
LdapValueFree( aValuesNamingContext );
if ( aValuesObjectClass )
LdapValueFree( aValuesObjectClass );
if ( pszNewADsPath )
FreeADsStr( pszNewADsPath );
if ( pszNewADsParent) {
FreeADsStr(pszNewADsParent);
}
if (pszNewADsCommonName) {
FreeADsStr(pszNewADsCommonName);
}
if ( ld ){
LdapCloseObject( ld );
}
RRETURN_ENUM_STATUS(hr);
}