COM_第四讲_保存GUID_优化使用代码

    优化以前的代码,让使用者更方便

一丶 优化思路

1.我们可以将我们写的GUID(类工厂的ID)保存到注册表中,并且保存一下DLL的文件路径,遍历注册表去DLL路径即可.

2.每个类工厂我们就要使用一个GUID,而我们就要写到注册表中GUID

注册表在系统的文件夹下:

C:\\ WINDWOS \\ System32\\config下面,当然这个使我们不能删除的,也不能查看的.我们需要调用API来操作

如果要查看,可以运行CMD命令:  

  regedit命令来查看注册表

二丶插件注册

问题一:

  当我们要把类工厂的ID写到注册表中,但是DLL的路径我们不知道,这些都是动态写入的,所以我们需要利用插件注册模式

问题二:

  我们要怎么注册,要写到哪里合适,要怎么注册

1.我们要写到的是注册表的

  计算机\HKEY_CLASSES_ROOT\CLSID 里面,其中项的名称就是类工厂的GUID

2.注册要调用系统的注册工具,我们需要实现3个API供他调用

HRESULT __stdcall DllCanUnloadNow()    /*能否卸载*/

HRESULT __stdcall DllRegisterServer()    /*是否注册*/

HRESULT __stdcall DllUnregisterServer()   /*卸载插件*/

HRESULT __stdcall DllGetClassObject(const GUID& clsid, const GUID& riid, void **ppObject)(其中这个就是COM前边讲的根据类ID返回类工厂)

 三丶设计注册表存储

上面已经准备好了,我们就要设计注册表了,然后依次写入到注册表

const char* g_szRegTable[][3] ={

{"CLSID\\{450A883B-F00A-46b3-AF3C-EC559997396A}", 0, "SuperMath"  /*你的类工厂的GUID*/

}, { "CLSID\\{450A883B-F00A-46b3-AF3C-EC559997396A}\\InprocServer32", 0 ,(const char*)-1(代表你可以动态的写入DLL路径) }, { "CLSID\\{450A883B-F00A-46b3-AF3C-EC559997396A}\\ 这个是可选的,因为你的GUID要存储进去,当我们取出来的时候要根据GUID查找,很麻烦,所以定义怎么一个文件夹,里面存放的就是GUID,可以直接通过他来获得GUID", 0, "SuperMathsrv.SuperMath.1"( }, { "CLSID\\{450A883B-F00A-46b3-AF3C-EC559997396A}\\TypeLib", 0, "{450A883B-F00A-46b3-AF3C-EC559997396A}"你要包含的头文件的路径,放在Typelib里面 }, { "SuperMathsrv.SuperMath", 0, "SuperMath"先定义一个目录,下面取消注册,删除插件的时候需要删除这个文件夹(也就是注册表的一项) }, { "SuperMathsrv.SuperMath\\CLSID", 0, "{450A883B-F00A-46b3-AF3C-EC559997396A}"写入类工厂的GUID }, { "SuperMathsrv.SuperMath\\CurVer", 0, "SuperMathsrv.SuperMath.1"写入类工厂的版本号 }, { "SuperMathsrv.SuperMath.1", 0, "" }, { "SuperMathsrv.SuperMath.1\\CLSID", 0, "{450A883B-F00A-46b3-AF3C-EC559997396A}" }, };

上面是一个二维数组里面保存了要创建注册表的信息

 详解二维数组里面的文件夹:

1.SuperMath 你的类工厂的名字,要创建怎么一个文件夹

2.InprocServer32 这个是注册表标准的文件夹,这里面的默认项填写的是当前COM(也就是DLL)的路径

3.ProID 可选写入 版本

4.TypeLib 这个是使用者需要用#include,所以我们把他的路径也封装一下

5. 这个是可选的,因为你的GUID要存储进去,当我们取出来的时候要根据GUID查找,很麻烦,所以定义怎么一个文件夹,里面存放的就是GUID,可以直接通过他来获得GUID

 这个是可选的,因为你的GUID要存储进去,当我们取出来的时候要根据GUID查找,很麻烦,所以定义怎么一个文件夹,里面存放的就是GUID,可以直接通过他来获得GUID,需要先定义怎么一个文件夹,我们删除的时候使用

最后一个GUID,写入的GUID

四丶在插件注册标准的函数中写注册表

 1  for (int i = 0; i < sizeof(g_szRegTable) / sizeof(g_szRegTable[0]); i++)
 2   {
 3     const char* pszKey = g_szRegTable[i][0]; /*找到数组的第第0项里面以为数组的第0项的值*/
 4     const char* pszValueName = g_szRegTable[i][1];/*一次类推*/
 5     const char* pszValue = g_szRegTable[i][2];
 6     char szModuleName[MAX_PATH];
 7 
 8     HKEY  hKey;
 9     if (RegCreateKeyEx(HKEY_CLASSES_ROOT, /*写入注册表*/
10                       pszKey, NULL, NULL, 
11                      REG_OPTION_VOLATILE, 
12                      KEY_ALL_ACCESS, 
13                      NULL, 
14                      &hKey, 
15                      NULL) != ERROR_SUCCESS)
16       return S_FALSE;
17 
18     if (pszValue == (const char*)-1)/*判断里面的内容是否是这个(主要是第二次循环要动态写入DLL路径)*/
19     {
20       GetModuleFileName(g_hInstan, szModuleName, sizeof(szModuleName));      /*获得DLL路径写入,g_Hinstan是从DLLmain里面保存的,是一个全局变量,因为要获取当前DLL的模块*/
21       pszValue = szModuleName;
22     }
23 
24     if (RegSetValueEx(hKey, pszValueName, NULL, REG_SZ, /*写入值*/
25       (BYTE*)pszValue, strlen(pszValue)) != ERROR_SUCCESS)
26       return S_FALSE;
27 
28     RegCloseKey(hKey); /*关闭Key*/
29   }
30 
31   return S_OK;卸载的时候需要反向卸载
 1 HRESULT __stdcall DllUnregisterServer()
 2 {
 3   int nCount = sizeof(g_szRegTable) / sizeof(g_szRegTable[0]);/*获得大小*/
 4   for (int i = nCount - 1; i >= 0; i--)
 5   {
 6     const char* pszKey = g_szRegTable[i][0];/*获得第x项的第0个下标(一维数组的第一项)的内容*/
 7     if (RegDeleteKey(HKEY_CLASSES_ROOT, pszKey) != ERROR_SUCCESS)
 8       return S_FALSE; /*删除*/
 9   }
10 
11 
12   return S_OK;
13 }

解决全新的字符串格式

我们的COM组件要跨平台使用,所以这个时候不得不考虑字符串了,C语言中的字符串是\0结尾,而Pascal是前边是字符串的长度,后面是字符串

全新的字符串格式 BSTR 他是他们两个的组合 前边是长度,中间是字符串,后边是\0结尾

而COM是C/c++程序员设计的,所以使用BSTR的时候,直接使用,他会默认帮我们移动到数据位,直到遇到\0结尾

而Pascal想要使用,就要自己减去一个,得到长度,然后获得字符串,不考虑\0结尾.

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏程序员互动联盟

【Windows编程】系列第十篇:文本插入符

大家知道,在使用微软的编程环境创建工程时会让你选择是控制台模式还是Windows应用程序。如果选择控制台的console模式,就会在运行时出现一个黑洞洞的字符模...

39712
来自专栏项勇

笔记89 | Canbus升级程序指南

为方便客户升级,我们定了将升级文件拷至U盘根目录,插到机器USB接口,点击我们的升级程序就可以进行升级

862
来自专栏IT杂记

关于MySQL DNS解析探究之二:unauthenticated user

把这篇没写完的文章写完,2015年的事就不留到2016了 开启DNS解析 mysql> show variables like 'skip_name_resol...

2058
来自专栏Python研发

pymysql

pymsql是python中操作的MYsql的模块,其使用方法和MySQLdb几乎相同

954
来自专栏我的博客

Smarty第一天

基本配置,以及数组和变量的使用和文件调用的基本应用 <?php require(“Smarty/Smarty.class.php”);//smarty类文件 ...

2745
来自专栏文渊之博

使用SQL Server 扩展事件来创建死锁的时间跟踪

我们通过SQL Server 2012图形界面来部署一个扩展事件跟踪会话。然后可以生成SQL脚本,在2008或2008 R2版本下运行类似的跟踪。 步骤1: 通...

1649
来自专栏码匠的流水账

storm drpc实例

592
来自专栏MasiMaro 的技术博文

COM学习(二)——COM的注册和卸载

COM组件是跨语言的,组件被注册到注册表中,在加载时由加载函数在注册表中查找到对应模块的路径并进行相关加载。它的存储规则如下: 1. 在注册表的HKEY_C...

1502
来自专栏安恒网络空间安全讲武堂

逆向专题 | Writeup分享一

逆向WP分享一 0x01.re4 首先我们先点开运行试玩一下,大意就是让你输入正确的用户名和密码就能拿到flag,接下来进入正题。 ? 丢进IDA中,先shi...

2239
来自专栏hadoop学习笔记

HanLP代码与词典分离方案与流程

之前在spark环境中一直用的是portable版本,词条数量不是很够,且有心想把jieba,swcs词典加进来,

712

扫码关注云+社区