mirror of https://github.com/tongzx/nt5src
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.
575 lines
13 KiB
575 lines
13 KiB
/*++
|
|
|
|
Copyright (c) 1998 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
registry.cpp
|
|
|
|
Abstract:
|
|
|
|
SIS Groveler/registry interface
|
|
|
|
Authors:
|
|
|
|
John Douceur, 1998
|
|
|
|
Environment:
|
|
|
|
User Mode
|
|
|
|
|
|
Revision History:
|
|
|
|
|
|
--*/
|
|
|
|
#include "all.hxx"
|
|
|
|
_TCHAR Registry::id_buffer[id_buffer_length];
|
|
|
|
bool
|
|
Registry::read(
|
|
HKEY base_key,
|
|
const _TCHAR *path,
|
|
int num_entries,
|
|
EntrySpec *entries)
|
|
{
|
|
ASSERT(base_key == HKEY_CLASSES_ROOT
|
|
|| base_key == HKEY_CURRENT_CONFIG
|
|
|| base_key == HKEY_CURRENT_USER
|
|
|| base_key == HKEY_LOCAL_MACHINE
|
|
|| base_key == HKEY_USERS);
|
|
ASSERT(path != 0);
|
|
ASSERT(num_entries > 0);
|
|
ASSERT(entries != 0);
|
|
for (int index = 0; index < num_entries; index++)
|
|
{
|
|
load_string_into_value(entries[index].type,
|
|
entries[index].default_value, entries[index].pointer);
|
|
}
|
|
HKEY path_key;
|
|
long result = RegOpenKeyEx(base_key, path, 0, KEY_ALL_ACCESS, &path_key);
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
return false;
|
|
}
|
|
ASSERT(path_key != 0);
|
|
for (index = 0; index < num_entries; index++)
|
|
{
|
|
DWORD data_type;
|
|
DWORD data_length = id_buffer_length;
|
|
result = RegQueryValueEx(path_key, entries[index].identifier, 0,
|
|
&data_type, (BYTE *)id_buffer, &data_length);
|
|
if (result == ERROR_SUCCESS && data_type == REG_SZ)
|
|
{
|
|
load_string_into_value(entries[index].type, id_buffer,
|
|
entries[index].pointer);
|
|
}
|
|
}
|
|
ASSERT(path_key != 0);
|
|
RegCloseKey(path_key);
|
|
path_key = 0;
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
Registry::write(
|
|
HKEY base_key,
|
|
const _TCHAR *path,
|
|
int num_entries,
|
|
EntrySpec *entries)
|
|
{
|
|
ASSERT(base_key == HKEY_CLASSES_ROOT
|
|
|| base_key == HKEY_CURRENT_CONFIG
|
|
|| base_key == HKEY_CURRENT_USER
|
|
|| base_key == HKEY_LOCAL_MACHINE
|
|
|| base_key == HKEY_USERS);
|
|
ASSERT(path != 0);
|
|
ASSERT(num_entries > 0);
|
|
ASSERT(entries != 0);
|
|
HKEY path_key;
|
|
DWORD disposition;
|
|
long result = RegCreateKeyEx(base_key, path, 0, 0, REG_OPTION_NON_VOLATILE,
|
|
KEY_ALL_ACCESS, 0, &path_key, &disposition);
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
PRINT_DEBUG_MSG((_T("GROVELER: RegCreateKeyEx() failed with error %d\n"),
|
|
result));
|
|
return false;
|
|
}
|
|
ASSERT(path_key != 0);
|
|
for (int index = 0; index < num_entries; index++)
|
|
{
|
|
DWORD data_type;
|
|
DWORD data_length = id_buffer_length;
|
|
result = RegQueryValueEx(path_key, entries[index].identifier, 0,
|
|
&data_type, (BYTE *)id_buffer, &data_length);
|
|
if (result != ERROR_SUCCESS || data_type != REG_SZ)
|
|
{
|
|
store_value_in_string(entries[index].type, entries[index].pointer,
|
|
id_buffer);
|
|
result =
|
|
RegSetValueEx(path_key, entries[index].identifier, 0, REG_SZ,
|
|
(BYTE *)id_buffer, (_tcslen(id_buffer) + 1) * sizeof(_TCHAR));
|
|
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
PRINT_DEBUG_MSG((_T("GROVELER: RegSetValueEx() failed with error %d\n"),
|
|
result));
|
|
}
|
|
}
|
|
}
|
|
ASSERT(path_key != 0);
|
|
RegCloseKey(path_key);
|
|
path_key = 0;
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
Registry::overwrite(
|
|
HKEY base_key,
|
|
const _TCHAR *path,
|
|
int num_entries,
|
|
EntrySpec *entries)
|
|
{
|
|
ASSERT(base_key == HKEY_CLASSES_ROOT
|
|
|| base_key == HKEY_CURRENT_CONFIG
|
|
|| base_key == HKEY_CURRENT_USER
|
|
|| base_key == HKEY_LOCAL_MACHINE
|
|
|| base_key == HKEY_USERS);
|
|
ASSERT(path != 0);
|
|
ASSERT(num_entries > 0);
|
|
ASSERT(entries != 0);
|
|
HKEY path_key;
|
|
DWORD disposition;
|
|
long result = RegCreateKeyEx(base_key, path, 0, 0, REG_OPTION_NON_VOLATILE,
|
|
KEY_ALL_ACCESS, 0, &path_key, &disposition);
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
PRINT_DEBUG_MSG((_T("GROVELER: RegCreateKeyEx() failed with error %d\n"),
|
|
result));
|
|
return false;
|
|
}
|
|
ASSERT(path_key != 0);
|
|
for (int index = 0; index < num_entries; index++)
|
|
{
|
|
store_value_in_string(entries[index].type, entries[index].pointer,
|
|
id_buffer);
|
|
result =
|
|
RegSetValueEx(path_key, entries[index].identifier, 0, REG_SZ,
|
|
(BYTE *)id_buffer, (_tcslen(id_buffer) + 1) * sizeof(_TCHAR));
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
PRINT_DEBUG_MSG((_T("GROVELER: RegSetValueEx() failed with error %d\n"),
|
|
result));
|
|
}
|
|
}
|
|
ASSERT(path_key != 0);
|
|
RegCloseKey(path_key);
|
|
path_key = 0;
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
Registry::write_defaults(
|
|
HKEY base_key,
|
|
const _TCHAR *path,
|
|
int num_entries,
|
|
EntrySpec *entries)
|
|
{
|
|
ASSERT(base_key == HKEY_CLASSES_ROOT
|
|
|| base_key == HKEY_CURRENT_CONFIG
|
|
|| base_key == HKEY_CURRENT_USER
|
|
|| base_key == HKEY_LOCAL_MACHINE
|
|
|| base_key == HKEY_USERS);
|
|
ASSERT(path != 0);
|
|
ASSERT(num_entries > 0);
|
|
ASSERT(entries != 0);
|
|
HKEY path_key;
|
|
DWORD disposition;
|
|
long result = RegCreateKeyEx(base_key, path, 0, 0, REG_OPTION_NON_VOLATILE,
|
|
KEY_ALL_ACCESS, 0, &path_key, &disposition);
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
PRINT_DEBUG_MSG((_T("GROVELER: RegCreateKeyEx() failed with error %d\n"),
|
|
result));
|
|
return false;
|
|
}
|
|
ASSERT(path_key != 0);
|
|
for (int index = 0; index < num_entries; index++)
|
|
{
|
|
DWORD data_type;
|
|
DWORD data_length = id_buffer_length;
|
|
result = RegQueryValueEx(path_key, entries[index].identifier, 0,
|
|
&data_type, (BYTE *)id_buffer, &data_length);
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
result = RegSetValueEx(path_key, entries[index].identifier, 0,
|
|
REG_SZ, (BYTE *)entries[index].default_value,
|
|
(_tcslen(entries[index].default_value) + 1) * sizeof(_TCHAR));
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
PRINT_DEBUG_MSG((_T("GROVELER: RegSetValueEx() failed with error %d\n"),
|
|
result));
|
|
}
|
|
}
|
|
}
|
|
ASSERT(path_key != 0);
|
|
RegCloseKey(path_key);
|
|
path_key = 0;
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
Registry::overwrite_defaults(
|
|
HKEY base_key,
|
|
const _TCHAR *path,
|
|
int num_entries,
|
|
EntrySpec *entries)
|
|
{
|
|
ASSERT(base_key == HKEY_CLASSES_ROOT
|
|
|| base_key == HKEY_CURRENT_CONFIG
|
|
|| base_key == HKEY_CURRENT_USER
|
|
|| base_key == HKEY_LOCAL_MACHINE
|
|
|| base_key == HKEY_USERS);
|
|
ASSERT(path != 0);
|
|
ASSERT(num_entries > 0);
|
|
ASSERT(entries != 0);
|
|
HKEY path_key;
|
|
DWORD disposition;
|
|
long result = RegCreateKeyEx(base_key, path, 0, 0, REG_OPTION_NON_VOLATILE,
|
|
KEY_ALL_ACCESS, 0, &path_key, &disposition);
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
PRINT_DEBUG_MSG((_T("GROVELER: RegCreateKeyEx() failed with error %d\n"),
|
|
result));
|
|
return false;
|
|
}
|
|
ASSERT(path_key != 0);
|
|
for (int index = 0; index < num_entries; index++)
|
|
{
|
|
result = RegSetValueEx(path_key, entries[index].identifier, 0, REG_SZ,
|
|
(BYTE *)entries[index].default_value,
|
|
(_tcslen(entries[index].default_value) + 1) * sizeof(_TCHAR));
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
PRINT_DEBUG_MSG((_T("GROVELER: RegSetValueEx() failed with error %d\n"),
|
|
result));
|
|
}
|
|
}
|
|
ASSERT(path_key != 0);
|
|
RegCloseKey(path_key);
|
|
path_key = 0;
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
Registry::read_string_set(
|
|
HKEY base_key,
|
|
const _TCHAR *path,
|
|
int *num_strings,
|
|
_TCHAR ***strings,
|
|
BYTE **buffer)
|
|
{
|
|
ASSERT(base_key == HKEY_CLASSES_ROOT
|
|
|| base_key == HKEY_CURRENT_CONFIG
|
|
|| base_key == HKEY_CURRENT_USER
|
|
|| base_key == HKEY_LOCAL_MACHINE
|
|
|| base_key == HKEY_USERS);
|
|
ASSERT(path != 0);
|
|
ASSERT(num_strings != 0);
|
|
ASSERT(strings != 0);
|
|
ASSERT(buffer != 0);
|
|
HKEY path_key;
|
|
*num_strings = 0;
|
|
*strings = 0;
|
|
*buffer = 0;
|
|
long result = RegOpenKeyEx(base_key, path, 0, KEY_ALL_ACCESS, &path_key);
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
return false;
|
|
}
|
|
unsigned long num_values;
|
|
unsigned long max_value_length;
|
|
unsigned long max_string_length;
|
|
result = RegQueryInfoKey(path_key, 0, 0, 0, 0, 0, 0,
|
|
&num_values, &max_value_length, &max_string_length, 0, 0);
|
|
if (result != ERROR_SUCCESS || num_values == 0)
|
|
{
|
|
return true;
|
|
}
|
|
_TCHAR *value = new _TCHAR[max_value_length + 1];
|
|
_TCHAR **string_set = new _TCHAR *[num_values];
|
|
BYTE *string_buffer =
|
|
new BYTE[num_values * max_string_length * sizeof(_TCHAR)];
|
|
int string_index = 0;
|
|
int buffer_offset = 0;
|
|
for (unsigned int index = 0; index < num_values; index++)
|
|
{
|
|
DWORD value_length = max_value_length + 1;
|
|
DWORD string_length = max_string_length;
|
|
DWORD data_type;
|
|
result = RegEnumValue(path_key, index, value, &value_length, 0,
|
|
&data_type, &string_buffer[buffer_offset], &string_length);
|
|
if (result == ERROR_SUCCESS && data_type == REG_SZ)
|
|
{
|
|
ASSERT(unsigned(string_index) < num_values);
|
|
string_set[string_index] = (_TCHAR *)&string_buffer[buffer_offset];
|
|
string_index++;
|
|
buffer_offset += string_length;
|
|
}
|
|
}
|
|
*num_strings = string_index;
|
|
ASSERT(value != 0);
|
|
delete[] value;
|
|
value = 0;
|
|
*strings = string_set;
|
|
*buffer = string_buffer;
|
|
ASSERT(path_key != 0);
|
|
RegCloseKey(path_key);
|
|
path_key = 0;
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
Registry::write_string_set(
|
|
HKEY base_key,
|
|
const _TCHAR *path,
|
|
int num_strings,
|
|
_TCHAR **strings,
|
|
_TCHAR **identifiers)
|
|
{
|
|
ASSERT(base_key == HKEY_CLASSES_ROOT
|
|
|| base_key == HKEY_CURRENT_CONFIG
|
|
|| base_key == HKEY_CURRENT_USER
|
|
|| base_key == HKEY_LOCAL_MACHINE
|
|
|| base_key == HKEY_USERS);
|
|
ASSERT(path != 0);
|
|
ASSERT(num_strings > 0);
|
|
ASSERT(strings != 0);
|
|
ASSERT(identifiers != 0);
|
|
HKEY path_key;
|
|
DWORD disposition;
|
|
long result = RegCreateKeyEx(base_key, path, 0, 0, REG_OPTION_NON_VOLATILE,
|
|
KEY_ALL_ACCESS, 0, &path_key, &disposition);
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
PRINT_DEBUG_MSG((_T("GROVELER: RegCreateKeyEx() failed with error %d\n"),
|
|
result));
|
|
return false;
|
|
}
|
|
ASSERT(path_key != 0);
|
|
for (int index = 0; index < num_strings; index++)
|
|
{
|
|
DWORD data_type;
|
|
DWORD data_length = id_buffer_length;
|
|
result = RegQueryValueEx(path_key, identifiers[index], 0,
|
|
&data_type, (BYTE *)id_buffer, &data_length);
|
|
if (result != ERROR_SUCCESS || data_type != REG_SZ)
|
|
{
|
|
result = RegSetValueEx(
|
|
path_key, identifiers[index], 0, REG_SZ, (BYTE *)strings[index],
|
|
(_tcslen(strings[index]) + 1) * sizeof(_TCHAR));
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
PRINT_DEBUG_MSG((_T("GROVELER: RegSetValueEx() failed with error %d\n"),
|
|
result));
|
|
}
|
|
}
|
|
}
|
|
ASSERT(path_key != 0);
|
|
RegCloseKey(path_key);
|
|
path_key = 0;
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
Registry::overwrite_string_set(
|
|
HKEY base_key,
|
|
const _TCHAR *path,
|
|
int num_strings,
|
|
_TCHAR **strings,
|
|
_TCHAR **identifiers)
|
|
{
|
|
ASSERT(base_key == HKEY_CLASSES_ROOT
|
|
|| base_key == HKEY_CURRENT_CONFIG
|
|
|| base_key == HKEY_CURRENT_USER
|
|
|| base_key == HKEY_LOCAL_MACHINE
|
|
|| base_key == HKEY_USERS);
|
|
ASSERT(path != 0);
|
|
ASSERT(num_strings > 0);
|
|
ASSERT(strings != 0);
|
|
ASSERT(identifiers != 0);
|
|
HKEY path_key;
|
|
DWORD disposition;
|
|
long result = RegCreateKeyEx(base_key, path, 0, 0, REG_OPTION_NON_VOLATILE,
|
|
KEY_ALL_ACCESS, 0, &path_key, &disposition);
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
PRINT_DEBUG_MSG((_T("GROVELER: RegCreateKeyEx() failed with error %d\n"),
|
|
result));
|
|
return false;
|
|
}
|
|
ASSERT(path_key != 0);
|
|
for (int index = 0; index < num_strings; index++)
|
|
{
|
|
result = RegSetValueEx(
|
|
path_key, identifiers[index], 0, REG_SZ, (BYTE *)strings[index],
|
|
(_tcslen(strings[index]) + 1) * sizeof(_TCHAR));
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
PRINT_DEBUG_MSG((_T("GROVELER: RegSetValueEx() failed with error %d\n"),
|
|
result));
|
|
}
|
|
}
|
|
ASSERT(path_key != 0);
|
|
RegCloseKey(path_key);
|
|
path_key = 0;
|
|
return true;
|
|
}
|
|
|
|
void
|
|
Registry::create_key_ex(
|
|
HKEY hKey,
|
|
LPCTSTR lpSubKey,
|
|
DWORD Reserved,
|
|
LPTSTR lpClass,
|
|
DWORD dwOptions,
|
|
REGSAM samDesired,
|
|
LPSECURITY_ATTRIBUTES lpSecurityAttributes,
|
|
PHKEY phkResult,
|
|
LPDWORD lpdwDisposition)
|
|
{
|
|
DWORD result = RegCreateKeyEx(hKey, lpSubKey, Reserved, lpClass, dwOptions,
|
|
samDesired, lpSecurityAttributes, phkResult, lpdwDisposition);
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
PRINT_DEBUG_MSG((_T("GROVELER: RegCreateKeyEx() failed with error %d\n"),
|
|
result));
|
|
throw result;
|
|
}
|
|
}
|
|
|
|
void
|
|
Registry::open_key_ex(
|
|
HKEY hKey,
|
|
LPCTSTR lpSubKey,
|
|
DWORD ulOptions,
|
|
REGSAM samDesired,
|
|
PHKEY phkResult)
|
|
{
|
|
DWORD result =
|
|
RegOpenKeyEx(hKey, lpSubKey, ulOptions, samDesired, phkResult);
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
throw result;
|
|
}
|
|
}
|
|
|
|
void
|
|
Registry::close_key(
|
|
HKEY hKey)
|
|
{
|
|
DWORD result = RegCloseKey(hKey);
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
PRINT_DEBUG_MSG((_T("GROVELER: RegCloseKey() failed with error %d\n"), result));
|
|
throw result;
|
|
}
|
|
}
|
|
|
|
void
|
|
Registry::query_value_ex(
|
|
HKEY hKey,
|
|
LPCTSTR lpValueName,
|
|
LPDWORD lpReserved,
|
|
LPDWORD lpType,
|
|
LPBYTE lpData,
|
|
LPDWORD lpcbData)
|
|
{
|
|
DWORD result = RegQueryValueEx(hKey, lpValueName,
|
|
lpReserved, lpType, lpData, lpcbData);
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
throw result;
|
|
}
|
|
}
|
|
|
|
void
|
|
Registry::set_value_ex(
|
|
HKEY hKey,
|
|
LPCTSTR lpValueName,
|
|
DWORD Reserved,
|
|
DWORD dwType,
|
|
CONST BYTE *lpData,
|
|
DWORD cbData)
|
|
{
|
|
DWORD result =
|
|
RegSetValueEx(hKey, lpValueName, Reserved, dwType, lpData, cbData);
|
|
if (result != ERROR_SUCCESS)
|
|
{
|
|
PRINT_DEBUG_MSG((_T("GROVELER: RegSetValueEx() failed with error %d\n"), result));
|
|
throw result;
|
|
}
|
|
}
|
|
|
|
void
|
|
Registry::load_string_into_value(
|
|
EntryType type,
|
|
const _TCHAR *string,
|
|
void *value)
|
|
{
|
|
ASSERT(string != 0);
|
|
ASSERT(value != 0);
|
|
switch (type)
|
|
{
|
|
case entry_bool:
|
|
*((bool *)value) = _ttoi(string) != 0;
|
|
break;
|
|
case entry_char:
|
|
*((_TCHAR *)value) = string[0];
|
|
break;
|
|
case entry_int:
|
|
_stscanf(string, _T("%d"), (int *)value);
|
|
break;
|
|
case entry_int64:
|
|
_stscanf(string, _T("%I64d"), (__int64 *)value);
|
|
break;
|
|
case entry_double:
|
|
_stscanf(string, _T("%lf"), (double *)value);
|
|
break;
|
|
default:
|
|
ASSERT(false);
|
|
}
|
|
}
|
|
|
|
void
|
|
Registry::store_value_in_string(
|
|
EntryType type,
|
|
void *value,
|
|
_TCHAR *string)
|
|
{
|
|
ASSERT(string != 0);
|
|
ASSERT(value != 0);
|
|
switch (type)
|
|
{
|
|
case entry_bool:
|
|
_stprintf(string, *((bool *)value) ? _T("1") : _T("0"));
|
|
break;
|
|
case entry_char:
|
|
_stprintf(string, _T("%c"), *((_TCHAR *)value));
|
|
break;
|
|
case entry_int:
|
|
_stprintf(string, _T("%d"), *((int *)value));
|
|
break;
|
|
case entry_int64:
|
|
_stprintf(string, _T("%I64d"), *((__int64 *)value));
|
|
break;
|
|
case entry_double:
|
|
_stprintf(string, _T("%g"), *((double *)value));
|
|
break;
|
|
default:
|
|
ASSERT(false);
|
|
}
|
|
}
|