680 lines
16 KiB
C++
680 lines
16 KiB
C++
//
|
||
// Author: ushaji
|
||
// Date: December 1996
|
||
//
|
||
//
|
||
// Providing support for Component Categories in Class Store
|
||
//
|
||
// This source file contains implementations for Enumerator Interfaces.
|
||
//
|
||
// Refer Doc "Design for Support of File Types and Component Categories
|
||
// in Class Store" ? (or may be Class Store Schema). Most of these uses
|
||
// OLE DB interfaces within the interfaces.
|
||
//
|
||
//----------------------------------------------------------------------------
|
||
|
||
#include "csadm.hxx"
|
||
|
||
|
||
//
|
||
// Private defines
|
||
//
|
||
|
||
#define MAX_ADS_FILTERS 10
|
||
#define MAX_ADS_ENUM 100
|
||
|
||
// BUGBUG:: Should this go to common
|
||
|
||
HRESULT
|
||
EnumCsObject(
|
||
IADsContainer * pADsContainer,
|
||
LPWSTR * lppPathNames,
|
||
DWORD dwPathNames,
|
||
IEnumVARIANT ** ppEnumVariant
|
||
)
|
||
{
|
||
ULONG cElementFetched = 0L;
|
||
VARIANT VarFilter;
|
||
HRESULT hr;
|
||
DWORD dwObjects = 0, dwEnumCount = 0, i = 0;
|
||
BOOL fContinue = TRUE;
|
||
|
||
|
||
if (dwPathNames)
|
||
{
|
||
VariantInit(&VarFilter);
|
||
|
||
hr = BuildVarArrayStr(
|
||
lppPathNames,
|
||
dwPathNames,
|
||
&VarFilter
|
||
);
|
||
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
hr = pADsContainer->put_Filter(VarFilter);
|
||
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
VariantClear(&VarFilter);
|
||
}
|
||
|
||
hr = ADsBuildEnumerator(
|
||
pADsContainer,
|
||
ppEnumVariant
|
||
);
|
||
|
||
return hr;
|
||
}
|
||
|
||
|
||
HRESULT
|
||
GetNextEnum(
|
||
IEnumVARIANT * pEnum,
|
||
IADs **ppADs
|
||
)
|
||
{
|
||
|
||
HRESULT hr;
|
||
VARIANT VariantArray[MAX_ADS_ENUM];
|
||
IDispatch *pDispatch = NULL;
|
||
|
||
|
||
hr = ADsEnumerateNext(
|
||
pEnum,
|
||
1,
|
||
VariantArray,
|
||
NULL
|
||
);
|
||
|
||
if (hr == S_FALSE)
|
||
return hr;
|
||
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
pDispatch = VariantArray[0].pdispVal;
|
||
memset(VariantArray, 0, sizeof(VARIANT)*MAX_ADS_ENUM);
|
||
hr = pDispatch->QueryInterface(IID_IADs, (void **) ppADs) ;
|
||
pDispatch->Release();
|
||
return(S_OK);
|
||
}
|
||
|
||
HRESULT GetCategoryProperty (IADs *pADs,
|
||
CATEGORYINFO *pcategoryinfo, LCID lcid)
|
||
{
|
||
HRESULT hr = S_OK;
|
||
WCHAR szName[_MAX_PATH];
|
||
LPOLESTR *pdesc = NULL;
|
||
ULONG i, cdesc;
|
||
|
||
hr = GetPropertyGuid(pADs, CATEGORYCATID, &(pcategoryinfo->catid));
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
hr = GetPropertyListAlloc(pADs, LOCALEDESCRIPTION, &cdesc, &pdesc);
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
pcategoryinfo->lcid = lcid;
|
||
|
||
hr = GetCategoryLocaleDesc(pdesc, cdesc, &(pcategoryinfo->lcid),
|
||
(pcategoryinfo->szDescription));
|
||
for (i = 0; i < cdesc; i++)
|
||
CoTaskMemFree(pdesc[i]);
|
||
CoTaskMemFree(pdesc);
|
||
|
||
if (FAILED(hr))
|
||
*(pcategoryinfo->szDescription) = L'\0';
|
||
|
||
return S_OK;
|
||
}
|
||
//----------------------------------------------
|
||
|
||
|
||
// IEnumCATEGORYINFO:
|
||
// IUnknown methods
|
||
HRESULT CSCEnumCategories::QueryInterface(REFIID riid, void** ppObject)
|
||
{
|
||
*ppObject = NULL; //gd
|
||
if ((riid==IID_IUnknown) || (riid==IID_IEnumCATEGORYINFO))
|
||
{
|
||
*ppObject=(IUnknown *)(IEnumCATEGORYINFO*) this;
|
||
}
|
||
else
|
||
{
|
||
return E_NOINTERFACE;
|
||
}
|
||
AddRef();
|
||
return S_OK;
|
||
}
|
||
|
||
ULONG CSCEnumCategories::AddRef()
|
||
{
|
||
InterlockedIncrement((long*) &m_dwRefCount);
|
||
return m_dwRefCount;
|
||
}
|
||
|
||
ULONG CSCEnumCategories::Release()
|
||
{
|
||
ULONG dwRefCount;
|
||
|
||
if ((dwRefCount = InterlockedDecrement((long*) &m_dwRefCount))==0)
|
||
{
|
||
delete this;
|
||
return 0;
|
||
}
|
||
return dwRefCount;
|
||
}
|
||
|
||
// IEnumCATEGORYINFO methods
|
||
//--------------------------------------------------------------------
|
||
// Next:
|
||
// celt: Number of elements to be fetched.
|
||
// rgelt: Buffer to return the elements.
|
||
// pceltFetched: ptr to Number of elements actually fetched.
|
||
// Takes care of that being NULL.
|
||
// Uses OLE DS interfaces to get the next enumerators.
|
||
//--------------------------------------------------------------------
|
||
|
||
HRESULT CSCEnumCategories::Next(ULONG celt, CATEGORYINFO *rgelt, ULONG *pceltFetched)
|
||
{
|
||
HRESULT hr = S_OK;
|
||
IADs *pCatADs = NULL;
|
||
ULONG dwCount;
|
||
|
||
if ((celt > 1) && (!pceltFetched))
|
||
return E_INVALIDARG;
|
||
|
||
if (pceltFetched)
|
||
(*pceltFetched) = 0;
|
||
|
||
if (m_pEnumVariant == NULL)
|
||
return E_UNEXPECTED;
|
||
|
||
if (celt < 0)
|
||
return E_INVALIDARG;
|
||
|
||
if (!IsValidPtrOut(rgelt, sizeof(CATEGORYINFO)*celt)) {
|
||
return E_INVALIDARG;
|
||
}
|
||
|
||
for ((dwCount) = 0; (dwCount) < celt;) {
|
||
hr = GetNextEnum(m_pEnumVariant, &pCatADs);
|
||
if ((FAILED(hr)) || (hr == S_FALSE))
|
||
break;
|
||
|
||
hr = GetCategoryProperty(pCatADs, &(rgelt[dwCount]), m_lcid);
|
||
pCatADs->Release();
|
||
|
||
if (FAILED(hr))
|
||
continue;
|
||
|
||
(dwCount)++;
|
||
}
|
||
|
||
m_dwPosition += dwCount;
|
||
|
||
if (pceltFetched)
|
||
(*pceltFetched) = dwCount;
|
||
return hr;
|
||
}
|
||
|
||
//---------------------------------------------------
|
||
// skip:
|
||
// skips elements.
|
||
// celt Number of elements to skip.
|
||
//---------------------------------------------------
|
||
|
||
HRESULT CSCEnumCategories::Skip(ULONG celt)
|
||
{
|
||
CATEGORYINFO *dummy;
|
||
ULONG got;
|
||
HRESULT hr;
|
||
|
||
dummy = new CATEGORYINFO[celt];
|
||
if (!dummy)
|
||
return E_OUTOFMEMORY;
|
||
|
||
if (m_pEnumVariant == NULL)
|
||
{
|
||
delete dummy;
|
||
return E_UNEXPECTED;
|
||
}
|
||
hr = Next(celt, dummy, &got);
|
||
delete dummy;
|
||
return hr;
|
||
}
|
||
|
||
//---------------------------------------------------
|
||
// Reset:
|
||
// Resets the pointer.
|
||
//---------------------------------------------------
|
||
HRESULT CSCEnumCategories::Reset(void)
|
||
{
|
||
LPOLESTR pszPathNames [2];
|
||
HRESULT hr;
|
||
|
||
pszPathNames [0] = CLASS_CS_CATEGORY;
|
||
|
||
if (m_pEnumVariant == NULL)
|
||
return E_UNEXPECTED;
|
||
m_pEnumVariant->Release();
|
||
|
||
hr = EnumCsObject(m_ADsCategoryContainer, &pszPathNames[0],
|
||
0, &m_pEnumVariant);
|
||
|
||
m_dwPosition = 0;
|
||
return hr;
|
||
}
|
||
//--------------------------------------------------------------
|
||
// Clone:
|
||
// returns another interface which points to the same data.
|
||
// ppenum: enumerator
|
||
//------------------------------------------------------------
|
||
////////////////////////////////////////////////////////////////////
|
||
HRESULT CSCEnumCategories::Clone(IEnumCATEGORYINFO **ppenum)
|
||
{
|
||
CSCEnumCategories* pClone=NULL;
|
||
|
||
if (!IsValidPtrOut(ppenum, sizeof(IEnumCATEGORYINFO *))) {
|
||
return E_INVALIDARG;
|
||
}
|
||
|
||
pClone=new CSCEnumCategories();
|
||
|
||
if (!pClone)
|
||
{
|
||
return E_OUTOFMEMORY;
|
||
}
|
||
if (FAILED(pClone->Initialize(m_ADsCategoryContainer, m_lcid)))
|
||
{
|
||
delete pClone;
|
||
return E_UNEXPECTED;
|
||
}
|
||
|
||
pClone->Skip(m_dwPosition);
|
||
|
||
if (SUCCEEDED(pClone->QueryInterface(IID_IEnumCATEGORYINFO, (void**) ppenum)))
|
||
{
|
||
return S_OK;
|
||
}
|
||
|
||
delete pClone;
|
||
return E_UNEXPECTED;
|
||
}
|
||
|
||
CSCEnumCategories::CSCEnumCategories()
|
||
{
|
||
m_dwRefCount=0;
|
||
m_pEnumVariant = NULL;
|
||
m_ADsCategoryContainer = NULL;
|
||
m_lcid=0;
|
||
m_dwPosition = 0;
|
||
}
|
||
|
||
HRESULT CSCEnumCategories::Initialize(IADsContainer *ADsCategoryContainer, LCID lcid)
|
||
{
|
||
LPOLESTR pszPathNames [2];
|
||
HRESULT hr;
|
||
|
||
pszPathNames [0] = CLASS_CS_CATEGORY;
|
||
m_ADsCategoryContainer = ADsCategoryContainer;
|
||
m_ADsCategoryContainer->AddRef();
|
||
|
||
hr = EnumCsObject(m_ADsCategoryContainer, &pszPathNames[0], 0, &m_pEnumVariant);
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
m_lcid = lcid;
|
||
return S_OK;
|
||
} /* EnumCategories */
|
||
|
||
|
||
CSCEnumCategories::~CSCEnumCategories()
|
||
{
|
||
if (m_ADsCategoryContainer)
|
||
m_ADsCategoryContainer->Release();
|
||
if (m_pEnumVariant)
|
||
m_pEnumVariant->Release();
|
||
}
|
||
|
||
// CSCEnumCategoriesOfClass:
|
||
// IUnknown methods
|
||
HRESULT CSCEnumCategoriesOfClass::QueryInterface(REFIID riid, void** ppObject)
|
||
{
|
||
if ((riid==IID_IUnknown) || (riid==IID_IEnumCATID))
|
||
{
|
||
*ppObject=(IEnumCATID*) this;
|
||
}
|
||
else
|
||
{
|
||
return E_NOINTERFACE;
|
||
}
|
||
AddRef();
|
||
return S_OK;
|
||
}
|
||
|
||
ULONG CSCEnumCategoriesOfClass::AddRef()
|
||
{
|
||
InterlockedIncrement((long*) &m_dwRefCount);
|
||
return m_dwRefCount;
|
||
}
|
||
|
||
ULONG CSCEnumCategoriesOfClass::Release()
|
||
{
|
||
ULONG dwRefCount;
|
||
if ((dwRefCount = InterlockedDecrement((long*) &m_dwRefCount))==0)
|
||
{
|
||
delete this;
|
||
return 0;
|
||
}
|
||
return dwRefCount;
|
||
}
|
||
|
||
// IEnumCATID methods
|
||
HRESULT CSCEnumCategoriesOfClass::Next(ULONG celt, GUID *rgelt, ULONG *pceltFetched)
|
||
{
|
||
ULONG dwCount;
|
||
|
||
if ((celt > 1) && (!pceltFetched))
|
||
return E_INVALIDARG;
|
||
|
||
if (pceltFetched)
|
||
(*pceltFetched) = 0;
|
||
|
||
if (celt < 0)
|
||
return E_INVALIDARG;
|
||
|
||
if (!IsValidPtrOut(rgelt, sizeof(GUID)*celt))
|
||
return E_INVALIDARG;
|
||
|
||
for ( (dwCount) = 0; (((dwCount) < celt) && (m_dwPosition < m_cCatid));
|
||
(dwCount)++, m_dwPosition++)
|
||
rgelt[(dwCount)] = m_catid[m_dwPosition];
|
||
|
||
if (pceltFetched)
|
||
(*pceltFetched) = dwCount;
|
||
if (dwCount == celt)
|
||
return S_OK;
|
||
return S_FALSE;
|
||
}
|
||
|
||
HRESULT CSCEnumCategoriesOfClass::Skip(ULONG celt)
|
||
{
|
||
if (m_cCatid >= (celt + m_dwPosition)) {
|
||
m_dwPosition += celt;
|
||
return S_OK;
|
||
}
|
||
m_dwPosition = m_cCatid;
|
||
return S_FALSE;
|
||
}
|
||
|
||
|
||
HRESULT CSCEnumCategoriesOfClass::Reset(void)
|
||
{
|
||
m_dwPosition = 0;
|
||
return S_OK;
|
||
}
|
||
|
||
HRESULT CSCEnumCategoriesOfClass::Clone(IEnumGUID **ppenum)
|
||
{
|
||
HRESULT hr = S_OK;
|
||
CSCEnumCategoriesOfClass *pEnumClone = NULL;
|
||
|
||
if (!IsValidPtrOut(ppenum, sizeof(IEnumGUID *)))
|
||
return E_POINTER;
|
||
|
||
pEnumClone = new CSCEnumCategoriesOfClass;
|
||
if (pEnumClone == NULL)
|
||
return E_OUTOFMEMORY;
|
||
|
||
pEnumClone->Initialize(m_catid, m_cCatid);
|
||
pEnumClone->m_dwPosition = m_dwPosition;
|
||
if (SUCCEEDED(hr = pEnumClone->QueryInterface(IID_IEnumCATID, (void**) ppenum)))
|
||
return S_OK;
|
||
|
||
delete pEnumClone;
|
||
return hr;
|
||
}
|
||
|
||
|
||
CSCEnumCategoriesOfClass::CSCEnumCategoriesOfClass()
|
||
{
|
||
m_dwRefCount=0;
|
||
m_dwPosition = 0;
|
||
m_catid = NULL;
|
||
}
|
||
|
||
HRESULT CSCEnumCategoriesOfClass::Initialize(CATID catid[], ULONG cCatid)
|
||
{
|
||
ULONG i;
|
||
|
||
m_catid = new CATID[cCatid];
|
||
if (!m_catid)
|
||
return E_OUTOFMEMORY;
|
||
m_cCatid = cCatid;
|
||
for (i = 0; i < cCatid; i++)
|
||
m_catid[i] = catid[i];
|
||
m_dwPosition = 0;
|
||
return S_OK;
|
||
}
|
||
|
||
CSCEnumCategoriesOfClass::~CSCEnumCategoriesOfClass()
|
||
{
|
||
if (m_catid)
|
||
delete m_catid;
|
||
}
|
||
|
||
// CEnumClassesOfCategories:
|
||
// IUnknown methods
|
||
HRESULT CSCEnumClassesOfCategories::QueryInterface(REFIID riid, void** ppObject)
|
||
{
|
||
if (riid==IID_IUnknown || riid==IID_IEnumCLSID)
|
||
{
|
||
*ppObject=(IEnumCLSID*) this;
|
||
}
|
||
else
|
||
{
|
||
return E_NOINTERFACE;
|
||
}
|
||
AddRef();
|
||
return S_OK;
|
||
}
|
||
|
||
ULONG CSCEnumClassesOfCategories::AddRef()
|
||
{
|
||
InterlockedIncrement((long*) &m_dwRefCount);
|
||
return m_dwRefCount;
|
||
}
|
||
|
||
ULONG CSCEnumClassesOfCategories::Release()
|
||
{
|
||
ULONG dwRefCount;
|
||
if ((dwRefCount = InterlockedDecrement((long*) &m_dwRefCount))==0)
|
||
{
|
||
delete this;
|
||
return 0;
|
||
}
|
||
return dwRefCount;
|
||
}
|
||
|
||
// IEnumGUID methods
|
||
HRESULT CSCEnumClassesOfCategories::Next(ULONG celt, GUID *rgelt, ULONG *pceltFetched)
|
||
{
|
||
ULONG cRead;
|
||
ULONG i, dwCount, cgot;
|
||
HRESULT hr;
|
||
CLSID clsid;
|
||
IADs *pclsid;
|
||
WCHAR szClsid[_MAX_PATH];
|
||
|
||
if ((celt > 1) && (!pceltFetched))
|
||
return E_INVALIDARG;
|
||
|
||
if (pceltFetched)
|
||
(*pceltFetched) = 0;
|
||
|
||
if ((celt < 0) || (!IsValidPtrOut(rgelt, sizeof(GUID)*celt)))
|
||
return E_INVALIDARG;
|
||
|
||
for ((dwCount) = 0; (dwCount) < celt;) {
|
||
|
||
hr = GetNextEnum(m_pEnumVariant, &pclsid);
|
||
if ((FAILED(hr)) || (hr == S_FALSE))
|
||
{
|
||
if (pceltFetched)
|
||
*pceltFetched = dwCount;
|
||
return S_FALSE;
|
||
}
|
||
|
||
hr = GetProperty (pclsid, CLASSCLSID, szClsid);
|
||
pclsid->Release();
|
||
|
||
if (FAILED(hr)) {
|
||
if (pceltFetched)
|
||
*pceltFetched = dwCount;
|
||
return S_FALSE;
|
||
}
|
||
|
||
GUIDFromString(szClsid, &clsid);
|
||
|
||
if ((ImplSatisfied(clsid, m_cImplemented, m_rgcatidImpl, m_pICatInfo) == S_OK) &&
|
||
(ReqSatisfied(clsid, m_cRequired, m_rgcatidReq, m_pICatInfo) == S_OK))
|
||
{
|
||
rgelt[dwCount] = clsid;
|
||
(dwCount)++;
|
||
}
|
||
}
|
||
|
||
m_dwPosition += dwCount;
|
||
|
||
if (pceltFetched)
|
||
*pceltFetched = dwCount;
|
||
|
||
return S_OK;
|
||
}
|
||
|
||
HRESULT CSCEnumClassesOfCategories::Skip(ULONG celt)
|
||
{
|
||
CLSID *dummyclasses;
|
||
ULONG celtFetched;
|
||
HRESULT hr;
|
||
|
||
dummyclasses = new CLSID[celt];
|
||
hr = Next(celt, dummyclasses, &celtFetched);
|
||
delete dummyclasses;
|
||
return hr;
|
||
}
|
||
|
||
HRESULT CSCEnumClassesOfCategories::Reset()
|
||
{
|
||
LPOLESTR pszPathNames [2];
|
||
HRESULT hr;
|
||
|
||
pszPathNames [0] = CLASS_CS_CLASS;
|
||
|
||
if (m_pEnumVariant == NULL)
|
||
return E_UNEXPECTED;
|
||
m_pEnumVariant->Release();
|
||
|
||
hr = EnumCsObject(m_ADsClassContainer, &pszPathNames[0],
|
||
0, &m_pEnumVariant);
|
||
m_dwPosition = 0;
|
||
return hr;
|
||
}
|
||
|
||
HRESULT CSCEnumClassesOfCategories::Clone(IEnumGUID **ppenum)
|
||
{
|
||
HRESULT hr;
|
||
CSCEnumClassesOfCategories *pEnumClone;
|
||
|
||
if (!ppenum)
|
||
return E_INVALIDARG;
|
||
|
||
pEnumClone = new CSCEnumClassesOfCategories;
|
||
if (!pEnumClone)
|
||
return E_OUTOFMEMORY;
|
||
hr = pEnumClone->Initialize(m_cRequired, m_rgcatidReq,
|
||
m_cImplemented, m_rgcatidImpl,
|
||
m_ADsClassContainer,
|
||
m_pICatInfo);
|
||
if (FAILED(hr))
|
||
{
|
||
delete pEnumClone;
|
||
return hr;
|
||
}
|
||
pEnumClone->Skip(m_dwPosition);
|
||
|
||
hr = pEnumClone->QueryInterface(IID_IEnumCLSID, (void **)ppenum);
|
||
if (FAILED(hr))
|
||
{
|
||
delete pEnumClone;
|
||
}
|
||
|
||
return hr;
|
||
}
|
||
|
||
CSCEnumClassesOfCategories::CSCEnumClassesOfCategories()
|
||
{
|
||
m_dwRefCount = 0;
|
||
m_rgcatidReq = NULL;
|
||
m_rgcatidImpl = NULL;
|
||
m_pICatInfo = NULL;
|
||
m_ADsClassContainer = NULL;
|
||
m_pEnumVariant = NULL;
|
||
m_dwPosition = 0;
|
||
}
|
||
|
||
HRESULT CSCEnumClassesOfCategories::Initialize(ULONG cRequired, CATID rgcatidReq[],
|
||
ULONG cImplemented, CATID rgcatidImpl[],
|
||
IADsContainer *ADsClassContainer,
|
||
ICatInformation *pICatInfo)
|
||
{
|
||
ULONG i;
|
||
HRESULT hr;
|
||
LPOLESTR pszPathNames [2];
|
||
|
||
pszPathNames [0] = CLASS_CS_CLASS;
|
||
m_ADsClassContainer = ADsClassContainer;
|
||
m_ADsClassContainer->AddRef();
|
||
|
||
hr = EnumCsObject(m_ADsClassContainer, &pszPathNames[0], 0, &m_pEnumVariant);
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
m_pICatInfo = pICatInfo;
|
||
RETURN_ON_FAILURE(m_pICatInfo->AddRef());
|
||
|
||
m_cRequired = cRequired;
|
||
if (cRequired != -1)
|
||
{
|
||
m_rgcatidReq = new CATID[cRequired];
|
||
if (!m_rgcatidReq)
|
||
return E_OUTOFMEMORY;
|
||
for (i = 0; i < m_cRequired; i++)
|
||
m_rgcatidReq[i] = rgcatidReq[i];
|
||
}
|
||
|
||
m_cImplemented = cImplemented;
|
||
if (cImplemented != -1)
|
||
{
|
||
m_rgcatidImpl = new CATID[cImplemented];
|
||
if (!m_rgcatidImpl)
|
||
return E_OUTOFMEMORY;
|
||
for (i = 0; i < m_cImplemented; i++)
|
||
m_rgcatidImpl[i] = rgcatidImpl[i];
|
||
}
|
||
return S_OK;
|
||
}
|
||
|
||
CSCEnumClassesOfCategories::~CSCEnumClassesOfCategories()
|
||
{
|
||
if (m_rgcatidReq)
|
||
delete m_rgcatidReq;
|
||
if (m_rgcatidImpl)
|
||
delete m_rgcatidImpl;
|
||
if (m_pICatInfo)
|
||
m_pICatInfo->Release();
|
||
if (m_ADsClassContainer)
|
||
m_ADsClassContainer->Release();
|
||
if (m_pEnumVariant)
|
||
m_pEnumVariant->Release();
|
||
}
|
||
|
||
|