maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[ASan] cleanup interceptors code - prefer ASAN_INTERCEPT_FUNCTION_NAME macro to _WIN32, __APPLE__ etc. 

llvm-svn: 161109
This commit is contained in:
Alexey Samsonov 2012-08-01 11:17:00 +00:00
parent fb78083b1c
commit cc7ea4269a
1 changed files with 95 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

147
compiler-rt/lib/asan/asan_interceptors.cc
View File

@ -27,8 +27,24 @@
// intercepted on a given platform. // intercepted on a given platform.
#if !defined(_WIN32) #if !defined(_WIN32)
# define ASAN_INTERCEPT_ATOLL_AND_STRTOLL 1 # define ASAN_INTERCEPT_ATOLL_AND_STRTOLL 1
# define ASAN_INTERCEPT__LONGJMP 1
# define ASAN_INTERCEPT_STRDUP 1
# define ASAN_INTERCEPT_STRCASECMP_AND_STRNCASECMP 1
# define ASAN_INTERCEPT_INDEX 1
# define ASAN_INTERCEPT_PTHREAD_CREATE 1
#else #else
# define ASAN_INTERCEPT_ATOLL_AND_STRTOLL 0 # define ASAN_INTERCEPT_ATOLL_AND_STRTOLL 0
# define ASAN_INTERCEPT__LONGJMP 0
# define ASAN_INTERCEPT_STRDUP 0
# define ASAN_INTERCEPT_STRCASECMP_AND_STRNCASECMP 0
# define ASAN_INTERCEPT_INDEX 0
# define ASAN_INTERCEPT_PTHREAD_CREATE 0
#endif
#if defined(__linux__)
# define ASAN_USE_ALIAS_ATTRIBUTE_FOR_INDEX 1
#else
# define ASAN_USE_ALIAS_ATTRIBUTE_FOR_INDEX 0
#endif #endif
#if !defined(__APPLE__) #if !defined(__APPLE__)
@ -37,10 +53,24 @@
# define ASAN_INTERCEPT_STRNLEN 0 # define ASAN_INTERCEPT_STRNLEN 0
#endif #endif
#if defined(ANDROID) || defined(_WIN32) #if !defined(ANDROID) && !defined(_WIN32)
# define ASAN_INTERCEPT_SIGNAL_AND_SIGACTION 0
#else
# define ASAN_INTERCEPT_SIGNAL_AND_SIGACTION 1 # define ASAN_INTERCEPT_SIGNAL_AND_SIGACTION 1
#else
# define ASAN_INTERCEPT_SIGNAL_AND_SIGACTION 0
#endif
// On Darwin siglongjmp tailcalls longjmp, so we don't want to intercept it
// there.
#if !defined(_WIN32) && !defined(__APPLE__)
# define ASAN_INTERCEPT_SIGLONGJMP 1
#else
# define ASAN_INTERCEPT_SIGLONGJMP 0
#endif
#if ASAN_HAS_EXCEPTIONS && !defined(_WIN32)
# define ASAN_INTERCEPT___CXA_THROW 1
#else
# define ASAN_INTERCEPT___CXA_THROW 0
#endif #endif
// Use extern declarations of intercepted functions on Mac and Windows // Use extern declarations of intercepted functions on Mac and Windows
@ -57,9 +87,12 @@ void *signal(int signum, void *handler);
// setjmp.h // setjmp.h
void longjmp(void* env, int value); void longjmp(void* env, int value);
# if !defined(_WIN32) # if ASAN_INTERCEPT__LONGJMP
void _longjmp(void *env, int value); void _longjmp(void *env, int value);
# endif # endif
# if ASAN_INTERCEPT___CXA_THROW
void __cxa_throw(void *a, void *b, void *c);
#endif
// string.h / strings.h // string.h / strings.h
int memcmp(const void *a1, const void *a2, uptr size); int memcmp(const void *a1, const void *a2, uptr size);
@ -67,24 +100,26 @@ void* memmove(void *to, const void *from, uptr size);
void* memcpy(void *to, const void *from, uptr size); void* memcpy(void *to, const void *from, uptr size);
void* memset(void *block, int c, uptr size); void* memset(void *block, int c, uptr size);
char* strchr(const char *str, int c); char* strchr(const char *str, int c);
# if defined(__APPLE__)
char* index(const char *string, int c);
# endif
char* strcat(char *to, const char* from); // NOLINT char* strcat(char *to, const char* from); // NOLINT
char *strncat(char *to, const char* from, uptr size); char *strncat(char *to, const char* from, uptr size);
char* strcpy(char *to, const char* from); // NOLINT char* strcpy(char *to, const char* from); // NOLINT
char* strncpy(char *to, const char* from, uptr size); char* strncpy(char *to, const char* from, uptr size);
int strcmp(const char *s1, const char* s2); int strcmp(const char *s1, const char* s2);
int strncmp(const char *s1, const char* s2, uptr size); int strncmp(const char *s1, const char* s2, uptr size);
# if !defined(_WIN32) uptr strlen(const char *s);
# if ASAN_INTERCEPT_STRCASECMP_AND_STRNCASECMP
int strcasecmp(const char *s1, const char *s2); int strcasecmp(const char *s1, const char *s2);
int strncasecmp(const char *s1, const char *s2, uptr n); int strncasecmp(const char *s1, const char *s2, uptr n);
# endif
# if ASAN_INTERCEPT_STRDUP
char* strdup(const char *s); char* strdup(const char *s);
# endif # endif
uptr strlen(const char *s);
# if ASAN_INTERCEPT_STRNLEN # if ASAN_INTERCEPT_STRNLEN
uptr strnlen(const char *s, uptr maxlen); uptr strnlen(const char *s, uptr maxlen);
# endif # endif
# if ASAN_INTERCEPT_INDEX
char* index(const char *string, int c);
# endif
// stdlib.h // stdlib.h
int atoi(const char *nptr); int atoi(const char *nptr);
@ -101,9 +136,8 @@ __declspec(dllimport)
void* __stdcall CreateThread(void *sec, uptr st, void* start, void* __stdcall CreateThread(void *sec, uptr st, void* start,
void *arg, DWORD fl, DWORD *id); void *arg, DWORD fl, DWORD *id);
# endif # endif
// Posix threads. // Posix threads.
# if !defined(_WIN32) # if ASAN_INTERCEPT_PTHREAD_CREATE
int pthread_create(void *thread, void *attr, void *(*start_routine)(void*), int pthread_create(void *thread, void *attr, void *(*start_routine)(void*),
void *arg); void *arg);
# endif # endif
@ -191,7 +225,7 @@ static thread_return_t THREAD_CALLING_CONV asan_thread_start(void *arg) {
return t->ThreadStart(); return t->ThreadStart();
} }
#ifndef _WIN32 #if ASAN_INTERCEPT_PTHREAD_CREATE
INTERCEPTOR(int, pthread_create, void *thread, INTERCEPTOR(int, pthread_create, void *thread,
void *attr, void *(*start_routine)(void*), void *arg) { void *attr, void *(*start_routine)(void*), void *arg) {
GET_STACK_TRACE_HERE(kStackTraceMax); GET_STACK_TRACE_HERE(kStackTraceMax);
@ -200,7 +234,7 @@ INTERCEPTOR(int, pthread_create, void *thread,
asanThreadRegistry().RegisterThread(t); asanThreadRegistry().RegisterThread(t);
return REAL(pthread_create)(thread, attr, asan_thread_start, t); return REAL(pthread_create)(thread, attr, asan_thread_start, t);
} }
#endif // !_WIN32 #endif // ASAN_INTERCEPT_PTHREAD_CREATE
#if ASAN_INTERCEPT_SIGNAL_AND_SIGACTION #if ASAN_INTERCEPT_SIGNAL_AND_SIGACTION
INTERCEPTOR(void*, signal, int signum, void *handler) { INTERCEPTOR(void*, signal, int signum, void *handler) {
@ -228,23 +262,21 @@ INTERCEPTOR(void, longjmp, void *env, int val) {
REAL(longjmp)(env, val); REAL(longjmp)(env, val);
} }
#if !defined(_WIN32) #if ASAN_INTERCEPT__LONGJMP
INTERCEPTOR(void, _longjmp, void *env, int val) { INTERCEPTOR(void, _longjmp, void *env, int val) {
__asan_handle_no_return(); __asan_handle_no_return();
REAL(_longjmp)(env, val); REAL(_longjmp)(env, val);
} }
#endif
#if ASAN_INTERCEPT_SIGLONGJMP
INTERCEPTOR(void, siglongjmp, void *env, int val) { INTERCEPTOR(void, siglongjmp, void *env, int val) {
__asan_handle_no_return(); __asan_handle_no_return();
REAL(siglongjmp)(env, val); REAL(siglongjmp)(env, val);
} }
#endif #endif
#if ASAN_HAS_EXCEPTIONS == 1 #if ASAN_INTERCEPT___CXA_THROW
#ifdef __APPLE__
extern "C" void __cxa_throw(void *a, void *b, void *c);
#endif // __APPLE__
INTERCEPTOR(void, __cxa_throw, void *a, void *b, void *c) { INTERCEPTOR(void, __cxa_throw, void *a, void *b, void *c) {
CHECK(REAL(__cxa_throw)); CHECK(REAL(__cxa_throw));
__asan_handle_no_return(); __asan_handle_no_return();
@ -367,26 +399,14 @@ INTERCEPTOR(char*, strchr, const char *str, int c) {
return result; return result;
} }
#ifdef __linux__ #if ASAN_INTERCEPT_INDEX
# if ASAN_USE_ALIAS_ATTRIBUTE_FOR_INDEX
INTERCEPTOR(char*, index, const char *string, int c) INTERCEPTOR(char*, index, const char *string, int c)
ALIAS(WRAPPER_NAME(strchr)); ALIAS(WRAPPER_NAME(strchr));
#else # else
DEFINE_REAL(char*, index, const char *string, int c) DEFINE_REAL(char*, index, const char *string, int c)
#endif # endif
#endif // ASAN_INTERCEPT_INDEX
INTERCEPTOR(int, strcasecmp, const char *s1, const char *s2) {
ENSURE_ASAN_INITED();
unsigned char c1, c2;
uptr i;
for (i = 0; ; i++) {
c1 = (unsigned char)s1[i];
c2 = (unsigned char)s2[i];
if (CharCaseCmp(c1, c2) != 0 || c1 == '\0') break;
}
ASAN_READ_RANGE(s1, i + 1);
ASAN_READ_RANGE(s2, i + 1);
return CharCaseCmp(c1, c2);
}
INTERCEPTOR(char*, strcat, char *to, const char *from) { // NOLINT INTERCEPTOR(char*, strcat, char *to, const char *from) { // NOLINT
ENSURE_ASAN_INITED(); ENSURE_ASAN_INITED();
@ -420,8 +440,8 @@ INTERCEPTOR(char*, strncat, char *to, const char *from, uptr size) {
} }
INTERCEPTOR(int, strcmp, const char *s1, const char *s2) { INTERCEPTOR(int, strcmp, const char *s1, const char *s2) {
if (!asan_inited) { if (asan_init_is_running) {
return internal_strcmp(s1, s2); return REAL(strcmp)(s1, s2);
} }
ENSURE_ASAN_INITED(); ENSURE_ASAN_INITED();
unsigned char c1, c2; unsigned char c1, c2;
@ -452,6 +472,7 @@ INTERCEPTOR(char*, strcpy, char *to, const char *from) { // NOLINT
return REAL(strcpy)(to, from); // NOLINT return REAL(strcpy)(to, from); // NOLINT
} }
#if ASAN_INTERCEPT_STRDUP
INTERCEPTOR(char*, strdup, const char *s) { INTERCEPTOR(char*, strdup, const char *s) {
ENSURE_ASAN_INITED(); ENSURE_ASAN_INITED();
if (flags()->replace_str) { if (flags()->replace_str) {
@ -460,6 +481,7 @@ INTERCEPTOR(char*, strdup, const char *s) {
} }
return REAL(strdup)(s); return REAL(strdup)(s);
} }
#endif
INTERCEPTOR(uptr, strlen, const char *s) { INTERCEPTOR(uptr, strlen, const char *s) {
// strlen is called from malloc_default_purgeable_zone() // strlen is called from malloc_default_purgeable_zone()
@ -475,6 +497,21 @@ INTERCEPTOR(uptr, strlen, const char *s) {
return length; return length;
} }
#if ASAN_INTERCEPT_STRCASECMP_AND_STRNCASECMP
INTERCEPTOR(int, strcasecmp, const char *s1, const char *s2) {
ENSURE_ASAN_INITED();
unsigned char c1, c2;
uptr i;
for (i = 0; ; i++) {
c1 = (unsigned char)s1[i];
c2 = (unsigned char)s2[i];
if (CharCaseCmp(c1, c2) != 0 || c1 == '\0') break;
}
ASAN_READ_RANGE(s1, i + 1);
ASAN_READ_RANGE(s2, i + 1);
return CharCaseCmp(c1, c2);
}
INTERCEPTOR(int, strncasecmp, const char *s1, const char *s2, uptr n) { INTERCEPTOR(int, strncasecmp, const char *s1, const char *s2, uptr n) {
ENSURE_ASAN_INITED(); ENSURE_ASAN_INITED();
unsigned char c1 = 0, c2 = 0; unsigned char c1 = 0, c2 = 0;
@ -488,6 +525,7 @@ INTERCEPTOR(int, strncasecmp, const char *s1, const char *s2, uptr n) {
ASAN_READ_RANGE(s2, Min(i + 1, n)); ASAN_READ_RANGE(s2, Min(i + 1, n));
return CharCaseCmp(c1, c2); return CharCaseCmp(c1, c2);
} }
#endif // ASAN_INTERCEPT_STRCASECMP_AND_STRNCASECMP
INTERCEPTOR(int, strncmp, const char *s1, const char *s2, uptr size) { INTERCEPTOR(int, strncmp, const char *s1, const char *s2, uptr size) {
// strncmp is called from malloc_default_purgeable_zone() // strncmp is called from malloc_default_purgeable_zone()
@ -679,18 +717,22 @@ void InitializeAsanInterceptors() {
ASAN_INTERCEPT_FUNC(strncat); ASAN_INTERCEPT_FUNC(strncat);
ASAN_INTERCEPT_FUNC(strncmp); ASAN_INTERCEPT_FUNC(strncmp);
ASAN_INTERCEPT_FUNC(strncpy); ASAN_INTERCEPT_FUNC(strncpy);
#if !defined(_WIN32) #if ASAN_INTERCEPT_STRCASECMP_AND_STRNCASECMP
ASAN_INTERCEPT_FUNC(strcasecmp); ASAN_INTERCEPT_FUNC(strcasecmp);
ASAN_INTERCEPT_FUNC(strdup);
ASAN_INTERCEPT_FUNC(strncasecmp); ASAN_INTERCEPT_FUNC(strncasecmp);
# ifndef __APPLE__ #endif
#if ASAN_INTERCEPT_STRDUP
ASAN_INTERCEPT_FUNC(strdup);
#endif
#if ASAN_INTERCEPT_STRNLEN
ASAN_INTERCEPT_FUNC(strnlen);
#endif
#if ASAN_INTERCEPT_INDEX
# if ASAN_USE_ALIAS_ATTRIBUTE_FOR_INDEX
ASAN_INTERCEPT_FUNC(index); ASAN_INTERCEPT_FUNC(index);
# else # else
CHECK(OVERRIDE_FUNCTION(index, WRAP(strchr))); CHECK(OVERRIDE_FUNCTION(index, WRAP(strchr)));
# endif # endif
#endif
#if ASAN_INTERCEPT_STRNLEN
ASAN_INTERCEPT_FUNC(strnlen);
#endif #endif
ASAN_INTERCEPT_FUNC(atoi); ASAN_INTERCEPT_FUNC(atoi);
@ -707,19 +749,20 @@ void InitializeAsanInterceptors() {
ASAN_INTERCEPT_FUNC(sigaction); ASAN_INTERCEPT_FUNC(sigaction);
ASAN_INTERCEPT_FUNC(signal); ASAN_INTERCEPT_FUNC(signal);
#endif #endif
#if ASAN_INTERCEPT__LONGJMP
#if !defined(_WIN32)
ASAN_INTERCEPT_FUNC(_longjmp); ASAN_INTERCEPT_FUNC(_longjmp);
INTERCEPT_FUNCTION(__cxa_throw); #endif
# if !defined(__APPLE__) #if ASAN_INTERCEPT_SIGLONGJMP
// On Darwin siglongjmp tailcalls longjmp, so we don't want to intercept it
// there.
ASAN_INTERCEPT_FUNC(siglongjmp); ASAN_INTERCEPT_FUNC(siglongjmp);
# endif #endif
// Intercept exception handling functions.
#if ASAN_INTERCEPT___CXA_THROW
INTERCEPT_FUNCTION(__cxa_throw);
#endif #endif
// Intercept threading-related functions // Intercept threading-related functions
#if !defined(_WIN32) #if ASAN_INTERCEPT_PTHREAD_CREATE
ASAN_INTERCEPT_FUNC(pthread_create); ASAN_INTERCEPT_FUNC(pthread_create);
#endif #endif