ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

ntdll: Run exception tests from executable memory.

This commit is contained in:
Peter Beutner 2006-11-30 17:53:48 +01:00 committed by Alexandre Julliard
parent 94b548fd62
commit b7393f29a4
1 changed files with 22 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
dlls/ntdll/tests/exception.c
View File

@ -38,6 +38,7 @@
static struct _TEB * (WINAPI *pNtCurrentTeb)(void); static struct _TEB * (WINAPI *pNtCurrentTeb)(void);
static NTSTATUS (WINAPI *pNtGetContextThread)(HANDLE,CONTEXT*); static NTSTATUS (WINAPI *pNtGetContextThread)(HANDLE,CONTEXT*);
static void *code_mem;
/* Test various instruction combinations that cause a protection fault on the i386, /* Test various instruction combinations that cause a protection fault on the i386,
* and check what the resulting exception looks like. * and check what the resulting exception looks like.
@ -162,18 +163,21 @@ static const struct exception
static int got_exception; static int got_exception;
static void run_exception_test(const void *handler, const void* context, const void *code) static void run_exception_test(const void *handler, const void* context,
const void *code, unsigned int code_size)
{ {
struct { struct {
EXCEPTION_REGISTRATION_RECORD frame; EXCEPTION_REGISTRATION_RECORD frame;
const void *context; const void *context;
} exc_frame; } exc_frame;
void (*func)(void) = code; void (*func)(void) = code_mem;
exc_frame.frame.Handler = handler; exc_frame.frame.Handler = handler;
exc_frame.frame.Prev = pNtCurrentTeb()->Tib.ExceptionList; exc_frame.frame.Prev = pNtCurrentTeb()->Tib.ExceptionList;
exc_frame.context = context; exc_frame.context = context;
memcpy(code_mem, code, code_size);
pNtCurrentTeb()->Tib.ExceptionList = &exc_frame.frame; pNtCurrentTeb()->Tib.ExceptionList = &exc_frame.frame;
func(); func();
pNtCurrentTeb()->Tib.ExceptionList = exc_frame.frame.Prev; pNtCurrentTeb()->Tib.ExceptionList = exc_frame.frame.Prev;
@ -191,9 +195,9 @@ static DWORD handler( EXCEPTION_RECORD *rec, EXCEPTION_REGISTRATION_RECORD *fram
ok( rec->ExceptionCode == except->status, ok( rec->ExceptionCode == except->status,
"%u: Wrong exception code %x/%x\n", entry, rec->ExceptionCode, except->status ); "%u: Wrong exception code %x/%x\n", entry, rec->ExceptionCode, except->status );
ok( rec->ExceptionAddress == except->code + except->offset, ok( rec->ExceptionAddress == (char*)code_mem + except->offset,
"%u: Wrong exception address %p/%p\n", entry, "%u: Wrong exception address %p/%p\n", entry,
rec->ExceptionAddress, except->code + except->offset ); rec->ExceptionAddress, (char*)code_mem + except->offset );
ok( rec->NumberParameters == except->nb_params, ok( rec->NumberParameters == except->nb_params,
"%u: Wrong number of parameters %u/%u\n", entry, rec->NumberParameters, except->nb_params ); "%u: Wrong number of parameters %u/%u\n", entry, rec->NumberParameters, except->nb_params );
@ -203,7 +207,7 @@ static DWORD handler( EXCEPTION_RECORD *rec, EXCEPTION_REGISTRATION_RECORD *fram
entry, i, rec->ExceptionInformation[i], except->params[i] ); entry, i, rec->ExceptionInformation[i], except->params[i] );
/* don't handle exception if it's not the address we expected */ /* don't handle exception if it's not the address we expected */
if (rec->ExceptionAddress != except->code + except->offset) return ExceptionContinueSearch; if (rec->ExceptionAddress != (char*)code_mem + except->offset) return ExceptionContinueSearch;
context->Eip += except->length; context->Eip += except->length;
return ExceptionContinueExecution; return ExceptionContinueExecution;
@ -216,7 +220,8 @@ static void test_prot_fault(void)
for (i = 0; i < sizeof(exceptions)/sizeof(exceptions[0]); i++) for (i = 0; i < sizeof(exceptions)/sizeof(exceptions[0]); i++)
{ {
got_exception = 0; got_exception = 0;
run_exception_test(handler, &exceptions[i], &exceptions[i].code); run_exception_test(handler, &exceptions[i], &exceptions[i].code,
sizeof(exceptions[i].code));
if (!i && !got_exception) if (!i && !got_exception)
{ {
trace( "No exception, assuming win9x, no point in testing further\n" ); trace( "No exception, assuming win9x, no point in testing further\n" );
@ -309,7 +314,7 @@ static void test_exceptions(void)
} }
/* test handling of debug registers */ /* test handling of debug registers */
run_exception_test(dreg_handler, NULL, &segfault_code); run_exception_test(dreg_handler, NULL, &segfault_code, sizeof(segfault_code));
ctx.ContextFlags = CONTEXT_DEBUG_REGISTERS; ctx.ContextFlags = CONTEXT_DEBUG_REGISTERS;
res = pNtGetContextThread(GetCurrentThread(), &ctx); res = pNtGetContextThread(GetCurrentThread(), &ctx);
@ -319,12 +324,12 @@ static void test_exceptions(void)
/* test single stepping behavior */ /* test single stepping behavior */
got_exception = 0; got_exception = 0;
run_exception_test(single_step_handler, NULL, &single_stepcode); run_exception_test(single_step_handler, NULL, &single_stepcode, sizeof(single_stepcode));
ok(got_exception == 1, "expected 1 single step exceptions, got %d\n", got_exception); ok(got_exception == 1, "expected 1 single step exceptions, got %d\n", got_exception);
/* test alignment exceptions */ /* test alignment exceptions */
got_exception = 0; got_exception = 0;
run_exception_test(align_check_handler, NULL, align_check_code); run_exception_test(align_check_handler, NULL, align_check_code, sizeof(align_check_code));
ok(got_exception == 0, "got %d alignment faults, expected 0\n", got_exception); ok(got_exception == 0, "got %d alignment faults, expected 0\n", got_exception);
} }
@ -340,7 +345,15 @@ START_TEST(exception)
return; return;
} }
/* 1024 byte should be sufficient */
code_mem = VirtualAlloc(NULL, 1024, MEM_RESERVE | MEM_COMMIT, PAGE_EXECUTE_READWRITE);
if(!code_mem) {
trace("VirtualAlloc failed\n");
return;
}
test_prot_fault(); test_prot_fault();
test_exceptions(); test_exceptions();
VirtualFree(code_mem, 1024, MEM_RELEASE);
#endif #endif
} }