1e5f5895bda30f374b0b51412fd4d837fa59aed66Alexey Samsonov//===-- asan_interceptors.cc ----------------------------------------------===//
21e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
31e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//                     The LLVM Compiler Infrastructure
41e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
51e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// This file is distributed under the University of Illinois Open Source
61e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// License. See LICENSE.TXT for details.
71e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
81e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//===----------------------------------------------------------------------===//
91e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// This file is a part of AddressSanitizer, an address sanity checker.
111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//
12d47189c1b63a4de755d7ec071f8d56c8d01cc667Kostya Serebryany// Intercept various libc functions.
131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//===----------------------------------------------------------------------===//
141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_interceptors.h"
151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_allocator.h"
1705bf9a51dfeac0f84f6dbc2dacf987249c0fc612Alexander Potapenko#include "asan_intercepted_functions.h"
181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_internal.h"
191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_mapping.h"
20487fee7f6f7497906a00d7d2fe2c75e6d5d4feb1Alexey Samsonov#include "asan_report.h"
211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_stack.h"
221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_stats.h"
234803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany#include "asan_thread_registry.h"
245b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov#include "interception/interception.h"
25c0d78c1de1f2607c874020d27b72cf989c5ce092Alexey Samsonov#include "sanitizer_common/sanitizer_libc.h"
26c0d78c1de1f2607c874020d27b72cf989c5ce092Alexey Samsonov
271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanynamespace __asan {
281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
29a84805f1ccb2b7d4b6c0ba384bd3541fa4eaf808Kostya Serebryany// Return true if we can quickly decide that the region is unpoisoned.
30a84805f1ccb2b7d4b6c0ba384bd3541fa4eaf808Kostya Serebryanystatic inline bool QuickCheckForUnpoisonedRegion(uptr beg, uptr size) {
31a84805f1ccb2b7d4b6c0ba384bd3541fa4eaf808Kostya Serebryany  if (size == 0) return true;
32a84805f1ccb2b7d4b6c0ba384bd3541fa4eaf808Kostya Serebryany  if (size <= 32)
33a84805f1ccb2b7d4b6c0ba384bd3541fa4eaf808Kostya Serebryany    return !AddressIsPoisoned(beg) &&
34a84805f1ccb2b7d4b6c0ba384bd3541fa4eaf808Kostya Serebryany           !AddressIsPoisoned(beg + size - 1) &&
35a84805f1ccb2b7d4b6c0ba384bd3541fa4eaf808Kostya Serebryany           !AddressIsPoisoned(beg + size / 2);
36a84805f1ccb2b7d4b6c0ba384bd3541fa4eaf808Kostya Serebryany  return false;
37a84805f1ccb2b7d4b6c0ba384bd3541fa4eaf808Kostya Serebryany}
38a84805f1ccb2b7d4b6c0ba384bd3541fa4eaf808Kostya Serebryany
391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// We implement ACCESS_MEMORY_RANGE, ASAN_READ_RANGE,
401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// and ASAN_WRITE_RANGE as macro instead of function so
411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// that no extra frames are created, and stack trace contains
421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// relevant information only.
43eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany// We check all shadow bytes.
44589dcdaa520de1033a0f6112c9b67ab9eb7931afEvgeniy Stepanov#define ACCESS_MEMORY_RANGE(offset, size, isWrite) do {                 \
45589dcdaa520de1033a0f6112c9b67ab9eb7931afEvgeniy Stepanov    uptr __offset = (uptr)(offset);                                     \
46589dcdaa520de1033a0f6112c9b67ab9eb7931afEvgeniy Stepanov    uptr __size = (uptr)(size);                                         \
471b057b20db71ec6cc2bac460c8b2848f0889d47dKostya Serebryany    uptr __bad = 0;                                                     \
48a84805f1ccb2b7d4b6c0ba384bd3541fa4eaf808Kostya Serebryany    if (!QuickCheckForUnpoisonedRegion(__offset, __size) &&             \
491b057b20db71ec6cc2bac460c8b2848f0889d47dKostya Serebryany        (__bad = __asan_region_is_poisoned(__offset, __size))) {        \
50589dcdaa520de1033a0f6112c9b67ab9eb7931afEvgeniy Stepanov      GET_CURRENT_PC_BP_SP;                                             \
511b057b20db71ec6cc2bac460c8b2848f0889d47dKostya Serebryany      __asan_report_error(pc, bp, sp, __bad, isWrite, __size);          \
52589dcdaa520de1033a0f6112c9b67ab9eb7931afEvgeniy Stepanov    }                                                                   \
53589dcdaa520de1033a0f6112c9b67ab9eb7931afEvgeniy Stepanov  } while (0)
541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
55eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany#define ASAN_READ_RANGE(offset, size) ACCESS_MEMORY_RANGE(offset, size, false)
56eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany#define ASAN_WRITE_RANGE(offset, size) ACCESS_MEMORY_RANGE(offset, size, true);
571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Behavior of functions like "memcpy" or "strcpy" is undefined
591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// if memory intervals overlap. We report error in this case.
601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Macro is used to avoid creation of new frames.
613f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryanystatic inline bool RangesOverlap(const char *offset1, uptr length1,
623f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany                                 const char *offset2, uptr length2) {
630985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany  return !((offset1 + length1 <= offset2) || (offset2 + length2 <= offset1));
641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
65c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryany#define CHECK_RANGES_OVERLAP(name, _offset1, length1, _offset2, length2) do { \
661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  const char *offset1 = (const char*)_offset1; \
671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  const char *offset2 = (const char*)_offset2; \
680985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany  if (RangesOverlap(offset1, length1, offset2, length2)) { \
69a30c8f9eac981dcf137e84226810b760e35c7be1Kostya Serebryany    GET_STACK_TRACE_FATAL_HERE; \
70487fee7f6f7497906a00d7d2fe2c75e6d5d4feb1Alexey Samsonov    ReportStringFunctionMemoryRangesOverlap(name, offset1, length1, \
71487fee7f6f7497906a00d7d2fe2c75e6d5d4feb1Alexey Samsonov                                            offset2, length2, &stack); \
721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  } \
73e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany} while (0)
741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
75e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany#define ENSURE_ASAN_INITED() do { \
76e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany  CHECK(!asan_init_is_running); \
77e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany  if (!asan_inited) { \
78e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany    __asan_init(); \
79e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany  } \
80e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany} while (0)
811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
82c925697df6626bb0ea27ea96539bf0580f8f3d3dAlexey Samsonovstatic inline uptr MaybeRealStrnlen(const char *s, uptr maxlen) {
8381a7a4ab53c94e76c3b131d6f0ed82894640900bAlexey Samsonov#if ASAN_INTERCEPT_STRNLEN
843f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  if (REAL(strnlen) != 0) {
8509672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov    return REAL(strnlen)(s, maxlen);
861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
87f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov#endif
88c925697df6626bb0ea27ea96539bf0580f8f3d3dAlexey Samsonov  return internal_strnlen(s, maxlen);
89a4ccf878e464d29a4a18756c5c4f626dc530a12eKostya Serebryany}
901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
91c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryanyvoid SetThreadName(const char *name) {
92c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany  AsanThread *t = asanThreadRegistry().GetCurrent();
93c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany  if (t)
94c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany    t->summary()->set_name(name);
95c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany}
96c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany
971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}  // namespace __asan
981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// ---------------------- Wrappers ---------------- {{{1
1001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyusing namespace __asan;  // NOLINT
1011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
10282a9080eaff95d69b270cd863e9df63e3b4e59adEvgeniy Stepanov#define COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ptr, size) \
10382a9080eaff95d69b270cd863e9df63e3b4e59adEvgeniy Stepanov  ASAN_WRITE_RANGE(ptr, size)
104996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov#define COMMON_INTERCEPTOR_READ_RANGE(ctx, ptr, size) ASAN_READ_RANGE(ptr, size)
105bee7415a86d9437659d09f034c346794bf15c2abEvgeniy Stepanov#define COMMON_INTERCEPTOR_ENTER(ctx, func, ...) \
10682a9080eaff95d69b270cd863e9df63e3b4e59adEvgeniy Stepanov  do {                                           \
10782a9080eaff95d69b270cd863e9df63e3b4e59adEvgeniy Stepanov    ctx = 0;                                     \
10882a9080eaff95d69b270cd863e9df63e3b4e59adEvgeniy Stepanov    (void)ctx;                                   \
10982a9080eaff95d69b270cd863e9df63e3b4e59adEvgeniy Stepanov    ENSURE_ASAN_INITED();                        \
11082a9080eaff95d69b270cd863e9df63e3b4e59adEvgeniy Stepanov  } while (false)
111996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov#define COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd) do { } while (false)
112996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov#define COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd) do { } while (false)
113996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov#define COMMON_INTERCEPTOR_SET_THREAD_NAME(ctx, name) SetThreadName(name)
1144f32c0beaa83ffbb84db23d2e6205bee57c39ce1Evgeniy Stepanov#include "sanitizer_common/sanitizer_common_interceptors.inc"
1158530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany
116600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanovstatic thread_return_t THREAD_CALLING_CONV asan_thread_start(void *arg) {
1174803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  AsanThread *t = (AsanThread*)arg;
1184803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  asanThreadRegistry().SetCurrent(t);
1194803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  return t->ThreadStart();
1204803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}
1214803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany
122fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#if ASAN_INTERCEPT_PTHREAD_CREATE
123b8ef925563f8dde5be837c4b0569082867c5f14cEvgeniy StepanovINTERCEPTOR(int, pthread_create, void *thread,
124b8ef925563f8dde5be837c4b0569082867c5f14cEvgeniy Stepanov    void *attr, void *(*start_routine)(void*), void *arg) {
125a30c8f9eac981dcf137e84226810b760e35c7be1Kostya Serebryany  GET_STACK_TRACE_THREAD;
126e0cff0bc20ae51790c8edfbceb817e18ebf5355eKostya Serebryany  u32 current_tid = asanThreadRegistry().GetCurrentTidOrInvalid();
12755cdfc6c5af92560bc0623b5a0d70af71511c3c8Alexey Samsonov  AsanThread *t = AsanThread::Create(current_tid, start_routine, arg, &stack);
12855cdfc6c5af92560bc0623b5a0d70af71511c3c8Alexey Samsonov  asanThreadRegistry().RegisterThread(t);
12909672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  return REAL(pthread_create)(thread, attr, asan_thread_start, t);
1304803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}
131fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#endif  // ASAN_INTERCEPT_PTHREAD_CREATE
1324803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany
13334a3202a2c22816a6da66959e266a2d078ded37bAlexey Samsonov#if ASAN_INTERCEPT_SIGNAL_AND_SIGACTION
134f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void*, signal, int signum, void *handler) {
135034bda5eab7cda14fac0bed3a39de4a3dbce3cffAlexander Potapenko  if (!AsanInterceptsSignal(signum)) {
136034bda5eab7cda14fac0bed3a39de4a3dbce3cffAlexander Potapenko    return REAL(signal)(signum, handler);
1374803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  }
1383f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  return 0;
1394803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}
1404803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany
141da13ba833b046553f50b54538e4828287ddbf5c0Alexey SamsonovINTERCEPTOR(int, sigaction, int signum, const struct sigaction *act,
142da13ba833b046553f50b54538e4828287ddbf5c0Alexey Samsonov                            struct sigaction *oldact) {
143034bda5eab7cda14fac0bed3a39de4a3dbce3cffAlexander Potapenko  if (!AsanInterceptsSignal(signum)) {
144034bda5eab7cda14fac0bed3a39de4a3dbce3cffAlexander Potapenko    return REAL(sigaction)(signum, act, oldact);
1454803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  }
146034bda5eab7cda14fac0bed3a39de4a3dbce3cffAlexander Potapenko  return 0;
1474803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}
14834a3202a2c22816a6da66959e266a2d078ded37bAlexey Samsonov#elif ASAN_POSIX
14934a3202a2c22816a6da66959e266a2d078ded37bAlexey Samsonov// We need to have defined REAL(sigaction) on posix systems.
15034a3202a2c22816a6da66959e266a2d078ded37bAlexey SamsonovDEFINE_REAL(int, sigaction, int signum, const struct sigaction *act,
15134a3202a2c22816a6da66959e266a2d078ded37bAlexey Samsonov    struct sigaction *oldact);
15234a3202a2c22816a6da66959e266a2d078ded37bAlexey Samsonov#endif  // ASAN_INTERCEPT_SIGNAL_AND_SIGACTION
1534803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany
1540870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov#if ASAN_INTERCEPT_SWAPCONTEXT
15557db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonovstatic void ClearShadowMemoryForContextStack(uptr stack, uptr ssize) {
15657db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonov  // Align to page size.
15757db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonov  uptr PageSize = GetPageSizeCached();
15857db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonov  uptr bottom = stack & ~(PageSize - 1);
15957db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonov  ssize += stack - bottom;
16057db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonov  ssize = RoundUpTo(ssize, PageSize);
16157db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonov  static const uptr kMaxSaneContextStackSize = 1 << 22;  // 4 Mb
162541cfb10f5daa17e48eb42365a74233cd551c545Kostya Serebryany  if (ssize && ssize <= kMaxSaneContextStackSize) {
16357db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonov    PoisonShadow(bottom, ssize, 0);
16457db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonov  }
16557db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonov}
16657db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonov
1670870028410087e67a0049c76cb7c64f02c260d24Alexey SamsonovINTERCEPTOR(int, swapcontext, struct ucontext_t *oucp,
1680870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov            struct ucontext_t *ucp) {
1690870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov  static bool reported_warning = false;
1700870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov  if (!reported_warning) {
1710870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov    Report("WARNING: ASan doesn't fully support makecontext/swapcontext "
1720870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov           "functions and may produce false positives in some cases!\n");
1730870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov    reported_warning = true;
1740870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov  }
1750870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov  // Clear shadow memory for new context (it may share stack
1760870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov  // with current context).
17757db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonov  uptr stack, ssize;
17857db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonov  ReadContextStack(ucp, &stack, &ssize);
17957db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonov  ClearShadowMemoryForContextStack(stack, ssize);
1800870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov  int res = REAL(swapcontext)(oucp, ucp);
1810870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov  // swapcontext technically does not return, but program may swap context to
1820870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov  // "oucp" later, that would look as if swapcontext() returned 0.
1830870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov  // We need to clear shadow for ucp once again, as it may be in arbitrary
1840870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov  // state.
18557db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonov  ClearShadowMemoryForContextStack(stack, ssize);
1860870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov  return res;
1870870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov}
18857db4bae483e7268dd7fc5bc42b0e55564014048Alexey Samsonov#endif  // ASAN_INTERCEPT_SWAPCONTEXT
1890870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov
190f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, longjmp, void *env, int val) {
191f54b1f9b73a6855d77ee270c282bd61407fa73a0Kostya Serebryany  __asan_handle_no_return();
19209672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  REAL(longjmp)(env, val);
1934803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}
1944803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany
195fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#if ASAN_INTERCEPT__LONGJMP
196f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, _longjmp, void *env, int val) {
197f54b1f9b73a6855d77ee270c282bd61407fa73a0Kostya Serebryany  __asan_handle_no_return();
19809672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  REAL(_longjmp)(env, val);
1994803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}
200fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#endif
2014803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany
202fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#if ASAN_INTERCEPT_SIGLONGJMP
203f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, siglongjmp, void *env, int val) {
204f54b1f9b73a6855d77ee270c282bd61407fa73a0Kostya Serebryany  __asan_handle_no_return();
20509672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  REAL(siglongjmp)(env, val);
2064803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}
20707bb9f1e3600195119aec1aae1aa48a6ed2f5febTimur Iskhodzhanov#endif
2084803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany
209fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#if ASAN_INTERCEPT___CXA_THROW
210f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(void, __cxa_throw, void *a, void *b, void *c) {
21109672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  CHECK(REAL(__cxa_throw));
212f54b1f9b73a6855d77ee270c282bd61407fa73a0Kostya Serebryany  __asan_handle_no_return();
21309672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  REAL(__cxa_throw)(a, b, c);
2144803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}
2154803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany#endif
2164803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany
2174803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany// intercept mlock and friends.
2184803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany// Since asan maps 16T of RAM, mlock is completely unfriendly to asan.
2194803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany// All functions return 0 (success).
2204803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryanystatic void MlockIsUnsupported() {
2214803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  static bool printed = 0;
2224803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  if (printed) return;
2234803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  printed = true;
2244803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  Printf("INFO: AddressSanitizer ignores mlock/mlockall/munlock/munlockall\n");
2254803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}
2263389b8ea81f51a28365a421b2646d594bac490f5Alexey Samsonov
227f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonovextern "C" {
22800f1c091441be5c2cacb99e21bd1965a5e418242Alexander PotapenkoINTERCEPTOR(int, mlock, const void *addr, uptr len) {
2294803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  MlockIsUnsupported();
2304803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  return 0;
2314803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}
2323389b8ea81f51a28365a421b2646d594bac490f5Alexey Samsonov
23300f1c091441be5c2cacb99e21bd1965a5e418242Alexander PotapenkoINTERCEPTOR(int, munlock, const void *addr, uptr len) {
2344803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  MlockIsUnsupported();
2354803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  return 0;
2364803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}
2373389b8ea81f51a28365a421b2646d594bac490f5Alexey Samsonov
23800f1c091441be5c2cacb99e21bd1965a5e418242Alexander PotapenkoINTERCEPTOR(int, mlockall, int flags) {
2394803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  MlockIsUnsupported();
2404803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  return 0;
2414803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}
2423389b8ea81f51a28365a421b2646d594bac490f5Alexey Samsonov
24300f1c091441be5c2cacb99e21bd1965a5e418242Alexander PotapenkoINTERCEPTOR(int, munlockall, void) {
2444803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  MlockIsUnsupported();
2454803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany  return 0;
2464803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}
2474803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany}  // extern "C"
2484803ab90ead451b55a5833f0fd38b10fd1fc83ebKostya Serebryany
24952fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryanystatic inline int CharCmp(unsigned char c1, unsigned char c2) {
25052fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany  return (c1 == c2) ? 0 : (c1 < c2) ? -1 : 1;
25152fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany}
25252fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany
25352fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryanystatic inline int CharCaseCmp(unsigned char c1, unsigned char c2) {
2548898102a6b1479a9af65857ccbbc1c329a4d9187Alexey Samsonov  int c1_low = ToLower(c1);
2558898102a6b1479a9af65857ccbbc1c329a4d9187Alexey Samsonov  int c2_low = ToLower(c2);
25652fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany  return c1_low - c2_low;
25752fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany}
25852fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany
2593f4c3875c42078e22c7e5356c5746fd18756d958Kostya SerebryanyINTERCEPTOR(int, memcmp, const void *a1, const void *a2, uptr size) {
26050f2e30497ba2a69ee5721e7235a296d7c13a495Alexander Potapenko  if (!asan_inited) return internal_memcmp(a1, a2, size);
26152fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany  ENSURE_ASAN_INITED();
2621b057b20db71ec6cc2bac460c8b2848f0889d47dKostya Serebryany  if (flags()->replace_intrin) {
2638bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko    if (flags()->strict_memcmp) {
2648bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko      // Check the entire regions even if the first bytes of the buffers are
2658bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko      // different.
2668bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko      ASAN_READ_RANGE(a1, size);
2678bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko      ASAN_READ_RANGE(a2, size);
2688bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko      // Fallthrough to REAL(memcmp) below.
2698bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko    } else {
2708bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko      unsigned char c1 = 0, c2 = 0;
2718bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko      const unsigned char *s1 = (const unsigned char*)a1;
2728bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko      const unsigned char *s2 = (const unsigned char*)a2;
2738bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko      uptr i;
2748bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko      for (i = 0; i < size; i++) {
2758bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko        c1 = s1[i];
2768bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko        c2 = s2[i];
2778bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko        if (c1 != c2) break;
2788bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko      }
2798bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko      ASAN_READ_RANGE(s1, Min(i + 1, size));
2808bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko      ASAN_READ_RANGE(s2, Min(i + 1, size));
2818bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko      return CharCmp(c1, c2);
2828bd5e74fa9d37a182088114918380e255e22e493Alexander Potapenko    }
28352fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany  }
2841b057b20db71ec6cc2bac460c8b2848f0889d47dKostya Serebryany  return REAL(memcmp(a1, a2, size));
28552fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany}
28652fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany
2873f4c3875c42078e22c7e5356c5746fd18756d958Kostya SerebryanyINTERCEPTOR(void*, memcpy, void *to, const void *from, uptr size) {
28850f2e30497ba2a69ee5721e7235a296d7c13a495Alexander Potapenko  if (!asan_inited) return internal_memcpy(to, from, size);
2891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  // memcpy is called during __asan_init() from the internals
2901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  // of printf(...).
2911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (asan_init_is_running) {
29209672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov    return REAL(memcpy)(to, from, size);
2931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
294e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany  ENSURE_ASAN_INITED();
295cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (flags()->replace_intrin) {
296c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany    if (to != from) {
297c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany      // We do not treat memcpy with to==from as a bug.
298c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany      // See http://llvm.org/bugs/show_bug.cgi?id=11763.
299c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany      CHECK_RANGES_OVERLAP("memcpy", to, size, from, size);
300c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany    }
301c75d8483e27528349620ccf4e997a72029a7ea17Alexander Potapenko    ASAN_READ_RANGE(from, size);
302c75d8483e27528349620ccf4e997a72029a7ea17Alexander Potapenko    ASAN_WRITE_RANGE(to, size);
3031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
304f1673e61768cb08d488a75584b61bfdf24ffdac1Alexander Potapenko  // Interposing of resolver functions is broken on Mac OS 10.7 and 10.8.
305f1673e61768cb08d488a75584b61bfdf24ffdac1Alexander Potapenko  // See also http://code.google.com/p/address-sanitizer/issues/detail?id=116.
306f1673e61768cb08d488a75584b61bfdf24ffdac1Alexander Potapenko  return internal_memcpy(to, from, size);
3071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
3081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
3093f4c3875c42078e22c7e5356c5746fd18756d958Kostya SerebryanyINTERCEPTOR(void*, memmove, void *to, const void *from, uptr size) {
3103e8458ab54c7cbaec45e367b2da909c6f94b291cAlexander Potapenko  if (!asan_inited) return internal_memmove(to, from, size);
311a1a25c1afe38e59ffb342c24bb5bb10d05679df7Alexey Samsonov  if (asan_init_is_running) {
312a1a25c1afe38e59ffb342c24bb5bb10d05679df7Alexey Samsonov    return REAL(memmove)(to, from, size);
313a1a25c1afe38e59ffb342c24bb5bb10d05679df7Alexey Samsonov  }
314e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany  ENSURE_ASAN_INITED();
315cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (flags()->replace_intrin) {
316c75d8483e27528349620ccf4e997a72029a7ea17Alexander Potapenko    ASAN_READ_RANGE(from, size);
317c75d8483e27528349620ccf4e997a72029a7ea17Alexander Potapenko    ASAN_WRITE_RANGE(to, size);
3181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
319f1673e61768cb08d488a75584b61bfdf24ffdac1Alexander Potapenko  // Interposing of resolver functions is broken on Mac OS 10.7 and 10.8.
320f1673e61768cb08d488a75584b61bfdf24ffdac1Alexander Potapenko  // See also http://code.google.com/p/address-sanitizer/issues/detail?id=116.
321f1673e61768cb08d488a75584b61bfdf24ffdac1Alexander Potapenko  return internal_memmove(to, from, size);
3221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
3231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
3243f4c3875c42078e22c7e5356c5746fd18756d958Kostya SerebryanyINTERCEPTOR(void*, memset, void *block, int c, uptr size) {
32550f2e30497ba2a69ee5721e7235a296d7c13a495Alexander Potapenko  if (!asan_inited) return internal_memset(block, c, size);
326ebb9702cff96192c6a6ea963037929ca7ed60eaeAlexander Potapenko  // memset is called inside Printf.
327e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany  if (asan_init_is_running) {
32809672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov    return REAL(memset)(block, c, size);
329e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany  }
330e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany  ENSURE_ASAN_INITED();
331cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (flags()->replace_intrin) {
3321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    ASAN_WRITE_RANGE(block, size);
3331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
33409672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  return REAL(memset)(block, c, size);
3351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
3361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
337f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(char*, strchr, const char *str, int c) {
33850f2e30497ba2a69ee5721e7235a296d7c13a495Alexander Potapenko  if (!asan_inited) return internal_strchr(str, c);
3398d6e3f7d011970e35dadf831964260bcf6a4d8a9Alexander Potapenko  // strchr is called inside create_purgeable_zone() when MallocGuardEdges=1 is
3408d6e3f7d011970e35dadf831964260bcf6a4d8a9Alexander Potapenko  // used.
3418d6e3f7d011970e35dadf831964260bcf6a4d8a9Alexander Potapenko  if (asan_init_is_running) {
3428d6e3f7d011970e35dadf831964260bcf6a4d8a9Alexander Potapenko    return REAL(strchr)(str, c);
3438d6e3f7d011970e35dadf831964260bcf6a4d8a9Alexander Potapenko  }
344e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany  ENSURE_ASAN_INITED();
34509672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  char *result = REAL(strchr)(str, c);
346cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (flags()->replace_str) {
3473f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany    uptr bytes_read = (result ? result - str : REAL(strlen)(str)) + 1;
3481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    ASAN_READ_RANGE(str, bytes_read);
3491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
3501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  return result;
3511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
3521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
353fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#if ASAN_INTERCEPT_INDEX
354fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov# if ASAN_USE_ALIAS_ATTRIBUTE_FOR_INDEX
3554b0c5f240e71be4f375627e57d0f1b89600918a7Evgeniy StepanovINTERCEPTOR(char*, index, const char *string, int c)
35672bbfd4f3a442afde281476bc6bd41b3b2152a4eAlexander Potapenko  ALIAS(WRAPPER_NAME(strchr));
357fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov# else
35850a002ecad3f0a10c136496e5b6289bd3c71590eAlexander Potapenko#  if defined(__APPLE__)
35950a002ecad3f0a10c136496e5b6289bd3c71590eAlexander PotapenkoDECLARE_REAL(char*, index, const char *string, int c)
36050a002ecad3f0a10c136496e5b6289bd3c71590eAlexander PotapenkoOVERRIDE_FUNCTION(index, strchr);
36150a002ecad3f0a10c136496e5b6289bd3c71590eAlexander Potapenko#  else
36250a002ecad3f0a10c136496e5b6289bd3c71590eAlexander PotapenkoDEFINE_REAL(char*, index, const char *string, int c);
36350a002ecad3f0a10c136496e5b6289bd3c71590eAlexander Potapenko#  endif
364fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov# endif
365fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#endif  // ASAN_INTERCEPT_INDEX
366af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany
36737b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko// For both strcat() and strncat() we need to check the validity of |to|
36837b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko// argument irrespective of the |from| length.
369f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(char*, strcat, char *to, const char *from) {  // NOLINT
3700985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany  ENSURE_ASAN_INITED();
371cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (flags()->replace_str) {
3723f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany    uptr from_length = REAL(strlen)(from);
3730985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany    ASAN_READ_RANGE(from, from_length + 1);
37437b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko    uptr to_length = REAL(strlen)(to);
37537b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko    ASAN_READ_RANGE(to, to_length);
37637b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko    ASAN_WRITE_RANGE(to + to_length, from_length + 1);
37737b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko    // If the copying actually happens, the |from| string should not overlap
37837b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko    // with the resulting string starting at |to|, which has a length of
37937b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko    // to_length + from_length + 1.
3800985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany    if (from_length > 0) {
38137b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko      CHECK_RANGES_OVERLAP("strcat", to, from_length + to_length + 1,
38237b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko                           from, from_length + 1);
3830985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany    }
3840985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany  }
38509672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  return REAL(strcat)(to, from);  // NOLINT
3860985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany}
3870985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany
388c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey SamsonovINTERCEPTOR(char*, strncat, char *to, const char *from, uptr size) {
389c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov  ENSURE_ASAN_INITED();
39037b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko  if (flags()->replace_str) {
391c925697df6626bb0ea27ea96539bf0580f8f3d3dAlexey Samsonov    uptr from_length = MaybeRealStrnlen(from, size);
39237b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko    uptr copy_length = Min(size, from_length + 1);
39337b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko    ASAN_READ_RANGE(from, copy_length);
394c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov    uptr to_length = REAL(strlen)(to);
395c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov    ASAN_READ_RANGE(to, to_length);
396c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov    ASAN_WRITE_RANGE(to + to_length, from_length + 1);
397c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov    if (from_length > 0) {
39837b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko      CHECK_RANGES_OVERLAP("strncat", to, to_length + copy_length + 1,
39937b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko                           from, copy_length);
400c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov    }
401c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov  }
402c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov  return REAL(strncat)(to, from, size);
403c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov}
404c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov
405f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(int, strcmp, const char *s1, const char *s2) {
40650f2e30497ba2a69ee5721e7235a296d7c13a495Alexander Potapenko  if (!asan_inited) return internal_strcmp(s1, s2);
407fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov  if (asan_init_is_running) {
408fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov    return REAL(strcmp)(s1, s2);
4091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
4108648e7734ff02a25706464e9e294bd217622bcaeKostya Serebryany  ENSURE_ASAN_INITED();
4111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  unsigned char c1, c2;
4123f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  uptr i;
4131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  for (i = 0; ; i++) {
4141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    c1 = (unsigned char)s1[i];
4151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    c2 = (unsigned char)s2[i];
4161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    if (c1 != c2 || c1 == '\0') break;
4171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
4181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  ASAN_READ_RANGE(s1, i + 1);
4191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  ASAN_READ_RANGE(s2, i + 1);
4201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  return CharCmp(c1, c2);
4211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
4221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
423f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(char*, strcpy, char *to, const char *from) {  // NOLINT
42469563986ca570ce750111a82264d51ddbf4107baAlexander Potapenko#if defined(__APPLE__)
425beda44f87166f91dbeef3d174d77395653bead4dAlexander Potapenko  if (!asan_inited) return REAL(strcpy)(to, from);  // NOLINT
4260ef531054fc25c11e372cbab1384f10954984219Alexander Potapenko#endif
4271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  // strcpy is called from malloc_default_purgeable_zone()
4281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  // in __asan::ReplaceSystemAlloc() on Mac.
4291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (asan_init_is_running) {
43009672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov    return REAL(strcpy)(to, from);  // NOLINT
4311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
432e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany  ENSURE_ASAN_INITED();
433cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (flags()->replace_str) {
4343f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany    uptr from_size = REAL(strlen)(from) + 1;
435c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryany    CHECK_RANGES_OVERLAP("strcpy", to, from_size, from, from_size);
4361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    ASAN_READ_RANGE(from, from_size);
4371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    ASAN_WRITE_RANGE(to, from_size);
4381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
43909672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  return REAL(strcpy)(to, from);  // NOLINT
4401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
4411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
442fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#if ASAN_INTERCEPT_STRDUP
443f2598fc21bf651d23feab396a7581d48c01c3be5Alexey SamsonovINTERCEPTOR(char*, strdup, const char *s) {
44469563986ca570ce750111a82264d51ddbf4107baAlexander Potapenko#if defined(__APPLE__)
44571578faf4eb8e61aefb3121e1945cc47f07cea0dAlexander Potapenko  // FIXME: because internal_strdup() uses InternalAlloc(), which currently
44671578faf4eb8e61aefb3121e1945cc47f07cea0dAlexander Potapenko  // just calls malloc() on Mac, we can't use internal_strdup() with the
44771578faf4eb8e61aefb3121e1945cc47f07cea0dAlexander Potapenko  // dynamic runtime. We can remove the call to REAL(strdup) once InternalAlloc
44871578faf4eb8e61aefb3121e1945cc47f07cea0dAlexander Potapenko  // starts using mmap() instead.
44971578faf4eb8e61aefb3121e1945cc47f07cea0dAlexander Potapenko  // See also http://code.google.com/p/address-sanitizer/issues/detail?id=123.
45071578faf4eb8e61aefb3121e1945cc47f07cea0dAlexander Potapenko  if (!asan_inited) return REAL(strdup)(s);
45171578faf4eb8e61aefb3121e1945cc47f07cea0dAlexander Potapenko#endif
45250f2e30497ba2a69ee5721e7235a296d7c13a495Alexander Potapenko  if (!asan_inited) return internal_strdup(s);
453e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany  ENSURE_ASAN_INITED();
454cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (flags()->replace_str) {
4553f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany    uptr length = REAL(strlen)(s);
4561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    ASAN_READ_RANGE(s, length + 1);
4571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
45809672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  return REAL(strdup)(s);
4591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
460fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#endif
4611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
4623f4c3875c42078e22c7e5356c5746fd18756d958Kostya SerebryanyINTERCEPTOR(uptr, strlen, const char *s) {
46350f2e30497ba2a69ee5721e7235a296d7c13a495Alexander Potapenko  if (!asan_inited) return internal_strlen(s);
4641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  // strlen is called from malloc_default_purgeable_zone()
4651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  // in __asan::ReplaceSystemAlloc() on Mac.
4661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (asan_init_is_running) {
46709672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov    return REAL(strlen)(s);
4681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
469e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany  ENSURE_ASAN_INITED();
4703f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  uptr length = REAL(strlen)(s);
471cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (flags()->replace_str) {
4721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    ASAN_READ_RANGE(s, length + 1);
4731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
4741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  return length;
4751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
4761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
477fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#if ASAN_INTERCEPT_STRCASECMP_AND_STRNCASECMP
478fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey SamsonovINTERCEPTOR(int, strcasecmp, const char *s1, const char *s2) {
479fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov  ENSURE_ASAN_INITED();
480fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov  unsigned char c1, c2;
481fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov  uptr i;
482fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov  for (i = 0; ; i++) {
483fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov    c1 = (unsigned char)s1[i];
484fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov    c2 = (unsigned char)s2[i];
485fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov    if (CharCaseCmp(c1, c2) != 0 || c1 == '\0') break;
486fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov  }
487fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov  ASAN_READ_RANGE(s1, i + 1);
488fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov  ASAN_READ_RANGE(s2, i + 1);
489fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov  return CharCaseCmp(c1, c2);
490fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov}
491fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov
4923f4c3875c42078e22c7e5356c5746fd18756d958Kostya SerebryanyINTERCEPTOR(int, strncasecmp, const char *s1, const char *s2, uptr n) {
493af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany  ENSURE_ASAN_INITED();
494af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany  unsigned char c1 = 0, c2 = 0;
4953f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  uptr i;
496f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov  for (i = 0; i < n; i++) {
497af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany    c1 = (unsigned char)s1[i];
498af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany    c2 = (unsigned char)s2[i];
499af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany    if (CharCaseCmp(c1, c2) != 0 || c1 == '\0') break;
500af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany  }
501f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov  ASAN_READ_RANGE(s1, Min(i + 1, n));
502f2598fc21bf651d23feab396a7581d48c01c3be5Alexey Samsonov  ASAN_READ_RANGE(s2, Min(i + 1, n));
503af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany  return CharCaseCmp(c1, c2);
504af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany}
505fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#endif  // ASAN_INTERCEPT_STRCASECMP_AND_STRNCASECMP
506af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany
5073f4c3875c42078e22c7e5356c5746fd18756d958Kostya SerebryanyINTERCEPTOR(int, strncmp, const char *s1, const char *s2, uptr size) {
50850f2e30497ba2a69ee5721e7235a296d7c13a495Alexander Potapenko  if (!asan_inited) return internal_strncmp(s1, s2, size);
5091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  // strncmp is called from malloc_default_purgeable_zone()
5101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  // in __asan::ReplaceSystemAlloc() on Mac.
5111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  if (asan_init_is_running) {
51209672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov    return REAL(strncmp)(s1, s2, size);
5131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
5148648e7734ff02a25706464e9e294bd217622bcaeKostya Serebryany  ENSURE_ASAN_INITED();
5151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  unsigned char c1 = 0, c2 = 0;
5163f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  uptr i;
5171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  for (i = 0; i < size; i++) {
5181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    c1 = (unsigned char)s1[i];
5191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    c2 = (unsigned char)s2[i];
5201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    if (c1 != c2 || c1 == '\0') break;
5211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
5222d8b3bdb112ebb8ed3f15ee41d4cebcd683b41b0Kostya Serebryany  ASAN_READ_RANGE(s1, Min(i + 1, size));
5232d8b3bdb112ebb8ed3f15ee41d4cebcd683b41b0Kostya Serebryany  ASAN_READ_RANGE(s2, Min(i + 1, size));
5241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  return CharCmp(c1, c2);
5251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
5261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
5273f4c3875c42078e22c7e5356c5746fd18756d958Kostya SerebryanyINTERCEPTOR(char*, strncpy, char *to, const char *from, uptr size) {
528e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany  ENSURE_ASAN_INITED();
529cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (flags()->replace_str) {
530c925697df6626bb0ea27ea96539bf0580f8f3d3dAlexey Samsonov    uptr from_size = Min(size, MaybeRealStrnlen(from, size) + 1);
531c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryany    CHECK_RANGES_OVERLAP("strncpy", to, from_size, from, from_size);
5321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    ASAN_READ_RANGE(from, from_size);
5331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany    ASAN_WRITE_RANGE(to, size);
5341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
53509672caefb5694f1981a1712fdefa44840a95e67Alexey Samsonov  return REAL(strncpy)(to, from, size);
5361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
5371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany
53881a7a4ab53c94e76c3b131d6f0ed82894640900bAlexey Samsonov#if ASAN_INTERCEPT_STRNLEN
5393f4c3875c42078e22c7e5356c5746fd18756d958Kostya SerebryanyINTERCEPTOR(uptr, strnlen, const char *s, uptr maxlen) {
540e130191a254064bfd1c8d373afdacf6d52979713Kostya Serebryany  ENSURE_ASAN_INITED();
5413f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  uptr length = REAL(strnlen)(s, maxlen);
542cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (flags()->replace_str) {
5432d8b3bdb112ebb8ed3f15ee41d4cebcd683b41b0Kostya Serebryany    ASAN_READ_RANGE(s, Min(length + 1, maxlen));
5441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  }
5451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany  return length;
5461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}
54781a7a4ab53c94e76c3b131d6f0ed82894640900bAlexey Samsonov#endif  // ASAN_INTERCEPT_STRNLEN
548547652c45fd7f7497dd214ec8ec6b6ee0f80359dKostya Serebryany
5498f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonovstatic inline bool IsValidStrtolBase(int base) {
5508f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov  return (base == 0) || (2 <= base && base <= 36);
5518f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov}
5528f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov
5538f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonovstatic inline void FixRealStrtolEndptr(const char *nptr, char **endptr) {
5543f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  CHECK(endptr != 0);
5558f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov  if (nptr == *endptr) {
5568f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov    // No digits were found at strtol call, we need to find out the last
5578f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov    // symbol accessed by strtoll on our own.
5588f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov    // We get this symbol by skipping leading blanks and optional +/- sign.
5598f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov    while (IsSpace(*nptr)) nptr++;
5608f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov    if (*nptr == '+' || *nptr == '-') nptr++;
5618f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov    *endptr = (char*)nptr;
5628f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov  }
5638f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov  CHECK(*endptr >= nptr);
56484ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov}
56584ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov
566847f932ab0405757946433b81d3b2952b306b0bcAlexey SamsonovINTERCEPTOR(long, strtol, const char *nptr,  // NOLINT
56784ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov            char **endptr, int base) {
56884ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov  ENSURE_ASAN_INITED();
569cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (!flags()->replace_str) {
570847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov    return REAL(strtol)(nptr, endptr, base);
57184ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov  }
57284ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov  char *real_endptr;
573847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  long result = REAL(strtol)(nptr, &real_endptr, base);  // NOLINT
5743f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  if (endptr != 0) {
57584ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov    *endptr = real_endptr;
57684ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov  }
5778f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov  if (IsValidStrtolBase(base)) {
5788f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov    FixRealStrtolEndptr(nptr, &real_endptr);
57984ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov    ASAN_READ_RANGE(nptr, (real_endptr - nptr) + 1);
58084ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov  }
58184ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov  return result;
58284ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov}
58384ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov
584847f932ab0405757946433b81d3b2952b306b0bcAlexey SamsonovINTERCEPTOR(int, atoi, const char *nptr) {
58569563986ca570ce750111a82264d51ddbf4107baAlexander Potapenko#if defined(__APPLE__)
5860ef531054fc25c11e372cbab1384f10954984219Alexander Potapenko  if (!asan_inited) return REAL(atoi)(nptr);
5870ef531054fc25c11e372cbab1384f10954984219Alexander Potapenko#endif
588847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  ENSURE_ASAN_INITED();
589cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (!flags()->replace_str) {
590847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov    return REAL(atoi)(nptr);
591847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  }
592847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  char *real_endptr;
593847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  // "man atoi" tells that behavior of atoi(nptr) is the same as
5943f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  // strtol(nptr, 0, 10), i.e. it sets errno to ERANGE if the
595847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  // parsed integer can't be stored in *long* type (even if it's
596847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  // different from int). So, we just imitate this behavior.
597847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  int result = REAL(strtol)(nptr, &real_endptr, 10);
598847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  FixRealStrtolEndptr(nptr, &real_endptr);
599847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  ASAN_READ_RANGE(nptr, (real_endptr - nptr) + 1);
600847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  return result;
601847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov}
602847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov
603847f932ab0405757946433b81d3b2952b306b0bcAlexey SamsonovINTERCEPTOR(long, atol, const char *nptr) {  // NOLINT
60469563986ca570ce750111a82264d51ddbf4107baAlexander Potapenko#if defined(__APPLE__)
6050ef531054fc25c11e372cbab1384f10954984219Alexander Potapenko  if (!asan_inited) return REAL(atol)(nptr);
6060ef531054fc25c11e372cbab1384f10954984219Alexander Potapenko#endif
607847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  ENSURE_ASAN_INITED();
608cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (!flags()->replace_str) {
609847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov    return REAL(atol)(nptr);
610847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  }
611847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  char *real_endptr;
612847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  long result = REAL(strtol)(nptr, &real_endptr, 10);  // NOLINT
613847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  FixRealStrtolEndptr(nptr, &real_endptr);
614847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  ASAN_READ_RANGE(nptr, (real_endptr - nptr) + 1);
615847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  return result;
616847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov}
617847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov
618847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov#if ASAN_INTERCEPT_ATOLL_AND_STRTOLL
619847f932ab0405757946433b81d3b2952b306b0bcAlexey SamsonovINTERCEPTOR(long long, strtoll, const char *nptr,  // NOLINT
6208f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov            char **endptr, int base) {
6218f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov  ENSURE_ASAN_INITED();
622cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (!flags()->replace_str) {
623847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov    return REAL(strtoll)(nptr, endptr, base);
6248f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov  }
6258f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov  char *real_endptr;
626847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  long long result = REAL(strtoll)(nptr, &real_endptr, base);  // NOLINT
6273f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany  if (endptr != 0) {
6288f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov    *endptr = real_endptr;
6298f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov  }
630847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  // If base has unsupported value, strtoll can exit with EINVAL
631847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  // without reading any characters. So do additional checks only
632847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  // if base is valid.
6338f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov  if (IsValidStrtolBase(base)) {
6348f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov    FixRealStrtolEndptr(nptr, &real_endptr);
6358f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov    ASAN_READ_RANGE(nptr, (real_endptr - nptr) + 1);
6368f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov  }
6378f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov  return result;
6388f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov}
6398f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov
640847f932ab0405757946433b81d3b2952b306b0bcAlexey SamsonovINTERCEPTOR(long long, atoll, const char *nptr) {  // NOLINT
641847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  ENSURE_ASAN_INITED();
642cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (!flags()->replace_str) {
643847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov    return REAL(atoll)(nptr);
644847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  }
645847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  char *real_endptr;
646847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  long long result = REAL(strtoll)(nptr, &real_endptr, 10);  // NOLINT
647847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  FixRealStrtolEndptr(nptr, &real_endptr);
648847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  ASAN_READ_RANGE(nptr, (real_endptr - nptr) + 1);
649847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov  return result;
650847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov}
651847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov#endif  // ASAN_INTERCEPT_ATOLL_AND_STRTOLL
652847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov
653580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov#define ASAN_INTERCEPT_FUNC(name) do { \
654cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov      if (!INTERCEPT_FUNCTION(name) && flags()->verbosity > 0) \
655580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov        Report("AddressSanitizer: failed to intercept '" #name "'\n"); \
656580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov    } while (0)
657580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov
658600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov#if defined(_WIN32)
659600972e3427173cc8904d741decd1af0ed5de9fdTimur IskhodzhanovINTERCEPTOR_WINAPI(DWORD, CreateThread,
6603f4c3875c42078e22c7e5356c5746fd18756d958Kostya Serebryany                   void* security, uptr stack_size,
661600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov                   DWORD (__stdcall *start_routine)(void*), void* arg,
662600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov                   DWORD flags, void* tid) {
663a30c8f9eac981dcf137e84226810b760e35c7be1Kostya Serebryany  GET_STACK_TRACE_THREAD;
664e0cff0bc20ae51790c8edfbceb817e18ebf5355eKostya Serebryany  u32 current_tid = asanThreadRegistry().GetCurrentTidOrInvalid();
665600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov  AsanThread *t = AsanThread::Create(current_tid, start_routine, arg, &stack);
666600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov  asanThreadRegistry().RegisterThread(t);
667600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov  return REAL(CreateThread)(security, stack_size,
668600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov                            asan_thread_start, t, flags, tid);
669600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov}
670600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov
671600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanovnamespace __asan {
672600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanovvoid InitializeWindowsInterceptors() {
673580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(CreateThread);
674600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov}
675600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov
676600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov}  // namespace __asan
677600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov#endif
678600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov
679547652c45fd7f7497dd214ec8ec6b6ee0f80359dKostya Serebryany// ---------------------- InitializeAsanInterceptors ---------------- {{{1
680547652c45fd7f7497dd214ec8ec6b6ee0f80359dKostya Serebryanynamespace __asan {
681547652c45fd7f7497dd214ec8ec6b6ee0f80359dKostya Serebryanyvoid InitializeAsanInterceptors() {
682fdbdab5ff8de4c23b58bb411ea2358a78e2d9b2fKostya Serebryany  static bool was_called_once;
683fdbdab5ff8de4c23b58bb411ea2358a78e2d9b2fKostya Serebryany  CHECK(was_called_once == false);
684fdbdab5ff8de4c23b58bb411ea2358a78e2d9b2fKostya Serebryany  was_called_once = true;
68569563986ca570ce750111a82264d51ddbf4107baAlexander Potapenko#if defined(__APPLE__)
6860ef531054fc25c11e372cbab1384f10954984219Alexander Potapenko  return;
68769563986ca570ce750111a82264d51ddbf4107baAlexander Potapenko#else
6888530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany  SANITIZER_COMMON_INTERCEPTORS_INIT;
6898530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany
69007bb9f1e3600195119aec1aae1aa48a6ed2f5febTimur Iskhodzhanov  // Intercept mem* functions.
691580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(memcmp);
692580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(memmove);
693580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(memset);
69438dd4ed885e714c376466f6fe0d69f5f22d37014Alexey Samsonov  if (PLATFORM_HAS_DIFFERENT_MEMCPY_AND_MEMMOVE) {
695580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov    ASAN_INTERCEPT_FUNC(memcpy);
696573fb4b102dda9d231a8dbd0c01e67e84e9b0874Alexander Potapenko  }
69707bb9f1e3600195119aec1aae1aa48a6ed2f5febTimur Iskhodzhanov
69807bb9f1e3600195119aec1aae1aa48a6ed2f5febTimur Iskhodzhanov  // Intercept str* functions.
699580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(strcat);  // NOLINT
700580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(strchr);
701580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(strcmp);
702580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(strcpy);  // NOLINT
703580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(strlen);
704c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov  ASAN_INTERCEPT_FUNC(strncat);
705580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(strncmp);
706580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(strncpy);
707fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#if ASAN_INTERCEPT_STRCASECMP_AND_STRNCASECMP
708580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(strcasecmp);
709580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(strncasecmp);
710fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#endif
711fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#if ASAN_INTERCEPT_STRDUP
712fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov  ASAN_INTERCEPT_FUNC(strdup);
713fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#endif
714fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#if ASAN_INTERCEPT_STRNLEN
715fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov  ASAN_INTERCEPT_FUNC(strnlen);
716fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#endif
71769563986ca570ce750111a82264d51ddbf4107baAlexander Potapenko#if ASAN_INTERCEPT_INDEX && ASAN_USE_ALIAS_ATTRIBUTE_FOR_INDEX
718580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(index);
71907bb9f1e3600195119aec1aae1aa48a6ed2f5febTimur Iskhodzhanov#endif
7205b29018cf422e7711fb760b733c32127397a43fcAlexey Samsonov
721580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(atoi);
722580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(atol);
723580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(strtol);
724847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov#if ASAN_INTERCEPT_ATOLL_AND_STRTOLL
725580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(atoll);
726580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(strtoll);
72784ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov#endif
72884ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov
729f0c6de3d91ba5cdfbb17a630e823523c7d433160Alexander Potapenko#if ASAN_INTERCEPT_MLOCKX
73000f1c091441be5c2cacb99e21bd1965a5e418242Alexander Potapenko  // Intercept mlock/munlock.
73100f1c091441be5c2cacb99e21bd1965a5e418242Alexander Potapenko  ASAN_INTERCEPT_FUNC(mlock);
73200f1c091441be5c2cacb99e21bd1965a5e418242Alexander Potapenko  ASAN_INTERCEPT_FUNC(munlock);
73300f1c091441be5c2cacb99e21bd1965a5e418242Alexander Potapenko  ASAN_INTERCEPT_FUNC(mlockall);
73400f1c091441be5c2cacb99e21bd1965a5e418242Alexander Potapenko  ASAN_INTERCEPT_FUNC(munlockall);
735f0c6de3d91ba5cdfbb17a630e823523c7d433160Alexander Potapenko#endif
73600f1c091441be5c2cacb99e21bd1965a5e418242Alexander Potapenko
73707bb9f1e3600195119aec1aae1aa48a6ed2f5febTimur Iskhodzhanov  // Intecept signal- and jump-related functions.
738580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(longjmp);
73934a3202a2c22816a6da66959e266a2d078ded37bAlexey Samsonov#if ASAN_INTERCEPT_SIGNAL_AND_SIGACTION
740580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(sigaction);
741580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(signal);
742919c24787a655597bdcf367e26c9dd1504e9f463Evgeniy Stepanov#endif
7430870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov#if ASAN_INTERCEPT_SWAPCONTEXT
7440870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov  ASAN_INTERCEPT_FUNC(swapcontext);
7450870028410087e67a0049c76cb7c64f02c260d24Alexey Samsonov#endif
746fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#if ASAN_INTERCEPT__LONGJMP
747580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(_longjmp);
748fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#endif
749fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#if ASAN_INTERCEPT_SIGLONGJMP
750580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(siglongjmp);
751fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#endif
752fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov
753fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov  // Intercept exception handling functions.
754fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#if ASAN_INTERCEPT___CXA_THROW
755fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov  INTERCEPT_FUNCTION(__cxa_throw);
7563e81fe43798c096d1d3565596f0717b9919de4fbTimur Iskhodzhanov#endif
7573e81fe43798c096d1d3565596f0717b9919de4fbTimur Iskhodzhanov
75807bb9f1e3600195119aec1aae1aa48a6ed2f5febTimur Iskhodzhanov  // Intercept threading-related functions
759fd2ae4fc7df0e5acd200f30d87bbc6b96830a989Alexey Samsonov#if ASAN_INTERCEPT_PTHREAD_CREATE
760580469d7e40e39319cb2d3750edac4bccca18105Dmitry Vyukov  ASAN_INTERCEPT_FUNC(pthread_create);
76107bb9f1e3600195119aec1aae1aa48a6ed2f5febTimur Iskhodzhanov#endif
76207bb9f1e3600195119aec1aae1aa48a6ed2f5febTimur Iskhodzhanov
763600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov  // Some Windows-specific interceptors.
764600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov#if defined(_WIN32)
765600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov  InitializeWindowsInterceptors();
766600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov#endif
767600972e3427173cc8904d741decd1af0ed5de9fdTimur Iskhodzhanov
768cb8c4dce691097718d5af41b36899b72ef4b1d84Alexey Samsonov  if (flags()->verbosity > 0) {
7692962f26071ebef1d5fec52b5569e5ae7aae45c9bAlexander Potapenko    Report("AddressSanitizer: libc interceptors initialized\n");
770547652c45fd7f7497dd214ec8ec6b6ee0f80359dKostya Serebryany  }
7712673fd8406197c42f16cede6d287f72169298c2eKostya Serebryany#endif  // __APPLE__
772547652c45fd7f7497dd214ec8ec6b6ee0f80359dKostya Serebryany}
773547652c45fd7f7497dd214ec8ec6b6ee0f80359dKostya Serebryany
774547652c45fd7f7497dd214ec8ec6b6ee0f80359dKostya Serebryany}  // namespace __asan
775