Registry.cpp
上传用户:biuytresa
上传日期:2007-12-07
资源大小:721k
文件大小:6k
源码类别:

DNA

开发平台:

Visual C++

  1. //
  2. // Registry.cpp
  3. //
  4. #include <objbase.h>
  5. #include <assert.h>
  6. #include "Registry.h"
  7. ////////////////////////////////////////////////////////
  8. //
  9. // Internal helper functions prototypes
  10. //
  11. //   - These helper functions were borrowed and modifed from
  12. //     Dale Rogerson's book Inside COM.
  13. // Set the given key and its value.
  14. BOOL SetKeyAndValue(const char* pszPath,
  15.                     const char* szSubkey,
  16.                     const char* szValue) ;
  17. // Convert a CLSID into a char string.
  18. void CLSIDtoString(const CLSID& clsid, 
  19.                  char* szCLSID,
  20.                  int length) ;
  21. // Delete szKeyChild and all of its descendents.
  22. LONG DeleteKey(HKEY hKeyParent, const char* szKeyString) ;
  23. ////////////////////////////////////////////////////////
  24. //
  25. // Constants
  26. //
  27. // Size of a CLSID as a string
  28. const int CLSID_STRING_SIZE = 39 ;
  29. /////////////////////////////////////////////////////////
  30. //
  31. // Public function implementation
  32. //
  33. //
  34. // Register the component in the registry.
  35. //
  36. HRESULT RegisterServer(const CLSID& clsid,         // Class ID
  37.                        const char *szFileName,     // EXE module handle
  38.                        const char* szProgID,       //   IDs
  39.                        const char* szDescription,  // Description String
  40.    const char* szVerIndProgID) // optional
  41. {
  42. // Convert the CLSID into a char.
  43. char szCLSID[CLSID_STRING_SIZE] ;
  44. CLSIDtoString(clsid, szCLSID, sizeof(szCLSID)) ;
  45. // Build the key CLSID\{...}
  46. char szKey[64] ;
  47. strcpy(szKey, "CLSID\") ;
  48. strcat(szKey, szCLSID) ;
  49.   
  50. // Add the CLSID to the registry.
  51. SetKeyAndValue(szKey, NULL, szDescription) ;
  52. // Add the server filename subkey under the CLSID key.
  53. SetKeyAndValue(szKey, "LocalServer32", szFileName) ;
  54. // Add the ProgID subkey under the CLSID key.
  55. if (szProgID != NULL) {
  56. SetKeyAndValue(szKey, "ProgID", szProgID) ;
  57. SetKeyAndValue(szProgID, "CLSID", szCLSID) ;
  58. }
  59. if (szVerIndProgID) {
  60. // Add the version-independent ProgID subkey under CLSID key.
  61. SetKeyAndValue(szKey, "VersionIndependentProgID",
  62.    szVerIndProgID) ;
  63. // Add the version-independent ProgID subkey under HKEY_CLASSES_ROOT.
  64. SetKeyAndValue(szVerIndProgID, NULL, szDescription) ; 
  65. SetKeyAndValue(szVerIndProgID, "CLSID", szCLSID) ;
  66. SetKeyAndValue(szVerIndProgID, "CurVer", szProgID) ;
  67. // Add the versioned ProgID subkey under HKEY_CLASSES_ROOT.
  68. SetKeyAndValue(szProgID, NULL, szDescription) ; 
  69. SetKeyAndValue(szProgID, "CLSID", szCLSID) ;
  70. }
  71. return S_OK ;
  72. }
  73. //
  74. // Remove the component from the registry.
  75. //
  76. HRESULT UnregisterServer(const CLSID& clsid,      // Class ID
  77.                       const char* szProgID,       //   IDs
  78.                       const char* szVerIndProgID) // Programmatic
  79. {
  80. // Convert the CLSID into a char.
  81. char szCLSID[CLSID_STRING_SIZE] ;
  82. CLSIDtoString(clsid, szCLSID, sizeof(szCLSID)) ;
  83. // Build the key CLSID\{...}
  84. char szKey[64] ;
  85. strcpy(szKey, "CLSID\") ;
  86. strcat(szKey, szCLSID) ;
  87. // Delete the CLSID Key - CLSID{...}
  88. LONG lResult = DeleteKey(HKEY_CLASSES_ROOT, szKey) ;
  89. // Delete the version-independent ProgID Key.
  90. if (szVerIndProgID != NULL)
  91. lResult = DeleteKey(HKEY_CLASSES_ROOT, szVerIndProgID) ;
  92. // Delete the ProgID key.
  93. if (szProgID != NULL)
  94. lResult = DeleteKey(HKEY_CLASSES_ROOT, szProgID) ;
  95. return S_OK ;
  96. }
  97. ///////////////////////////////////////////////////////////
  98. //
  99. // Internal helper functions
  100. //
  101. // Convert a CLSID to a char string.
  102. void CLSIDtoString(const CLSID& clsid,
  103.                  char* szCLSID,
  104.                  int length)
  105. {
  106. assert(length >= CLSID_STRING_SIZE) ;
  107. // Get CLSID
  108. LPOLESTR wszCLSID = NULL ;
  109. HRESULT hr = StringFromCLSID(clsid, &wszCLSID) ;
  110. assert(SUCCEEDED(hr)) ;
  111. // Covert from wide characters to non-wide.
  112. wcstombs(szCLSID, wszCLSID, length) ;
  113. // Free memory.
  114. CoTaskMemFree(wszCLSID) ;
  115. }
  116. //
  117. // Delete a key and all of its descendents.
  118. //
  119. LONG DeleteKey(HKEY hKeyParent,           // Parent of key to delete
  120.                const char* lpszKeyChild)  // Key to delete
  121. {
  122. // Open the child.
  123. HKEY hKeyChild ;
  124. LONG lRes = RegOpenKeyEx(hKeyParent, lpszKeyChild, 0,
  125.                          KEY_ALL_ACCESS, &hKeyChild) ;
  126. if (lRes != ERROR_SUCCESS)
  127. {
  128. return lRes ;
  129. }
  130. // Enumerate all of the decendents of this child.
  131. FILETIME time ;
  132. char szBuffer[256] ;
  133. DWORD dwSize = 256 ;
  134. while (RegEnumKeyEx(hKeyChild, 0, szBuffer, &dwSize, NULL,
  135.                     NULL, NULL, &time) == S_OK)
  136. {
  137. // Delete the decendents of this child.
  138. lRes = DeleteKey(hKeyChild, szBuffer) ;
  139. if (lRes != ERROR_SUCCESS)
  140. {
  141. // Cleanup before exiting.
  142. RegCloseKey(hKeyChild) ;
  143. return lRes;
  144. }
  145. dwSize = 256 ;
  146. }
  147. // Close the child.
  148. RegCloseKey(hKeyChild) ;
  149. // Delete this child.
  150. return RegDeleteKey(hKeyParent, lpszKeyChild) ;
  151. }
  152. //
  153. // Create a key and set its value.
  154. //
  155. BOOL SetKeyAndValue(const char* szKey,
  156.                     const char* szSubkey,
  157.                     const char* szValue)
  158. {
  159. HKEY hKey;
  160. char szKeyBuf[1024] ;
  161. // Copy keyname into buffer.
  162. strcpy(szKeyBuf, szKey) ;
  163. // Add subkey name to buffer.
  164. if (szSubkey != NULL)
  165. {
  166. strcat(szKeyBuf, "\") ;
  167. strcat(szKeyBuf, szSubkey ) ;
  168. }
  169. // Create and open key and subkey.
  170. long lResult = RegCreateKeyEx(HKEY_CLASSES_ROOT ,
  171.                               szKeyBuf, 
  172.                               0, NULL, REG_OPTION_NON_VOLATILE,
  173.                               KEY_ALL_ACCESS, NULL, 
  174.                               &hKey, NULL) ;
  175. if (lResult != ERROR_SUCCESS)
  176. {
  177. return FALSE ;
  178. }
  179. // Set the Value.
  180. if (szValue != NULL)
  181. {
  182. RegSetValueEx(hKey, NULL, 0, REG_SZ, 
  183.               (BYTE *)szValue, 
  184.               strlen(szValue)+1) ;
  185. }
  186. RegCloseKey(hKey) ;
  187. return TRUE ;
  188. }