NTSTATUS status = STATUS_SUCCESS;
HANDLE hRegKey = NULL;
OBJECT_ATTRIBUTES RegAttribute = { 0 };
ULONG KeyOp = 0;

InitializeObjectAttributes(&RegAttribute, RegistryPath, OBJ_CASE_INSENSITIVE, NULL, NULL);
status = ZwCreateKey(&hRegKey, KEY_ALL_ACCESS, &RegAttribute, 0, NULL, REG_OPTION_NON_VOLATILE, &KeyOp);
if (NT_SUCCESS(status)) {
    if (KeyOp == REG_CREATED_NEW_KEY) {
        DbgPrint("Create New key\n");
    }
    else if (KeyOp == REG_OPENED_EXISTING_KEY) {
        DbgPrint("Registry key opened\n");
    }
    else
    {
        DbgPrint("Error\n");
    }
}

NTSTATUS KernelReg(PUNICODE_STRING RegistryPath) {
    NTSTATUS status = STATUS_SUCCESS;
    HANDLE hRegKey = NULL;
    OBJECT_ATTRIBUTES RegAttribute = { 0 };
    ULONG KeyOp = 0;

    PVOID KeyInfo = ExAllocatePool2(POOL_FLAG_NON_PAGED, 0x1000, 'kcaH');
    if (KeyInfo == NULL) {
        DbgPrint("Allocate Mem Failed\n");
        return status;
    }
    RtlZeroMemory(KeyInfo, 0x1000);
    UNICODE_STRING regKeyName = { 0 };
    RtlInitUnicodeString(&regKeyName, L"ImagePath");

    InitializeObjectAttributes(&RegAttribute, RegistryPath, OBJ_CASE_INSENSITIVE, NULL, NULL);
    //status = ZwOpenKey(&hRegKey, KEY_ALL_ACCESS, &RegAttribute, 0, NULL, REG_OPTION_NON_VOLATILE, &KeyOp);
    status = ZwOpenKey(&hRegKey, KEY_ALL_ACCESS, &RegAttribute);
    if (!NT_SUCCESS(status)) {
        DbgPrint("Open Reg Key Failed\n");
        return status;
    }
    
    status = ZwQueryValueKey(hRegKey, &regKeyName, KeyValuePartialInformation, KeyInfo, 0x1000 - 1, &KeyOp);
    if (!NT_SUCCESS(status)) {
        DbgPrint("Read Reg Key Failed\n");
        ZwClose(hRegKey);
        ExFreePool(KeyInfo);
        return status;
    }
    PKEY_VALUE_PARTIAL_INFORMATION tempinfo = (PKEY_VALUE_PARTIAL_INFORMATION)KeyInfo;
    PWCHAR tempName = (PWCHAR)(tempinfo->Data);
    DbgPrint("reg: %wZ, key: %wZ, Value: %ws\n", RegistryPath,regKeyName, tempName);

    ZwClose(hRegKey);
    ExFreePool(KeyInfo);
    return status;
}