From 097a8b873300c09c6373e5c881ab0626a6d21ea6 Mon Sep 17 00:00:00 2001
From: Thomas Faber <thomas.faber@reactos.org>
Date: Wed, 24 Aug 2011 14:09:53 +0000
Subject: [PATCH] [KMTESTS/IO] - Add IoInterrupt test with a simple test for
 KeSynchronizeExecution

svn path=/branches/GSoC_2011/KMTestSuite/; revision=53419
---
 kmtests/CMakeLists.txt        |  1 +
 kmtests/kmtest_drv.rbuild     |  1 +
 kmtests/kmtest_drv/testlist.c |  2 +
 kmtests/ntos_io/IoInterrupt.c | 95 +++++++++++++++++++++++++++++++++++
 4 files changed, 99 insertions(+)
 create mode 100644 kmtests/ntos_io/IoInterrupt.c

diff --git a/kmtests/CMakeLists.txt b/kmtests/CMakeLists.txt
index ca5c0d1feda..864d202e835 100644
--- a/kmtests/CMakeLists.txt
+++ b/kmtests/CMakeLists.txt
@@ -27,6 +27,7 @@ list(APPEND KMTEST_DRV_SOURCE
     ntos_ex/ExTimer.c
     ntos_fsrtl/FsRtlExpression.c
     ntos_io/IoDeviceInterface.c
+    ntos_io/IoInterrupt.c
     ntos_io/IoIrp.c
     ntos_io/IoMdl.c
     ntos_ke/KeApc.c
diff --git a/kmtests/kmtest_drv.rbuild b/kmtests/kmtest_drv.rbuild
index 22268078ca5..f77b3cdb956 100644
--- a/kmtests/kmtest_drv.rbuild
+++ b/kmtests/kmtest_drv.rbuild
@@ -30,6 +30,7 @@
 	</directory>
 	<directory name="ntos_io">
 		<file>IoDeviceInterface.c</file>
+		<file>IoInterrupt.c</file>
 		<file>IoIrp.c</file>
 		<file>IoMdl.c</file>
 	</directory>
diff --git a/kmtests/kmtest_drv/testlist.c b/kmtests/kmtest_drv/testlist.c
index 6562c0e1042..c57cad9ada6 100644
--- a/kmtests/kmtest_drv/testlist.c
+++ b/kmtests/kmtest_drv/testlist.c
@@ -20,6 +20,7 @@ KMT_TESTFUNC Test_ExSingleList;
 KMT_TESTFUNC Test_ExTimer;
 KMT_TESTFUNC Test_FsRtlExpression;
 KMT_TESTFUNC Test_IoDeviceInterface;
+KMT_TESTFUNC Test_IoInterrupt;
 KMT_TESTFUNC Test_IoIrp;
 KMT_TESTFUNC Test_IoMdl;
 KMT_TESTFUNC Test_KeApc;
@@ -51,6 +52,7 @@ const KMT_TEST TestList[] =
     { "Example",                            Test_Example },
     { "FsRtlExpression",                    Test_FsRtlExpression },
     { "IoDeviceInterface",                  Test_IoDeviceInterface },
+    { "IoInterrupt",                        Test_IoInterrupt },
     { "IoIrp",                              Test_IoIrp },
     { "IoMdl",                              Test_IoMdl },
     { "KeApc",                              Test_KeApc },
diff --git a/kmtests/ntos_io/IoInterrupt.c b/kmtests/ntos_io/IoInterrupt.c
new file mode 100644
index 00000000000..0f4d8de9027
--- /dev/null
+++ b/kmtests/ntos_io/IoInterrupt.c
@@ -0,0 +1,95 @@
+/*
+ * PROJECT:         ReactOS kernel-mode tests
+ * LICENSE:         GPLv2+ - See COPYING in the top level directory
+ * PURPOSE:         Kernel-Mode Test Suite Interrupt test
+ * PROGRAMMER:      Thomas Faber <thfabba@gmx.de>
+ */
+
+#include <kmt_test.h>
+
+#define NDEBUG
+#include <debug.h>
+
+#define CheckSpinLock(Lock, Locked) do                  \
+{                                                       \
+    if (KmtIsMultiProcessorBuild)                       \
+        ok_eq_ulongptr(*(Lock), (Locked) != 0);         \
+    else                                                \
+        ok_eq_ulongptr(*(Lock), 0);                     \
+} while (0)
+
+typedef struct
+{
+    BOOLEAN ReturnValue;
+    KIRQL ExpectedIrql;
+    PKINTERRUPT Interrupt;
+} TEST_CONTEXT, *PTEST_CONTEXT;
+
+static KSYNCHRONIZE_ROUTINE SynchronizeRoutine;
+
+static
+BOOLEAN
+NTAPI
+SynchronizeRoutine(
+    IN PVOID Context)
+{
+    PTEST_CONTEXT TestContext = Context;
+
+    ok_irql(TestContext->ExpectedIrql);
+
+    CheckSpinLock(TestContext->Interrupt->ActualLock, TRUE);
+
+    return TestContext->ReturnValue;
+}
+
+static
+VOID
+TestSynchronizeExecution(VOID)
+{
+    KINTERRUPT Interrupt;
+    TEST_CONTEXT TestContext;
+    KIRQL SynchIrql;
+    KIRQL OriginalIrql;
+    KIRQL Irql;
+    KSPIN_LOCK ActualLock;
+    BOOLEAN Ret;
+
+    RtlFillMemory(&Interrupt, sizeof Interrupt, 0x55);
+    Interrupt.ActualLock = &ActualLock;
+    KeInitializeSpinLock(Interrupt.ActualLock);
+    CheckSpinLock(Interrupt.ActualLock, FALSE);
+
+    TestContext.Interrupt = &Interrupt;
+    TestContext.ReturnValue = TRUE;
+
+    for (TestContext.ReturnValue = 0; TestContext.ReturnValue <= 2; ++TestContext.ReturnValue)
+    {
+        for (OriginalIrql = PASSIVE_LEVEL; OriginalIrql <= HIGH_LEVEL; ++OriginalIrql)
+        {
+            /* TODO: don't hardcode this :| */
+            if (OriginalIrql == 3 || (OriginalIrql >= 11 && OriginalIrql <= 26) || OriginalIrql == 30)
+                continue;
+            KeRaiseIrql(OriginalIrql, &Irql);
+            for (SynchIrql = max(DISPATCH_LEVEL, OriginalIrql); SynchIrql <= HIGH_LEVEL; ++SynchIrql)
+            {
+                if (SynchIrql == 3 || (SynchIrql >= 11 && SynchIrql <= 26) || SynchIrql == 30)
+                    continue;
+                Interrupt.SynchronizeIrql = SynchIrql;
+                ok_irql(OriginalIrql);
+                CheckSpinLock(Interrupt.ActualLock, FALSE);
+                TestContext.ExpectedIrql = SynchIrql;
+                Ret = KeSynchronizeExecution(&Interrupt, SynchronizeRoutine, &TestContext);
+                ok_eq_int(Ret, TestContext.ReturnValue);
+                ok_irql(OriginalIrql);
+                CheckSpinLock(Interrupt.ActualLock, FALSE);
+                /* TODO: Check that all other fields of the interrupt are untouched */
+            }
+            KeLowerIrql(Irql);
+        }
+    }
+}
+
+START_TEST(IoInterrupt)
+{
+    TestSynchronizeExecution();
+}
-- 
2.17.1