Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-06-29 01:12:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

[KMTESTS:RTL]

Browse source 
- Add test for REG_MULTI_SZ in RtlQueryRegistryValues

svn path=/trunk/; revision=59544
This commit is contained in:
Thomas Faber 2013-07-20 19:33:13 +00:00
parent 38f41f733e
commit 2d6aa7e6da
4 changed files with 253 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
rostests/kmtests/CMakeLists.txt
View file

@ -13,6 +13,7 @@ list(APPEND COMMON_SOURCE
rtl/RtlAvlTree.c
rtl/RtlException.c
rtl/RtlMemory.c
rtl/RtlRegistry.c
rtl/RtlSplayTree.c
rtl/RtlUnicodeString.c)

2
rostests/kmtests/kmtest/testlist.c
View file

@ -13,6 +13,7 @@ KMT_TESTFUNC Test_IoDeviceObject;
KMT_TESTFUNC Test_RtlAvlTree;
KMT_TESTFUNC Test_RtlException;
KMT_TESTFUNC Test_RtlMemory;
KMT_TESTFUNC Test_RtlRegistry;
KMT_TESTFUNC Test_RtlSplayTree;
KMT_TESTFUNC Test_RtlUnicodeString;
@ -25,6 +26,7 @@ const KMT_TEST TestList[] =
{ "RtlAvlTree", Test_RtlAvlTree },
{ "RtlException", Test_RtlException },
{ "RtlMemory", Test_RtlMemory },
{ "RtlRegistry", Test_RtlRegistry },
{ "RtlSplayTree", Test_RtlSplayTree },
{ "RtlUnicodeString", Test_RtlUnicodeString },
{ NULL, NULL },

2
rostests/kmtests/kmtest_drv/testlist.c
View file

@ -48,6 +48,7 @@ KMT_TESTFUNC Test_SeQueryInfoToken;
KMT_TESTFUNC Test_RtlAvlTree;
KMT_TESTFUNC Test_RtlException;
KMT_TESTFUNC Test_RtlMemory;
KMT_TESTFUNC Test_RtlRegistry;
KMT_TESTFUNC Test_RtlSplayTree;
KMT_TESTFUNC Test_ZwAllocateVirtualMemory;
KMT_TESTFUNC Test_ZwCreateSection;
@ -97,6 +98,7 @@ const KMT_TEST TestList[] =
{ "RtlAvlTreeKM", Test_RtlAvlTree },
{ "RtlExceptionKM", Test_RtlException },
{ "RtlMemoryKM", Test_RtlMemory },
{ "RtlRegistryKM", Test_RtlRegistry },
{ "RtlSplayTreeKM", Test_RtlSplayTree },
{ "ZwAllocateVirtualMemory", Test_ZwAllocateVirtualMemory },
{ "ZwCreateSection", Test_ZwCreateSection },

248
rostests/kmtests/rtl/RtlRegistry.c Normal file
View file

@ -0,0 +1,248 @@
/*
* PROJECT: ReactOS kernel-mode tests
* LICENSE: GPLv2+ - See COPYING in the top level directory
* PURPOSE: Test for RtlQueryRegistryValues
* PROGRAMMER: Thomas Faber <thfabba@gmx.de>
*/
#define KMT_EMULATE_KERNEL
#include <kmt_test.h>
#ifdef KMT_KERNEL_MODE
#define KMT_KERNEL_HANDLE OBJ_KERNEL_HANDLE
#else
#define KMT_KERNEL_HANDLE 0
#endif
#ifndef RTL_NUMBER_OF
#define RTL_NUMBER_OF(x) (sizeof(x) / sizeof(x[0]))
#endif
typedef struct
{
PCWSTR ValueName;
ULONG ValueType;
PVOID ValueData;
ULONG ValueLength;
} EXPECTED_VALUE, *PEXPECTED_VALUE;
typedef struct
{
ULONG Count;
ULONG CurrentIndex;
EXPECTED_VALUE Values[20];
} EXPECTED_VALUES, *PEXPECTED_VALUES;
//static RTL_QUERY_REGISTRY_ROUTINE QueryRoutine;
static
NTSTATUS
NTAPI
QueryRoutine(
_In_ PWSTR ValueName,
_In_ ULONG ValueType,
_In_ PVOID ValueData,
_In_ ULONG ValueLength,
_In_ PVOID Context,
_In_ PVOID EntryContext)
{
PEXPECTED_VALUES ExpectedValues = Context;
PEXPECTED_VALUE Expected;
SIZE_T EqualBytes;
ok(ExpectedValues->CurrentIndex < ExpectedValues->Count,
"Call number %lu, expected only %lu\n",
ExpectedValues->CurrentIndex, ExpectedValues->Count);
if (!skip(ExpectedValues->CurrentIndex < ExpectedValues->Count, "Out of bounds\n"))
{
Expected = &ExpectedValues->Values[ExpectedValues->CurrentIndex];
if (EntryContext)
ok_eq_pointer(EntryContext, Expected);
ok_eq_wstr(ValueName, Expected->ValueName);
ok_eq_ulong(ValueType, Expected->ValueType);
ok_eq_ulong(ValueLength, Expected->ValueLength);
EqualBytes = RtlCompareMemory(ValueData,
Expected->ValueData,
min(ValueLength, Expected->ValueLength));
ok_eq_size(EqualBytes, Expected->ValueLength);
}
ExpectedValues->CurrentIndex++;
return STATUS_SUCCESS;
}
static
VOID
TestRtlQueryRegistryValues(
_In_ HANDLE KeyHandle)
{
NTSTATUS Status;
UNICODE_STRING ValueName = RTL_CONSTANT_STRING(L"TestValue");
RTL_QUERY_REGISTRY_TABLE QueryTable[] =
{
{ QueryRoutine, 0, L"TestValue", NULL, REG_NONE, NULL, 0 },
{ NULL }
};
EXPECTED_VALUES Expected;
typedef struct
{
PWSTR Str;
ULONG Len;
} STR_AND_LEN;
#define CONST_STR_AND_LEN(d) { (d), sizeof(d) }
#define CSAL CONST_STR_AND_LEN
#define NO_AUTO_LEN 1
#define NO_DEFAULT 2
#define AUTO_DIFFERS 4
#define DEFAULT_DIFFERS 8
struct
{
STR_AND_LEN Value;
ULONG ExpectedCount;
STR_AND_LEN Expected[20];
ULONG Flags;
ULONG DefaultExpectedCount;
STR_AND_LEN DefaultExpected[20];
} Tests[] =
{
{ { NULL, 0 }, 0, { { NULL, 0 } }, NO_AUTO_LEN | NO_DEFAULT },
{ CSAL(L""), 0, { { NULL, 0 } }, NO_AUTO_LEN },
{ CSAL(L"\0"), 1, { CSAL(L"") },
AUTO_DIFFERS | DEFAULT_DIFFERS, 0, { { NULL, 0 } } },
{ CSAL(L"String"), 1, { CSAL(L"String") }, NO_AUTO_LEN },
{ CSAL(L"String\0"), 1, { CSAL(L"String") } },
{ CSAL(L"String1\0String2"), 2, { CSAL(L"String1"), CSAL(L"String2") }, NO_AUTO_LEN },
{ CSAL(L"String1\0String2\0"), 2, { CSAL(L"String1"), CSAL(L"String2") } },
{ CSAL(L"String1\0\0String3"), 3, { CSAL(L"String1"), CSAL(L""), CSAL(L"String3") }, NO_AUTO_LEN },
{ CSAL(L"String1\0\0String3\0"), 3, { CSAL(L"String1"), CSAL(L""), CSAL(L"String3") },
AUTO_DIFFERS, 1, { CSAL(L"String1") } },
};
#define DO_QUERY(ExpectedArray, ExpectedCount) do \
{ \
ULONG _i; \
for (_i = 0; _i < (ExpectedCount); _i++) \
{ \
Expected.Values[_i].ValueName = ValueName.Buffer; \
Expected.Values[_i].ValueType = REG_SZ; \
Expected.Values[_i].ValueData = (ExpectedArray)[_i].Str; \
Expected.Values[_i].ValueLength = (ExpectedArray)[_i].Len; \
} \
Expected.CurrentIndex = 0; \
Expected.Count = (ExpectedCount); \
if ((ExpectedCount) == 1) \
QueryTable[0].EntryContext = &Expected.Values[0]; \
else \
QueryTable[0].EntryContext = NULL; \
Status = RtlQueryRegistryValues(RTL_REGISTRY_HANDLE, \
(PCWSTR)KeyHandle, \
QueryTable, \
&Expected, \
NULL); \
ok_eq_hex(Status, STATUS_SUCCESS); \
ok_eq_ulong(Expected.CurrentIndex, Expected.Count); \
} while(0)
ULONG TestCount = RTL_NUMBER_OF(Tests);
ULONG i;
for (i = 0; i < TestCount; i++)
{
trace("Set: %lu\n", i);
Status = ZwSetValueKey(KeyHandle,
&ValueName,
0,
REG_MULTI_SZ,
Tests[i].Value.Str,
Tests[i].Value.Len);
ok_eq_hex(Status, STATUS_SUCCESS);
DO_QUERY(Tests[i].Expected, Tests[i].ExpectedCount);
}
/* Delete value to test default values */
Status = ZwDeleteValueKey(KeyHandle, &ValueName);
ok(Status == STATUS_SUCCESS || Status == STATUS_OBJECT_NAME_NOT_FOUND,
"ZwDeleteValueKey returned %lx\n", Status);
/* Default: REG_NONE */
DO_QUERY((STR_AND_LEN *)NULL, 0);
for (i = 0; i < TestCount; i++)
{
if (Tests[i].Flags & NO_DEFAULT)
continue;
trace("Default: %lu\n", i);
QueryTable[0].DefaultType = REG_MULTI_SZ;
QueryTable[0].DefaultData = Tests[i].Value.Str;
QueryTable[0].DefaultLength = Tests[i].Value.Len;
if (Tests[i].Flags & DEFAULT_DIFFERS)
DO_QUERY(Tests[i].DefaultExpected, Tests[i].DefaultExpectedCount);
else
DO_QUERY(Tests[i].Expected, Tests[i].ExpectedCount);
}
for (i = 0; i < TestCount; i++)
{
if (Tests[i].Flags & NO_AUTO_LEN)
continue;
trace("Auto: %lu\n", i);
QueryTable[0].DefaultType = REG_MULTI_SZ;
QueryTable[0].DefaultData = Tests[i].Value.Str;
QueryTable[0].DefaultLength = 0;
if (Tests[i].Flags & AUTO_DIFFERS)
DO_QUERY(Tests[i].DefaultExpected, Tests[i].DefaultExpectedCount);
else
DO_QUERY(Tests[i].Expected, Tests[i].ExpectedCount);
}
}
START_TEST(RtlRegistry)
{
NTSTATUS Status;
UNICODE_STRING KeyName;
OBJECT_ATTRIBUTES ObjectAttributes;
HANDLE SoftwareHandle;
HANDLE KeyHandle;
RtlInitUnicodeString(&KeyName, L"\\Registry\\MACHINE\\Software");
InitializeObjectAttributes(&ObjectAttributes,
&KeyName,
OBJ_CASE_INSENSITIVE | KMT_KERNEL_HANDLE,
NULL,
NULL);
Status = ZwOpenKey(&SoftwareHandle,
KEY_CREATE_SUB_KEY,
&ObjectAttributes);
ok_eq_hex(Status, STATUS_SUCCESS);
if (skip(NT_SUCCESS(Status) && SoftwareHandle != NULL, "No software key\n"))
return;
RtlInitUnicodeString(&KeyName, L"RtlRegistryKmtestKey");
InitializeObjectAttributes(&ObjectAttributes,
&KeyName,
OBJ_CASE_INSENSITIVE | KMT_KERNEL_HANDLE,
SoftwareHandle,
NULL);
Status = ZwCreateKey(&KeyHandle,
KEY_QUERY_VALUE | KEY_SET_VALUE | DELETE,
&ObjectAttributes,
0,
NULL,
REG_OPTION_VOLATILE,
NULL);
ok_eq_hex(Status, STATUS_SUCCESS);
if (!skip(NT_SUCCESS(Status) && KeyHandle != NULL, "No test key\n"))
{
TestRtlQueryRegistryValues(KeyHandle);
Status = ZwDeleteKey(KeyHandle);
ok_eq_hex(Status, STATUS_SUCCESS);
Status = ZwClose(KeyHandle);
ok_eq_hex(Status, STATUS_SUCCESS);
}
}