/************************************************************************
* 文件名：    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;
}
}