--- /dev/null
+/*
+ * PROJECT: ReactOS kernel-mode tests
+ * LICENSE: GPLv2+ - See COPYING in the top level directory
+ * PURPOSE: Kernel-Mode Test Suite Object types test
+ * PROGRAMMER: Thomas Faber <thfabba@gmx.de>
+ */
+
+#include <kmt_test.h>
+
+#define NDEBUG
+#include <debug.h>
+
+static
+POBJECT_TYPE
+GetObjectType(
+ IN PCWSTR TypeName)
+{
+ NTSTATUS Status;
+ UNICODE_STRING Name;
+ OBJECT_ATTRIBUTES ObjectAttributes;
+ HANDLE Handle;
+ PVOID ObjectType = NULL;
+
+ RtlInitUnicodeString(&Name, TypeName);
+ InitializeObjectAttributes(&ObjectAttributes, &Name, OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE, NULL, NULL);
+ Status = ObOpenObjectByName(&ObjectAttributes, NULL, KernelMode, NULL, 0, NULL, &Handle);
+ ok_eq_hex(Status, STATUS_SUCCESS);
+ ok(Handle != NULL, "ObjectTypeHandle = NULL\n");
+ if (!skip(Status == STATUS_SUCCESS && Handle, "No handle\n"))
+ {
+ Status = ObReferenceObjectByHandle(Handle, 0, NULL, KernelMode, &ObjectType, NULL);
+ ok_eq_hex(Status, STATUS_SUCCESS);
+ ok(ObjectType != NULL, "ObjectType = NULL\n");
+ Status = ZwClose(Handle);
+ ok_eq_hex(Status, STATUS_SUCCESS);
+ }
+ return ObjectType;
+}
+
+#define ok_eq_ustr(value, expected) ok(RtlEqualUnicodeString(value, expected, FALSE), #value " = \"%wZ\", expected \"%%wZ\"\n", value, expected)
+
+#define CheckObjectType(TypeName, Variable, Flags, InvalidAttr, \
+ ReadMapping, WriteMapping, ExecMapping, AllMapping, \
+ ValidMask) do \
+{ \
+ POBJECT_TYPE ObjectType; \
+ UNICODE_STRING Name; \
+ ULONG Key; \
+ BOOLEAN UseDefault = ((Flags) & OBT_NO_DEFAULT) == 0; \
+ BOOLEAN CustomSecurityProc = ((Flags) & OBT_CUSTOM_SECURITY_PROC) != 0; \
+ BOOLEAN SecurityRequired = ((Flags) & OBT_SECURITY_REQUIRED) != 0; \
+ BOOLEAN CaseInsensitive = ((Flags) & OBT_CASE_INSENSITIVE) != 0; \
+ BOOLEAN MaintainTypeList = ((Flags) & OBT_MAINTAIN_TYPE_LIST) != 0; \
+ BOOLEAN MaintainHandleCount = ((Flags) & OBT_MAINTAIN_HANDLE_COUNT) != 0; \
+ POOL_TYPE PoolType = ((Flags) & OBT_PAGED_POOL) ? PagedPool : NonPagedPool; \
+ BOOLEAN CustomKey = ((Flags) & OBT_CUSTOM_KEY) != 0; \
+ \
+ trace(#TypeName "\n"); \
+ ObjectType = GetObjectType(L"\\ObjectTypes\\" L ## #TypeName); \
+ ok(ObjectType != NULL, "ObjectType = NULL\n"); \
+ if (!skip(ObjectType != NULL, "No ObjectType\n")) \
+ { \
+ ok(!Variable || Variable == ObjectType, \
+ #Variable "is %p, expected %p\n", Variable, ObjectType); \
+ RtlInitUnicodeString(&Name, L ## #TypeName); \
+ ok_eq_ustr(&ObjectType->Name, &Name); \
+ ok_eq_ulong(ObjectType->Index, Index); \
+ Index = ObjectType->Index + 1; \
+ /* apparently, File and WaitablePort are evil and have other stuff \
+ * in DefaultObject. All others are NULL */ \
+ if (UseDefault) \
+ ok_eq_pointer(ObjectType->DefaultObject, ObpDefaultObject); \
+ /*ok(ObjectType->TotalNumberOfObjects >= 1, \
+ "Number of objects = %lu\n", ObjectType->TotalNumberOfObjects); \
+ ok(ObjectType->TotalNumberOfHandles >= ObjectType->TotalNumberOfObjects, \
+ "%lu objects, but %lu handles\n", \
+ ObjectType->TotalNumberOfObjects, ObjectType->TotalNumberOfHandles);*/ \
+ ok(ObjectType->HighWaterNumberOfObjects >= ObjectType->TotalNumberOfObjects,\
+ "%lu objects, but high water %lu\n", \
+ ObjectType->TotalNumberOfObjects, ObjectType->HighWaterNumberOfObjects); \
+ ok(ObjectType->HighWaterNumberOfHandles >= ObjectType->TotalNumberOfHandles,\
+ "%lu handles, but high water %lu\n", \
+ ObjectType->TotalNumberOfHandles, ObjectType->HighWaterNumberOfHandles); \
+ ok_eq_ulong(ObjectType->TypeInfo.Length, sizeof(OBJECT_TYPE_INITIALIZER)); \
+ ok_eq_bool(ObjectType->TypeInfo.UseDefaultObject, UseDefault); \
+ ok_eq_bool(ObjectType->TypeInfo.CaseInsensitive, CaseInsensitive); \
+ ok_eq_hex(ObjectType->TypeInfo.InvalidAttributes, InvalidAttr); \
+ ok_eq_hex(ObjectType->TypeInfo.GenericMapping.GenericRead, ReadMapping); \
+ ok_eq_hex(ObjectType->TypeInfo.GenericMapping.GenericWrite, WriteMapping); \
+ ok_eq_hex(ObjectType->TypeInfo.GenericMapping.GenericExecute, ExecMapping); \
+ ok_eq_hex(ObjectType->TypeInfo.GenericMapping.GenericAll, AllMapping); \
+ ok_eq_hex(ObjectType->TypeInfo.ValidAccessMask, ValidMask); \
+ ok_eq_bool(ObjectType->TypeInfo.SecurityRequired, SecurityRequired); \
+ ok_eq_bool(ObjectType->TypeInfo.MaintainHandleCount, MaintainHandleCount); \
+ ok_eq_bool(ObjectType->TypeInfo.MaintainTypeList, MaintainTypeList); \
+ ok_eq_ulong(ObjectType->TypeInfo.PoolType, PoolType); \
+ /* DefaultPagedPoolCharge */ \
+ /* DefaultNonPagedPoolCharge */ \
+ /* DumpProcedure */ \
+ /* OpenProcedure */ \
+ /* CloseProcedure */ \
+ /* DeleteProcedure */ \
+ /* ParseProcedure */ \
+ if (CustomSecurityProc) \
+ ok(ObjectType->TypeInfo.SecurityProcedure != NULL, \
+ "No Security Proc\n"); \
+ else \
+ ok_eq_pointer(ObjectType->TypeInfo.SecurityProcedure, \
+ SeDefaultObjectMethod); \
+ /* QueryNameProcedure */ \
+ /* OkayToCloseProcedure */ \
+ Key = *(PULONG)#TypeName; \
+ if (sizeof(#TypeName) <= 4) Key |= ' ' << 24; \
+ if (sizeof(#TypeName) <= 3) Key |= ' ' << 16; \
+ if (sizeof(#TypeName) <= 2) Key |= ' ' << 8; \
+ if (!CustomKey) \
+ ok_eq_hex(ObjectType->Key, Key); \
+ ObDereferenceObject(ObjectType); \
+ } \
+} while (0)
+
+static POBJECT_TYPE ObpTypeObjectType;
+#define ObpDirectoryObjectType NULL
+#define ObpSymbolicLinkObjectType NULL
+#define DbgkDebugObjectType NULL
+#define ExEventPairObjectType NULL
+#define ExMutantObjectType NULL
+#define ExCallbackObjectType NULL
+#define ExTimerObjectType NULL
+#define ExProfileObjectType NULL
+#define ExpKeyedEventObjectType NULL
+#define CmpKeyObjectType NULL
+#define LpcWaitablePortObjectType NULL
+#define IoControllerObjectType NULL
+#define IoCompletionObjectType NULL
+#define WmipGuidObjectType NULL
+
+static POBJECT_TYPE ObpDefaultObject;
+static OB_SECURITY_METHOD SeDefaultObjectMethod;
+
+#define OBT_NO_DEFAULT 0x01
+#define OBT_CUSTOM_SECURITY_PROC 0x02
+#define OBT_SECURITY_REQUIRED 0x04
+#define OBT_CASE_INSENSITIVE 0x08
+#define OBT_MAINTAIN_TYPE_LIST 0x10
+#define OBT_MAINTAIN_HANDLE_COUNT 0x20
+#define OBT_PAGED_POOL 0x40
+#define OBT_CUSTOM_KEY 0x80
+
+#define TAG(x) RtlUlongByteSwap(x)
+
+static
+VOID
+TestObjectTypes(VOID)
+{
+ ULONG Index;
+
+ ObpTypeObjectType = GetObjectType(L"\\ObjectTypes\\Type");
+ if (skip(ObpTypeObjectType != NULL, "No Type object type\n"))
+ return;
+
+ ObpDefaultObject = ObpTypeObjectType->DefaultObject;
+ ok(ObpDefaultObject != NULL, "No ObpDefaultObject\n");
+ SeDefaultObjectMethod = ObpTypeObjectType->TypeInfo.SecurityProcedure;
+ ok(SeDefaultObjectMethod != NULL, "No SeDefaultObjectMethod\n");
+
+ Index = 1;
+ CheckObjectType(Type, ObpTypeObjectType, OBT_MAINTAIN_TYPE_LIST | OBT_CUSTOM_KEY, 0x100, 0x020000, 0x020000, 0x020000, 0x0f0001, 0x1f0001);
+ ok_eq_hex(ObpTypeObjectType->Key, TAG('ObjT'));
+ CheckObjectType(Directory, ObpDirectoryObjectType, OBT_CASE_INSENSITIVE | OBT_PAGED_POOL, 0x100, 0x020003, 0x02000c, 0x020003, 0x0f000f, 0x0f000f);
+ CheckObjectType(SymbolicLink, ObpSymbolicLinkObjectType, OBT_CASE_INSENSITIVE | OBT_PAGED_POOL, 0x100, 0x020001, 0x020000, 0x020001, 0x0f0001, 0x0f0001);
+ CheckObjectType(Token, SeTokenObjectType, OBT_SECURITY_REQUIRED | OBT_PAGED_POOL, 0x100, 0x020008, 0x0200e0, 0x020000, 0x0f01ff, 0x1f01ff);
+ CheckObjectType(Process, PsProcessType, OBT_NO_DEFAULT | OBT_SECURITY_REQUIRED, 0x0b0, 0x020410, 0x020beb, 0x120000, 0x1f0fff, 0x1f0fff);
+ CheckObjectType(Thread, PsThreadType, OBT_NO_DEFAULT | OBT_SECURITY_REQUIRED, 0x0b0, 0x020048, 0x020037, 0x120000, 0x1f03ff, 0x1f03ff);
+ CheckObjectType(Job, PsJobType, OBT_NO_DEFAULT | OBT_SECURITY_REQUIRED, 0x000, 0x020004, 0x02000b, 0x120000, 0x1f03ff, 0x1f001f);
+ CheckObjectType(DebugObject, DbgkDebugObjectType, OBT_NO_DEFAULT | OBT_SECURITY_REQUIRED, 0x000, 0x020001, 0x020002, 0x120000, 0x1f000f, 0x1f000f);
+ CheckObjectType(Event, ExEventObjectType, OBT_NO_DEFAULT, 0x100, 0x020001, 0x020002, 0x120000, 0x1f0003, 0x1f0003);
+ CheckObjectType(EventPair, ExEventPairObjectType, 0, 0x100, 0x120000, 0x120000, 0x120000, 0x1f0000, 0x1f0000);
+ CheckObjectType(Mutant, ExMutantObjectType, OBT_NO_DEFAULT, 0x100, 0x020001, 0x020000, 0x120000, 0x1f0001, 0x1f0001);
+ CheckObjectType(Callback, ExCallbackObjectType, OBT_NO_DEFAULT, 0x100, 0x020000, 0x020001, 0x120000, 0x1f0001, 0x1f0001);
+ CheckObjectType(Semaphore, ExSemaphoreObjectType, OBT_NO_DEFAULT, 0x100, 0x020001, 0x020002, 0x120000, 0x1f0003, 0x1f0003);
+ CheckObjectType(Timer, ExTimerObjectType, OBT_NO_DEFAULT, 0x100, 0x020001, 0x020002, 0x120000, 0x1f0003, 0x1f0003);
+ CheckObjectType(Profile, ExProfileObjectType, OBT_NO_DEFAULT, 0x100, 0x020001, 0x020001, 0x020001, 0x0f0001, 0x0f0001);
+ CheckObjectType(KeyedEvent, ExpKeyedEventObjectType, OBT_PAGED_POOL, 0x000, 0x020001, 0x020002, 0x020000, 0x0f0003, 0x1f0003);
+ CheckObjectType(WindowStation, ExWindowStationObjectType, OBT_NO_DEFAULT | OBT_SECURITY_REQUIRED | OBT_MAINTAIN_HANDLE_COUNT,
+ 0x130, 0x020303, 0x02001c, 0x020060, 0x0f037f, 0x0f037f);
+ CheckObjectType(Desktop, ExDesktopObjectType, OBT_NO_DEFAULT | OBT_SECURITY_REQUIRED | OBT_MAINTAIN_HANDLE_COUNT,
+ 0x130, 0x020041, 0x0200be, 0x020100, 0x0f01ff, 0x0f01ff);
+ CheckObjectType(Section, MmSectionObjectType, OBT_PAGED_POOL, 0x100, 0x020005, 0x020002, 0x020008, 0x0f001f, 0x1f001f);
+ CheckObjectType(Key, CmpKeyObjectType, OBT_CUSTOM_SECURITY_PROC | OBT_SECURITY_REQUIRED | OBT_PAGED_POOL,
+ 0x030, 0x020019, 0x020006, 0x020019, 0x0f003f, 0x1f003f);
+ CheckObjectType(Port, LpcPortObjectType, OBT_PAGED_POOL, 0x7b2, 0x020001, 0x010001, 0x000000, 0x1f0001, 0x1f0001);
+ CheckObjectType(WaitablePort, LpcWaitablePortObjectType, OBT_NO_DEFAULT, 0x7b2, 0x020001, 0x010001, 0x000000, 0x1f0001, 0x1f0001);
+ CheckObjectType(Adapter, IoAdapterObjectType, 0, 0x100, 0x120089, 0x120116, 0x1200a0, 0x1f01ff, 0x1f01ff);
+ CheckObjectType(Controller, IoControllerObjectType, 0, 0x100, 0x120089, 0x120116, 0x1200a0, 0x1f01ff, 0x1f01ff);
+ CheckObjectType(Device, IoDeviceObjectType, OBT_CUSTOM_SECURITY_PROC | OBT_CASE_INSENSITIVE,
+ 0x100, 0x120089, 0x120116, 0x1200a0, 0x1f01ff, 0x1f01ff);
+ CheckObjectType(Driver, IoDriverObjectType, OBT_CASE_INSENSITIVE, 0x100, 0x120089, 0x120116, 0x1200a0, 0x1f01ff, 0x1f01ff);
+ CheckObjectType(IoCompletion, IoCompletionObjectType, OBT_CASE_INSENSITIVE, 0x110, 0x020001, 0x020002, 0x120000, 0x1f0003, 0x1f0003);
+ CheckObjectType(File, IoFileObjectType, OBT_NO_DEFAULT | OBT_CUSTOM_SECURITY_PROC | OBT_CASE_INSENSITIVE | OBT_MAINTAIN_HANDLE_COUNT,
+ 0x130, 0x120089, 0x120116, 0x1200a0, 0x1f01ff, 0x1f01ff);
+ CheckObjectType(WmiGuid, WmipGuidObjectType, OBT_NO_DEFAULT | OBT_CUSTOM_SECURITY_PROC | OBT_SECURITY_REQUIRED,
+ 0x100, 0x000001, 0x000002, 0x000010, 0x120fff, 0x1f0fff);
+ CheckObjectType(FilterConnectionPort, NULL, OBT_NO_DEFAULT | OBT_SECURITY_REQUIRED, 0x100, 0x020001, 0x010001, 0x000000, 0x1f0001, 0x1f0001);
+ CheckObjectType(FilterCommunicationPort, NULL, OBT_NO_DEFAULT, 0x100, 0x020001, 0x010001, 0x000000, 0x1f0001, 0x1f0001);
+
+ // exported but not created
+ ok_eq_pointer(IoDeviceHandlerObjectType, NULL);
+
+ // my Win7/x64 additionally has:
+ // ALPC Port
+ // EtwConsumer
+ // EtwRegistration
+ // IoCompletionReserve
+ // PcwObject
+ // PowerRequest
+ // Session
+ // TmEn
+ // TmRm
+ // TmTm
+ // TmTx
+ // TpWorkerFactory
+ // UserApcReserve
+ // ... and does not have:
+ // Port
+ // WaitablePort
+
+ ObDereferenceObject(ObpTypeObjectType);
+}
+
+START_TEST(ObTypes)
+{
+ TestObjectTypes();
+}
projects / reactos.git / commitdiff