[tfaber] 69566: [KMTESTS:OB] - Add ObHandle test that verifies a couple ZwDuplicateObject behaviors
Author: tfaber Date: Sat Oct 17 15:50:25 2015 New Revision: 69566
URL: http://svn.reactos.org/svn/reactos?rev=69566&view=rev Log: [KMTESTS:OB] - Add ObHandle test that verifies a couple ZwDuplicateObject behaviors
Added: trunk/rostests/kmtests/ntos_ob/ObHandle.c (with props) Modified: trunk/rostests/kmtests/CMakeLists.txt trunk/rostests/kmtests/kmtest_drv/testlist.c
Modified: trunk/rostests/kmtests/CMakeLists.txt URL: http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/CMakeLists.txt?rev... ============================================================================== --- trunk/rostests/kmtests/CMakeLists.txt [iso-8859-1] (original) +++ trunk/rostests/kmtests/CMakeLists.txt [iso-8859-1] Sat Oct 17 15:50:25 2015 @@ -35,6 +35,7 @@ npfs/NpfsFileInfo.c npfs/NpfsHelpers.c npfs/NpfsReadWrite.c + npfs/NpfsSecurity.c npfs/NpfsVolumeInfo.c ntos_cm/CmSecurity.c ntos_ex/ExCallback.c @@ -53,6 +54,7 @@ ntos_io/IoCreateFile.c ntos_io/IoDeviceInterface.c ntos_io/IoEvent.c + ntos_io/IoFilesystem.c ntos_io/IoInterrupt.c ntos_io/IoIrp.c ntos_io/IoMdl.c @@ -60,6 +62,7 @@ ntos_ke/KeDevQueue.c ntos_ke/KeDpc.c ntos_ke/KeEvent.c + ntos_ke/KeFloatingPoint.c ntos_ke/KeGuardedMutex.c ntos_ke/KeIrql.c ntos_ke/KeMutex.c @@ -72,6 +75,7 @@ ntos_mm/ZwAllocateVirtualMemory.c ntos_mm/ZwCreateSection.c ntos_mm/ZwMapViewOfSection.c + ntos_ob/ObHandle.c ntos_ob/ObReference.c ntos_ob/ObType.c ntos_ob/ObTypes.c
Modified: trunk/rostests/kmtests/kmtest_drv/testlist.c URL: http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/kmtest_drv/testlis... ============================================================================== --- trunk/rostests/kmtests/kmtest_drv/testlist.c [iso-8859-1] (original) +++ trunk/rostests/kmtests/kmtest_drv/testlist.c [iso-8859-1] Sat Oct 17 15:50:25 2015 @@ -26,6 +26,7 @@ KMT_TESTFUNC Test_IoCreateFile; KMT_TESTFUNC Test_IoDeviceInterface; KMT_TESTFUNC Test_IoEvent; +KMT_TESTFUNC Test_IoFilesystem; KMT_TESTFUNC Test_IoInterrupt; KMT_TESTFUNC Test_IoIrp; KMT_TESTFUNC Test_IoMdl; @@ -33,6 +34,7 @@ KMT_TESTFUNC Test_KeDeviceQueue; KMT_TESTFUNC Test_KeDpc; KMT_TESTFUNC Test_KeEvent; +KMT_TESTFUNC Test_KeFloatingPoint; KMT_TESTFUNC Test_KeGuardedMutex; KMT_TESTFUNC Test_KeIrql; KMT_TESTFUNC Test_KeMutex; @@ -47,7 +49,9 @@ KMT_TESTFUNC Test_NpfsCreate; KMT_TESTFUNC Test_NpfsFileInfo; KMT_TESTFUNC Test_NpfsReadWrite; +KMT_TESTFUNC Test_NpfsSecurity; KMT_TESTFUNC Test_NpfsVolumeInfo; +KMT_TESTFUNC Test_ObHandle; KMT_TESTFUNC Test_ObReference; KMT_TESTFUNC Test_ObType; KMT_TESTFUNC Test_ObTypeClean; @@ -89,6 +93,7 @@ { "IoCreateFile", Test_IoCreateFile }, { "IoDeviceInterface", Test_IoDeviceInterface }, { "IoEvent", Test_IoEvent }, + { "IoFilesystem", Test_IoFilesystem }, { "IoInterrupt", Test_IoInterrupt }, { "IoIrp", Test_IoIrp }, { "IoMdl", Test_IoMdl }, @@ -96,6 +101,7 @@ { "KeDeviceQueue", Test_KeDeviceQueue }, { "KeDpc", Test_KeDpc }, { "KeEvent", Test_KeEvent }, + { "KeFloatingPoint", Test_KeFloatingPoint }, { "KeGuardedMutex", Test_KeGuardedMutex }, { "KeIrql", Test_KeIrql }, { "KeMutex", Test_KeMutex }, @@ -110,7 +116,9 @@ { "NpfsCreate", Test_NpfsCreate }, { "NpfsFileInfo", Test_NpfsFileInfo }, { "NpfsReadWrite", Test_NpfsReadWrite }, + { "NpfsSecurity", Test_NpfsSecurity }, { "NpfsVolumeInfo", Test_NpfsVolumeInfo }, + { "ObHandle", Test_ObHandle }, { "ObReference", Test_ObReference }, { "ObType", Test_ObType }, { "-ObTypeClean", Test_ObTypeClean },
Added: trunk/rostests/kmtests/ntos_ob/ObHandle.c URL: http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/ntos_ob/ObHandle.c... ============================================================================== --- trunk/rostests/kmtests/ntos_ob/ObHandle.c (added) +++ trunk/rostests/kmtests/ntos_ob/ObHandle.c [iso-8859-1] Sat Oct 17 15:50:25 2015 @@ -0,0 +1,136 @@ +/* + * PROJECT: ReactOS kernel-mode tests + * LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory + * PURPOSE: Kernel-Mode Test Suite Object Handle test + * PROGRAMMER: Thomas Faber thomas.faber@reactos.org + */ + +#include <kmt_test.h> + +#include <ndk/obfuncs.h> + +#define CheckObject(Handle, Pointers, Handles, Attrib, Access) do \ +{ \ + PUBLIC_OBJECT_BASIC_INFORMATION ObjectInfo; \ + Status = ZwQueryObject(Handle, ObjectBasicInformation, \ + &ObjectInfo, sizeof ObjectInfo, NULL); \ + ok_eq_hex(Status, STATUS_SUCCESS); \ + ok_eq_hex(ObjectInfo.Attributes, Attrib); \ + ok_eq_hex(ObjectInfo.GrantedAccess, Access); \ + ok_eq_ulong(ObjectInfo.PointerCount, Pointers); \ + ok_eq_ulong(ObjectInfo.HandleCount, Handles); \ +} while (0) + +#define KERNEL_HANDLE_FLAG ((ULONG_PTR)0xFFFFFFFF80000000) +#define IsUserHandle(h) (((ULONG_PTR)(h) & KERNEL_HANDLE_FLAG) == 0) +#define IsKernelHandle(h) (((ULONG_PTR)(h) & KERNEL_HANDLE_FLAG) == KERNEL_HANDLE_FLAG) + +static +VOID +TestDuplicate( + _In_ HANDLE Handle) +{ + NTSTATUS Status; + HANDLE NewHandle; + struct + { + ACCESS_MASK DesiredAccess; + ULONG RequestedAttributes; + ULONG Options; + ACCESS_MASK GrantedAccess; + ULONG ExpectedAttributes; + } Tests[] = + { + { DIRECTORY_ALL_ACCESS, 0, 0, + DIRECTORY_ALL_ACCESS, 0 }, + { DIRECTORY_ALL_ACCESS, OBJ_KERNEL_HANDLE, 0, + DIRECTORY_ALL_ACCESS, 0 }, + { DIRECTORY_QUERY, 0, 0, + DIRECTORY_QUERY, 0 }, + { DIRECTORY_QUERY, OBJ_INHERIT, 0, + DIRECTORY_QUERY, OBJ_INHERIT }, + { DIRECTORY_QUERY, OBJ_INHERIT, DUPLICATE_SAME_ACCESS, + DIRECTORY_ALL_ACCESS, OBJ_INHERIT }, + /* 5 */ + { DIRECTORY_QUERY, OBJ_INHERIT, DUPLICATE_SAME_ATTRIBUTES, + DIRECTORY_QUERY, 0 }, + { DIRECTORY_QUERY, OBJ_INHERIT, DUPLICATE_SAME_ACCESS | DUPLICATE_SAME_ATTRIBUTES, + DIRECTORY_ALL_ACCESS, 0 }, + }; + ULONG i; + + for (i = 0; i < RTL_NUMBER_OF(Tests); i++) + { + trace("Test %lu\n", i); + Status = ZwDuplicateObject(ZwCurrentProcess(), + Handle, + ZwCurrentProcess(), + &NewHandle, + Tests[i].DesiredAccess, + Tests[i].RequestedAttributes, + Tests[i].Options); + ok_eq_hex(Status, STATUS_SUCCESS); + if (!skip(NT_SUCCESS(Status), "DuplicateHandle failed\n")) + { + ok(IsUserHandle(NewHandle), "New handle = %p\n", NewHandle); + CheckObject(NewHandle, 3UL, 2UL, Tests[i].ExpectedAttributes, Tests[i].GrantedAccess); + CheckObject(Handle, 3UL, 2UL, 0UL, DIRECTORY_ALL_ACCESS); + + Status = ObCloseHandle(NewHandle, UserMode); + ok_eq_hex(Status, STATUS_SUCCESS); + CheckObject(Handle, 2UL, 1UL, 0UL, DIRECTORY_ALL_ACCESS); + } + } +} + +START_TEST(ObHandle) +{ + NTSTATUS Status; + OBJECT_ATTRIBUTES ObjectAttributes; + HANDLE KernelDirectoryHandle; + HANDLE UserDirectoryHandle; + + InitializeObjectAttributes(&ObjectAttributes, + NULL, + 0, + NULL, + NULL); + Status = ZwCreateDirectoryObject(&UserDirectoryHandle, + DIRECTORY_ALL_ACCESS, + &ObjectAttributes); + ok_eq_hex(Status, STATUS_SUCCESS); + if (!skip(NT_SUCCESS(Status), "No directory handle\n")) + { + ok(IsUserHandle(UserDirectoryHandle), "User handle = %p\n", UserDirectoryHandle); + CheckObject(UserDirectoryHandle, 2UL, 1UL, 0UL, DIRECTORY_ALL_ACCESS); + + TestDuplicate(UserDirectoryHandle); + + Status = ObCloseHandle(UserDirectoryHandle, UserMode); + ok_eq_hex(Status, STATUS_SUCCESS); + } + + InitializeObjectAttributes(&ObjectAttributes, + NULL, + OBJ_KERNEL_HANDLE, + NULL, + NULL); + Status = ZwCreateDirectoryObject(&KernelDirectoryHandle, + DIRECTORY_ALL_ACCESS, + &ObjectAttributes); + ok_eq_hex(Status, STATUS_SUCCESS); + if (!skip(NT_SUCCESS(Status), "No directory handle\n")) + { + ok(IsKernelHandle(KernelDirectoryHandle), "Kernel handle = %p\n", KernelDirectoryHandle); + CheckObject(KernelDirectoryHandle, 2UL, 1UL, 0UL, DIRECTORY_ALL_ACCESS); + + TestDuplicate(KernelDirectoryHandle); + + Status = ObCloseHandle(KernelDirectoryHandle, UserMode); + ok_eq_hex(Status, STATUS_INVALID_HANDLE); + CheckObject(KernelDirectoryHandle, 2UL, 1UL, 0UL, DIRECTORY_ALL_ACCESS); + + Status = ObCloseHandle(KernelDirectoryHandle, KernelMode); + ok_eq_hex(Status, STATUS_SUCCESS); + } +}
Propchange: trunk/rostests/kmtests/ntos_ob/ObHandle.c ------------------------------------------------------------------------------ svn:eol-style = native
-
tfaber＠svn.reactos.org