[USP10_WINETEST] Sync with Wine Staging 2.9. CORE-13362

[reactos.git] / rostests / kmtests / rtl / RtlException.c

diff --git a/rostests/kmtests/rtl/RtlException.c b/rostests/kmtests/rtl/RtlException.c
index 4dfb5b9..a8b76a3 100644 (file)
--- a/rostests/kmtests/rtl/RtlException.c
+++ b/rostests/kmtests/rtl/RtlException.c
@@ -2,16 +2,113 @@
  * PROJECT:         ReactOS kernel-mode tests
  * LICENSE:         GPLv2+ - See COPYING in the top level directory
  * PURPOSE:         Kernel-Mode Test Suite Exception test
- * PROGRAMMER:      Thomas Faber <thfabba@gmx.de>
+ * PROGRAMMER:      Thomas Faber <thomas.faber@reactos.org>
  */
 
+#define KMT_EMULATE_KERNEL
 #include <kmt_test.h>
 
+static
+VOID
+PossiblyRaise(
+    _In_ BOOLEAN Raise)
+{
+    if (Raise)
+    {
+        ExRaiseStatus(STATUS_ASSERTION_FAILURE);
+    }
+}
+
+static
+VOID
+InnerFunction(
+    _Inout_ PULONG State,
+    _In_ BOOLEAN Raise)
+{
+    _SEH2_VOLATILE INT Var = 123;
+    static _SEH2_VOLATILE INT *AddressOfVar;
+
+    AddressOfVar = &Var;
+    ok_eq_ulong(*State, 1);
+    _SEH2_TRY
+    {
+        *State = 2;
+        PossiblyRaise(Raise);
+        ok_eq_ulong(*State, 2);
+        *State = 3;
+    }
+    _SEH2_FINALLY
+    {
+        ok_eq_int(Var, 123);
+        ok_eq_pointer(&Var, AddressOfVar);
+        if (Raise)
+            ok_eq_ulong(*State, 2);
+        else
+            ok_eq_ulong(*State, 3);
+        *State = 4;
+    }
+    _SEH2_END;
+
+    ok_eq_int(Var, 123);
+    ok_eq_pointer(&Var, AddressOfVar);
+    ok_eq_ulong(*State, 4);
+    *State = 5;
+}
+
+static
+VOID
+OuterFunction(
+    _Inout_ PULONG State,
+    _In_ BOOLEAN Raise)
+{
+    _SEH2_VOLATILE INT Var = 456;
+    static _SEH2_VOLATILE INT *AddressOfVar;
+
+    AddressOfVar = &Var;
+    ok_eq_ulong(*State, 0);
+    _SEH2_TRY
+    {
+        *State = 1;
+        InnerFunction(State, Raise);
+        ok_eq_ulong(*State, 5);
+        *State = 6;
+    }
+    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
+    {
+        ok_eq_int(Var, 456);
+        ok_eq_pointer(&Var, AddressOfVar);
+        ok_eq_ulong(*State, 4);
+        *State = 7;
+    }
+    _SEH2_END;
+
+    ok_eq_int(Var, 456);
+    ok_eq_pointer(&Var, AddressOfVar);
+    if (Raise)
+        ok_eq_ulong(*State, 7);
+    else
+        ok_eq_ulong(*State, 6);
+    *State = 8;
+}
+
+static
+VOID
+TestNestedExceptionHandler(VOID)
+{
+    ULONG State;
+
+    State = 0;
+    OuterFunction(&State, FALSE);
+    ok_eq_ulong(State, 8);
+
+    State = 0;
+    OuterFunction(&State, TRUE);
+    ok_eq_ulong(State, 8);
+}
+
 START_TEST(RtlException)
 {
-    NTSTATUS ExceptionStatus;
     PCHAR Buffer[128];
-    CHAR Value;
 
     /* Access a valid pointer - must not trigger SEH */
     KmtStartSeh()
@@ -20,7 +117,7 @@ START_TEST(RtlException)
 
     /* Read from a NULL pointer - must cause an access violation */
     KmtStartSeh()
-        Value = *(volatile CHAR *)NULL;
+        (void)*(volatile CHAR *)NULL;
     KmtEndSeh(STATUS_ACCESS_VIOLATION);
 
     /* Write to a NULL pointer - must cause an access violation */
@@ -32,7 +129,7 @@ START_TEST(RtlException)
 #if 0 //def KMT_KERNEL_MODE
     /* Read from MmBadPointer - must cause an access violation */
     KmtStartSeh()
-        Value = *(volatile CHAR *)MmBadPointer;
+        (void)*(volatile CHAR *)MmBadPointer;
     KmtEndSeh(STATUS_ACCESS_VIOLATION);
 
     /* Write to MmBadPointer - must cause an access violation */
@@ -41,13 +138,31 @@ START_TEST(RtlException)
     KmtEndSeh(STATUS_ACCESS_VIOLATION);
 #endif
 
+    KmtStartSeh()
+        ExRaiseStatus(STATUS_ACCESS_VIOLATION);
+    KmtEndSeh(STATUS_ACCESS_VIOLATION);
+
+    KmtStartSeh()
+        ExRaiseStatus(STATUS_TIMEOUT);
+    KmtEndSeh(STATUS_TIMEOUT);
+
+    KmtStartSeh()
+        ExRaiseStatus(STATUS_STACK_OVERFLOW);
+    KmtEndSeh(STATUS_STACK_OVERFLOW);
+
+    KmtStartSeh()
+        ExRaiseStatus(STATUS_GUARD_PAGE_VIOLATION);
+    KmtEndSeh(STATUS_GUARD_PAGE_VIOLATION);
+
+    TestNestedExceptionHandler();
+
     /* We cannot test this in kernel mode easily - the stack is just "somewhere"
      * in system space, and there's no guard page below it */
 #if CORE_6640_IS_FIXED
 #ifdef KMT_USER_MODE
     /* Overflow the stack - must cause a special exception */
     KmtStartSeh()
-        PCHAR Pointer;
+        volatile CHAR *Pointer;
 
         while (1)
         {