forked from OSchip/llvm-project
759 lines
23 KiB
C++
759 lines
23 KiB
C++
//===-- asan_interceptors.cc ----------------------------------------------===//
|
|
//
|
|
// 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_intercepted_functions.h"
|
|
#include "asan_internal.h"
|
|
#include "asan_mapping.h"
|
|
#include "asan_report.h"
|
|
#include "asan_stack.h"
|
|
#include "asan_stats.h"
|
|
#include "asan_thread_registry.h"
|
|
#include "interception/interception.h"
|
|
#include "sanitizer_common/sanitizer_libc.h"
|
|
|
|
namespace __asan {
|
|
|
|
// Return true if we can quickly decide that the region is unpoisoned.
|
|
static inline bool QuickCheckForUnpoisonedRegion(uptr beg, uptr size) {
|
|
if (size == 0) return true;
|
|
if (size <= 32)
|
|
return !AddressIsPoisoned(beg) &&
|
|
!AddressIsPoisoned(beg + size - 1) &&
|
|
!AddressIsPoisoned(beg + size / 2);
|
|
return false;
|
|
}
|
|
|
|
// 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.
|
|
// We check all shadow bytes.
|
|
#define ACCESS_MEMORY_RANGE(offset, size, isWrite) do { \
|
|
uptr __offset = (uptr)(offset); \
|
|
uptr __size = (uptr)(size); \
|
|
uptr __bad = 0; \
|
|
if (!QuickCheckForUnpoisonedRegion(__offset, __size) && \
|
|
(__bad = __asan_region_is_poisoned(__offset, __size))) { \
|
|
GET_CURRENT_PC_BP_SP; \
|
|
__asan_report_error(pc, bp, sp, __bad, isWrite, __size); \
|
|
} \
|
|
} while (0)
|
|
|
|
#define ASAN_READ_RANGE(offset, size) ACCESS_MEMORY_RANGE(offset, size, false)
|
|
#define ASAN_WRITE_RANGE(offset, size) ACCESS_MEMORY_RANGE(offset, size, true);
|
|
|
|
// 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)) { \
|
|
GET_STACK_TRACE_FATAL_HERE; \
|
|
ReportStringFunctionMemoryRangesOverlap(name, offset1, length1, \
|
|
offset2, length2, &stack); \
|
|
} \
|
|
} while (0)
|
|
|
|
#define ENSURE_ASAN_INITED() do { \
|
|
CHECK(!asan_init_is_running); \
|
|
if (!asan_inited) { \
|
|
__asan_init(); \
|
|
} \
|
|
} while (0)
|
|
|
|
static inline uptr MaybeRealStrnlen(const char *s, uptr maxlen) {
|
|
#if ASAN_INTERCEPT_STRNLEN
|
|
if (REAL(strnlen) != 0) {
|
|
return REAL(strnlen)(s, maxlen);
|
|
}
|
|
#endif
|
|
return internal_strnlen(s, maxlen);
|
|
}
|
|
|
|
void SetThreadName(const char *name) {
|
|
AsanThread *t = asanThreadRegistry().GetCurrent();
|
|
if (t)
|
|
t->summary()->set_name(name);
|
|
}
|
|
|
|
} // namespace __asan
|
|
|
|
// ---------------------- Wrappers ---------------- {{{1
|
|
using namespace __asan; // NOLINT
|
|
|
|
#define COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ptr, size) \
|
|
ASAN_WRITE_RANGE(ptr, size)
|
|
#define COMMON_INTERCEPTOR_READ_RANGE(ctx, ptr, size) ASAN_READ_RANGE(ptr, size)
|
|
#define COMMON_INTERCEPTOR_ENTER(ctx, func, ...) \
|
|
do { \
|
|
ctx = 0; \
|
|
(void)ctx; \
|
|
ENSURE_ASAN_INITED(); \
|
|
} while (false)
|
|
#define COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd) do { } while (false)
|
|
#define COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd) do { } while (false)
|
|
#define COMMON_INTERCEPTOR_SET_THREAD_NAME(ctx, name) SetThreadName(name)
|
|
#include "sanitizer_common/sanitizer_common_interceptors.inc"
|
|
|
|
static thread_return_t THREAD_CALLING_CONV asan_thread_start(void *arg) {
|
|
AsanThread *t = (AsanThread*)arg;
|
|
asanThreadRegistry().SetCurrent(t);
|
|
return t->ThreadStart();
|
|
}
|
|
|
|
#if ASAN_INTERCEPT_PTHREAD_CREATE
|
|
INTERCEPTOR(int, pthread_create, void *thread,
|
|
void *attr, void *(*start_routine)(void*), void *arg) {
|
|
GET_STACK_TRACE_THREAD;
|
|
u32 current_tid = asanThreadRegistry().GetCurrentTidOrInvalid();
|
|
AsanThread *t = AsanThread::Create(current_tid, start_routine, arg, &stack);
|
|
asanThreadRegistry().RegisterThread(t);
|
|
return REAL(pthread_create)(thread, attr, asan_thread_start, t);
|
|
}
|
|
#endif // ASAN_INTERCEPT_PTHREAD_CREATE
|
|
|
|
#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
|
|
|
|
#if ASAN_INTERCEPT_SWAPCONTEXT
|
|
static void ClearShadowMemoryForContextStack(uptr stack, uptr ssize) {
|
|
// Align to page size.
|
|
uptr PageSize = GetPageSizeCached();
|
|
uptr bottom = stack & ~(PageSize - 1);
|
|
ssize += stack - bottom;
|
|
ssize = RoundUpTo(ssize, PageSize);
|
|
static const uptr kMaxSaneContextStackSize = 1 << 22; // 4 Mb
|
|
if (ssize && ssize <= kMaxSaneContextStackSize) {
|
|
PoisonShadow(bottom, ssize, 0);
|
|
}
|
|
}
|
|
|
|
INTERCEPTOR(int, swapcontext, struct ucontext_t *oucp,
|
|
struct ucontext_t *ucp) {
|
|
static bool reported_warning = false;
|
|
if (!reported_warning) {
|
|
Report("WARNING: ASan doesn't fully support makecontext/swapcontext "
|
|
"functions and may produce false positives in some cases!\n");
|
|
reported_warning = true;
|
|
}
|
|
// Clear shadow memory for new context (it may share stack
|
|
// with current context).
|
|
uptr stack, ssize;
|
|
ReadContextStack(ucp, &stack, &ssize);
|
|
ClearShadowMemoryForContextStack(stack, ssize);
|
|
int res = REAL(swapcontext)(oucp, ucp);
|
|
// swapcontext technically does not return, but program may swap context to
|
|
// "oucp" later, that would look as if swapcontext() returned 0.
|
|
// We need to clear shadow for ucp once again, as it may be in arbitrary
|
|
// state.
|
|
ClearShadowMemoryForContextStack(stack, ssize);
|
|
return res;
|
|
}
|
|
#endif // ASAN_INTERCEPT_SWAPCONTEXT
|
|
|
|
INTERCEPTOR(void, longjmp, void *env, int val) {
|
|
__asan_handle_no_return();
|
|
REAL(longjmp)(env, val);
|
|
}
|
|
|
|
#if ASAN_INTERCEPT__LONGJMP
|
|
INTERCEPTOR(void, _longjmp, void *env, int val) {
|
|
__asan_handle_no_return();
|
|
REAL(_longjmp)(env, val);
|
|
}
|
|
#endif
|
|
|
|
#if ASAN_INTERCEPT_SIGLONGJMP
|
|
INTERCEPTOR(void, siglongjmp, void *env, int val) {
|
|
__asan_handle_no_return();
|
|
REAL(siglongjmp)(env, val);
|
|
}
|
|
#endif
|
|
|
|
#if ASAN_INTERCEPT___CXA_THROW
|
|
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(int, mlock, const void *addr, uptr len) {
|
|
MlockIsUnsupported();
|
|
return 0;
|
|
}
|
|
|
|
INTERCEPTOR(int, munlock, const void *addr, uptr len) {
|
|
MlockIsUnsupported();
|
|
return 0;
|
|
}
|
|
|
|
INTERCEPTOR(int, mlockall, int flags) {
|
|
MlockIsUnsupported();
|
|
return 0;
|
|
}
|
|
|
|
INTERCEPTOR(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) {
|
|
if (!asan_inited) return internal_memcmp(a1, a2, size);
|
|
ENSURE_ASAN_INITED();
|
|
if (flags()->replace_intrin) {
|
|
// We check the entire regions even if the first bytes of the buffers
|
|
// are different.
|
|
ASAN_READ_RANGE(a1, size);
|
|
ASAN_READ_RANGE(a2, size);
|
|
}
|
|
return REAL(memcmp(a1, a2, size));
|
|
}
|
|
|
|
INTERCEPTOR(void*, memcpy, void *to, const void *from, uptr size) {
|
|
if (!asan_inited) return internal_memcpy(to, from, 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 (flags()->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_READ_RANGE(from, size);
|
|
ASAN_WRITE_RANGE(to, size);
|
|
}
|
|
// Interposing of resolver functions is broken on Mac OS 10.7 and 10.8.
|
|
// See also http://code.google.com/p/address-sanitizer/issues/detail?id=116.
|
|
return internal_memcpy(to, from, size);
|
|
}
|
|
|
|
INTERCEPTOR(void*, memmove, void *to, const void *from, uptr size) {
|
|
if (!asan_inited) return internal_memmove(to, from, size);
|
|
if (asan_init_is_running) {
|
|
return REAL(memmove)(to, from, size);
|
|
}
|
|
ENSURE_ASAN_INITED();
|
|
if (flags()->replace_intrin) {
|
|
ASAN_READ_RANGE(from, size);
|
|
ASAN_WRITE_RANGE(to, size);
|
|
}
|
|
// Interposing of resolver functions is broken on Mac OS 10.7 and 10.8.
|
|
// See also http://code.google.com/p/address-sanitizer/issues/detail?id=116.
|
|
return internal_memmove(to, from, size);
|
|
}
|
|
|
|
INTERCEPTOR(void*, memset, void *block, int c, uptr size) {
|
|
if (!asan_inited) return internal_memset(block, c, size);
|
|
// memset is called inside Printf.
|
|
if (asan_init_is_running) {
|
|
return REAL(memset)(block, c, size);
|
|
}
|
|
ENSURE_ASAN_INITED();
|
|
if (flags()->replace_intrin) {
|
|
ASAN_WRITE_RANGE(block, size);
|
|
}
|
|
return REAL(memset)(block, c, size);
|
|
}
|
|
|
|
INTERCEPTOR(char*, strchr, const char *str, int c) {
|
|
if (!asan_inited) return internal_strchr(str, c);
|
|
// strchr is called inside create_purgeable_zone() when MallocGuardEdges=1 is
|
|
// used.
|
|
if (asan_init_is_running) {
|
|
return REAL(strchr)(str, c);
|
|
}
|
|
ENSURE_ASAN_INITED();
|
|
char *result = REAL(strchr)(str, c);
|
|
if (flags()->replace_str) {
|
|
uptr bytes_read = (result ? result - str : REAL(strlen)(str)) + 1;
|
|
ASAN_READ_RANGE(str, bytes_read);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
#if ASAN_INTERCEPT_INDEX
|
|
# if ASAN_USE_ALIAS_ATTRIBUTE_FOR_INDEX
|
|
INTERCEPTOR(char*, index, const char *string, int c)
|
|
ALIAS(WRAPPER_NAME(strchr));
|
|
# else
|
|
# if defined(__APPLE__)
|
|
DECLARE_REAL(char*, index, const char *string, int c)
|
|
OVERRIDE_FUNCTION(index, strchr);
|
|
# else
|
|
DEFINE_REAL(char*, index, const char *string, int c);
|
|
# endif
|
|
# endif
|
|
#endif // ASAN_INTERCEPT_INDEX
|
|
|
|
// For both strcat() and strncat() we need to check the validity of |to|
|
|
// argument irrespective of the |from| length.
|
|
INTERCEPTOR(char*, strcat, char *to, const char *from) { // NOLINT
|
|
ENSURE_ASAN_INITED();
|
|
if (flags()->replace_str) {
|
|
uptr from_length = REAL(strlen)(from);
|
|
ASAN_READ_RANGE(from, from_length + 1);
|
|
uptr to_length = REAL(strlen)(to);
|
|
ASAN_READ_RANGE(to, to_length);
|
|
ASAN_WRITE_RANGE(to + to_length, from_length + 1);
|
|
// If the copying actually happens, the |from| string should not overlap
|
|
// with the resulting string starting at |to|, which has a length of
|
|
// to_length + from_length + 1.
|
|
if (from_length > 0) {
|
|
CHECK_RANGES_OVERLAP("strcat", to, from_length + to_length + 1,
|
|
from, from_length + 1);
|
|
}
|
|
}
|
|
return REAL(strcat)(to, from); // NOLINT
|
|
}
|
|
|
|
INTERCEPTOR(char*, strncat, char *to, const char *from, uptr size) {
|
|
ENSURE_ASAN_INITED();
|
|
if (flags()->replace_str) {
|
|
uptr from_length = MaybeRealStrnlen(from, size);
|
|
uptr copy_length = Min(size, from_length + 1);
|
|
ASAN_READ_RANGE(from, copy_length);
|
|
uptr to_length = REAL(strlen)(to);
|
|
ASAN_READ_RANGE(to, to_length);
|
|
ASAN_WRITE_RANGE(to + to_length, from_length + 1);
|
|
if (from_length > 0) {
|
|
CHECK_RANGES_OVERLAP("strncat", to, to_length + copy_length + 1,
|
|
from, copy_length);
|
|
}
|
|
}
|
|
return REAL(strncat)(to, from, size);
|
|
}
|
|
|
|
INTERCEPTOR(int, strcmp, const char *s1, const char *s2) {
|
|
if (!asan_inited) return internal_strcmp(s1, s2);
|
|
if (asan_init_is_running) {
|
|
return REAL(strcmp)(s1, 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 (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
|
|
#if defined(__APPLE__)
|
|
if (!asan_inited) return REAL(strcpy)(to, from); // NOLINT
|
|
#endif
|
|
// 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 (flags()->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
|
|
}
|
|
|
|
#if ASAN_INTERCEPT_STRDUP
|
|
INTERCEPTOR(char*, strdup, const char *s) {
|
|
#if defined(__APPLE__)
|
|
// FIXME: because internal_strdup() uses InternalAlloc(), which currently
|
|
// just calls malloc() on Mac, we can't use internal_strdup() with the
|
|
// dynamic runtime. We can remove the call to REAL(strdup) once InternalAlloc
|
|
// starts using mmap() instead.
|
|
// See also http://code.google.com/p/address-sanitizer/issues/detail?id=123.
|
|
if (!asan_inited) return REAL(strdup)(s);
|
|
#endif
|
|
if (!asan_inited) return internal_strdup(s);
|
|
ENSURE_ASAN_INITED();
|
|
if (flags()->replace_str) {
|
|
uptr length = REAL(strlen)(s);
|
|
ASAN_READ_RANGE(s, length + 1);
|
|
}
|
|
return REAL(strdup)(s);
|
|
}
|
|
#endif
|
|
|
|
INTERCEPTOR(uptr, strlen, const char *s) {
|
|
if (!asan_inited) return internal_strlen(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 (flags()->replace_str) {
|
|
ASAN_READ_RANGE(s, length + 1);
|
|
}
|
|
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) {
|
|
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);
|
|
}
|
|
#endif // ASAN_INTERCEPT_STRCASECMP_AND_STRNCASECMP
|
|
|
|
INTERCEPTOR(int, strncmp, const char *s1, const char *s2, uptr size) {
|
|
if (!asan_inited) return internal_strncmp(s1, s2, 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);
|
|
}
|
|
ENSURE_ASAN_INITED();
|
|
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 (flags()->replace_str) {
|
|
uptr from_size = Min(size, MaybeRealStrnlen(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 (flags()->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 (!flags()->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) {
|
|
#if defined(__APPLE__)
|
|
if (!asan_inited) return REAL(atoi)(nptr);
|
|
#endif
|
|
ENSURE_ASAN_INITED();
|
|
if (!flags()->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
|
|
#if defined(__APPLE__)
|
|
if (!asan_inited) return REAL(atol)(nptr);
|
|
#endif
|
|
ENSURE_ASAN_INITED();
|
|
if (!flags()->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 (!flags()->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 (!flags()->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) && flags()->verbosity > 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_THREAD;
|
|
u32 current_tid = asanThreadRegistry().GetCurrentTidOrInvalid();
|
|
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;
|
|
#if defined(__APPLE__)
|
|
return;
|
|
#else
|
|
SANITIZER_COMMON_INTERCEPTORS_INIT;
|
|
|
|
// 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);
|
|
}
|
|
|
|
// 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(strncat);
|
|
ASAN_INTERCEPT_FUNC(strncmp);
|
|
ASAN_INTERCEPT_FUNC(strncpy);
|
|
#if ASAN_INTERCEPT_STRCASECMP_AND_STRNCASECMP
|
|
ASAN_INTERCEPT_FUNC(strcasecmp);
|
|
ASAN_INTERCEPT_FUNC(strncasecmp);
|
|
#endif
|
|
#if ASAN_INTERCEPT_STRDUP
|
|
ASAN_INTERCEPT_FUNC(strdup);
|
|
#endif
|
|
#if ASAN_INTERCEPT_STRNLEN
|
|
ASAN_INTERCEPT_FUNC(strnlen);
|
|
#endif
|
|
#if ASAN_INTERCEPT_INDEX && ASAN_USE_ALIAS_ATTRIBUTE_FOR_INDEX
|
|
ASAN_INTERCEPT_FUNC(index);
|
|
#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
|
|
|
|
#if ASAN_INTERCEPT_MLOCKX
|
|
// Intercept mlock/munlock.
|
|
ASAN_INTERCEPT_FUNC(mlock);
|
|
ASAN_INTERCEPT_FUNC(munlock);
|
|
ASAN_INTERCEPT_FUNC(mlockall);
|
|
ASAN_INTERCEPT_FUNC(munlockall);
|
|
#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 ASAN_INTERCEPT_SWAPCONTEXT
|
|
ASAN_INTERCEPT_FUNC(swapcontext);
|
|
#endif
|
|
#if ASAN_INTERCEPT__LONGJMP
|
|
ASAN_INTERCEPT_FUNC(_longjmp);
|
|
#endif
|
|
#if ASAN_INTERCEPT_SIGLONGJMP
|
|
ASAN_INTERCEPT_FUNC(siglongjmp);
|
|
#endif
|
|
|
|
// Intercept exception handling functions.
|
|
#if ASAN_INTERCEPT___CXA_THROW
|
|
INTERCEPT_FUNCTION(__cxa_throw);
|
|
#endif
|
|
|
|
// Intercept threading-related functions
|
|
#if ASAN_INTERCEPT_PTHREAD_CREATE
|
|
ASAN_INTERCEPT_FUNC(pthread_create);
|
|
#endif
|
|
|
|
// Some Windows-specific interceptors.
|
|
#if defined(_WIN32)
|
|
InitializeWindowsInterceptors();
|
|
#endif
|
|
|
|
if (flags()->verbosity > 0) {
|
|
Report("AddressSanitizer: libc interceptors initialized\n");
|
|
}
|
|
#endif // __APPLE__
|
|
}
|
|
|
|
} // namespace __asan
|