[tfaber] 53280: [KMTESTS] - ObType: take advantage of r53211 - IoDriverObject: add the remaining functions from the old drvobj and devobj tests - fix some rbuild files

Author: tfaber
Date: Wed Aug 17 11:50:46 2011
New Revision: 53280

URL: http://svn.reactos.org/svn/reactos?rev=53280&view=rev
Log:
[KMTESTS]
- ObType: take advantage of r53211
- IoDriverObject: add the remaining functions from the old drvobj and devobj tests
- fix some rbuild files

Added:
    branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/iohelper_drv.rbuild
      - copied unchanged from r53279,
branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoHelper_drv.rbuild
Removed:
    branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoHelper_drv.rbuild
Modified:
    branches/GSoC_2011/KMTestSuite/kmtests/directory.rbuild
    branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoDriverObject_drv.c
    branches/GSoC_2011/KMTestSuite/kmtests/ntos_ob/ObType.c

Modified: branches/GSoC_2011/KMTestSuite/kmtests/directory.rbuild
URL:
http://svn.reactos.org/svn/reactos/branches/GSoC_2011/KMTestSuite/kmtests/d…
==============================================================================
--- branches/GSoC_2011/KMTestSuite/kmtests/directory.rbuild [iso-8859-1] (original)
+++ branches/GSoC_2011/KMTestSuite/kmtests/directory.rbuild [iso-8859-1] Wed Aug 17
11:50:46 2011
@@ -6,6 +6,7 @@
 	</directory>
 	<directory name="ntos_io">
 		<xi:include href="ntos_io/iodriverobject_drv.rbuild" />
+		<xi:include href="ntos_io/iohelper_drv.rbuild" />
 	</directory>
 	<xi:include href="kmtest.rbuild" />
 	<xi:include href="kmtest_drv.rbuild" />

Modified: branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoDriverObject_drv.c
URL:
http://svn.reactos.org/svn/reactos/branches/GSoC_2011/KMTestSuite/kmtests/n…
==============================================================================
--- branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoDriverObject_drv.c [iso-8859-1]
(original)
+++ branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoDriverObject_drv.c [iso-8859-1] Wed
Aug 17 11:50:46 2011
@@ -21,6 +21,10 @@
 static KMT_IRP_HANDLER TestIrpHandler;
 static VOID TestDriverObject(IN PDRIVER_OBJECT DriverObject, IN DRIVER_STATUS
DriverStatus);
 
+static PDEVICE_OBJECT MainDeviceObject;
+static PDEVICE_OBJECT AttachDeviceObject;
+static PDRIVER_OBJECT ThisDriverObject;
+
 NTSTATUS
 TestEntry(
     IN PDRIVER_OBJECT DriverObject,
@@ -132,3 +136,300 @@
         }
     }
 }
+
+BOOLEAN ZwLoadTest(PDRIVER_OBJECT DriverObject, PUNICODE_STRING DriverRegistryPath,
PWCHAR NewDriverRegPath)
+{
+    UNICODE_STRING RegPath;
+    NTSTATUS Status;
+
+    /* Try to load ourself */
+    Status = ZwLoadDriver(DriverRegistryPath);
+    ok (Status == STATUS_IMAGE_ALREADY_LOADED, "Expected NTSTATUS
STATUS_IMAGE_ALREADY_LOADED, got 0x%lX", Status);
+
+    if (Status != STATUS_IMAGE_ALREADY_LOADED)
+    {
+        DbgPrint("WARNING: Loading this a second time will cause
BUGCHECK!\n");
+    }
+
+    /* Try to load with a Registry Path that doesnt exist */
+    RtlInitUnicodeString(&RegPath,
L"\\Registry\\Machine\\System\\CurrentControlSet\\Services\\deadbeef");
+    Status = ZwLoadDriver(&RegPath);
+    ok (Status == STATUS_OBJECT_NAME_NOT_FOUND, "Expected NTSTATUS
STATUS_OBJECT_NAME_NOT_FOUND, got 0x%lX", Status);
+
+    /* Load the driver */
+    RtlInitUnicodeString(&RegPath, NewDriverRegPath);
+    Status = ZwLoadDriver(&RegPath);
+    ok(Status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got 0x%lX",
Status);
+
+    if (!NT_SUCCESS(Status))
+    {
+        return FALSE;
+    }
+
+    return TRUE;
+}
+
+BOOLEAN ZwUnloadTest(PDRIVER_OBJECT DriverObject, PUNICODE_STRING DriverRegistryPath,
PWCHAR NewDriverRegPath)
+{
+    UNICODE_STRING RegPath;
+    NTSTATUS Status;
+
+    /* Try to unload ourself, which should fail as our Unload routine hasnt been set yet.
*/
+    Status = ZwUnloadDriver(DriverRegistryPath);
+    ok (Status == STATUS_INVALID_DEVICE_REQUEST, "Expected NTSTATUS
STATUS_INVALID_DEVICE_REQUEST, got 0x%lX", Status);
+
+    /* Try to unload with a Registry Path that doesnt exist */
+    RtlInitUnicodeString(&RegPath,
L"\\Registry\\Machine\\System\\CurrentControlSet\\Services\\deadbeef");
+    Status = ZwUnloadDriver(&RegPath);
+    ok (Status == STATUS_OBJECT_NAME_NOT_FOUND, "Expected NTSTATUS
STATUS_OBJECT_NAME_NOT_FOUND, got 0x%lX", Status);
+
+    /* Unload the driver */
+    RtlInitUnicodeString(&RegPath, NewDriverRegPath);
+    Status = ZwUnloadDriver(&RegPath);
+    ok(Status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got 0x%lX",
Status);
+
+    if (!NT_SUCCESS(Status))
+    {
+        return FALSE;
+    }
+
+    return TRUE;
+}
+
+VOID LowerDeviceKernelAPITest(PDEVICE_OBJECT DeviceObject, BOOLEAN UnLoading)
+{
+    PDEVICE_OBJECT RetObject;
+
+    RetObject = IoGetLowerDeviceObject(DeviceObject);
+
+    if (UnLoading)
+    {
+        ok(RetObject == 0,
+            "Expected no Lower DeviceObject, got %p", RetObject);
+    }
+    else
+    {
+        ok(RetObject == AttachDeviceObject,
+            "Expected an Attached DeviceObject %p, got %p", AttachDeviceObject,
RetObject);
+    }
+
+    if (RetObject)
+    {
+        ObDereferenceObject(RetObject);
+    }
+
+    RetObject = IoGetDeviceAttachmentBaseRef(DeviceObject);
+    ok(RetObject == DeviceObject,
+        "Expected an Attached DeviceObject %p, got %p", DeviceObject,
RetObject);
+
+    if (RetObject)
+    {
+        ObDereferenceObject(RetObject);
+    }
+
+}
+VOID DeviceCreatedTest(PDEVICE_OBJECT DeviceObject, BOOLEAN ExclusiveAccess)
+{
+    PEXTENDED_DEVOBJ_EXTENSION extdev;
+
+    /*Check the device object members */
+    ok(DeviceObject->Type==3, "Expected Type = 3, got %x",
DeviceObject->Type);
+    ok(DeviceObject->Size = 0xb8, "Expected Size = 0xba, got %x",
DeviceObject->Size);
+    ok(DeviceObject->ReferenceCount == 0, "Expected ReferenceCount = 0, got
%lu",
+        DeviceObject->ReferenceCount);
+    ok(DeviceObject->DriverObject == ThisDriverObject,
+        "Expected DriverObject member to match this DriverObject %p, got %p",
+        ThisDriverObject, DeviceObject->DriverObject);
+    ok(DeviceObject->NextDevice == NULL, "Expected NextDevice to be NULL, got
%p", DeviceObject->NextDevice);
+    ok(DeviceObject->AttachedDevice == NULL, "Expected AttachDevice to be NULL,
got %p", DeviceObject->AttachedDevice);
+    ok(DeviceObject->Characteristics == 0, "Expected Characteristics to be
0");
+    if (ExclusiveAccess)
+    {
+        ok((DeviceObject->Flags == (DO_DEVICE_HAS_NAME | DO_DEVICE_INITIALIZING |
DO_EXCLUSIVE)),
+            "Expected Flags DO_DEVICE_HAS_NAME | DO_DEVICE_INITIALIZING |
DO_EXCLUSIVE, got %lu", DeviceObject->Flags);
+    }
+    else
+    {
+        ok((DeviceObject->Flags == (DO_DEVICE_HAS_NAME | DO_DEVICE_INITIALIZING)),
+            "Expected Flags DO_DEVICE_HAS_NAME | DO_DEVICE_INITIALIZING, got
%lu", DeviceObject->Flags);
+    }
+    ok(DeviceObject->DeviceType == FILE_DEVICE_UNKNOWN,
+        "Expected DeviceType to match creation parameter FILE_DEVICE_UNKNWOWN, got
%lu",
+        DeviceObject->DeviceType);
+    ok(DeviceObject->ActiveThreadCount == 0, "Expected ActiveThreadCount = 0, got
%lu\n", DeviceObject->ActiveThreadCount);
+
+    /*Check the extended extension */
+    extdev = (PEXTENDED_DEVOBJ_EXTENSION)DeviceObject->DeviceObjectExtension;
+    ok(extdev->ExtensionFlags == 0, "Expected Extended ExtensionFlags to be 0,
got %lu", extdev->ExtensionFlags);
+    ok (extdev->Type == 13, "Expected Type of 13, got %d",
extdev->Type);
+    ok (extdev->Size == 0, "Expected Size of 0, got %d", extdev->Size);
+    ok (extdev->DeviceObject == DeviceObject, "Expected DeviceOject to match
newly created device %p, got %p",
+        DeviceObject, extdev->DeviceObject);
+    ok(extdev->AttachedTo == NULL, "Expected AttachTo to be NULL, got %p",
extdev->AttachedTo);
+    ok(extdev->StartIoCount == 0, "Expected StartIoCount = 0, got %lu",
extdev->StartIoCount);
+    ok(extdev->StartIoKey == 0, "Expected StartIoKey = 0, got %lu",
extdev->StartIoKey);
+    ok(extdev->StartIoFlags == 0, "Expected StartIoFlags = 0, got %lu",
extdev->StartIoFlags);
+}
+
+VOID DeviceDeletionTest(PDEVICE_OBJECT DeviceObject, BOOLEAN Lower)
+{
+    PEXTENDED_DEVOBJ_EXTENSION extdev;
+
+    /*Check the device object members */
+    ok(DeviceObject->Type==3, "Expected Type = 3, got %d",
DeviceObject->Type);
+    ok(DeviceObject->Size = 0xb8, "Expected Size = 0xba, got %d",
DeviceObject->Size);
+    ok(DeviceObject->ReferenceCount == 0, "Expected ReferenceCount = 0, got
%lu",
+        DeviceObject->ReferenceCount);
+    if (!Lower)
+    {
+        ok(DeviceObject->DriverObject == ThisDriverObject,
+            "Expected DriverObject member to match this DriverObject %p, got
%p",
+            ThisDriverObject, DeviceObject->DriverObject);
+    }
+    ok(DeviceObject->NextDevice == NULL, "Expected NextDevice to be NULL, got
%p", DeviceObject->NextDevice);
+
+    if (Lower)
+    {
+        ok(DeviceObject->AttachedDevice == MainDeviceObject,
+            "Expected AttachDevice to be %p, got %p", MainDeviceObject,
DeviceObject->AttachedDevice);
+    }
+    else
+    {
+        ok(DeviceObject->AttachedDevice == NULL, "Expected AttachDevice to be
NULL, got %p", DeviceObject->AttachedDevice);
+    }
+
+    ok(DeviceObject->Flags ==FILE_VIRTUAL_VOLUME,
+        "Expected Flags FILE_VIRTUAL_VOLUME, got %lu",
DeviceObject->Flags);
+    ok(DeviceObject->DeviceType == FILE_DEVICE_UNKNOWN,
+        "Expected DeviceType to match creation parameter FILE_DEVICE_UNKNWOWN, got
%lu",
+        DeviceObject->DeviceType);
+    ok(DeviceObject->ActiveThreadCount == 0, "Expected ActiveThreadCount = 0, got
%lu\n", DeviceObject->ActiveThreadCount);
+
+    /*Check the extended extension */
+    extdev = (PEXTENDED_DEVOBJ_EXTENSION)DeviceObject->DeviceObjectExtension;
+    ok(extdev->ExtensionFlags == DOE_UNLOAD_PENDING,
+        "Expected Extended ExtensionFlags to be DOE_UNLOAD_PENDING, got %lu",
extdev->ExtensionFlags);
+    ok (extdev->Type == 13, "Expected Type of 13, got %d",
extdev->Type);
+    ok (extdev->Size == 0, "Expected Size of 0, got %d", extdev->Size);
+    ok (extdev->DeviceObject == DeviceObject, "Expected DeviceOject to match
newly created device %p, got %p",
+        DeviceObject, extdev->DeviceObject);
+    if (Lower)
+    {
+        /* Skip this for now */
+        //ok(extdev->AttachedTo == MainDeviceObject, "Expected AttachTo to %p,
got %p", MainDeviceObject, extdev->AttachedTo);
+    }
+    else
+    {
+        ok(extdev->AttachedTo == NULL, "Expected AttachTo to be NULL, got
%p", extdev->AttachedTo);
+    }
+    ok(extdev->StartIoCount == 0, "Expected StartIoCount = 0, got %lu",
extdev->StartIoCount);
+    ok(extdev->StartIoKey == 0, "Expected StartIoKey = 0, got %lu",
extdev->StartIoKey);
+    ok(extdev->StartIoFlags == 0, "Expected StartIoFlags = 0, got %lu",
extdev->StartIoFlags);
+}
+
+VOID DeviceCreateDeleteTest(PDRIVER_OBJECT DriverObject)
+{
+    NTSTATUS Status;
+    UNICODE_STRING DeviceString;
+    UNICODE_STRING DosDeviceString;
+    PDEVICE_OBJECT DeviceObject;
+
+    /* Create using wrong directory */
+    RtlInitUnicodeString(&DeviceString, L"\\Device1\\Kmtest");
+    Status = IoCreateDevice(DriverObject,
+                            0,
+                            &DeviceString,
+                            FILE_DEVICE_UNKNOWN,
+                            0,
+                            FALSE,
+                            &DeviceObject);
+    ok(Status == STATUS_OBJECT_PATH_NOT_FOUND, "Expected
STATUS_OBJECT_PATH_NOT_FOUND, got 0x%lX", Status);
+
+    /* Create using correct params with exlusice access */
+    RtlInitUnicodeString(&DeviceString, L"\\Device\\Kmtest");
+    Status = IoCreateDevice(DriverObject,
+                            0,
+                            &DeviceString,
+                            FILE_DEVICE_UNKNOWN,
+                            0,
+                            TRUE,
+                            &DeviceObject);
+    ok(Status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got 0x%lX",
Status);
+
+    DeviceCreatedTest(DeviceObject, TRUE);
+
+    /* Delete the device */
+    if (NT_SUCCESS(Status))
+    {
+        IoDeleteDevice(DeviceObject);
+        ok(DriverObject->DeviceObject == 0, "Expected
DriverObject->DeviceObject to be NULL, got %p",
+            DriverObject->DeviceObject);
+    }
+
+    /* Create using correct params with exlusice access */
+    Status = IoCreateDevice(DriverObject,
+                            0,
+                            &DeviceString,
+                            FILE_DEVICE_UNKNOWN,
+                            0,
+                            FALSE,
+                            &DeviceObject);
+    ok(Status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got 0x%lX",
Status);
+
+    DeviceCreatedTest(DeviceObject, FALSE);
+
+    /* Delete the device */
+    if (NT_SUCCESS(Status))
+    {
+        IoDeleteDevice(DeviceObject);
+        ok(DriverObject->DeviceObject == 0, "Expected
DriverObject->DeviceObject to be NULL, got %p",
+            DriverObject->DeviceObject);
+    }
+
+    /* Recreate device */
+    Status = IoCreateDevice(DriverObject,
+                            0,
+                            &DeviceString,
+                            FILE_DEVICE_UNKNOWN,
+                            0,
+                            FALSE,
+                            &DeviceObject);
+    ok(Status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got 0x%lX",
Status);
+
+    RtlInitUnicodeString(&DosDeviceString, L"\\DosDevices\\kmtest");
+    Status = IoCreateSymbolicLink(&DosDeviceString, &DeviceString);
+
+    if (!NT_SUCCESS(Status))
+    {
+            /* Delete device object if not successful */
+            IoDeleteDevice(DeviceObject);
+            return;
+    }
+
+    MainDeviceObject = DeviceObject;
+
+    return;
+}
+
+BOOLEAN AttachDeviceTest(PDEVICE_OBJECT DeviceObject,  PWCHAR NewDriverRegPath)
+{
+    NTSTATUS Status;
+    UNICODE_STRING LowerDeviceName;
+
+    RtlInitUnicodeString(&LowerDeviceName, NewDriverRegPath);
+    Status = IoAttachDevice(DeviceObject, &LowerDeviceName,
&AttachDeviceObject);
+
+    /* TODO: Add more tests */
+
+    return TRUE;
+}
+
+BOOLEAN DetachDeviceTest(PDEVICE_OBJECT AttachedDevice)
+{
+
+    IoDetachDevice(AttachedDevice);
+
+    /* TODO: Add more tests */
+
+    return TRUE;
+}

Removed: branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoHelper_drv.rbuild
URL:
http://svn.reactos.org/svn/reactos/branches/GSoC_2011/KMTestSuite/kmtests/n…
==============================================================================
--- branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoHelper_drv.rbuild [iso-8859-1]
(original)
+++ branches/GSoC_2011/KMTestSuite/kmtests/ntos_io/IoHelper_drv.rbuild (removed)
@@ -1,14 +1,0 @@
-<module name="iohelper_drv" type="kernelmodedriver"
installbase="bin" installname="iohelper_drv.sys">
-	<include base="kmtest_drv">include</include>
-	<library>ntoskrnl</library>
-	<library>hal</library>
-	<library>pseh</library>
-	<library>kmtest_printf</library>
-	<define name="KMT_STANDALONE_DRIVER" />
-	<file>IoHelper_drv.c</file>
-	<directory name="..">
-		<directory name="kmtest_drv">
-			<file>kmtest_standalone.c</file>
-		</directory>
-	</directory>
-</module>

Modified: branches/GSoC_2011/KMTestSuite/kmtests/ntos_ob/ObType.c
URL:
http://svn.reactos.org/svn/reactos/branches/GSoC_2011/KMTestSuite/kmtests/n…
==============================================================================
--- branches/GSoC_2011/KMTestSuite/kmtests/ntos_ob/ObType.c [iso-8859-1] (original)
+++ branches/GSoC_2011/KMTestSuite/kmtests/ntos_ob/ObType.c [iso-8859-1] Wed Aug 17
11:50:46 2011
@@ -133,9 +133,8 @@
     return STATUS_OBJECT_NAME_NOT_FOUND;//STATUS_SUCCESS;
 }
 
-/* TODO: this does not actually return an NTSTATUS! */
-static
-NTSTATUS
+static
+BOOLEAN
 NTAPI
 OkayToCloseProc(
     IN PEPROCESS Process OPTIONAL,