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.
 
 
 
 
 
 

137 lines
3.8 KiB

/***************************************************************************/
/** Microsoft Windows **/
/** Copyright(c) Microsoft Corp., 1995-1996 **/
/***************************************************************************/
/****************************************************************************
regentry.h
Oct. 95 LenS
Wrapper for registry access
Construct a RegEntry object by specifying the subkey (under
HKEY_CURRENT_USER by default, but can be overridden.)
All member functions (except the destructor) set an internal
error state which can be retrieved with GetError().
Zero indicates no error.
RegEntry works only with strings and DWORDS which are both set
using the overloaded function SetValue()
SetValue("valuename", "string");
SetValue("valuename", 42);
Values are retrieved with GetString() and GetNumber().
GetNumber() allows you to specificy a default if the valuename doesn't
exist.
GetString() returns a pointer to a string internal to RegEntry that is
invalidated when another fuction is called on the same RegEntry object
(e.g. its destructor) so, if you want to use the string beyond this
time, then you must copy it out of the RegEntry object first.
DeleteValue() removes the valuename and value pair.
Registry flushes are automatic when RegEntry is destroys or moves to
another key.
****************************************************************************/
#ifndef REGENTRY_INC
#define REGENTRY_INC
class RegEntry
{
public:
RegEntry( LPCTSTR pszSubKey,
HKEY hkey = HKEY_CURRENT_USER,
BOOL fCreate = TRUE,
REGSAM samDesired = 0
);
~RegEntry();
long GetError() { return m_error;}
VOID ClearError() {m_error = ERROR_SUCCESS;}
long SetValue(LPCTSTR pszValueName, LPCTSTR string);
long SetValue(LPCTSTR pszValueName, unsigned long dwNumber);
long SetValue(LPCTSTR pszValue, void* pData, DWORD cbLength);
LPTSTR GetString(LPCTSTR pszValueName);
DWORD GetBinary(LPCTSTR pszValueName, void** ppvData);
long GetNumber(LPCTSTR pszValueName, long dwDefault = 0);
ULONG GetNumberIniStyle(LPCTSTR pszValueName, ULONG dwDefault = 0);
long DeleteValue(LPCTSTR pszValueName);
long FlushKey();
VOID MoveToSubKey(LPCTSTR pszSubKeyName);
HKEY GetKey() { return m_hkey; }
private:
VOID ChangeKey(HKEY hNewKey);
VOID UpdateWrittenStatus();
VOID ResizeValueBuffer(DWORD length);
HKEY m_hkey;
long m_error;
BOOL m_fhkeyValid;
LPBYTE m_pbValueBuffer;
DWORD m_cbValueBuffer;
BOOL m_fValuesWritten;
TCHAR m_szNULL;
};
inline long
RegEntry::FlushKey()
{
if (m_fhkeyValid) {
m_error = ::RegFlushKey(m_hkey);
}
return m_error;
}
class RegEnumValues
{
public:
RegEnumValues(RegEntry *pRegEntry);
~RegEnumValues();
long Next();
LPTSTR GetName() {return m_pchName;}
DWORD GetType() {return m_dwType;}
LPBYTE GetData() {return m_pbValue;}
DWORD GetDataLength() {return m_dwDataLength;}
DWORD GetCount() {return m_cEntries;}
private:
RegEntry * m_pRegEntry;
DWORD m_iEnum;
DWORD m_cEntries;
LPTSTR m_pchName;
LPBYTE m_pbValue;
DWORD m_dwType;
DWORD m_dwDataLength;
DWORD m_cMaxValueName;
DWORD m_cMaxData;
LONG m_error;
};
class RegEnumSubKeys
{
public:
RegEnumSubKeys(RegEntry *pRegEntry);
~RegEnumSubKeys();
long Next();
LPTSTR GetName() {return m_pchName;}
DWORD GetCount() {return m_cEntries;}
protected:
RegEntry * m_pRegEntry;
DWORD m_iEnum;
DWORD m_cEntries;
LPTSTR m_pchName;
DWORD m_cMaxKeyName;
LONG m_error;
};
#endif // REGENTRY_INC