419 lines
10 KiB
C++
419 lines
10 KiB
C++
/*++
|
|
|
|
Copyright (c) 1997 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
regentry.cpp
|
|
|
|
Abstract:
|
|
|
|
Author:
|
|
|
|
Vlad Sadovsky (vlads) 26-Jan-1997
|
|
|
|
Revision History:
|
|
|
|
26-Jan-1997 VladS created
|
|
|
|
--*/
|
|
|
|
#include "cplusinc.h"
|
|
#include "sticomm.h"
|
|
|
|
RegEntry::RegEntry()
|
|
{
|
|
m_hkey = NULL;
|
|
bhkeyValid = FALSE;
|
|
}
|
|
|
|
RegEntry::RegEntry(const TCHAR *pszSubKey, HKEY hkey)
|
|
{
|
|
m_hkey = NULL;
|
|
bhkeyValid = FALSE;
|
|
|
|
Open(pszSubKey, hkey);
|
|
}
|
|
|
|
RegEntry::~RegEntry()
|
|
{
|
|
Close();
|
|
}
|
|
|
|
BOOL RegEntry::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 RegEntry::Close()
|
|
{
|
|
if (bhkeyValid) {
|
|
RegCloseKey(m_hkey);
|
|
}
|
|
m_hkey = NULL;
|
|
bhkeyValid = FALSE;
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
|
|
long RegEntry::SetValue(const TCHAR *pszValue, const TCHAR *string)
|
|
{
|
|
if (bhkeyValid) {
|
|
m_error = RegSetValueEx(m_hkey, pszValue, 0, REG_SZ,
|
|
(BYTE *)string, sizeof(TCHAR) * (lstrlen(string) + 1));
|
|
}
|
|
return m_error;
|
|
}
|
|
|
|
long RegEntry::SetValue(const TCHAR *pszValue, const TCHAR *string, DWORD dwType)
|
|
{
|
|
DWORD cbData;
|
|
cbData = sizeof(TCHAR) * (lstrlen(string) + 1);
|
|
if (REG_MULTI_SZ == dwType)
|
|
{
|
|
// account for second null
|
|
cbData+= sizeof(TCHAR);
|
|
}
|
|
if (bhkeyValid) {
|
|
|
|
m_error = RegSetValueEx(m_hkey, pszValue, 0, dwType,
|
|
(BYTE *)string, cbData);
|
|
}
|
|
return m_error;
|
|
}
|
|
|
|
|
|
long RegEntry::SetValue(const TCHAR *pszValue, unsigned long dwNumber)
|
|
{
|
|
if (bhkeyValid) {
|
|
m_error = RegSetValueEx(m_hkey, pszValue, 0, REG_BINARY,
|
|
(BYTE *)&dwNumber, sizeof(dwNumber));
|
|
}
|
|
return m_error;
|
|
}
|
|
|
|
long RegEntry::SetValue(const TCHAR *pszValue, BYTE * pValue,unsigned long dwNumber)
|
|
{
|
|
if (bhkeyValid) {
|
|
m_error = RegSetValueEx(m_hkey, pszValue, 0, REG_BINARY,
|
|
pValue, dwNumber);
|
|
}
|
|
return m_error;
|
|
}
|
|
|
|
long RegEntry::DeleteValue(const TCHAR *pszValue)
|
|
{
|
|
if (bhkeyValid) {
|
|
m_error = RegDeleteValue(m_hkey, (LPTSTR) pszValue);
|
|
}
|
|
return m_error;
|
|
}
|
|
|
|
|
|
TCHAR *RegEntry::GetString(const TCHAR *pszValue, TCHAR *string, unsigned long length)
|
|
{
|
|
DWORD dwType = REG_SZ;
|
|
|
|
if (bhkeyValid) {
|
|
DWORD le;
|
|
|
|
m_error = RegQueryValueEx(m_hkey, (LPTSTR) pszValue, 0, &dwType, (LPBYTE)string,
|
|
&length);
|
|
le = ::GetLastError();
|
|
|
|
}
|
|
if (!m_error) {
|
|
//
|
|
// Expand string if indicated
|
|
//
|
|
if (dwType == REG_EXPAND_SZ) {
|
|
|
|
DWORD dwReqSize = 0;
|
|
LPTSTR pszExpanded = new TCHAR[length];
|
|
|
|
if (pszExpanded) {
|
|
|
|
*pszExpanded = TEXT('\0');
|
|
|
|
dwReqSize = ExpandEnvironmentStrings(string,pszExpanded,length);
|
|
|
|
if (dwReqSize && dwReqSize <= length) {
|
|
lstrcpy(string,pszExpanded);
|
|
}
|
|
|
|
delete[] pszExpanded;
|
|
}
|
|
}
|
|
|
|
}
|
|
else {
|
|
*string = '\0';
|
|
}
|
|
|
|
return string;
|
|
}
|
|
|
|
long RegEntry::GetNumber(const TCHAR *pszValue, long dwDefault)
|
|
{
|
|
DWORD dwType = REG_BINARY;
|
|
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 RegEntry::GetValue(const TCHAR *pszValueName, BUFFER *pValue)
|
|
{
|
|
DWORD dwType = REG_SZ;
|
|
DWORD length;
|
|
|
|
m_error = NOERROR;
|
|
|
|
if (bhkeyValid) {
|
|
m_error = RegQueryValueEx( m_hkey,
|
|
(LPTSTR) pszValueName,
|
|
0,
|
|
&dwType,
|
|
NULL,
|
|
&length );
|
|
if (m_error == ERROR_SUCCESS) {
|
|
|
|
pValue->Resize(length);
|
|
|
|
if (length > (UINT)pValue->QuerySize()) {
|
|
m_error = ERROR_NOT_ENOUGH_MEMORY;
|
|
}
|
|
}
|
|
|
|
if (NOERROR == m_error ) {
|
|
|
|
m_error = RegQueryValueEx( m_hkey,
|
|
(LPTSTR) pszValueName,
|
|
0,
|
|
&dwType,
|
|
(LPBYTE) pValue->QueryPtr(),
|
|
&length );
|
|
}
|
|
}
|
|
|
|
if (m_error != ERROR_SUCCESS) {
|
|
pValue->Resize(0);
|
|
}
|
|
}
|
|
|
|
VOID RegEntry::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 RegEntry::FlushKey()
|
|
{
|
|
if (bhkeyValid) {
|
|
m_error = RegFlushKey(m_hkey);
|
|
}
|
|
return m_error;
|
|
}
|
|
|
|
BOOL RegEntry::GetSubKeyInfo(DWORD *pNumberOfSubKeys, DWORD *pMaxSubKeyLength)
|
|
{
|
|
BOOL fResult = FALSE;
|
|
if (bhkeyValid) {
|
|
m_error = RegQueryInfoKey ( m_hkey, // Key
|
|
NULL, // Buffer for class string
|
|
NULL, // Size of class string buffer
|
|
NULL, // Reserved
|
|
pNumberOfSubKeys, // Number of subkeys
|
|
pMaxSubKeyLength, // Longest subkey name
|
|
NULL, // Longest class string
|
|
NULL, // Number of value entries
|
|
NULL, // Longest value name
|
|
NULL, // Longest value data
|
|
NULL, // Security descriptor
|
|
NULL ); // Last write time
|
|
if (m_error == ERROR_SUCCESS) {
|
|
fResult = TRUE;
|
|
}
|
|
}
|
|
return fResult;
|
|
}
|
|
|
|
|
|
BOOL RegEntry::EnumSubKey(DWORD index, StiCString *pstrString)
|
|
{
|
|
BOOL fResult = FALSE;
|
|
|
|
m_error = NOERROR;
|
|
|
|
if (!bhkeyValid) {
|
|
return fResult;
|
|
}
|
|
|
|
m_error = RegEnumKey( m_hkey,
|
|
index,
|
|
(LPTSTR)(LPCTSTR)*pstrString,
|
|
pstrString->GetAllocLength() );
|
|
|
|
if (m_error == ERROR_SUCCESS) {
|
|
fResult = TRUE;
|
|
}
|
|
|
|
return fResult;
|
|
}
|
|
|
|
|
|
RegEnumValues::RegEnumValues(RegEntry *pReqRegEntry)
|
|
: pRegEntry(pReqRegEntry),
|
|
iEnum(0),
|
|
pchName(NULL),
|
|
pbValue(NULL)
|
|
{
|
|
m_error = pRegEntry->GetError();
|
|
if (m_error == ERROR_SUCCESS) {
|
|
m_error = RegQueryInfoKey ( pRegEntry->GetKey(), // Key
|
|
NULL, // Buffer for class string
|
|
NULL, // Size of class string buffer
|
|
NULL, // Reserved
|
|
NULL, // Number of subkeys
|
|
NULL, // Longest subkey name
|
|
NULL, // Longest class string
|
|
&cEntries, // Number of value entries
|
|
&cMaxValueName, // Longest value name
|
|
&cMaxData, // Longest value data
|
|
NULL, // Security descriptor
|
|
NULL ); // Last write time
|
|
}
|
|
if (m_error == ERROR_SUCCESS) {
|
|
if (cEntries != 0) {
|
|
cMaxValueName = cMaxValueName + 1; // REG_SZ needs one more for null
|
|
cMaxData = cMaxData + 1; // REG_SZ needs one more for null
|
|
pchName = new TCHAR[cMaxValueName];
|
|
if (!pchName) {
|
|
m_error = ERROR_NOT_ENOUGH_MEMORY;
|
|
}
|
|
else {
|
|
if (cMaxData) {
|
|
pbValue = new BYTE[cMaxData];
|
|
if (!pbValue) {
|
|
m_error = ERROR_NOT_ENOUGH_MEMORY;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
RegEnumValues::~RegEnumValues()
|
|
{
|
|
delete[] pchName;
|
|
delete[] pbValue;
|
|
}
|
|
|
|
long RegEnumValues::Next()
|
|
{
|
|
if (m_error != ERROR_SUCCESS) {
|
|
return m_error;
|
|
}
|
|
if (cEntries == iEnum) {
|
|
return ERROR_NO_MORE_ITEMS;
|
|
}
|
|
|
|
DWORD cchName = cMaxValueName;
|
|
|
|
dwDataLength = cMaxData;
|
|
m_error = RegEnumValue ( pRegEntry->GetKey(), // Key
|
|
iEnum, // Index of value
|
|
pchName, // Address of buffer for value name
|
|
&cchName, // Address for size of buffer
|
|
NULL, // Reserved
|
|
&dwType, // Data type
|
|
pbValue, // Address of buffer for value data
|
|
&dwDataLength ); // Address for size of data
|
|
iEnum++;
|
|
return m_error;
|
|
}
|
|
|
|
|
|
|
|
//
|
|
// Temporarily here
|
|
//
|
|
VOID
|
|
TokenizeIntoStringArray(
|
|
STRArray& array,
|
|
LPCTSTR lpstrIn,
|
|
TCHAR tcSplitter
|
|
)
|
|
{
|
|
|
|
//
|
|
array.RemoveAll();
|
|
|
|
if (IS_EMPTY_STRING(lpstrIn)) {
|
|
return;
|
|
}
|
|
|
|
while (*lpstrIn) {
|
|
|
|
// First, strip off any leading blanks
|
|
|
|
while (*lpstrIn && *lpstrIn == _TEXT(' '))
|
|
lpstrIn++;
|
|
|
|
for (LPCTSTR lpstrMoi = lpstrIn;
|
|
*lpstrMoi && *lpstrMoi != tcSplitter;
|
|
lpstrMoi++)
|
|
;
|
|
// If we hit the end, just add the whole thing to the array
|
|
if (!*lpstrMoi) {
|
|
if (*lpstrIn)
|
|
array.Add(lpstrIn);
|
|
return;
|
|
}
|
|
|
|
//
|
|
// Otherwise, just add the string up to the splitter
|
|
//
|
|
TCHAR szNew[MAX_PATH];
|
|
SIZE_T uiLen = (SIZE_T)(lpstrMoi - lpstrIn) + 1;
|
|
|
|
if (uiLen < (sizeof(szNew) / sizeof(szNew[0])) - 1) {
|
|
|
|
lstrcpyn(szNew,lpstrIn,(UINT)uiLen);
|
|
szNew[uiLen] = TCHAR('\0');
|
|
|
|
array.Add((LPCTSTR) szNew);
|
|
}
|
|
|
|
lpstrIn = lpstrMoi + 1;
|
|
}
|
|
|
|
}
|
|
|
|
|