settings.c File Reference

#include <ntddk.h>
#include "sndblst.h"
#include "sbdebug.h"

Include dependency graph for settings.c:

Go to the source code of this file.

Functions
NTSTATUS NTAPI	EnumDeviceKeys (IN PUNICODE_STRING RegistryPath, IN PWSTR SubKey, IN PREGISTRY_CALLBACK_ROUTINE Callback, IN PVOID Context)
NTSTATUS NTAPI	LoadSettings (IN PWSTR ValueName, IN ULONG ValueType, IN PVOID ValueData, IN ULONG ValueLength, IN PVOID Context, IN PVOID EntryContext)

Function Documentation

EnumDeviceKeys()

NTSTATUS NTAPI EnumDeviceKeys	(	IN PUNICODE_STRING	RegistryPath,
		IN PWSTR	SubKey,
		IN PREGISTRY_CALLBACK_ROUTINE	Callback,
		IN PVOID	Context
	)

Definition at line 70 of file settings.c.

{
    NTSTATUS s;
    OBJECT_ATTRIBUTES oa;
    HANDLE hKey, hSubKey;
    UNICODE_STRING SubkeyName;
    ULONG i;
 
    // Attempt to open the key
 
    InitializeObjectAttributes(&oa, RegistryPath, OBJ_CASE_INSENSITIVE,
                                NULL, (PSECURITY_DESCRIPTOR)NULL);
 
    s = ZwOpenKey(&hKey, KEY_READ, &oa);
 
        TEST_STATUS(s); // debugging
 
    if (! NT_SUCCESS(s))
        return s;   // Problem
 
    RtlInitUnicodeString(&SubkeyName, SubKey);
 
    DPRINT("Subkey: %wZ\n", &SubkeyName);
 
    InitializeObjectAttributes(&oa, &SubkeyName, OBJ_CASE_INSENSITIVE,
                                hKey, (PSECURITY_DESCRIPTOR)NULL);
 
    s = ZwOpenKey(&hSubKey, KEY_ENUMERATE_SUB_KEYS, &oa);
 
    ZwClose(hKey);
 
        TEST_STATUS(s); // debugging
 
    if (! NT_SUCCESS(s))
        return s;
 
 
    // And now, the enumeration
 
    for (i = 0;; i ++)
    {
        KEY_BASIC_INFORMATION Info;
        PKEY_BASIC_INFORMATION pInfo;
        ULONG ResultLength = 0;
        ULONG Size = 0;
        PWSTR Pos;
        PWSTR Name;
 
        // Find the length of the subkey data
 
//        Info.NameLength = 0;    // TEMPORARY!
 
        s = ZwEnumerateKey(hSubKey, i, KeyBasicInformation, &Info,
                            sizeof(Info), &ResultLength);
 
        if (s == STATUS_NO_MORE_ENTRIES)
            break;
 
        DPRINT("Found an entry, allocating memory...\n");
 
//        Size = Info.NameLength + FIELD_OFFSET(KEY_BASIC_INFORMATION, Name[0]);
        Size = ResultLength + FIELD_OFFSET(KEY_BASIC_INFORMATION, Name[0]);
 
        DPRINT("Size is %d\n", Size);
 
        pInfo = (PKEY_BASIC_INFORMATION) ExAllocatePool(PagedPool, Size);
 
        if (pInfo == NULL)
        {
            DPRINT("INSUFFICIENT RESOURCES!\n");
            s = STATUS_INSUFFICIENT_RESOURCES;
            break;
        }
 
        DPRINT("Re-enumerating...\n");
 
        s = ZwEnumerateKey(hSubKey, i, KeyBasicInformation, pInfo, Size,
                            &ResultLength);
 
//        TEST_STATUS(s); // debugging
 
        if (! NT_SUCCESS(s))
        {
            ExFreePool((PVOID) pInfo);
            s = STATUS_INTERNAL_ERROR;
            break;
        }
 
        DPRINT("Allocating memory for name...\n");
 
        Name = ExAllocatePool(PagedPool,
                          RegistryPath->Length + sizeof(WCHAR) +
                          SubkeyName.Length + sizeof(WCHAR) +
                          pInfo->NameLength + sizeof(UNICODE_NULL));
 
        if (Name == NULL)
        {
            DPRINT("INSUFFICIENT RESOURCES!");
            ExFreePool((PVOID) pInfo);
            return STATUS_INSUFFICIENT_RESOURCES;
        }
 
        // Copy the key name
        RtlCopyMemory((PVOID)Name, (PVOID)RegistryPath->Buffer, RegistryPath->Length);
        Pos = Name + (RegistryPath->Length / sizeof(WCHAR));
        Pos[0] = '\\';
        Pos++;
 
        // Copy the parameters sub key name
        RtlCopyMemory((PVOID)Pos, (PVOID)SubKey, SubkeyName.Length);    //SubkeyName?
        Pos += SubkeyName.Length / sizeof(WCHAR);
        Pos[0] = '\\';
        Pos ++;
 
        // Copy the device sub key name
        RtlCopyMemory((PVOID)Pos, (PVOID)pInfo->Name, pInfo->NameLength);
        Pos += pInfo->NameLength / sizeof(WCHAR);
        Pos[0] = UNICODE_NULL;
 
        ExFreePool((PVOID)pInfo);
 
        DPRINT("Calling callback...\n");
 
        s = (*Callback)(Name, Context);
 
        if (! NT_SUCCESS(s))
        {   DPRINT("Callback FAILED\n");
            break;}
    }
 
    ZwClose(hSubKey);
 
    DPRINT("%d device registry keys found\n", i);
 
    if ((i == 0) && (s == STATUS_NO_MORE_ENTRIES))
        return STATUS_DEVICE_CONFIGURATION_ERROR;
 
    return s == STATUS_NO_MORE_ENTRIES ? STATUS_SUCCESS : s;
}

LoadSettings()

NTSTATUS NTAPI LoadSettings	(	IN PWSTR	ValueName,
		IN ULONG	ValueType,
		IN PVOID	ValueData,
		IN ULONG	ValueLength,
		IN PVOID	Context,
		IN PVOID	EntryContext
	)

Definition at line 230 of file settings.c.

{
    PDEVICE_EXTENSION DeviceInfo = Context;
 
    if (ValueType == REG_DWORD)
    {
        if (! _wcsicmp(ValueName, REGISTRY_PORT))
        {
            DeviceInfo->Port = *(PULONG) ValueData;
            DPRINT("Registry port = 0x%x\n", DeviceInfo->Port);
        }
 
        // More to come... (config.c)
    }
 
    else
    {
        // ?
    }
 
    return STATUS_SUCCESS;
}