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.
1479 lines
45 KiB
1479 lines
45 KiB
//---------------------------------------------------------------------------
|
|
//
|
|
//
|
|
// Microsoft Windows
|
|
// Copyright (C) Microsoft Corporation, 1992 - 1995
|
|
//
|
|
// File: cdssrch.cxx
|
|
//
|
|
// Contents: Microsoft ADs NDS Provider Generic Object
|
|
//
|
|
//
|
|
// History: 03-02-97 ShankSh Created.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
#include "nds.hxx"
|
|
#pragma hdrstop
|
|
|
|
const int NO_NDS_RESULT_HANDLES = 32;
|
|
|
|
static
|
|
HRESULT
|
|
NdsValueToADsColumn(
|
|
LPWSTR pszColumnName,
|
|
DWORD dwSyntaxId,
|
|
DWORD dwValues,
|
|
LPBYTE lpValue,
|
|
ADS_SEARCH_COLUMN * pColumn
|
|
);
|
|
|
|
static
|
|
HRESULT
|
|
NdsValueToADsColumnAppend(
|
|
DWORD dwSyntaxId,
|
|
DWORD dwValues,
|
|
LPBYTE lpValue,
|
|
ADS_SEARCH_COLUMN * pColumn
|
|
);
|
|
|
|
static
|
|
HRESULT
|
|
NdsAddAttributes(
|
|
IN ADS_SEARCH_HANDLE hSearchHandle,
|
|
OUT HANDLE *phSearchResult
|
|
);
|
|
|
|
//
|
|
// Sets the appropriate search preferences.
|
|
//
|
|
HRESULT
|
|
CNDSGenObject::SetSearchPreference(
|
|
IN PADS_SEARCHPREF_INFO pSearchPrefs,
|
|
IN DWORD dwNumPrefs
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
BOOL fWarning = FALSE;
|
|
DWORD i;
|
|
|
|
if (!pSearchPrefs && dwNumPrefs > 0) {
|
|
RRETURN (E_ADS_BAD_PARAMETER);
|
|
}
|
|
|
|
for (i=0; i<dwNumPrefs; i++) {
|
|
|
|
pSearchPrefs[i].dwStatus = ADS_STATUS_S_OK;
|
|
|
|
switch(pSearchPrefs[i].dwSearchPref) {
|
|
case ADS_SEARCHPREF_ASYNCHRONOUS:
|
|
case ADS_SEARCHPREF_SIZE_LIMIT:
|
|
case ADS_SEARCHPREF_TIME_LIMIT:
|
|
case ADS_SEARCHPREF_TIMEOUT:
|
|
case ADS_SEARCHPREF_PAGESIZE:
|
|
case ADS_SEARCHPREF_PAGED_TIME_LIMIT:
|
|
case ADS_SEARCHPREF_CHASE_REFERRALS:
|
|
//
|
|
// Can't be set
|
|
//
|
|
pSearchPrefs[i].dwStatus = ADS_STATUS_INVALID_SEARCHPREF;
|
|
fWarning = TRUE;
|
|
continue;
|
|
|
|
case ADS_SEARCHPREF_DEREF_ALIASES:
|
|
if (pSearchPrefs[i].vValue.dwType != ADSTYPE_INTEGER) {
|
|
pSearchPrefs[i].dwStatus = ADS_STATUS_INVALID_SEARCHPREFVALUE;
|
|
fWarning = TRUE;
|
|
continue;
|
|
}
|
|
|
|
switch (pSearchPrefs[i].vValue.Integer) {
|
|
case ADS_DEREF_NEVER:
|
|
_SearchPref._fDerefAliases = FALSE;
|
|
break;
|
|
|
|
case ADS_DEREF_ALWAYS:
|
|
_SearchPref._fDerefAliases = TRUE;
|
|
break;
|
|
|
|
default:
|
|
pSearchPrefs[i].dwStatus = ADS_STATUS_INVALID_SEARCHPREFVALUE;
|
|
fWarning = TRUE;
|
|
continue;
|
|
}
|
|
break;
|
|
|
|
case ADS_SEARCHPREF_ATTRIBTYPES_ONLY:
|
|
if (pSearchPrefs[i].vValue.dwType != ADSTYPE_BOOLEAN) {
|
|
pSearchPrefs[i].dwStatus = ADS_STATUS_INVALID_SEARCHPREFVALUE;
|
|
fWarning = TRUE;
|
|
continue;
|
|
}
|
|
_SearchPref._fAttrsOnly = pSearchPrefs[i].vValue.Boolean;
|
|
break;
|
|
|
|
case ADS_SEARCHPREF_SEARCH_SCOPE:
|
|
if (pSearchPrefs[i].vValue.dwType != ADSTYPE_INTEGER) {
|
|
pSearchPrefs[i].dwStatus = ADS_STATUS_INVALID_SEARCHPREFVALUE;
|
|
fWarning = TRUE;
|
|
continue;
|
|
}
|
|
|
|
switch (pSearchPrefs[i].vValue.Integer) {
|
|
case ADS_SCOPE_ONELEVEL:
|
|
_SearchPref._iScope = 0;
|
|
break;
|
|
|
|
case ADS_SCOPE_SUBTREE:
|
|
_SearchPref._iScope = 1;
|
|
break;
|
|
|
|
case ADS_SCOPE_BASE:
|
|
_SearchPref._iScope = 2;
|
|
break;
|
|
|
|
default:
|
|
pSearchPrefs[i].dwStatus = ADS_STATUS_INVALID_SEARCHPREFVALUE;
|
|
fWarning = TRUE;
|
|
continue;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
pSearchPrefs[i].dwStatus = ADS_STATUS_INVALID_SEARCHPREF;
|
|
fWarning = TRUE;
|
|
continue;
|
|
|
|
}
|
|
}
|
|
|
|
RRETURN (fWarning ? S_ADS_ERRORSOCCURRED : S_OK);
|
|
|
|
}
|
|
|
|
|
|
HRESULT
|
|
CNDSGenObject::ExecuteSearch(
|
|
IN LPWSTR pszSearchFilter,
|
|
IN LPWSTR * pAttributeNames,
|
|
IN DWORD dwNumberAttributes,
|
|
OUT PADS_SEARCH_HANDLE phSearchHandle
|
|
)
|
|
{
|
|
PNDS_SEARCHINFO phSearchInfo = NULL;
|
|
LPWSTR pszNDSContext = NULL, szCurrAttr = NULL;
|
|
DWORD dwAttrNamesLen = 0;
|
|
HRESULT hr = S_OK;
|
|
ULONG i, j;
|
|
LPWSTR pszAttrNameBuffer = NULL, *ppszAttrs = NULL;
|
|
|
|
if (!phSearchHandle) {
|
|
RRETURN (E_ADS_BAD_PARAMETER);
|
|
}
|
|
|
|
//
|
|
// Allocate search handle
|
|
//
|
|
phSearchInfo = (PNDS_SEARCHINFO) AllocADsMem(sizeof(NDS_SEARCHINFO));
|
|
if(!phSearchInfo)
|
|
BAIL_ON_FAILURE (hr = E_OUTOFMEMORY);
|
|
|
|
if (pszSearchFilter) {
|
|
phSearchInfo->_pszSearchFilter = AllocADsStr(pszSearchFilter);
|
|
}
|
|
else {
|
|
phSearchInfo->_pszSearchFilter = AllocADsStr(L"(object class=*)");
|
|
}
|
|
if(!(phSearchInfo->_pszSearchFilter))
|
|
BAIL_ON_FAILURE (hr = E_OUTOFMEMORY);
|
|
|
|
|
|
hr = BuildNDSPathFromADsPath(
|
|
_ADsPath,
|
|
&phSearchInfo->_pszBindContext
|
|
);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
hr = AdsNdsGenerateParseTree(
|
|
phSearchInfo->_pszSearchFilter,
|
|
phSearchInfo->_pszBindContext,
|
|
&phSearchInfo->_pQueryNode
|
|
);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
phSearchInfo->_fADsPathPresent = FALSE;
|
|
phSearchInfo->_fADsPathReturned = FALSE;
|
|
|
|
if (dwNumberAttributes == -1) {
|
|
//
|
|
// Specifies returning all attributes
|
|
//
|
|
|
|
phSearchInfo->_ppszAttrs = NULL;
|
|
phSearchInfo->_pszAttrNameBuffer = NULL;
|
|
phSearchInfo->_fADsPathPresent = TRUE;
|
|
}
|
|
else {
|
|
ppszAttrs = (LPWSTR *) AllocADsMem(
|
|
sizeof(LPWSTR) *
|
|
(dwNumberAttributes + 1)
|
|
);
|
|
if (!ppszAttrs)
|
|
BAIL_ON_FAILURE(hr = E_OUTOFMEMORY);
|
|
|
|
for (i = 0; i < dwNumberAttributes; i++)
|
|
dwAttrNamesLen+= (wcslen(pAttributeNames[i]) + 1) * sizeof(WCHAR);
|
|
|
|
pszAttrNameBuffer = (LPWSTR) AllocADsMem(
|
|
dwAttrNamesLen
|
|
);
|
|
if (!pszAttrNameBuffer)
|
|
BAIL_ON_FAILURE(hr = E_OUTOFMEMORY);
|
|
|
|
szCurrAttr = pszAttrNameBuffer;
|
|
for (i = 0, j = 0; i < dwNumberAttributes; i++) {
|
|
wcscpy(szCurrAttr, pAttributeNames[i]);
|
|
ppszAttrs[j] = szCurrAttr;
|
|
szCurrAttr += wcslen(ppszAttrs[j]) + 1;
|
|
|
|
if(_wcsicmp(ppszAttrs[j], L"ADsPath") == 0) {
|
|
//
|
|
// ADsPath need not be sent
|
|
//
|
|
|
|
phSearchInfo->_fADsPathPresent = TRUE;
|
|
}
|
|
else {
|
|
|
|
j++;
|
|
}
|
|
|
|
}
|
|
ppszAttrs[j] = NULL;
|
|
|
|
phSearchInfo->_ppszAttrs = ppszAttrs;
|
|
phSearchInfo->_pszAttrNameBuffer = pszAttrNameBuffer;
|
|
}
|
|
|
|
phSearchInfo->_hConnection = NULL;
|
|
phSearchInfo->_dwIterHandle = NDS_INITIAL_SEARCH;
|
|
phSearchInfo->_pSearchResults = NULL;
|
|
phSearchInfo->_cSearchResults = 0;
|
|
phSearchInfo->_dwCurrResult = 0;
|
|
phSearchInfo->_fResultPrefetched = FALSE;
|
|
phSearchInfo->_fCheckForDuplicates = TRUE;
|
|
phSearchInfo->_dwCurrAttr = 0;
|
|
phSearchInfo->_SearchPref = _SearchPref;
|
|
|
|
*phSearchHandle = phSearchInfo;
|
|
|
|
RRETURN(S_OK);
|
|
|
|
error:
|
|
|
|
if(phSearchInfo) {
|
|
if(phSearchInfo->_pszBindContext)
|
|
FreeADsStr(phSearchInfo->_pszBindContext);
|
|
|
|
if(phSearchInfo->_pszSearchFilter)
|
|
FreeADsStr(phSearchInfo->_pszSearchFilter);
|
|
|
|
if(phSearchInfo->_ppszAttrs)
|
|
FreeADsMem(phSearchInfo->_ppszAttrs);
|
|
|
|
if(phSearchInfo->_pszAttrNameBuffer)
|
|
FreeADsMem(phSearchInfo->_pszAttrNameBuffer);
|
|
|
|
FreeADsMem(phSearchInfo);
|
|
}
|
|
|
|
RRETURN (hr);
|
|
|
|
}
|
|
|
|
|
|
HRESULT
|
|
CNDSGenObject::AbandonSearch(
|
|
IN ADS_SEARCH_HANDLE hSearchHandle
|
|
)
|
|
{
|
|
RRETURN(E_NOTIMPL);
|
|
}
|
|
|
|
HRESULT
|
|
CNDSGenObject::CloseSearchHandle (
|
|
IN ADS_SEARCH_HANDLE hSearchHandle
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
PNDS_SEARCHINFO phSearchInfo = (PNDS_SEARCHINFO) hSearchHandle;
|
|
DWORD dwStatus;
|
|
|
|
if (!phSearchInfo)
|
|
RRETURN (E_ADS_BAD_PARAMETER);
|
|
|
|
if (phSearchInfo->_pQueryNode) {
|
|
dwStatus = NwNdsDeleteQueryTree(phSearchInfo->_pQueryNode);
|
|
if (dwStatus) {
|
|
hr = HRESULT_FROM_WIN32(GetLastError());
|
|
}
|
|
}
|
|
|
|
if (phSearchInfo->_hConnection) {
|
|
dwStatus = NwNdsCloseObject(phSearchInfo->_hConnection);
|
|
if (dwStatus) {
|
|
hr = HRESULT_FROM_WIN32(GetLastError());
|
|
}
|
|
}
|
|
|
|
if(phSearchInfo->_pszBindContext)
|
|
FreeADsStr(phSearchInfo->_pszBindContext);
|
|
|
|
if(phSearchInfo->_pszSearchFilter)
|
|
FreeADsStr(phSearchInfo->_pszSearchFilter);
|
|
|
|
if(phSearchInfo->_ppszAttrs)
|
|
FreeADsMem(phSearchInfo->_ppszAttrs);
|
|
|
|
if(phSearchInfo->_pszAttrNameBuffer)
|
|
FreeADsMem(phSearchInfo->_pszAttrNameBuffer);
|
|
|
|
if (phSearchInfo->_pSearchResults) {
|
|
for (DWORD i=0; i <= phSearchInfo->_dwCurrResult; i++) {
|
|
NwNdsFreeBuffer(phSearchInfo->_pSearchResults[i]._hSearchResult);
|
|
}
|
|
FreeADsMem(phSearchInfo->_pSearchResults);
|
|
}
|
|
|
|
FreeADsMem(phSearchInfo);
|
|
RRETURN (hr);
|
|
}
|
|
|
|
|
|
HRESULT
|
|
CNDSGenObject::GetFirstRow(
|
|
IN ADS_SEARCH_HANDLE hSearchHandle
|
|
)
|
|
{
|
|
RRETURN(E_NOTIMPL);
|
|
}
|
|
|
|
HRESULT
|
|
CNDSGenObject::GetNextRow(
|
|
IN ADS_SEARCH_HANDLE hSearchHandle
|
|
)
|
|
{
|
|
HRESULT hr;
|
|
DWORD dwStatus = NO_ERROR;
|
|
PNDS_SEARCH_RESULT pResult, pNextResult;
|
|
LPNDS_OBJECT_INFO pObject, pNextObject;
|
|
PNDS_SEARCHINFO phSearchInfo = (PNDS_SEARCHINFO) hSearchHandle;
|
|
|
|
if (!phSearchInfo) {
|
|
RRETURN(E_ADS_BAD_PARAMETER);
|
|
}
|
|
|
|
if (phSearchInfo->_fCheckForDuplicates) {
|
|
phSearchInfo->_dwCurrAttr = 0;
|
|
phSearchInfo->_fADsPathReturned = FALSE;
|
|
}
|
|
|
|
if (!phSearchInfo->_hConnection) {
|
|
dwStatus = ADsNwNdsOpenObject(
|
|
phSearchInfo->_pszBindContext,
|
|
_Credentials,
|
|
&phSearchInfo->_hConnection,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
NULL
|
|
);
|
|
if (dwStatus) {
|
|
hr = HRESULT_FROM_WIN32(GetLastError());
|
|
RRETURN (hr);
|
|
}
|
|
}
|
|
|
|
if (phSearchInfo->_pSearchResults) {
|
|
pResult = &(phSearchInfo->_pSearchResults[phSearchInfo->_dwCurrResult]);
|
|
|
|
if (pResult->_pObjects &&
|
|
((pResult->_lObjectCurrent+1) < pResult->_lObjects)) {
|
|
pResult->_lObjectCurrent++;
|
|
RRETURN(S_OK);
|
|
}
|
|
if (pResult->_lObjectCurrent+1 == pResult->_lObjects &&
|
|
phSearchInfo->_fResultPrefetched) {
|
|
phSearchInfo->_dwCurrResult++;
|
|
pNextResult = &(phSearchInfo->_pSearchResults[phSearchInfo->_dwCurrResult]);
|
|
pNextResult->_lObjectCurrent = 0;
|
|
phSearchInfo->_fResultPrefetched = FALSE;
|
|
|
|
if(phSearchInfo->_fCheckForDuplicates) {
|
|
pObject = pResult->_pObjects + pResult->_lObjectCurrent;
|
|
pNextObject = pNextResult->_pObjects + pNextResult->_lObjectCurrent;
|
|
if (!_wcsicmp(pObject->szObjectName, pNextObject->szObjectName)) {
|
|
//
|
|
// Duplicates; Skip one more result
|
|
//
|
|
if (pNextResult->_lObjectCurrent+1 < pNextResult->_lObjects)
|
|
pNextResult->_lObjectCurrent++;
|
|
else
|
|
{
|
|
pNextResult->_lObjectCurrent++;
|
|
RRETURN(S_ADS_NOMORE_ROWS);
|
|
}
|
|
}
|
|
}
|
|
if( pNextResult->_lObjectCurrent >= pNextResult->_lObjects &&
|
|
phSearchInfo->_dwIterHandle == NDS_NO_MORE_ITERATIONS)
|
|
RRETURN(S_ADS_NOMORE_ROWS);
|
|
else
|
|
RRETURN(S_OK);
|
|
}
|
|
else if( pResult->_lObjectCurrent+1 >= pResult->_lObjects &&
|
|
phSearchInfo->_dwIterHandle == NDS_NO_MORE_ITERATIONS)
|
|
{
|
|
// Make sure _lObjectCurrent doesn't exceed _lObjects. If the
|
|
// result set is empty, _lObjectCurrent should stay at -1
|
|
if( ((pResult->_lObjectCurrent+1) == pResult->_lObjects) &&
|
|
(pResult->_lObjectCurrent != -1) )
|
|
pResult->_lObjectCurrent++;
|
|
RRETURN(S_ADS_NOMORE_ROWS);
|
|
}
|
|
}
|
|
|
|
if (!phSearchInfo->_pQueryNode) {
|
|
//
|
|
// querynode not setup yet
|
|
//
|
|
RRETURN (E_FAIL);
|
|
}
|
|
|
|
if(!phSearchInfo->_pSearchResults) {
|
|
//
|
|
// Allocate an array of handles to Search Handles
|
|
//
|
|
phSearchInfo->_pSearchResults = (PNDS_SEARCH_RESULT) AllocADsMem(
|
|
sizeof(NDS_SEARCH_RESULT) *
|
|
NO_NDS_RESULT_HANDLES);
|
|
if(!phSearchInfo->_pSearchResults) {
|
|
hr = E_OUTOFMEMORY;
|
|
goto error;
|
|
}
|
|
phSearchInfo->_dwCurrResult = 0;
|
|
phSearchInfo->_cSearchResults = NO_NDS_RESULT_HANDLES;
|
|
}
|
|
else {
|
|
phSearchInfo->_dwCurrResult++;
|
|
if (phSearchInfo->_dwCurrResult >= phSearchInfo->_cSearchResults) {
|
|
//
|
|
// Need to allocate more memory for handles
|
|
//
|
|
|
|
phSearchInfo->_pSearchResults = (PNDS_SEARCH_RESULT) ReallocADsMem(
|
|
(void *) phSearchInfo->_pSearchResults,
|
|
sizeof(NDS_SEARCH_RESULT) *
|
|
phSearchInfo->_cSearchResults,
|
|
sizeof(NDS_SEARCH_RESULT) *
|
|
(phSearchInfo->_cSearchResults +
|
|
NO_NDS_RESULT_HANDLES)
|
|
);
|
|
if(!phSearchInfo->_pSearchResults) {
|
|
hr = E_OUTOFMEMORY;
|
|
goto error;
|
|
}
|
|
phSearchInfo->_cSearchResults += NO_NDS_RESULT_HANDLES;
|
|
|
|
}
|
|
}
|
|
|
|
pNextResult = &(phSearchInfo->_pSearchResults[phSearchInfo->_dwCurrResult]);
|
|
|
|
hr = NdsAddAttributes(phSearchInfo,
|
|
&pNextResult->_hSearchResult
|
|
);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
pNextResult->_lObjects = 0;
|
|
pNextResult->_pObjects = NULL;
|
|
// Set _lObjectCurrent to -1 so that empty result set is handled correctly.
|
|
// If it is set to 0 and the result set is empty, then a subsequent call
|
|
// to GetNextRow will not return S_ADS_NOMORE_ROWS. Instead, it will try to
|
|
// search again (in the process, possibly allocate memory for
|
|
// _pSearchResults etc.). Also, setting it to -1 ensures that a call to
|
|
// GetColumn returns error if the result set is empty.
|
|
// Setting it to -1 also ensures that if an error occurs during the search,
|
|
// subsequent calls to GetNextRow and GetColumn are handled correctly.
|
|
pNextResult->_lObjectCurrent = -1;
|
|
|
|
dwStatus = NwNdsSearch(
|
|
phSearchInfo->_hConnection,
|
|
_SearchPref._fAttrsOnly ?
|
|
NDS_INFO_NAMES : NDS_INFO_ATTR_NAMES_VALUES,
|
|
_SearchPref._iScope,
|
|
_SearchPref._fDerefAliases,
|
|
phSearchInfo->_pQueryNode,
|
|
&phSearchInfo->_dwIterHandle,
|
|
&pNextResult->_hSearchResult
|
|
);
|
|
if (dwStatus) {
|
|
hr = HRESULT_FROM_WIN32(GetLastError());
|
|
RRETURN (hr);
|
|
}
|
|
|
|
DWORD dwType;
|
|
dwStatus = NwNdsGetObjectListFromBuffer(
|
|
pNextResult->_hSearchResult,
|
|
(DWORD *) (&pNextResult->_lObjects),
|
|
&dwType,
|
|
&pNextResult->_pObjects
|
|
);
|
|
if (dwStatus) {
|
|
dwStatus = GetLastError();
|
|
if (dwStatus == ERROR_NO_DATA)
|
|
RRETURN(S_ADS_NOMORE_ROWS);
|
|
else
|
|
RRETURN (HRESULT_FROM_WIN32(dwStatus));
|
|
}
|
|
|
|
if (pNextResult->_lObjects > 0) {
|
|
pNextResult->_lObjectCurrent = 0;
|
|
if(phSearchInfo->_fCheckForDuplicates && phSearchInfo->_dwCurrResult > 0) {
|
|
pResult = &(phSearchInfo->_pSearchResults[phSearchInfo->_dwCurrResult-1]);
|
|
pObject = pResult->_pObjects + pResult->_lObjectCurrent;
|
|
pNextObject = pNextResult->_pObjects + pNextResult->_lObjectCurrent;
|
|
if (!_wcsicmp(pObject->szObjectName, pNextObject->szObjectName)) {
|
|
//
|
|
// Duplicates; Skip one more result
|
|
//
|
|
pNextResult->_lObjectCurrent++;
|
|
}
|
|
}
|
|
if( pNextResult->_lObjectCurrent >= pNextResult->_lObjects &&
|
|
phSearchInfo->_dwIterHandle == NDS_NO_MORE_ITERATIONS)
|
|
RRETURN(S_ADS_NOMORE_ROWS);
|
|
|
|
RRETURN(S_OK);
|
|
}
|
|
else
|
|
RRETURN(E_FAIL);
|
|
|
|
error:
|
|
RRETURN(hr);
|
|
}
|
|
|
|
HRESULT
|
|
CNDSGenObject::GetPreviousRow(
|
|
IN ADS_SEARCH_HANDLE hSearchHandle
|
|
)
|
|
{
|
|
PNDS_SEARCH_RESULT pResult, pPrevResult;
|
|
LPNDS_OBJECT_INFO pPrevObject, pObject;
|
|
PNDS_SEARCHINFO phSearchInfo = (PNDS_SEARCHINFO) hSearchHandle;
|
|
|
|
if(!phSearchInfo || !phSearchInfo->_pSearchResults)
|
|
RRETURN(E_FAIL);
|
|
|
|
if (phSearchInfo->_fCheckForDuplicates) {
|
|
phSearchInfo->_dwCurrAttr = 0;
|
|
phSearchInfo->_fADsPathReturned = FALSE;
|
|
}
|
|
|
|
pResult = &(phSearchInfo->_pSearchResults[phSearchInfo->_dwCurrResult]);
|
|
|
|
if (pResult->_lObjectCurrent > 0)
|
|
{
|
|
pResult->_lObjectCurrent--;
|
|
|
|
if(phSearchInfo->_fCheckForDuplicates &&
|
|
(phSearchInfo->_dwCurrResult > 0) && (0 == pResult->_lObjectCurrent))
|
|
{
|
|
pPrevResult = &(phSearchInfo->_pSearchResults[
|
|
phSearchInfo->_dwCurrResult - 1]);
|
|
pPrevObject = pPrevResult->_pObjects + pPrevResult->_lObjects - 1;
|
|
pObject = pResult->_pObjects + pResult->_lObjectCurrent;
|
|
|
|
if(!_wcsicmp(pObject->szObjectName, pPrevObject->szObjectName)) {
|
|
// Current row is a duplicate. Go to previous result
|
|
phSearchInfo->_dwCurrResult--;
|
|
pResult = &(phSearchInfo->_pSearchResults[
|
|
phSearchInfo->_dwCurrResult]);
|
|
pResult->_lObjectCurrent = pResult->_lObjects-1;
|
|
phSearchInfo->_fResultPrefetched = TRUE;
|
|
}
|
|
}
|
|
}
|
|
else if (phSearchInfo->_dwCurrResult > 0) {
|
|
phSearchInfo->_dwCurrResult--;
|
|
pResult = &(phSearchInfo->_pSearchResults[phSearchInfo->_dwCurrResult]);
|
|
pResult->_lObjectCurrent = pResult->_lObjects-1;
|
|
phSearchInfo->_fResultPrefetched = TRUE;
|
|
}
|
|
else if(0 == pResult->_lObjectCurrent)
|
|
// we are at the very beginning of the result set
|
|
pResult->_lObjectCurrent--;
|
|
else
|
|
RRETURN(S_ADS_NOMORE_ROWS);
|
|
|
|
RRETURN(S_OK);
|
|
|
|
}
|
|
|
|
|
|
HRESULT
|
|
CNDSGenObject::GetColumn(
|
|
IN ADS_SEARCH_HANDLE hSearchHandle,
|
|
IN LPWSTR pszColumnName,
|
|
OUT PADS_SEARCH_COLUMN pColumn
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
DWORD dwStatus;
|
|
DWORD dwSyntaxId = 0;
|
|
DWORD dwNumValues = 0;
|
|
LPNDS_ATTR_INFO pAttribute;
|
|
PNDS_SEARCH_RESULT pResult, pNextResult;
|
|
LPNDS_OBJECT_INFO pObject, pNextObject;
|
|
DWORD cAttr;
|
|
BOOL fRowAdvanced = FALSE;
|
|
PNDS_SEARCHINFO phSearchInfo = (PNDS_SEARCHINFO) hSearchHandle;
|
|
|
|
if( !pColumn ||
|
|
!phSearchInfo ||
|
|
!phSearchInfo->_pSearchResults )
|
|
RRETURN (E_ADS_BAD_PARAMETER);
|
|
|
|
pColumn->pszAttrName = NULL;
|
|
pColumn->dwADsType = ADSTYPE_INVALID;
|
|
pColumn->pADsValues = NULL;
|
|
pColumn->dwNumValues = 0;
|
|
pColumn->hReserved = NULL;
|
|
|
|
pResult = &(phSearchInfo->_pSearchResults[phSearchInfo->_dwCurrResult]);
|
|
|
|
if( pResult->_lObjectCurrent < 0 )
|
|
RRETURN (E_ADS_BAD_PARAMETER);
|
|
|
|
pObject = pResult->_pObjects + pResult->_lObjectCurrent;
|
|
|
|
pColumn->pszAttrName = AllocADsStr(pszColumnName);
|
|
if (pColumn->pszAttrName)
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
if(!_wcsicmp (pszColumnName, L"ADsPath")) {
|
|
LPWSTR szNDSPath = pObject->szObjectFullName;
|
|
|
|
//
|
|
// Build the ADsPathName
|
|
//
|
|
|
|
WCHAR szTree[MAX_PATH];
|
|
WCHAR szCN[MAX_PATH];
|
|
WCHAR szADsPath[MAX_PATH];
|
|
|
|
// Building the CN and the TreeName
|
|
LPWSTR szCurrent = szNDSPath;
|
|
szCurrent+=2;
|
|
while ((WCHAR)(*szCurrent) != (WCHAR)'\\')
|
|
szCurrent++;
|
|
wcsncpy( szTree,
|
|
szNDSPath,
|
|
(UINT) (szCurrent-szNDSPath) );
|
|
|
|
// Make the first two characters "//" instead of "\\"
|
|
szTree[0] = (WCHAR)'/';
|
|
szTree[1] = (WCHAR)'/';
|
|
|
|
szTree[szCurrent-szNDSPath] = (WCHAR)'\0';
|
|
szCurrent++;
|
|
wcscpy( szCN,
|
|
szCurrent );
|
|
|
|
// Building the ADsPath
|
|
hr = BuildADsPathFromNDSPath(
|
|
szTree,
|
|
szCN,
|
|
szADsPath
|
|
);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
if(*szADsPath) {
|
|
pColumn->pADsValues = (PADSVALUE) AllocADsMem(sizeof(ADSVALUE));
|
|
if (!pColumn->pADsValues)
|
|
BAIL_ON_FAILURE(hr = E_OUTOFMEMORY);
|
|
|
|
pColumn->dwADsType = ADSTYPE_CASE_IGNORE_STRING;
|
|
pColumn->dwNumValues = 1;
|
|
pColumn->pADsValues[0].dwType = ADSTYPE_CASE_IGNORE_STRING;
|
|
|
|
pColumn->pADsValues[0].CaseIgnoreString = AllocADsStr(szADsPath);
|
|
if (!pColumn->pADsValues[0].CaseIgnoreString)
|
|
BAIL_ON_FAILURE(hr = E_OUTOFMEMORY);
|
|
|
|
pColumn->hReserved = pColumn->pADsValues[0].CaseIgnoreString;
|
|
}
|
|
RRETURN(S_OK);
|
|
}
|
|
|
|
if (phSearchInfo->_SearchPref._fAttrsOnly) {
|
|
//
|
|
// Only Names got. So, don't return any values
|
|
//
|
|
RRETURN (S_OK);
|
|
}
|
|
|
|
pAttribute = (LPNDS_ATTR_INFO)pObject->lpAttribute;
|
|
|
|
for (cAttr=0;cAttr<pObject->dwNumberOfAttributes;cAttr++,pAttribute++) {
|
|
if (_wcsicmp(
|
|
pAttribute->szAttributeName,
|
|
pszColumnName
|
|
) == 0)
|
|
break;
|
|
}
|
|
if (cAttr == pObject->dwNumberOfAttributes) {
|
|
if(pResult->_lObjectCurrent+1 != pResult->_lObjects ||
|
|
(phSearchInfo->_dwIterHandle == NDS_NO_MORE_ITERATIONS &&
|
|
!phSearchInfo->_fResultPrefetched)) {
|
|
//
|
|
// No need to look in the next result set;
|
|
//
|
|
BAIL_ON_FAILURE(hr = E_ADS_COLUMN_NOT_SET);
|
|
}
|
|
else {
|
|
//
|
|
// There is a chance that the column may come in the next
|
|
// result set. So, fetch the next set of results.
|
|
//
|
|
phSearchInfo->_fCheckForDuplicates = FALSE;
|
|
hr = GetNextRow(
|
|
phSearchInfo
|
|
);
|
|
phSearchInfo->_fCheckForDuplicates = TRUE;
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
if (hr == S_ADS_NOMORE_ROWS) {
|
|
BAIL_ON_FAILURE(hr = E_ADS_COLUMN_NOT_SET);
|
|
}
|
|
|
|
fRowAdvanced = TRUE;
|
|
|
|
pNextResult = &(phSearchInfo->_pSearchResults[phSearchInfo->_dwCurrResult]);
|
|
pNextObject = pNextResult->_pObjects + pNextResult->_lObjectCurrent;
|
|
if (_wcsicmp(pObject->szObjectName, pNextObject->szObjectName)) {
|
|
//
|
|
// No need to look in the next object;
|
|
//
|
|
BAIL_ON_FAILURE(hr = E_ADS_COLUMN_NOT_SET);
|
|
}
|
|
else {
|
|
//
|
|
// Look in the next object
|
|
//
|
|
pAttribute = (LPNDS_ATTR_INFO)pNextObject->lpAttribute;
|
|
for (cAttr=0;cAttr<pNextObject->dwNumberOfAttributes;cAttr++,pAttribute++) {
|
|
if (_wcsicmp(
|
|
pAttribute->szAttributeName,
|
|
pszColumnName
|
|
) == 0)
|
|
break;
|
|
}
|
|
if (cAttr == pNextObject->dwNumberOfAttributes) {
|
|
//
|
|
// Didn't find in the next result set containing the row too
|
|
//
|
|
BAIL_ON_FAILURE(hr = E_ADS_COLUMN_NOT_SET);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
hr = NdsValueToADsColumn(
|
|
pszColumnName,
|
|
pAttribute->dwSyntaxId,
|
|
pAttribute->dwNumberOfValues,
|
|
pAttribute->lpValue,
|
|
pColumn
|
|
);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
//
|
|
// Added in to support the case when one multivalue attribute is split into 2 packets. The
|
|
// following case checks
|
|
// 1) if we haven't advanced the row, if we have advanced already, the whole
|
|
// Attribute will already be completely residing in the second packet
|
|
// 2) the attribute was the last attribute from the last packet, thus
|
|
// the next attribute, (the first attribute of the next row) might be
|
|
// the same.
|
|
//
|
|
if ((!fRowAdvanced) &&
|
|
(cAttr == (pObject->dwNumberOfAttributes - 1))) {
|
|
|
|
//
|
|
// If there is indeed a need to try out an extra packet
|
|
//
|
|
if(pResult->_lObjectCurrent+1 != pResult->_lObjects ||
|
|
(phSearchInfo->_dwIterHandle == NDS_NO_MORE_ITERATIONS &&
|
|
!phSearchInfo->_fResultPrefetched)) {
|
|
//
|
|
// No need to look in the next result set;
|
|
//
|
|
hr = S_OK;
|
|
goto done;
|
|
}
|
|
else {
|
|
//
|
|
// There is a chance that the column may come in the next
|
|
// result set. So, fetch the next set of results.
|
|
//
|
|
phSearchInfo->_fCheckForDuplicates = FALSE;
|
|
hr = GetNextRow(
|
|
phSearchInfo
|
|
);
|
|
phSearchInfo->_fCheckForDuplicates = TRUE;
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
if (hr == S_ADS_NOMORE_ROWS) {
|
|
hr = S_OK;
|
|
goto done;
|
|
}
|
|
|
|
fRowAdvanced = TRUE;
|
|
|
|
pNextResult = &(phSearchInfo->_pSearchResults[phSearchInfo->_dwCurrResult]);
|
|
pNextObject = pNextResult->_pObjects + pNextResult->_lObjectCurrent;
|
|
if (_wcsicmp(pObject->szObjectName, pNextObject->szObjectName)) {
|
|
//
|
|
// No need to look in the next object, since objname is different
|
|
//
|
|
hr = S_OK;
|
|
goto done;
|
|
}
|
|
else {
|
|
//
|
|
// Look in the next object, look for the same attribute
|
|
//
|
|
pAttribute = (LPNDS_ATTR_INFO)pNextObject->lpAttribute;
|
|
for (cAttr=0;cAttr<pNextObject->dwNumberOfAttributes;cAttr++,pAttribute++) {
|
|
if (_wcsicmp(
|
|
pAttribute->szAttributeName,
|
|
pszColumnName
|
|
) == 0)
|
|
break;
|
|
}
|
|
if (cAttr == pNextObject->dwNumberOfAttributes) {
|
|
//
|
|
// Didn't find in the next result set containing the row too
|
|
//
|
|
hr = S_OK;
|
|
goto done;
|
|
}
|
|
}
|
|
}
|
|
|
|
//
|
|
// If found, we'll append it to the last column
|
|
//
|
|
hr = NdsValueToADsColumnAppend(
|
|
pAttribute->dwSyntaxId,
|
|
pAttribute->dwNumberOfValues,
|
|
pAttribute->lpValue,
|
|
pColumn
|
|
);
|
|
BAIL_ON_FAILURE(hr);
|
|
}
|
|
done:
|
|
if (fRowAdvanced) {
|
|
phSearchInfo->_fCheckForDuplicates = FALSE;
|
|
GetPreviousRow(phSearchInfo);
|
|
phSearchInfo->_fCheckForDuplicates = TRUE;
|
|
}
|
|
|
|
RRETURN(S_OK);
|
|
|
|
error:
|
|
|
|
if (fRowAdvanced) {
|
|
phSearchInfo->_fCheckForDuplicates = FALSE;
|
|
GetPreviousRow(phSearchInfo);
|
|
phSearchInfo->_fCheckForDuplicates = TRUE;
|
|
}
|
|
|
|
FreeColumn(pColumn);
|
|
|
|
RRETURN (hr);
|
|
}
|
|
|
|
|
|
|
|
HRESULT
|
|
CNDSGenObject::GetNextColumnName(
|
|
IN ADS_SEARCH_HANDLE hSearchHandle,
|
|
OUT LPWSTR * ppszColumnName
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
LPNDS_ATTR_INFO pAttribute;
|
|
LPNDS_ATTR_INFO pAttributeOld = NULL;
|
|
LPNDS_NAME_ONLY pNameOnlyAttr;
|
|
PNDS_SEARCH_RESULT pResult, pNextResult;
|
|
LPNDS_OBJECT_INFO pObject, pNextObject;
|
|
BOOL fRowAdvanced = FALSE;
|
|
PNDS_SEARCHINFO phSearchInfo = (PNDS_SEARCHINFO) hSearchHandle;
|
|
|
|
if( !phSearchInfo ||
|
|
!phSearchInfo->_pSearchResults ||
|
|
!ppszColumnName)
|
|
RRETURN (E_ADS_BAD_PARAMETER);
|
|
|
|
*ppszColumnName = NULL;
|
|
|
|
pResult = &(phSearchInfo->_pSearchResults[phSearchInfo->_dwCurrResult]);
|
|
|
|
if( pResult->_lObjectCurrent < 0 )
|
|
RRETURN (E_ADS_BAD_PARAMETER);
|
|
|
|
pObject = pResult->_pObjects + pResult->_lObjectCurrent;
|
|
|
|
pNameOnlyAttr = (LPNDS_NAME_ONLY)pObject->lpAttribute +
|
|
phSearchInfo->_dwCurrAttr;
|
|
pAttribute = (LPNDS_ATTR_INFO)pObject->lpAttribute +
|
|
phSearchInfo->_dwCurrAttr;
|
|
|
|
//
|
|
// Get the last attribute's name to test it to avoid getting duplicate
|
|
// column names. This will happen if a multi-value got divided into two
|
|
// packets. In that case, both attribute names would be the same.
|
|
// We are only getting the last attribute if this object has greater than
|
|
// 1 object, or else if this attribute is the first attribute, there would
|
|
// not be a one before
|
|
//
|
|
if (phSearchInfo->_dwCurrAttr > 0) {
|
|
pAttributeOld = pAttribute - 1;
|
|
}
|
|
|
|
if (phSearchInfo->_dwCurrAttr >= pObject->dwNumberOfAttributes) {
|
|
if(pResult->_lObjectCurrent+1 != pResult->_lObjects ||
|
|
(phSearchInfo->_dwIterHandle == NDS_NO_MORE_ITERATIONS &&
|
|
!phSearchInfo->_fResultPrefetched)) {
|
|
//
|
|
// No need to look in the next result set;
|
|
//
|
|
hr = S_ADS_NOMORE_COLUMNS;
|
|
goto error;
|
|
}
|
|
else {
|
|
//
|
|
// There is a chance that the column may come in the next
|
|
// result set. So, fetch the next set of results.
|
|
//
|
|
phSearchInfo->_fCheckForDuplicates = FALSE;
|
|
hr = GetNextRow(
|
|
phSearchInfo
|
|
);
|
|
phSearchInfo->_fCheckForDuplicates = TRUE;
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
if (hr == S_ADS_NOMORE_ROWS) {
|
|
hr = S_ADS_NOMORE_COLUMNS;
|
|
goto error;
|
|
}
|
|
|
|
fRowAdvanced = TRUE;
|
|
|
|
pNextResult = &(phSearchInfo->_pSearchResults[phSearchInfo->_dwCurrResult]);
|
|
pNextObject = pNextResult->_pObjects + pNextResult->_lObjectCurrent;
|
|
if (_wcsicmp(pObject->szObjectName, pNextObject->szObjectName)) {
|
|
//
|
|
// No need to look in the next object;
|
|
//
|
|
hr = S_ADS_NOMORE_COLUMNS;
|
|
goto error;
|
|
}
|
|
else {
|
|
//
|
|
// Look in the next object
|
|
//
|
|
pNameOnlyAttr = (LPNDS_NAME_ONLY)pNextObject->lpAttribute +
|
|
phSearchInfo->_dwCurrAttr -
|
|
pObject->dwNumberOfAttributes;
|
|
pAttribute = (LPNDS_ATTR_INFO)pNextObject->lpAttribute +
|
|
phSearchInfo->_dwCurrAttr -
|
|
pObject->dwNumberOfAttributes;
|
|
//
|
|
// If the new attribute is after the first attribute in the new object,
|
|
// we'll reset AttributeOld to point to the attribute before this.
|
|
// Because the old attribute will be the one before the current one
|
|
// in this case.
|
|
//
|
|
if ((phSearchInfo->_dwCurrAttr - pObject->dwNumberOfAttributes) > 0) {
|
|
pAttributeOld = pAttribute - 1;
|
|
}
|
|
|
|
if (phSearchInfo->_dwCurrAttr >= (pObject->dwNumberOfAttributes +
|
|
pNextObject->dwNumberOfAttributes)) {
|
|
//
|
|
// Didn't find in the next result set
|
|
// containing the row too
|
|
//
|
|
hr = S_ADS_NOMORE_COLUMNS;
|
|
goto error;
|
|
}
|
|
|
|
//
|
|
// If it is a duplicate column, go on to the next one
|
|
//
|
|
if (pAttributeOld) {
|
|
if(wcscmp(pAttribute->szAttributeName,
|
|
pAttributeOld->szAttributeName) == 0) {
|
|
phSearchInfo->_dwCurrAttr++;
|
|
if (phSearchInfo->_dwCurrAttr >= (pObject->dwNumberOfAttributes +
|
|
pNextObject->dwNumberOfAttributes)) {
|
|
//
|
|
// Didn't find in the next result set
|
|
// containing the row too
|
|
//
|
|
hr = S_ADS_NOMORE_COLUMNS;
|
|
goto error;
|
|
}
|
|
pNameOnlyAttr = (LPNDS_NAME_ONLY)pNextObject->lpAttribute +
|
|
phSearchInfo->_dwCurrAttr -
|
|
pObject->dwNumberOfAttributes;
|
|
pAttribute = (LPNDS_ATTR_INFO)pNextObject->lpAttribute +
|
|
phSearchInfo->_dwCurrAttr -
|
|
pObject->dwNumberOfAttributes;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (phSearchInfo->_SearchPref._fAttrsOnly)
|
|
*ppszColumnName = AllocADsStr(
|
|
pNameOnlyAttr->szName
|
|
);
|
|
else
|
|
*ppszColumnName = AllocADsStr(
|
|
pAttribute->szAttributeName
|
|
);
|
|
|
|
phSearchInfo->_dwCurrAttr++;
|
|
|
|
if (fRowAdvanced) {
|
|
phSearchInfo->_fCheckForDuplicates = FALSE;
|
|
GetPreviousRow(phSearchInfo);
|
|
phSearchInfo->_fCheckForDuplicates = TRUE;
|
|
}
|
|
|
|
RRETURN(S_OK);
|
|
|
|
error:
|
|
|
|
if (fRowAdvanced) {
|
|
phSearchInfo->_fCheckForDuplicates = FALSE;
|
|
GetPreviousRow(phSearchInfo);
|
|
phSearchInfo->_fCheckForDuplicates = TRUE;
|
|
}
|
|
|
|
if (*ppszColumnName)
|
|
FreeADsStr(*ppszColumnName);
|
|
|
|
|
|
if (hr == S_ADS_NOMORE_COLUMNS && phSearchInfo->_fADsPathPresent) {
|
|
|
|
//
|
|
// If ADsPath was specified, return it as the last column in the row
|
|
//
|
|
|
|
if (!phSearchInfo->_fADsPathReturned) {
|
|
|
|
*ppszColumnName = AllocADsStr(L"ADsPath");
|
|
phSearchInfo->_fADsPathReturned = TRUE;
|
|
hr = S_OK;
|
|
}
|
|
else {
|
|
|
|
hr = S_ADS_NOMORE_COLUMNS;
|
|
}
|
|
|
|
}
|
|
|
|
RRETURN (hr);
|
|
}
|
|
|
|
|
|
HRESULT
|
|
CNDSGenObject::FreeColumn(
|
|
IN PADS_SEARCH_COLUMN pColumn
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
if(!pColumn)
|
|
RRETURN (E_ADS_BAD_PARAMETER);
|
|
|
|
if (pColumn->pszAttrName)
|
|
FreeADsStr(pColumn->pszAttrName);
|
|
|
|
switch(pColumn->dwADsType) {
|
|
case ADSTYPE_CASE_IGNORE_STRING:
|
|
if ((PNDSOBJECT)pColumn->hReserved) {
|
|
//ADsPath does not have hReserved, it is a direct string allocation
|
|
|
|
if (pColumn->pADsValues) {
|
|
FreeADsStr(pColumn->pADsValues[0].CaseIgnoreString);
|
|
}
|
|
}
|
|
break;
|
|
|
|
default:
|
|
// Nothing to free
|
|
break;
|
|
}
|
|
|
|
if (pColumn->pADsValues)
|
|
FreeADsMem(pColumn->pADsValues);
|
|
|
|
RRETURN(hr);
|
|
}
|
|
|
|
|
|
HRESULT
|
|
NdsValueToADsColumn(
|
|
LPWSTR pszColumnName,
|
|
DWORD dwSyntaxId,
|
|
DWORD dwValues,
|
|
LPBYTE lpValue,
|
|
ADS_SEARCH_COLUMN * pColumn
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
LPASN1_TYPE_1 lpASN1_1;
|
|
LPASN1_TYPE_7 lpASN1_7;
|
|
LPASN1_TYPE_8 lpASN1_8;
|
|
LPASN1_TYPE_9 lpASN1_9;
|
|
LPASN1_TYPE_14 lpASN1_14;
|
|
LPASN1_TYPE_24 lpASN1_24;
|
|
DWORD i, j;
|
|
|
|
if(!pszColumnName || !pColumn)
|
|
RRETURN(E_ADS_BAD_PARAMETER);
|
|
|
|
pColumn->hReserved = NULL;
|
|
pColumn->dwNumValues = dwValues;
|
|
pColumn->pADsValues = (PADSVALUE) AllocADsMem(
|
|
sizeof(ADSVALUE) * dwValues
|
|
);
|
|
if (!pColumn->pADsValues)
|
|
BAIL_ON_FAILURE(hr = E_OUTOFMEMORY);
|
|
|
|
if (dwSyntaxId >= g_cMapNdsTypeToADsType)
|
|
pColumn->dwADsType = ADSTYPE_INVALID;
|
|
else
|
|
pColumn->dwADsType = g_MapNdsTypeToADsType[dwSyntaxId];
|
|
|
|
switch (dwSyntaxId) {
|
|
// WIDE STRING
|
|
case NDS_SYNTAX_ID_1:
|
|
case NDS_SYNTAX_ID_2:
|
|
case NDS_SYNTAX_ID_3:
|
|
case NDS_SYNTAX_ID_4:
|
|
case NDS_SYNTAX_ID_5:
|
|
case NDS_SYNTAX_ID_10:
|
|
case NDS_SYNTAX_ID_11:
|
|
case NDS_SYNTAX_ID_20:
|
|
for (i=0; i < dwValues; i++) {
|
|
lpASN1_1 = (LPASN1_TYPE_1) lpValue + i;
|
|
pColumn->pADsValues[i].dwType = pColumn->dwADsType;
|
|
pColumn->pADsValues[i].CaseIgnoreString = lpASN1_1->DNString;
|
|
}
|
|
break;
|
|
|
|
case NDS_SYNTAX_ID_14 :
|
|
for (i=0; i < dwValues; i++) {
|
|
lpASN1_14 = (LPASN1_TYPE_14) lpValue + i;
|
|
pColumn->pADsValues[i].dwType = pColumn->dwADsType;
|
|
pColumn->pADsValues[i].CaseIgnoreString = lpASN1_14->Address;
|
|
}
|
|
break;
|
|
|
|
// BYTE STREAM
|
|
case NDS_SYNTAX_ID_9:
|
|
for (i=0; i < dwValues; i++) {
|
|
lpASN1_9 = (LPASN1_TYPE_9) lpValue + i;
|
|
pColumn->pADsValues[i].dwType = pColumn->dwADsType;
|
|
pColumn->pADsValues[i].OctetString.dwLength =lpASN1_9->Length;
|
|
pColumn->pADsValues[i].OctetString.lpValue = lpASN1_9->OctetString;
|
|
}
|
|
break;
|
|
|
|
// BOOLEAN
|
|
case NDS_SYNTAX_ID_7:
|
|
for (i=0; i < dwValues; i++) {
|
|
lpASN1_7 = (LPASN1_TYPE_7) lpValue + i;
|
|
pColumn->pADsValues[i].dwType = pColumn->dwADsType;
|
|
pColumn->pADsValues[i].Boolean = lpASN1_7->Boolean;
|
|
}
|
|
break;
|
|
|
|
// INTEGER
|
|
case NDS_SYNTAX_ID_8:
|
|
case NDS_SYNTAX_ID_22:
|
|
case NDS_SYNTAX_ID_27:
|
|
for (i=0; i < dwValues; i++) {
|
|
lpASN1_8 = (LPASN1_TYPE_8) lpValue + i;
|
|
pColumn->pADsValues[i].dwType = pColumn->dwADsType;
|
|
pColumn->pADsValues[i].Integer = lpASN1_8->Integer;
|
|
}
|
|
break;
|
|
|
|
case NDS_SYNTAX_ID_24 :
|
|
for (i=0; i < dwValues; i++) {
|
|
lpASN1_24 = (LPASN1_TYPE_24) lpValue + i;
|
|
pColumn->pADsValues[i].dwType = pColumn->dwADsType;
|
|
hr = ConvertDWORDtoSYSTEMTIME(
|
|
lpASN1_24->Time,
|
|
&(pColumn->pADsValues[i].UTCTime)
|
|
);
|
|
BAIL_ON_FAILURE(hr);
|
|
}
|
|
break;
|
|
|
|
case NDS_SYNTAX_ID_6 :
|
|
case NDS_SYNTAX_ID_12 :
|
|
case NDS_SYNTAX_ID_13 :
|
|
case NDS_SYNTAX_ID_15 :
|
|
case NDS_SYNTAX_ID_16 :
|
|
case NDS_SYNTAX_ID_17 :
|
|
case NDS_SYNTAX_ID_18 :
|
|
case NDS_SYNTAX_ID_19 :
|
|
case NDS_SYNTAX_ID_21 :
|
|
case NDS_SYNTAX_ID_23 :
|
|
case NDS_SYNTAX_ID_25 :
|
|
case NDS_SYNTAX_ID_26 :
|
|
default:
|
|
for (i=0; i < dwValues; i++) {
|
|
pColumn->pADsValues[i].dwType = pColumn->dwADsType;
|
|
pColumn->pADsValues[i].ProviderSpecific.dwLength =0;
|
|
pColumn->pADsValues[i].ProviderSpecific.lpValue = NULL;
|
|
}
|
|
break;
|
|
}
|
|
RRETURN(hr);
|
|
|
|
error:
|
|
|
|
if (pColumn->pADsValues)
|
|
FreeADsMem(pColumn->pADsValues);
|
|
|
|
RRETURN(hr);
|
|
|
|
|
|
}
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Given a column, this function will append more values to the end
|
|
|
|
Return Value:
|
|
|
|
S_OK on success, error code otherwise
|
|
|
|
--*/
|
|
HRESULT
|
|
NdsValueToADsColumnAppend(
|
|
DWORD dwSyntaxId,
|
|
DWORD dwValues,
|
|
LPBYTE lpValue,
|
|
ADS_SEARCH_COLUMN * pColumn
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
LPASN1_TYPE_1 lpASN1_1;
|
|
LPASN1_TYPE_7 lpASN1_7;
|
|
LPASN1_TYPE_8 lpASN1_8;
|
|
LPASN1_TYPE_9 lpASN1_9;
|
|
LPASN1_TYPE_14 lpASN1_14;
|
|
LPASN1_TYPE_24 lpASN1_24;
|
|
DWORD i, j;
|
|
PADSVALUE pADsValuesNew = NULL;
|
|
DWORD dwValuesBase;
|
|
|
|
if(!pColumn)
|
|
RRETURN(E_ADS_BAD_PARAMETER);
|
|
|
|
dwValuesBase = pColumn->dwNumValues;
|
|
|
|
//
|
|
// Allocate memory for new values + old values
|
|
//
|
|
pADsValuesNew = (PADSVALUE) AllocADsMem(
|
|
sizeof(ADSVALUE) * (pColumn->dwNumValues + dwValues)
|
|
);
|
|
if (!pADsValuesNew) {
|
|
BAIL_ON_FAILURE(hr = E_OUTOFMEMORY);
|
|
}
|
|
|
|
//
|
|
// Copy old values into new array, and free the old one
|
|
//
|
|
memcpy(pADsValuesNew,
|
|
pColumn->pADsValues,
|
|
sizeof(ADSVALUE) * dwValuesBase);
|
|
FreeADsMem(pColumn->pADsValues);
|
|
pColumn->pADsValues = pADsValuesNew;
|
|
|
|
switch (dwSyntaxId) {
|
|
// WIDE STRING
|
|
case NDS_SYNTAX_ID_1:
|
|
case NDS_SYNTAX_ID_2:
|
|
case NDS_SYNTAX_ID_3:
|
|
case NDS_SYNTAX_ID_4:
|
|
case NDS_SYNTAX_ID_5:
|
|
case NDS_SYNTAX_ID_10:
|
|
case NDS_SYNTAX_ID_11:
|
|
case NDS_SYNTAX_ID_20:
|
|
for (i=0; i < dwValues; i++) {
|
|
lpASN1_1 = (LPASN1_TYPE_1) lpValue + i;
|
|
pColumn->pADsValues[dwValuesBase+i].dwType = pColumn->dwADsType;
|
|
pColumn->pADsValues[dwValuesBase+i].CaseIgnoreString = lpASN1_1->DNString;
|
|
}
|
|
break;
|
|
|
|
case NDS_SYNTAX_ID_14 :
|
|
for (i=0; i < dwValues; i++) {
|
|
lpASN1_14 = (LPASN1_TYPE_14) lpValue + i;
|
|
pColumn->pADsValues[dwValuesBase+i].dwType = pColumn->dwADsType;
|
|
pColumn->pADsValues[dwValuesBase+i].CaseIgnoreString = lpASN1_14->Address;
|
|
}
|
|
break;
|
|
|
|
// BYTE STREAM
|
|
case NDS_SYNTAX_ID_9:
|
|
for (i=0; i < dwValues; i++) {
|
|
lpASN1_9 = (LPASN1_TYPE_9) lpValue + i;
|
|
pColumn->pADsValues[dwValuesBase+i].dwType = pColumn->dwADsType;
|
|
pColumn->pADsValues[dwValuesBase+i].OctetString.dwLength =lpASN1_9->Length;
|
|
pColumn->pADsValues[dwValuesBase+i].OctetString.lpValue = lpASN1_9->OctetString;
|
|
}
|
|
break;
|
|
|
|
// BOOLEAN
|
|
case NDS_SYNTAX_ID_7:
|
|
for (i=0; i < dwValues; i++) {
|
|
lpASN1_7 = (LPASN1_TYPE_7) lpValue + i;
|
|
pColumn->pADsValues[dwValuesBase+i].dwType = pColumn->dwADsType;
|
|
pColumn->pADsValues[dwValuesBase+i].Boolean = lpASN1_7->Boolean;
|
|
}
|
|
break;
|
|
|
|
// INTEGER
|
|
case NDS_SYNTAX_ID_8:
|
|
case NDS_SYNTAX_ID_22:
|
|
case NDS_SYNTAX_ID_27:
|
|
for (i=0; i < dwValues; i++) {
|
|
lpASN1_8 = (LPASN1_TYPE_8) lpValue + i;
|
|
pColumn->pADsValues[dwValuesBase+i].dwType = pColumn->dwADsType;
|
|
pColumn->pADsValues[dwValuesBase+i].Integer = lpASN1_8->Integer;
|
|
}
|
|
break;
|
|
|
|
case NDS_SYNTAX_ID_24 :
|
|
for (i=0; i < dwValues; i++) {
|
|
lpASN1_24 = (LPASN1_TYPE_24) lpValue + i;
|
|
pColumn->pADsValues[dwValuesBase+i].dwType = pColumn->dwADsType;
|
|
hr = ConvertDWORDtoSYSTEMTIME(
|
|
lpASN1_24->Time,
|
|
&(pColumn->pADsValues[dwValuesBase+i].UTCTime)
|
|
);
|
|
BAIL_ON_FAILURE(hr);
|
|
}
|
|
break;
|
|
|
|
case NDS_SYNTAX_ID_6 :
|
|
case NDS_SYNTAX_ID_12 :
|
|
case NDS_SYNTAX_ID_13 :
|
|
case NDS_SYNTAX_ID_15 :
|
|
case NDS_SYNTAX_ID_16 :
|
|
case NDS_SYNTAX_ID_17 :
|
|
case NDS_SYNTAX_ID_18 :
|
|
case NDS_SYNTAX_ID_19 :
|
|
case NDS_SYNTAX_ID_21 :
|
|
case NDS_SYNTAX_ID_23 :
|
|
case NDS_SYNTAX_ID_25 :
|
|
case NDS_SYNTAX_ID_26 :
|
|
default:
|
|
for (i=0; i < dwValues; i++) {
|
|
pColumn->pADsValues[dwValuesBase+i].dwType = pColumn->dwADsType;
|
|
pColumn->pADsValues[dwValuesBase+i].ProviderSpecific.dwLength =0;
|
|
pColumn->pADsValues[dwValuesBase+i].ProviderSpecific.lpValue = NULL;
|
|
}
|
|
break;
|
|
}
|
|
pColumn->dwNumValues = pColumn->dwNumValues + dwValues;
|
|
|
|
error:
|
|
//
|
|
// We don't need to free memory even in error case because it has been
|
|
// put into the column
|
|
//
|
|
RRETURN(hr);
|
|
}
|
|
|
|
|
|
|
|
|
|
HRESULT
|
|
NdsAddAttributes(
|
|
IN ADS_SEARCH_HANDLE hSearchHandle,
|
|
OUT HANDLE *phSearchResult
|
|
)
|
|
{
|
|
DWORD dwStatus = NO_ERROR;
|
|
HANDLE hSearchResult = NULL;
|
|
HRESULT hr;
|
|
DWORD i;
|
|
PNDS_SEARCHINFO phSearchInfo = (PNDS_SEARCHINFO) hSearchHandle;
|
|
|
|
if (!phSearchResult || !phSearchResult)
|
|
RRETURN( E_FAIL );
|
|
|
|
*phSearchResult = NULL;
|
|
|
|
if(!phSearchInfo->_ppszAttrs)
|
|
RRETURN(S_OK);
|
|
|
|
dwStatus = NwNdsCreateBuffer( NDS_SEARCH,
|
|
&hSearchResult );
|
|
|
|
if (dwStatus) {
|
|
hr = HRESULT_FROM_WIN32(GetLastError());
|
|
goto error;
|
|
}
|
|
|
|
for (i=0; phSearchInfo->_ppszAttrs[i]; i++) {
|
|
dwStatus = NwNdsPutInBuffer( phSearchInfo->_ppszAttrs[i],
|
|
0,
|
|
NULL,
|
|
0,
|
|
0,
|
|
hSearchResult );
|
|
if (dwStatus) {
|
|
hr = HRESULT_FROM_WIN32(GetLastError());
|
|
goto error;
|
|
}
|
|
}
|
|
|
|
*phSearchResult = hSearchResult;
|
|
RRETURN( S_OK );
|
|
|
|
error:
|
|
if(hSearchResult)
|
|
NwNdsFreeBuffer(phSearchInfo);
|
|
|
|
RRETURN(hr);
|
|
}
|
|
|
|
|