為ActiveX控制元件增加安全認證
阿新 • • 發佈:2019-02-11
首先在工程名Ctrl.cpp檔案中增加這三個函式
然後修改這個檔案中的UpdateRegistry函式HRESULT CreateComponentCategory(CATID catid, WCHAR* catDescription) { ICatRegister* pcr = NULL ; HRESULT hr = S_OK ; hr = CoCreateInstance(CLSID_StdComponentCategoriesMgr, NULL, CLSCTX_INPROC_SERVER, IID_ICatRegister, (void**)&pcr); if (FAILED(hr)) { return hr; } CATEGORYINFO catinfo; catinfo.catid = catid; catinfo.lcid = 0x0409 ; // english int len = wcslen(catDescription); if (len>127) len = 127; wcsncpy(catinfo.szDescription, catDescription, len); // Make sure the description is null terminated catinfo.szDescription[len] = '\0'; hr = pcr->RegisterCategories(1, &catinfo); pcr->Release(); return hr; } HRESULT RegisterCLSIDInCategory(REFCLSID clsid, CATID catid) { ICatRegister* pcr = NULL ; HRESULT hr = S_OK ; 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 ; hr = CoCreateInstance(CLSID_StdComponentCategoriesMgr, NULL, CLSCTX_INPROC_SERVER, IID_ICatRegister, (void**)&pcr); if (SUCCEEDED(hr)) { // Unregister this category as being "implemented" by // the class. CATID rgcatid[1] ; rgcatid[0] = catid; hr = pcr->UnRegisterClassImplCategories(clsid, 1, rgcatid); } if (pcr != NULL) { pcr->Release(); } return hr; }
BOOL CXXXXCtrl::CXXXXCtrlFactory::UpdateRegistry(BOOL bRegister) { // TODO: Verify that your control follows apartment-model threading rules. // Refer to MFC TechNote 64 for more information. // If your control does not conform to the apartment-model rules, then // you must modify the code below, changing the 6th parameter from // afxRegApartmentThreading to 0. if (bRegister) { HRESULT hr = S_OK ; // register as safe for scripting hr = CreateComponentCategory(CATID_SafeForScripting, L"Controls that are safely scriptable"); if (FAILED(hr)) { return FALSE; } hr = RegisterCLSIDInCategory(m_clsid, CATID_SafeForScripting); if (FAILED(hr)) { return FALSE; } // register as safe for initializing hr = CreateComponentCategory(CATID_SafeForInitializing, L"Controls safely initializable from persistent data"); if (FAILED(hr)) { return FALSE; } hr = RegisterCLSIDInCategory(m_clsid, CATID_SafeForInitializing); if (FAILED(hr)) { return FALSE; } return AfxOleRegisterControlClass( AfxGetInstanceHandle(), m_clsid, m_lpszProgID, IDS_XXXX, IDB_XXXX, afxRegApartmentThreading, _dwTvWallOleMisc, _tlid, _wVerMajor, _wVerMinor); } else { HRESULT hr = S_OK ; hr = UnRegisterCLSIDInCategory(m_clsid, CATID_SafeForScripting); if (FAILED(hr)) { return FALSE; } hr = UnRegisterCLSIDInCategory(m_clsid, CATID_SafeForInitializing); if (FAILED(hr)) { return FALSE; } return AfxOleUnregisterClass(m_clsid, m_lpszProgID); } }
最後別忘了包含兩個標頭檔案
#include <ComCat.h>
#include <objsafe.h>
重新編譯,即可。