regtools.cpp File Reference

#include "regtools.h"

Include dependency graph for regtools.cpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
NTSTATUS	RegTGetKeyHandle (IN HKEY hRootKey, IN PWCHAR KeyName, OUT HKEY *hKey)
VOID	RegTCloseKeyHandle (IN HKEY hKey)
BOOLEAN	RegTGetDwordValue (IN HKEY hRootKey, IN PCWSTR RegistryPath, IN PCWSTR Name, IN PULONG pUlong)
BOOLEAN	RegTGetStringValue (IN HKEY hRootKey, IN PCWSTR RegistryPath, IN PCWSTR Name, IN PWCHAR pStr, IN ULONG MaxLen)

Function Documentation

RegTCloseKeyHandle()

VOID RegTCloseKeyHandle

(

IN HKEY

hKey

)

Definition at line 86 of file regtools.cpp.

{
    if(!hKey) {
        return;
    }
    RegCloseKey(hKey);
} // end RegTCloseKeyHandle()

Referenced by RegTGetDwordValue(), RegTGetStringValue(), and UDFRegCheckParameterValue().

RegTGetDwordValue()

BOOLEAN RegTGetDwordValue	(	IN HKEY	hRootKey,
		IN PCWSTR	RegistryPath,
		IN PCWSTR	Name,
		IN PULONG	pUlong
	)

Definition at line 99 of file regtools.cpp.

{
#ifndef WIN_32_MODE
    UNICODE_STRING NameString;
    PKEY_VALUE_PARTIAL_INFORMATION ValInfo;
#endif //WIN_32_MODE
    ULONG len;
    NTSTATUS status;
    HKEY hKey;
    BOOLEAN retval = FALSE;
    BOOLEAN free_h = FALSE;
 
#ifdef WIN_32_MODE
    if(!hRootKey)
        hRootKey = HKEY_LOCAL_MACHINE;
#endif //WIN_32_MODE
 
    if(RegistryPath && RegistryPath[0]) {
        status = RegTGetKeyHandle(hRootKey, RegistryPath, &hKey);
#ifdef WIN_32_MODE
        if(status != ERROR_SUCCESS)
#else //WIN_32_MODE
        if(!NT_SUCCESS(status))
#endif //WIN_32_MODE
            return FALSE;
        free_h = TRUE;
    } else {
        hKey = hRootKey;
    }
    if(!hKey)
        return FALSE;
 
#ifndef WIN_32_MODE
/*
    UDFPrint(("h=%x|%S, %S (%x)\n",
        hRootKey, RegistryPath, Name, *pUlong));
*/
    len = sizeof(KEY_VALUE_PARTIAL_INFORMATION) + sizeof(ULONG) + 0x20;
    ValInfo = (PKEY_VALUE_PARTIAL_INFORMATION)
        MyAllocatePool__(NonPagedPool, len);
    if(!ValInfo) {
        if(free_h) {
            RegTCloseKeyHandle(hKey);
        }
        return FALSE;
    }
 
    RtlInitUnicodeString(&NameString, Name);
 
    status = ZwQueryValueKey(hKey,
                             &NameString,
                             KeyValuePartialInformation,
                             ValInfo,
                             len,
                             &len);
    if(NT_SUCCESS(status) &&
       ValInfo->DataLength == sizeof(ULONG)) {
        RtlCopyMemory(pUlong, ValInfo->Data, sizeof(ULONG));
        retval = TRUE;
        //UDFPrint(("  -> %x\n",*pUlong));
    } else {
        //UDFPrint(("  err %x\n",status));
    }
 
    MyFreePool__(ValInfo);
#else //WIN_32_MODE
    len = sizeof(ULONG);
    if (ERROR_SUCCESS == RegQueryValueExW(
        hKey,               // handle of key to query
        Name,            // address of name of value to query
        0,                  // reserved
        NULL,            // address of buffer for value type
        (BYTE *)pUlong,    // address of data buffer
        &len          // address of data buffer size
        ) && len == sizeof(ULONG)) {
        retval = TRUE;
    }
#endif //WIN_32_MODE
    if(free_h) {
        RegTCloseKeyHandle(hKey);
    }
    return retval;
} // end RegTGetDwordValue()

Referenced by UDFRegCheckParameterValue().

RegTGetKeyHandle()

NTSTATUS RegTGetKeyHandle	(	IN HKEY	hRootKey,
		IN PWCHAR	KeyName,
		OUT HKEY *	hKey
	)

Definition at line 59 of file regtools.cpp.

{
    LONG status;
 
    if(!hRootKey)
        hRootKey = HKEY_LOCAL_MACHINE;
 
    status = RegOpenKeyExW(
                hRootKey,
                KeyName,
                0,
                KEY_WRITE | KEY_READ,
                hKey
                );
 
    if(status != ERROR_SUCCESS) {
        *hKey = NULL;
    }
 
    return status;
} // end RegTGetKeyHandle()

Referenced by DriverEntry(), RegTGetDwordValue(), RegTGetStringValue(), and UDFRegCheckParameterValue().

RegTGetStringValue()

BOOLEAN RegTGetStringValue	(	IN HKEY	hRootKey,
		IN PCWSTR	RegistryPath,
		IN PCWSTR	Name,
		IN PWCHAR	pStr,
		IN ULONG	MaxLen
	)

Definition at line 189 of file regtools.cpp.

{
#ifndef WIN_32_MODE
    UNICODE_STRING NameString;
    PKEY_VALUE_PARTIAL_INFORMATION ValInfo;
#endif //USER_MODE
    ULONG len;
    NTSTATUS status;
    HKEY hKey;
    BOOLEAN retval = FALSE;
    BOOLEAN free_h = FALSE;
 
#ifdef WIN_32_MODE
    if(!hRootKey)
        hRootKey = HKEY_LOCAL_MACHINE;
#endif //WIN_32_MODE
 
    if(RegistryPath && RegistryPath[0]) {
        status = RegTGetKeyHandle(hRootKey, RegistryPath, &hKey);
#ifdef WIN_32_MODE
        if(status != ERROR_SUCCESS)
#else //WIN_32_MODE
        if(!NT_SUCCESS(status))
#endif //WIN_32_MODE
            return FALSE;
        free_h = TRUE;
    } else {
        hKey = hRootKey;
    }
    if(!hKey)
        return FALSE;
 
    pStr[0] = 0;
#ifndef WIN_32_MODE
    len = sizeof(KEY_VALUE_PARTIAL_INFORMATION) + MaxLen + 0x20;
    ValInfo = (PKEY_VALUE_PARTIAL_INFORMATION)
        MyAllocatePool__(NonPagedPool, len);
    if(!ValInfo) {
        if(free_h) {
            RegTCloseKeyHandle(hKey);
        }
        return FALSE;
    }
 
    RtlInitUnicodeString(&NameString, Name);
 
    status = ZwQueryValueKey(hKey,
                             &NameString,
                             KeyValuePartialInformation,
                             ValInfo,
                             len,
                             &len);
    if(NT_SUCCESS(status) &&
       ValInfo->DataLength) {
        RtlCopyMemory(pStr, ValInfo->Data, min(ValInfo->DataLength, MaxLen) );
        if(pStr[(ValInfo->DataLength)/sizeof(WCHAR)-1]) {
            pStr[(ValInfo->DataLength)/sizeof(WCHAR)-1] = 0;
        }
        retval = TRUE;
    }
 
    MyFreePool__(ValInfo);
#else //WIN_32_MODE
    len = MaxLen;
    if (ERROR_SUCCESS == RegQueryValueExW(
        hKey,               // handle of key to query
        Name,            // address of name of value to query
        0,                  // reserved
        NULL,            // address of buffer for value type
        (BYTE *)pStr,    // address of data buffer
        &len             // address of data buffer size
        ) && len) {
        if(pStr[len-1]) {
            pStr[len-1] = 0;
        }
        retval = TRUE;
    }
#endif //WIN_32_MODE
 
    if(free_h) {
        RegTCloseKeyHandle(hKey);
    }
    return retval;
} // end RegTGetStringValue()