- Add unloading ability.

- Don't forget to wrap any testing module into StartTest / FinishTest.

svn path=/trunk/; revision=34951

rostests/drivers/kmtest/kmtest.c

@@ -108,6 +108,16 @@ VOID NtoskrnlObTest();
 VOID NtoskrnlExecutiveTests();
 VOID NtoskrnlPoolsTest();
 
+/*
+ * KmtestUnload
+ */
+VOID
+NTAPI
+KmtestUnload(IN PDRIVER_OBJECT DriverObject)
+{
+    /* Nothing to do here */
+}
+
 /*
  * DriverEntry
  */
@@ -117,6 +127,11 @@ DriverEntry(PDRIVER_OBJECT DriverObject,
             PUNICODE_STRING RegistryPath)
 {
     DbgPrint("\n===============================================\nKernel Mode Regression Test driver starting...\n");
+
+    /* Set necessary routines */
+    DriverObject->DriverUnload = KmtestUnload;
+
+
     //NtoskrnlExecutiveTests();
     //NtoskrnlIoDeviceInterface();
     NtoskrnlIoTests();

rostests/drivers/kmtest/ntos_io.c

@@ -89,6 +89,8 @@ VOID NtoskrnlIoIrpTest()
     USHORT size;
     IRP *iorp;
 
+    StartTest();
+
     // 1st test
     size = sizeof(IRP) + 5 * sizeof(IO_STACK_LOCATION);
     iorp = ExAllocatePool(NonPagedPool, size);
@@ -163,6 +165,8 @@ VOID NtoskrnlIoIrpTest()
 
         IoFreeIrp(iorp);
     }
+
+    FinishTest("NTOSKRNL Io Irp");
 }
 
 VOID NtoskrnlIoTests()