一个ActiveX控件安全注册的例子

#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);
}
*/

//用途: 注册组件分类
//说明: 组件安全性种类就是通过组件分类的注册才得以设置组件的安全性
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

    //说明: 确保提供的描述不会超长, 仅仅复制前127个字符
    int len = wcslen(catDescription);
    if (len>127)
  len = 127;
    wcsncpy(catinfo.szDescription, catDescription, len);
  // 确保描述使用"/0"结束
  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 ;

//获得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;
}

//注册服务器
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("注册类型库失败");
}

   //创建初始化安全组件分类
    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;

    //创建脚本编程安全组件分类
    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;
}

//用途: 反注册服务器
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;
}


注意: 上面的CLSID_XXX就是你的自己的控件的CLSID替代
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值