Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

122 lines
3.3 KiB

//
// Registry.h
//
#pragma once
#ifdef __cplusplus
// Note: For strong type checking in the Query functions, #define _REG_STRONGTYPES
#ifndef _REG_STRONGTYPES
#define _REG_STRONGTYPES 1
#endif
// Set _REG_ALLOCMEM to 0 to disable functions which allocate memory
#ifndef _REG_ALLOCMEM
#define _REG_ALLOCMEM 1
#endif
class CRegistry
{
public:
CRegistry();
CRegistry(HKEY hkeyParent, LPCTSTR pszKey, DWORD dwAccessFlags = KEY_ALL_ACCESS, BOOL bCreateIfMissing = TRUE);
~CRegistry();
void CloseKey();
BOOL OpenKey(HKEY hkeyParent, LPCTSTR pszKey, REGSAM dwAccessFlags = KEY_ALL_ACCESS);
BOOL CreateKey(HKEY hkeyParent, LPCTSTR pszKey, REGSAM dwAccessFlags = KEY_ALL_ACCESS);
BOOL OpenSubKey(LPCTSTR pszKey, REGSAM dwAccessFlags = KEY_ALL_ACCESS);
BOOL CreateSubKey(LPCTSTR pszKey, REGSAM dwAccessFlags = KEY_ALL_ACCESS);
BOOL DeleteSubKey(LPCTSTR pszKey);
DWORD GetValueSize(LPCTSTR pszValueName);
BOOL DoesValueExist(LPCTSTR pszValueName);
BOOL DeleteValue(LPCTSTR pszValueName);
#ifdef _AFX
BOOL QueryStringValue(LPCTSTR pszValueName, CString& strResult);
CString QueryStringValue(LPCTSTR pszValueName);
BOOL SetStringValue(LPCTSTR pszValueName, const CString& strData);
#endif
BOOL QueryStringValue(LPCTSTR pszValueName, LPTSTR pszBuf, int cchBuf, int* pNumCharsWritten);
int QueryStringValue(LPCTSTR pszValueName, LPTSTR pszBuf, int cchBuf);
BOOL SetStringValue(LPCTSTR pszValueName, LPCTSTR pszData);
LPTSTR QueryStringValue(LPCTSTR pszValueName, int* pNumCharsWritten);
DWORD QueryDwordValue(LPCTSTR pszValueName, DWORD dwDefault = 0);
BOOL QueryDwordValue(LPCTSTR pszValueName, DWORD* pVal);
BOOL SetDwordValue(LPCTSTR pszValueName, DWORD dwVal);
BOOL SetBinaryValue(LPCTSTR pszValueName, LPCVOID pvData, DWORD cbData);
int QueryIntValue(LPCTSTR pszValueName);
BOOL QueryIntValue(LPCTSTR pszValueName, int* pVal);
BOOL SetIntValue(LPCTSTR pszValueName, int nVal);
BOOL DeleteAllValues();
BOOL CloneSubKey(LPCTSTR pszExistingSubKey, CRegistry& regDest, BOOL bRecursive);
BOOL CloneSubKey(LPCTSTR pszExistingSubKey, LPCTSTR pszNewSubKey, BOOL bRecursive);
public:
HKEY m_hKey;
};
//
// Inline functions
//
inline CRegistry::CRegistry()
{
m_hKey = NULL;
}
inline BOOL CRegistry::DoesValueExist(LPCTSTR pszValueName)
{
return (BOOL)GetValueSize(pszValueName);
}
inline BOOL CRegistry::DeleteValue(LPCTSTR pszValueName)
{
if (m_hKey)
return (ERROR_SUCCESS == RegDeleteValue(m_hKey, pszValueName));
else
return FALSE;
}
inline int CRegistry::QueryStringValue(LPCTSTR pszValueName, LPTSTR pszBuf, int cchBuf)
{
if (QueryStringValue(pszValueName, pszBuf, cchBuf, &cchBuf))
return cchBuf;
else
return 0;
}
inline DWORD CRegistry::QueryDwordValue(LPCTSTR pszValueName, DWORD dwDefault /*=0*/)
{
DWORD dwVal = dwDefault;
QueryDwordValue(pszValueName, &dwVal);
return dwVal;
}
inline int CRegistry::QueryIntValue(LPCTSTR pszValueName)
{
return (int)QueryDwordValue(pszValueName);
}
inline BOOL CRegistry::QueryIntValue(LPCTSTR pszValueName, int* pVal)
{
return QueryDwordValue(pszValueName, (DWORD*)pVal);
}
inline BOOL CRegistry::SetIntValue(LPCTSTR pszValueName, int nVal)
{
return SetDwordValue(pszValueName, (DWORD)nVal);
}
#endif // __cplusplus