registry.c File Reference

#include <win32k.h>
#include <debug.h>

Include dependency graph for registry.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Functions
NTSTATUS NTAPI	RegOpenKey (LPCWSTR pwszKeyName, PHKEY phkey)
NTSTATUS NTAPI	RegQueryValue (IN HKEY hkey, IN PCWSTR pwszValueName, IN ULONG ulType, OUT PVOID pvData, IN OUT PULONG pcbValue)
VOID NTAPI	RegWriteSZ (HKEY hkey, PCWSTR pwszValue, PWSTR pwszData)
VOID NTAPI	RegWriteDWORD (HKEY hkey, PCWSTR pwszValue, DWORD dwData)
BOOL NTAPI	RegReadDWORD (HKEY hkey, PCWSTR pwszValue, PDWORD pdwData)
NTSTATUS NTAPI	RegOpenSectionKey (LPCWSTR pszSection, PHKEY phkey)
DWORD NTAPI	RegGetSectionDWORD (LPCWSTR pszSection, PCWSTR pszValue, DWORD dwDefault)
	_Success_ (return!=FALSE)
	_Success_ (return !=FALSE)

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 11 of file registry.c.

Function Documentation

_Success_() [1/2]

_Success_

(

return !

= FALSE

)

Definition at line 295 of file registry.c.

{
    NTSTATUS Status;
    OBJECT_ATTRIBUTES ObjectAttributes;
    UNICODE_STRING usCurrentUserKey, usKeyName, usValueName;
    WCHAR awcBuffer[MAX_PATH];
    HKEY hkey;
 
    // FIXME: Logged in user versus current process user?
    /* Get the path of the current user's profile */
    Status = RtlFormatCurrentUserKeyPath(&usCurrentUserKey);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("RtlFormatCurrentUserKeyPath failed\n");
        return FALSE;
    }
 
    /* Initialize empty key name */
    RtlInitEmptyUnicodeString(&usKeyName, awcBuffer, sizeof(awcBuffer));
 
    /* Append the current user key name */
    Status = RtlAppendUnicodeStringToString(&usKeyName, &usCurrentUserKey);
    if (!NT_SUCCESS(Status))
    {
        return FALSE;
    }
 
    /* Free the current user key name */
    RtlFreeUnicodeString(&usCurrentUserKey);
 
    /* Append a '\', we can trust in enough space left. */
    usKeyName.Buffer[usKeyName.Length / sizeof(WCHAR)] = '\\';
    usKeyName.Length += sizeof(WCHAR);
 
    /* Append the subkey name */
    Status = RtlAppendUnicodeToString(&usKeyName, pwszKeyName);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("RtlAppendUnicodeToString failed with Status=0x%lx, buf:%u,%u\n",
                Status, usKeyName.Length, usKeyName.MaximumLength);
        return FALSE;
    }
 
    /* Initialize object attributes */
    InitializeObjectAttributes(&ObjectAttributes,
                               &usKeyName,
                               OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE,
                               NULL,
                               NULL);
 
    /* Open or create the key */
    Status = ZwCreateKey((PHANDLE)&hkey,
                         KEY_READ | KEY_WRITE,
                         &ObjectAttributes,
                         0,
                         NULL,
                         0,
                         NULL);
    if(!NT_SUCCESS(Status))
    {
        DPRINT1("Failed to create key: 0x%x\n", Status);
        return FALSE;
    }
 
    /* Initialize the value name string */
    RtlInitUnicodeString(&usValueName, pwszValueName);
 
    Status = ZwSetValueKey(hkey, &usValueName, 0, ulType, (PVOID)pvData, cbDataSize);
    if(!NT_SUCCESS(Status))
    {
        DPRINT1("Failed to write reg key '%S' value '%S', Status = %lx\n",
                pwszKeyName, pwszValueName, Status);
    }
 
    /* Cleanup */
    ZwClose(hkey);
 
    return NT_SUCCESS(Status);
}

_Success_() [2/2]

_Success_

(

return!

= FALSE

)

Definition at line 187 of file registry.c.

{
    NTSTATUS Status;
    OBJECT_ATTRIBUTES ObjectAttributes;
    UNICODE_STRING usCurrentUserKey, usKeyName, usValueName;
    WCHAR awcBuffer[MAX_PATH];
    HKEY hkey;
    PKEY_VALUE_PARTIAL_INFORMATION pInfo;
    ULONG cbInfoSize, cbReqSize;
 
    /* Get the path of the current user's profile */
    Status = RtlFormatCurrentUserKeyPath(&usCurrentUserKey);
    if (!NT_SUCCESS(Status))
    {
        return FALSE;
    }
 
    /* Initialize empty key name */
    RtlInitEmptyUnicodeString(&usKeyName, awcBuffer, sizeof(awcBuffer));
 
    /* Append the current user key name */
    Status = RtlAppendUnicodeStringToString(&usKeyName, &usCurrentUserKey);
 
    /* Free the current user key name */
    RtlFreeUnicodeString(&usCurrentUserKey);
 
    /* Check for success */
    if (!NT_SUCCESS(Status))
    {
        return FALSE;
    }
 
    /* Append a '\', we can trust in enough space left. */
    usKeyName.Buffer[usKeyName.Length / sizeof(WCHAR)] = '\\';
    usKeyName.Length += sizeof(WCHAR);
 
    /* Append the subkey name */
    Status = RtlAppendUnicodeToString(&usKeyName, pwszKeyName);
    if (!NT_SUCCESS(Status))
    {
        return FALSE;
    }
 
    /* Initialize object attributes */
    InitializeObjectAttributes(&ObjectAttributes,
                               &usKeyName,
                               OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE,
                               NULL,
                               NULL);
 
    /* Open the key */
    Status = ZwOpenKey((PHANDLE)&hkey, KEY_READ, &ObjectAttributes);
    if (!NT_SUCCESS(Status))
    {
        return FALSE;
    }
 
    /* Check if the local buffer is sufficient */
    cbInfoSize = sizeof(KEY_VALUE_PARTIAL_INFORMATION) + cbDataSize;
    if (cbInfoSize <= sizeof(awcBuffer))
    {
        pInfo = (PVOID)awcBuffer;
    }
    else
    {
        /* It's not, allocate a sufficient buffer */
        pInfo = ExAllocatePoolWithTag(PagedPool, cbInfoSize, TAG_TEMP);
        if (!pInfo)
        {
            ZwClose(hkey);
            return FALSE;
        }
    }
 
    /* Query the value */
    RtlInitUnicodeString(&usValueName, pwszValueName);
    Status = ZwQueryValueKey(hkey,
                             &usValueName,
                             KeyValuePartialInformation,
                             (PVOID)pInfo,
                             cbInfoSize,
                             &cbReqSize);
    if (NT_SUCCESS(Status))
    {
        /* Did we get the right type */
        if (pInfo->Type == ulType)
        {
            /* Copy the contents to the caller */
            RtlCopyMemory(pvData, pInfo->Data, cbDataSize);
        }
    }
 
    /* Cleanup */
    ZwClose(hkey);
    if (pInfo != (PVOID)awcBuffer)
        ExFreePoolWithTag(pInfo, TAG_TEMP);
 
    return NT_SUCCESS(Status);
}

RegGetSectionDWORD()

DWORD NTAPI RegGetSectionDWORD	(	LPCWSTR	pszSection,
		PCWSTR	pszValue,
		DWORD	dwDefault
	)

Definition at line 172 of file registry.c.

{
    HKEY hKey;
    DWORD dwValue;
 
    if (NT_ERROR(RegOpenSectionKey(pszSection, &hKey)))
        return dwDefault;
 
    if (!RegReadDWORD(hKey, pszValue, &dwValue))
        dwValue = dwDefault;
 
    ZwClose(hKey);
    return dwValue;
}

Referenced by UserIsCiceroEnabled(), and UserIsIMMEnabled().

RegOpenKey()

NTSTATUS NTAPI RegOpenKey	(	LPCWSTR	pwszKeyName,
		PHKEY	phkey
	)

Definition at line 16 of file registry.c.

{
    NTSTATUS Status;
    OBJECT_ATTRIBUTES ObjectAttributes;
    UNICODE_STRING ustrKeyName;
    HKEY hkey;
 
    /* Initialize the key name */
    RtlInitUnicodeString(&ustrKeyName, pwszKeyName);
 
    /* Initialize object attributes */
    InitializeObjectAttributes(&ObjectAttributes,
                               &ustrKeyName,
                               OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE,
                               NULL,
                               NULL);
 
    /* Open the key */
    Status = ZwOpenKey((PHANDLE)&hkey, KEY_READ, &ObjectAttributes);
    if (NT_SUCCESS(Status))
    {
        *phkey = hkey;
    }
 
    return Status;
}

RegOpenSectionKey()

NTSTATUS NTAPI RegOpenSectionKey	(	LPCWSTR	pszSection,
		PHKEY	phkey
	)

Definition at line 159 of file registry.c.

{
    WCHAR szKey[MAX_PATH] =
        L"\\Registry\\Machine\\Software\\Microsoft\\Windows NT\\CurrentVersion\\";
 
    RtlStringCchCatW(szKey, _countof(szKey), pszSection);
    return RegOpenKey(szKey, phkey);
}

Referenced by RegGetSectionDWORD().

RegQueryValue()

NTSTATUS NTAPI RegQueryValue	(	IN HKEY	hkey,
		IN PCWSTR	pwszValueName,
		IN ULONG	ulType,
		OUT PVOID	pvData,
		IN OUT PULONG	pcbValue
	)

Definition at line 47 of file registry.c.

{
    NTSTATUS Status;
    UNICODE_STRING ustrValueName;
    BYTE ajBuffer[100];
    PKEY_VALUE_PARTIAL_INFORMATION pInfo;
    ULONG cbInfoSize, cbDataSize;
 
    /* Check if the local buffer is sufficient */
    cbInfoSize = FIELD_OFFSET(KEY_VALUE_PARTIAL_INFORMATION, Data[*pcbValue]);
    if (cbInfoSize <= sizeof(ajBuffer))
    {
        pInfo = (PVOID)ajBuffer;
    }
    else
    {
        /* It's not, allocate a sufficient buffer */
        pInfo = ExAllocatePoolWithTag(PagedPool, cbInfoSize, TAG_TEMP);
        if (!pInfo)
        {
            return STATUS_INSUFFICIENT_RESOURCES;
        }
    }
 
    /* Query the value */
    RtlInitUnicodeString(&ustrValueName, pwszValueName);
    Status = ZwQueryValueKey(hkey,
                             &ustrValueName,
                             KeyValuePartialInformation,
                             (PVOID)pInfo,
                             cbInfoSize,
                             &cbInfoSize);
 
    /* Note: STATUS_BUFFER_OVERFLOW is not a success */
    if (NT_SUCCESS(Status))
    {
        cbDataSize = pInfo->DataLength;
 
        /* Did we get the right type */
        if (pInfo->Type != ulType)
        {
            Status = STATUS_OBJECT_TYPE_MISMATCH;
        }
        else if (cbDataSize > *pcbValue)
        {
            Status = STATUS_BUFFER_TOO_SMALL;
        }
        else
        {
            /* Copy the contents to the caller */
            RtlCopyMemory(pvData, pInfo->Data, cbDataSize);
        }
    }
    else if ((Status == STATUS_BUFFER_OVERFLOW) || (Status == STATUS_BUFFER_TOO_SMALL))
    {
        _PRAGMA_WARNING_SUPPRESS(6102); /* cbInfoSize is initialized here! */
        cbDataSize = cbInfoSize - FIELD_OFFSET(KEY_VALUE_PARTIAL_INFORMATION, Data);
    }
    else
    {
        cbDataSize = 0;
    }
 
    /* Return the data size to the caller */
    *pcbValue = cbDataSize;
 
    /* Cleanup */
    if (pInfo != (PVOID)ajBuffer)
        ExFreePoolWithTag(pInfo, TAG_TEMP);
 
    return Status;
}

RegReadDWORD()

BOOL NTAPI RegReadDWORD	(	HKEY	hkey,
		PCWSTR	pwszValue,
		PDWORD	pdwData
	)

Definition at line 149 of file registry.c.

{
    NTSTATUS Status;
    ULONG cbSize = sizeof(DWORD);
    Status = RegQueryValue(hkey, pwszValue, REG_DWORD, pdwData, &cbSize);
    return NT_SUCCESS(Status);
}

Referenced by EngpUpdateGraphicsDeviceList(), InitMetrics(), InitUserImpl(), and RegGetSectionDWORD().

RegWriteDWORD()

VOID NTAPI RegWriteDWORD	(	HKEY	hkey,
		PCWSTR	pwszValue,
		DWORD	dwData
	)

Definition at line 139 of file registry.c.

{
    UNICODE_STRING ustrValue;
 
    RtlInitUnicodeString(&ustrValue, pwszValue);
    ZwSetValueKey(hkey, &ustrValue, 0, REG_DWORD, &dwData, sizeof(DWORD));
}

Referenced by RegWriteDisplaySettings().

RegWriteSZ()

VOID NTAPI RegWriteSZ	(	HKEY	hkey,
		PCWSTR	pwszValue,
		PWSTR	pwszData
	)

Definition at line 127 of file registry.c.

{
    UNICODE_STRING ustrValue;
    UNICODE_STRING ustrData;
 
    RtlInitUnicodeString(&ustrValue, pwszValue);
    RtlInitUnicodeString(&ustrData, pwszData);
    ZwSetValueKey(hkey, &ustrValue, 0, REG_SZ, &ustrData, ustrData.Length + sizeof(WCHAR));
}