关于VC向导生成的COM的注册与反注册
通過編程實踐可以發現，如果通過ATL向導生成的COM，自動會生成DllRegisterServer及DllUnregisterServer函數，可供regsvr32等調用進行
注冊與反注冊。而如果通過普通的DLL向導并選擇Automation支持，則只會自動生成DllRegisterServer，而沒有DllUnregisterServer接口，因
此需要手工添加一個DllUnregisterServer函數。當試圖在此函數中調用COleObjectFactory::UnregisterAll()進行反注冊時，跟蹤源代碼可知
最終只是簡單地返回了TRUE，所以根本沒有反注冊。
為了能夠仿ATL實現COM的反注冊，需要研究ATL注冊/反注冊源代碼：
STDAPI DllRegisterServer(void)
{
??? // registers object, typelib and all interfaces in typelib
??? return _Module.RegisterServer(TRUE);
}
/
// DllUnregisterServer - Removes entries from the system registry
STDAPI DllUnregisterServer(void)
{
??? return _Module.UnregisterServer(TRUE);
}
_Module為一個CComModule的全局對象，就類似于CXXXApp的全局對象，上面兩個函數是CComModule的成員函數，如下：
HRESULT RegisterServer(BOOL bRegTypeLib = FALSE, const CLSID* pCLSID = NULL)
{
?return AtlModuleRegisterServer(this, bRegTypeLib, pCLSID);
}
HRESULT UnregisterServer(BOOL bUnRegTypeLib, const CLSID* pCLSID = NULL)
{
?return AtlModuleUnregisterServerEx(this, bUnRegTypeLib, pCLSID);
}
由于注冊與反注冊是類似的，下面以普通支持自動化的DLL中沒有的反注冊功能為例
ATLINLINE ATLAPI AtlModuleUnregisterServerEx(_ATL_MODULE* pM, BOOL bUnRegTypeLib, const CLSID* pCLSID)
{
?ATLASSERT(pM != NULL);
?if (pM == NULL)
??return E_INVALIDARG;
?ATLASSERT(pM->m_hInst != NULL);
?ATLASSERT(pM->m_pObjMap != NULL);
?_ATL_OBJMAP_ENTRY* pEntry = pM->m_pObjMap;
?for (;pEntry->pclsid != NULL; pEntry = _NextObjectMapEntry(pM, pEntry))
?{
??if (pCLSID == NULL)
??{
???if (pEntry->pfnGetObjectDescription != NULL
????&& pEntry->pfnGetObjectDescription() != NULL)
????continue;
??}
??else
??{
???if (!IsEqualGUID(*pCLSID, *pEntry->pclsid))
????continue;
??}
??pEntry->pfnUpdateRegistry(FALSE); //unregister
??if (pM->cbSize == sizeof(_ATL_MODULE) && pEntry->pfnGetCategoryMap != NULL)
???AtlRegisterClassCategoriesHelper( *pEntry->pclsid,
????pEntry->pfnGetCategoryMap(), FALSE );
?}
?if (bUnRegTypeLib)
??AtlModuleUnRegisterTypeLib(pM, 0);
?return S_OK;
}
關鍵處是這句pEntry->pfnUpdateRegistry(FALSE); //unregister
那么pfnUpdateRegistry是哪來的呢？它是_ATL_OBJMAP_ENTRY中的一員：
struct _ATL_OBJMAP_ENTRY
{
?const CLSID* pclsid;
?HRESULT (WINAPI *pfnUpdateRegistry)(BOOL bRegister);????? // ------------- 這里
?_ATL_CREATORFUNC* pfnGetClassObject;
?_ATL_CREATORFUNC* pfnCreateInstance;
?IUnknown* pCF;
?DWORD dwRegister;
?_ATL_DESCRIPTIONFUNC* pfnGetObjectDescription;
?_ATL_CATMAPFUNC* pfnGetCategoryMap;
?HRESULT WINAPI RevokeClassObject()
?{
??return CoRevokeClassObject(dwRegister);
?}
?HRESULT WINAPI RegisterClassObject(DWORD dwClsContext, DWORD dwFlags)
?{
??IUnknown* p = NULL;
??if (pfnGetClassObject == NULL)
???return S_OK;
??HRESULT hRes = pfnGetClassObject(pfnCreateInstance, IID_IUnknown, (LPVOID*) &p);
??if (SUCCEEDED(hRes))
???hRes = CoRegisterClassObject(*pclsid, p, dwClsContext, dwFlags, &dwRegister);
??if (p != NULL)
???p->Release();
??return hRes;
?}
// Added in ATL 3.0
?void (WINAPI *pfnObjectMain)(bool bStarting);
};
這個結構是怎么構造起來的呢？ATL采用了類似MFC中構建message map及serialize的使用宏構建列表的方法：
BEGIN_OBJECT_MAP/OBJECT_ENTRY/END_OBJECT_MAP
#define BEGIN_OBJECT_MAP(x) static _ATL_OBJMAP_ENTRY x[] = {
#define END_OBJECT_MAP()?? {NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL}};
#define OBJECT_ENTRY(clsid, class) {&clsid, class::UpdateRegistry, class::_ClassFactoryCreatorClass::CreateInstance,
class::_CreatorClass::CreateInstance, NULL, 0, class::GetObjectDescription, class::GetCategoryMap, class::ObjectMain },
這三個宏可以構建一個COM類靜態鏈表（數組），鏈表的元素就是上面的_ATL_OBJMAP_ENTRY結構，它的第二個成員就是注冊時被調用的pfnUpda
teRegistry。
下面是一例：
BEGIN_OBJECT_MAP(ObjectMap)
OBJECT_ENTRY(CLSID_HtmlSelParse, CHtmlSelParse)
END_OBJECT_MAP()
通過上面的分析可以知道，pfnUpdateRegistry =
CHtmlSelParse::UpdateRegistry，那么UpdateRegistry又是什么呢？它是通過DECLARE_REGISTRY_RESOURCEID宏定義的一個靜態函數：
#define DECLARE_REGISTRY_RESOURCEID(x)\
?static HRESULT WINAPI UpdateRegistry(BOOL bRegister)\
?{\
?return _Module.UpdateRegistryFromResource(x, bRegister);\
?}
下面再來看UpdateRegistryFromResource：
#ifdef _ATL_STATIC_REGISTRY
#define UpdateRegistryFromResource UpdateRegistryFromResourceS
#else
#define UpdateRegistryFromResource UpdateRegistryFromResourceD
#endif
UpdateRegistryFromResourceS與UpdateRegistryFromResourceD分別為ATL靜態鏈接與動態鏈接版本，下面看UpdateRegistryFromResourceD動
態鏈接版本：
HRESULT WINAPI UpdateRegistryFromResourceD(LPCTSTR lpszRes, BOOL bRegister,
?struct _ATL_REGMAP_ENTRY* pMapEntries = NULL)
{
?USES_CONVERSION;
?return AtlModuleUpdateRegistryFromResourceD(this, T2COLE(lpszRes), bRegister,
??pMapEntries);
}
它調用的是AtlModuleUpdateRegistryFromResourceD，其位于ATLBASE.h中，下面是其原型：
ATLINLINE ATLAPI AtlModuleUpdateRegistryFromResourceD(_ATL_MODULE* pM, LPCOLESTR lpszRes,
?BOOL bRegister, struct _ATL_REGMAP_ENTRY* pMapEntries, IRegistrar* pReg)
查看它的源代碼就會發現，它調用的是IRegistrar::ResourceRegister/ResourceUnregister等進行注冊與反注冊。IRegistar是什么？它應該
是一個用來注冊COM的Shell接口，具體可查看MSDN的“The ATL Registry Component (Registrar)”主題。由于到了這里不能看到它的源代碼
了，所以不知道它的具體實現，但通過搜索ResourceRegister/ResourceUnregister，意外地發現它們也是另一個類CRegObject的成員函數。它
們最終調用的是CRegObject::RegisterFromResource，其中調用了CRegParser::RegisterBuffer，這個可以看作是注冊與反注冊的終極靶標了
。通過查看這個函數，而且根據前面的函數名及需要傳遞的資源ID，你會恍然大悟：它實際上是通過解析ATL向導生成的rgs文件實現注冊與反
注冊了，其根本操作就是添加或刪除注冊表項。這又回到了注冊與反注冊的最原始的方法了。
使用這種方法要求有一個rgs文件，并把它以資源方式添加到工程中，資源類型命名必須為"REGISTRY"(這是從函數的源代碼中可以看到)，函數
通過這個資源類型找到rgs文件。
關于rgs文件，下面是一小段解釋：
HKCR
{
??? NoRemove txtfile
??? {
??????? NoRemove ShellEx
??????? {
??????????? NoRemove ContextMenuHandlers
??????????? {
??????????????? ForceRemove SimpleShlExt = s '{5E2121EE-0300-11D4-8D3B-444553540000}'
??????????? }
??????? }
??? }
}
每一行代表一個注冊表鍵, "HKCR"是 HKEY_CLASSES_ROOT 的縮寫. NoRemove 關鍵字表示當該COM服務器注銷時該鍵 不用被刪除. 最后一行有
些復雜. ForceRemove 關鍵字表示如果該鍵已存在, 那么在新鍵添加之前該鍵先應被刪除. 這行腳本的余下部分指定一個字符串，它將被存為
SimpleShlExt 鍵的默認值.
下面是兩種方式來實現普通支持Automation的DLL的反注冊代碼：
需要寫一個rgs文件，并把它作為資源添加到工程中，假設資源類型為“REGISTRY”，ID為IDR_WFDOWNLOAD，其內容是類似下面的：
HKCR
{
?WFDownload.AddURL = s 'WFDownload.AddURL'
?{
??CLSID = s '{C7CFF70F-3F33-4F34-ACEF-CFCE14F1792D}'
?}
?NoRemove CLSID
?{
??ForceRemove {C7CFF70F-3F33-4F34-ACEF-CFCE14F1792D} = s 'WFDownload.AddURL'
??{
???ProgID = s 'WFDownload.AddURL'
???InprocServer32 = s '%MODULE%'
??}
?}
}
方法一：利用IRegistrar實現反注冊（代碼摘抄自ATLBASE.h中的AtlModuleUpdateRegistryFromResourceD函數）
STDAPI DllUnregisterServer(void)
{
?AFX_MANAGE_STATE(AfxGetStaticModuleState());
?USES_CONVERSION;
?HRESULT hRes = S_OK;
?CComPtr<IRegistrar> p;
?hRes = CoCreateInstance(CLSID_Registrar, NULL,
???CLSCTX_INPROC_SERVER, IID_IRegistrar, (void**)&p);
?if (SUCCEEDED(hRes))
?{
??TCHAR szModule[_MAX_PATH];
??GetModuleFileName(AfxGetInstanceHandle(), szModule, _MAX_PATH);
?
??LPOLESTR pszModule;
??pszModule = T2OLE(szModule);
??int nLen = ocslen(pszModule);
??LPOLESTR pszModuleQuote = (LPOLESTR)alloca((nLen*2+1)*sizeof(OLECHAR));
??CComModule::ReplaceSingleQuote(pszModuleQuote, pszModule);
??p->AddReplacement(OLESTR("Module"), pszModuleQuote);
??LPCOLESTR szType = OLESTR("REGISTRY");
??LPCOLESTR lpszRes = (LPCOLESTR)MAKEINTRESOURCE(IDR_WFDOWNLOAD);
??if (HIWORD(lpszRes)==0)
??{
???hRes = p->ResourceUnregister(pszModule, ((UINT)LOWORD((DWORD)lpszRes)), szType);
??}
??else
??{
??hRes = p->ResourceUnregisterSz(pszModule, lpszRes, szType);
??}
??
?}
?return hRes;
}
方法二：利用CRegObject
STDAPI DllUnregisterServer(void)
{
?AFX_MANAGE_STATE(AfxGetStaticModuleState());
?USES_CONVERSION;
?HRESULT hRes = S_OK;
?TCHAR szModule[_MAX_PATH];
?GetModuleFileName(AfxGetInstanceHandle(), szModule, _MAX_PATH);
?
?LPOLESTR pszModule;
?pszModule = T2OLE(szModule);
?int nLen = ocslen(pszModule);
?LPOLESTR pszModuleQuote = (LPOLESTR)alloca((nLen*2+1)*sizeof(OLECHAR));
?CComModule::ReplaceSingleQuote(pszModuleQuote, pszModule);
?LPCOLESTR szType = OLESTR("REGISTRY");
?
?CRegObject objReg;
?objReg.AddReplacement(OLESTR("Module"), pszModule);
?hRes = objReg.ResourceUnregister(pszModule, IDR_WFDOWNLOAD, szType);
?return hRes;
?
}
總結
以上是生活随笔為你收集整理的关于VC向导生成的COM的注册与反注册的全部內容，希望文章能夠幫你解決所遇到的問題。
- 上一篇： 为treeview添加客户端事件
- 下一篇： 在VC++中创建DLL文件并加载