#include <comcat.h>
#include <afxctl.h>
#include <objsafe.h>
...

/
// DllRegisterServer - Adds entries to the system registry
/* 原來的代碼(被注釋)
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);
}
*/

//用途: 注冊組件分類
//說明: 組件安全性種類就是通過組件分類的注冊才得以設(shè)置組件的安全性
HRESULT CreateComponentCategory(CATID catid, WCHAR* catDescription){
????//初始化ICatRegister對象實例指針
????ICatRegister* pcr = NULL ;
????HRESULT hr = S_OK ;

????//獲得ICatRegister對象實例
//說明: 這個接口原來注冊組件分類
????hr = CoCreateInstance(CLSID_StdComponentCategoriesMgr,
?????NULL, CLSCTX_INPROC_SERVER, IID_ICatRegister, (void**)&pcr);
????if (FAILED(hr))
??return hr;

????//初始化分類信息
//說明: 確保注冊 HKCR/Component Categories/{..catid...}
????CATEGORYINFO catinfo;
????catinfo.catid = catid;
????catinfo.lcid = 0x0409 ; // english

????//說明: 確保提供的描述不會超長, 僅僅復(fù)制前127個字符
????int len = wcslen(catDescription);
????if (len>127)
??len = 127;
????wcsncpy(catinfo.szDescription, catDescription, len);
??// 確保描述使用"/0"結(jié)束
??catinfo.szDescription[len] = '/0';

????hr = pcr->RegisterCategories(1, &catinfo);
??pcr->Release();

??return hr;
}


//用途: 在已經(jīng)存在的組件分類中進(jìn)行接口類的注冊
HRESULT RegisterCLSIDInCategory(REFCLSID clsid, CATID catid) {
????ICatRegister* pcr = NULL ;
????HRESULT hr = S_OK ;

//獲得CCM的ICatRegister接口
????hr = CoCreateInstance(CLSID_StdComponentCategoriesMgr,
???NULL, CLSCTX_INPROC_SERVER, IID_ICatRegister, (void**)&pcr);
????if (SUCCEEDED(hr))
????{
???????//在組件分類之中注冊接口類
???????CATID rgcatid[1] ;
???????rgcatid[0] = catid;
???????hr = pcr->RegisterClassImplCategories(clsid, 1, rgcatid);
????}

????if (pcr != NULL)
??pcr->Release();

??return hr;
}

//用途: 反注冊已存在組件分類中的接口類
HRESULT UnRegisterCLSIDInCategory(REFCLSID clsid, CATID catid){
????ICatRegister* pcr = NULL ;
????HRESULT hr = S_OK ;

//獲得CCM的ICatRegister接口
????hr = CoCreateInstance(CLSID_StdComponentCategoriesMgr,
?????NULL, CLSCTX_INPROC_SERVER, IID_ICatRegister, (void**)&pcr);
????if (SUCCEEDED(hr))
????{
???????//注銷已存在的組件分類中的接口類
???????CATID rgcatid[1] ;
???????rgcatid[0] = catid;
???????hr = pcr->UnRegisterClassImplCategories(clsid, 1, rgcatid);
????}

????if (pcr != NULL)
??pcr->Release();

??return hr;
}

//注冊服務(wù)器
STDAPI DllRegisterServer(void){
????HRESULT??hr;????// return for safety functions

????AFX_MANAGE_STATE(_afxModuleAddrThis);
/* 原來的代碼
????if (!AfxOleRegisterTypeLib(AfxGetInstanceHandle(), _tlid))
??return ResultFromScode(SELFREG_E_TYPELIB);

????if (!COleObjectFactoryEx::UpdateRegistryAll(TRUE))
??return ResultFromScode(SELFREG_E_CLASS);
*/
????if (FAILED(_Module.RegisterServer(TRUE))){
??AfxMessageBox("注冊類型庫失敗");
}

???//創(chuàng)建初始化安全組件分類
????hr = CreateComponentCategory(CATID_SafeForInitializing,
??L"Controls safely initializable from persistent data!");
????if (FAILED(hr))
??return hr;
//在上面的分組之中注冊接口類
????hr = RegisterCLSIDInCategory(CLSID_psSub, CATID_SafeForInitializing);
????if (FAILED(hr))
??return hr;

????//創(chuàng)建腳本編程安全組件分類
????hr = CreateComponentCategory(CATID_SafeForScripting,
??L"Controls safely scriptable!");
????if (FAILED(hr))
??return hr;
//在上面的分組之中注冊接口類
????hr = RegisterCLSIDInCategory(CLSID_psSub, CATID_SafeForScripting);
????if (FAILED(hr))
??return hr;

????return NOERROR;
}

//用途: 反注冊服務(wù)器
STDAPI DllUnregisterServer(void) {
????HRESULT hr;????// HResult used by Safety Functions

????AFX_MANAGE_STATE(_afxModuleAddrThis);

/* 原來的代碼
????if (!AfxOleUnregisterTypeLib(_tlid))
??return ResultFromScode(SELFREG_E_TYPELIB);
????if (!COleObjectFactoryEx::UpdateRegistryAll(FALSE))
??return ResultFromScode(SELFREG_E_CLASS);
*/

if(FAILED(_Module.RegisterServer(TRUE))){
??AfxMessageBox("反注冊類型庫失敗");
}

????//刪除注冊表入口
????hr=UnRegisterCLSIDInCategory(CLSID_psSub, CATID_SafeForInitializing);
????if (FAILED(hr))
??return hr;

????hr=UnRegisterCLSIDInCategory(CLSID_psSub, CATID_SafeForScripting);
????if (FAILED(hr))
??return hr;

????return NOERROR;
}

或者

class ....
????public IObjectSafetyImpl<CAgent,INTERFACESAFE_FOR_UNTRUSTED_CALLER|INTERFACESAFE_FOR_UNTRUSTED_DATA>,
...
...
BEGIN_COM_MAP(CAgent)
????...
????COM_INTERFACE_ENTRY(IObjectSafety)
????...
END_COM_MAP()

或者

//2001.12.07 - 安全注冊支持
const CATID CATID_SafeForScripting?????=
????{0x7dd95801,0x9882,0x11cf,{0x9f,0xa9,0x00,0xaa,0x00,0x6c,0x42,0xc4}};
const CATID CATID_SafeForInitializing??=
????{0x7dd95802,0x9882,0x11cf,{0x9f,0xa9,0x00,0xaa,0x00,0x6c,0x42,0xc4}};

....
//2001.12.07 - 安全注冊支持
BEGIN_CATEGORY_MAP(CsubAgent)
????IMPLEMENTED_CATEGORY(CATID_SafeForInitializing)
????IMPLEMENTED_CATEGORY(CATID_SafeForScripting)
END_CATEGORY_MAP()


自己挑吧.

總結(jié)

以上是生活随笔為你收集整理的控件安全注册方法3选的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

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