128 lines
2.6 KiB
C++
128 lines
2.6 KiB
C++
|
//
|
||
|
|
// Registry.cpp
|
||
|
|
//
|
||
|
|
|
||
|
|
#include "stdafx.h"
|
||
|
|
#include "Registry.h"
|
||
|
|
|
||
|
|
CRegistry::CRegistry()
|
||
|
|
{
|
||
|
|
m_hkey = NULL;
|
||
|
|
bhkeyValid = FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
CRegistry::~CRegistry()
|
||
|
|
{
|
||
|
|
Close();
|
||
|
|
}
|
||
|
|
|
||
|
|
CRegistry::CRegistry(const TCHAR *pszSubKey, HKEY hkey)
|
||
|
|
{
|
||
|
|
m_hkey = NULL;
|
||
|
|
bhkeyValid = FALSE;
|
||
|
|
Open(pszSubKey, hkey);
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL CRegistry::Open(const TCHAR *pszSubKey, HKEY hkey)
|
||
|
|
{
|
||
|
|
Close();
|
||
|
|
m_error = RegCreateKey(hkey, pszSubKey, &m_hkey);
|
||
|
|
if (m_error)
|
||
|
|
bhkeyValid = FALSE;
|
||
|
|
else
|
||
|
|
bhkeyValid = TRUE;
|
||
|
|
return bhkeyValid;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL CRegistry::Close()
|
||
|
|
{
|
||
|
|
if (bhkeyValid)
|
||
|
|
RegCloseKey(m_hkey);
|
||
|
|
m_hkey = NULL;
|
||
|
|
bhkeyValid = FALSE;
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL CRegistry::CreateKey(const TCHAR *pszSubKey)
|
||
|
|
{
|
||
|
|
HKEY hKey;
|
||
|
|
m_error = RegCreateKey(m_hkey, pszSubKey, &hKey);
|
||
|
|
if(m_error)
|
||
|
|
return FALSE;
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL CRegistry::DeleteKey(const TCHAR *pszSubKey)
|
||
|
|
{
|
||
|
|
m_error = RegDeleteKey(m_hkey,pszSubKey);
|
||
|
|
if(m_error)
|
||
|
|
return FALSE;
|
||
|
|
else
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
LONG CRegistry::SetValue(const TCHAR *pszValue, DWORD dwNumber)
|
||
|
|
{
|
||
|
|
if (bhkeyValid) {
|
||
|
|
m_error = RegSetValueEx(m_hkey,
|
||
|
|
pszValue,
|
||
|
|
0,
|
||
|
|
REG_DWORD,
|
||
|
|
(BYTE *)&dwNumber,
|
||
|
|
sizeof(dwNumber));
|
||
|
|
}
|
||
|
|
return m_error;
|
||
|
|
}
|
||
|
|
|
||
|
|
LONG CRegistry::GetValue(const TCHAR *pszValue, DWORD dwDefault)
|
||
|
|
{
|
||
|
|
DWORD dwType = REG_DWORD;
|
||
|
|
long dwNumber = 0L;
|
||
|
|
DWORD dwSize = sizeof(dwNumber);
|
||
|
|
|
||
|
|
if (bhkeyValid) {
|
||
|
|
m_error = RegQueryValueEx(m_hkey,
|
||
|
|
(LPTSTR) pszValue,
|
||
|
|
0,
|
||
|
|
&dwType,
|
||
|
|
(LPBYTE)&dwNumber,
|
||
|
|
&dwSize);
|
||
|
|
}
|
||
|
|
if (m_error)
|
||
|
|
dwNumber = dwDefault;
|
||
|
|
return dwNumber;
|
||
|
|
}
|
||
|
|
|
||
|
|
VOID CRegistry::MoveToSubKey(const TCHAR *pszSubKeyName)
|
||
|
|
{
|
||
|
|
HKEY _hNewKey;
|
||
|
|
if (bhkeyValid) {
|
||
|
|
m_error = RegOpenKey ( m_hkey,
|
||
|
|
pszSubKeyName,
|
||
|
|
&_hNewKey );
|
||
|
|
if (m_error == ERROR_SUCCESS) {
|
||
|
|
RegCloseKey(m_hkey);
|
||
|
|
m_hkey = _hNewKey;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
LONG CRegistry::EnumerateKeys(DWORD dwIndex,TCHAR *pszKeyName, DWORD dwSize)
|
||
|
|
{
|
||
|
|
FILETIME ft;
|
||
|
|
LONG lError = 0;
|
||
|
|
memset(pszKeyName,0,sizeof(pszKeyName));
|
||
|
|
|
||
|
|
lError = RegEnumKeyEx(m_hkey, // handle to key to enumerate
|
||
|
|
dwIndex, // index of subkey to enumerate
|
||
|
|
pszKeyName, // address of buffer for subkey name
|
||
|
|
&dwSize, // address for size of subkey buffer
|
||
|
|
NULL, // reserved
|
||
|
|
NULL, // address of buffer for class string
|
||
|
|
NULL, // address for size of class buffer
|
||
|
|
&ft // address for time key last written to
|
||
|
|
);
|
||
|
|
|
||
|
|
return lError;
|
||
|
|
}
|