|
|
//+---------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1992 - 1996.
//
// File: cprops.cxx
//
// Contents: Property Cache functionality for LDAP
//
// Functions:
// CPropertyCache::addproperty
// CPropertyCache::updateproperty
// CPropertyCache::findproperty
// CPropertyCache::getproperty
// CPropertyCache::putproperty
// CProperyCache::CPropertyCache
// CPropertyCache::~CPropertyCache
// CPropertyCache::createpropertycache
//
// History: 15-Jun-96 yihsins Created.
//
//----------------------------------------------------------------------------
#include "ldap.hxx"
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::addproperty
//
// Synopsis:
//
//
//
// Arguments: [szPropertyName] --
// [vt] --
// [vaData] --
//
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache:: addproperty( LPWSTR szPropertyName ) { HRESULT hr = S_OK; PPROPERTY pNewProperty = NULL; LPWSTR tempString1 = NULL; LPWSTR tempString2 = NULL;
PPROPERTY pNewProperties = NULL;
PDISPPROPERTY pDispNewProperty = NULL; PDISPPROPERTY pDispNewProperties = NULL; DWORD dwDispLoc = 0;
//
// Allocate the string first
//
tempString1 = AllocADsStr(szPropertyName);
if (!tempString1) BAIL_ON_FAILURE(hr=E_OUTOFMEMORY);
//
// Make a copy for the Dispatch Mgr Table.
//
tempString2 = AllocADsStr(szPropertyName);
if (!tempString2) BAIL_ON_FAILURE(hr=E_OUTOFMEMORY);
//
// extend the property cache by adding a new property entry
//
pNewProperties = (PPROPERTY)ReallocADsMem( _pProperties, _cb, _cb + sizeof(PROPERTY) ); if (!pNewProperties) { hr = E_OUTOFMEMORY; BAIL_ON_FAILURE(hr); }
_pProperties = pNewProperties;
pNewProperty = (PPROPERTY)((LPBYTE)_pProperties + _cb);
//
// Since the memory has already been allocated in tempString
// just set the value/pointer now.
//
pNewProperty->szPropertyName = tempString1;
//
// Update the index
//
_dwMaxProperties++; _cb += sizeof(PROPERTY);
//
// extend the property cache by adding a new property entry
//
//
// Need to check if this property is already there in the
// dispatch table - otherwise we are going to keep on growing
// forever - AjayR 7-31-98.
//
hr = DispatchFindProperty(szPropertyName, &dwDispLoc);
if (hr == S_OK) { // we do not need this string in this case
if (tempString2) { FreeADsStr(tempString2); tempString2 = NULL; } } else {
//
// reset the hr otherwise we will return an
// error incorrectly when there was none.
//
hr = S_OK;
pDispNewProperties = (PDISPPROPERTY)ReallocADsMem( _pDispProperties, _cbDisp, _cbDisp + sizeof(DISPPROPERTY) ); if (!pDispNewProperties) { hr = E_OUTOFMEMORY; BAIL_ON_FAILURE(hr); }
_pDispProperties = pDispNewProperties;
pDispNewProperty = (PDISPPROPERTY)((LPBYTE)_pDispProperties + _cbDisp);
//
// Since the memory has already been allocated in tempString
// just set the value/pointer now.
//
pDispNewProperty->szPropertyName = tempString2;
//
// Update the index
//
_dwDispMaxProperties++; _cbDisp += sizeof(DISPPROPERTY);
} // else clause - that is property not found in disp
RRETURN(hr); error:
// if we failed but we already increased the property count and property size, we need to revert them
if(pNewProperties) { _dwMaxProperties--; _cb -= sizeof(PROPERTY); }
if (tempString1){ FreeADsStr(tempString1); }
if (tempString2) { FreeADsStr(tempString2); }
RRETURN(hr); }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::putpropertyext
//
// Synopsis: Similar to put property only unlike update it will add
// the property to the cahce if it is not already there !
//
//
// Arguments: [szPropertyName] --
// [vaData] --
//
// History
//-------------------------------------------------------------------------
HRESULT CPropertyCache:: putpropertyext( LPWSTR szPropertyName, DWORD dwFlags, DWORD dwSyntaxId, LDAPOBJECTARRAY ldapObjectArray ) { HRESULT hr; DWORD dwIndex; BOOL fFound = FALSE; PPROPERTY pThisProperty = NULL;
hr = findproperty( szPropertyName, &dwIndex );
//
// If the property is not in the cache we need to add it
// as updateproperty expects it to be in the cache.
//
if (hr == E_ADS_PROPERTY_NOT_FOUND) { hr = addproperty( szPropertyName ); } else { fFound = TRUE; }
BAIL_ON_FAILURE(hr);
//
// at this time we can call putproperty
//
if (fFound) { hr = putproperty( dwIndex, dwFlags, dwSyntaxId, ldapObjectArray ); } else { hr = putproperty( szPropertyName, dwFlags, dwSyntaxId, ldapObjectArray ); }
error:
RRETURN(hr); }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::updateproperty
//
// Synopsis:
//
//
//
// Arguments: [szPropertyName] --
// [vaData] --
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache:: updateproperty( LPWSTR szPropertyName, DWORD dwSyntaxId, LDAPOBJECTARRAY ldapObjectArray, BOOL fExplicit ) { HRESULT hr; DWORD dwIndex; PPROPERTY pThisProperty = NULL;
hr = findproperty( szPropertyName, &dwIndex ); BAIL_ON_FAILURE(hr);
pThisProperty = _pProperties + dwIndex;
if (!fExplicit) { if ( PROPERTY_FLAGS(pThisProperty) == PROPERTY_UPDATE ) { hr = S_OK; goto error; } }
// Free the old values first
LdapTypeFreeLdapObjects( &(PROPERTY_LDAPOBJECTARRAY(pThisProperty)));
PROPERTY_SYNTAX(pThisProperty) = dwSyntaxId; PROPERTY_FLAGS(pThisProperty) = PROPERTY_INIT;
hr = LdapTypeCopyConstruct( ldapObjectArray, &(PROPERTY_LDAPOBJECTARRAY(pThisProperty)) ); BAIL_ON_FAILURE(hr);
error:
RRETURN(hr); }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::findproperty
//
// Synopsis:
//
//
//
// Arguments: [szPropertyName] --
// [pdwIndex] --
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache:: findproperty( LPWSTR szPropertyName, PDWORD pdwIndex ) { DWORD i = 0; PPROPERTY pThisProperty = NULL;
for (i = 0; i < _dwMaxProperties; i++) {
pThisProperty = _pProperties + i;
if (!_wcsicmp(pThisProperty->szPropertyName, szPropertyName)) { *pdwIndex = i; RRETURN(S_OK); } }
*pdwIndex = 0; RRETURN(E_ADS_PROPERTY_NOT_FOUND);
}
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::getproperty
//
// Synopsis:
//
//
//
// Arguments: [szPropertyName] -- Property to retrieve from the cache
// [pvaData] -- Data returned in a variant
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache:: getproperty( LPWSTR szPropertyName, PDWORD pdwSyntaxId, PDWORD pdwStatusFlag, LDAPOBJECTARRAY *pLdapObjectArray ) { HRESULT hr; DWORD dwIndex = 0L;
//
// retrieve index of property in cache
//
hr = findproperty( szPropertyName, &dwIndex );
//
// if property not already in cache, try get properties from svr
//
//
// INDEX_EMPTY(???) ???
//
if ((hr == E_ADS_PROPERTY_NOT_FOUND || (INDEX_EMPTY(dwIndex) && !PROP_DELETED(dwIndex)) ) && !_fGetInfoDone) { BOOL fResult = FindSavingEntry(szPropertyName);
if(!fResult) { hr = _pCoreADsObject->GetInfo(FALSE);
// workaround to avoid confusing callers of getproperty.
if (hr == E_NOTIMPL) { hr = E_ADS_PROPERTY_NOT_FOUND; } BAIL_ON_FAILURE(hr);
hr = findproperty(szPropertyName, &dwIndex);
} else { hr = E_ADS_PROPERTY_NOT_FOUND; } } BAIL_ON_FAILURE(hr);
//
// get property based on index in cache
//
hr = unboundgetproperty( dwIndex, pdwSyntaxId, pdwStatusFlag, pLdapObjectArray );
error:
RRETURN(hr);
}
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::putproperty
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache:: putproperty( LPWSTR szPropertyName, DWORD dwFlags, DWORD dwSyntaxId, LDAPOBJECTARRAY ldapObjectArray ) { HRESULT hr = S_OK; DWORD dwIndex = 0L;
hr = findproperty(szPropertyName, &dwIndex); if (SUCCEEDED(hr)) hr = putproperty(dwIndex, dwFlags, dwSyntaxId, ldapObjectArray);
RRETURN(hr); }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::putproperty
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache:: putproperty( DWORD dwIndex, DWORD dwFlags, DWORD dwSyntaxId, LDAPOBJECTARRAY ldapObjectArray ) { HRESULT hr = S_OK; PPROPERTY pThisProperty = NULL;
pThisProperty = _pProperties + dwIndex;
// Free the old values first
LdapTypeFreeLdapObjects( &(PROPERTY_LDAPOBJECTARRAY(pThisProperty)) );
PROPERTY_SYNTAX(pThisProperty) = dwSyntaxId;
switch ( dwFlags ) {
case PROPERTY_INIT: if ( ldapObjectArray.dwCount > 0 ) { hr = LdapTypeCopyConstruct( ldapObjectArray, &(PROPERTY_LDAPOBJECTARRAY(pThisProperty)) ); BAIL_ON_FAILURE(hr); }
PROPERTY_FLAGS(pThisProperty) = PROPERTY_INIT; break;
case PROPERTY_DELETE: PROPERTY_FLAGS(pThisProperty) = PROPERTY_DELETE; break;
case PROPERTY_UPDATE: if ( ldapObjectArray.dwCount > 0 ) { hr = LdapTypeCopyConstruct( ldapObjectArray, &(PROPERTY_LDAPOBJECTARRAY(pThisProperty)) ); BAIL_ON_FAILURE(hr); }
PROPERTY_FLAGS(pThisProperty) = ldapObjectArray.dwCount? PROPERTY_UPDATE : PROPERTY_DELETE; break;
case PROPERTY_DELETE_VALUE: if ( ldapObjectArray.dwCount > 0 ) { hr = LdapTypeCopyConstruct( ldapObjectArray, &(PROPERTY_LDAPOBJECTARRAY(pThisProperty)) ); BAIL_ON_FAILURE(hr); }
PROPERTY_FLAGS(pThisProperty) = PROPERTY_DELETE_VALUE;
break;
case PROPERTY_ADD: hr = LdapTypeCopyConstruct( ldapObjectArray, &(PROPERTY_LDAPOBJECTARRAY(pThisProperty)) ); BAIL_ON_FAILURE(hr);
PROPERTY_FLAGS(pThisProperty) = PROPERTY_ADD; break;
}
error:
RRETURN(hr); }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::CPropertyCache
//
// Synopsis:
//
//
//
// Arguments:
//
//
//-------------------------------------------------------------------------
CPropertyCache:: CPropertyCache(): _dwMaxProperties(0), _pProperties(NULL), _cb(0), _dwCurrentIndex(0), _pCoreADsObject(NULL), _pGetAttributeSyntax(NULL), _fGetInfoDone(FALSE), _pDispProperties(NULL), _dwDispMaxProperties(0), _cbDisp(0), _pCredentials(NULL), _pszServerName(NULL), _dwPort(0) { InitializeListHead(&_ListSavingEntries);
}
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::~CPropertyCache
//
// Synopsis:
//
//
//
// Arguments:
//
//
//-------------------------------------------------------------------------
CPropertyCache:: ~CPropertyCache() { PPROPERTY pThisProperty = NULL; PDISPPROPERTY pThisDispProperty = NULL; if (_pProperties) {
for ( DWORD i = 0; i < _dwMaxProperties; i++ ) {
pThisProperty = _pProperties + i;
if (pThisProperty->szPropertyName) { FreeADsStr(pThisProperty->szPropertyName); pThisProperty->szPropertyName = NULL; }
LdapTypeFreeLdapObjects(&(PROPERTY_LDAPOBJECTARRAY(pThisProperty))); }
FreeADsMem(_pProperties); }
if (_pDispProperties) {
for ( DWORD i = 0; i < _dwDispMaxProperties; i++ ) {
pThisDispProperty = _pDispProperties + i;
if (pThisDispProperty->szPropertyName) { FreeADsStr(pThisDispProperty->szPropertyName); pThisDispProperty->szPropertyName = NULL; }
}
FreeADsMem(_pDispProperties); }
if (_pszServerName) { FreeADsStr(_pszServerName); _pszServerName = NULL; }
DeleteSavingEntry();
//
// The property cache is deleted before the object is
// so the object will handle freeing the credentials.
// We just keep a pointer to the credentials.
//
if (_pCredentials) { _pCredentials = NULL; } }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::ClearAllPropertyFlags
//
// Synopsis:
//
//
//
// Arguments:
//
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache::ClearAllPropertyFlags(VOID) { PPROPERTY pThisProperty = NULL;
if (_pProperties) {
for ( DWORD i = 0; i < _dwMaxProperties; i++ ) {
pThisProperty = _pProperties + i; PROPERTY_FLAGS(pThisProperty) = PROPERTY_INIT; } }
RRETURN(S_OK); }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::ClearPropertyFlag
//
// Synopsis:
//
//
//
// Arguments:
//
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache::ClearPropertyFlag( LPWSTR szPropertyName ) { PPROPERTY pThisProperty = NULL; HRESULT hr = S_OK; DWORD dwIndex;
hr = findproperty( szPropertyName, &dwIndex ); BAIL_ON_FAILURE(hr);
pThisProperty = _pProperties + dwIndex; if (PROPERTY_LDAPOBJECTARRAY(pThisProperty).pLdapObjects) {
PROPERTY_FLAGS(pThisProperty) = PROPERTY_INIT; }
error:
RRETURN(hr); }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::ClearMarshalledProperties
//
// Synopsis: Once the properties have been marshalled and
// the set has been done, the properties on the cache are no
// longer valid. This method must be called to keep the property
// cache in a coherrent state.
// The method frees the 'dirty' entries, sets implicit get
// flag. If the dirty entries cannot be cleared, then the
// entire contents are flushed and they will be picked up at the
// next GetInfo call -- AjayR
//
// Arguments: None.
//
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache::ClearMarshalledProperties() { HRESULT hr = S_OK; DWORD dwIndx = 0; DWORD dwCtr = 0; DWORD dwChng = 0; PPROPERTY pNewProperties = NULL; PPROPERTY pThisProperty = NULL; PPROPERTY pNewCurProperty = NULL; DWORD dwNewProps = 0;
//
// Go through properties to see how many have changed
//
for (dwCtr = 0; dwCtr < _dwMaxProperties; dwCtr++ ) { pThisProperty = _pProperties + dwCtr;
if (PROPERTY_FLAGS(pThisProperty) != PROPERTY_INIT) dwChng++; }
if (dwChng == 0) { RRETURN(S_OK); }
//
// Need to remove those entries which were changed
//
dwNewProps = _dwMaxProperties - dwChng;
if (dwNewProps != 0) { pNewProperties = (PPROPERTY) AllocADsMem( dwNewProps * sizeof(PROPERTY) );
if (!pNewProperties) { BAIL_ON_FAILURE(hr = E_OUTOFMEMORY); // if this fails, then we cannot recover
// effectively. What alternative is there ?
// We do not want to flush the cache.
} }
for (dwCtr = 0, dwIndx = 0; dwCtr < _dwMaxProperties; dwCtr++ ) {
pThisProperty = _pProperties + dwCtr;
if (PROPERTY_FLAGS(pThisProperty) != PROPERTY_INIT) { //
// delete the property
//
if (pThisProperty->szPropertyName) { FreeADsStr(pThisProperty->szPropertyName); pThisProperty->szPropertyName = NULL; }
LdapTypeFreeLdapObjects(&(PROPERTY_LDAPOBJECTARRAY(pThisProperty)));
} else {
//
// Sanity Check, should not hit this if Assert preferable
//
if (dwIndx > dwNewProps || dwIndx == dwNewProps) { BAIL_ON_FAILURE(hr = E_FAIL); }
pNewCurProperty = pNewProperties + dwIndx;
pNewCurProperty->szPropertyName = pThisProperty->szPropertyName;
pNewCurProperty->ldapObjectArray = PROPERTY_LDAPOBJECTARRAY(pThisProperty);
pNewCurProperty->dwFlags = pThisProperty->dwFlags;
pNewCurProperty->dwSyntaxId = pThisProperty->dwSyntaxId;
dwIndx++; } } // for, copying the old elements to new buffer
_dwMaxProperties -= dwChng;
_cb = dwNewProps * sizeof(PROPERTY);
if (_pProperties) FreeADsMem(_pProperties);
_pProperties = pNewProperties;
// Need to set this flag to implicitly fetch properties
// the next time somebody asks for a poperty not in the cache.
_fGetInfoDone = FALSE;
RRETURN(S_OK);
error:
if (pNewProperties) { FreeADsMem(pNewProperties); }
RRETURN(hr);
}
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::SetPropertyFlag
//
// Synopsis:
//
//
//
// Arguments:
//
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache::SetPropertyFlag( LPWSTR szPropertyName, DWORD dwFlag ) { PPROPERTY pThisProperty = NULL; HRESULT hr = S_OK; DWORD dwIndex;
hr = findproperty( szPropertyName, &dwIndex ); BAIL_ON_FAILURE(hr);
pThisProperty = _pProperties + dwIndex; if (PROPERTY_LDAPOBJECTARRAY(pThisProperty).pLdapObjects) {
PROPERTY_FLAGS(pThisProperty) = dwFlag; }
error:
RRETURN(hr); }
HRESULT CPropertyCache::GetPropertyFlag( LPWSTR szPropertyName, DWORD* pdwFlag ) { PPROPERTY pThisProperty = NULL; HRESULT hr = S_OK; DWORD dwIndex;
hr = findproperty( szPropertyName, &dwIndex ); BAIL_ON_FAILURE(hr);
pThisProperty = _pProperties + dwIndex;
*pdwFlag = PROPERTY_FLAGS(pThisProperty);
error:
RRETURN(hr); }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::IsPropertyUpdated
//
// Synopsis:
//
//
//
// Arguments: [szPropertyName] --
// [pdwIndex] --
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache:: IsPropertyUpdated( LPWSTR szPropertyName, BOOL *pfUpdated ) { PPROPERTY pThisProperty = NULL; HRESULT hr = S_OK; DWORD dwIndex;
*pfUpdated = FALSE;
hr = findproperty( szPropertyName, &dwIndex ); BAIL_ON_FAILURE(hr);
pThisProperty = _pProperties + dwIndex; if (PROPERTY_LDAPOBJECTARRAY(pThisProperty).pLdapObjects) {
if ( PROPERTY_FLAGS(pThisProperty) == PROPERTY_UPDATE ) *pfUpdated = TRUE; }
error:
RRETURN(hr); }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::createpropertycache
//
// Synopsis:
//
//
//
// Arguments:
//
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache:: createpropertycache( CCoreADsObject *pCoreADsObject, IGetAttributeSyntax *pGetAttributeSyntax, CPropertyCache **ppPropertyCache ) { CPropertyCache FAR * pPropertyCache = NULL;
pPropertyCache = new CPropertyCache();
if (!pPropertyCache) { RRETURN_EXP_IF_ERR(E_FAIL); }
pPropertyCache->_pCoreADsObject = pCoreADsObject; pPropertyCache->_pGetAttributeSyntax = pGetAttributeSyntax; pPropertyCache->_fGetInfoDone = FALSE;
*ppPropertyCache = pPropertyCache;
RRETURN(S_OK); }
HRESULT CPropertyCache::SetObjInformation( CCredentials* pCredentials, LPWSTR pszServerName, DWORD dwPortNo ) { //
// We need the credentials to be valid
//
if (!pCredentials) { ADsAssert(!"InvalidCredentials to prop cache"); } else { _pCredentials = pCredentials; }
//
// This can be NULL, so it is better to allocate and dealloc
// in destructor
//
if (_pszServerName) { FreeADsStr(_pszServerName); _pszServerName = NULL; }
if (pszServerName) { _pszServerName = AllocADsStr(pszServerName); if (_pszServerName == NULL) { RRETURN(E_OUTOFMEMORY); } }
_dwPort = dwPortNo;
RRETURN(S_OK); } //+------------------------------------------------------------------------
//
// Function: CPropertyCache::flushpropertycache
//
// Synopsis: Flushes the property cache of all data.
//
// The name <-> index mappings need to stay until the property cache is
// destructed, because the indexes are also used as the DISPIDs of the
// properties. So this neither deallocates the names nor the array itself.
//
//-------------------------------------------------------------------------
void CPropertyCache:: flushpropertycache() { DWORD i = 0; PPROPERTY pThisProperty = NULL;
if (_pProperties) {
for (i = 0; i < _dwMaxProperties; i++) {
pThisProperty = _pProperties + i;
if (pThisProperty->szPropertyName) { FreeADsStr(pThisProperty->szPropertyName); pThisProperty->szPropertyName = NULL; }
LdapTypeFreeLdapObjects(&(PROPERTY_LDAPOBJECTARRAY(pThisProperty))); PROPERTY_FLAGS(pThisProperty) = PROPERTY_INIT; }
FreeADsMem(_pProperties);
_pProperties = NULL; _dwMaxProperties = 0; _cb = 0;
}
//
// Reset the property cache
//
_dwCurrentIndex = 0; _fGetInfoDone = FALSE; }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::unmarshallproperty
//
// Synopsis:
//
//
//
// Arguments:
//
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache:: unmarshallproperty( LPWSTR szPropertyName, PADSLDP pLdapHandle, LDAPMessage *entry, DWORD dwSyntaxId, BOOL fExplicit, BOOL * pfRangeRetrieval // defaulted to NULL
) { DWORD dwIndex = 0; HRESULT hr = S_OK; LDAPOBJECTARRAY ldapObjectArray; LPWSTR pszTemp = NULL; WCHAR* pszTempPropertyName = NULL; WCHAR* pszToken = NULL; WCHAR* pszPlaceHolder = NULL;
LDAPOBJECTARRAY_INIT(ldapObjectArray);
//
// If arg is valid default value to false.
//
if (pfRangeRetrieval) { *pfRangeRetrieval = FALSE; }
hr = UnMarshallLDAPToLDAPSynID( szPropertyName, pLdapHandle, entry, dwSyntaxId, &ldapObjectArray );
if(FAILED(hr)) { return hr; }
//
// Need to look for ; when subtype of attribute is used
//
if ((pszTemp = wcschr(szPropertyName, L';')) != NULL ) { // we don't want to break the existing behavior of range retrieval, so keep it
// as a special case. For the others, we will not strip whatever is after ;
pszTempPropertyName= new WCHAR[wcslen(szPropertyName) + 1]; if(!pszTempPropertyName) { BAIL_ON_FAILURE(hr = E_OUTOFMEMORY); }
wcsncpy(pszTempPropertyName, szPropertyName, pszTemp-szPropertyName); pszTempPropertyName[pszTemp-szPropertyName] = '\0';
pszToken = wcstok( pszTemp+1, L";" ); while( pszToken != NULL ) { if(_wcsnicmp(pszToken, L"Range=", wcslen(L"Range="))) { wcscat(pszTempPropertyName, L";"); wcscat(pszTempPropertyName, pszToken); }
pszPlaceHolder= pszToken - 1;
pszToken = wcstok( NULL, L";" ); *pszPlaceHolder = L';'; } } //
// Find this property in the cache
//
hr = findproperty( pszTempPropertyName? pszTempPropertyName : szPropertyName, &dwIndex );
//
// If this property does not exist in the
// cache, add this property into the cache.
//
if (FAILED(hr)) { hr = addproperty( pszTempPropertyName? pszTempPropertyName : szPropertyName );
//
// If the operation fails for some reason
// move on to the next property
//
BAIL_ON_FAILURE(hr);
}
//
// Now update the property in the cache
//
hr = updateproperty( pszTempPropertyName? pszTempPropertyName : szPropertyName, dwSyntaxId, ldapObjectArray, fExplicit ); BAIL_ON_FAILURE(hr); //
// Put the ; back if we replaced it.
//
if (pszTemp) {
//
// Do we need to update the flag ?
//
if (pfRangeRetrieval) { //
// See if this was members and update flag.
//
if (!_wcsicmp(L"member", pszTempPropertyName? pszTempPropertyName : szPropertyName)) { *pfRangeRetrieval = TRUE; } }
}
error:
if ( ldapObjectArray.fIsString ) LdapValueFree( (TCHAR **) ldapObjectArray.pLdapObjects ); else LdapValueFreeLen( (struct berval **) ldapObjectArray.pLdapObjects );
if(pszTempPropertyName) delete [] pszTempPropertyName;
RRETURN_EXP_IF_ERR(hr);
}
HRESULT CPropertyCache:: LDAPUnMarshallProperties( LPWSTR pszServerPath, PADSLDP pLdapHandle, LDAPMessage *ldapmsg, BOOL fExplicit, CCredentials& Credentials ) { int nNumberOfEntries = 0L; int nNumberOfValues = 0L; HRESULT hr = S_OK; DWORD i = 0; LDAPMessage *entry; LPWSTR pszAttrName = NULL; BSTR bstrADsPath = NULL; void *ptr;
OBJECTINFO ObjectInfo; POBJECTINFO pObjectInfo = &ObjectInfo;
memset(pObjectInfo, 0, sizeof(OBJECTINFO));
//
// Compute the number of attributes in the
// read buffer.
nNumberOfEntries = LdapCountEntries( pLdapHandle, ldapmsg );
if ( nNumberOfEntries == 0 ) RRETURN(S_OK);
hr = _pCoreADsObject->get_CoreADsPath(&bstrADsPath); BAIL_ON_FAILURE(hr);
hr = ADsObject(bstrADsPath, pObjectInfo); BAIL_ON_FAILURE(hr);
hr = LdapFirstEntry( pLdapHandle, ldapmsg, &entry );
BAIL_ON_FAILURE(hr);
hr = LdapFirstAttribute( pLdapHandle, entry, &ptr, &pszAttrName ); BAIL_ON_FAILURE(hr);
while ( pszAttrName != NULL ) { DWORD dwSyntax = LDAPTYPE_UNKNOWN;
//
// unmarshall this property into the
// property cache.
// LdapGetSyntax takes care of ; while looking up
// the schema no need to handle at this level.
//
hr = LdapGetSyntaxOfAttributeOnServer( pszServerPath, pszAttrName, &dwSyntax, Credentials, pObjectInfo->PortNumber, TRUE // fForce
);
if ( SUCCEEDED(hr) && (dwSyntax != LDAPTYPE_UNKNOWN)) { if ( (!_wcsicmp(pszAttrName, L"ntSecurityDescriptor")) && (dwSyntax == LDAPTYPE_OCTETSTRING) ) { dwSyntax = LDAPTYPE_SECURITY_DESCRIPTOR; }
(VOID) unmarshallproperty( pszAttrName, pLdapHandle, entry, dwSyntax, fExplicit ); }
LdapAttributeFree( pszAttrName ); pszAttrName = NULL;
//
// If we cannot find the syntax, ignore the property and
// continue with the next property
//
hr = S_OK;
hr = LdapNextAttribute( pLdapHandle, entry, ptr, &pszAttrName );
BAIL_ON_FAILURE(hr);
FreeObjectInfo(pObjectInfo);
memset(pObjectInfo, 0, sizeof(OBJECTINFO));
}
error:
if(bstrADsPath) { ADsFreeString(bstrADsPath); } FreeObjectInfo(pObjectInfo);
RRETURN_EXP_IF_ERR(hr);
}
////////////////////////////////////////////////////////////////////////
//
// Unmarshall attributes (& their values) in [ldapmsg] into cache.
// Syntaxes of attributes are read from schema on server [pszServerPath].
// If an attribute not in the schema (e.g. not in our default schema used
// in case of schemaless server), the attribute is unmarshalled as ldap
// binary data with type = LDAPTYPE_UNKWNON.
//
// [Credentials]
// - used to access pszServerPath
//
// [pLdapHandle]
// - handle assoc with [ldapmsg]
// - used to retrive attributes and their values from [ldapmsg]
//
// [fExplicit]
// - overwrite value of exiting attributes in cache iff = TRUE
//
// NOTE: This function modified LDAPUnMarshallProperties to allow
// unmarshalling of attributes not in the schema.
//
////////////////////////////////////////////////////////////////////////
HRESULT CPropertyCache:: LDAPUnMarshallProperties2( IN LPWSTR pszServerPath, IN PADSLDP pLdapHandle, IN LDAPMessage *ldapmsg, IN BOOL fExplicit, IN CCredentials& Credentials, OUT BOOL * pfRangeRetrieval ) {
HRESULT hr = S_OK; int nNumberOfEntries = 0L; LDAPMessage *entry = NULL; void *ptr = NULL; LPWSTR pszAttrName = NULL; DWORD dwSyntax = LDAPTYPE_UNKNOWN; BSTR bstrADsPath = NULL; OBJECTINFO ObjectInfo; BOOL fRange = FALSE; memset(&ObjectInfo, 0, sizeof(OBJECTINFO));
ADsAssert(pfRangeRetrieval); *pfRangeRetrieval = FALSE;
if (!pLdapHandle || !ldapmsg) RRETURN(E_ADS_BAD_PARAMETER);
//
// Compute the number of attributes in the read buffer.
//
nNumberOfEntries = LdapCountEntries(pLdapHandle, ldapmsg);
if ( nNumberOfEntries == 0 ) RRETURN(S_OK);
//
// get port number to talk to server on which schema locate ???
//
hr = _pCoreADsObject->get_CoreADsPath(&bstrADsPath); BAIL_ON_FAILURE(hr);
hr = ADsObject(bstrADsPath, &ObjectInfo); BAIL_ON_FAILURE(hr);
//
// Get first entry from ldapmsg first. Should be only one entry.
//
hr = LdapFirstEntry( pLdapHandle, ldapmsg, &entry ); BAIL_ON_FAILURE(hr);
//
// get first attribute's name
//
hr = LdapFirstAttribute( pLdapHandle, entry, &ptr, &pszAttrName ); BAIL_ON_FAILURE(hr);
while ( pszAttrName != NULL ) { //
// get syntax of attribute from schema on sever (may be cached);
// continue to unmarshall the attribute even if it isn't in the
// schema.
//
dwSyntax = LDAPTYPE_UNKNOWN;
(VOID) LdapGetSyntaxOfAttributeOnServer( pszServerPath, pszAttrName, &dwSyntax, Credentials, ObjectInfo.PortNumber, TRUE // fForce
);
//
// There is currently no such syntax as "SecurityDescriptor" on
// server, ADSI will unmarshall "OctetString" security descriptor
// as as ntSecurityDescriptor
//
if ( (!_wcsicmp(pszAttrName, L"ntSecurityDescriptor")) && (dwSyntax == LDAPTYPE_OCTETSTRING) ) { dwSyntax = LDAPTYPE_SECURITY_DESCRIPTOR; }
//
// unmarshall the property into cache, LDAPTYPE_UNWKNOWN
// (dwSyntax) will be unmarshalled as binary data.
//
(VOID) unmarshallproperty( pszAttrName, pLdapHandle, entry, dwSyntax, fExplicit, fRange ? NULL : pfRangeRetrieval );
//
// Small trick to make sure we do not loose the range
// retrieval information for members attribute.
//
fRange = *pfRangeRetrieval;
//
// get next attribute
//
LdapAttributeFree( pszAttrName ); pszAttrName = NULL; hr = LdapNextAttribute( pLdapHandle, entry, ptr, &pszAttrName ); BAIL_ON_FAILURE(hr); }
error:
if (bstrADsPath) ADsFreeString(bstrADsPath);
FreeObjectInfo(&ObjectInfo);
RRETURN_EXP_IF_ERR(hr);
}
////////////////////////////////////////////////////////////////////////
//
// Unmarshall attributes (& their values) in [ldapmsg] into cache.
// Syntaxes of attributes are read from schema on server [pszServerPath].
// If an attribute not in the schema (e.g. not in our default schema used
// in case of schemaless server), the attribute is unmarshalled as dwDefaultSyntaxID
//
// [Credentials]
// - used to access pszServerPath
//
// [pLdapHandle]
// - handle assoc with [ldapmsg]
// - used to retrive attributes and their values from [ldapmsg]
//
// [fExplicit]
// - overwrite value of exiting attributes in cache iff = TRUE
//
// NOTE: This function modified LDAPUnMarshallProperties2 to allow
// unmarshalling of attributes not in the schema with dwDefaultSyntaxID passed in
// It is currently only used by CLDAPRootDSE::GetInfoEx, since most of the attributes on
// RootDSE on not defined in schema
//
////////////////////////////////////////////////////////////////////////
HRESULT CPropertyCache:: LDAPUnMarshallProperties3( LPWSTR pszServerPath, PADSLDP pLdapHandle, LDAPMessage *ldapmsg, BOOL fExplicit, DWORD dwDefaultSyntaxID, CCredentials& Credentials ) { HRESULT hr = S_OK; int nNumberOfEntries = 0L; LDAPMessage *entry = NULL; void *ptr = NULL; LPWSTR pszAttrName = NULL; DWORD dwSyntax = LDAPTYPE_UNKNOWN; BSTR bstrADsPath = NULL; OBJECTINFO ObjectInfo; memset(&ObjectInfo, 0, sizeof(OBJECTINFO));
if (!pLdapHandle || !ldapmsg) RRETURN(E_ADS_BAD_PARAMETER);
//
// Compute the number of attributes in the read buffer.
//
nNumberOfEntries = LdapCountEntries(pLdapHandle, ldapmsg);
if ( nNumberOfEntries == 0 ) RRETURN(S_OK);
//
// get port number to talk to server on which schema locate ???
//
hr = _pCoreADsObject->get_CoreADsPath(&bstrADsPath); BAIL_ON_FAILURE(hr);
hr = ADsObject(bstrADsPath, &ObjectInfo); BAIL_ON_FAILURE(hr);
//
// Get first entry from ldapmsg first. Should be only one entry.
//
hr = LdapFirstEntry( pLdapHandle, ldapmsg, &entry ); BAIL_ON_FAILURE(hr);
//
// get first attribute's name
//
hr = LdapFirstAttribute( pLdapHandle, entry, &ptr, &pszAttrName ); BAIL_ON_FAILURE(hr);
while ( pszAttrName != NULL ) { //
// get syntax of attribute from schema on sever (may be cached);
// continue to unmarshall the attribute even if it isn't in the
// schema.
//
dwSyntax = LDAPTYPE_UNKNOWN;
(VOID) LdapGetSyntaxOfAttributeOnServer( pszServerPath, pszAttrName, &dwSyntax, Credentials, ObjectInfo.PortNumber, TRUE // fForce
);
if(dwSyntax == LDAPTYPE_UNKNOWN) { dwSyntax = dwDefaultSyntaxID; }
//
// unmarshall the property into cache, LDAPTYPE_UNWKNOWN
// (dwSyntax) will be unmarshalled as binary data.
//
(VOID) unmarshallproperty( pszAttrName, pLdapHandle, entry, dwSyntax, fExplicit ); //
// get next attribute
//
LdapAttributeFree( pszAttrName ); pszAttrName = NULL; hr = LdapNextAttribute( pLdapHandle, entry, ptr, &pszAttrName ); BAIL_ON_FAILURE(hr); }
error:
if (bstrADsPath) ADsFreeString(bstrADsPath);
FreeObjectInfo(&ObjectInfo); RRETURN_EXP_IF_ERR(hr);
}
HRESULT CPropertyCache:: LDAPMarshallProperties( LDAPModW ***aMods, PBOOL pfNTSecDes, SECURITY_INFORMATION *pSeInfo ) {
HRESULT hr = S_OK; DWORD i = 0; DWORD j = 0; PPROPERTY pThisProperty = NULL; int dwCount = 0; LDAPModW *aModsBuffer = NULL; LDAPOBJECTARRAY ldapObjectArray; PSECURITY_DESCRIPTOR pSecurityDescriptor = NULL;
BOOL fDaclDefaulted = FALSE; BOOL fSaclDefaulted = FALSE; BOOL fOwnerDefaulted = FALSE; BOOL fGroupDefaulted = FALSE;
PSID pOwnerSid = NULL; PSID pGroupSid = NULL; PACL pDacl = NULL; PACL pSacl = NULL; BOOL DaclPresent = FALSE; BOOL SaclPresent = FALSE;
BOOL fSecDesProp = FALSE;
*pSeInfo = INVALID_SE_VALUE; *pfNTSecDes = FALSE;
for (i = 0; i < _dwMaxProperties ; i++) {
pThisProperty = _pProperties + i;
//
// Bypass any property that has not been
// modified
//
if (PROPERTY_FLAGS(pThisProperty) != PROPERTY_INIT) dwCount++; }
if ( dwCount == 0 ) // Nothing to change
{ *aMods = NULL; RRETURN(S_OK); }
*aMods = (LDAPModW **) AllocADsMem((dwCount+1) * sizeof(LDAPModW *));
if ( *aMods == NULL ) { hr = E_OUTOFMEMORY; BAIL_ON_FAILURE(hr); }
aModsBuffer = (LDAPModW *) AllocADsMem( dwCount * sizeof(LDAPModW));
if ( aModsBuffer == NULL ) { hr = E_OUTOFMEMORY; BAIL_ON_FAILURE(hr); }
for (i = 0, j = 0; i < _dwMaxProperties; i++) {
pThisProperty = _pProperties + i;
//
// Bypass any property that has not been
// modified
//
if (PROPERTY_FLAGS(pThisProperty) == PROPERTY_INIT ) {
continue; }
if (!_wcsicmp(PROPERTY_NAME(pThisProperty),L"ntSecurityDescriptor")) { *pfNTSecDes = TRUE; fSecDesProp = TRUE; } else { fSecDesProp = FALSE; }
ldapObjectArray = PROPERTY_LDAPOBJECTARRAY(pThisProperty);
(*aMods)[j] = &aModsBuffer[j]; aModsBuffer[j].mod_type = PROPERTY_NAME(pThisProperty);
if ( ldapObjectArray.fIsString ) { aModsBuffer[j].mod_values = (TCHAR **) ldapObjectArray.pLdapObjects; } else { aModsBuffer[j].mod_bvalues = (struct berval **) ldapObjectArray.pLdapObjects; aModsBuffer[j].mod_op = LDAP_MOD_BVALUES; if (fSecDesProp && ldapObjectArray.pLdapObjects) { pSecurityDescriptor = LDAPOBJECT_BERVAL_VAL(ldapObjectArray.pLdapObjects);
if ( GetSecurityDescriptorOwner( pSecurityDescriptor, &pOwnerSid, &fOwnerDefaulted ) && GetSecurityDescriptorGroup( pSecurityDescriptor, &pGroupSid, &fGroupDefaulted ) &&
GetSecurityDescriptorDacl( pSecurityDescriptor, &DaclPresent, &pDacl, &fDaclDefaulted ) && GetSecurityDescriptorSacl( pSecurityDescriptor, &SaclPresent, &pSacl, &fSaclDefaulted ) ) { //
// All the calls succeeded, so we should reset to 0
// instead of the invalid value.
//
*pSeInfo = 0;
if (!fOwnerDefaulted) { *pSeInfo = *pSeInfo | OWNER_SECURITY_INFORMATION; }
if (!fGroupDefaulted) { *pSeInfo = *pSeInfo | GROUP_SECURITY_INFORMATION; }
//
// If the DACL is present we need to send DACL bit.
//
if (DaclPresent) { *pSeInfo = *pSeInfo | DACL_SECURITY_INFORMATION; }
//
// If SACL present then we set the SACL bit.
if (SaclPresent) { *pSeInfo = *pSeInfo | SACL_SECURITY_INFORMATION; }
}
} }
switch( PROPERTY_FLAGS(pThisProperty)) { case PROPERTY_UPDATE: aModsBuffer[j].mod_op |= LDAP_MOD_REPLACE; break;
case PROPERTY_ADD: aModsBuffer[j].mod_op |= LDAP_MOD_ADD; break;
case PROPERTY_DELETE: aModsBuffer[j].mod_op |= LDAP_MOD_DELETE; break;
case PROPERTY_DELETE_VALUE: aModsBuffer[j].mod_op |= LDAP_MOD_DELETE; break;
}
j++;
}
RRETURN(hr);
error:
FreeADsMem( aModsBuffer ); FreeADsMem( *aMods ); *aMods = NULL;
RRETURN_EXP_IF_ERR(hr); }
HRESULT CPropertyCache:: LDAPMarshallProperties2( LDAPModW ***aMods, DWORD *pdwNumOfMods ) {
HRESULT hr = S_OK; DWORD i = 0; DWORD j = 0; PPROPERTY pThisProperty = NULL; int dwCount = 0; LDAPModW *aModsBuffer = NULL; LDAPOBJECTARRAY ldapObjectArray;
for (i = 0; i < _dwMaxProperties ; i++) {
pThisProperty = _pProperties + i;
//
// Bypass any property that has not been
// modified
//
if (PROPERTY_FLAGS(pThisProperty) != PROPERTY_INIT) dwCount++; }
if ( dwCount == 0 ) // Nothing to change
RRETURN(S_OK);
if ( *aMods == NULL ) { *aMods = (LDAPModW **) AllocADsMem((dwCount+1) * sizeof(LDAPModW *));
if ( *aMods == NULL ) { hr = E_OUTOFMEMORY; BAIL_ON_FAILURE(hr); }
aModsBuffer = (LDAPModW *) AllocADsMem( dwCount * sizeof(LDAPModW));
if ( aModsBuffer == NULL ) { FreeADsMem( *aMods ); *aMods = NULL; hr = E_OUTOFMEMORY; BAIL_ON_FAILURE(hr); } } else { LDAPModW **aModsTemp = NULL;
aModsTemp = (LDAPModW **) AllocADsMem( (*pdwNumOfMods+ dwCount + 1) * sizeof(LDAPModW *));
if ( aModsTemp == NULL ) { hr = E_OUTOFMEMORY; BAIL_ON_FAILURE(hr); }
aModsBuffer = (LDAPModW *) AllocADsMem( (*pdwNumOfMods + dwCount) * sizeof(LDAPModW));
if ( aModsBuffer == NULL ) { FreeADsMem( aModsTemp ); hr = E_OUTOFMEMORY; BAIL_ON_FAILURE(hr); }
memcpy( aModsBuffer, **aMods, *pdwNumOfMods * sizeof(LDAPModW)); FreeADsMem( **aMods ); FreeADsMem( *aMods );
*aMods = aModsTemp;
for ( j = 0; j < *pdwNumOfMods; j++ ) { (*aMods)[j] = &aModsBuffer[j]; } }
for (i = 0; i < _dwMaxProperties; i++) {
pThisProperty = _pProperties + i;
//
// Bypass any property that has not been
// modified
//
if (PROPERTY_FLAGS(pThisProperty) == PROPERTY_INIT ) {
continue; }
ldapObjectArray = PROPERTY_LDAPOBJECTARRAY(pThisProperty);
(*aMods)[j] = &aModsBuffer[j]; aModsBuffer[j].mod_type = PROPERTY_NAME(pThisProperty);
if ( ldapObjectArray.fIsString ) { aModsBuffer[j].mod_values = (TCHAR **) ldapObjectArray.pLdapObjects; } else { aModsBuffer[j].mod_bvalues = (struct berval **) ldapObjectArray.pLdapObjects; aModsBuffer[j].mod_op = LDAP_MOD_BVALUES; }
switch( PROPERTY_FLAGS(pThisProperty)) { case PROPERTY_UPDATE: aModsBuffer[j].mod_op |= LDAP_MOD_REPLACE; break;
case PROPERTY_ADD: aModsBuffer[j].mod_op |= LDAP_MOD_ADD; break;
case PROPERTY_DELETE: aModsBuffer[j].mod_op |= LDAP_MOD_DELETE; break; }
j++;
}
*pdwNumOfMods += dwCount;
error:
RRETURN_EXP_IF_ERR(hr); }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::unboundgetproperty
//
// Synopsis:
//
//
//
// Arguments: [szPropertyName] -- Property to retrieve from the cache
// [pvaData] -- Data returned in a variant
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache:: unboundgetproperty( LPWSTR szPropertyName, PDWORD pdwSyntaxId, PDWORD pdwStatusFlag, LDAPOBJECTARRAY *pLdapObjectArray ) { HRESULT hr; DWORD dwIndex = 0L;
//
// get index of property in cache
//
hr = findproperty( szPropertyName, &dwIndex ); BAIL_ON_FAILURE(hr);
//
// get property based on index in cache
//
hr = unboundgetproperty( dwIndex, pdwSyntaxId, pdwStatusFlag, pLdapObjectArray );
error:
RRETURN_EXP_IF_ERR(hr); }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::unboundgetproperty
//
// Synopsis: Note that this version takes the index of the element to
// fetch. It also returns the control code of the item in the cache.
//
//
// Arguments: [dwIndex] -- Index of the property to retrieve.
// [pdwSytnaxId] -- SyntaxId of the data.
// [pdwStatusFlag] -- Status of this property in cache.
// [pLdapObjectArray] -- Array of ldapObjects returned.
//
// NOTE: [dwIndex] is invalid -> E_ADS_PROPERTY_NOT_FOUND//E_FAIL ???
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache:: unboundgetproperty( DWORD dwIndex, PDWORD pdwSyntaxId, PDWORD pdwStatusFlag, LDAPOBJECTARRAY *pLdapObjectArray ) { HRESULT hr = S_OK; PPROPERTY pThisProperty = NULL;
if (!index_valid(dwIndex)) RRETURN(E_ADS_PROPERTY_NOT_FOUND);
pThisProperty = _pProperties + dwIndex;
// Status flag has to be valid if we found the property
*pdwStatusFlag = PROPERTY_FLAGS(pThisProperty);
if (PROPERTY_LDAPOBJECTARRAY(pThisProperty).pLdapObjects) {
//
// property has non-empty values
//
*pdwSyntaxId = (DWORD)PROPERTY_SYNTAX(pThisProperty);
hr = LdapTypeCopyConstruct( PROPERTY_LDAPOBJECTARRAY(pThisProperty), pLdapObjectArray ); BAIL_ON_FAILURE(hr);
}else {
//
// property has empty values: E.g. status flag indicate delete
// operation (or empty values allowed on non-ntds ldap server?)
//
pLdapObjectArray->pLdapObjects = NULL; pLdapObjectArray->dwCount = 0; *pdwSyntaxId = LDAPTYPE_UNKNOWN; //hr = E_FAIL;
}
error:
RRETURN(hr); }
void CPropertyCache:: reset_propindex( ) { _dwCurrentIndex = 0;
}
HRESULT CPropertyCache:: skip_propindex( DWORD dwElements ) { DWORD newIndex = _dwCurrentIndex + dwElements;
if (!index_valid()) RRETURN_EXP_IF_ERR(E_FAIL);
//
// - allow current index to go from within range to out of range by 1
// - by 1 since initial state is out of range by 1
//
if ( newIndex > _dwMaxProperties ) RRETURN_EXP_IF_ERR(E_FAIL);
_dwCurrentIndex = newIndex; RRETURN(S_OK);
}
HRESULT CPropertyCache:: get_PropertyCount( PDWORD pdwMaxProperties ) { ADsAssert(pdwMaxProperties); // function private -> use assertion
*pdwMaxProperties = _dwMaxProperties;
RRETURN(S_OK); }
DWORD CPropertyCache:: get_CurrentIndex( ) { return(_dwCurrentIndex); }
LPWSTR CPropertyCache:: get_CurrentPropName( )
{ PPROPERTY pThisProperty = NULL;
if (!index_valid()) return(NULL);
pThisProperty = _pProperties + _dwCurrentIndex;
return(PROPERTY_NAME(pThisProperty)); }
LPWSTR CPropertyCache:: get_PropName( DWORD dwIndex )
{ PPROPERTY pThisProperty = NULL;
if (!index_valid(dwIndex)) return(NULL);
pThisProperty = _pProperties + dwIndex;
return(PROPERTY_NAME(pThisProperty)); }
HRESULT CPropertyCache:: LDAPUnMarshallPropertyAs( LPWSTR pszServerPath, PADSLDP pLdapHandle, LDAPMessage *ldapmsg, LPWSTR szPropertyName, DWORD dwSyntaxId, BOOL fExplicit, CCredentials& Credentials ) { int nNumberOfEntries = 0L; int nNumberOfValues = 0L; HRESULT hr = S_OK; DWORD i = 0; LDAPMessage *entry; LPWSTR pszAttrName = NULL; void *ptr;
OBJECTINFO ObjectInfo; POBJECTINFO pObjectInfo = &ObjectInfo;
memset(pObjectInfo, 0, sizeof(OBJECTINFO));
//
// Compute the number of attributes in the
// read buffer.
//
nNumberOfEntries = LdapCountEntries( pLdapHandle, ldapmsg );
if ( nNumberOfEntries == 0 ) RRETURN(S_OK);
hr = LdapFirstEntry( pLdapHandle, ldapmsg, &entry );
BAIL_ON_FAILURE(hr);
hr = LdapFirstAttribute( pLdapHandle, entry, &ptr, &pszAttrName ); BAIL_ON_FAILURE(hr);
while ( pszAttrName != NULL ) { DWORD dwSyntax = LDAPTYPE_UNKNOWN;
LPWSTR pszADsPath; hr = _pCoreADsObject->get_CoreADsPath(&pszADsPath); BAIL_ON_FAILURE(hr);
hr = ADsObject(pszADsPath, pObjectInfo); BAIL_ON_FAILURE(hr);
//
// unmarshall this property into the
// property cache
//
hr = LdapGetSyntaxOfAttributeOnServer( pszServerPath, pszAttrName, &dwSyntax, Credentials, pObjectInfo->PortNumber ); ADsFreeString(pszADsPath);
if ( SUCCEEDED(hr) && (dwSyntax != LDAPTYPE_UNKNOWN)) { if ( (!_wcsicmp(pszAttrName, L"ntSecurityDescriptor")) && (dwSyntax == LDAPTYPE_OCTETSTRING) ) { dwSyntax = LDAPTYPE_SECURITY_DESCRIPTOR; }
(VOID) unmarshallproperty( pszAttrName, pLdapHandle, entry, dwSyntax, fExplicit ); }else {
if (!_wcsicmp(pszAttrName, szPropertyName)) {
(VOID) unmarshallproperty( pszAttrName, pLdapHandle, entry, dwSyntaxId, fExplicit );
}
}
LdapAttributeFree( pszAttrName ); pszAttrName = NULL;
//
// If we cannot find the syntax, ignore the property and
// continue with the next property
//
hr = S_OK;
hr = LdapNextAttribute( pLdapHandle, entry, ptr, &pszAttrName );
BAIL_ON_FAILURE(hr);
FreeObjectInfo(pObjectInfo);
memset(pObjectInfo, 0, sizeof(OBJECTINFO));
}
error:
FreeObjectInfo(pObjectInfo);
RRETURN_EXP_IF_ERR(hr);
}
HRESULT CPropertyCache:: LDAPUnMarshallPropertiesAs( LPWSTR pszServerPath, PADSLDP pLdapHandle, LDAPMessage *ldapmsg, DWORD dwSyntaxId, BOOL fExplicit, CCredentials& Credentials ) { int nNumberOfEntries = 0L; int nNumberOfValues = 0L; HRESULT hr = S_OK; DWORD i = 0; LDAPMessage *entry; LPWSTR pszAttrName = NULL; void *ptr;
//
// Compute the number of attributes in the
// read buffer.
//
nNumberOfEntries = LdapCountEntries( pLdapHandle, ldapmsg );
if ( nNumberOfEntries == 0 ) RRETURN(S_OK);
hr = LdapFirstEntry( pLdapHandle, ldapmsg, &entry );
BAIL_ON_FAILURE(hr);
hr = LdapFirstAttribute( pLdapHandle, entry, &ptr, &pszAttrName ); BAIL_ON_FAILURE(hr);
while ( pszAttrName != NULL ) { (VOID) unmarshallproperty( pszAttrName, pLdapHandle, entry, dwSyntaxId, fExplicit );
LdapAttributeFree( pszAttrName ); pszAttrName = NULL;
hr = LdapNextAttribute( pLdapHandle, entry, ptr, &pszAttrName );
BAIL_ON_FAILURE(hr); }
error:
RRETURN_EXP_IF_ERR(hr);
}
HRESULT CPropertyCache:: deleteproperty( DWORD dwIndex ) { HRESULT hr = S_OK; PPROPERTY pNewProperties = NULL; PPROPERTY pThisProperty = _pProperties + dwIndex;
if (!index_valid(dwIndex)) { hr = E_FAIL; BAIL_ON_FAILURE(hr); }
if (_dwMaxProperties == 1) { //
// Deleting everything
//
FreeADsStr(pThisProperty->szPropertyName); LdapTypeFreeLdapObjects( &(PROPERTY_LDAPOBJECTARRAY(pThisProperty)) ); FreeADsMem(_pProperties); _pProperties = NULL; _dwMaxProperties = 0; _cb = 0; //
// Need to reset the current index too just in case.
//
_dwCurrentIndex = 0; RRETURN(hr); }
pNewProperties = (PPROPERTY)AllocADsMem( _cb - sizeof(PROPERTY) ); if (!pNewProperties) { hr = E_OUTOFMEMORY; BAIL_ON_FAILURE(hr); }
//
// Copying the memory before the deleted item
//
if (dwIndex != 0) { memcpy( pNewProperties, _pProperties, dwIndex * sizeof(PROPERTY)); }
//
// Copying the memory following the deleted item
//
if (dwIndex != (_dwMaxProperties-1)) { memcpy( pNewProperties + dwIndex, _pProperties + dwIndex + 1, (_dwMaxProperties - dwIndex - 1) * sizeof(PROPERTY)); }
FreeADsStr(pThisProperty->szPropertyName); LdapTypeFreeLdapObjects( &(PROPERTY_LDAPOBJECTARRAY(pThisProperty)) ); FreeADsMem(_pProperties); _pProperties = pNewProperties; _dwMaxProperties--; _cb -= sizeof(PROPERTY);
//
// Reset the current index if necesary so we do not skip a property.
//
if (_dwCurrentIndex > dwIndex) { _dwCurrentIndex--; } error:
RRETURN_EXP_IF_ERR(hr); }
//
// For dynamic dispid's.
//
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::locateproperty
//
// Synopsis: Finds a property in the cache by name.
//
// This differs from findproperty() in return code; this returns no ADSI
// errors, since it's going to a VB interface.
//
// If the property is not found in the cache, this uses the IDirectoryObject
// interface of the containing object to get the attributes of the object.
// If GetObjectAttributes doesn't return any information about the property,
// this method returns DISP_E_UNKNOWNNAME.
//
// Arguments: [szPropertyName] -- Name of the property.
// [pdwIndex] -- Index in the array of properties.
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache::locateproperty( LPWSTR szPropertyName, PDWORD pdwIndex ) { HRESULT hr = findproperty(szPropertyName, pdwIndex);
if (hr == E_ADS_PROPERTY_NOT_FOUND) { DWORD dwLdapSyntaxId = 0; PPROPERTY pProperty = NULL;
hr = _pGetAttributeSyntax->GetAttributeSyntax( szPropertyName, &dwLdapSyntaxId ); BAIL_ON_FAILURE(hr);
hr = addproperty(szPropertyName); BAIL_ON_FAILURE(hr);
hr = findproperty(szPropertyName, pdwIndex); BAIL_ON_FAILURE(hr);
pProperty = _pProperties + *pdwIndex; PROPERTY_SYNTAX(pProperty) = dwLdapSyntaxId; PROPERTY_FLAGS(pProperty) = PROPERTY_INIT; PROPERTY_LDAPOBJECTARRAY(pProperty).dwCount = 0; PROPERTY_LDAPOBJECTARRAY(pProperty).pLdapObjects = NULL; }
error: //
// Automation return code would be DISP_E_UNKNOWNNAME.
// ADSI return code would be E_ADS_PROPERTY_NOT_FOUND (like findproperty.)
//
if (FAILED(hr)) hr = E_ADS_PROPERTY_NOT_FOUND;
RRETURN(hr); }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::getproperty
//
// Synopsis: Get the values of a property from the cache.
// The values are returned as a VARIANT.
//
// Arguments: [dwIndex] -- Index of property to retrieve
// [pVarResult] -- Data returned as a VARIANT
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache::getproperty( DWORD dwIndex, PDWORD dwStatusFlag, VARIANT *pVarResult, CCredentials &Credentials ) { HRESULT hr = S_OK; DWORD dwSyntaxId; LDAPOBJECTARRAY LdapObjectArray;
LDAPOBJECTARRAY_INIT(LdapObjectArray);
//
// This should not return E_ADS_PROPERTY_NOT_FOUND in this case.
// We have an index, which should indicate that we have an array entry.
// If we weren't dealing with VB, I'd be tempted to make this an Assert.
//
if (!index_valid(dwIndex)) BAIL_ON_FAILURE(hr = DISP_E_MEMBERNOTFOUND);
if (INDEX_EMPTY(dwIndex) && !PROP_DELETED(dwIndex) && !_fGetInfoDone) { hr = _pCoreADsObject->GetInfo(FALSE);
// workaround to avoid confusing callers of getproperty.
if (hr == E_NOTIMPL) hr = DISP_E_MEMBERNOTFOUND; BAIL_ON_FAILURE(hr); }
hr = unboundgetproperty( dwIndex, &dwSyntaxId, dwStatusFlag, &LdapObjectArray );
// For backward compatibility
if (!LdapObjectArray.pLdapObjects && SUCCEEDED(hr)) { hr = E_FAIL; }
if (SUCCEEDED(hr)) { if (LdapObjectArray.dwCount == 1) { hr = LdapTypeToVarTypeCopy( NULL, Credentials, LdapObjectArray.pLdapObjects, dwSyntaxId, pVarResult); } else { hr = LdapTypeToVarTypeCopyConstruct( NULL, Credentials, LdapObjectArray, dwSyntaxId, pVarResult); } } else { //
// unboundgetproperty() returns E_FAIL on failure.
// But getproperty() should return E_ADS_PROPERTY_NOT_FOUND.
//
if (hr == E_FAIL) hr = E_ADS_PROPERTY_NOT_FOUND;
//
// A proper Automation return value would be
// hr = DISP_E_MEMBERNOTFOUND;
//
ADsAssert(pVarResult); V_VT(pVarResult) = VT_ERROR; }
error: LdapTypeFreeLdapObjects(&LdapObjectArray);
RRETURN(hr); }
//
//
// This is here just so that we can compile, no one should be
// calling this
//
// - The comment above and within the function is INCORRECT.
// - At present, this function is called and works with known bug.
// This function should NOT be a wrap around of getproperty with
// 3 param. The [dwIndex] in this function should have a DIFFERENT
// meaning than the index in the cache.
// - Please LEAVE this for irenef to fix before beta2.
//
HRESULT CPropertyCache::getproperty( DWORD dwIndex, VARIANT *pVarResult, CCredentials &Credentials ) { HRESULT hr = S_OK; DWORD dwStatus = 0;
// Ideally we need to get rid of this but this would mean that
// we need to change the cdispmgr code and the IPropertyCache
// interface.
hr = getproperty( dwIndex, &dwStatus, pVarResult, Credentials );
if (hr == E_ADS_PROPERTY_NOT_FOUND) { hr = DISP_E_MEMBERNOTFOUND; }
RRETURN(hr); }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::putproperty
//
// Synopsis: Updates a property in the property cache.
// The property is specified by its index in the array of
// properties (which is also its DISPID), and the new value
// is given by a VARIANT.
//
// Arguments: [dwIndex] -- Index of the property.
// [varValue] -- Value of the property.
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache::putproperty( DWORD dwIndex, VARIANT vValue ) { HRESULT hr; LDAPOBJECTARRAY LdapObjectArray; DWORD dwLdapType;
LDAPOBJECTARRAY_INIT(LdapObjectArray);
VARIANT *pvValue = &vValue, *pvArray = NULL; DWORD dwNumValues = 1;
//
// This should not return E_ADS_PROPERTY_NOT_FOUND in this case.
// We have an index, which should indicate that we have an array entry.
// If we weren't dealing with VB, I'd be tempted to make this an Assert.
//
if (!index_valid(dwIndex)) BAIL_ON_FAILURE(hr = DISP_E_MEMBERNOTFOUND);
//
// If we get a safe array of variants, convert it to a regular array
// of variants, so we can pass the first one to GetLdapSyntaxFromVariant.
// We assume that all the elements of the array are of the same type
//
if (V_VT(pvValue) == (VT_VARIANT|VT_ARRAY)) { hr = ConvertSafeArrayToVariantArray(vValue, &pvArray, &dwNumValues); BAIL_ON_FAILURE(hr); pvValue = pvArray; }
//
// - dwLdapType in cache can be LDAPTYPE_UNKNOWN.
// - for consistency btw this function and Put/PutEx, will
// get type from input VARIANT and overwrite existing type in
// cache if any
//
ADsAssert(_pCredentials);
hr = GetLdapSyntaxFromVariant( pvValue, &dwLdapType, _pszServerName, (_pProperties+dwIndex)->szPropertyName, *_pCredentials, _dwPort );
BAIL_ON_FAILURE(hr);
hr = VarTypeToLdapTypeCopyConstruct( _pszServerName, *_pCredentials, dwLdapType, pvValue, dwNumValues, &LdapObjectArray ); BAIL_ON_FAILURE(hr);
hr = putproperty(dwIndex, PROPERTY_UPDATE, dwLdapType, LdapObjectArray); BAIL_ON_FAILURE(hr);
error: // if (hr == E_ADS_CANT_CONVERT_DATATYPE)
// hr = DISP_E_TYPEMISMATCH;
LdapTypeFreeLdapObjects(&LdapObjectArray);
if (pvArray) {
for (DWORD i=0; i < dwNumValues; i++) { VariantClear(pvArray + i); }
FreeADsMem(pvArray); }
RRETURN(hr); }
//
// This method is called by _pCoreADsObject->GetInfo(). It signifies
// exactly that the property cache has been filled with all the data
// the server has, i.e. that a GetInfo() has been done. When this is
// set, further implicit calls to GetInfo are skipped. This flag is
// cleared only by "flushpropertycache".
//
void CPropertyCache::setGetInfoFlag() { _fGetInfoDone = TRUE; }
//+------------------------------------------------------------------------
//
// Function: CPropertyCache::findproperty
//
// Synopsis:
//
//
//
// Arguments: [szPropertyName] --
// [pdwIndex] --
//
//-------------------------------------------------------------------------
HRESULT CPropertyCache:: DispatchFindProperty( LPWSTR szPropertyName, PDWORD pdwIndex ) { DWORD i = 0; PDISPPROPERTY pThisDispProperty = NULL;
for (i = 0; i < _dwDispMaxProperties; i++) {
pThisDispProperty = _pDispProperties + i;
if (!_wcsicmp(pThisDispProperty->szPropertyName, szPropertyName)) { *pdwIndex = i; RRETURN(S_OK); } }
*pdwIndex = 0; RRETURN(E_ADS_PROPERTY_NOT_FOUND);
}
//+---------------------------------------------------------------------------
// Function: CPropertyCache::GetPropertyNames.
//
// Synopsis: Gets a list of the names of the properties in this object.
//
// Arguments: ppUmiPropVals - Contains the return value.
//
// Returns: S_OK, or any appropriate error code.
//
// Modifies: N/A.
//
//----------------------------------------------------------------------------
HRESULT CPropertyCache::GetPropertyNames( UMI_PROPERTY_VALUES **ppUmiPropVals ) { HRESULT hr = S_OK; PUMI_PROPERTY_VALUES pUmiPropVals = NULL; PUMI_PROPERTY pUmiProps = NULL; PPROPERTY pNextProperty = NULL; DWORD dwCtr = 0;
ADsAssert(ppUmiPropVals);
//
// Always have only 1 value.
//
pUmiPropVals = (PUMI_PROPERTY_VALUES) AllocADsMem(sizeof(UMI_PROPERTY_VALUES)); if (!pUmiPropVals) { BAIL_ON_FAILURE(hr = E_OUTOFMEMORY); }
if (!_dwMaxProperties) { //
// No properties, we need to special case this.
//
*ppUmiPropVals = pUmiPropVals; RRETURN(S_OK); }
pUmiProps = (PUMI_PROPERTY) AllocADsMem( _dwMaxProperties * sizeof(UMI_PROPERTY) );
if (!pUmiProps) { BAIL_ON_FAILURE(hr = E_OUTOFMEMORY); }
for(dwCtr = 0; dwCtr < _dwMaxProperties; dwCtr++) { pNextProperty = _pProperties + dwCtr;
pUmiProps[dwCtr].pszPropertyName = (LPWSTR) AllocADsStr(pNextProperty->szPropertyName); if(pUmiProps[dwCtr].pszPropertyName == NULL) { BAIL_ON_FAILURE(hr = E_OUTOFMEMORY); }
pUmiProps[dwCtr].uCount = 0; //
// Put correct operation type
//
hr = ConvertLdapCodeToUmiPropCode( pNextProperty->dwFlags, pUmiProps[dwCtr].uOperationType ); //
// Verify that this is the right thing to do.
//
pUmiProps[dwCtr].uType = UMI_TYPE_NULL; }
pUmiPropVals->uCount = _dwMaxProperties; pUmiPropVals->pPropArray = pUmiProps;
*ppUmiPropVals = pUmiPropVals;
RRETURN(S_OK);
error:
if(pUmiProps != NULL) { for(dwCtr = 0; dwCtr < _dwMaxProperties; dwCtr++) { if(pUmiProps[dwCtr].pszPropertyName != NULL) { FreeADsStr(pUmiProps[dwCtr].pszPropertyName); } } FreeADsMem(pUmiProps); }
if (pUmiProps) { FreeADsMem(pUmiPropVals); }
RRETURN(hr); }
HRESULT CPropertyCache:: AddSavingEntry( LPWSTR propertyName ) { HRESULT hr = S_OK; PSAVINGENTRY tempEntry = NULL; BOOL fResult = FALSE;
fResult = FindSavingEntry(propertyName);
if(!fResult) { tempEntry = new SAVINGENTRY; if(!tempEntry) { hr = E_OUTOFMEMORY; BAIL_ON_FAILURE(hr); } tempEntry->entryData = AllocADsStr(propertyName); if(!(tempEntry->entryData)) { hr = E_OUTOFMEMORY; delete tempEntry; BAIL_ON_FAILURE(hr); } InsertTailList( &_ListSavingEntries, &tempEntry->ListEntry ); } error:
return hr; }
BOOL CPropertyCache:: FindSavingEntry( LPWSTR propertyName ) { PLIST_ENTRY listEntry = NULL; PSAVINGENTRY EntryInfo = NULL; BOOL fResult = FALSE;
listEntry = _ListSavingEntries.Flink; while (listEntry != &_ListSavingEntries) { EntryInfo = CONTAINING_RECORD( listEntry, SAVINGENTRY, ListEntry ); if (!_wcsicmp(EntryInfo->entryData, propertyName)) { fResult = TRUE; break; }
listEntry = listEntry->Flink; }
return fResult;
}
HRESULT CPropertyCache:: DeleteSavingEntry() { PLIST_ENTRY pEntry; PSAVINGENTRY EntryInfo;
while (!IsListEmpty (&_ListSavingEntries)) { pEntry = RemoveHeadList (&_ListSavingEntries); EntryInfo = CONTAINING_RECORD (pEntry, SAVINGENTRY, ListEntry);
if(EntryInfo->entryData) { FreeADsStr(EntryInfo->entryData); EntryInfo->entryData = NULL; }
delete EntryInfo; }
RRETURN(S_OK); }
|