|
通过编程实践可以发现，如果通过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 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;
}
|
