Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-05-28 13:38:19 +00:00
Code Issues Projects Releases Packages Wiki Activity

[NTDLL_APITEST] Add a test for KeyValuePartialInformationAlign64. CORE-17358

Browse source
This commit is contained in:
Thomas Faber 2020-10-31 13:08:06 +01:00
parent 33714797f4
commit b2cf64094c
No known key found for this signature in database
GPG key ID: 076E7C3D44720826
3 changed files with 118 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
modules/rostests/apitests/ntdll/CMakeLists.txt
View file

@ -30,6 +30,7 @@ list(APPEND SOURCE
NtQueryKey.c
NtQuerySystemEnvironmentValue.c
NtQuerySystemInformation.c
NtQueryValueKey.c
NtQueryVolumeInformationFile.c
NtReadFile.c
NtSaveKey.c

115
modules/rostests/apitests/ntdll/NtQueryValueKey.c Normal file
View file

@ -0,0 +1,115 @@
/*
* PROJECT: ReactOS API Tests
* LICENSE: LGPL-2.1-or-later (https://spdx.org/licenses/LGPL-2.1-or-later)
* PURPOSE: Test for NtQueryValueKey
* COPYRIGHT: Copyright 2020 Thomas Faber (thomas.faber@reactos.org)
*/
#include "precomp.h"
#include <winreg.h>
START_TEST(NtQueryValueKey)
{
NTSTATUS Status;
HANDLE KeyHandle;
UNICODE_STRING KeyName = RTL_CONSTANT_STRING(L"\\Registry\\Machine\\SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion");
OBJECT_ATTRIBUTES ObjectAttributes;
UNICODE_STRING ValueName = RTL_CONSTANT_STRING(L"SystemRoot");
PKEY_VALUE_PARTIAL_INFORMATION_ALIGN64 Info;
PKEY_VALUE_PARTIAL_INFORMATION_ALIGN64 InfoUnaligned;
ULONG InfoLength;
ULONG ResultLength;
const WCHAR *StringData;
ULONG StringLength;
InitializeObjectAttributes(&ObjectAttributes,
&KeyName,
OBJ_CASE_INSENSITIVE,
NULL,
NULL);
Status = NtOpenKey(&KeyHandle,
KEY_QUERY_VALUE,
&ObjectAttributes);
if (!NT_SUCCESS(Status))
{
skip("Test key unavailable\n");
return;
}
/* Specify zero-size buffer, get the length */
ResultLength = 0x55555555;
Status = NtQueryValueKey(KeyHandle,
&ValueName,
KeyValuePartialInformationAlign64,
NULL,
0,
&ResultLength);
ok_hex(Status, STATUS_BUFFER_TOO_SMALL);
ok(ResultLength > FIELD_OFFSET(KEY_VALUE_PARTIAL_INFORMATION_ALIGN64, Data) && ResultLength < 0x10000,
"Invalid result length %lu\n", ResultLength);
if (ResultLength < FIELD_OFFSET(KEY_VALUE_PARTIAL_INFORMATION_ALIGN64, Data))
{
skip("Result length %lu too small\n", ResultLength);
goto Exit;
}
InfoLength = ResultLength;
Info = RtlAllocateHeap(RtlGetProcessHeap(),
0,
InfoLength + 4);
if (Info == NULL)
{
skip("Could not alloc %lu bytes\n", InfoLength);
goto Exit;
}
/* Successful call */
Status = NtQueryValueKey(KeyHandle,
&ValueName,
KeyValuePartialInformationAlign64,
Info,
InfoLength,
&ResultLength);
ok_hex(Status, STATUS_SUCCESS);
ok_int(ResultLength, InfoLength);
ok_int(Info->Type, REG_SZ);
StringLength = InfoLength - FIELD_OFFSET(KEY_VALUE_PARTIAL_INFORMATION_ALIGN64, Data);
ok_int(Info->DataLength, StringLength);
StringData = (PWCHAR)Info->Data;
ok(Info->DataLength >= 5 * sizeof(WCHAR), "DataLength %lu is too small for path\n", Info->DataLength);
if (Info->DataLength >= 5 * sizeof(WCHAR))
{
trace("SystemRoot: %.*ls\n", (int)(Info->DataLength / sizeof(WCHAR) - 1), StringData);
ok(StringData[0] >= 'A' && StringData[0] <= 'Z', "Data[0] = %x\n", StringData[0]);
ok(StringData[1] == ':', "Data[1] = %x\n", StringData[1]);
ok(StringData[2] == '\\', "Data[2] = %x\n", StringData[2]);
ok(iswalnum(StringData[3]), "Data[3] = %x\n", StringData[3]);
ok(StringData[Info->DataLength / sizeof(WCHAR) - 1] == UNICODE_NULL,
"Data[%lu] = %x\n", Info->DataLength / sizeof(WCHAR) - 1, StringData[Info->DataLength / sizeof(WCHAR) - 1]);
}
/* If the buffer isn't 64 bit aligned, the data won't be, either */
InfoUnaligned = (PVOID)((PUCHAR)Info + 4);
Status = NtQueryValueKey(KeyHandle,
&ValueName,
KeyValuePartialInformationAlign64,
InfoUnaligned,
InfoLength,
&ResultLength);
ok_hex(Status, STATUS_SUCCESS);
ok_int(ResultLength, InfoLength);
ok_int(InfoUnaligned->Type, REG_SZ);
StringData = (PWCHAR)InfoUnaligned->Data;
ok(InfoUnaligned->DataLength >= 2 * sizeof(WCHAR), "DataLength %lu is too small for path\n", InfoUnaligned->DataLength);
if (InfoUnaligned->DataLength >= 2 * sizeof(WCHAR))
{
ok(StringData[1] == ':', "Data[1] = %x\n", StringData[1]);
}
RtlFreeHeap(RtlGetProcessHeap(), 0, Info);
Exit:
Status = NtClose(KeyHandle);
ok_hex(Status, STATUS_SUCCESS);
}

2
modules/rostests/apitests/ntdll/testlist.c
View file

@ -28,6 +28,7 @@ extern void func_NtQueryInformationThread(void);
extern void func_NtQueryKey(void);
extern void func_NtQuerySystemEnvironmentValue(void);
extern void func_NtQuerySystemInformation(void);
extern void func_NtQueryValueKey(void);
extern void func_NtQueryVolumeInformationFile(void);
extern void func_NtReadFile(void);
extern void func_NtSaveKey(void);
@ -102,6 +103,7 @@ const struct test winetest_testlist[] =
{ "NtQueryKey", func_NtQueryKey },
{ "NtQuerySystemEnvironmentValue", func_NtQuerySystemEnvironmentValue },
{ "NtQuerySystemInformation", func_NtQuerySystemInformation },
{ "NtQueryValueKey", func_NtQueryValueKey },
{ "NtQueryVolumeInformationFile", func_NtQueryVolumeInformationFile },
{ "NtReadFile", func_NtReadFile },
{ "NtSaveKey", func_NtSaveKey},