1b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar//===-- clear_cache_test.c - Test clear_cache -----------------------------===// 2b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar// 3b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar// The LLVM Compiler Infrastructure 4b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar// 59ad441ffec97db647fee3725b3424284fb913e14Howard Hinnant// This file is dual licensed under the MIT and the University of Illinois Open 69ad441ffec97db647fee3725b3424284fb913e14Howard Hinnant// Source Licenses. See LICENSE.TXT for details. 7b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar// 8b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar//===----------------------------------------------------------------------===// 9b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar 10b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar 11b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar#include <stdio.h> 12b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar#include <string.h> 13b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar#include <stdint.h> 147e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov#if defined(_WIN32) 157e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov#include <windows.h> 167e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikovvoid __clear_cache(void* start, void* end) 177e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov{ 187e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov if (!FlushInstructionCache(GetCurrentProcess(), start, end-start)) 197e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov exit(1); 207e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov} 217e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov#else 22b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar#include <sys/mman.h> 237e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikovextern void __clear_cache(void* start, void* end); 247e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov#endif 25b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar 26b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar 27b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar 28b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar 29b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbartypedef int (*pfunc)(void); 30b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar 31b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbarint func1() 32b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar{ 33b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar return 1; 34b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar} 35b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar 36b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbarint func2() 37b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar{ 38b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar return 2; 39b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar} 40b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar 4186277eb844c4983c81de62d7c050e92fe7155788Stephen Hinesvoid *__attribute__((noinline)) 4286277eb844c4983c81de62d7c050e92fe7155788Stephen Hinesmemcpy_f(void *dst, const void *src, size_t n) { 4386277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// ARM and MIPS nartually align functions, but use the LSB for ISA selection 4486277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// (THUMB, MIPS16/uMIPS respectively). Ensure that the ISA bit is ignored in 4586277eb844c4983c81de62d7c050e92fe7155788Stephen Hines// the memcpy 4686277eb844c4983c81de62d7c050e92fe7155788Stephen Hines#if defined(__arm__) || defined(__mips__) 4786277eb844c4983c81de62d7c050e92fe7155788Stephen Hines return (void *)((uintptr_t)memcpy(dst, (void *)((uintptr_t)src & ~1), n) | 4886277eb844c4983c81de62d7c050e92fe7155788Stephen Hines ((uintptr_t)src & 1)); 4986277eb844c4983c81de62d7c050e92fe7155788Stephen Hines#else 5086277eb844c4983c81de62d7c050e92fe7155788Stephen Hines return memcpy(dst, (void *)((uintptr_t)src), n); 5186277eb844c4983c81de62d7c050e92fe7155788Stephen Hines#endif 5286277eb844c4983c81de62d7c050e92fe7155788Stephen Hines} 53b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar 54b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbarunsigned char execution_buffer[128]; 55b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar 56b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbarint main() 57b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar{ 58b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar // make executable the page containing execution_buffer 59b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar char* start = (char*)((uintptr_t)execution_buffer & (-4095)); 60b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar char* end = (char*)((uintptr_t)(&execution_buffer[128+4096]) & (-4095)); 617e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov#if defined(_WIN32) 627e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov DWORD dummy_oldProt; 637e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov MEMORY_BASIC_INFORMATION b; 647e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov if (!VirtualQuery(start, &b, sizeof(b))) 657e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov return 1; 667e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov if (!VirtualProtect(b.BaseAddress, b.RegionSize, PAGE_EXECUTE_READWRITE, &b.Protect)) 677e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov#else 687e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov if (mprotect(start, end-start, PROT_READ|PROT_WRITE|PROT_EXEC) != 0) 697e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov#endif 70b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar return 1; 71b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar 72b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar // verify you can copy and execute a function 7386277eb844c4983c81de62d7c050e92fe7155788Stephen Hines pfunc f1 = (pfunc)memcpy_f(execution_buffer, func1, 128); 74b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar __clear_cache(execution_buffer, &execution_buffer[128]); 757e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov if ((*f1)() != 1) 76b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar return 1; 77b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar 78b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar // verify you can overwrite a function with another 7986277eb844c4983c81de62d7c050e92fe7155788Stephen Hines pfunc f2 = (pfunc)memcpy_f(execution_buffer, func2, 128); 80b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar __clear_cache(execution_buffer, &execution_buffer[128]); 817e8904f4b2b6b084ea2b4203d62554b42c068296Anton Korobeynikov if ((*f2)() != 2) 82b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar return 1; 83b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar 84b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar return 0; 85b3a6901e66f55b35aa9e01bcb24134e6a65ea004Daniel Dunbar} 86