NtCreateKey.c File Reference

#include "precomp.h"

Include dependency graph for NtCreateKey.c:

Go to the source code of this file.

Macros
#define	VerifyAccess(h, e)   VerifyAccess_(h, e, __FILE__, __LINE__)
#define	TestCreateOpen(d, ea, es)   TestCreateOpen_(d, ea, es, __FILE__, __LINE__)

Functions
static VOID	VerifyAccess_ (_In_ HANDLE Handle, _In_ ACCESS_MASK ExpectedAccess, _In_ PCSTR File, _In_ INT Line)
static VOID	TestCreateOpen_ (_In_ ACCESS_MASK DesiredAccess, _In_ ACCESS_MASK ExpectedAccess, _In_ NTSTATUS ExpectedStatus, _In_ PCSTR File, _In_ INT Line)
	START_TEST (NtCreateKey)

Macro Definition Documentation

TestCreateOpen

#define TestCreateOpen	(		d,
			ea,
			es
	)		TestCreateOpen_(d, ea, es, __FILE__, __LINE__)

Definition at line 95 of file NtCreateKey.c.

VerifyAccess

#define VerifyAccess	(		h,
			e
	)		VerifyAccess_(h, e, __FILE__, __LINE__)

Definition at line 32 of file NtCreateKey.c.

Function Documentation

START_TEST()

START_TEST

(

NtCreateKey

)

Definition at line 97 of file NtCreateKey.c.

{
    TestCreateOpen(0, 0, STATUS_ACCESS_DENIED);
    TestCreateOpen(KEY_WOW64_32KEY, 0, STATUS_ACCESS_DENIED);
    TestCreateOpen(KEY_WOW64_64KEY, 0, STATUS_ACCESS_DENIED);
    TestCreateOpen(KEY_WOW64_32KEY | KEY_WOW64_64KEY, 0, STATUS_ACCESS_DENIED); // STATUS_INVALID_PARAMETER on Win7
    TestCreateOpen(READ_CONTROL, READ_CONTROL, STATUS_SUCCESS);
    TestCreateOpen(READ_CONTROL | KEY_WOW64_32KEY, READ_CONTROL, STATUS_SUCCESS);
    TestCreateOpen(READ_CONTROL | KEY_WOW64_64KEY, READ_CONTROL, STATUS_SUCCESS);
    TestCreateOpen(READ_CONTROL | KEY_WOW64_32KEY | KEY_WOW64_64KEY, READ_CONTROL, STATUS_SUCCESS); // STATUS_INVALID_PARAMETER on Win7
}

TestCreateOpen_()

static VOID TestCreateOpen_ ( _In_ ACCESS_MASK  DesiredAccess, _In_ ACCESS_MASK  ExpectedAccess, _In_ NTSTATUS  ExpectedStatus, _In_ PCSTR  File, _In_ INT  Line  )

static

Definition at line 36 of file NtCreateKey.c.

{
    NTSTATUS Status;
    HANDLE KeyHandle;
    UNICODE_STRING KeyName = RTL_CONSTANT_STRING(L"\\Registry\\Machine\\Software");
    OBJECT_ATTRIBUTES ObjectAttributes;
 
    InitializeObjectAttributes(&ObjectAttributes,
                               &KeyName,
                               OBJ_CASE_INSENSITIVE,
                               NULL,
                               NULL);
    Status = NtCreateKey(&KeyHandle,
                         DesiredAccess,
                         &ObjectAttributes,
                         0,
                         NULL,
                         REG_OPTION_NON_VOLATILE,
                         NULL);
    ok_(File, Line)(Status == ExpectedStatus,
                    "NtCreateKey returned 0x%lx, expected 0x%lx\n",
                    Status, ExpectedStatus);
    if (NT_SUCCESS(Status))
    {
        VerifyAccess_(KeyHandle, ExpectedAccess, File, Line);
        Status = NtClose(KeyHandle);
        ok_(File, Line)(Status == STATUS_SUCCESS,
                        "NtClose from NtCreateKey returned 0x%lx\n",
                        Status);
    }
    else if (NT_SUCCESS(ExpectedStatus))
    {
        skip_(File, Line)("NtCreateKey failed, skipping\n");
    }
 
    Status = NtOpenKey(&KeyHandle,
                       DesiredAccess,
                       &ObjectAttributes);
    ok_(File, Line)(Status == ExpectedStatus,
                    "NtOpenKey returned 0x%lx, expected 0x%lx\n",
                    Status, ExpectedStatus);
    if (NT_SUCCESS(Status))
    {
        VerifyAccess_(KeyHandle, ExpectedAccess, File, Line);
        Status = NtClose(KeyHandle);
        ok_(File, Line)(Status == STATUS_SUCCESS,
                        "NtClose from NtOpenKey returned 0x%lx\n",
                        Status);
    }
    else if (NT_SUCCESS(ExpectedStatus))
    {
        skip_(File, Line)("NtOpenKey failed, skipping\n");
    }
}

VerifyAccess_()

static VOID VerifyAccess_ ( _In_ HANDLE  Handle, _In_ ACCESS_MASK  ExpectedAccess, _In_ PCSTR  File, _In_ INT  Line  )

static

Definition at line 12 of file NtCreateKey.c.

{
    NTSTATUS Status;
    OBJECT_BASIC_INFORMATION BasicInfo;
    ULONG Length;
 
    Status = NtQueryObject(Handle,
                           ObjectBasicInformation,
                           &BasicInfo,
                           sizeof(BasicInfo),
                           &Length);
    ok_(File, Line)(Status == STATUS_SUCCESS, "NtQueryObject returned 0x%lx\n", Status);
    ok_(File, Line)(BasicInfo.GrantedAccess == ExpectedAccess,
                    "GrantedAccess is 0x%lx, expected 0x%lx\n",
                    BasicInfo.GrantedAccess, ExpectedAccess);
}

Referenced by TestCreateOpen_().