进程内组件注册表注册

原创 2015年11月18日 13:26:46

头文件:

#ifndef __Registry_H__
#define __Registry_H__
//
// Registry.h
//   - Helper functions registering and unregistering a component.
//
//   - These helper functions were borrowed and modifed from
//     Dale Rogerson's book Inside COM.


// This function will register a component in the Registry.
// DllRegisterServer function should call this function.
HRESULT RegisterServer(const CLSID& clsid,
                       const char *szFileName, 
                       const char* szProgID,
                       const char* szDescription,
                       const char* szVerIndProgID) ;


// This function will unregister a component.  Components
// DllUnregisterServer function should call this function.
HRESULT UnregisterServer(const CLSID& clsid,
                         const char* szProgID,
                         const char* szVerIndProgID) ;


#endif

源文件:

//
// Registry.cpp
//


#include <objbase.h>
#include <assert.h>


#include "Registry.h"


////////////////////////////////////////////////////////
//
// Internal helper functions prototypes
//
//   - These helper functions were borrowed and modifed from
//     Dale Rogerson's book Inside COM.


// Set the given key and its value.
BOOL SetKeyAndValue(const char* pszPath,
                    const char* szSubkey,
                    const char* szValue) ;


// Convert a CLSID into a char string.
void CLSIDtoString(const CLSID& clsid, 
                 char* szCLSID,
                 int length) ;


// Delete szKeyChild and all of its descendents.
LONG DeleteKey(HKEY hKeyParent, const char* szKeyString) ;


////////////////////////////////////////////////////////
//
// Constants
//


// Size of a CLSID as a string
const int CLSID_STRING_SIZE = 39 ;


/////////////////////////////////////////////////////////
//
// Public function implementation
//


//
// Register the component in the registry.
//
HRESULT RegisterServer(const CLSID& clsid,         // Class ID
                       const char *szFileName,     // DLL module handle
                       const char* szProgID,       //   IDs
                       const char* szDescription,  // Description String
  const char* szVerIndProgID) // optional


{
// Convert the CLSID into a char.
char szCLSID[CLSID_STRING_SIZE] ;
CLSIDtoString(clsid, szCLSID, sizeof(szCLSID)) ;


// Build the key CLSID\\{...}
char szKey[64] ;
strcpy(szKey, "CLSID\\") ;
strcat(szKey, szCLSID) ;
  
// Add the CLSID to the registry.
SetKeyAndValue(szKey, NULL, szDescription) ;


// Add the server filename subkey under the CLSID key.
SetKeyAndValue(szKey, "InprocServer32", szFileName) ;


// Add the ProgID subkey under the CLSID key.
if (szProgID != NULL) {
SetKeyAndValue(szKey, "ProgID", szProgID) ;
SetKeyAndValue(szProgID, "CLSID", szCLSID) ;
}


if (szVerIndProgID) {
// Add the version-independent ProgID subkey under CLSID key.
SetKeyAndValue(szKey, "VersionIndependentProgID",
  szVerIndProgID) ;


// Add the version-independent ProgID subkey under HKEY_CLASSES_ROOT.
SetKeyAndValue(szVerIndProgID, NULL, szDescription) ; 
SetKeyAndValue(szVerIndProgID, "CLSID", szCLSID) ;
SetKeyAndValue(szVerIndProgID, "CurVer", szProgID) ;


// Add the versioned ProgID subkey under HKEY_CLASSES_ROOT.
SetKeyAndValue(szProgID, NULL, szDescription) ; 
SetKeyAndValue(szProgID, "CLSID", szCLSID) ;
}


return S_OK ;
}


//
// Remove the component from the registry.
//
HRESULT UnregisterServer(const CLSID& clsid,      // Class ID
                      const char* szProgID,       //   IDs
                      const char* szVerIndProgID) // Programmatic
{
// Convert the CLSID into a char.
char szCLSID[CLSID_STRING_SIZE] ;
CLSIDtoString(clsid, szCLSID, sizeof(szCLSID)) ;


// Build the key CLSID\\{...}
char szKey[64] ;
strcpy(szKey, "CLSID\\") ;
strcat(szKey, szCLSID) ;


// Delete the CLSID Key - CLSID\{...}
LONG lResult = DeleteKey(HKEY_CLASSES_ROOT, szKey) ;


// Delete the version-independent ProgID Key.
if (szVerIndProgID != NULL)
lResult = DeleteKey(HKEY_CLASSES_ROOT, szVerIndProgID) ;


// Delete the ProgID key.
if (szProgID != NULL)
lResult = DeleteKey(HKEY_CLASSES_ROOT, szProgID) ;


return S_OK ;
}


///////////////////////////////////////////////////////////
//
// Internal helper functions
//


// Convert a CLSID to a char string.
void CLSIDtoString(const CLSID& clsid,
                 char* szCLSID,
                 int length)
{
assert(length >= CLSID_STRING_SIZE) ;
// Get CLSID
LPOLESTR wszCLSID = NULL ;
HRESULT hr = StringFromCLSID(clsid, &wszCLSID) ;
assert(SUCCEEDED(hr)) ;


// Covert from wide characters to non-wide.
wcstombs(szCLSID, wszCLSID, length) ;


// Free memory.
CoTaskMemFree(wszCLSID) ;
}


//
// Delete a key and all of its descendents.
//
LONG DeleteKey(HKEY hKeyParent,           // Parent of key to delete
               const char* lpszKeyChild)  // Key to delete
{
// Open the child.
HKEY hKeyChild ;
LONG lRes = RegOpenKeyEx(hKeyParent, lpszKeyChild, 0,
                        KEY_ALL_ACCESS, &hKeyChild) ;
if (lRes != ERROR_SUCCESS)
{
return lRes ;
}


// Enumerate all of the decendents of this child.
FILETIME time ;
char szBuffer[256] ;
DWORD dwSize = 256 ;
while (RegEnumKeyEx(hKeyChild, 0, szBuffer, &dwSize, NULL,
                   NULL, NULL, &time) == S_OK)
{
// Delete the decendents of this child.
lRes = DeleteKey(hKeyChild, szBuffer) ;
if (lRes != ERROR_SUCCESS)
{
// Cleanup before exiting.
RegCloseKey(hKeyChild) ;
return lRes;
}
dwSize = 256 ;
}


// Close the child.
RegCloseKey(hKeyChild) ;


// Delete this child.
return RegDeleteKey(hKeyParent, lpszKeyChild) ;
}


//
// Create a key and set its value.
//
BOOL SetKeyAndValue(const char* szKey,
                    const char* szSubkey,
                    const char* szValue)
{
HKEY hKey;
char szKeyBuf[1024] ;


// Copy keyname into buffer.
strcpy(szKeyBuf, szKey) ;


// Add subkey name to buffer.
if (szSubkey != NULL)
{
strcat(szKeyBuf, "\\") ;
strcat(szKeyBuf, szSubkey ) ;
}


// Create and open key and subkey.
long lResult = RegCreateKeyEx(HKEY_CLASSES_ROOT ,
                             szKeyBuf, 
                             0, NULL, REG_OPTION_NON_VOLATILE,
                             KEY_ALL_ACCESS, NULL, 
                             &hKey, NULL) ;
if (lResult != ERROR_SUCCESS)
{
return FALSE ;
}


// Set the Value.
if (szValue != NULL)
{
RegSetValueEx(hKey, NULL, 0, REG_SZ, 
             (BYTE *)szValue, 
             strlen(szValue)+1) ;
}


RegCloseKey(hKey) ;
return TRUE ;
}


DLL函数

extern "C" HRESULT __stdcall DllRegisterServer()
{
char szModule[1024];
DWORD dwResult = ::GetModuleFileName((HMODULE)g_hModule, szModule, 1024);
if (dwResult == 0)
return SELFREG_E_CLASS;
return RegisterServer(CLSID_Dictionary,
                     szModule, 
 "Dictionary.Object",
 "Dictionary Component",
 NULL);
}




//
// Server unregistration
//
extern "C" HRESULT __stdcall DllUnregisterServer()
{
return UnregisterServer(CLSID_Dictionary,
                       "Dictionary.Object",NULL);
}

版权声明:本文为博主原创文章,未经博主允许不得转载。

相关文章推荐

Binder进程间通信机制的Service Manager组件响应Service组件的注册过程

在Android系统的Binder进程间通信机制中,Service Manager组件在启动之后,主线程会进入到Binder驱动程序的函数binder_thread_read中等待todo队列有新的工...

在注册表中注册应用程序的数据文件

注册表是Windows操作系统的信息存储中心,存放着包括计算机硬件配置、已安装软件的设置信息、当前用户的环境设置及某些文件类型与对其进行访问和操作的应用程序之间的联系等重要信息。Windows操作系统...

Windows CE 6.0注册表示例,及注册表监控函数

Windows CE 6.0有4个基本的注册表键值, HKEY_CLASSES_ROOT, HKEY_CURRENT_USER, HKEY_LOCAL_MACHINE, HKEY_USERS,其它的都...

如何完全卸载oracle和删除oracle在注册表中的注册信息

/************************************************************/ >卸载步骤介绍 /****************************...

[Win32] 注册表操作(2)枚举注册表

本文由CSDN用户zuishikonghuan所作,转载请注明出处:http://blog.csdn.net/zuishikonghuan/article/details/46981391 上一...

MS Com注册表信息及其注册过程

Com注册表信息及其注册过程

关于一个模块中所有rgs文件中有注册项无法全部写入注册表的问题

1、问题情形 创建ATL项目时,使用VC每增加一个COM类,就会生成一个对应的rgs文件。 在vista及win7系统中,如果增加了HKLM注册表项,而生成的程序并不是以管理员身份运行时,就会出现...
  • wcyoot
  • wcyoot
  • 2012-02-22 10:54
  • 1735

注册表处理之写入和删除带子项的注册表

写入带子项的注册表bool AddRegInfoStruct(PREG_INFO_CTRL pRegInfoCtrl) { bool success = true; PREG_INFO_CT...

编写注册表reg文件及批处理操作注册表

编写注册表文件新建一个文本文件,在其中输入以下内容: Windows Registry Editor Version 5.00 [HKEY_CURRENT_USER\Software\Microsof...

锁定注册表与注册表解锁

锁定注册表与注册表解锁  HKEY_CURRENT_USER/Software/Microsoft/Windows/CurrentVerssion/Policies/System 分支下,新建DWOR...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)