Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248/* * PROJECT: ReactOS kernel-mode tests * LICENSE: GPLv2+ - See COPYING in the top level directory * PURPOSE: Kernel-Mode Test Suite Object Referencing test * PROGRAMMER: Thomas Faber <thomas.faber@reactos.org> */
#include <kmt_test.h>#define NDEBUG#include <debug.h>
#define CheckObject(Handle, Pointers, Handles) do \{ \ PUBLIC_OBJECT_BASIC_INFORMATION ObjectInfo; \ Status = ZwQueryObject(Handle, ObjectBasicInformation, \ &ObjectInfo, sizeof ObjectInfo, NULL); \ ok_eq_hex(Status, STATUS_SUCCESS); \ ok_eq_ulong(ObjectInfo.PointerCount, Pointers); \ ok_eq_ulong(ObjectInfo.HandleCount, Handles); \} while (0)
static POBJECT_TYPE ObDirectoryObjectType;
staticVOIDTestReference( IN HANDLE Handle, IN PUNICODE_STRING Name OPTIONAL, IN PUNICODE_STRING NameUpper OPTIONAL, IN BOOLEAN CaseSensitive, IN ULONG AdditionalReferences, IN BOOLEAN Permanent){ NTSTATUS Status; LONG_PTR Ret; PVOID Object = NULL; PVOID Object2 = NULL; PVOID Object3 = NULL; PVOID Object4 = NULL;
CheckObject(Handle, 2LU + AdditionalReferences, 1LU);
Status = ObReferenceObjectByHandle(Handle, DIRECTORY_ALL_ACCESS, NULL, KernelMode, &Object, NULL); ok_eq_hex(Status, STATUS_SUCCESS); ok(Object != NULL, "ObReferenceObjectByHandle returned NULL object\n"); CheckObject(Handle, 3LU + AdditionalReferences, 1LU);
Status = ObReferenceObjectByHandle(Handle, DIRECTORY_ALL_ACCESS, NULL, KernelMode, &Object2, NULL); ok_eq_hex(Status, STATUS_SUCCESS); ok(Object != NULL, "ObReferenceObjectByHandle returned NULL object\n"); ok_eq_pointer(Object, Object2); CheckObject(Handle, 4LU + AdditionalReferences, 1LU);
if (!skip(Object != NULL, "No object to reference!\n")) { Ret = ObReferenceObject(Object); ok_eq_longptr(Ret, (LONG_PTR)4 + AdditionalReferences); CheckObject(Handle, 5LU + AdditionalReferences, 1LU);
Ret = ObReferenceObject(Object); ok_eq_longptr(Ret, (LONG_PTR)5 + AdditionalReferences); CheckObject(Handle, 6LU + AdditionalReferences, 1LU);
Status = ObReferenceObjectByPointer(Object, DIRECTORY_ALL_ACCESS, NULL, KernelMode); ok_eq_hex(Status, STATUS_SUCCESS); CheckObject(Handle, 7LU + AdditionalReferences, 1LU);
Status = ObReferenceObjectByPointer(Object, DIRECTORY_ALL_ACCESS, NULL, KernelMode); ok_eq_hex(Status, STATUS_SUCCESS); CheckObject(Handle, 8LU + AdditionalReferences, 1LU);
Ret = ObDereferenceObject(Object); ok_eq_longptr(Ret, (LONG_PTR)6 + AdditionalReferences); CheckObject(Handle, 7LU + AdditionalReferences, 1LU);
Ret = ObDereferenceObject(Object); ok_eq_longptr(Ret, (LONG_PTR)5 + AdditionalReferences); CheckObject(Handle, 6LU + AdditionalReferences, 1LU);
Ret = ObDereferenceObject(Object); ok_eq_longptr(Ret, (LONG_PTR)4 + AdditionalReferences); CheckObject(Handle, 5LU + AdditionalReferences, 1LU);
Ret = ObDereferenceObject(Object); ok_eq_longptr(Ret, (LONG_PTR)3 + AdditionalReferences); CheckObject(Handle, 4LU + AdditionalReferences, 1LU); }
if (Name && !skip(ObDirectoryObjectType != NULL, "No directory object type\n")) { Status = ObReferenceObjectByName(Name, 0, NULL, DIRECTORY_ALL_ACCESS, ObDirectoryObjectType, KernelMode, NULL, &Object3); ok_eq_hex(Status, STATUS_SUCCESS); CheckObject(Handle, 5LU + AdditionalReferences, 1LU); }
if (NameUpper && !skip(ObDirectoryObjectType != NULL, "No directory object type\n")) { Status = ObReferenceObjectByName(NameUpper, 0, NULL, DIRECTORY_ALL_ACCESS, ObDirectoryObjectType, KernelMode, NULL, &Object4); ok_eq_hex(Status, CaseSensitive ? STATUS_OBJECT_NAME_NOT_FOUND : STATUS_SUCCESS); CheckObject(Handle, 5LU + AdditionalReferences + !CaseSensitive, 1LU); }
if (NameUpper && !skip(Object4 != NULL, "No object to dereference\n")) { Ret = ObDereferenceObject(Object4); ok_eq_longptr(Ret, (LONG_PTR)4 + AdditionalReferences); CheckObject(Handle, 5LU + AdditionalReferences, 1LU); } if (Name && !skip(Object3 != NULL, "No object to dereference\n")) { Ret = ObDereferenceObject(Object3); ok_eq_longptr(Ret, (LONG_PTR)3 + AdditionalReferences); CheckObject(Handle, 4LU + AdditionalReferences, 1LU); } if (!skip(Object2 != NULL, "No object to dereference\n")) { Ret = ObDereferenceObject(Object2); ok_eq_longptr(Ret, (LONG_PTR)2 + AdditionalReferences); CheckObject(Handle, 3LU + AdditionalReferences, 1LU); } if (!skip(Object != NULL, "No object to dereference\n")) { Ret = ObDereferenceObject(Object); ok_eq_longptr(Ret, (LONG_PTR)1 + AdditionalReferences); CheckObject(Handle, 2LU + AdditionalReferences, 1LU); }
CheckObject(Handle, 2LU + AdditionalReferences, 1LU);
if (Permanent) { Status = ZwMakeTemporaryObject(Handle); ok_eq_hex(Status, STATUS_SUCCESS); CheckObject(Handle, 2LU + AdditionalReferences, 1LU);
Status = ZwMakeTemporaryObject(Handle); ok_eq_hex(Status, STATUS_SUCCESS); CheckObject(Handle, 2LU + AdditionalReferences, 1LU); }}
START_TEST(ObReference){ NTSTATUS Status; HANDLE DirectoryHandle = NULL; OBJECT_ATTRIBUTES ObjectAttributes; UNICODE_STRING Name, *pName; UNICODE_STRING NameUpper, *pNameUpper; ULONG i; struct { PWSTR Name; ULONG Flags; ULONG AdditionalReferences; } Tests[] = { { NULL, 0, 0 }, { NULL, OBJ_CASE_INSENSITIVE, 0 }, { NULL, OBJ_KERNEL_HANDLE, 0 }, { NULL, OBJ_PERMANENT, 0 }, { NULL, OBJ_CASE_INSENSITIVE | OBJ_PERMANENT, 0 }, { NULL, OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE, 0 }, { NULL, OBJ_KERNEL_HANDLE | OBJ_PERMANENT, 0 }, { NULL, OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE | OBJ_PERMANENT, 0 }, { L"\\YayDirectory0", 0, 1 }, { L"\\YayDirectory1", OBJ_CASE_INSENSITIVE, 1 }, { L"\\YayDirectory2", OBJ_KERNEL_HANDLE, 1 }, { L"\\YayDirectory3", OBJ_PERMANENT, 1 }, { L"\\YayDirectory4", OBJ_CASE_INSENSITIVE | OBJ_PERMANENT, 1 }, { L"\\YayDirectory5", OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE, 1 }, { L"\\YayDirectory6", OBJ_KERNEL_HANDLE | OBJ_PERMANENT, 1 }, { L"\\YayDirectory7", OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE | OBJ_PERMANENT, 1 }, }; HANDLE ObjectTypeHandle;
/* ObReferenceObjectByName needs the object type... so get it... */ RtlInitUnicodeString(&Name, L"\\ObjectTypes\\Directory"); InitializeObjectAttributes(&ObjectAttributes, &Name, OBJ_KERNEL_HANDLE, NULL, NULL); Status = ObOpenObjectByName(&ObjectAttributes, NULL, KernelMode, NULL, 0, NULL, &ObjectTypeHandle); ok_eq_hex(Status, STATUS_SUCCESS); ok(ObjectTypeHandle != NULL, "ObjectTypeHandle = NULL\n"); if (!skip(Status == STATUS_SUCCESS && ObjectTypeHandle, "No handle\n")) { Status = ObReferenceObjectByHandle(ObjectTypeHandle, 0, NULL, KernelMode, (PVOID)&ObDirectoryObjectType, NULL); ok_eq_hex(Status, STATUS_SUCCESS); ok(ObDirectoryObjectType != NULL, "ObDirectoryObjectType = NULL\n"); Status = ZwClose(ObjectTypeHandle); ok_eq_hex(Status, STATUS_SUCCESS); }
for (i = 0; i < sizeof Tests / sizeof Tests[0]; ++i) { DPRINT("Run %d\n", i); if (Tests[i].Name) { RtlInitUnicodeString(&Name, Tests[i].Name); pName = &Name; Status = RtlUpcaseUnicodeString(&NameUpper, &Name, TRUE); ok_eq_hex(Status, STATUS_SUCCESS); if (skip(Status == STATUS_SUCCESS, "No upper case name\n")) pNameUpper = NULL; else pNameUpper = &NameUpper; } else { pName = NULL; pNameUpper = NULL; } InitializeObjectAttributes(&ObjectAttributes, pName, Tests[i].Flags, NULL, NULL); Status = ZwCreateDirectoryObject(&DirectoryHandle, DIRECTORY_ALL_ACCESS, &ObjectAttributes); ok_eq_hex(Status, STATUS_SUCCESS); ok(DirectoryHandle != NULL, "DirectoryHandle = NULL\n");
if (!skip(Status == STATUS_SUCCESS && DirectoryHandle, "Cannot proceed without an object\n")) { TestReference(DirectoryHandle, pName, pNameUpper, FALSE, Tests[i].AdditionalReferences, (Tests[i].Flags & OBJ_PERMANENT) != 0); /* try again for good measure */ TestReference(DirectoryHandle, pName, pNameUpper, FALSE, Tests[i].AdditionalReferences, FALSE);
Status = ZwClose(DirectoryHandle); ok_eq_hex(Status, STATUS_SUCCESS); DirectoryHandle = NULL; }
if (pNameUpper) RtlFreeUnicodeString(pNameUpper); }
if (DirectoryHandle) { Status = ZwClose(DirectoryHandle); ok_eq_hex(Status, STATUS_SUCCESS); }
/* parameter tests */ /* bugcheck at APC_LEVEL Status = ObReferenceObject(NULL); Status = ObReferenceObjectByPointer(NULL, 0, NULL, UserMode); Status = ObReferenceObjectByPointer(NULL, 0, NULL, KernelMode);*/
if (ObDirectoryObjectType) { ObDereferenceObject(ObDirectoryObjectType); ObDirectoryObjectType = NULL; }}