https://git.reactos.org/?p=reactos.git;a=commitdiff;h=eb7550f76730105582625e84210a376cf676c112
commit eb7550f76730105582625e84210a376cf676c112 Author: George Bișoc <[email protected]> AuthorDate: Wed Mar 15 19:52:29 2023 +0100 Commit: George Bișoc <[email protected]> CommitDate: Thu Mar 16 20:04:31 2023 +0100 [NTDLL_APITEST] Write some tests for NtQueryOpenSubKeys --- modules/rostests/apitests/ntdll/CMakeLists.txt | 1 + .../rostests/apitests/ntdll/NtQueryOpenSubKeys.c | 101 +++++++++++++++++++++ modules/rostests/apitests/ntdll/testlist.c | 2 + 3 files changed, 104 insertions(+) diff --git a/modules/rostests/apitests/ntdll/CMakeLists.txt b/modules/rostests/apitests/ntdll/CMakeLists.txt index 2e38c4ec7b5..ed7cf1ff86f 100644 --- a/modules/rostests/apitests/ntdll/CMakeLists.txt +++ b/modules/rostests/apitests/ntdll/CMakeLists.txt @@ -37,6 +37,7 @@ list(APPEND SOURCE NtQueryInformationThread.c NtQueryInformationToken.c NtQueryKey.c + NtQueryOpenSubKeys.c NtQuerySystemEnvironmentValue.c NtQuerySystemInformation.c NtQueryValueKey.c diff --git a/modules/rostests/apitests/ntdll/NtQueryOpenSubKeys.c b/modules/rostests/apitests/ntdll/NtQueryOpenSubKeys.c new file mode 100644 index 00000000000..3f3aa6dfbc6 --- /dev/null +++ b/modules/rostests/apitests/ntdll/NtQueryOpenSubKeys.c @@ -0,0 +1,101 @@ +/* + * PROJECT: ReactOS API tests + * LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later) + * PURPOSE: Tests for the NtQueryOpenSubKeys API + * COPYRIGHT: Copyright 2023 George Bișoc <[email protected]> + */ + +#include "precomp.h" + +START_TEST(NtQueryOpenSubKeys) +{ + NTSTATUS Status; + OBJECT_ATTRIBUTES ObjectAttributes; + ULONG Subkeys; + UNICODE_STRING RegistryKey = RTL_CONSTANT_STRING(L"\\Registry"); + UNICODE_STRING SystemKey = RTL_CONSTANT_STRING(L"\\Registry\\Machine\\SYSTEM"); + UNICODE_STRING SoftwareKey = RTL_CONSTANT_STRING(L"\\Registry\\Machine\\SOFTWARE"); + UNICODE_STRING DefaultUserKey = RTL_CONSTANT_STRING(L"\\Registry\\User\\.DEFAULT"); + + /* We give no object attributes and no return variable */ + Status = NtQueryOpenSubKeys(NULL, NULL); + ok_ntstatus(Status, STATUS_ACCESS_VIOLATION); + + /* Build a key path to the main registry tree */ + InitializeObjectAttributes(&ObjectAttributes, + &RegistryKey, + OBJ_CASE_INSENSITIVE, + NULL, + NULL); + + /* We give object attributes but no return variable */ + Status = NtQueryOpenSubKeys(&ObjectAttributes, NULL); + ok_ntstatus(Status, STATUS_ACCESS_VIOLATION); + + /* + * We give no object attributes but return variable. + * + * NOTE: Windows 10 and Server 2003 return different kinds of status + * codes. In Server 2003 it returns STATUS_ACCESS_VIOLATION because + * this function implements more probe checks against the object + * attributes parameter (namely the path name of the object) so it + * fails earlier. In Windows 10 instead the function only probes the + * memory address of the object attributes so that it resides in the boundary + * of the UM memory range so the function lets this NULL parameter + * slide through until ObOpenObjectByName hits this parameter as being + * NULL and returns STATUS_INVALID_PARAMETER. Currently ReactOS follows + * the behavior of Windows 10. + */ + Status = NtQueryOpenSubKeys(NULL, &Subkeys); + ok(Status == STATUS_ACCESS_VIOLATION || Status == STATUS_INVALID_PARAMETER, + "STATUS_ACCESS_VIOLATION or STATUS_INVALID_PARAMETER expected, got 0x%lx\n", Status); + + /* Garbage return variable, this function doesn't check for alignment */ + Status = NtQueryOpenSubKeys(&ObjectAttributes, (PVOID)1); + ok_ntstatus(Status, STATUS_ACCESS_VIOLATION); + + /* Return the open subkeys of this key now */ + Status = NtQueryOpenSubKeys(&ObjectAttributes, &Subkeys); + ok_ntstatus(Status, STATUS_SUCCESS); + + trace("\\Registry has %lu opened subkeys\n", Subkeys); + + /* Build a key path to the SYSTEM key */ + InitializeObjectAttributes(&ObjectAttributes, + &SystemKey, + OBJ_CASE_INSENSITIVE, + NULL, + NULL); + + /* Return the open subkeys of this key now */ + Status = NtQueryOpenSubKeys(&ObjectAttributes, &Subkeys); + ok_ntstatus(Status, STATUS_SUCCESS); + + trace("\\Registry\\Machine\\SYSTEM has %lu opened subkeys\n", Subkeys); + + /* Build a key path to the SYSTEM key */ + InitializeObjectAttributes(&ObjectAttributes, + &SoftwareKey, + OBJ_CASE_INSENSITIVE, + NULL, + NULL); + + /* Return the open subkeys of this key now */ + Status = NtQueryOpenSubKeys(&ObjectAttributes, &Subkeys); + ok_ntstatus(Status, STATUS_SUCCESS); + + trace("\\Registry\\Machine\\SOFTWARE has %lu opened subkeys\n", Subkeys); + + /* Build a key path to the default user key */ + InitializeObjectAttributes(&ObjectAttributes, + &DefaultUserKey, + OBJ_CASE_INSENSITIVE, + NULL, + NULL); + + /* Return the open subkeys of this key now */ + Status = NtQueryOpenSubKeys(&ObjectAttributes, &Subkeys); + ok_ntstatus(Status, STATUS_SUCCESS); + + trace("\\Registry\\User\\.DEFAULT has %lu opened subkeys\n", Subkeys); +} diff --git a/modules/rostests/apitests/ntdll/testlist.c b/modules/rostests/apitests/ntdll/testlist.c index 29b3ae4ba21..03857e1852a 100644 --- a/modules/rostests/apitests/ntdll/testlist.c +++ b/modules/rostests/apitests/ntdll/testlist.c @@ -33,6 +33,7 @@ extern void func_NtQueryInformationProcess(void); extern void func_NtQueryInformationThread(void); extern void func_NtQueryInformationToken(void); extern void func_NtQueryKey(void); +extern void func_NtQueryOpenSubKeys(void); extern void func_NtQuerySystemEnvironmentValue(void); extern void func_NtQuerySystemInformation(void); extern void func_NtQueryValueKey(void); @@ -125,6 +126,7 @@ const struct test winetest_testlist[] = { "NtQueryInformationThread", func_NtQueryInformationThread }, { "NtQueryInformationToken", func_NtQueryInformationToken }, { "NtQueryKey", func_NtQueryKey }, + { "NtQueryOpenSubKeys", func_NtQueryOpenSubKeys }, { "NtQuerySystemEnvironmentValue", func_NtQuerySystemEnvironmentValue }, { "NtQuerySystemInformation", func_NtQuerySystemInformation }, { "NtQueryValueKey", func_NtQueryValueKey },
