[KMTESTS:RTL]

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

diff --git a/rostests/kmtests/rtl/RtlException.c b/rostests/kmtests/rtl/RtlException.c
index 1b38098..d5cae4c 100644 (file)
--- a/rostests/kmtests/rtl/RtlException.c
+++ b/rostests/kmtests/rtl/RtlException.c
@@ -2,60 +2,74 @@
  * 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>
 
-#define StartSeh()                  ExceptionStatus = STATUS_SUCCESS; _SEH2_TRY {
-#define EndSeh(ExpectedStatus) }    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) { ExceptionStatus = _SEH2_GetExceptionCode(); } _SEH2_END; ok_eq_hex(ExceptionStatus, ExpectedStatus)
-
 START_TEST(RtlException)
 {
-    NTSTATUS ExceptionStatus;
     PCHAR Buffer[128];
-    CHAR Value;
 
     /* Access a valid pointer - must not trigger SEH */
-    StartSeh()
+    KmtStartSeh()
         RtlFillMemory(Buffer, sizeof(Buffer), 0x12);
-    EndSeh(STATUS_SUCCESS);
+    KmtEndSeh(STATUS_SUCCESS);
 
     /* Read from a NULL pointer - must cause an access violation */
-    StartSeh()
-        Value = *(volatile CHAR *)NULL;
-    EndSeh(STATUS_ACCESS_VIOLATION);
+    KmtStartSeh()
+        (void)*(volatile CHAR *)NULL;
+    KmtEndSeh(STATUS_ACCESS_VIOLATION);
 
     /* Write to a NULL pointer - must cause an access violation */
-    StartSeh()
+    KmtStartSeh()
         *(volatile CHAR *)NULL = 5;
-    EndSeh(STATUS_ACCESS_VIOLATION);
+    KmtEndSeh(STATUS_ACCESS_VIOLATION);
 
     /* TODO: Find where MmBadPointer is defined - gives an unresolved external */
 #if 0 //def KMT_KERNEL_MODE
     /* Read from MmBadPointer - must cause an access violation */
-    StartSeh()
-        Value = *(volatile CHAR *)MmBadPointer;
-    EndSeh(STATUS_ACCESS_VIOLATION);
+    KmtStartSeh()
+        (void)*(volatile CHAR *)MmBadPointer;
+    KmtEndSeh(STATUS_ACCESS_VIOLATION);
 
     /* Write to MmBadPointer - must cause an access violation */
-    StartSeh()
+    KmtStartSeh()
         *(volatile CHAR *)MmBadPointer = 5;
-    EndSeh(STATUS_ACCESS_VIOLATION);
+    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);
+
     /* 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 */
-    StartSeh()
-        PCHAR Pointer;
+    KmtStartSeh()
+        volatile CHAR *Pointer;
 
         while (1)
         {
             Pointer = _alloca(1024);
             *Pointer = 5;
         }
-    EndSeh(STATUS_STACK_OVERFLOW);
+    KmtEndSeh(STATUS_STACK_OVERFLOW);
 #endif
+#endif /* CORE_6640_IS_FIXED */
 }