mirror of
https://github.com/reactos/reactos.git
synced 2024-10-29 19:13:58 +00:00
108 lines
3.8 KiB
C
108 lines
3.8 KiB
C
/*
|
|
* PROJECT: ReactOS API tests
|
|
* LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory
|
|
* PURPOSE: Test for NtCreateKey
|
|
* PROGRAMMER: Thomas Faber <thomas.faber@reactos.org>
|
|
*/
|
|
|
|
#include "precomp.h"
|
|
|
|
static
|
|
VOID
|
|
VerifyAccess_(
|
|
_In_ HANDLE Handle,
|
|
_In_ ACCESS_MASK ExpectedAccess,
|
|
_In_ PCSTR File,
|
|
_In_ INT Line)
|
|
{
|
|
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);
|
|
}
|
|
#define VerifyAccess(h, e) VerifyAccess_(h, e, __FILE__, __LINE__)
|
|
|
|
static
|
|
VOID
|
|
TestCreateOpen_(
|
|
_In_ ACCESS_MASK DesiredAccess,
|
|
_In_ ACCESS_MASK ExpectedAccess,
|
|
_In_ NTSTATUS ExpectedStatus,
|
|
_In_ PCSTR File,
|
|
_In_ INT Line)
|
|
{
|
|
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");
|
|
}
|
|
}
|
|
#define TestCreateOpen(d, ea, es) TestCreateOpen_(d, ea, es, __FILE__, __LINE__)
|
|
|
|
START_TEST(NtCreateKey)
|
|
{
|
|
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
|
|
}
|