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.
675 lines
16 KiB
675 lines
16 KiB
/*++
|
|
|
|
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);
|
|
}
|