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.
448 lines
14 KiB
448 lines
14 KiB
//
|
|
// Author: ushaji
|
|
// Date: December/Jan 1996
|
|
//
|
|
//
|
|
// Providing support for Component Categories in Class Store
|
|
//
|
|
// This source file contains implementations for ICatRegister interfaces. ¦
|
|
//
|
|
// Refer Doc "Design for Support of File Types and Component Categories
|
|
// in Class Store" ? (or may be Class Store Schema)
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
#include "cstore.hxx"
|
|
|
|
//-------------------------------------------------------------
|
|
// RegisterCategories:
|
|
// registering categories in the class store.
|
|
// cCategories: Number of Categories
|
|
// rgCategoryInfo: Size cCategories
|
|
//
|
|
// Returns as soon as one of them fails.
|
|
//-------------------------------------------------------------
|
|
HRESULT STDMETHODCALLTYPE CClassContainer::RegisterCategories(
|
|
/* [in] */ ULONG cCategories,
|
|
/* [size_is][in] */ CATEGORYINFO __RPC_FAR rgCategoryInfo[ ])
|
|
{
|
|
/* Registering a list of Category ID in the Class Store */
|
|
|
|
HRESULT hr;
|
|
IADs *pADs = NULL;
|
|
IDispatch *pUnknown = NULL;
|
|
STRINGGUIDRDN szCatid;
|
|
ULONG i, j, cdesc, posn;
|
|
LPOLESTR *pszDescExisting, *pszDesc;
|
|
WCHAR *localedescription = NULL;
|
|
// sizeof description + seperator length + locale in hex
|
|
|
|
/* BUGBUG::Should check whether write permissions exist? */
|
|
if (!IsValidReadPtrIn(this, sizeof(*this))) {
|
|
return E_ACCESSDENIED;
|
|
}
|
|
|
|
if (!IsValidReadPtrIn(rgCategoryInfo, sizeof(rgCategoryInfo[0])*cCategories))
|
|
{
|
|
return E_INVALIDARG; // gd
|
|
}
|
|
|
|
if (!m_fOpen)
|
|
return E_FAIL;
|
|
|
|
for (i = 0; i < cCategories; i++)
|
|
{
|
|
localedescription = (WCHAR *)CoTaskMemAlloc(sizeof(WCHAR)*(128+16));
|
|
if (!localedescription)
|
|
return E_OUTOFMEMORY;
|
|
|
|
RdnFromGUID(rgCategoryInfo[i].catid, szCatid);
|
|
|
|
wsprintf(localedescription, L"%x %s %s", rgCategoryInfo[i].lcid, CATSEPERATOR,
|
|
rgCategoryInfo[i].szDescription);
|
|
|
|
hr = m_ADsCategoryContainer->GetObject(NULL, szCatid, (IDispatch **)&pADs);
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = GetPropertyListAlloc (pADs, LOCALEDESCRIPTION, &cdesc, &pszDescExisting);
|
|
RETURN_ON_FAILURE(hr);
|
|
// Existing list of descriptions
|
|
|
|
pszDesc = (LPOLESTR *)CoTaskMemAlloc(sizeof(LPOLESTR)*(cdesc+1));
|
|
if (!pszDesc)
|
|
return E_OUTOFMEMORY;
|
|
for (j = 0; j < cdesc; j++)
|
|
pszDesc[j] = pszDescExisting[j];
|
|
|
|
if (!(posn = FindDescription(pszDescExisting, cdesc, &(rgCategoryInfo[i].lcid), NULL, 0)))
|
|
{
|
|
// if no description exists for the lcid.
|
|
pszDesc[cdesc] = localedescription;
|
|
cdesc++;
|
|
}
|
|
else
|
|
{ // overwrite the old value
|
|
CoTaskMemFree(pszDesc[posn-1]);
|
|
pszDesc[posn-1] = localedescription;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
hr = m_ADsCategoryContainer->Create(
|
|
CLASS_CS_CATEGORY,
|
|
szCatid,
|
|
&pUnknown
|
|
);
|
|
|
|
RETURN_ON_FAILURE(hr);
|
|
pszDesc = (LPOLESTR *)CoTaskMemAlloc(sizeof(LPOLESTR));
|
|
if (!pszDesc)
|
|
return E_OUTOFMEMORY;
|
|
cdesc = 1;
|
|
pszDesc[0] = localedescription;
|
|
|
|
hr = pUnknown->QueryInterface(IID_IADs, (void **)&pADs);
|
|
RETURN_ON_FAILURE(hr);
|
|
|
|
pUnknown->Release();
|
|
}
|
|
|
|
// StringFromGUID(rgCategoryInfo[i].catid, szCatid);
|
|
// SetProperty(pADs, CATEGORYCATID, szCatid);
|
|
hr = SetPropertyGuid(pADs, CATEGORYCATID, rgCategoryInfo[i].catid);
|
|
RETURN_ON_FAILURE(hr);
|
|
|
|
SetPropertyList(pADs, LOCALEDESCRIPTION, cdesc, pszDesc);
|
|
for (j = 0; j < cdesc; j++)
|
|
CoTaskMemFree(pszDesc[j]);
|
|
CoTaskMemFree(pszDesc);
|
|
RETURN_ON_FAILURE(hr);
|
|
|
|
hr = StoreIt (pADs);
|
|
RETURN_ON_FAILURE(hr);
|
|
|
|
pADs->Release();
|
|
}
|
|
return hr;
|
|
} /* RegisterCategories */
|
|
|
|
|
|
//--------------------------------------------------------
|
|
// Unregistering categories from the class store
|
|
// cCategories: Number of Categories
|
|
// rgcatid: catids of the categories.
|
|
//
|
|
// Stops after any one of them returns a error.
|
|
// Doesn't remove the category ids from each of the class ids.
|
|
|
|
HRESULT STDMETHODCALLTYPE CClassContainer::UnRegisterCategories(
|
|
/* [in] */ ULONG cCategories,
|
|
/* [size_is][in] */ CATID __RPC_FAR rgcatid[ ])
|
|
{
|
|
STRINGGUIDRDN szCatid;
|
|
ULONG i;
|
|
HRESULT hr;
|
|
|
|
|
|
if (!IsValidPtrOut(this, sizeof(*this))){
|
|
return E_ACCESSDENIED;
|
|
}
|
|
if (!IsValidReadPtrIn(rgcatid, sizeof(rgcatid[0])*cCategories)){
|
|
return E_INVALIDARG;
|
|
}
|
|
// Checking whether the caller has permissions
|
|
|
|
if (!m_fOpen)
|
|
return E_FAIL;
|
|
|
|
for (i = 0; i < cCategories; i++)
|
|
{
|
|
RdnFromGUID(rgcatid[i], szCatid);
|
|
|
|
hr = m_ADsCategoryContainer->Delete(CLASS_CS_CATEGORY,
|
|
szCatid);
|
|
}
|
|
return S_OK;
|
|
} /* UnRegisterCategories */
|
|
|
|
//------------------------------------------------------------------
|
|
// RegisterClassXXXCategories:
|
|
// rclsid: This category will be registered with this clsid.
|
|
// cCategories: The number of categories to be added.
|
|
// rgcatid The categories to be added (cCategories)
|
|
// impl_or_req The property to which this category will be added.
|
|
// "Implemented Categories" or "Required Categories"
|
|
//
|
|
//
|
|
// add all the categories given to the class store for this class.
|
|
// The previous entries will be lost and on error it would not be
|
|
// restored or made empty. A PRIVATE METHOD called by the 2 public methods.
|
|
//------------------------------------------------------------------
|
|
|
|
HRESULT STDMETHODCALLTYPE CClassContainer::RegisterClassXXXCategories(
|
|
/* [in] */ REFCLSID rclsid,
|
|
/* [in] */ ULONG cCategories,
|
|
/* [size_is][in] */ CATID __RPC_FAR rgcatid[ ],BSTR impl_or_req)
|
|
{
|
|
HRESULT hr;
|
|
STRINGGUIDRDN szName;
|
|
IADs *pADs = NULL;
|
|
ULONG i, j, tobeadded;
|
|
VARIANT Var;
|
|
GUID *pCatids, *pOldCatids=NULL;
|
|
IDispatch *pUnknown = NULL;
|
|
STRINGGUID szGUID;
|
|
|
|
if (!m_fOpen)
|
|
return E_FAIL;
|
|
|
|
if (!IsValidPtrOut(this, sizeof(*this))) {
|
|
return E_ACCESSDENIED;
|
|
}
|
|
|
|
if (!IsValidReadPtrIn(rgcatid, sizeof(rgcatid[0])*cCategories)){
|
|
return E_INVALIDARG; // gd
|
|
}
|
|
|
|
if (IsNullGuid(rclsid))
|
|
return E_INVALIDARG;
|
|
|
|
// Get the ADs interface corresponding to the clsid that is mentioned.
|
|
StringFromGUID(rclsid, szGUID);
|
|
wsprintf(szName, L"CN=%s", szGUID);
|
|
|
|
hr = m_ADsClassContainer->GetObject(NULL,
|
|
szName,
|
|
(IDispatch **)&pADs
|
|
);
|
|
|
|
// if it didn't exist create it.
|
|
if (FAILED(hr)) {
|
|
hr = m_ADsClassContainer->Create(
|
|
CLASS_CS_CLASS,
|
|
szName,
|
|
&pUnknown
|
|
);
|
|
|
|
RETURN_ON_FAILURE(hr);
|
|
|
|
hr = pUnknown->QueryInterface(
|
|
IID_IADs,
|
|
(void **)&pADs
|
|
);
|
|
|
|
pUnknown->Release();
|
|
}
|
|
|
|
hr = SetProperty (pADs, CLASSCLSID, szGUID);
|
|
RETURN_ON_FAILURE(hr);
|
|
|
|
hr = GetPropertyListAllocGuid(pADs, impl_or_req, &tobeadded, &pOldCatids);
|
|
RETURN_ON_FAILURE(hr);
|
|
|
|
pCatids = (GUID *)CoTaskMemAlloc(sizeof(GUID)*(tobeadded+cCategories));
|
|
if (!pCatids)
|
|
return E_OUTOFMEMORY;
|
|
|
|
for (i = 0; i < tobeadded; i++)
|
|
pCatids[i] = pOldCatids[i];
|
|
|
|
for (i=0; i < cCategories; ++i) {
|
|
for (j = 0; j < tobeadded; j++)
|
|
{
|
|
if (memcmp(&rgcatid[i], &pCatids[j], sizeof(GUID)) == 0)
|
|
break;
|
|
}
|
|
if (j < tobeadded)
|
|
continue;
|
|
// The ith element is already there in the array.
|
|
// Make sure of this when the property name changes.
|
|
|
|
pCatids[tobeadded] = rgcatid[i];
|
|
tobeadded++;
|
|
}
|
|
|
|
VariantInit(&Var);
|
|
hr = PackGuidArray2Variant(pCatids, tobeadded, &Var);
|
|
RETURN_ON_FAILURE(hr);
|
|
|
|
if (tobeadded)
|
|
hr = pADs->Put(impl_or_req, Var);
|
|
else
|
|
hr = pADs->PutEx(ADS_PROPERTY_CLEAR, impl_or_req, Var);
|
|
|
|
RETURN_ON_FAILURE(hr);
|
|
|
|
VariantClear(&Var);
|
|
|
|
// save the data modified
|
|
hr = StoreIt(pADs);
|
|
pADs->Release();
|
|
|
|
CoTaskMemFree(pCatids);
|
|
if (pOldCatids)
|
|
CoTaskMemFree(pOldCatids);
|
|
return hr;
|
|
} /* RegisterClassXXXCategories */
|
|
|
|
|
|
//---------------------------------------------------------------------
|
|
// UnRegisterClassXXXCategories
|
|
// rclsid: classid from which the categories have to be removed.
|
|
// cCategories: Number of Categories
|
|
// rgcatid: Categories
|
|
// impl_or_req: The property to which this has to be added.
|
|
//
|
|
// Presently gets all the categories from the class. parses through it
|
|
// removes the ones that match in rgcatid and reregister the category.
|
|
//---------------------------------------------------------------------
|
|
|
|
HRESULT STDMETHODCALLTYPE CClassContainer::UnRegisterClassXXXCategories(
|
|
/* [in] */ REFCLSID rclsid,
|
|
/* [in] */ ULONG cCategories,
|
|
/* [size_is][in] */ CATID __RPC_FAR rgcatid[ ],BSTR impl_or_req)
|
|
{
|
|
HRESULT hr;
|
|
STRINGGUIDRDN szName;
|
|
IADs *pADs = NULL;
|
|
ULONG i, j;
|
|
ULONG cNewCatids, cOldCatids;
|
|
VARIANT Var;
|
|
GUID *pOldCatids=NULL, *pNewCatids;
|
|
|
|
// BUGBUG:: Have to decide some way of removing clsids once all categories
|
|
// are unregistered.
|
|
|
|
if (!m_fOpen)
|
|
return E_FAIL;
|
|
|
|
if (IsNullGuid(rclsid))
|
|
return E_INVALIDARG;
|
|
|
|
if (cCategories == 0)
|
|
return S_OK;
|
|
|
|
if (!IsValidPtrOut(this, sizeof(*this))) {
|
|
return E_ACCESSDENIED;
|
|
}
|
|
|
|
if (!IsValidReadPtrIn(rgcatid, sizeof(rgcatid[0])*cCategories)){
|
|
return E_INVALIDARG; // gd
|
|
}
|
|
|
|
if (cCategories == 0)
|
|
return S_OK;
|
|
|
|
// Get all the catids corresp to this clsid.
|
|
// Get the ADs interface corresponding to the clsid that is mentioned.
|
|
RdnFromGUID(rclsid, szName);
|
|
hr = m_ADsClassContainer->GetObject(NULL,
|
|
szName,
|
|
(IDispatch **)&pADs
|
|
);
|
|
|
|
RETURN_ON_FAILURE(hr);
|
|
|
|
hr = GetPropertyListAllocGuid(pADs,
|
|
impl_or_req,
|
|
&cOldCatids,
|
|
&pOldCatids);
|
|
|
|
RETURN_ON_FAILURE(hr);
|
|
|
|
// parse through this list and delete all the catids that is part of
|
|
// the user supplied list.
|
|
|
|
pNewCatids = (GUID *)CoTaskMemAlloc(sizeof(GUID)*cOldCatids);
|
|
if (!pNewCatids)
|
|
return E_OUTOFMEMORY;
|
|
|
|
for (i = 0, cNewCatids = 0; i < cOldCatids; i++)
|
|
{
|
|
for (j = 0; j < cCategories; j++)
|
|
if (memcmp(&pOldCatids[i], &rgcatid[j], sizeof(GUID)) == 0)
|
|
break;
|
|
if (j == cCategories)
|
|
pNewCatids[cNewCatids++] = pOldCatids[i];
|
|
}
|
|
|
|
// reregister this.
|
|
|
|
VariantInit(&Var);
|
|
hr = PackGuidArray2Variant(pNewCatids, cNewCatids, &Var);
|
|
RETURN_ON_FAILURE(hr);
|
|
|
|
if (cNewCatids)
|
|
hr = pADs->Put(impl_or_req, Var);
|
|
else
|
|
hr = pADs->PutEx(ADS_PROPERTY_CLEAR, impl_or_req, Var);
|
|
|
|
RETURN_ON_FAILURE(hr);
|
|
|
|
VariantClear(&Var);
|
|
|
|
// save the data modified
|
|
hr = StoreIt(pADs);
|
|
pADs->Release();
|
|
|
|
CoTaskMemFree(pNewCatids);
|
|
if (pOldCatids)
|
|
CoTaskMemFree(pOldCatids);
|
|
|
|
return hr;
|
|
} /* UnRegisterClassXXXCategories */
|
|
|
|
|
|
|
|
HRESULT STDMETHODCALLTYPE CClassContainer::RegisterClassImplCategories(
|
|
/* [in] */ REFCLSID rclsid,
|
|
/* [in] */ ULONG cCategories,
|
|
/* [size_is][in] */ CATID __RPC_FAR rgcatid[ ])
|
|
{
|
|
return RegisterClassXXXCategories(rclsid, cCategories, rgcatid,
|
|
IMPL_CATEGORIES);
|
|
} /* RegisterClassImplCategories */
|
|
|
|
|
|
HRESULT STDMETHODCALLTYPE CClassContainer::UnRegisterClassImplCategories(
|
|
/* [in] */ REFCLSID rclsid,
|
|
/* [in] */ ULONG cCategories,
|
|
/* [size_is][in] */ CATID __RPC_FAR rgcatid[ ])
|
|
{
|
|
|
|
return UnRegisterClassXXXCategories(rclsid, cCategories, rgcatid,
|
|
IMPL_CATEGORIES);
|
|
|
|
} /* UnRegisterClassImplCategories */
|
|
|
|
|
|
|
|
|
|
HRESULT STDMETHODCALLTYPE CClassContainer::RegisterClassReqCategories(
|
|
/* [in] */ REFCLSID rclsid,
|
|
/* [in] */ ULONG cCategories,
|
|
/* [size_is][in] */ CATID __RPC_FAR rgcatid[ ])
|
|
{
|
|
return RegisterClassXXXCategories(rclsid, cCategories, rgcatid,
|
|
REQ_CATEGORIES);
|
|
|
|
} /* RegisterClassReqCategories */
|
|
|
|
|
|
|
|
|
|
HRESULT STDMETHODCALLTYPE CClassContainer::UnRegisterClassReqCategories(
|
|
/* [in] */ REFCLSID rclsid,
|
|
/* [in] */ ULONG cCategories,
|
|
/* [size_is][in] */ CATID __RPC_FAR rgcatid[ ])
|
|
{
|
|
return UnRegisterClassXXXCategories(rclsid, cCategories, rgcatid,
|
|
REQ_CATEGORIES);
|
|
|
|
} /* UnRegisterClassReqCategories */
|
|
|
|
|
|
|
|
//--------------------------------------------------------------------
|
|
|