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.
2818 lines
64 KiB
2818 lines
64 KiB
//+---------------------------------------------------------------------------
|
|
//
|
|
// 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 (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);
|
|
}
|
|
|
|
//+------------------------------------------------------------------------
|
|
//
|
|
// 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;
|
|
|
|
LDAPOBJECTARRAY_INIT(ldapObjectArray);
|
|
|
|
//
|
|
// If arg is valid default value to false.
|
|
//
|
|
if (pfRangeRetrieval) {
|
|
*pfRangeRetrieval = FALSE;
|
|
}
|
|
|
|
hr = UnMarshallLDAPToLDAPSynID(
|
|
szPropertyName,
|
|
pLdapHandle,
|
|
entry,
|
|
dwSyntaxId,
|
|
&ldapObjectArray
|
|
);
|
|
|
|
//
|
|
// Need to look for ; as in members;range or value;binary
|
|
// and strip the ; out before adding to cache.
|
|
//
|
|
if ((pszTemp = wcschr(szPropertyName, L';')) != NULL ) {
|
|
*pszTemp = L'\0';
|
|
}
|
|
//
|
|
// Find this property in the cache
|
|
//
|
|
|
|
hr = findproperty(
|
|
szPropertyName,
|
|
&dwIndex
|
|
);
|
|
|
|
//
|
|
// If this property does not exist in the
|
|
// cache, add this property into the cache.
|
|
//
|
|
|
|
|
|
if (FAILED(hr)) {
|
|
hr = addproperty(
|
|
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(
|
|
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", szPropertyName)) {
|
|
*pfRangeRetrieval = TRUE;
|
|
}
|
|
}
|
|
|
|
*pszTemp = L';';
|
|
|
|
}
|
|
|
|
if ( ldapObjectArray.fIsString )
|
|
LdapValueFree( (TCHAR **) ldapObjectArray.pLdapObjects );
|
|
else
|
|
LdapValueFreeLen( (struct berval **) ldapObjectArray.pLdapObjects );
|
|
|
|
error:
|
|
|
|
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;
|
|
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.
|
|
// 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
|
|
);
|
|
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
|
|
);
|
|
}
|
|
|
|
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);
|
|
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// 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;
|
|
LPWSTR pszADsPath = 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(&pszADsPath);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
hr = ADsObject(pszADsPath, &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 (pszADsPath)
|
|
ADsFreeString(pszADsPath);
|
|
|
|
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) {
|
|
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 );
|
|
|
|
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);
|
|
}
|
|
|
|
|
|
|
|
|
|
|