From 3a6d53e5d09da1d09211985947344b426860cdf3 Mon Sep 17 00:00:00 2001
From: Marcus Meissner <marcus@jet.franken.de>
Date: Sat, 30 Sep 2006 15:09:55 +0200
Subject: [PATCH] ntdll: Added debug registers test case.

---
 dlls/ntdll/tests/exception.c | 55 ++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)

diff --git a/dlls/ntdll/tests/exception.c b/dlls/ntdll/tests/exception.c
index 64e683bd5f6..1d00816a0a7 100644
--- a/dlls/ntdll/tests/exception.c
+++ b/dlls/ntdll/tests/exception.c
@@ -37,6 +37,7 @@
 #ifdef __i386__
 
 static struct _TEB * (WINAPI *pNtCurrentTeb)(void);
+static NTSTATUS  (WINAPI *pNtGetContextThread)(HANDLE,CONTEXT*);
 
 /* Test various instruction combinations that cause a protection fault on the i386,
  * and check what the resulting exception looks like.
@@ -227,11 +228,65 @@ static void test_prot_fault(void)
     pNtCurrentTeb()->Tib.ExceptionList = exc_frame.frame.Prev;
 }
 
+static DWORD dreg_handler( EXCEPTION_RECORD *rec, EXCEPTION_REGISTRATION_RECORD *frame,
+                      CONTEXT *context, EXCEPTION_REGISTRATION_RECORD **dispatcher )
+{
+    context->Eip += 2;	/* Skips the popl (%eax) */
+    context->Dr0 = 0;
+    context->Dr1 = 0;
+    context->Dr2 = 0;
+    context->Dr3 = 0;
+    context->Dr6 = 0;
+    context->Dr7 = 0x155;
+    return ExceptionContinueExecution;
+}
+
+static BYTE code[5] = {
+	0x31, 0xc0, /* xor    %eax,%eax */
+	0x8f, 0x00, /* popl   (%eax) - cause exception */
+        0xc3        /* ret */
+};
+
+static void test_debug_regs(void)
+{
+    CONTEXT ctx;
+    NTSTATUS res;
+    struct
+    {
+        EXCEPTION_REGISTRATION_RECORD frame;
+    } exc_frame;
+    void (*func)(void) = (void*)code;
+
+    pNtCurrentTeb = (void *)GetProcAddress( GetModuleHandleA("ntdll.dll"), "NtCurrentTeb" );
+    if (!pNtCurrentTeb)
+    {
+        trace( "NtCurrentTeb not found, skipping tests\n" );
+        return;
+    }
+    pNtGetContextThread = (void *)GetProcAddress( GetModuleHandleA("ntdll.dll"), "NtGetContextThread" );
+    if (!pNtGetContextThread)
+    {
+        trace( "NtGetContextThread not found, skipping tests\n" );
+        return;
+    }
+
+    exc_frame.frame.Handler = dreg_handler;
+    exc_frame.frame.Prev = pNtCurrentTeb()->Tib.ExceptionList;
+    pNtCurrentTeb()->Tib.ExceptionList = &exc_frame.frame;
+    func();
+    pNtCurrentTeb()->Tib.ExceptionList = exc_frame.frame.Prev;
+    ctx.ContextFlags = CONTEXT_DEBUG_REGISTERS;
+    res = pNtGetContextThread(GetCurrentThread(), &ctx);
+    ok (res == STATUS_SUCCESS,"NtGetContextThread failed with %lx", res);
+    ok(ctx.Dr7 == 0x155,"failed to set debugregister 7 to 0x155");
+}
+
 #endif  /* __i386__ */
 
 START_TEST(exception)
 {
 #ifdef __i386__
     test_prot_fault();
+    test_debug_regs();
 #endif
 }