注册表操作的类

#if !defined _REG_H
#define _REG_H

/************************************************************************
* 文件名：    reg.h
* 文件描述：  注册表读写
*************************************************************************/

class CReg : public CObject
{
public:
 CReg(HKEY hRootKey = HKEY_LOCAL_MACHINE); //构造函数带有默认参数
 virtual ~CReg();

public:
 BOOL VerifyKey (LPCTSTR pszPath);
 BOOL VerifyValue (LPCTSTR pszValue);

 BOOL CreateKey (LPCTSTR pszPath);
 void Close();

 BOOL DeleteValue (LPCTSTR pszValue);
 BOOL DeleteKey (LPCTSTR pszPath);

 BOOL Write (LPCTSTR pszKey, int iVal);
 BOOL Write (LPCTSTR pszKey, DWORD dwVal);
 BOOL Write (LPCTSTR pszKey, LPCTSTR pszVal);

 BOOL Read (LPCTSTR pszKey, int& iVal);
 BOOL Read (LPCTSTR pszKey, DWORD& dwVal);
 BOOL Read (LPCTSTR pszKey, CString& sVal);

 BOOL IsEqual(LPCTSTR pszValue,int nn);
 BOOL IsEqual(LPCTSTR pszValue,DWORD dw);
 BOOL IsEqual(LPCTSTR pszValue,LPCTSTR lpsz);

protected: 
 HKEY  m_hSubKey;    //保存打开的子键句柄
 HKEY    m_hRootKey;   //保存根键句柄
};

#endif

#include "stdafx.h"
#include "reg.h"

/*================================================================
* 函数名：    CReg
* 参数：      (HKEY hRootKey)
* 说明：      如果构造函数不带参数，则使用默认的参数，m_hRootKey被初始化
     为HKEY_LOCAL_MACHINE, 如果带有参数则 m_hRootKey为指定的值
================================================================*/
CReg::CReg(HKEY hRootKey)
{
 m_hRootKey = hRootKey;
}

CReg::~CReg() //在析构函数中关闭打开注册表句柄
{
 Close();
}

/*================================================================
* 函数名：    VerifyKey
* 参数：      (HKEY hRootKey, LPCTSTR pszPath)
* 功能描述:   判断给定的路径是否存在 (兼有打开的功能)
     如果第一个参数为NULL，则使用默认的根键。
* 返回值：    BOOL
================================================================*/
BOOL CReg::VerifyKey (LPCTSTR pszPath)
{
  LONG ReturnValue = ::RegOpenKeyEx (m_hRootKey, pszPath, 0L,
          KEY_ALL_ACCESS, &m_hSubKey);

 if(ReturnValue == ERROR_SUCCESS)
  return TRUE;

 return FALSE;
}

/*================================================================
* 函数名：    VerifyValue
* 参数：      (LPCTSTR pszValue)
* 功能描述:   判断给定的值是否存在 （请先调用VerifyKey，然后在使用该函数）
* 返回值：    BOOL
================================================================*/
BOOL CReg::VerifyValue (LPCTSTR pszValue)
{
 LONG lReturn = ::RegQueryValueEx(m_hSubKey, pszValue, NULL,
  NULL, NULL, NULL);

 if(lReturn == ERROR_SUCCESS)
  return TRUE;

 return FALSE;
}

/*================================================================
* 函数名：    VerifyValue
* 参数：      (LPCTSTR pszValue)
* 功能描述:   判断指定的键名是否等于某个值
* 返回值：    BOOL
================================================================*/
BOOL CReg::IsEqual(LPCTSTR pszValue,int nn)
{
 int nTemp;
 this->Read(pszValue,nTemp);
 if(nTemp==nn)
  return TRUE;
 return FALSE;
}
BOOL CReg::IsEqual(LPCTSTR pszValue,DWORD dw)
{
 DWORD dwTemp;
 this->Read(pszValue,dwTemp);
 if(dwTemp==dw)
  return TRUE;
 return FALSE;
}
BOOL CReg::IsEqual(LPCTSTR pszValue,LPCTSTR lpsz)
{
 CString str;
 this->Read(pszValue,str);
 if(str.CompareNoCase(lpsz)==0)
  return TRUE;
 return FALSE;
}

/*================================================================
* 函数名：    CreateKey
* 参数：      (HKEY hRootKey, LPCTSTR pszPath)
* 功能描述:   创建路径
* 返回值：    BOOL
================================================================*/
BOOL CReg::CreateKey (LPCTSTR pszPath)
{
 DWORD dw;

 LONG ReturnValue = ::RegCreateKeyEx (m_hRootKey, pszPath, 0L, NULL,
       REG_OPTION_VOLATILE, KEY_ALL_ACCESS, NULL,
       &m_hSubKey, &dw);

 if(ReturnValue == ERROR_SUCCESS)
  return TRUE;

 return FALSE;
}

/*================================================================
* 函数名：    Write
* 参数：      (LPCTSTR lpszKeyName, int iVal)
* 功能描述:   写入整型值
* 返回值：    BOOL
================================================================*/
BOOL CReg::Write (LPCTSTR lpszKeyName, int iVal)
{
 DWORD dwValue;

 dwValue = (DWORD)iVal;
 LONG ReturnValue = ::RegSetValueEx (m_hSubKey, lpszKeyName, 0L, REG_DWORD,
  (CONST BYTE*) &dwValue, sizeof(DWORD));

 if(ReturnValue == ERROR_SUCCESS)
  return TRUE;
 
 return FALSE;
}

/*================================================================
* 函数名：    Write
* 参数：      (LPCTSTR lpszKeyName, DWORD dwVal)
* 功能描述:   写入DWORD值
* 返回值：    BOOL
================================================================*/
BOOL CReg::Write (LPCTSTR lpszKeyName, DWORD dwVal)
{
 return ::RegSetValueEx (m_hSubKey, lpszKeyName, 0L, REG_DWORD,
  (CONST BYTE*) &dwVal, sizeof(DWORD));
}

/*================================================================
* 函数名：    Write
* 参数：      (LPCTSTR lpszKeyName, LPCTSTR pszData)
* 功能描述:   写入字符串值
* 返回值：    BOOL
================================================================*/
BOOL CReg::Write (LPCTSTR lpszKeyName, LPCTSTR pszData)
{

 LONG ReturnValue = ::RegSetValueEx (m_hSubKey, lpszKeyName, 0L, REG_SZ,
  (CONST BYTE*) pszData, strlen(pszData) + 1);

 if(ReturnValue == ERROR_SUCCESS)
  return TRUE;
 
 return FALSE;
}

/*================================================================
* 函数名：    Read
* 参数：      (LPCTSTR lpszKeyName, int& iVal) 第2个参数通过引用传递，可以在函数中修改实参
* 功能描述:   读取整数
* 返回值：    BOOL
================================================================*/
BOOL CReg::Read(LPCTSTR lpszKeyName, int& iVal)
{

 DWORD dwType;
 DWORD dwSize = sizeof (DWORD);
 DWORD dwDest;

 LONG lReturn = ::RegQueryValueEx (m_hSubKey, (LPSTR) lpszKeyName, NULL,
  &dwType, (BYTE *) &dwDest, &dwSize);

 if(lReturn == ERROR_SUCCESS)
 {
  iVal = (int)dwDest;
  return TRUE;
 }

 return FALSE;
}

/*================================================================
* 函数名：    Read
* 参数：      (LPCTSTR lpszKeyName, DWORD& dwVal) 第2个参数通过引用传递，可以在函数中修改实参
* 功能描述:   读取DWORD值
* 返回值：    BOOL
================================================================*/
BOOL CReg::Read (LPCTSTR lpszKeyName, DWORD& dwVal)
{

 DWORD dwType;
 DWORD dwSize = sizeof (DWORD);
 DWORD dwDest;

 LONG lReturn = ::RegQueryValueEx (m_hSubKey, (LPSTR) lpszKeyName, NULL,
  &dwType, (BYTE *) &dwDest, &dwSize);

 if(lReturn == ERROR_SUCCESS)
 {
  dwVal = dwDest;
  return TRUE;
 }

 return FALSE;
}

/*================================================================
* 函数名：    Read
* 参数：      (LPCTSTR lpszKeyName, CString& sVal) 第2个参数通过引用传递，可以在函数中修改实参
* 功能描述:   读取字符串值
* 返回值：    BOOL
================================================================*/
BOOL CReg::Read (LPCTSTR lpszKeyName, CString& sVal)
{

 DWORD dwType;
 DWORD dwSize = 200;
 char  szString[255];

 LONG lReturn = ::RegQueryValueEx (m_hSubKey, (LPSTR) lpszKeyName, NULL,
  &dwType, (BYTE *) szString, &dwSize);

 if(lReturn == ERROR_SUCCESS)
 {
  sVal = szString;
  return TRUE;
 }

 return FALSE;
}

/*================================================================
* 函数名：    DeleteValue
* 参数：      (LPCTSTR pszValue)
* 功能描述:   删除值
* 返回值：    BOOL
================================================================*/
BOOL CReg::DeleteValue (LPCTSTR pszValue)
{
 if(::RegDeleteValue(m_hSubKey, pszValue)== ERROR_SUCCESS)  
  return TRUE;
 else
  return FALSE;
}

/*================================================================
* 函数名：    DeleteKey
* 参数：      (HKEY hRootKey, LPCTSTR pszPath)
* 功能描述:   删除路径
* 返回值：    BOOL
================================================================*/
BOOL CReg::DeleteKey (LPCTSTR pszPath)
{

 if(::RegDeleteKey(m_hRootKey, pszPath) == ERROR_SUCCESS)
  return TRUE;
 else
  return FALSE;
}

/*================================================================
* 函数名：    Close
* 参数：     
* 功能描述:   关闭注册表
* 返回值：    void
================================================================*/
 void CReg::Close()
{
 if (m_hSubKey)
 {
  ::RegCloseKey (m_hSubKey);
  m_hSubKey = NULL;
 }
}