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.
2100 lines
42 KiB
2100 lines
42 KiB
/*++
|
|
|
|
Copyright (c) 1997 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
reg.c
|
|
|
|
Abstract:
|
|
|
|
Implements utilities to parse a registry key string, and also implements
|
|
wrappers to the registry API. There are three groups of APIs in this
|
|
source file: query functions, open and create functions, and registry
|
|
string parsing functions.
|
|
|
|
The query functions allow simplified querying, where the caller receives
|
|
a MemAlloc'd pointer to the data and does not have to worry about managing
|
|
the numerous parameters needed to do the query. The query functions
|
|
also allow filtering of values that are not the expected type. All
|
|
query functions have a version with 2 appended to the function name which
|
|
allow the caller to specify an alternative allocator and deallocator.
|
|
|
|
The open and create functions simplify the process of obtaining a key
|
|
handle. They allow the caller to specify a key string as input and return
|
|
the key handle as output.
|
|
|
|
The registry string parsing functions are utilities that can be used when
|
|
processing registry key strings. The functions extract the registry root
|
|
from a string, convert it into a handle, convert a hive handle into a
|
|
string, and so on.
|
|
|
|
Author:
|
|
|
|
Jim Schmidt (jimschm) 20-Mar-1997
|
|
|
|
Revisions:
|
|
|
|
jimschm 18-Sep-2000 Added cache
|
|
ovidiut 22-Feb-1999 Added GetRegSubkeysCount
|
|
calinn 23-Sep-1998 Fixed REG_SZ filtering
|
|
jimschm 25-Mar-1998 Added CreateEncodedRegistryStringEx
|
|
jimschm 21-Oct-1997 Added EnumFirstKeyInTree/EnumNextKeyInTree
|
|
marcw 16-Jul-1997 Added CreateEncodedRegistryString/FreeEncodedRegistryString
|
|
jimschm 22-Jun-1997 Added GetRegData
|
|
|
|
--*/
|
|
|
|
#include "pch.h"
|
|
#include "regp.h"
|
|
|
|
#ifdef DEBUG
|
|
#undef RegCloseKey
|
|
#endif
|
|
|
|
HKEY g_Root = HKEY_ROOT;
|
|
REGSAM g_OpenSam = KEY_ALL_ACCESS;
|
|
REGSAM g_CreateSam = KEY_ALL_ACCESS;
|
|
INT g_RegRefs;
|
|
|
|
#define DBG_REG "Reg"
|
|
|
|
//
|
|
// Implementation
|
|
//
|
|
|
|
|
|
BOOL
|
|
RegInitialize (
|
|
VOID
|
|
)
|
|
{
|
|
BOOL b = TRUE;
|
|
|
|
MYASSERT (g_RegRefs >= 0);
|
|
|
|
g_RegRefs++;
|
|
|
|
if (g_RegRefs == 1) {
|
|
|
|
RegInitializeCache (0);
|
|
}
|
|
|
|
return b;
|
|
}
|
|
|
|
|
|
VOID
|
|
RegTerminate (
|
|
VOID
|
|
)
|
|
{
|
|
MYASSERT (g_RegRefs > 0);
|
|
|
|
g_RegRefs--;
|
|
|
|
if (!g_RegRefs) {
|
|
RegTerminateCache ();
|
|
}
|
|
|
|
#ifdef DEBUG
|
|
RegTrackTerminate();
|
|
#endif
|
|
}
|
|
|
|
|
|
VOID
|
|
SetRegRoot (
|
|
IN HKEY Root
|
|
)
|
|
{
|
|
g_Root = Root;
|
|
}
|
|
|
|
HKEY
|
|
GetRegRoot (
|
|
VOID
|
|
)
|
|
{
|
|
return g_Root;
|
|
}
|
|
|
|
|
|
REGSAM
|
|
SetRegOpenAccessMode (
|
|
REGSAM Mode
|
|
)
|
|
{
|
|
REGSAM OldMode;
|
|
|
|
OldMode = g_OpenSam;
|
|
g_OpenSam = Mode;
|
|
|
|
return OldMode;
|
|
}
|
|
|
|
REGSAM
|
|
GetRegOpenAccessMode (
|
|
REGSAM Mode
|
|
)
|
|
{
|
|
return g_OpenSam;
|
|
}
|
|
|
|
REGSAM
|
|
SetRegCreateAccessMode (
|
|
REGSAM Mode
|
|
)
|
|
{
|
|
REGSAM OldMode;
|
|
|
|
OldMode = g_CreateSam;
|
|
g_CreateSam = Mode;
|
|
|
|
return OldMode;
|
|
}
|
|
|
|
REGSAM
|
|
GetRegCreateAccessMode (
|
|
REGSAM Mode
|
|
)
|
|
{
|
|
return g_CreateSam;
|
|
}
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
OpenRegKeyStrA and OpenRegKeyStrW parse a text string that specifies a
|
|
registry key into the hive and subkey, and then they open the subkey
|
|
and return the handle.
|
|
|
|
Arguments:
|
|
|
|
RegKey - Specifies the complete path to the registry subkey, including
|
|
the hive.
|
|
|
|
Return Value:
|
|
|
|
A non-NULL registry handle if successful, or NULL if either the subkey
|
|
could not be opened or the string is malformed.
|
|
|
|
--*/
|
|
|
|
HKEY
|
|
RealOpenRegKeyStrA (
|
|
IN PCSTR RegKey
|
|
DEBUG_TRACKING_PARAMS
|
|
)
|
|
{
|
|
DWORD End;
|
|
HKEY RootKey;
|
|
HKEY Key;
|
|
HKEY parentKey;
|
|
PCSTR lastWack;
|
|
|
|
//
|
|
// Attempt to use cache
|
|
//
|
|
|
|
Key = RegGetKeyFromCacheA (RegKey, NULL, g_OpenSam, TRUE);
|
|
if (Key) {
|
|
TRACK_KEYA (Key, RegKey);
|
|
return Key;
|
|
}
|
|
|
|
//
|
|
// Attempt to use cache for parent
|
|
//
|
|
|
|
lastWack = _mbsrchr (RegKey, '\\');
|
|
if (lastWack) {
|
|
parentKey = RegGetKeyFromCacheA (RegKey, lastWack, g_OpenSam, FALSE);
|
|
if (parentKey) {
|
|
Key = OpenRegKeyWorkerA (parentKey, lastWack + 1 /* , */ DEBUG_TRACKING_ARGS);
|
|
RegAddKeyToCacheA (RegKey, Key, g_OpenSam);
|
|
return Key;
|
|
}
|
|
}
|
|
|
|
//
|
|
// Not in cache; use full api
|
|
//
|
|
|
|
DEBUGMSGA ((DBG_REG, "Opening %s", RegKey));
|
|
|
|
RootKey = ConvertRootStringToKeyA (RegKey, &End);
|
|
if (!RootKey) {
|
|
return NULL;
|
|
}
|
|
|
|
if (!RegKey[End]) {
|
|
OurRegOpenRootKeyA (RootKey, RegKey /* , */ DEBUG_TRACKING_ARGS);
|
|
return RootKey;
|
|
}
|
|
|
|
Key = OpenRegKeyWorkerA (RootKey, &RegKey[End] /* , */ DEBUG_TRACKING_ARGS);
|
|
|
|
if (Key) {
|
|
RegAddKeyToCacheA (RegKey, Key, g_OpenSam);
|
|
RegRecordParentInCacheA (RegKey, lastWack);
|
|
}
|
|
|
|
return Key;
|
|
}
|
|
|
|
|
|
HKEY
|
|
RealOpenRegKeyStrW (
|
|
IN PCWSTR RegKey
|
|
DEBUG_TRACKING_PARAMS
|
|
)
|
|
{
|
|
DWORD End;
|
|
HKEY RootKey;
|
|
HKEY Key;
|
|
HKEY parentKey;
|
|
PCWSTR lastWack;
|
|
|
|
//
|
|
// Attempt to use cache
|
|
//
|
|
|
|
Key = RegGetKeyFromCacheW (RegKey, NULL, g_OpenSam, TRUE);
|
|
if (Key) {
|
|
TRACK_KEYW (Key, RegKey);
|
|
return Key;
|
|
}
|
|
|
|
//
|
|
// Attempt to use cache for parent
|
|
//
|
|
|
|
lastWack = wcsrchr (RegKey, L'\\');
|
|
if (lastWack) {
|
|
parentKey = RegGetKeyFromCacheW (RegKey, lastWack, g_OpenSam, FALSE);
|
|
if (parentKey) {
|
|
Key = OpenRegKeyWorkerW (parentKey, lastWack + 1 /* , */ DEBUG_TRACKING_ARGS);
|
|
RegAddKeyToCacheW (RegKey, Key, g_OpenSam);
|
|
return Key;
|
|
}
|
|
}
|
|
|
|
//
|
|
// Not in cache; use full api
|
|
//
|
|
|
|
DEBUGMSGW ((DBG_REG, "Opening %s", RegKey));
|
|
|
|
RootKey = ConvertRootStringToKeyW (RegKey, &End);
|
|
if (!RootKey) {
|
|
return NULL;
|
|
}
|
|
|
|
if (!RegKey[End]) {
|
|
OurRegOpenRootKeyW (RootKey, RegKey /* , */ DEBUG_TRACKING_ARGS);
|
|
return RootKey;
|
|
}
|
|
|
|
Key = OpenRegKeyWorkerW (RootKey, &RegKey[End] /* , */ DEBUG_TRACKING_ARGS);
|
|
|
|
if (Key) {
|
|
RegAddKeyToCacheW (RegKey, Key, g_OpenSam);
|
|
RegRecordParentInCacheW (RegKey, lastWack);
|
|
}
|
|
|
|
return Key;
|
|
}
|
|
|
|
HKEY
|
|
RealOpenRegKeyStrW1 (
|
|
IN PCWSTR RegKey
|
|
DEBUG_TRACKING_PARAMS
|
|
)
|
|
{
|
|
PCSTR AnsiRegKey;
|
|
HKEY Key;
|
|
|
|
AnsiRegKey = ConvertWtoA (RegKey);
|
|
if (!AnsiRegKey) {
|
|
return NULL;
|
|
}
|
|
|
|
Key = RealOpenRegKeyStrA (AnsiRegKey /* , */ DEBUG_TRACKING_ARGS);
|
|
|
|
FreeConvertedStr (AnsiRegKey);
|
|
|
|
return Key;
|
|
}
|
|
|
|
BOOL
|
|
DeleteRegKeyStrA (
|
|
IN PCSTR RegKey
|
|
)
|
|
{
|
|
DWORD End;
|
|
HKEY RootKey;
|
|
|
|
RootKey = ConvertRootStringToKeyA (RegKey, &End);
|
|
if (!RootKey) {
|
|
return FALSE;
|
|
}
|
|
|
|
if (!RegKey[End]) {
|
|
return FALSE;
|
|
}
|
|
|
|
return (RegDeleteKeyA (RootKey, &RegKey[End]) == ERROR_SUCCESS);
|
|
}
|
|
|
|
BOOL
|
|
DeleteRegKeyStrW (
|
|
IN PCWSTR RegKey
|
|
)
|
|
{
|
|
DWORD End;
|
|
HKEY RootKey;
|
|
|
|
RootKey = ConvertRootStringToKeyW (RegKey, &End);
|
|
if (!RootKey) {
|
|
return FALSE;
|
|
}
|
|
|
|
if (!RegKey[End]) {
|
|
return FALSE;
|
|
}
|
|
|
|
return (RegDeleteKeyW (RootKey, &RegKey[End]) == ERROR_SUCCESS);
|
|
}
|
|
|
|
BOOL
|
|
DeleteRegKeyStrW1 (
|
|
IN PCWSTR RegKey
|
|
)
|
|
{
|
|
PCSTR AnsiRegKey;
|
|
BOOL result = FALSE;
|
|
|
|
AnsiRegKey = ConvertWtoA (RegKey);
|
|
if (!AnsiRegKey) {
|
|
return FALSE;
|
|
}
|
|
|
|
result = DeleteRegKeyStrA (AnsiRegKey);
|
|
|
|
FreeConvertedStr (AnsiRegKey);
|
|
|
|
return result;
|
|
}
|
|
|
|
BOOL
|
|
DeleteEmptyRegKeyStrA (
|
|
IN PCSTR RegKey
|
|
)
|
|
{
|
|
DWORD End;
|
|
LONG rc;
|
|
DWORD subKeys;
|
|
DWORD values;
|
|
HKEY rootKey;
|
|
HKEY subKey;
|
|
|
|
rootKey = ConvertRootStringToKeyA (RegKey, &End);
|
|
if (!rootKey) {
|
|
SetLastError (ERROR_INVALID_PARAMETER);
|
|
return FALSE;
|
|
}
|
|
|
|
if (!RegKey[End]) {
|
|
SetLastError (ERROR_INVALID_PARAMETER);
|
|
return FALSE;
|
|
}
|
|
|
|
subKey = OpenRegKeyA (rootKey, &RegKey[End]);
|
|
if (!subKey) {
|
|
return TRUE;
|
|
}
|
|
|
|
rc = RegQueryInfoKey (subKey, NULL, NULL, NULL, &subKeys, NULL, NULL, &values, NULL, NULL, NULL, NULL);
|
|
|
|
CloseRegKey (subKey);
|
|
|
|
if (rc != ERROR_SUCCESS) {
|
|
SetLastError (rc);
|
|
return FALSE;
|
|
}
|
|
|
|
if (subKeys || values) {
|
|
SetLastError (ERROR_ACCESS_DENIED);
|
|
return FALSE;
|
|
}
|
|
|
|
rc = RegDeleteKeyA (rootKey, &RegKey[End]);
|
|
if (rc != ERROR_SUCCESS) {
|
|
SetLastError (rc);
|
|
return FALSE;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL
|
|
DeleteEmptyRegKeyStrW (
|
|
IN PCWSTR RegKey
|
|
)
|
|
{
|
|
DWORD End;
|
|
LONG rc;
|
|
DWORD subKeys;
|
|
DWORD values;
|
|
HKEY rootKey;
|
|
HKEY subKey;
|
|
|
|
rootKey = ConvertRootStringToKeyW (RegKey, &End);
|
|
if (!rootKey) {
|
|
SetLastError (ERROR_INVALID_PARAMETER);
|
|
return FALSE;
|
|
}
|
|
|
|
if (!RegKey[End]) {
|
|
SetLastError (ERROR_INVALID_PARAMETER);
|
|
return FALSE;
|
|
}
|
|
|
|
subKey = OpenRegKeyW (rootKey, &RegKey[End]);
|
|
if (!subKey) {
|
|
return TRUE;
|
|
}
|
|
|
|
rc = RegQueryInfoKey (subKey, NULL, NULL, NULL, &subKeys, NULL, NULL, &values, NULL, NULL, NULL, NULL);
|
|
|
|
CloseRegKey (subKey);
|
|
|
|
if (rc != ERROR_SUCCESS) {
|
|
SetLastError (rc);
|
|
return FALSE;
|
|
}
|
|
|
|
if (subKeys || values) {
|
|
SetLastError (ERROR_ACCESS_DENIED);
|
|
return FALSE;
|
|
}
|
|
|
|
rc = RegDeleteKeyW (rootKey, &RegKey[End]);
|
|
if (rc != ERROR_SUCCESS) {
|
|
SetLastError (rc);
|
|
return FALSE;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL
|
|
DeleteEmptyRegKeyStrW1 (
|
|
IN PCWSTR RegKey
|
|
)
|
|
{
|
|
PCSTR AnsiRegKey;
|
|
BOOL result = FALSE;
|
|
|
|
AnsiRegKey = ConvertWtoA (RegKey);
|
|
if (!AnsiRegKey) {
|
|
return FALSE;
|
|
}
|
|
|
|
result = DeleteEmptyRegKeyStrA (AnsiRegKey);
|
|
|
|
FreeConvertedStr (AnsiRegKey);
|
|
|
|
return result;
|
|
}
|
|
|
|
PVOID
|
|
MemAllocWrapper (
|
|
IN DWORD Size
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
pemAllocWrapper implements a default allocation routine. The APIs
|
|
that have a "2" at the end allow the caller to supply an alternative
|
|
allocator or deallocator. The routines without the "2" use this
|
|
default allocator.
|
|
|
|
Arguments:
|
|
|
|
Size - Specifies the amount of memory (in bytes) to allocate
|
|
|
|
Return Value:
|
|
|
|
A pointer to a block of memory that can hold Size bytes, or NULL
|
|
if allocation fails.
|
|
|
|
--*/
|
|
|
|
{
|
|
return MemAlloc (g_hHeap, 0, Size);
|
|
}
|
|
|
|
|
|
VOID
|
|
MemFreeWrapper (
|
|
IN PCVOID Mem
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
MemFreeWrapper implements a default deallocation routine.
|
|
See MemAllocWrapper above.
|
|
|
|
Arguments:
|
|
|
|
Mem - Specifies the block of memory to free, and was allocated by the
|
|
MemAllocWrapper function.
|
|
|
|
Return Value:
|
|
|
|
none
|
|
|
|
--*/
|
|
|
|
{
|
|
MemFree (g_hHeap, 0, Mem);
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
GetRegValueData2A and GetRegValueData2W query a registry value and
|
|
return the data as a pointer. They use the specified Alloc and Free
|
|
routines to allocate and free the memory as needed.
|
|
|
|
A GetRegValueData macro is defined, and it uses the default allocators,
|
|
simplifying the function parameters and allowing the caller to free
|
|
the return value via MemFree.
|
|
|
|
Arguments:
|
|
|
|
hKey - Specifies the registry key that holds the specified value.
|
|
|
|
Value - Specifies the value name to query.
|
|
|
|
Alloc - Specifies the allocation routine, called to allocate a block of
|
|
memory for the return data.
|
|
|
|
FreeRoutine - Specifies the deallocation routine, called if an error is encountered
|
|
during processing.
|
|
|
|
Return Value:
|
|
|
|
A pointer to the data retrieved, or NULL if the value does not exist or an
|
|
error occurred. Call GetLastError to obtian the failure code.
|
|
|
|
--*/
|
|
|
|
PBYTE
|
|
GetRegValueData2A (
|
|
IN HKEY hKey,
|
|
IN PCSTR Value,
|
|
IN ALLOCATOR AllocRoutine,
|
|
IN DEALLOCATOR FreeRoutine
|
|
)
|
|
{
|
|
LONG rc;
|
|
DWORD BufSize;
|
|
PBYTE DataBuf;
|
|
|
|
rc = RegQueryValueExA (hKey, Value, NULL, NULL, NULL, &BufSize);
|
|
if (rc != ERROR_SUCCESS) {
|
|
SetLastError ((DWORD)rc);
|
|
return NULL;
|
|
}
|
|
|
|
DataBuf = (PBYTE) AllocRoutine (BufSize + sizeof (CHAR));
|
|
rc = RegQueryValueExA (hKey, Value, NULL, NULL, DataBuf, &BufSize);
|
|
|
|
if (rc == ERROR_SUCCESS) {
|
|
*((PSTR) DataBuf + BufSize) = 0;
|
|
return DataBuf;
|
|
}
|
|
|
|
FreeRoutine (DataBuf);
|
|
SetLastError ((DWORD)rc);
|
|
return NULL;
|
|
}
|
|
|
|
|
|
PBYTE
|
|
GetRegValueData2W (
|
|
IN HKEY hKey,
|
|
IN PCWSTR Value,
|
|
IN ALLOCATOR AllocRoutine,
|
|
IN DEALLOCATOR FreeRoutine
|
|
)
|
|
{
|
|
LONG rc;
|
|
DWORD BufSize = 0;
|
|
PBYTE DataBuf;
|
|
|
|
rc = RegQueryValueExW (hKey, Value, NULL, NULL, NULL, &BufSize);
|
|
if (rc != ERROR_SUCCESS) {
|
|
SetLastError ((DWORD)rc);
|
|
return NULL;
|
|
}
|
|
|
|
|
|
DataBuf = (PBYTE) AllocRoutine (BufSize + sizeof(WCHAR));
|
|
rc = RegQueryValueExW (hKey, Value, NULL, NULL, DataBuf, &BufSize);
|
|
|
|
if (rc == ERROR_SUCCESS) {
|
|
*((PWSTR) (DataBuf + BufSize)) = 0;
|
|
return DataBuf;
|
|
}
|
|
|
|
FreeRoutine (DataBuf);
|
|
SetLastError ((DWORD)rc);
|
|
return NULL;
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
GetRegValueDataOfType2A and GetRegValueDataOfType2W are extensions of
|
|
GetRegValueData. They only return a data pointer when the data stored
|
|
in the registry value is the correct type.
|
|
|
|
Arguments:
|
|
|
|
hKey - Specifies the registry key to query
|
|
|
|
Value - Specifies the value name to query
|
|
|
|
MustBeType - Specifies the type of data (a REG_* constant). If the specified
|
|
value has data but is a different type, NULL will be returned.
|
|
|
|
AllocRoutine - Specifies the allocation routine, called to allocate the return data.
|
|
|
|
FreeRoutine - Specifies the deallocation routine, called when an error is encountered.
|
|
|
|
Return Value:
|
|
|
|
If successful, returns a pointer to data that matches the specified type.
|
|
If the data is a different type, the value name does not exist, or an
|
|
error occurs during the query, NULL is returned, and the failure code
|
|
can be obtained from GetLastError.
|
|
|
|
--*/
|
|
|
|
|
|
PBYTE
|
|
GetRegValueDataOfType2A (
|
|
IN HKEY hKey,
|
|
IN PCSTR Value,
|
|
IN DWORD MustBeType,
|
|
IN ALLOCATOR AllocRoutine,
|
|
IN DEALLOCATOR FreeRoutine
|
|
)
|
|
{
|
|
LONG rc;
|
|
DWORD BufSize = 0;
|
|
PBYTE DataBuf;
|
|
DWORD Type;
|
|
|
|
rc = RegQueryValueExA (hKey, Value, NULL, &Type, NULL, &BufSize);
|
|
if (rc != ERROR_SUCCESS) {
|
|
SetLastError ((DWORD)rc);
|
|
return NULL;
|
|
}
|
|
|
|
switch (MustBeType) {
|
|
|
|
case REG_SZ:
|
|
case REG_EXPAND_SZ:
|
|
if (Type == REG_SZ) {
|
|
break;
|
|
}
|
|
if (Type == REG_EXPAND_SZ) {
|
|
break;
|
|
}
|
|
return NULL;
|
|
|
|
default:
|
|
if (Type == MustBeType) {
|
|
break;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
DataBuf = (PBYTE) AllocRoutine (BufSize + sizeof (WORD));
|
|
rc = RegQueryValueExA (hKey, Value, NULL, NULL, DataBuf, &BufSize);
|
|
|
|
if (rc == ERROR_SUCCESS) {
|
|
*((PWORD) (DataBuf + BufSize)) = 0;
|
|
return DataBuf;
|
|
}
|
|
|
|
MYASSERT (FALSE); //lint !e506
|
|
FreeRoutine (DataBuf);
|
|
SetLastError ((DWORD)rc);
|
|
return NULL;
|
|
}
|
|
|
|
|
|
PBYTE
|
|
GetRegValueDataOfType2W (
|
|
IN HKEY hKey,
|
|
IN PCWSTR Value,
|
|
IN DWORD MustBeType,
|
|
IN ALLOCATOR AllocRoutine,
|
|
IN DEALLOCATOR FreeRoutine
|
|
)
|
|
{
|
|
LONG rc;
|
|
DWORD BufSize;
|
|
PBYTE DataBuf;
|
|
DWORD Type;
|
|
|
|
rc = RegQueryValueExW (hKey, Value, NULL, &Type, NULL, &BufSize);
|
|
if (rc != ERROR_SUCCESS) {
|
|
SetLastError ((DWORD)rc);
|
|
return NULL;
|
|
}
|
|
switch (MustBeType) {
|
|
case REG_SZ:
|
|
case REG_EXPAND_SZ:
|
|
if (Type == REG_SZ) break;
|
|
if (Type == REG_EXPAND_SZ) break;
|
|
return NULL;
|
|
case REG_DWORD:
|
|
case REG_DWORD_BIG_ENDIAN:
|
|
if (Type == REG_DWORD) break;
|
|
if (Type == REG_DWORD_BIG_ENDIAN) break;
|
|
return NULL;
|
|
default:
|
|
if (Type == MustBeType) break;
|
|
return NULL;
|
|
}
|
|
|
|
DataBuf = (PBYTE) AllocRoutine (BufSize + sizeof(WCHAR));
|
|
rc = RegQueryValueExW (hKey, Value, NULL, NULL, DataBuf, &BufSize);
|
|
|
|
if (rc == ERROR_SUCCESS) {
|
|
*((PWSTR) (DataBuf + BufSize)) = 0;
|
|
return DataBuf;
|
|
}
|
|
|
|
MYASSERT (FALSE); //lint !e506
|
|
FreeRoutine (DataBuf);
|
|
SetLastError ((DWORD)rc);
|
|
return NULL;
|
|
}
|
|
|
|
|
|
BOOL
|
|
GetRegValueTypeAndSizeA (
|
|
IN HKEY Key,
|
|
IN PCSTR ValueName,
|
|
OUT PDWORD OutType, OPTIONAL
|
|
OUT PDWORD OutSize OPTIONAL
|
|
)
|
|
{
|
|
LONG rc;
|
|
DWORD Type;
|
|
DWORD Size = 0;
|
|
|
|
rc = RegQueryValueExA (Key, ValueName, NULL, &Type, NULL, &Size);
|
|
|
|
if (rc == ERROR_SUCCESS) {
|
|
if (OutType) {
|
|
*OutType = Type;
|
|
}
|
|
|
|
if (OutSize) {
|
|
*OutSize = Size;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
|
|
BOOL
|
|
GetRegValueTypeAndSizeW (
|
|
IN HKEY Key,
|
|
IN PCWSTR ValueName,
|
|
OUT PDWORD OutType, OPTIONAL
|
|
OUT PDWORD OutSize OPTIONAL
|
|
)
|
|
{
|
|
LONG rc;
|
|
DWORD Type;
|
|
DWORD Size;
|
|
|
|
rc = RegQueryValueExW (Key, ValueName, NULL, &Type, NULL, &Size);
|
|
|
|
if (rc == ERROR_SUCCESS) {
|
|
if (OutType) {
|
|
*OutType = Type;
|
|
}
|
|
|
|
if (OutSize) {
|
|
*OutSize = Size;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
GetRegKeyData2A and GetRegKeyData2W return default data associated
|
|
with a registry key. They open the specified subkey, query the value,
|
|
close the subkey and return the data.
|
|
|
|
Arguments:
|
|
|
|
Parent - Specifies the key that contains SubKey.
|
|
|
|
SubKey - Specifies the name of the subkey to obtain the default value for.
|
|
|
|
AllocRoutine - Specifies the allocation routine, called to allocate a block of
|
|
memory for the registry data.
|
|
|
|
FreeRoutine - Specifies the deallocation routine, called to free the block of
|
|
data if an error occurs.
|
|
|
|
Return Value:
|
|
|
|
A pointer to the block of data obtained from the subkey's default value,
|
|
or NULL if the subkey does not exist or an error was encountered. Call
|
|
GetLastError for a failure code.
|
|
|
|
--*/
|
|
|
|
PBYTE
|
|
GetRegKeyData2A (
|
|
IN HKEY Parent,
|
|
IN PCSTR SubKey,
|
|
IN ALLOCATOR AllocRoutine,
|
|
IN DEALLOCATOR FreeRoutine
|
|
)
|
|
{
|
|
HKEY SubKeyHandle;
|
|
PBYTE Data;
|
|
|
|
SubKeyHandle = OpenRegKeyA (Parent, SubKey);
|
|
if (!SubKeyHandle) {
|
|
return NULL;
|
|
}
|
|
|
|
Data = GetRegValueData2A (SubKeyHandle, "", AllocRoutine, FreeRoutine);
|
|
|
|
CloseRegKey (SubKeyHandle);
|
|
|
|
return Data;
|
|
}
|
|
|
|
|
|
PBYTE
|
|
GetRegKeyData2W (
|
|
IN HKEY Parent,
|
|
IN PCWSTR SubKey,
|
|
IN ALLOCATOR AllocRoutine,
|
|
IN DEALLOCATOR FreeRoutine
|
|
)
|
|
{
|
|
HKEY SubKeyHandle;
|
|
PBYTE Data;
|
|
|
|
SubKeyHandle = OpenRegKeyW (Parent, SubKey);
|
|
if (!SubKeyHandle) {
|
|
return NULL;
|
|
}
|
|
|
|
Data = GetRegValueData2W (SubKeyHandle, L"", AllocRoutine, FreeRoutine);
|
|
|
|
CloseRegKey (SubKeyHandle);
|
|
|
|
return Data;
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
GetRegData2A and GetRegData2W open a registry key, query a value,
|
|
close the registry key and return the value.
|
|
|
|
Arguments:
|
|
|
|
KeyString - Specifies the registry key to open
|
|
|
|
ValueName - Specifies the value to query
|
|
|
|
AllocRoutine - Specifies the allocation routine, used to allocate a block of
|
|
memory to hold the value data
|
|
|
|
FreeRoutine - Specifies the deallocation routine, used to free the block of
|
|
memory when an error is encountered.
|
|
|
|
Return Value:
|
|
|
|
A pointer to the registry data retrieved, or NULL if the key or value
|
|
does not exist, or if an error occurs. Call GetLastError for a failure code.
|
|
|
|
--*/
|
|
|
|
PBYTE
|
|
GetRegData2A (
|
|
IN PCSTR KeyString,
|
|
IN PCSTR ValueName,
|
|
IN ALLOCATOR AllocRoutine,
|
|
IN DEALLOCATOR FreeRoutine
|
|
)
|
|
{
|
|
HKEY Key;
|
|
PBYTE Data;
|
|
|
|
Key = OpenRegKeyStrA (KeyString);
|
|
if (!Key) {
|
|
return NULL;
|
|
}
|
|
|
|
Data = GetRegValueData2A (Key, ValueName, AllocRoutine, FreeRoutine);
|
|
|
|
CloseRegKey (Key);
|
|
|
|
return Data;
|
|
}
|
|
|
|
|
|
PBYTE
|
|
GetRegData2W (
|
|
IN PCWSTR KeyString,
|
|
IN PCWSTR ValueName,
|
|
IN ALLOCATOR AllocRoutine,
|
|
IN DEALLOCATOR FreeRoutine
|
|
)
|
|
{
|
|
HKEY Key;
|
|
PBYTE Data;
|
|
|
|
Key = OpenRegKeyStrW (KeyString);
|
|
if (!Key) {
|
|
return NULL;
|
|
}
|
|
|
|
Data = GetRegValueData2W (Key, ValueName, AllocRoutine, FreeRoutine);
|
|
|
|
CloseRegKey (Key);
|
|
|
|
return Data;
|
|
}
|
|
|
|
|
|
BOOL
|
|
GetRegSubkeysCount (
|
|
IN HKEY ParentKey,
|
|
OUT PDWORD SubKeyCount, OPTIONAL
|
|
OUT PDWORD MaxSubKeyLen OPTIONAL
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
GetRegSubkeysCount retrieves the number of subkeys of a given parent key.
|
|
|
|
Arguments:
|
|
|
|
ParentKey - Specifies a handle to the parent registry key.
|
|
|
|
SubKeyCount - Receives the number of subkeys
|
|
|
|
MaxSubKeyLen - Receives the length, in chars, of the longest subkey string
|
|
|
|
Return Value:
|
|
|
|
TRUE if the count was retrieved successfully, FALSE otherwise.
|
|
In this case, call GetLastError for a failure code.
|
|
|
|
--*/
|
|
|
|
{
|
|
LONG rc;
|
|
|
|
rc = RegQueryInfoKey (
|
|
ParentKey,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
SubKeyCount,
|
|
MaxSubKeyLen,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
NULL
|
|
);
|
|
if (rc != ERROR_SUCCESS) {
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
CreateRegKeyA and CreateRegKeyW create a subkey if it does not
|
|
exist already, or open a subkey if it already exists.
|
|
|
|
Arguments:
|
|
|
|
ParentKey - Specifies a handle to the parent registry key to contain
|
|
the new key.
|
|
|
|
NewKeyName - Specifies the name of the subkey to create or open.
|
|
|
|
Return Value:
|
|
|
|
The handle to an open registry key upon success, or NULL if an
|
|
error occurred. Call GetLastError for a failure code.
|
|
|
|
--*/
|
|
|
|
HKEY
|
|
pCreateRegKeyWorkerA (
|
|
IN HKEY ParentKey,
|
|
IN PCSTR NewKeyName
|
|
DEBUG_TRACKING_PARAMS
|
|
)
|
|
{
|
|
LONG rc;
|
|
HKEY SubKey;
|
|
DWORD DontCare;
|
|
|
|
rc = OurRegCreateKeyExA (
|
|
ParentKey,
|
|
NewKeyName,
|
|
0,
|
|
NULL,
|
|
0,
|
|
g_CreateSam,
|
|
NULL,
|
|
&SubKey,
|
|
&DontCare
|
|
DEBUG_TRACKING_ARGS
|
|
);
|
|
|
|
if (rc != ERROR_SUCCESS) {
|
|
SetLastError ((DWORD)rc);
|
|
return NULL;
|
|
}
|
|
|
|
return SubKey;
|
|
}
|
|
|
|
|
|
HKEY
|
|
RealCreateRegKeyA (
|
|
IN HKEY ParentKey,
|
|
IN PCSTR NewKeyName
|
|
DEBUG_TRACKING_PARAMS
|
|
)
|
|
{
|
|
HKEY result;
|
|
|
|
result = pCreateRegKeyWorkerA (ParentKey, NewKeyName /* , */ DEBUG_TRACKING_ARGS);
|
|
RegAddKeyToCacheA ("", result, g_CreateSam);
|
|
|
|
return result;
|
|
}
|
|
|
|
HKEY
|
|
pCreateRegKeyWorkerW (
|
|
IN HKEY ParentKey,
|
|
IN PCWSTR NewKeyName
|
|
DEBUG_TRACKING_PARAMS
|
|
)
|
|
{
|
|
LONG rc;
|
|
HKEY SubKey;
|
|
DWORD DontCare;
|
|
|
|
rc = OurRegCreateKeyExW (
|
|
ParentKey,
|
|
NewKeyName,
|
|
0,
|
|
NULL,
|
|
0,
|
|
g_CreateSam,
|
|
NULL,
|
|
&SubKey,
|
|
&DontCare
|
|
DEBUG_TRACKING_ARGS
|
|
);
|
|
|
|
if (rc != ERROR_SUCCESS) {
|
|
SetLastError ((DWORD)rc);
|
|
return NULL;
|
|
}
|
|
|
|
return SubKey;
|
|
}
|
|
|
|
|
|
HKEY
|
|
RealCreateRegKeyW (
|
|
IN HKEY ParentKey,
|
|
IN PCWSTR NewKeyName
|
|
DEBUG_TRACKING_PARAMS
|
|
)
|
|
{
|
|
HKEY result;
|
|
|
|
result = pCreateRegKeyWorkerW (ParentKey, NewKeyName /* , */ DEBUG_TRACKING_ARGS);
|
|
RegAddKeyToCacheW (L"", result, g_CreateSam);
|
|
|
|
return result;
|
|
}
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
CreateRegKeyStrA and CreateRegKeyStrW create a subkey if it does not
|
|
exist already, or open a subkey if it already exists.
|
|
|
|
Arguments:
|
|
|
|
NewKeyName - Specifies the full path to the key to create or open.
|
|
|
|
Return Value:
|
|
|
|
The handle to an open registry key upon success, or NULL if an
|
|
error occurred. Call GetLastError for a failure code.
|
|
|
|
--*/
|
|
|
|
HKEY
|
|
RealCreateRegKeyStrA (
|
|
IN PCSTR NewKeyName
|
|
DEBUG_TRACKING_PARAMS
|
|
)
|
|
{
|
|
LONG rc;
|
|
DWORD DontCare;
|
|
CHAR RegKey[MAX_REGISTRY_KEYA];
|
|
PCSTR Start;
|
|
PCSTR End;
|
|
HKEY Parent, NewKey;
|
|
BOOL CloseParent = FALSE;
|
|
DWORD EndPos;
|
|
PCSTR lastWack;
|
|
HKEY parentKey;
|
|
|
|
//
|
|
// Attempt to use cache
|
|
//
|
|
|
|
NewKey = RegGetKeyFromCacheA (NewKeyName, NULL, g_CreateSam, TRUE);
|
|
if (NewKey) {
|
|
TRACK_KEYA (NewKey, NewKeyName);
|
|
return NewKey;
|
|
}
|
|
|
|
//
|
|
// Attempt to use cache for parent
|
|
//
|
|
|
|
lastWack = _mbsrchr (NewKeyName, '\\');
|
|
if (lastWack) {
|
|
parentKey = RegGetKeyFromCacheA (NewKeyName, lastWack, g_CreateSam, FALSE);
|
|
if (parentKey) {
|
|
NewKey = pCreateRegKeyWorkerA (parentKey, lastWack + 1 /* , */ DEBUG_TRACKING_ARGS);
|
|
RegAddKeyToCacheA (NewKeyName, NewKey, g_CreateSam);
|
|
return NewKey;
|
|
}
|
|
}
|
|
|
|
//
|
|
// Get the root
|
|
//
|
|
|
|
Parent = ConvertRootStringToKeyA (NewKeyName, &EndPos);
|
|
if (!Parent) {
|
|
return NULL;
|
|
}
|
|
|
|
Start = &NewKeyName[EndPos];
|
|
|
|
if (!(*Start)) {
|
|
OurRegOpenRootKeyA (Parent, NewKeyName/* , */ DEBUG_TRACKING_ARGS);
|
|
return Parent;
|
|
}
|
|
|
|
//
|
|
// Create each node until entire key exists
|
|
//
|
|
|
|
NewKey = NULL;
|
|
|
|
do {
|
|
//
|
|
// Find end of this node
|
|
//
|
|
|
|
End = _mbschr (Start, '\\');
|
|
if (!End) {
|
|
End = GetEndOfStringA (Start);
|
|
}
|
|
|
|
StringCopyABA (RegKey, Start, End);
|
|
|
|
//
|
|
// Try to open the key (unless it's the last in the string)
|
|
//
|
|
|
|
if (*End) { //lint !e613
|
|
rc = OurRegOpenKeyExA (
|
|
Parent,
|
|
RegKey,
|
|
0,
|
|
KEY_READ|KEY_CREATE_SUB_KEY,
|
|
&NewKey
|
|
DEBUG_TRACKING_ARGS
|
|
);
|
|
if (rc != ERROR_SUCCESS) {
|
|
NewKey = NULL;
|
|
}
|
|
} else {
|
|
NewKey = NULL;
|
|
}
|
|
|
|
//
|
|
// If open failed, create the key
|
|
//
|
|
|
|
if (NewKey) {
|
|
rc = ERROR_SUCCESS;
|
|
} else {
|
|
rc = OurRegCreateKeyExA (
|
|
Parent,
|
|
RegKey,
|
|
0,
|
|
NULL,
|
|
0,
|
|
g_CreateSam,
|
|
NULL,
|
|
&NewKey,
|
|
&DontCare
|
|
DEBUG_TRACKING_ARGS
|
|
);
|
|
}
|
|
|
|
if (CloseParent) {
|
|
CloseRegKey (Parent);
|
|
}
|
|
|
|
if (rc != ERROR_SUCCESS) {
|
|
SetLastError ((DWORD)rc);
|
|
return NULL;
|
|
}
|
|
|
|
Parent = NewKey;
|
|
CloseParent = TRUE;
|
|
|
|
//
|
|
// Go to next node
|
|
//
|
|
|
|
Start = End;
|
|
if (*Start) { //lint !e613
|
|
Start = _mbsinc (Start);
|
|
}
|
|
|
|
} while (*Start); //lint !e613
|
|
|
|
if (Parent) {
|
|
RegAddKeyToCacheA (NewKeyName, Parent, g_CreateSam);
|
|
RegRecordParentInCacheA (NewKeyName, lastWack);
|
|
}
|
|
|
|
return Parent;
|
|
}
|
|
|
|
|
|
HKEY
|
|
RealCreateRegKeyStrW (
|
|
IN PCWSTR NewKeyName
|
|
DEBUG_TRACKING_PARAMS
|
|
)
|
|
{
|
|
LONG rc;
|
|
DWORD DontCare;
|
|
WCHAR RegKey[MAX_REGISTRY_KEYW];
|
|
PCWSTR Start;
|
|
PCWSTR End;
|
|
HKEY Parent, NewKey;
|
|
BOOL CloseParent = FALSE;
|
|
DWORD EndPos;
|
|
PCWSTR lastWack;
|
|
HKEY parentKey;
|
|
|
|
//
|
|
// Attempt to use cache
|
|
//
|
|
|
|
NewKey = RegGetKeyFromCacheW (NewKeyName, NULL, g_CreateSam, TRUE);
|
|
if (NewKey) {
|
|
TRACK_KEYW (NewKey, NewKeyName);
|
|
return NewKey;
|
|
}
|
|
|
|
//
|
|
// Attempt to use cache for parent
|
|
//
|
|
|
|
lastWack = wcsrchr (NewKeyName, L'\\');
|
|
if (lastWack) {
|
|
parentKey = RegGetKeyFromCacheW (NewKeyName, lastWack, g_CreateSam, FALSE);
|
|
if (parentKey) {
|
|
NewKey = pCreateRegKeyWorkerW (parentKey, lastWack + 1 /* , */ DEBUG_TRACKING_ARGS);
|
|
RegAddKeyToCacheW (NewKeyName, NewKey, g_CreateSam);
|
|
return NewKey;
|
|
}
|
|
}
|
|
|
|
//
|
|
// Get the root
|
|
//
|
|
|
|
Parent = ConvertRootStringToKeyW (NewKeyName, &EndPos);
|
|
if (!Parent) {
|
|
return NULL;
|
|
}
|
|
|
|
Start = &NewKeyName[EndPos];
|
|
|
|
if (!(*Start)) {
|
|
OurRegOpenRootKeyW (Parent, NewKeyName/* , */ DEBUG_TRACKING_ARGS);
|
|
return Parent;
|
|
}
|
|
|
|
//
|
|
// Create each node until entire key exists
|
|
//
|
|
|
|
NewKey = NULL;
|
|
|
|
do {
|
|
//
|
|
// Find end of this node
|
|
//
|
|
|
|
End = wcschr (Start, '\\');
|
|
if (!End) {
|
|
End = GetEndOfStringW (Start);
|
|
}
|
|
|
|
StringCopyABW (RegKey, Start, End);
|
|
|
|
//
|
|
// Try to open the key (unless it's the last in the string)
|
|
//
|
|
|
|
if (*End) {
|
|
rc = OurRegOpenKeyExW (
|
|
Parent,
|
|
RegKey,
|
|
0,
|
|
KEY_READ|KEY_CREATE_SUB_KEY,
|
|
&NewKey
|
|
DEBUG_TRACKING_ARGS
|
|
);
|
|
if (rc != ERROR_SUCCESS) {
|
|
NewKey = NULL;
|
|
}
|
|
} else {
|
|
NewKey = NULL;
|
|
}
|
|
|
|
//
|
|
// If open failed, create the key
|
|
//
|
|
|
|
if (NewKey) {
|
|
rc = ERROR_SUCCESS;
|
|
} else {
|
|
rc = OurRegCreateKeyExW (
|
|
Parent,
|
|
RegKey,
|
|
0,
|
|
NULL,
|
|
0,
|
|
g_CreateSam,
|
|
NULL,
|
|
&NewKey,
|
|
&DontCare
|
|
DEBUG_TRACKING_ARGS
|
|
);
|
|
}
|
|
|
|
if (CloseParent) {
|
|
CloseRegKey (Parent);
|
|
}
|
|
|
|
if (rc != ERROR_SUCCESS) {
|
|
SetLastError ((DWORD)rc);
|
|
return NULL;
|
|
}
|
|
|
|
Parent = NewKey;
|
|
CloseParent = TRUE;
|
|
|
|
//
|
|
// Go to next node
|
|
//
|
|
|
|
Start = End;
|
|
if (*Start) {
|
|
Start++;
|
|
}
|
|
} while (*Start);
|
|
|
|
if (Parent) {
|
|
RegAddKeyToCacheW (NewKeyName, Parent, g_CreateSam);
|
|
RegRecordParentInCacheW (NewKeyName, lastWack);
|
|
}
|
|
|
|
return Parent;
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
OpenRegKeyA and OpenRegKeyW open a subkey.
|
|
|
|
Arguments:
|
|
|
|
ParentKey - Specifies a handle to the parent registry key to contain
|
|
the subkey.
|
|
|
|
KeyToOpen - Specifies the name of the subkey to open.
|
|
|
|
Return Value:
|
|
|
|
The handle to an open registry key upon success, or NULL if an
|
|
error occurred. Call GetLastError for a failure code.
|
|
|
|
--*/
|
|
|
|
HKEY
|
|
OpenRegKeyWorkerA (
|
|
IN HKEY ParentKey,
|
|
IN PCSTR KeyToOpen OPTIONAL
|
|
DEBUG_TRACKING_PARAMS
|
|
)
|
|
{
|
|
HKEY SubKey;
|
|
LONG rc;
|
|
|
|
rc = OurRegOpenKeyExA (
|
|
ParentKey,
|
|
KeyToOpen,
|
|
0,
|
|
g_OpenSam,
|
|
&SubKey
|
|
DEBUG_TRACKING_ARGS
|
|
);
|
|
|
|
if (rc != ERROR_SUCCESS) {
|
|
SetLastError ((DWORD)rc);
|
|
return NULL;
|
|
}
|
|
|
|
return SubKey;
|
|
}
|
|
|
|
|
|
HKEY
|
|
RealOpenRegKeyA (
|
|
IN HKEY ParentKey,
|
|
IN PCSTR KeyToOpen OPTIONAL
|
|
DEBUG_TRACKING_PARAMS
|
|
)
|
|
{
|
|
HKEY result;
|
|
|
|
result = OpenRegKeyWorkerA (ParentKey, KeyToOpen /* , */ DEBUG_TRACKING_ARGS);
|
|
RegAddKeyToCacheA ("", result, g_OpenSam);
|
|
|
|
return result;
|
|
}
|
|
|
|
|
|
HKEY
|
|
OpenRegKeyWorkerW (
|
|
IN HKEY ParentKey,
|
|
IN PCWSTR KeyToOpen
|
|
DEBUG_TRACKING_PARAMS
|
|
)
|
|
{
|
|
LONG rc;
|
|
HKEY SubKey;
|
|
|
|
rc = OurRegOpenKeyExW (
|
|
ParentKey,
|
|
KeyToOpen,
|
|
0,
|
|
g_OpenSam,
|
|
&SubKey
|
|
DEBUG_TRACKING_ARGS
|
|
);
|
|
|
|
if (rc != ERROR_SUCCESS) {
|
|
SetLastError ((DWORD)rc);
|
|
return NULL;
|
|
}
|
|
|
|
return SubKey;
|
|
}
|
|
|
|
|
|
HKEY
|
|
RealOpenRegKeyW (
|
|
IN HKEY ParentKey,
|
|
IN PCWSTR KeyToOpen OPTIONAL
|
|
DEBUG_TRACKING_PARAMS
|
|
)
|
|
{
|
|
HKEY result;
|
|
|
|
result = OpenRegKeyWorkerW (ParentKey, KeyToOpen /* , */ DEBUG_TRACKING_ARGS);
|
|
RegAddKeyToCacheW (L"", result, g_OpenSam);
|
|
|
|
return result;
|
|
}
|
|
|
|
|
|
LONG
|
|
CloseRegKeyWorker (
|
|
IN HKEY Key
|
|
)
|
|
{
|
|
LONG rc = ERROR_INVALID_HANDLE;
|
|
|
|
if (!Key) {
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
if (GetOffsetOfRootKey (Key)) {
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
__try {
|
|
rc = RegCloseKey (Key);
|
|
}
|
|
__except (TRUE) {
|
|
DEBUGMSG ((DBG_WHOOPS, "RegCloseKey threw an exception!"));
|
|
}
|
|
|
|
MYASSERT (rc == ERROR_SUCCESS);
|
|
|
|
return rc;
|
|
}
|
|
|
|
|
|
LONG
|
|
RealCloseRegKey (
|
|
IN HKEY Key
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
RealCloseRegKey closes the reg handle supplied, unless the handle is
|
|
a pre-defined Win32 handle. The CloseRegKey macro resolves directly
|
|
to this function in the free build, and to OurCloseRegKey in the
|
|
checked build.
|
|
|
|
Arguments:
|
|
|
|
Key - Specifies the reg handle to close
|
|
|
|
Return Value:
|
|
|
|
A standard Win32 error code indicating outcome.
|
|
|
|
--*/
|
|
|
|
{
|
|
if (RegDecrementRefCount (Key)) {
|
|
//
|
|
// Key is in the cache; don't call CloseRegKeyWorker. This will
|
|
// be done by the cache code.
|
|
//
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
return CloseRegKeyWorker (Key);
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
GetOffsetOfRootString returns a non-zero offset to the g_RegRoots table
|
|
below. The offset can be used with GetRootStringFromOffset and
|
|
GetRootKeyFromOffset.
|
|
|
|
Arguments:
|
|
|
|
RootString - A pointer to a string containing the path to a registry key
|
|
LengthPtr - A pointer to a variable that receives the length of the
|
|
registry root, including the joining backslash if it exists.
|
|
|
|
Return Value:
|
|
|
|
A non-zero offset to the g_RegRoots table, or zero if RootString does not
|
|
contain a registry root.
|
|
|
|
--*/
|
|
|
|
typedef struct {
|
|
PCSTR RootText;
|
|
PCWSTR WideRootText;
|
|
UINT TextLength;
|
|
HKEY RootKey;
|
|
} REGISTRYROOT, *PREGISTRYROOT;
|
|
|
|
static
|
|
REGISTRYROOT g_RegRoots[] = {
|
|
"HKR", L"HKR", 3, HKEY_ROOT,
|
|
"HKEY_ROOT", L"HKEY_ROOT", 9, HKEY_ROOT,
|
|
"HKLM", L"HKLM", 4, HKEY_LOCAL_MACHINE,
|
|
"HKEY_LOCAL_MACHINE", L"HKEY_LOCAL_MACHINE", 18, HKEY_LOCAL_MACHINE,
|
|
"HKU", L"HKU", 3, HKEY_USERS,
|
|
"HKEY_USERS", L"HKEY_USERS", 10, HKEY_USERS,
|
|
"HKCU", L"HKCU", 4, HKEY_CURRENT_USER,
|
|
"HKEY_CURRENT_USER", L"HKEY_CURRENT_USER", 17, HKEY_CURRENT_USER,
|
|
"HKCC", L"HKCC", 4, HKEY_CURRENT_CONFIG,
|
|
"HKEY_CURRENT_CONFIG", L"HKEY_CURRENT_CONFIG", 19, HKEY_CURRENT_CONFIG,
|
|
"HKCR", L"HKCR", 4, HKEY_CLASSES_ROOT,
|
|
"HKEY_CLASSES_ROOT", L"HKEY_CLASSES_ROOT", 17, HKEY_CLASSES_ROOT,
|
|
"HKDD", L"HKDD", 4, HKEY_DYN_DATA,
|
|
"HKEY_DYN_DATA", L"HKEY_DYN_DATA", 13, HKEY_DYN_DATA,
|
|
NULL, NULL, 0, NULL
|
|
};
|
|
|
|
#define REGROOTS 14
|
|
|
|
INT
|
|
GetOffsetOfRootStringA (
|
|
IN PCSTR RootString,
|
|
OUT PDWORD LengthPtr OPTIONAL
|
|
)
|
|
{
|
|
int i;
|
|
MBCHAR c;
|
|
|
|
for (i = 0 ; g_RegRoots[i].RootText ; i++) {
|
|
if (StringIMatchCharCountA (
|
|
RootString,
|
|
g_RegRoots[i].RootText,
|
|
g_RegRoots[i].TextLength
|
|
)) {
|
|
|
|
c = _mbsgetc (RootString, g_RegRoots[i].TextLength);
|
|
if (c && c != '\\') {
|
|
continue;
|
|
}
|
|
|
|
if (LengthPtr) {
|
|
*LengthPtr = g_RegRoots[i].TextLength;
|
|
if (c) {
|
|
*LengthPtr += 1;
|
|
}
|
|
}
|
|
|
|
return i + 1;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
INT
|
|
GetOffsetOfRootStringW (
|
|
IN PCWSTR RootString,
|
|
OUT PDWORD LengthPtr OPTIONAL
|
|
)
|
|
{
|
|
int i;
|
|
WCHAR c;
|
|
|
|
for (i = 0 ; g_RegRoots[i].RootText ; i++) {
|
|
if (!_wcsnicmp (RootString, g_RegRoots[i].WideRootText,
|
|
g_RegRoots[i].TextLength)
|
|
) {
|
|
c = _wcsgetc (RootString, g_RegRoots[i].TextLength);
|
|
if (c && c != L'\\') {
|
|
continue;
|
|
}
|
|
|
|
if (LengthPtr) {
|
|
*LengthPtr = g_RegRoots[i].TextLength;
|
|
if (c) {
|
|
*LengthPtr += 1;
|
|
}
|
|
}
|
|
|
|
return i + 1;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
GetOffsetOfRootKey returns a non-zero offset to the g_RegRoots table
|
|
corresponding to the root that matches the supplied HKEY. This offset
|
|
can be used with GetRootStringFromOffset and GetRootKeyFromOffset.
|
|
|
|
Arguments:
|
|
|
|
RootKey - Supplies the handle to locate in g_RegRoots table
|
|
|
|
Return Value:
|
|
|
|
A non-zero offset to the g_RegRoots table, or zero if the handle is not
|
|
a registry root.
|
|
|
|
--*/
|
|
|
|
INT
|
|
GetOffsetOfRootKey (
|
|
IN HKEY RootKey
|
|
)
|
|
{
|
|
INT i;
|
|
|
|
if (RootKey == g_Root) {
|
|
return 1;
|
|
}
|
|
|
|
for (i = 0 ; g_RegRoots[i].RootText ; i++) {
|
|
if (g_RegRoots[i].RootKey == RootKey) {
|
|
return i + 1;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
GetRootStringFromOffset and GetRootKeyFromOffset return a pointer to a
|
|
static string or HKEY, respectively. If the offset supplied is invalid,
|
|
these functions return NULL.
|
|
|
|
Arguments:
|
|
|
|
i - The offset as returned by GetOffsetOfRootString or GetOffsetOfRootKey
|
|
|
|
Return Value:
|
|
|
|
A pointer to a static string/HKEY, or NULL if offset is invalid
|
|
|
|
--*/
|
|
|
|
PCSTR
|
|
GetRootStringFromOffsetA (
|
|
IN INT i
|
|
)
|
|
{
|
|
if (i < 1 || i > REGROOTS) {
|
|
return NULL;
|
|
}
|
|
|
|
return g_RegRoots[i - 1].RootText;
|
|
}
|
|
|
|
PCWSTR
|
|
GetRootStringFromOffsetW (
|
|
IN INT i
|
|
)
|
|
{
|
|
if (i < 1 || i > REGROOTS) {
|
|
return NULL;
|
|
}
|
|
|
|
return g_RegRoots[i - 1].WideRootText;
|
|
}
|
|
|
|
HKEY
|
|
GetRootKeyFromOffset (
|
|
IN INT i
|
|
)
|
|
{
|
|
HKEY Ret;
|
|
|
|
if (i < 1 || i > REGROOTS) {
|
|
return NULL;
|
|
}
|
|
|
|
Ret = g_RegRoots[i - 1].RootKey;
|
|
if (Ret == HKEY_ROOT) {
|
|
Ret = g_Root;
|
|
}
|
|
|
|
return Ret;
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
ConvertRootStringToKey converts a registry key path's root to an HKEY.
|
|
|
|
Arguments:
|
|
|
|
RegPath - A pointer to a registry string that has a root at the begining
|
|
LengthPtr - An optional pointer to a variable that receives the length of
|
|
the root, including the joining backslash if it exists.
|
|
|
|
Return Value:
|
|
|
|
A handle to the registry key, or NULL if RegPath does not have a root
|
|
|
|
--*/
|
|
|
|
HKEY
|
|
ConvertRootStringToKeyA (
|
|
PCSTR RegPath,
|
|
PDWORD LengthPtr OPTIONAL
|
|
)
|
|
{
|
|
return GetRootKeyFromOffset (GetOffsetOfRootStringA (RegPath, LengthPtr));
|
|
}
|
|
|
|
HKEY
|
|
ConvertRootStringToKeyW (
|
|
PCWSTR RegPath,
|
|
PDWORD LengthPtr OPTIONAL
|
|
)
|
|
{
|
|
return GetRootKeyFromOffset (GetOffsetOfRootStringW (RegPath, LengthPtr));
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
ConvertKeyToRootString converts a root HKEY to a registry root string
|
|
|
|
Arguments:
|
|
|
|
RegRoot - A handle to a registry root
|
|
|
|
Return Value:
|
|
|
|
A pointer to a static string, or NULL if RegRoot is not a valid registry
|
|
root handle
|
|
|
|
--*/
|
|
|
|
PCSTR
|
|
ConvertKeyToRootStringA (
|
|
HKEY RegRoot
|
|
)
|
|
{
|
|
return GetRootStringFromOffsetA (GetOffsetOfRootKey (RegRoot));
|
|
}
|
|
|
|
PCWSTR
|
|
ConvertKeyToRootStringW (
|
|
HKEY RegRoot
|
|
)
|
|
{
|
|
return GetRootStringFromOffsetW (GetOffsetOfRootKey (RegRoot));
|
|
}
|
|
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
CreateEncodedRegistryStringEx is used to create a registry string in the format commonly
|
|
expected by w95upg reg routines. This format is:
|
|
|
|
EncodedKey\[EncodedValue]
|
|
|
|
Encoding is used to safely represent "special" characters
|
|
(such as MBS chars and certain punctuation marks.)
|
|
|
|
The [EncodedValue] part will exist only if Value is non null.
|
|
|
|
Arguments:
|
|
|
|
Key - Contains an unencoded registry key.
|
|
Value - Optionally contains an unencoded registry value.
|
|
Tree - Specifies that the registry key refers to the entire key
|
|
|
|
Return Value:
|
|
|
|
Returns a pointer to the encoded registry string, or NULL if there was an error.
|
|
|
|
--*/
|
|
|
|
PCSTR
|
|
CreateEncodedRegistryStringExA (
|
|
IN PCSTR Key,
|
|
IN PCSTR Value, OPTIONAL
|
|
IN BOOL Tree
|
|
)
|
|
{
|
|
PSTR rEncodedString = NULL;
|
|
DWORD requiredSize;
|
|
PSTR end;
|
|
|
|
//
|
|
// Determine required size and allocate buffer large enough to hold
|
|
// the encoded string.
|
|
//
|
|
requiredSize = SizeOfStringA(Key)*6 + (Value ? SizeOfStringA(Value)*6 : 0) + 10;
|
|
rEncodedString = AllocPathStringA(requiredSize);
|
|
|
|
//
|
|
// Encode the key portion of the string.
|
|
//
|
|
EncodeRuleCharsA(rEncodedString, Key);
|
|
|
|
//
|
|
// Finally, if a value exists, append it in encoded form. If a value does not exist,
|
|
// then add an '*' to the line.
|
|
//
|
|
if (Value) {
|
|
|
|
StringCopyA (AppendWackA (rEncodedString), "[");
|
|
end = GetEndOfStringA (rEncodedString);
|
|
EncodeRuleCharsA(end, Value);
|
|
StringCatA(end, "]");
|
|
|
|
} else if (Tree) {
|
|
StringCopyA (AppendWackA (rEncodedString), "*");
|
|
}
|
|
|
|
return rEncodedString;
|
|
}
|
|
|
|
|
|
PCWSTR
|
|
CreateEncodedRegistryStringExW (
|
|
IN PCWSTR Key,
|
|
IN PCWSTR Value, OPTIONAL
|
|
IN BOOL Tree
|
|
)
|
|
{
|
|
PWSTR rEncodedString = NULL;
|
|
DWORD requiredSize;
|
|
PWSTR end;
|
|
|
|
//
|
|
// Determine required size and allocate buffer large enough to hold
|
|
// the encoded string.
|
|
//
|
|
requiredSize = SizeOfStringW(Key)*6 + (Value ? SizeOfStringW(Value)*6 : 0) + 10;
|
|
rEncodedString = AllocPathStringW(requiredSize);
|
|
|
|
//
|
|
// Encode the key portion of the string.
|
|
//
|
|
EncodeRuleCharsW(rEncodedString, Key);
|
|
|
|
//
|
|
// Finally, if a value exists, append it in encoded form.
|
|
// If a value doesn't exist, add na '*' to the line.
|
|
//
|
|
if (Value) {
|
|
|
|
StringCopyW (AppendWackW (rEncodedString), L"[");
|
|
end = GetEndOfStringW (rEncodedString);
|
|
EncodeRuleCharsW(end, Value);
|
|
StringCatW(end, L"]");
|
|
} else if (Tree) {
|
|
StringCopyW (AppendWackW (rEncodedString), L"*");
|
|
}
|
|
|
|
return rEncodedString;
|
|
}
|
|
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
FreeEncodedRegistryString frees the memory allocated by a call to CreateEncodedRegistryString.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
|
|
Return Value:
|
|
|
|
None.
|
|
|
|
--*/
|
|
VOID
|
|
FreeEncodedRegistryStringA (
|
|
IN OUT PCSTR RegString
|
|
)
|
|
{
|
|
FreePathStringA(RegString);
|
|
}
|
|
|
|
|
|
VOID
|
|
FreeEncodedRegistryStringW (
|
|
IN OUT PCWSTR RegString
|
|
)
|
|
{
|
|
FreePathStringW(RegString);
|
|
}
|
|
|
|
|
|
|