mirror of
https://github.com/reactos/reactos.git
synced 2024-08-28 14:18:41 +00:00
112 lines
3.8 KiB
C
112 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 <apitest.h>
|
||
|
|
||
|
#include <winreg.h>
|
||
|
#include <ndk/cmfuncs.h>
|
||
|
#include <ndk/obfuncs.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
|
||
|
}
|