?

通過(guò)編程實(shí)踐可以發(fā)現(xiàn)，如果通過(guò)ATL向?qū)傻腃OM，自動(dòng)會(huì)生成DllRegisterServer及DllUnregisterServer函數(shù)，可供regsvr32等調(diào)用進(jìn)行

注冊(cè)與反注冊(cè)。而如果通過(guò)普通的DLL向?qū)Р⑦x擇Automation支持，則只會(huì)自動(dòng)生成DllRegisterServer，而沒(méi)有DllUnregisterServer接口，因

此需要手工添加一個(gè)DllUnregisterServer函數(shù)。當(dāng)試圖在此函數(shù)中調(diào)用COleObjectFactory::UnregisterAll()進(jìn)行反注冊(cè)時(shí)，跟蹤源代碼可知

最終只是簡(jiǎn)單地返回了TRUE，所以根本沒(méi)有反注冊(cè)。

為了能夠仿ATL實(shí)現(xiàn)COM的反注冊(cè)，需要研究ATL注冊(cè)/反注冊(cè)源代碼：
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為一個(gè)CComModule的全局對(duì)象，就類似于CXXXApp的全局對(duì)象，上面兩個(gè)函數(shù)是CComModule的成員函數(shù)，如下：
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);
}

由于注冊(cè)與反注冊(cè)是類似的，下面以普通支持自動(dòng)化的DLL中沒(méi)有的反注冊(cè)功能為例
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;
}

關(guān)鍵處是這句pEntry->pfnUpdateRegistry(FALSE); //unregister
那么pfnUpdateRegistry是哪來(lái)的呢？它是_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);
};

這個(gè)結(jié)構(gòu)是怎么構(gòu)造起來(lái)的呢？ATL采用了類似MFC中構(gòu)建message map及serialize的使用宏構(gòu)建列表的方法：
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 },
這三個(gè)宏可以構(gòu)建一個(gè)COM類靜態(tài)鏈表（數(shù)組），鏈表的元素就是上面的_ATL_OBJMAP_ENTRY結(jié)構(gòu)，它的第二個(gè)成員就是注冊(cè)時(shí)被調(diào)用的pfnUpda

teRegistry。

下面是一例：
BEGIN_OBJECT_MAP(ObjectMap)
OBJECT_ENTRY(CLSID_HtmlSelParse, CHtmlSelParse)
END_OBJECT_MAP()

通過(guò)上面的分析可以知道，pfnUpdateRegistry =

CHtmlSelParse::UpdateRegistry，那么UpdateRegistry又是什么呢？它是通過(guò)DECLARE_REGISTRY_RESOURCEID宏定義的一個(gè)靜態(tài)函數(shù)：
#define DECLARE_REGISTRY_RESOURCEID(x)\
?static HRESULT WINAPI UpdateRegistry(BOOL bRegister)\
?{\
?return _Module.UpdateRegistryFromResource(x, bRegister);\
?}
下面再來(lái)看UpdateRegistryFromResource：
#ifdef _ATL_STATIC_REGISTRY
#define UpdateRegistryFromResource UpdateRegistryFromResourceS
#else
#define UpdateRegistryFromResource UpdateRegistryFromResourceD
#endif

UpdateRegistryFromResourceS與UpdateRegistryFromResourceD分別為ATL靜態(tài)鏈接與動(dòng)態(tài)鏈接版本，下面看UpdateRegistryFromResourceD動(dòng)

態(tài)鏈接版本：
HRESULT WINAPI UpdateRegistryFromResourceD(LPCTSTR lpszRes, BOOL bRegister,
?struct _ATL_REGMAP_ENTRY* pMapEntries = NULL)
{
?USES_CONVERSION;
?return AtlModuleUpdateRegistryFromResourceD(this, T2COLE(lpszRes), bRegister,
??pMapEntries);
}

它調(diào)用的是AtlModuleUpdateRegistryFromResourceD，其位于ATLBASE.h中，下面是其原型：
ATLINLINE ATLAPI AtlModuleUpdateRegistryFromResourceD(_ATL_MODULE* pM, LPCOLESTR lpszRes,
?BOOL bRegister, struct _ATL_REGMAP_ENTRY* pMapEntries, IRegistrar* pReg)

查看它的源代碼就會(huì)發(fā)現(xiàn)，它調(diào)用的是IRegistrar::ResourceRegister/ResourceUnregister等進(jìn)行注冊(cè)與反注冊(cè)。IRegistar是什么？它應(yīng)該

是一個(gè)用來(lái)注冊(cè)COM的Shell接口，具體可查看MSDN的“The ATL Registry Component (Registrar)”主題。由于到了這里不能看到它的源代碼

了，所以不知道它的具體實(shí)現(xiàn)，但通過(guò)搜索ResourceRegister/ResourceUnregister，意外地發(fā)現(xiàn)它們也是另一個(gè)類CRegObject的成員函數(shù)。它

們最終調(diào)用的是CRegObject::RegisterFromResource，其中調(diào)用了CRegParser::RegisterBuffer，這個(gè)可以看作是注冊(cè)與反注冊(cè)的終極靶標(biāo)了

。通過(guò)查看這個(gè)函數(shù)，而且根據(jù)前面的函數(shù)名及需要傳遞的資源ID，你會(huì)恍然大悟：它實(shí)際上是通過(guò)解析ATL向?qū)傻膔gs文件實(shí)現(xiàn)注冊(cè)與反

注冊(cè)了，其根本操作就是添加或刪除注冊(cè)表項(xiàng)。這又回到了注冊(cè)與反注冊(cè)的最原始的方法了。

使用這種方法要求有一個(gè)rgs文件，并把它以資源方式添加到工程中，資源類型命名必須為"REGISTRY"(這是從函數(shù)的源代碼中可以看到)，函數(shù)

通過(guò)這個(gè)資源類型找到rgs文件。

關(guān)于rgs文件，下面是一小段解釋：

HKCR
{
??? NoRemove txtfile
??? {
??????? NoRemove ShellEx
??????? {
??????????? NoRemove ContextMenuHandlers
??????????? {
??????????????? ForceRemove SimpleShlExt = s '{5E2121EE-0300-11D4-8D3B-444553540000}'
??????????? }
??????? }
??? }
}
每一行代表一個(gè)注冊(cè)表鍵, "HKCR"是 HKEY_CLASSES_ROOT 的縮寫. NoRemove 關(guān)鍵字表示當(dāng)該COM服務(wù)器注銷時(shí)該鍵 不用被刪除. 最后一行有

些復(fù)雜. ForceRemove 關(guān)鍵字表示如果該鍵已存在, 那么在新鍵添加之前該鍵先應(yīng)被刪除. 這行腳本的余下部分指定一個(gè)字符串，它將被存為

SimpleShlExt 鍵的默認(rèn)值.

下面是兩種方式來(lái)實(shí)現(xiàn)普通支持Automation的DLL的反注冊(cè)代碼：

需要寫一個(gè)rgs文件，并把它作為資源添加到工程中，假設(shè)資源類型為“REGISTRY”，ID為IDR_WFDOWNLOAD，其內(nèi)容是類似下面的：
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實(shí)現(xiàn)反注冊(cè)（代碼摘抄自ATLBASE.h中的AtlModuleUpdateRegistryFromResourceD函數(shù)）

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;
?
}

總結(jié)

以上是生活随笔為你收集整理的关于VC向导生成的COM的注册与反注册的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。