windows-nt/Source/XPSP1/NT/ds/adsi/winnt/cenmfpre.cxx

676 lines
16 KiB
C++
Raw Normal View History

2020-09-26 03:20:57 -05:00
/*++
Copyright (c) 1996 Microsoft Corporation
Module Name:
cenmfpre.cxx
Abstract:
Contains methods for implementing the Enumeration of resources(open files)
on a server. Has methods for the CFPNWResourcesCollection object
as well as the CFPNWResourcesEnumVar object.
Author:
Ram Viswanathan (ramv) 03/12/96
Revision History:
--*/
#include "winnt.hxx"
#pragma hdrstop
CFPNWResourcesCollection::CFPNWResourcesCollection()
{
_pszServerADsPath = NULL;
_pszServerName = NULL;
_pszBasePath = NULL;
_pszUserName = NULL;
_pDispMgr = NULL;
_pCResourcesEnumVar = NULL;
ENLIST_TRACKING(CFPNWResourcesCollection);
}
CFPNWResourcesCollection::~CFPNWResourcesCollection()
{
if(_pszServerADsPath){
FreeADsStr(_pszServerADsPath);
}
if(_pszServerName){
FreeADsStr(_pszServerName);
}
if(_pszBasePath){
FreeADsStr(_pszBasePath);
}
if(_pszUserName){
FreeADsStr(_pszUserName);
}
delete _pDispMgr;
if(_pCResourcesEnumVar){
_pCResourcesEnumVar->Release();
}
}
HRESULT
CFPNWResourcesCollection::Create(LPTSTR pszServerADsPath,
LPTSTR pszBasePath,
CWinNTCredentials& Credentials,
CFPNWResourcesCollection
** ppCFPNWResourcesCollection
)
{
BOOL fStatus = FALSE, LastError;
HRESULT hr;
CFPNWResourcesCollection *pCFPNWResourcesCollection = NULL;
POBJECTINFO pServerObjectInfo = NULL;
//
// create the Resources collection object
//
pCFPNWResourcesCollection = new CFPNWResourcesCollection();
if(pCFPNWResourcesCollection == NULL){
hr = E_OUTOFMEMORY;
goto cleanup;
}
pCFPNWResourcesCollection->_pszServerADsPath =
AllocADsStr(pszServerADsPath);
if(!pCFPNWResourcesCollection->_pszServerADsPath){
hr = E_OUTOFMEMORY;
goto cleanup;
}
hr = BuildObjectInfo(pszServerADsPath,
&pServerObjectInfo
);
BAIL_IF_ERROR(hr);
pCFPNWResourcesCollection->_pszServerName =
AllocADsStr(pServerObjectInfo->ComponentArray[1]);
if(!pCFPNWResourcesCollection->_pszServerName){
hr = E_OUTOFMEMORY;
goto cleanup;
}
pCFPNWResourcesCollection->_Credentials = Credentials;
hr = pCFPNWResourcesCollection->_Credentials.RefServer(
pCFPNWResourcesCollection->_pszServerName);
BAIL_IF_ERROR(hr);
if(pszBasePath){
pCFPNWResourcesCollection->_pszBasePath =
AllocADsStr(pszBasePath);
if(!pCFPNWResourcesCollection->_pszBasePath){
hr = E_OUTOFMEMORY;
goto cleanup;
}
}
pCFPNWResourcesCollection->_pDispMgr = new CAggregatorDispMgr;
if (pCFPNWResourcesCollection->_pDispMgr == NULL){
hr = E_OUTOFMEMORY;
goto cleanup;
}
hr = LoadTypeInfoEntry(pCFPNWResourcesCollection->_pDispMgr,
LIBID_ADs,
IID_IADsCollection,
(IADsCollection *)pCFPNWResourcesCollection,
DISPID_NEWENUM);
BAIL_IF_ERROR(hr);
hr = CFPNWResourcesEnumVar::Create(pszServerADsPath,
pszBasePath,
pCFPNWResourcesCollection->_Credentials,
&pCFPNWResourcesCollection->_pCResourcesEnumVar);
BAIL_IF_ERROR(hr);
*ppCFPNWResourcesCollection =pCFPNWResourcesCollection;
cleanup:
if(pServerObjectInfo){
FreeObjectInfo(pServerObjectInfo);
}
if(SUCCEEDED(hr)){
RRETURN(hr);
}
delete pCFPNWResourcesCollection;
RRETURN_EXP_IF_ERR(hr);
}
/* IUnknown methods for Resources collection object */
STDMETHODIMP
CFPNWResourcesCollection::QueryInterface(REFIID riid, LPVOID FAR* ppvObj)
{
if(!ppvObj){
RRETURN(E_POINTER);
}
if (IsEqualIID(riid, IID_IUnknown))
{
*ppvObj = this;
}
else if (IsEqualIID(riid, IID_IDispatch))
{
*ppvObj = (IDispatch *)this;
}
else if (IsEqualIID(riid, IID_ISupportErrorInfo))
{
*ppvObj = (ISupportErrorInfo FAR *) this;
}
else if (IsEqualIID(riid, IID_IADsCollection))
{
*ppvObj = (IADsCollection FAR *)this;
}
else
{
*ppvObj = NULL;
return E_NOINTERFACE;
}
((LPUNKNOWN)*ppvObj)->AddRef();
RRETURN(S_OK);
}
/* ISupportErrorInfo method */
STDMETHODIMP
CFPNWResourcesCollection::InterfaceSupportsErrorInfo(
THIS_ REFIID riid
)
{
if (IsEqualIID(riid, IID_IADsCollection)) {
RRETURN(S_OK);
} else {
RRETURN(S_FALSE);
}
}
DEFINE_IDispatch_Implementation(CFPNWResourcesCollection);
/* IADsCollection methods */
STDMETHODIMP
CFPNWResourcesCollection::get__NewEnum(THIS_ IUnknown * FAR* retval)
{
HRESULT hr;
CFPNWResourcesEnumVar *pCResourcesEnumVar = NULL;
if(!retval){
RRETURN_EXP_IF_ERR(E_POINTER);
}
*retval = NULL;
ADsAssert(_pCResourcesEnumVar);
hr = _pCResourcesEnumVar->QueryInterface(IID_IUnknown, (void **)retval);
RRETURN_EXP_IF_ERR(hr);
}
STDMETHODIMP
CFPNWResourcesCollection::GetObject(THIS_ BSTR bstrResourceName,
VARIANT *pvar
)
{
//
// scan the buffer _pbSessions to find one where the ConnectionId
// matches the one in bstrSessionName and get this object
//
HRESULT hr;
hr = _pCResourcesEnumVar->GetObject(bstrResourceName, pvar);
RRETURN_EXP_IF_ERR(hr);
/*
HRESULT hr = S_OK;
DWORD dwFileId;
IDispatch *pDispatch = NULL;
if(!bstrResourceName || !pvar){
RRETURN_EXP_IF_ERR(E_ADS_BAD_PARAMETER);
}
//
// convert the job name into a fileid
//
dwFileId = (DWORD)_wtol(bstrResourceName);
hr = CFPNWResource::Create((LPTSTR)_bstrServerADsPath,
ADS_OBJECT_BOUND,
dwFileId,
IID_IDispatch,
_Credentials,
(void **)&pDispatch);
BAIL_IF_ERROR(hr);
//
// stick this IDispatch pointer into caller provided variant
//
VariantInit(pvar);
V_VT(pvar) = VT_DISPATCH;
V_DISPATCH(pvar) = pDispatch;
cleanup:
RRETURN_EXP_IF_ERR(hr);
*/
}
STDMETHODIMP
CFPNWResourcesCollection::Add(THIS_ BSTR bstrName, VARIANT varNewItem)
{
RRETURN_EXP_IF_ERR(E_NOTIMPL);
}
STDMETHODIMP
CFPNWResourcesCollection::Remove(THIS_ BSTR bstrResourceName)
{
RRETURN_EXP_IF_ERR(E_NOTIMPL);
}
//
// CFPNWResourcesEnumVar methods follow
//
//+---------------------------------------------------------------------------
//
// Function: CFPNWResourcesEnumVar::CFPNWResourcesEnumVar
//
// Synopsis:
//
//
// Arguments:
//
//
// Returns:
//
// Modifies:
//
// History: 11-22-95 RamV Created.
//
//----------------------------------------------------------------------------
CFPNWResourcesEnumVar::CFPNWResourcesEnumVar()
{
_pszServerADsPath = NULL;
_pszServerName = NULL;
_pszBasePath = NULL;
_pszUserName = NULL;
_pbResources = NULL;
_cElements = 0;
_lLBound = 0;
_lCurrentPosition = _lLBound;
_dwTotalEntries = 0;
_dwResumeHandle = 0;
}
//+---------------------------------------------------------------------------
//
// Function: CFPNWResourcesEnumVar::~CFPNWResourcesEnumVar
//
// Synopsis:
//
//
// Arguments:
//
// Returns:
//
// Modifies:
//
// History: 11-22-95 RamV Created.
//
//----------------------------------------------------------------------------
CFPNWResourcesEnumVar::~CFPNWResourcesEnumVar()
{
if(_pszServerADsPath){
FreeADsStr(_pszServerADsPath);
}
if(_pszServerName){
FreeADsStr(_pszServerName);
}
if(_pszBasePath){
FreeADsStr(_pszBasePath);
}
if(_pszUserName){
FreeADsStr(_pszUserName);
}
if(_pbResources){
NetApiBufferFree(_pbResources);
}
}
HRESULT CFPNWResourcesEnumVar::Create(LPTSTR pszServerADsPath,
LPTSTR pszBasePath,
CWinNTCredentials& Credentials,
CFPNWResourcesEnumVar
**ppCResourcesEnumVar)
{
HRESULT hr;
BOOL fStatus = FALSE;
POBJECTINFO pServerObjectInfo = NULL;
CFPNWResourcesEnumVar FAR* pCResourcesEnumVar = NULL;
*ppCResourcesEnumVar = NULL;
pCResourcesEnumVar = new CFPNWResourcesEnumVar();
if (pCResourcesEnumVar == NULL){
hr = E_OUTOFMEMORY;
goto cleanup;
}
pCResourcesEnumVar->_pszServerADsPath =
AllocADsStr(pszServerADsPath);
if(!pCResourcesEnumVar->_pszServerADsPath){
hr = E_OUTOFMEMORY;
goto cleanup;
}
hr = BuildObjectInfo(pszServerADsPath,
&pServerObjectInfo);
BAIL_IF_ERROR(hr);
pCResourcesEnumVar->_pszServerName =
AllocADsStr(pServerObjectInfo->ComponentArray[1]);
if(!pCResourcesEnumVar->_pszServerName){
hr = E_OUTOFMEMORY;
goto cleanup;
}
pCResourcesEnumVar->_Credentials = Credentials;
hr = pCResourcesEnumVar->_Credentials.RefServer(
pCResourcesEnumVar->_pszServerName);
BAIL_IF_ERROR(hr);
if(pszBasePath){
pCResourcesEnumVar->_pszBasePath =
AllocADsStr(pszBasePath);
if(!pCResourcesEnumVar->_pszBasePath){
hr = E_OUTOFMEMORY;
goto cleanup;
}
}
*ppCResourcesEnumVar = pCResourcesEnumVar;
cleanup:
if(pServerObjectInfo){
FreeObjectInfo(pServerObjectInfo);
}
if(SUCCEEDED(hr)){
RRETURN(hr);
}
delete pCResourcesEnumVar;
RRETURN_EXP_IF_ERR(hr);
}
//+---------------------------------------------------------------------------
//
// Function: CFPNWResourcesEnumVar::Next
//
// Synopsis: Returns cElements number of requested Resource objects in the
// array supplied in pvar.
//
// Arguments: [cElements] -- The number of elements requested by client
// [pvar] -- ptr to array of VARIANTs to for return objects
// [pcElementFetched] -- if non-NULL, then number of elements
// -- actually returned is placed here
//
// Returns: HRESULT -- S_OK if number of elements requested are returned
// -- S_FALSE if number of elements is < requested
//
// Modifies:
//
// History: 11-27-95 RamV Created.
//
//----------------------------------------------------------------------------
STDMETHODIMP
CFPNWResourcesEnumVar::Next(ULONG ulNumElementsRequested,
VARIANT FAR* pvar,
ULONG FAR* pulNumFetched)
{
HRESULT hresult;
ULONG l;
LONG lNewCurrent;
ULONG lNumFetched;
PNWFILEINFO pFileInfo;
VARIANT v;
IDispatch * pDispatch = NULL;
if (pulNumFetched != NULL){
*pulNumFetched = 0;
}
//
// Initialize the elements to be returned
//
for (l=0; l<ulNumElementsRequested; l++){
VariantInit(&pvar[l]);
}
if(!_pbResources || (_lCurrentPosition == _lLBound +(LONG)_cElements) ){
if (_pbResources){
NetApiBufferFree(_pbResources);
_pbResources = NULL;
}
hresult = FPNWEnumResources(_pszServerName,
_pszBasePath,
&_pbResources,
&_cElements,
&_dwResumeHandle);
if(hresult == S_FALSE){
RRETURN(S_FALSE);
}
_lLBound = 0;
_lCurrentPosition = _lLBound;
}
//
// Get each element and place it into the return array
// Don't request more than we have
//
for (lNewCurrent=_lCurrentPosition, lNumFetched=0;
lNewCurrent<(LONG)(_lLBound+_cElements) &&
lNumFetched < ulNumElementsRequested;
lNewCurrent++, lNumFetched++){
pFileInfo = (PNWFILEINFO)(_pbResources + \
lNewCurrent*sizeof(NWFILEINFO));
hresult = CFPNWResource::Create(_pszServerADsPath,
pFileInfo,
ADS_OBJECT_BOUND,
IID_IDispatch,
_Credentials,
(void **)&pDispatch);
BAIL_ON_FAILURE(hresult);
VariantInit(&v);
V_VT(&v) = VT_DISPATCH;
V_DISPATCH(&v) = pDispatch;
pvar[lNumFetched] = v;
}
//
// Tell the caller how many we got (which may be less than the number
// requested), and save the current position
//
if (pulNumFetched != NULL)
*pulNumFetched = lNumFetched;
_lCurrentPosition = lNewCurrent;
//
// If we're returning less than they asked for return S_FALSE, but
// they still have the data (S_FALSE is a success code)
//
return (lNumFetched < ulNumElementsRequested) ?
S_FALSE
: S_OK;
error:
RRETURN(S_FALSE);
}
HRESULT
FPNWEnumResources(LPTSTR pszServerName,
LPTSTR pszBasePath,
LPBYTE * ppMem,
LPDWORD pdwEntriesRead,
LPDWORD pdwResumeHandle
)
{
HRESULT hr;
DWORD dwErrorCode;
dwErrorCode = ADsNwFileEnum(pszServerName,
1,
pszBasePath,
(PNWFILEINFO *) ppMem,
pdwEntriesRead,
pdwResumeHandle);
if(*ppMem == NULL|| (dwErrorCode != NERR_Success)){
//
// no more entries returned by NwFileEnum
//
RRETURN(S_FALSE);
}
RRETURN(S_OK);
}
//
// helper function
//
HRESULT
CFPNWResourcesEnumVar::GetObject(BSTR bstrResourceName, VARIANT *pvar)
{
HRESULT hr = S_OK;
DWORD dwFileId;
PNWFILEINFO pFileInfo = NULL;
IDispatch *pDispatch = NULL;
DWORD i;
if(!_pbResources){
RRETURN_EXP_IF_ERR(E_ADS_UNKNOWN_OBJECT);
}
//
// scan the buffer _pbResources to find one where the ConnectionId
// matches the one in bstrResourceName and get this object
//
dwFileId = (DWORD)_wtol(bstrResourceName);
for( i=0; i<_cElements; i++){
pFileInfo = (PNWFILEINFO)(_pbResources+ i*sizeof(PNWFILEINFO));
if(pFileInfo->dwFileId = dwFileId){
//
// return this struct in the static create for the object
//
hr = CFPNWResource::Create(_pszServerADsPath,
pFileInfo,
ADS_OBJECT_BOUND,
IID_IDispatch,
_Credentials,
(void **) &pDispatch );
BAIL_IF_ERROR(hr);
break;
}
}
if(i == _cElements){
//
// no such element
//
hr = E_ADS_UNKNOWN_OBJECT;
goto cleanup;
}
//
// stick this IDispatch pointer into caller provided variant
//
VariantInit(pvar);
V_VT(pvar) = VT_DISPATCH;
V_DISPATCH(pvar) = pDispatch;
cleanup:
RRETURN_EXP_IF_ERR(hr);
}