利用RegSvr32来注册.NET COM组件

  • 背景

  前一段时间由于业务关系,须要在一个比较旧的系统编写一个补丁。原系统使用VB编程,但如果要使用VB来实现这个补丁的内容,对于我这个并不怎么接触VB的人来说太头痛了。由于平时使用的是C#来开发,于是便想到能否用编写.NET COM组件的方式来让VB调用。想到这两三下便写了个.NET COM测试组件,用VB测试调用测试OK。接下来就是进行后续的开发,可等到开发完成后到了补丁打包环节却傻眼了。原来的补丁打包工具在诞生时还不知道.NET为何物,所以根本没有使用RegAsm来注册.NET COM组件的功能。现在更改打补丁工具那是不现实的,可要使得VB调用.NET COM又必须使用RegAsm注册,这下怎么办呢?

  • 解决方式

  原打包工具是支持使用RegSvr32来注册Com组件的,忽然想到能否使用RegSvr32来注册.NET COM组件呢?印象中regsvr32在注册DLL的时候会有个CallBack函数的,于是赶紧Google找找相关资料最终解决了这个问题。主要原理是写一个代理的dll,在这个DLL的注册响应函数DllRegisterServer里调用RegAsm来注册.NET COM,同理在DllUnregisterServer里反注册。

  • 代理DLL代码

   stdafx.h

#pragma once
#include <windows.h>
#include <tchar.h>

  Register.cpp

复制代码
/*----------------------------------------------
此组件主要解决C#编写的COM组件无法用RegSvr32注册的问题
原理是增加一个代理DLL,在代理DLL注册时,响应DllRegisterServer注册函数,
在响应函数中使用RegAsm来注册.Net Com组件,反注册同理。
----------------------------------------------
*/

#include "stdafx.h"
#define MAX_PATH 260

//全局变量,存放注册代理dll路劲
TCHAR g_FilePath[MAX_PATH];

BOOL APIENTRY DllMain( HMODULE hModule,
DWORD ul_reason_for_call,
LPVOID lpReserved
)
{
//记录代理DLL的路径,主要是为了方便后面获取代理DLL所在目录
GetModuleFileName(hModule, g_FilePath, 255);
switch (ul_reason_for_call)
{
case DLL_PROCESS_ATTACH:
case DLL_THREAD_ATTACH:
case DLL_THREAD_DETACH:
case DLL_PROCESS_DETACH:
break;
}
return TRUE;
}

//注册响应函数
extern "C" _declspec(dllexport) long DllRegisterServer(void)
{
return Register(TRUE);
}

//反注册响应函数
extern "C" _declspec(dllexport) long DllUnregisterServer(void)
{
return Register(FALSE);
}

//注册组件
//install : TRUE for install, FALSE for unstall
BOOL Register(BOOL install){
TCHAR cmd[MAX_PATH];
PROCESS_INFORMATION processInfo;
STARTUPINFO stInfo;

//获取注册命令
GetCmdLine(cmd, install);
ZeroMemory( &processInfo, sizeof(processInfo) );
ZeroMemory( &stInfo, sizeof(stInfo) );
stInfo.cb = sizeof(stInfo);

BOOL ret;
//创建注册进程
ret = CreateProcess(NULL,cmd,NULL,NULL,FALSE,CREATE_NO_WINDOW,NULL,NULL,&stInfo,&processInfo);
if(ret==TRUE){
CloseHandle(processInfo.hProcess);
CloseHandle(processInfo.hThread);
return TRUE;
}else{
//DWORD error = GetLastError();
return FALSE;
}

return FALSE;
}


//获取注册命令,加/s开关可以免除确认步骤。
//注册: RegAsm.exe /s Test.dll
//反注册:RegAsm.exe /s /u Test.dll
void GetCmdLine(LPTSTR cmd, BOOL install){
TCHAR regAsmPath[MAX_PATH];
TCHAR regDllPath[MAX_PATH];
TCHAR temp[10];

regAsmPath[0] = '\0';
regDllPath[0] = '\0';
temp[0] = '\0';

GetRegAsmPath(regAsmPath); //获取RegAsm.exe程序绝对路径
GetRegDllPath(regDllPath); //获取需要注册的.NET COM组件绝对路径

if(install){
lstrcpy(temp, _T(" /s "));
}else{
lstrcpy(temp, _T(" /s /u "));
}

cmd[0] = '\0';
lstrcat(cmd, regAsmPath);
lstrcat(cmd, temp);
lstrcat(cmd, regDllPath);
}

//获取需要注册 .Net Com 组件路径,此处写死为Test.dll,与代理dll在同一个目录
//后期可以改成读取ini配置文件
void GetRegDllPath(LPTSTR path){
int count = 0;
for(int i=0; i < MAX_PATH; i++){
if(g_FilePath[i] == _T('\\')){
count = i + 2;
}
}

TCHAR temp[MAX_PATH];
lstrcpyn(temp, g_FilePath, count); //代理DLL所在目录
lstrcat (path, _T("\""));
lstrcat (path, temp);
lstrcat (path, _T("Test.dll\""));
}

//获取 .NET COM 注册程序 RegAsm.exe 的路径
//方法:读取[HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\.NETFramework\InstallRoot]的值,
//然后拼接上"v2.0.50727\RegAsm.exe"
void GetRegAsmPath(LPTSTR path){
HKEY hKey;
DWORD dwBufLen = MAX_PATH;
LONG lRet;

lRet = RegOpenKeyEx(HKEY_LOCAL_MACHINE,_T("SOFTWARE\\Microsoft\\.NETFramework"),0, KEY_READ, &hKey);
lRet = RegQueryValueEx(hKey, _T("InstallRoot"), NULL, NULL, (LPBYTE)path, &dwBufLen);

RegCloseKey(hKey);

lstrcat(path, _T("v2.0.50727\\RegAsm.exe"));
}
复制代码

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值