VC中操作注册表的实现源码

#pragma once
const UINT BUFFER_LENGTH = 1024;
const UINT BUFFER_LENGTH_BINARY = 3;
enum err
{ 
 ERR_OPENREGKEY,
 ERR_SETREGVALUE,
 ERR_QUERYVALUE,
 ERR_DELETEVALUE,
 ERR_SUBKEYEXIST,
 ERR_CREATESUBKEY,
 ERR_DELETESUBKEY, 
 ERR_SUCCESS,
 ERR_KEYNOTEXIST
};
enum hkey
{
 DHKEY_CLASSES_ROOT = 0x80000000,
 DHKEY_CURRENT_USER = 0x80000001,
 DHKEY_LOCAL_MACHINE = 0x80000002,
 DHKEY_USERS = 0x80000003,
 DHKEY_CURRENT_CONFIG = 0x80000005
};
class CRegeditOp
{
public:
 CRegeditOp();
 virtual ~CRegeditOp();
public:
 //Set or modify the type and value of the key
 err SetStrValue(HKEY hMainKey, CString szSubKey, CString szItem,  CString szValue);
 //query the value and the type of the key
 err GetStrValue(HKEY hMainKey, CString szSubKey, CString szItem, CString& szValue, DWORD& strType,DWORD& length);
 //delete the key value
 err DeleteStrValue(HKEY hMainKey, CString szSubKey, CString szItem);
 //create a sub key
 err AddSubKey(HKEY hMainKey, CString szSubKey);
 //delete a sub key
 err DeleteSubKey(HKEY hMainKey, CString szSubKey);
 //check the specified key path wheather exist
 err CheckRegeditFoler(HKEY hMainKey, CString szSubKey);
};

实现文件

#include "StdAfx.h"
#include "RegeditOp.h"
#include "./regeditop.h"
#include "MyFile.h"

CRegeditOp::CRegeditOp(void)
{
}

CRegeditOp::~CRegeditOp(void)
{
}

/**************************************************************************
* Name    : SetStrValue
* Function: set the value of the stecial key
* Input   :
      hMainKey:Handle to an open key,It is returned by RegOpenKeyEx or RegCreateKeyEx.it can be one of the follows predefined keys
           HKEY_CLASSES_ROOT
     HKEY_CURRENT_USER
     HKEY_LOCAL_MACHINE
     HKEY_USERS
   szSubKey:The path of the key which is going to be setted.
   szItem:The name of the key which is going to be setted
   szValue:The value of the key which is going to be setted
* Output  :
* In&Out  :
* Return  : IF success, return ERR_SUCCESS(7);if open key failed, return  ERR_OPENREGKEY(0);If set value failed, return ERR_SETREGVALUE(1)
* Others  :
* Date        Version       People Name     Content of modify
* -----------------------------------------------------------
* 2008-8-4    V1.0         jet             Create
* -----------------------------------------------------------
**************************************************************************/
err CRegeditOp::SetStrValue(HKEY hMainKey, CString szSubKey, CString szItem,  CString szValue)
{
 HKEY hRsltKey;

 //=========================================打开注册表
 long lRslt = RegOpenKeyEx(hMainKey, szSubKey, 0,KEY_SET_VALUE,&hRsltKey);
 if (lRslt != ERROR_SUCCESS)
 {  
  return ERR_OPENREGKEY;
 }

 //===================================设置注册表值
 lRslt = RegSetValueEx(hRsltKey,
       szItem,
       NULL,
       REG_SZ,
       LPBYTE(szValue.GetBuffer(szValue.GetLength())),
       szValue.GetLength());
 if (lRslt != ERROR_SUCCESS)
 {  
  return ERR_SETREGVALUE;
 }

 //============================================关闭注册表键
 if (hMainKey)
 {
  RegCloseKey(hMainKey);
 }
 if (hRsltKey)
 {
  RegCloseKey(hRsltKey);
 }
 return ERR_SUCCESS;
}

/**************************************************************************
* Name    : GetStrValue
* Function: Query the key value and type of the specify key.
* Input   :
      hMainKey:Handle to an open key,It is returned by RegOpenKeyEx or RegCreateKeyEx.it can be one of the follows predefined keys
           HKEY_CLASSES_ROOT
     HKEY_CURRENT_USER
     HKEY_LOCAL_MACHINE
     HKEY_USERS
   szSubKey:The path of the key which is going to be got.
   szItem:The name of the key which is going to be got.   
   
* Output  : szValue:A string parameter that receives the value's #
   strType:A DWORD parameter that receives the value's type
* In&Out  : length:[IN]  A DWORD parameter that spcify the value's length,if the length is smaller than the the real key length, the keyValue cann't be ruturn.
          [OUT] A DWORD parameter that receives the value's length
* Return  : IF success, return ERR_SUCCESS(7);if open key failed, return  ERR_OPENREGKEY(0);If set value failed, return ERR_SETREGVALUE(1)
* Others  :
* Date        Version       People Name     Content of modify
* -----------------------------------------------------------
* 2008-8-4    V1.0         jet             Create
* -----------------------------------------------------------
**************************************************************************/
err CRegeditOp::GetStrValue(HKEY hMainKey, CString szSubKey, CString szItem, CString& szValue, DWORD &strType,DWORD& length)
{ 
 HKEY hRsltKey;
 //打开注册表
 if (ERROR_SUCCESS != RegOpenKeyEx(hMainKey, szSubKey,0,KEY_READ, &hRsltKey))
 {  
  return ERR_OPENREGKEY;
 }
 LPBYTE lp# = new BYTE[length];
 memset(lp#,0,length);
 //查询注册表项值
 long lRslt = RegQueryValueEx(hRsltKey, szItem, 0, &strType, lp#,&length);
 szValue = (CString)lp#;
 
 if (lRslt != ERROR_SUCCESS)
 {  
  delete [] lp#;
  return ERR_QUERYVALUE;
 } 
 delete [] lp#;
 //关闭打开的键
 if (hMainKey)
 {
  RegCloseKey(hMainKey);
 }
 if (hRsltKey)
 {
  RegCloseKey(hRsltKey);
 }
 return ERR_SUCCESS;
}

/**************************************************************************
* Name    : AddSubKey
* Function: Add a new key
* Input   :
      hMainKey:Handle to an open key,It is returned by RegOpenKeyEx or RegCreateKeyEx.it can be one of the follows predefined keys
           HKEY_CLASSES_ROOT
     HKEY_CURRENT_USER
     HKEY_LOCAL_MACHINE
     HKEY_USERS
   szSubKey:The path of the key which is going to be created.
      
* Output  :
* In&Out  :
* Return  : If success, return ERR_SUCCESS(7);if the key was exist, return ERR_SUBKEYEXIST(4);if create key failed, return  ERR_CREATESUBKEY(5)
* Others  :
* Date        Version       People Name     Content of modify
* -----------------------------------------------------------
* 2008-8-4    V1.0         jet             Create
* -----------------------------------------------------------
**************************************************************************/
err CRegeditOp::AddSubKey(HKEY hMainKey, CString szSubKey)
{
 HKEY hRsltKey;
 //================================判断要增加的注册表子键是否已存在
 //LONG iReturn = RegOpenKeyEx(hMainKey, szSubKey, 0,0,&hRsltKey);
 if (ERROR_SUCCESS == RegOpenKeyEx(hMainKey, szSubKey,0,KEY_SET_VALUE,&hRsltKey))
 { 
  return ERR_SUBKEYEXIST;
 }

 //==================================如果不存在，则增加此子键
 long lRslt = RegCreateKeyEx(hMainKey, szSubKey,0,NULL,REG_OPTION_NON_VOLATILE,KEY_CREATE_SUB_KEY ,NULL, &hRsltKey,NULL);
 //long lRslt = RegCreateKey(hMainKey,szSubKey, &hRsltKey);

 if (lRslt != ERROR_SUCCESS)
 { 
  return ERR_CREATESUBKEY; 
 }

 //===============================关闭打开的注册表键句柄
 if(hMainKey)
 {
  RegCloseKey(hMainKey);
 }
 if (hRsltKey)
 {
  RegCloseKey(hRsltKey);
 }
 return ERR_SUCCESS;
}
/**************************************************************************
* Name    : DeleteStrValue
* Function: Delete the value of a key
* Input   :
      hMainKey:Handle to an open key,It is returned by RegOpenKeyEx or RegCreateKeyEx.it can be one of the follows predefined keys
           HKEY_CLASSES_ROOT
     HKEY_CURRENT_USER
     HKEY_LOCAL_MACHINE
     HKEY_USERS
   szSubKey:The path of the key which value is going to be deleted.
   szItem:The key which to be deleted.      
* Output  :
* In&Out  :
* Return  : If success, return ERR_SUCCESS(7);if open key failed, return ERR_OPENREGKEY(0);if delete key value failed, return  ERR_DELETEVALUE(3)
* Others  :
* Date        Version       People Name     Content of modify
* -----------------------------------------------------------
* 2008-8-4    V1.0         jet             Create
* -----------------------------------------------------------
**************************************************************************/
err CRegeditOp::DeleteStrValue(HKEY hMainKey, CString szSubKey, CString szItem)
{
 HKEY hRsltKey;
 //打开注册表
 long lRslt = RegOpenKeyEx(hMainKey, szSubKey,0,KEY_CREATE_SUB_KEY, &hRsltKey);
 if (lRslt != ERROR_SUCCESS)
 { 
  return ERR_OPENREGKEY;
 }
 //删除注册表项
 lRslt = RegDeleteValue(hRsltKey, szItem);
 if (lRslt != ERROR_SUCCESS)
 { 
  return ERR_DELETEVALUE;
 }
 return ERR_SUCCESS;
}

/**************************************************************************
* Name    : DeleteSubKey
* Function: Delete a key
* Input   :
      hMainKey:Handle to an open key,It is returned by RegOpenKeyEx or RegCreateKeyEx.it can be one of the follows predefined keys
           HKEY_CLASSES_ROOT
     HKEY_CURRENT_USER
     HKEY_LOCAL_MACHINE
     HKEY_USERS
   szSubKey:The path of the key which is going to be deleted.    
* Output  :
* In&Out  :
* Return  : If success, return ERR_SUCCESS(7);if delete key failed, return  ERR_DELETESUBKEY(3)
* Others  :
* Date        Version       People Name     Content of modify
* -----------------------------------------------------------
* 2008-8-4    V1.0         jet             Create
* -----------------------------------------------------------
**************************************************************************/
err CRegeditOp::DeleteSubKey(HKEY hMainKey, CString szSubKey)
{
 //=================================删除子键
 long lRslt = RegDeleteKey(hMainKey, szSubKey);
 if (lRslt)
 { 
  return ERR_KEYNOTEXIST;
 }

 //关闭打开的键
 if (hMainKey)
 {
  RegCloseKey(hMainKey);
 }
 return ERR_SUCCESS;
}

/**************************************************************************
* Name    : CheckRegeditFoler
* Function: Check wheathe the specified Key path is exist.
* Input   :
      hMainKey:Handle to an open key,It is returned by RegOpenKeyEx or RegCreateKeyEx.it can be one of the follows predefined keys
           HKEY_CLASSES_ROOT
     HKEY_CURRENT_USER
     HKEY_LOCAL_MACHINE
     HKEY_USERS
   szSubKey:The path of the key which is going to be checked.    
* Output  :
* In&Out  :
* Return  : If success, return ERR_SUCCESS(7);if delete key failed, return  ERR_DELETESUBKEY(3)
* Others  :
* Date        Version       People Name     Content of modify
* -----------------------------------------------------------
* 2008-8-4    V1.0         jet             Create
* -----------------------------------------------------------
**************************************************************************/
err CRegeditOp::CheckRegeditFoler(HKEY hMainKey, CString szSubKey)
{
 HKEY hRsltKey;
 //打开注册表
 if (ERROR_SUCCESS == RegOpenKeyEx(hMainKey, szSubKey,0,KEY_READ, &hRsltKey))
 { 
  if(hMainKey)
  {
   RegCloseKey(hMainKey);
  }
  if (hRsltKey)
  {
   RegCloseKey(hRsltKey);
  }
  return ERR_SUCCESS;
 }
 else
 {
  return FALSE;
 }
}