[ros-diffs] [fireball] 22295: - Add create/insert object tests - Add simple referencing tests (now disabled) - All tests pass currently on Windows 2003
fireball at svn.reactos.org
fireball at svn.reactos.org
Sat Jun 10 18:37:35 CEST 2006
Author: fireball
Date: Sat Jun 10 20:37:34 2006
New Revision: 22295
URL: http://svn.reactos.ru/svn/reactos?rev=22295&view=rev
Log:
- Add create/insert object tests
- Add simple referencing tests (now disabled)
- All tests pass currently on Windows 2003
Modified:
trunk/reactos/drivers/test/kmtest/ntos_ob.c
Modified: trunk/reactos/drivers/test/kmtest/ntos_ob.c
URL: http://svn.reactos.ru/svn/reactos/trunk/reactos/drivers/test/kmtest/ntos_ob.c?rev=22295&r1=22294&r2=22295&view=diff
==============================================================================
--- trunk/reactos/drivers/test/kmtest/ntos_ob.c (original)
+++ trunk/reactos/drivers/test/kmtest/ntos_ob.c Sat Jun 10 20:37:34 2006
@@ -60,6 +60,9 @@
PVOID ObBody[NUM_OBTYPES];
PMY_OBJECT1 ObObject1;
PMY_OBJECT2 ObObject2;
+HANDLE ObHandle1[NUM_OBTYPES];
+HANDLE ObHandle2[NUM_OBTYPES];
+HANDLE DirectoryHandle;
/* PRIVATE FUNCTIONS **********************************************************/
@@ -70,30 +73,34 @@
DbgPrint("DumpProc() called\n");
}
+// prototype doesn't match Win2003! (causes BSOD)
VOID
OpenProc(IN OB_OPEN_REASON OpenReason,
- IN PEPROCESS Process,
+ IN PEPROCESS Process,
+ IN PVOID Object,
+ IN ACCESS_MASK GrantedAccess,
+ IN ULONG HandleCount)
+{
+ DbgPrint("OpenProc() called\n");
+ DbgBreakPoint();
+}
+
+// Tested in Win2k3
+VOID
+CloseProc(IN PEPROCESS Process,
IN PVOID Object,
IN ACCESS_MASK GrantedAccess,
- IN ULONG HandleCount)
-{
- DbgPrint("OpenProc() called\n");
-}
-
-VOID
-CloseProc(IN PEPROCESS Process,
- IN PVOID Object,
- IN ACCESS_MASK GrantedAccess,
- IN ULONG ProcessHandleCount,
- IN ULONG SystemHandleCount)
-{
- DbgPrint("CloseProc() called\n");
-}
-
+ IN ULONG ProcessHandleCount,
+ IN ULONG SystemHandleCount)
+{
+ DPRINT("CloseProc() called for Object=0x%p\n", Object);
+}
+
+// Tested in Win2k3
VOID
DeleteProc(IN PVOID Object)
{
- DbgPrint("DeleteProc()called\n");
+ DPRINT("DeleteProc() called for Object=0x%p\n", Object);
}
NTSTATUS
@@ -115,7 +122,7 @@
VOID
ObtCreateObjectTypes()
{
- UCHAR i;
+ USHORT i;
NTSTATUS Status;
for (i=0; i<NUM_OBTYPES; i++)
@@ -135,11 +142,19 @@
ObTypeInitializer[i].MaintainHandleCount = TRUE;
ObTypeInitializer[i].ValidAccessMask = OBJECT_TYPE_ALL_ACCESS;
+ // Test for invalid parameter
+ // FIXME: Make it more exact, to see which params Win2k3 checks
+ // existence of
+ Status = ObCreateObjectType(&ObTypeName[i], &ObTypeInitializer[i],
+ (PSECURITY_DESCRIPTOR)NULL, &ObTypes[i]);
+ ok(Status == STATUS_INVALID_PARAMETER,
+ "ObCreateObjectType returned 0x%lX", Status);
+
// Object procedures
ObTypeInitializer[i].CloseProcedure = (OB_CLOSE_METHOD)CloseProc;
ObTypeInitializer[i].DeleteProcedure = (OB_DELETE_METHOD)DeleteProc;
ObTypeInitializer[i].DumpProcedure = (OB_DUMP_METHOD)DumpProc;
- ObTypeInitializer[i].OpenProcedure = (OB_OPEN_METHOD)OpenProc;
+ //ObTypeInitializer[i].OpenProcedure = (OB_OPEN_METHOD)OpenProc;
ObTypeInitializer[i].ParseProcedure = (OB_PARSE_METHOD)ParseProc;
Status = ObCreateObjectType(&ObTypeName[i], &ObTypeInitializer[i],
@@ -153,7 +168,6 @@
ObtCreateDirectory()
{
NTSTATUS Status;
- HANDLE DirectoryHandle;
// Directory will have permanent and case insensitive flags
RtlInitUnicodeString(&ObDirectoryName, L"\\ObtDirectory");
@@ -163,17 +177,15 @@
Status = ZwCreateDirectoryObject(&DirectoryHandle, 0, &ObDirectoryAttributes);
ok(Status == STATUS_SUCCESS,
"Failed to create directory object with status=0x%lX", Status);
-
- Status = ZwClose(DirectoryHandle);
- ok(Status == STATUS_SUCCESS,
- "Failed to close handle with status=0x%lX", Status);
}
VOID
ObtCreateObjects()
{
+ PVOID ObBody1[2];
NTSTATUS Status;
+ // Create two objects
RtlInitUnicodeString(&ObName[0], L"\\ObtDirectory\\MyObject1");
InitializeObjectAttributes(&ObAttributes[0], &ObName[0],
OBJ_CASE_INSENSITIVE, NULL, NULL);
@@ -187,20 +199,175 @@
(PVOID *)&ObBody[0]);
ok(Status == STATUS_SUCCESS,
"Failed to create object with status=0x%lX", Status);
- DPRINT("Created Object 0\n");
Status = ObCreateObject(KernelMode, ObTypes[1], &ObAttributes[1],
KernelMode, NULL, (ULONG)sizeof(MY_OBJECT2), 0L, 0L,
(PVOID *)&ObBody[1]);
ok(Status == STATUS_SUCCESS,
"Failed to create object with status=0x%lX", Status);
- DPRINT("Created Object 1\n");
+
+ // Insert them
+ Status = ObInsertObject(ObBody[0], NULL, STANDARD_RIGHTS_ALL, 0,
+ &ObBody[0], &ObHandle1[0]);
+ ok(Status == STATUS_SUCCESS,
+ "Failed to insert object 0 with status=0x%lX", Status);
+ ok(ObBody[0] != NULL, "Object body = NULL");
+ ok(ObHandle1[0] != NULL, "Handle = NULL");
+
+ Status = ObInsertObject(ObBody[1], NULL, GENERIC_ALL, 0,
+ &ObBody[1], &ObHandle1[1]);
+ ok(Status == STATUS_SUCCESS,
+ "Failed to insert object 1 with status=0x%lX", Status);
+ ok(ObBody[1] != NULL, "Object body = NULL");
+ ok(ObHandle1[1] != NULL, "Handle = NULL");
+
+ // Now create an object of type 0, of the same name and expect it to fail
+ // inserting, but success creation
+ RtlInitUnicodeString(&ObName[0], L"\\ObtDirectory\\MyObject1");
+ InitializeObjectAttributes(&ObAttributes[0], &ObName[0], OBJ_OPENIF,
+ NULL, NULL);
+
+ Status = ObCreateObject(KernelMode, ObTypes[0], &ObAttributes[0], KernelMode,
+ NULL, (ULONG)sizeof(MY_OBJECT1), 0L, 0L, (PVOID *)&ObBody1[0]);
+ ok(Status == STATUS_SUCCESS,
+ "Failed to create object with status=0x%lX", Status);
+
+ Status = ObInsertObject(ObBody1[0], NULL, GENERIC_ALL, 0,
+ &ObBody1[1], &ObHandle2[0]);
+ ok(Status == STATUS_OBJECT_NAME_EXISTS,
+ "Object insertion should have failed, but got 0x%lX", Status);
+ ok(ObBody[0] == ObBody1[1],
+ "Object bodies doesn't match, 0x%p != 0x%p", ObBody[0], ObBody1[1]);
+ ok(ObHandle2[0] != NULL, "Bad handle returned 0x%lX", (ULONG)ObHandle2[0]);
+
+ // Close its handle
+ Status = ZwClose(ObHandle2[0]);
+ ok(Status == STATUS_SUCCESS,
+ "Failed to close handle status=0x%lX", Status);
+
+ // Object referenced 2 times:
+ // 1) ObInsertObject
+ // 2) AdditionalReferences
+ ObDereferenceObject(ObBody1[1]);
}
VOID
ObtClose()
{
- // TODO: Close what we have opened and free what we allocated
+ PVOID DirObject;
+ NTSTATUS Status;
+ //PVOID TypeObject;
+ //USHORT i;
+ //UNICODE_STRING ObPathName[NUM_OBTYPES];
+
+ // Close what we have opened and free what we allocated
+ ZwClose(ObHandle1[0]);
+ ZwClose(ObHandle1[1]);
+ ZwClose(ObHandle2[0]);
+ ZwClose(ObHandle2[1]);
+
+ // Now we have to get rid of a directory object
+ // Since it is permanent, we have to firstly make it temporary
+ // and only then kill
+ // (this procedure is described in DDK)
+ Status = ObReferenceObjectByHandle(DirectoryHandle, 0L, NULL,
+ KernelMode, &DirObject, NULL);
+ ok(Status == STATUS_SUCCESS,
+ "Failed to reference object by handle with status=0x%lX", Status);
+
+ // Dereference 2 times - first for just previous referencing
+ // and 2nd time for creation of permanent object itself
+ ObDereferenceObject(DirObject);
+ ObDereferenceObject(DirObject);
+
+ Status = ZwMakeTemporaryObject(DirectoryHandle);
+ ok(Status == STATUS_SUCCESS,
+ "Failed to make temp object with status=0x%lX", Status);
+
+ // Close the handle now and we are done
+ Status = ZwClose(DirectoryHandle);
+ ok(Status == STATUS_SUCCESS,
+ "Failed to close handle with status=0x%lX", Status);
+
+ // Now delete the last piece - object types
+ // FIXME: How to do this correctly?
+ /*
+ for (i=0; i<NUM_OBTYPES; i++)
+ {
+ ObDereferenceObject(ObTypes[i]);
+ }*/
+ /*
+ RtlInitUnicodeString(&ObPathName[0], L"\\ObjectTypes\\MyObjectType1");
+ RtlInitUnicodeString(&ObPathName[1], L"\\ObjectTypes\\MyObjectType2");
+
+ for (i=0; i<NUM_OBTYPES; i++)
+ {
+ Status = ObReferenceObjectByName(&ObPathName[i],
+ OBJ_CASE_INSENSITIVE, NULL, 0L, NULL, KernelMode, NULL,
+ &TypeObject);
+
+ ObDereferenceObject(TypeObject);
+ ObDereferenceObject(TypeObject);
+ DPRINT("Reference Name %S = %p, ObTypes[%d] = %p\n",
+ ObPathName[i], TypeObject, i, ObTypes[i]);
+ }*/
+}
+
+VOID
+ObtReferenceTests()
+{
+ USHORT i;
+ NTSTATUS Status;
+ UNICODE_STRING ObPathName[NUM_OBTYPES];
+
+ // Reference them by handle
+ for (i=0; i<NUM_OBTYPES; i++)
+ {
+ Status = ObReferenceObjectByHandle(ObHandle1[i], 0L, ObTypes[i],
+ KernelMode, &ObBody[i], NULL);
+ ok(Status == STATUS_SUCCESS,
+ "Failed to reference object by handle, status=0x%lX", Status);
+ DPRINT("Ref by handle %lx = %p\n", ObHandle1[i], ObBody[i]);
+ }
+
+ // Reference them by pointer
+ for (i=0; i<NUM_OBTYPES; i++)
+ {
+ Status = ObReferenceObjectByPointer(ObBody[i], 0L, ObTypes[i], KernelMode);
+ ok(Status == STATUS_SUCCESS,
+ "Failed to reference object by pointer, status=0x%lX", Status);
+ }
+
+ // Reference them by name
+ RtlInitUnicodeString(&ObPathName[0], L"\\ObtDirectory\\MyObject1");
+ RtlInitUnicodeString(&ObPathName[1], L"\\ObtDirectory\\MyObject2");
+
+ for (i=0; i<NUM_OBTYPES; i++)
+ {
+ Status = ObReferenceObjectByName(&ObPathName[i],
+ OBJ_CASE_INSENSITIVE, NULL, 0L, ObTypes[i], KernelMode, NULL,
+ &ObBody[0]);
+
+ DPRINT("Ref by name %S = %p\n", ObPathName[i], ObBody[i]);
+ }
+
+ // Dereference now all of them
+
+ // For ObInsertObject, AdditionalReference
+ ObDereferenceObject(ObBody[0]);
+ ObDereferenceObject(ObBody[1]);
+
+ // For ByHandle
+ ObDereferenceObject(ObBody[0]);
+ ObDereferenceObject(ObBody[1]);
+
+ // For ByPointer
+ ObDereferenceObject(ObBody[0]);
+ ObDereferenceObject(ObBody[1]);
+
+ // For ByName
+ ObDereferenceObject(ObBody[0]);
+ ObDereferenceObject(ObBody[1]);
}
/* PUBLIC FUNCTIONS ***********************************************************/
@@ -223,7 +390,13 @@
ObtCreateObjects();
DPRINT("ObtCreateObjects() done\n");
+ // Reference them in a variety of ways
+ // FIXME: Disabled due to ParseProcedure call
+ //ObtReferenceTests();
+
// Clean up
+ // FIXME: Disabled to see results of creating objects in usermode
+ // and also due to problems with object types removal
ObtClose();
DPRINT("Cleanup done\n");
More information about the Ros-diffs
mailing list