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.
422 lines
10 KiB
422 lines
10 KiB
//+---------------------------------------------------------------------------
|
|
//
|
|
// Microsoft Windows
|
|
// Copyright (C) Microsoft Corporation, 1992 - 2000.
|
|
//
|
|
// File: umisrch.cxx
|
|
//
|
|
// Contents: This file contains the query object for the LDAP provider.
|
|
// IUmiQuery is the interface supported by this object. The
|
|
// properties on the interface property list of this object are
|
|
// mapped to the preferences you can set for IDirectorySearch.
|
|
//
|
|
// History: 03-27-00 AjayR Created.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
#include "ldap.hxx"
|
|
|
|
//+---------------------------------------------------------------------------
|
|
// Function: CUmiLDAPQuery::CUmiLDAPQuery --- Constructor.
|
|
//
|
|
// Synopsis: Standard constructor.
|
|
//
|
|
// Arguments: N/A.
|
|
//
|
|
// Returns: N/A.
|
|
//
|
|
// Modifies: N/A.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
CUmiLDAPQuery::CUmiLDAPQuery():
|
|
_pIntfPropMgr(NULL),
|
|
_pszQueryText(NULL),
|
|
_pszLanguage(NULL),
|
|
_ulStatus(0)
|
|
{
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
// Function: CUmiLDAPQuery::~CUmiLDAPQuery --- Destructor.
|
|
//
|
|
// Synopsis: Standard destructor.
|
|
//
|
|
// Arguments: N/A.
|
|
//
|
|
// Returns: N/A.
|
|
//
|
|
// Modifies: N/A.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
CUmiLDAPQuery::~CUmiLDAPQuery()
|
|
{
|
|
if (_pIntfPropMgr) {
|
|
delete _pIntfPropMgr;
|
|
}
|
|
|
|
if (_pszQueryText) {
|
|
FreeADsStr(_pszQueryText);
|
|
}
|
|
|
|
if (_pszLanguage) {
|
|
FreeADsStr(_pszLanguage);
|
|
}
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
// Function: CUmiLDAPQuery::CreateUmiLDAPQuery --- STATIC constructor.
|
|
//
|
|
// Synopsis: Static constructor.
|
|
//
|
|
// Arguments: riid - Interface needed on new object.
|
|
// ppObj - Return ptr for newly created object.
|
|
//
|
|
// Returns: S_OK on success or any suitable error code.
|
|
//
|
|
// Modifies: N/A.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
HRESULT
|
|
CUmiLDAPQuery::CreateUmiLDAPQuery(
|
|
IID riid,
|
|
void FAR* FAR* ppObj
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
CUmiLDAPQuery *pQuery;
|
|
CPropertyManager *pIntfPropMgr = NULL;
|
|
|
|
if (!ppObj) {
|
|
RRETURN(E_INVALIDARG);
|
|
}
|
|
|
|
pQuery = new CUmiLDAPQuery();
|
|
if (!pQuery) {
|
|
BAIL_ON_FAILURE(hr = E_OUTOFMEMORY);
|
|
}
|
|
|
|
hr = CPropertyManager::CreatePropertyManager(
|
|
(IUmiQuery *) pQuery,
|
|
NULL, // IADs ptr
|
|
NULL, // pCreds
|
|
IntfPropsQuery,
|
|
&pIntfPropMgr
|
|
);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
|
|
hr = pQuery->QueryInterface(riid, ppObj);
|
|
BAIL_ON_FAILURE(hr)
|
|
|
|
pQuery->Release();
|
|
|
|
pQuery->_pIntfPropMgr = pIntfPropMgr;
|
|
|
|
RRETURN(hr);
|
|
|
|
error :
|
|
|
|
if (pQuery) {
|
|
delete pQuery;
|
|
}
|
|
|
|
if (pIntfPropMgr) {
|
|
delete pIntfPropMgr;
|
|
}
|
|
|
|
RRETURN(hr);
|
|
}
|
|
|
|
//
|
|
// IUnknown support - standard query interface method.
|
|
//
|
|
STDMETHODIMP
|
|
CUmiLDAPQuery::QueryInterface(REFIID iid, LPVOID FAR* ppv)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
if (ppv == NULL) {
|
|
RRETURN(E_POINTER);
|
|
}
|
|
|
|
if (IsEqualIID(iid, IID_IUnknown)){
|
|
*ppv = (IUnknown FAR *) this;
|
|
}
|
|
else if (IsEqualIID(iid, IID_IUmiPropList)) {
|
|
*ppv = (IUmiConnection FAR *) this;
|
|
}
|
|
else if (IsEqualIID(iid, IID_IUmiBaseObject)) {
|
|
*ppv = (IUmiConnection FAR *) this;
|
|
}
|
|
else if (IsEqualIID(iid, IID_IUmiQuery)) {
|
|
*ppv = (IUmiQuery FAR *) this;
|
|
}
|
|
else {
|
|
*ppv = NULL;
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
AddRef();
|
|
return NOERROR;
|
|
}
|
|
|
|
//
|
|
// IUmiQuery methods.
|
|
//
|
|
|
|
//+---------------------------------------------------------------------------
|
|
// Function: CUmiLDAPQuery::Set --- IUmiQuery support.
|
|
//
|
|
// Synopsis: Sets the query string and language.
|
|
//
|
|
// Arguments: pszLanguage - Language used to specify the query,
|
|
// only LDAP and SQL/WQL are supported.
|
|
// uFlags - Flags, only 0 is allowed now.
|
|
// pszText - Query text.
|
|
//
|
|
// Returns: N/A.
|
|
//
|
|
// Modifies: Internal query and language strings.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
STDMETHODIMP
|
|
CUmiLDAPQuery::Set(
|
|
IN LPCWSTR pszLanguage,
|
|
IN ULONG uFlags,
|
|
IN LPCWSTR pszText
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
SetLastStatus(0);
|
|
//
|
|
// Validate params
|
|
//
|
|
if (!pszLanguage || !pszText) {
|
|
BAIL_ON_FAILURE(hr = E_INVALIDARG);
|
|
}
|
|
|
|
if (uFlags != 0) {
|
|
BAIL_ON_FAILURE(hr = UMI_E_INVALID_FLAGS);
|
|
}
|
|
|
|
if (_wcsicmp(pszLanguage, L"LDAP")
|
|
&& _wcsicmp(pszLanguage, L"SQL")
|
|
&& _wcsicmp(pszLanguage, L"WQL")
|
|
) {
|
|
//
|
|
// We do not support this language.
|
|
//
|
|
BAIL_ON_FAILURE(hr = E_INVALIDARG);
|
|
}
|
|
|
|
//
|
|
// Free exisitng stuff if needed.
|
|
//
|
|
if (_pszLanguage) {
|
|
FreeADsStr(_pszLanguage);
|
|
_pszLanguage = NULL;
|
|
}
|
|
|
|
if (_pszQueryText) {
|
|
FreeADsStr(_pszQueryText);
|
|
}
|
|
|
|
_pszLanguage = AllocADsStr(pszLanguage);
|
|
if (!_pszLanguage) {
|
|
BAIL_ON_FAILURE(hr = E_OUTOFMEMORY);
|
|
}
|
|
|
|
_pszQueryText = AllocADsStr(pszText);
|
|
if (!_pszQueryText) {
|
|
BAIL_ON_FAILURE(hr = E_OUTOFMEMORY);
|
|
}
|
|
|
|
error:
|
|
|
|
if (FAILED(hr)) {
|
|
SetLastStatus(hr);
|
|
hr = MapHrToUmiError(hr);
|
|
}
|
|
|
|
RRETURN(hr);
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
// Function: CUmiLDAPQuery::GetQuery --- IUmiQuery support.
|
|
//
|
|
// Synopsis: Returns the query language and text in the user allocated
|
|
// buffers provided.
|
|
//
|
|
// Arguments: puLangBufSize - Size of buffer for language string.
|
|
// pszLangBuf - The actual buffer.
|
|
// puQyeryTextBufSiz - Size of buffer for queyr text.
|
|
// pszQueryTextBuf - Buffer for query text.
|
|
//
|
|
// Returns: S_OK on success or any appropriate error code.
|
|
//
|
|
// Modifies: pszLangBuf and pszQueryText on success. On failure,
|
|
// * puLangBufSize and *puQueryTextBufSize are updated with
|
|
// the length of the buffers needed (length in bytes).
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
STDMETHODIMP
|
|
CUmiLDAPQuery::GetQuery(
|
|
IN OUT ULONG * puLangBufSize,
|
|
IN OUT LPWSTR pszLangBuf,
|
|
IN OUT ULONG * puQueryTextBufSize,
|
|
IN OUT LPWSTR pszQueryTextBuf
|
|
)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
SetLastStatus(0);
|
|
|
|
if (!_pszLanguage || !_pszQueryText) {
|
|
BAIL_ON_FAILURE(hr = UMI_E_NOT_FOUND);
|
|
}
|
|
|
|
if (!puLangBufSize || !puQueryTextBufSize) {
|
|
BAIL_ON_FAILURE(hr = E_INVALIDARG);
|
|
}
|
|
|
|
if ((*puLangBufSize < ((wcslen(_pszLanguage) + 1) * sizeof(WCHAR)))
|
|
|| *puQueryTextBufSize < ((wcslen(_pszQueryText) + 1) * sizeof(WCHAR))
|
|
) {
|
|
//
|
|
// We really need an insufficient buffer error for this.
|
|
//
|
|
*puLangBufSize = (wcslen(_pszLanguage) + 1) * sizeof(WCHAR);
|
|
*puQueryTextBufSize = (wcslen(_pszQueryText) + 1) * sizeof(WCHAR);
|
|
BAIL_ON_FAILURE( hr = E_OUTOFMEMORY);
|
|
}
|
|
|
|
//
|
|
// We have enough space in the provided buffers.
|
|
//
|
|
wcscpy(pszLangBuf, _pszLanguage);
|
|
wcscpy(pszQueryTextBuf, _pszQueryText);
|
|
|
|
error:
|
|
|
|
if (FAILED(hr)) {
|
|
SetLastStatus(hr);
|
|
hr = MapHrToUmiError(hr);
|
|
}
|
|
|
|
RRETURN(hr);
|
|
}
|
|
|
|
//
|
|
// IUmiBaseObject methods.
|
|
//
|
|
|
|
//+---------------------------------------------------------------------------
|
|
// Function: CUmiLDAPQuery::GetLastStatus (IUmiBaseObject method).
|
|
//
|
|
// Synopsis: Returns only numeric status code from the last operation.
|
|
//
|
|
// Arguments: uFlags - Only 0 is supported for now.
|
|
// puSpecificStatus - Returns status/error code.
|
|
// riid - Not used.
|
|
// pStatusObj - NULL, not used currently.
|
|
//
|
|
// Returns: UMI_S_NO_ERROR on success. Error code otherwise.
|
|
//
|
|
// Modifies: *puSpecificStatus to return appropriate status code.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
STDMETHODIMP
|
|
CUmiLDAPQuery::GetLastStatus(
|
|
IN ULONG uFlags,
|
|
OUT ULONG *puSpecificStatus,
|
|
IN REFIID riid,
|
|
OUT LPVOID *pStatusObj
|
|
)
|
|
{
|
|
if (uFlags != 0) {
|
|
RRETURN(UMI_E_INVALID_FLAGS);
|
|
}
|
|
|
|
if (!puSpecificStatus) {
|
|
RRETURN(E_INVALIDARG);
|
|
}
|
|
|
|
if (pStatusObj) {
|
|
//
|
|
// Should we error out ?
|
|
//
|
|
*pStatusObj = NULL;
|
|
}
|
|
|
|
*puSpecificStatus = _ulStatus;
|
|
|
|
RRETURN(UMI_S_NO_ERROR);
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
// Function: CUmiLDAPQuery::GetInterfacePropList (IUmiBaseObject method).
|
|
//
|
|
// Synopsis: Returns a pointer to the interface property list for
|
|
// cursor object.
|
|
//
|
|
// Arguments: uFlags - Flags, only 0 is supported.
|
|
// ppPropList - Return value.
|
|
//
|
|
// Returns: UMI_S_NO_ERROR on success. Error code otherwise.
|
|
//
|
|
// Modifies: *pPropList changed to IUmiPropList pointer
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
STDMETHODIMP
|
|
CUmiLDAPQuery::GetInterfacePropList(
|
|
IN ULONG uFlags,
|
|
OUT IUmiPropList **pPropList
|
|
)
|
|
{
|
|
HRESULT hr = UMI_S_NO_ERROR;
|
|
|
|
SetLastStatus(0);
|
|
|
|
if (uFlags != 0) {
|
|
SetLastStatus(hr);
|
|
RRETURN(UMI_E_INVALID_FLAGS);
|
|
}
|
|
|
|
//
|
|
// QI will check for bad pointer so no need to check here.
|
|
//
|
|
hr = _pIntfPropMgr->QueryInterface(IID_IUmiPropList, (void **)pPropList);
|
|
|
|
if (FAILED(hr)) {
|
|
SetLastStatus(hr);
|
|
}
|
|
|
|
RRETURN(hr);
|
|
}
|
|
|
|
//
|
|
// Private/Protected Methods.
|
|
//
|
|
|
|
//+---------------------------------------------------------------------------
|
|
// Function: CUmiCursor::SetLastStatus (internal private helper routine).
|
|
//
|
|
// Synopsis: Sets the status of the last operation. If the status is one
|
|
// of the pre-defined error codes, then the status is just set to
|
|
// 0 since we are not adding any value by returning the same
|
|
// status as the error code.
|
|
//
|
|
// Arguments: ulStatus - Status to be set.
|
|
//
|
|
// Returns: Nothing
|
|
//
|
|
// Modifies: Internal member status variable.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
void
|
|
CUmiLDAPQuery::SetLastStatus(ULONG ulStatus)
|
|
{
|
|
this->_ulStatus = ulStatus;
|
|
|
|
return;
|
|
}
|