[ros-diffs] [fireball] 34951: - Add unloading ability. - Don't forget to wrap any testing module into StartTest / FinishTest.

Author: fireball
Date: Wed Jul 30 06:11:25 2008
New Revision: 34951

URL: http://svn.reactos.org/svn/reactos?rev=34951&view=rev
Log:
- Add unloading ability.
- Don't forget to wrap any testing module into StartTest / FinishTest.

Modified:
    trunk/rostests/drivers/kmtest/kmtest.c
    trunk/rostests/drivers/kmtest/ntos_io.c

Modified: trunk/rostests/drivers/kmtest/kmtest.c
URL: http://svn.reactos.org/svn/reactos/trunk/rostests/drivers/kmtest/kmtest.c?re...
==============================================================================
--- trunk/rostests/drivers/kmtest/kmtest.c [iso-8859-1] (original)
+++ trunk/rostests/drivers/kmtest/kmtest.c [iso-8859-1] Wed Jul 30 06:11:25 2008
@@ -109,6 +109,16 @@
 VOID NtoskrnlPoolsTest();

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

Modified: trunk/rostests/drivers/kmtest/ntos_io.c
URL: http://svn.reactos.org/svn/reactos/trunk/rostests/drivers/kmtest/ntos_io.c?r...
==============================================================================
--- trunk/rostests/drivers/kmtest/ntos_io.c [iso-8859-1] (original)
+++ trunk/rostests/drivers/kmtest/ntos_io.c [iso-8859-1] Wed Jul 30 06:11:25 2008
@@ -89,6 +89,8 @@
     USHORT size;
     IRP *iorp;

+    StartTest();
+
     // 1st test
     size = sizeof(IRP) + 5 * sizeof(IO_STACK_LOCATION);
     iorp = ExAllocatePool(NonPagedPool, size);
@@ -163,6 +165,8 @@

IoFreeIrp(iorp);
     }
+
+    FinishTest("NTOSKRNL Io Irp");
 }

VOID NtoskrnlIoTests()