| //===-- asan_interceptors.cc ------------------------------------*- C++ -*-===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is distributed under the University of Illinois Open Source |
| // License. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| // |
| // This file is a part of AddressSanitizer, an address sanity checker. |
| // |
| // Intercept various libc functions. |
| //===----------------------------------------------------------------------===// |
| #include "asan_interceptors.h" |
| |
| #include "asan_allocator.h" |
| #include "asan_interface.h" |
| #include "asan_internal.h" |
| #include "asan_mapping.h" |
| #include "asan_stack.h" |
| #include "asan_stats.h" |
| #include "asan_thread_registry.h" |
| #include "interception/interception.h" |
| #include "sanitizer_common/sanitizer_libc.h" |
| |
| using namespace __sanitizer; // NOLINT |
| |
| // Use macro to describe if specific function should be |
| // intercepted on a given platform. |
| #if !defined(_WIN32) |
| # define ASAN_INTERCEPT_ATOLL_AND_STRTOLL 1 |
| #else |
| # define ASAN_INTERCEPT_ATOLL_AND_STRTOLL 0 |
| #endif |
| |
| #if !defined(__APPLE__) |
| # define ASAN_INTERCEPT_STRNLEN 1 |
| #else |
| # define ASAN_INTERCEPT_STRNLEN 0 |
| #endif |
| |
| #if defined(ANDROID) || defined(_WIN32) |
| # define ASAN_INTERCEPT_SIGNAL_AND_SIGACTION 0 |
| #else |
| # define ASAN_INTERCEPT_SIGNAL_AND_SIGACTION 1 |
| #endif |
| |
| // Use extern declarations of intercepted functions on Mac and Windows |
| // to avoid including system headers. |
| #if defined(__APPLE__) || (defined(_WIN32) && !defined(_DLL)) |
| extern "C" { |
| // signal.h |
| # if ASAN_INTERCEPT_SIGNAL_AND_SIGACTION |
| struct sigaction; |
| int sigaction(int sig, const struct sigaction *act, |
| struct sigaction *oldact); |
| void *signal(int signum, void *handler); |
| # endif |
| |
| // setjmp.h |
| void longjmp(void* env, int value); |
| # if !defined(_WIN32) |
| void _longjmp(void *env, int value); |
| # endif |
| |
| // string.h / strings.h |
| int memcmp(const void *a1, const void *a2, uptr size); |
| void* memmove(void *to, const void *from, uptr size); |
| void* memcpy(void *to, const void *from, uptr size); |
| void* memset(void *block, int c, uptr size); |
| 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* strcpy(char *to, const char* from); // NOLINT |
| char* strncpy(char *to, const char* from, uptr size); |
| int strcmp(const char *s1, const char* s2); |
| int strncmp(const char *s1, const char* s2, uptr size); |
| # if !defined(_WIN32) |
| int strcasecmp(const char *s1, const char *s2); |
| int strncasecmp(const char *s1, const char *s2, uptr n); |
| char* strdup(const char *s); |
| # endif |
| uptr strlen(const char *s); |
| # if ASAN_INTERCEPT_STRNLEN |
| uptr strnlen(const char *s, uptr maxlen); |
| # endif |
| |
| // stdlib.h |
| int atoi(const char *nptr); |
| long atol(const char *nptr); // NOLINT |
| long strtol(const char *nptr, char **endptr, int base); // NOLINT |
| # if ASAN_INTERCEPT_ATOLL_AND_STRTOLL |
| long long atoll(const char *nptr); // NOLINT |
| long long strtoll(const char *nptr, char **endptr, int base); // NOLINT |
| # endif |
| |
| // Windows threads. |
| # if defined(_WIN32) |
| __declspec(dllimport) |
| void* __stdcall CreateThread(void *sec, uptr st, void* start, |
| void *arg, DWORD fl, DWORD *id); |
| # endif |
| |
| // Posix threads. |
| # if !defined(_WIN32) |
| int pthread_create(void *thread, void *attr, void *(*start_routine)(void*), |
| void *arg); |
| # endif |
| } // extern "C" |
| #endif |
| |
| namespace __asan { |
| |
| // Instruments read/write access to a single byte in memory. |
| // On error calls __asan_report_error, which aborts the program. |
| #define ACCESS_ADDRESS(address, isWrite) do { \ |
| if (AddressIsPoisoned(address)) { \ |
| GET_CURRENT_PC_BP_SP; \ |
| __asan_report_error(pc, bp, sp, address, isWrite, /* access_size */ 1); \ |
| } \ |
| } while (0) |
| |
| // We implement ACCESS_MEMORY_RANGE, ASAN_READ_RANGE, |
| // and ASAN_WRITE_RANGE as macro instead of function so |
| // that no extra frames are created, and stack trace contains |
| // relevant information only. |
| |
| // Instruments read/write access to a memory range. |
| // More complex implementation is possible, for now just |
| // checking the first and the last byte of a range. |
| #define ACCESS_MEMORY_RANGE(offset, size, isWrite) do { \ |
| if (size > 0) { \ |
| uptr ptr = (uptr)(offset); \ |
| ACCESS_ADDRESS(ptr, isWrite); \ |
| ACCESS_ADDRESS(ptr + (size) - 1, isWrite); \ |
| } \ |
| } while (0) |
| |
| #define ASAN_READ_RANGE(offset, size) do { \ |
| ACCESS_MEMORY_RANGE(offset, size, false); \ |
| } while (0) |
| |
| #define ASAN_WRITE_RANGE(offset, size) do { \ |
| ACCESS_MEMORY_RANGE(offset, size, true); \ |
| } while (0) |
| |
| // Behavior of functions like "memcpy" or "strcpy" is undefined |
| // if memory intervals overlap. We report error in this case. |
| // Macro is used to avoid creation of new frames. |
| static inline bool RangesOverlap(const char *offset1, uptr length1, |
| const char *offset2, uptr length2) { |
| return !((offset1 + length1 <= offset2) || (offset2 + length2 <= offset1)); |
| } |
| #define CHECK_RANGES_OVERLAP(name, _offset1, length1, _offset2, length2) do { \ |
| const char *offset1 = (const char*)_offset1; \ |
| const char *offset2 = (const char*)_offset2; \ |
| if (RangesOverlap(offset1, length1, offset2, length2)) { \ |
| Report("ERROR: AddressSanitizer %s-param-overlap: " \ |
| "memory ranges [%p,%p) and [%p, %p) overlap\n", \ |
| name, offset1, offset1 + length1, offset2, offset2 + length2); \ |
| PRINT_CURRENT_STACK(); \ |
| ShowStatsAndAbort(); \ |
| } \ |
| } while (0) |
| |
| #define ENSURE_ASAN_INITED() do { \ |
| CHECK(!asan_init_is_running); \ |
| if (!asan_inited) { \ |
| __asan_init(); \ |
| } \ |
| } while (0) |
| |
| static inline bool IsSpace(int c) { |
| return (c == ' ') || (c == '\n') || (c == '\t') || |
| (c == '\f') || (c == '\r') || (c == '\v'); |
| } |
| |
| static inline bool IsDigit(int c) { |
| return (c >= '0') && (c <= '9'); |
| } |
| |
| static inline int ToLower(int c) { |
| return (c >= 'A' && c <= 'Z') ? (c + 'a' - 'A') : c; |
| } |
| |
| // ---------------------- Internal string functions ---------------- {{{1 |
| |
| s64 internal_simple_strtoll(const char *nptr, char **endptr, int base) { |
| CHECK(base == 10); |
| while (IsSpace(*nptr)) nptr++; |
| int sgn = 1; |
| u64 res = 0; |
| bool have_digits = false; |
| char *old_nptr = (char*)nptr; |
| if (*nptr == '+') { |
| sgn = 1; |
| nptr++; |
| } else if (*nptr == '-') { |
| sgn = -1; |
| nptr++; |
| } |
| while (IsDigit(*nptr)) { |
| res = (res <= UINT64_MAX / 10) ? res * 10 : UINT64_MAX; |
| int digit = ((*nptr) - '0'); |
| res = (res <= UINT64_MAX - digit) ? res + digit : UINT64_MAX; |
| have_digits = true; |
| nptr++; |
| } |
| if (endptr != 0) { |
| *endptr = (have_digits) ? (char*)nptr : old_nptr; |
| } |
| if (sgn > 0) { |
| return (s64)(Min((u64)INT64_MAX, res)); |
| } else { |
| return (res > INT64_MAX) ? INT64_MIN : ((s64)res * -1); |
| } |
| } |
| |
| s64 internal_atoll(const char *nptr) { |
| return internal_simple_strtoll(nptr, (char**)0, 10); |
| } |
| |
| uptr internal_strlen(const char *s) { |
| uptr i = 0; |
| while (s[i]) i++; |
| return i; |
| } |
| |
| uptr internal_strnlen(const char *s, uptr maxlen) { |
| #if ASAN_INTERCEPT_STRNLEN |
| if (REAL(strnlen) != 0) { |
| return REAL(strnlen)(s, maxlen); |
| } |
| #endif |
| uptr i = 0; |
| while (i < maxlen && s[i]) i++; |
| return i; |
| } |
| |
| char* internal_strchr(const char *s, int c) { |
| while (true) { |
| if (*s == (char)c) |
| return (char*)s; |
| if (*s == 0) |
| return 0; |
| s++; |
| } |
| } |
| |
| void* internal_memchr(const void* s, int c, uptr n) { |
| const char* t = (char*)s; |
| for (uptr i = 0; i < n; ++i, ++t) |
| if (*t == c) |
| return (void*)t; |
| return 0; |
| } |
| |
| int internal_memcmp(const void* s1, const void* s2, uptr n) { |
| const char* t1 = (char*)s1; |
| const char* t2 = (char*)s2; |
| for (uptr i = 0; i < n; ++i, ++t1, ++t2) |
| if (*t1 != *t2) |
| return *t1 < *t2 ? -1 : 1; |
| return 0; |
| } |
| |
| // Should not be used in performance-critical places. |
| void* internal_memset(void* s, int c, uptr n) { |
| // The next line prevents Clang from making a call to memset() instead of the |
| // loop below. |
| // FIXME: building the runtime with -ffreestanding is a better idea. However |
| // there currently are linktime problems due to PR12396. |
| char volatile *t = (char*)s; |
| for (uptr i = 0; i < n; ++i, ++t) { |
| *t = c; |
| } |
| return s; |
| } |
| |
| char *internal_strstr(const char *haystack, const char *needle) { |
| // This is O(N^2), but we are not using it in hot places. |
| uptr len1 = internal_strlen(haystack); |
| uptr len2 = internal_strlen(needle); |
| if (len1 < len2) return 0; |
| for (uptr pos = 0; pos <= len1 - len2; pos++) { |
| if (internal_memcmp(haystack + pos, needle, len2) == 0) |
| return (char*)haystack + pos; |
| } |
| return 0; |
| } |
| |
| char *internal_strncat(char *dst, const char *src, uptr n) { |
| uptr len = internal_strlen(dst); |
| uptr i; |
| for (i = 0; i < n && src[i]; i++) |
| dst[len + i] = src[i]; |
| dst[len + i] = 0; |
| return dst; |
| } |
| |
| } // namespace __asan |
| |
| // ---------------------- Wrappers ---------------- {{{1 |
| using namespace __asan; // NOLINT |
| |
| static thread_return_t THREAD_CALLING_CONV asan_thread_start(void *arg) { |
| AsanThread *t = (AsanThread*)arg; |
| asanThreadRegistry().SetCurrent(t); |
| return t->ThreadStart(); |
| } |
| |
| #ifndef _WIN32 |
| INTERCEPTOR(int, pthread_create, void *thread, |
| void *attr, void *(*start_routine)(void*), void *arg) { |
| GET_STACK_TRACE_HERE(kStackTraceMax); |
| int current_tid = asanThreadRegistry().GetCurrentTidOrMinusOne(); |
| AsanThread *t = AsanThread::Create(current_tid, start_routine, arg, &stack); |
| asanThreadRegistry().RegisterThread(t); |
| return REAL(pthread_create)(thread, attr, asan_thread_start, t); |
| } |
| #endif // !_WIN32 |
| |
| #if ASAN_INTERCEPT_SIGNAL_AND_SIGACTION |
| INTERCEPTOR(void*, signal, int signum, void *handler) { |
| if (!AsanInterceptsSignal(signum)) { |
| return REAL(signal)(signum, handler); |
| } |
| return 0; |
| } |
| |
| INTERCEPTOR(int, sigaction, int signum, const struct sigaction *act, |
| struct sigaction *oldact) { |
| if (!AsanInterceptsSignal(signum)) { |
| return REAL(sigaction)(signum, act, oldact); |
| } |
| return 0; |
| } |
| #elif ASAN_POSIX |
| // We need to have defined REAL(sigaction) on posix systems. |
| DEFINE_REAL(int, sigaction, int signum, const struct sigaction *act, |
| struct sigaction *oldact); |
| #endif // ASAN_INTERCEPT_SIGNAL_AND_SIGACTION |
| |
| INTERCEPTOR(void, longjmp, void *env, int val) { |
| __asan_handle_no_return(); |
| REAL(longjmp)(env, val); |
| } |
| |
| #if !defined(_WIN32) |
| INTERCEPTOR(void, _longjmp, void *env, int val) { |
| __asan_handle_no_return(); |
| REAL(_longjmp)(env, val); |
| } |
| |
| INTERCEPTOR(void, siglongjmp, void *env, int val) { |
| __asan_handle_no_return(); |
| REAL(siglongjmp)(env, val); |
| } |
| #endif |
| |
| #if ASAN_HAS_EXCEPTIONS == 1 |
| #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) { |
| CHECK(REAL(__cxa_throw)); |
| __asan_handle_no_return(); |
| REAL(__cxa_throw)(a, b, c); |
| } |
| #endif |
| |
| // intercept mlock and friends. |
| // Since asan maps 16T of RAM, mlock is completely unfriendly to asan. |
| // All functions return 0 (success). |
| static void MlockIsUnsupported() { |
| static bool printed = 0; |
| if (printed) return; |
| printed = true; |
| Printf("INFO: AddressSanitizer ignores mlock/mlockall/munlock/munlockall\n"); |
| } |
| |
| extern "C" { |
| INTERCEPTOR_ATTRIBUTE |
| int mlock(const void *addr, uptr len) { |
| MlockIsUnsupported(); |
| return 0; |
| } |
| |
| INTERCEPTOR_ATTRIBUTE |
| int munlock(const void *addr, uptr len) { |
| MlockIsUnsupported(); |
| return 0; |
| } |
| |
| INTERCEPTOR_ATTRIBUTE |
| int mlockall(int flags) { |
| MlockIsUnsupported(); |
| return 0; |
| } |
| |
| INTERCEPTOR_ATTRIBUTE |
| int munlockall(void) { |
| MlockIsUnsupported(); |
| return 0; |
| } |
| } // extern "C" |
| |
| static inline int CharCmp(unsigned char c1, unsigned char c2) { |
| return (c1 == c2) ? 0 : (c1 < c2) ? -1 : 1; |
| } |
| |
| static inline int CharCaseCmp(unsigned char c1, unsigned char c2) { |
| int c1_low = ToLower(c1); |
| int c2_low = ToLower(c2); |
| return c1_low - c2_low; |
| } |
| |
| INTERCEPTOR(int, memcmp, const void *a1, const void *a2, uptr size) { |
| ENSURE_ASAN_INITED(); |
| unsigned char c1 = 0, c2 = 0; |
| const unsigned char *s1 = (const unsigned char*)a1; |
| const unsigned char *s2 = (const unsigned char*)a2; |
| uptr i; |
| for (i = 0; i < size; i++) { |
| c1 = s1[i]; |
| c2 = s2[i]; |
| if (c1 != c2) break; |
| } |
| ASAN_READ_RANGE(s1, Min(i + 1, size)); |
| ASAN_READ_RANGE(s2, Min(i + 1, size)); |
| return CharCmp(c1, c2); |
| } |
| |
| INTERCEPTOR(void*, memcpy, void *to, const void *from, uptr size) { |
| // memcpy is called during __asan_init() from the internals |
| // of printf(...). |
| if (asan_init_is_running) { |
| return REAL(memcpy)(to, from, size); |
| } |
| ENSURE_ASAN_INITED(); |
| if (FLAG_replace_intrin) { |
| if (to != from) { |
| // We do not treat memcpy with to==from as a bug. |
| // See http://llvm.org/bugs/show_bug.cgi?id=11763. |
| CHECK_RANGES_OVERLAP("memcpy", to, size, from, size); |
| } |
| ASAN_WRITE_RANGE(from, size); |
| ASAN_READ_RANGE(to, size); |
| } |
| return REAL(memcpy)(to, from, size); |
| } |
| |
| INTERCEPTOR(void*, memmove, void *to, const void *from, uptr size) { |
| ENSURE_ASAN_INITED(); |
| if (FLAG_replace_intrin) { |
| ASAN_WRITE_RANGE(from, size); |
| ASAN_READ_RANGE(to, size); |
| } |
| return REAL(memmove)(to, from, size); |
| } |
| |
| INTERCEPTOR(void*, memset, void *block, int c, uptr size) { |
| // memset is called inside Printf. |
| if (asan_init_is_running) { |
| return REAL(memset)(block, c, size); |
| } |
| ENSURE_ASAN_INITED(); |
| if (FLAG_replace_intrin) { |
| ASAN_WRITE_RANGE(block, size); |
| } |
| return REAL(memset)(block, c, size); |
| } |
| |
| INTERCEPTOR(char*, strchr, const char *str, int c) { |
| ENSURE_ASAN_INITED(); |
| char *result = REAL(strchr)(str, c); |
| if (FLAG_replace_str) { |
| uptr bytes_read = (result ? result - str : REAL(strlen)(str)) + 1; |
| ASAN_READ_RANGE(str, bytes_read); |
| } |
| return result; |
| } |
| |
| #ifdef __linux__ |
| INTERCEPTOR(char*, index, const char *string, int c) |
| ALIAS(WRAPPER_NAME(strchr)); |
| #else |
| DEFINE_REAL(char*, index, const char *string, int c); |
| #endif |
| |
| 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 |
| ENSURE_ASAN_INITED(); |
| if (FLAG_replace_str) { |
| uptr from_length = REAL(strlen)(from); |
| ASAN_READ_RANGE(from, from_length + 1); |
| if (from_length > 0) { |
| uptr to_length = REAL(strlen)(to); |
| ASAN_READ_RANGE(to, to_length); |
| ASAN_WRITE_RANGE(to + to_length, from_length + 1); |
| CHECK_RANGES_OVERLAP("strcat", to, to_length + 1, from, from_length + 1); |
| } |
| } |
| return REAL(strcat)(to, from); // NOLINT |
| } |
| |
| INTERCEPTOR(int, strcmp, const char *s1, const char *s2) { |
| if (!asan_inited) { |
| return internal_strcmp(s1, s2); |
| } |
| unsigned char c1, c2; |
| uptr i; |
| for (i = 0; ; i++) { |
| c1 = (unsigned char)s1[i]; |
| c2 = (unsigned char)s2[i]; |
| if (c1 != c2 || c1 == '\0') break; |
| } |
| ASAN_READ_RANGE(s1, i + 1); |
| ASAN_READ_RANGE(s2, i + 1); |
| return CharCmp(c1, c2); |
| } |
| |
| INTERCEPTOR(char*, strcpy, char *to, const char *from) { // NOLINT |
| // strcpy is called from malloc_default_purgeable_zone() |
| // in __asan::ReplaceSystemAlloc() on Mac. |
| if (asan_init_is_running) { |
| return REAL(strcpy)(to, from); // NOLINT |
| } |
| ENSURE_ASAN_INITED(); |
| if (FLAG_replace_str) { |
| uptr from_size = REAL(strlen)(from) + 1; |
| CHECK_RANGES_OVERLAP("strcpy", to, from_size, from, from_size); |
| ASAN_READ_RANGE(from, from_size); |
| ASAN_WRITE_RANGE(to, from_size); |
| } |
| return REAL(strcpy)(to, from); // NOLINT |
| } |
| |
| INTERCEPTOR(char*, strdup, const char *s) { |
| ENSURE_ASAN_INITED(); |
| if (FLAG_replace_str) { |
| uptr length = REAL(strlen)(s); |
| ASAN_READ_RANGE(s, length + 1); |
| } |
| return REAL(strdup)(s); |
| } |
| |
| INTERCEPTOR(uptr, strlen, const char *s) { |
| // strlen is called from malloc_default_purgeable_zone() |
| // in __asan::ReplaceSystemAlloc() on Mac. |
| if (asan_init_is_running) { |
| return REAL(strlen)(s); |
| } |
| ENSURE_ASAN_INITED(); |
| uptr length = REAL(strlen)(s); |
| if (FLAG_replace_str) { |
| ASAN_READ_RANGE(s, length + 1); |
| } |
| return length; |
| } |
| |
| INTERCEPTOR(int, strncasecmp, const char *s1, const char *s2, uptr n) { |
| ENSURE_ASAN_INITED(); |
| unsigned char c1 = 0, c2 = 0; |
| uptr i; |
| for (i = 0; i < n; i++) { |
| c1 = (unsigned char)s1[i]; |
| c2 = (unsigned char)s2[i]; |
| if (CharCaseCmp(c1, c2) != 0 || c1 == '\0') break; |
| } |
| ASAN_READ_RANGE(s1, Min(i + 1, n)); |
| ASAN_READ_RANGE(s2, Min(i + 1, n)); |
| return CharCaseCmp(c1, c2); |
| } |
| |
| INTERCEPTOR(int, strncmp, const char *s1, const char *s2, uptr size) { |
| // strncmp is called from malloc_default_purgeable_zone() |
| // in __asan::ReplaceSystemAlloc() on Mac. |
| if (asan_init_is_running) { |
| return REAL(strncmp)(s1, s2, size); |
| } |
| unsigned char c1 = 0, c2 = 0; |
| uptr i; |
| for (i = 0; i < size; i++) { |
| c1 = (unsigned char)s1[i]; |
| c2 = (unsigned char)s2[i]; |
| if (c1 != c2 || c1 == '\0') break; |
| } |
| ASAN_READ_RANGE(s1, Min(i + 1, size)); |
| ASAN_READ_RANGE(s2, Min(i + 1, size)); |
| return CharCmp(c1, c2); |
| } |
| |
| INTERCEPTOR(char*, strncpy, char *to, const char *from, uptr size) { |
| ENSURE_ASAN_INITED(); |
| if (FLAG_replace_str) { |
| uptr from_size = Min(size, internal_strnlen(from, size) + 1); |
| CHECK_RANGES_OVERLAP("strncpy", to, from_size, from, from_size); |
| ASAN_READ_RANGE(from, from_size); |
| ASAN_WRITE_RANGE(to, size); |
| } |
| return REAL(strncpy)(to, from, size); |
| } |
| |
| #if ASAN_INTERCEPT_STRNLEN |
| INTERCEPTOR(uptr, strnlen, const char *s, uptr maxlen) { |
| ENSURE_ASAN_INITED(); |
| uptr length = REAL(strnlen)(s, maxlen); |
| if (FLAG_replace_str) { |
| ASAN_READ_RANGE(s, Min(length + 1, maxlen)); |
| } |
| return length; |
| } |
| #endif // ASAN_INTERCEPT_STRNLEN |
| |
| static inline bool IsValidStrtolBase(int base) { |
| return (base == 0) || (2 <= base && base <= 36); |
| } |
| |
| static inline void FixRealStrtolEndptr(const char *nptr, char **endptr) { |
| CHECK(endptr != 0); |
| if (nptr == *endptr) { |
| // No digits were found at strtol call, we need to find out the last |
| // symbol accessed by strtoll on our own. |
| // We get this symbol by skipping leading blanks and optional +/- sign. |
| while (IsSpace(*nptr)) nptr++; |
| if (*nptr == '+' || *nptr == '-') nptr++; |
| *endptr = (char*)nptr; |
| } |
| CHECK(*endptr >= nptr); |
| } |
| |
| INTERCEPTOR(long, strtol, const char *nptr, // NOLINT |
| char **endptr, int base) { |
| ENSURE_ASAN_INITED(); |
| if (!FLAG_replace_str) { |
| return REAL(strtol)(nptr, endptr, base); |
| } |
| char *real_endptr; |
| long result = REAL(strtol)(nptr, &real_endptr, base); // NOLINT |
| if (endptr != 0) { |
| *endptr = real_endptr; |
| } |
| if (IsValidStrtolBase(base)) { |
| FixRealStrtolEndptr(nptr, &real_endptr); |
| ASAN_READ_RANGE(nptr, (real_endptr - nptr) + 1); |
| } |
| return result; |
| } |
| |
| INTERCEPTOR(int, atoi, const char *nptr) { |
| ENSURE_ASAN_INITED(); |
| if (!FLAG_replace_str) { |
| return REAL(atoi)(nptr); |
| } |
| char *real_endptr; |
| // "man atoi" tells that behavior of atoi(nptr) is the same as |
| // strtol(nptr, 0, 10), i.e. it sets errno to ERANGE if the |
| // parsed integer can't be stored in *long* type (even if it's |
| // different from int). So, we just imitate this behavior. |
| int result = REAL(strtol)(nptr, &real_endptr, 10); |
| FixRealStrtolEndptr(nptr, &real_endptr); |
| ASAN_READ_RANGE(nptr, (real_endptr - nptr) + 1); |
| return result; |
| } |
| |
| INTERCEPTOR(long, atol, const char *nptr) { // NOLINT |
| ENSURE_ASAN_INITED(); |
| if (!FLAG_replace_str) { |
| return REAL(atol)(nptr); |
| } |
| char *real_endptr; |
| long result = REAL(strtol)(nptr, &real_endptr, 10); // NOLINT |
| FixRealStrtolEndptr(nptr, &real_endptr); |
| ASAN_READ_RANGE(nptr, (real_endptr - nptr) + 1); |
| return result; |
| } |
| |
| #if ASAN_INTERCEPT_ATOLL_AND_STRTOLL |
| INTERCEPTOR(long long, strtoll, const char *nptr, // NOLINT |
| char **endptr, int base) { |
| ENSURE_ASAN_INITED(); |
| if (!FLAG_replace_str) { |
| return REAL(strtoll)(nptr, endptr, base); |
| } |
| char *real_endptr; |
| long long result = REAL(strtoll)(nptr, &real_endptr, base); // NOLINT |
| if (endptr != 0) { |
| *endptr = real_endptr; |
| } |
| // If base has unsupported value, strtoll can exit with EINVAL |
| // without reading any characters. So do additional checks only |
| // if base is valid. |
| if (IsValidStrtolBase(base)) { |
| FixRealStrtolEndptr(nptr, &real_endptr); |
| ASAN_READ_RANGE(nptr, (real_endptr - nptr) + 1); |
| } |
| return result; |
| } |
| |
| INTERCEPTOR(long long, atoll, const char *nptr) { // NOLINT |
| ENSURE_ASAN_INITED(); |
| if (!FLAG_replace_str) { |
| return REAL(atoll)(nptr); |
| } |
| char *real_endptr; |
| long long result = REAL(strtoll)(nptr, &real_endptr, 10); // NOLINT |
| FixRealStrtolEndptr(nptr, &real_endptr); |
| ASAN_READ_RANGE(nptr, (real_endptr - nptr) + 1); |
| return result; |
| } |
| #endif // ASAN_INTERCEPT_ATOLL_AND_STRTOLL |
| |
| #define ASAN_INTERCEPT_FUNC(name) do { \ |
| if (!INTERCEPT_FUNCTION(name) && FLAG_v > 0) \ |
| Report("AddressSanitizer: failed to intercept '" #name "'\n"); \ |
| } while (0) |
| |
| #if defined(_WIN32) |
| INTERCEPTOR_WINAPI(DWORD, CreateThread, |
| void* security, uptr stack_size, |
| DWORD (__stdcall *start_routine)(void*), void* arg, |
| DWORD flags, void* tid) { |
| GET_STACK_TRACE_HERE(kStackTraceMax); |
| int current_tid = asanThreadRegistry().GetCurrentTidOrMinusOne(); |
| AsanThread *t = AsanThread::Create(current_tid, start_routine, arg, &stack); |
| asanThreadRegistry().RegisterThread(t); |
| return REAL(CreateThread)(security, stack_size, |
| asan_thread_start, t, flags, tid); |
| } |
| |
| namespace __asan { |
| void InitializeWindowsInterceptors() { |
| ASAN_INTERCEPT_FUNC(CreateThread); |
| } |
| |
| } // namespace __asan |
| #endif |
| |
| // ---------------------- InitializeAsanInterceptors ---------------- {{{1 |
| namespace __asan { |
| void InitializeAsanInterceptors() { |
| static bool was_called_once; |
| CHECK(was_called_once == false); |
| was_called_once = true; |
| // Intercept mem* functions. |
| ASAN_INTERCEPT_FUNC(memcmp); |
| ASAN_INTERCEPT_FUNC(memmove); |
| ASAN_INTERCEPT_FUNC(memset); |
| if (PLATFORM_HAS_DIFFERENT_MEMCPY_AND_MEMMOVE) { |
| ASAN_INTERCEPT_FUNC(memcpy); |
| } else { |
| REAL(memcpy) = REAL(memmove); |
| } |
| |
| // Intercept str* functions. |
| ASAN_INTERCEPT_FUNC(strcat); // NOLINT |
| ASAN_INTERCEPT_FUNC(strchr); |
| ASAN_INTERCEPT_FUNC(strcmp); |
| ASAN_INTERCEPT_FUNC(strcpy); // NOLINT |
| ASAN_INTERCEPT_FUNC(strlen); |
| ASAN_INTERCEPT_FUNC(strncmp); |
| ASAN_INTERCEPT_FUNC(strncpy); |
| #if !defined(_WIN32) |
| ASAN_INTERCEPT_FUNC(strcasecmp); |
| ASAN_INTERCEPT_FUNC(strdup); |
| ASAN_INTERCEPT_FUNC(strncasecmp); |
| # ifndef __APPLE__ |
| ASAN_INTERCEPT_FUNC(index); |
| # else |
| CHECK(OVERRIDE_FUNCTION(index, WRAP(strchr))); |
| # endif |
| #endif |
| #if ASAN_INTERCEPT_STRNLEN |
| ASAN_INTERCEPT_FUNC(strnlen); |
| #endif |
| |
| ASAN_INTERCEPT_FUNC(atoi); |
| ASAN_INTERCEPT_FUNC(atol); |
| ASAN_INTERCEPT_FUNC(strtol); |
| #if ASAN_INTERCEPT_ATOLL_AND_STRTOLL |
| ASAN_INTERCEPT_FUNC(atoll); |
| ASAN_INTERCEPT_FUNC(strtoll); |
| #endif |
| |
| // Intecept signal- and jump-related functions. |
| ASAN_INTERCEPT_FUNC(longjmp); |
| #if ASAN_INTERCEPT_SIGNAL_AND_SIGACTION |
| ASAN_INTERCEPT_FUNC(sigaction); |
| ASAN_INTERCEPT_FUNC(signal); |
| #endif |
| |
| #if !defined(_WIN32) |
| ASAN_INTERCEPT_FUNC(_longjmp); |
| INTERCEPT_FUNCTION(__cxa_throw); |
| # if !defined(__APPLE__) |
| // On Darwin siglongjmp tailcalls longjmp, so we don't want to intercept it |
| // there. |
| ASAN_INTERCEPT_FUNC(siglongjmp); |
| # endif |
| #endif |
| |
| // Intercept threading-related functions |
| #if !defined(_WIN32) |
| ASAN_INTERCEPT_FUNC(pthread_create); |
| #endif |
| |
| // Some Windows-specific interceptors. |
| #if defined(_WIN32) |
| InitializeWindowsInterceptors(); |
| #endif |
| |
| // Some Mac-specific interceptors. |
| #if defined(__APPLE__) |
| InitializeMacInterceptors(); |
| #endif |
| |
| if (FLAG_v > 0) { |
| Report("AddressSanitizer: libc interceptors initialized\n"); |
| } |
| } |
| |
| } // namespace __asan |