From 653bb6745595746d04ad4acacaaab66caaaeb69b Mon Sep 17 00:00:00 2001 From: ed Date: Thu, 21 Oct 2010 19:02:02 +0000 Subject: Import compiler-rt r117047. --- test/Unit/enable_execute_stack_test.c | 57 +++++++++++++++++++++++++++++++++++ 1 file changed, 57 insertions(+) create mode 100644 test/Unit/enable_execute_stack_test.c (limited to 'test/Unit/enable_execute_stack_test.c') diff --git a/test/Unit/enable_execute_stack_test.c b/test/Unit/enable_execute_stack_test.c new file mode 100644 index 0000000..a0e97bb --- /dev/null +++ b/test/Unit/enable_execute_stack_test.c @@ -0,0 +1,57 @@ +//===-- enable_execute_stack_test.c - Test __enable_execute_stack ----------===// +// +// The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// + + +#include +#include +#include +#include + + + +extern void __clear_cache(void* start, void* end); +extern void __enable_execute_stack(void* addr); + +typedef int (*pfunc)(void); + +int func1() +{ + return 1; +} + +int func2() +{ + return 2; +} + + + + +int main() +{ + unsigned char execution_buffer[128]; + // mark stack page containing execution_buffer to be executable + __enable_execute_stack(execution_buffer); + + // verify you can copy and execute a function + memcpy(execution_buffer, (void *)(uintptr_t)&func1, 128); + __clear_cache(execution_buffer, &execution_buffer[128]); + pfunc f1 = (pfunc)(uintptr_t)execution_buffer; + if ( (*f1)() != 1 ) + return 1; + + // verify you can overwrite a function with another + memcpy(execution_buffer, (void *)(uintptr_t)&func2, 128); + __clear_cache(execution_buffer, &execution_buffer[128]); + pfunc f2 = (pfunc)(uintptr_t)execution_buffer; + if ( (*f2)() != 2 ) + return 1; + + return 0; +} -- cgit v1.1