asan_rtl.cc revision f4f51f2cc6fa936f0c65577f82e6b62989d546ee 
1//===-- asan_rtl.cc -------------------------------------------------------===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file is a part of AddressSanitizer, an address sanity checker.
11//
12// Main file of the ASan run-time library.
13//===----------------------------------------------------------------------===//
14#include "asan_allocator.h"
15#include "asan_interceptors.h"
16#include "asan_internal.h"
17#include "asan_mapping.h"
18#include "asan_report.h"
19#include "asan_stack.h"
20#include "asan_stats.h"
21#include "asan_thread.h"
22#include "asan_thread_registry.h"
23#include "sanitizer/asan_interface.h"
24#include "sanitizer_common/sanitizer_atomic.h"
25#include "sanitizer_common/sanitizer_flags.h"
26#include "sanitizer_common/sanitizer_libc.h"
27#include "sanitizer_common/sanitizer_symbolizer.h"
28
29namespace __asan {
30
31static void AsanDie() {
32  static atomic_uint32_t num_calls;
33  if (atomic_fetch_add(&num_calls, 1, memory_order_relaxed) != 0) {
34    // Don't die twice - run a busy loop.
35    while (1) { }
36  }
37  if (flags()->sleep_before_dying) {
38    Report("Sleeping for %d second(s)\n", flags()->sleep_before_dying);
39    SleepForSeconds(flags()->sleep_before_dying);
40  }
41  if (flags()->unmap_shadow_on_exit)
42    UnmapOrDie((void*)kLowShadowBeg, kHighShadowEnd - kLowShadowBeg);
43  if (death_callback)
44    death_callback();
45  if (flags()->abort_on_error)
46    Abort();
47  Exit(flags()->exitcode);
48}
49
50static void AsanCheckFailed(const char *file, int line, const char *cond,
51                            u64 v1, u64 v2) {
52  Report("AddressSanitizer CHECK failed: %s:%d \"%s\" (0x%zx, 0x%zx)\n",
53             file, line, cond, (uptr)v1, (uptr)v2);
54  // FIXME: check for infinite recursion without a thread-local counter here.
55  PRINT_CURRENT_STACK();
56  Die();
57}
58
59// -------------------------- Flags ------------------------- {{{1
60static const int kDeafultMallocContextSize = 30;
61
62static Flags asan_flags;
63
64Flags *flags() {
65  return &asan_flags;
66}
67
68static const char *MaybeCallAsanDefaultOptions() {
69  return (&__asan_default_options) ? __asan_default_options() : "";
70}
71
72static void ParseFlagsFromString(Flags *f, const char *str) {
73  ParseFlag(str, &f->quarantine_size, "quarantine_size");
74  ParseFlag(str, &f->symbolize, "symbolize");
75  ParseFlag(str, &f->verbosity, "verbosity");
76  ParseFlag(str, &f->redzone, "redzone");
77  CHECK(f->redzone >= 16);
78  CHECK(IsPowerOfTwo(f->redzone));
79
80  ParseFlag(str, &f->debug, "debug");
81  ParseFlag(str, &f->report_globals, "report_globals");
82  ParseFlag(str, &f->check_initialization_order, "initialization_order");
83  ParseFlag(str, &f->malloc_context_size, "malloc_context_size");
84  CHECK((uptr)f->malloc_context_size <= kStackTraceMax);
85
86  ParseFlag(str, &f->replace_str, "replace_str");
87  ParseFlag(str, &f->replace_intrin, "replace_intrin");
88  ParseFlag(str, &f->replace_cfallocator, "replace_cfallocator");
89  ParseFlag(str, &f->mac_ignore_invalid_free, "mac_ignore_invalid_free");
90  ParseFlag(str, &f->use_fake_stack, "use_fake_stack");
91  ParseFlag(str, &f->max_malloc_fill_size, "max_malloc_fill_size");
92  ParseFlag(str, &f->exitcode, "exitcode");
93  ParseFlag(str, &f->allow_user_poisoning, "allow_user_poisoning");
94  ParseFlag(str, &f->sleep_before_dying, "sleep_before_dying");
95  ParseFlag(str, &f->handle_segv, "handle_segv");
96  ParseFlag(str, &f->use_sigaltstack, "use_sigaltstack");
97  ParseFlag(str, &f->check_malloc_usable_size, "check_malloc_usable_size");
98  ParseFlag(str, &f->unmap_shadow_on_exit, "unmap_shadow_on_exit");
99  ParseFlag(str, &f->abort_on_error, "abort_on_error");
100  ParseFlag(str, &f->atexit, "atexit");
101  ParseFlag(str, &f->disable_core, "disable_core");
102  ParseFlag(str, &f->strip_path_prefix, "strip_path_prefix");
103  ParseFlag(str, &f->allow_reexec, "allow_reexec");
104  ParseFlag(str, &f->print_full_thread_history, "print_full_thread_history");
105  ParseFlag(str, &f->log_path, "log_path");
106  ParseFlag(str, &f->fast_unwind_on_fatal, "fast_unwind_on_fatal");
107  ParseFlag(str, &f->fast_unwind_on_malloc, "fast_unwind_on_malloc");
108  ParseFlag(str, &f->poison_heap, "poison_heap");
109  ParseFlag(str, &f->alloc_dealloc_mismatch, "alloc_dealloc_mismatch");
110  ParseFlag(str, &f->use_stack_depot, "use_stack_depot");
111}
112
113void InitializeFlags(Flags *f, const char *env) {
114  internal_memset(f, 0, sizeof(*f));
115
116  f->quarantine_size = (ASAN_LOW_MEMORY) ? 1UL << 26 : 1UL << 28;
117  f->symbolize = false;
118  f->verbosity = 0;
119  f->redzone = ASAN_ALLOCATOR_VERSION == 2 ? 16 : (ASAN_LOW_MEMORY) ? 64 : 128;
120  f->debug = false;
121  f->report_globals = 1;
122  f->check_initialization_order = true;
123  f->malloc_context_size = kDeafultMallocContextSize;
124  f->replace_str = true;
125  f->replace_intrin = true;
126  f->replace_cfallocator = true;
127  f->mac_ignore_invalid_free = false;
128  f->use_fake_stack = true;
129  f->max_malloc_fill_size = 0;
130  f->exitcode = ASAN_DEFAULT_FAILURE_EXITCODE;
131  f->allow_user_poisoning = true;
132  f->sleep_before_dying = 0;
133  f->handle_segv = ASAN_NEEDS_SEGV;
134  f->use_sigaltstack = false;
135  f->check_malloc_usable_size = true;
136  f->unmap_shadow_on_exit = false;
137  f->abort_on_error = false;
138  f->atexit = false;
139  f->disable_core = (SANITIZER_WORDSIZE == 64);
140  f->strip_path_prefix = "";
141  f->allow_reexec = true;
142  f->print_full_thread_history = true;
143  f->log_path = 0;
144  f->fast_unwind_on_fatal = true;
145  f->fast_unwind_on_malloc = true;
146  f->poison_heap = true;
147  f->alloc_dealloc_mismatch = true;
148  f->use_stack_depot = true;  // Only affects allocator2.
149
150  // Override from user-specified string.
151  ParseFlagsFromString(f, MaybeCallAsanDefaultOptions());
152  if (flags()->verbosity) {
153    Report("Using the defaults from __asan_default_options: %s\n",
154           MaybeCallAsanDefaultOptions());
155  }
156
157  // Override from command line.
158  ParseFlagsFromString(f, env);
159}
160
161// -------------------------- Globals --------------------- {{{1
162int asan_inited;
163bool asan_init_is_running;
164void (*death_callback)(void);
165
166// -------------------------- Misc ---------------- {{{1
167void ShowStatsAndAbort() {
168  __asan_print_accumulated_stats();
169  Die();
170}
171
172// ---------------------- mmap -------------------- {{{1
173// Reserve memory range [beg, end].
174static void ReserveShadowMemoryRange(uptr beg, uptr end) {
175  CHECK((beg % GetPageSizeCached()) == 0);
176  CHECK(((end + 1) % GetPageSizeCached()) == 0);
177  uptr size = end - beg + 1;
178  void *res = MmapFixedNoReserve(beg, size);
179  if (res != (void*)beg) {
180    Report("ReserveShadowMemoryRange failed while trying to map 0x%zx bytes. "
181           "Perhaps you're using ulimit -v\n", size);
182    Abort();
183  }
184}
185
186// --------------- LowLevelAllocateCallbac ---------- {{{1
187static void OnLowLevelAllocate(uptr ptr, uptr size) {
188  PoisonShadow(ptr, size, kAsanInternalHeapMagic);
189}
190
191// -------------------------- Run-time entry ------------------- {{{1
192// exported functions
193#define ASAN_REPORT_ERROR(type, is_write, size)                     \
194extern "C" NOINLINE INTERFACE_ATTRIBUTE                        \
195void __asan_report_ ## type ## size(uptr addr);                \
196void __asan_report_ ## type ## size(uptr addr) {               \
197  GET_CALLER_PC_BP_SP;                                              \
198  __asan_report_error(pc, bp, sp, addr, is_write, size);            \
199}
200
201ASAN_REPORT_ERROR(load, false, 1)
202ASAN_REPORT_ERROR(load, false, 2)
203ASAN_REPORT_ERROR(load, false, 4)
204ASAN_REPORT_ERROR(load, false, 8)
205ASAN_REPORT_ERROR(load, false, 16)
206ASAN_REPORT_ERROR(store, true, 1)
207ASAN_REPORT_ERROR(store, true, 2)
208ASAN_REPORT_ERROR(store, true, 4)
209ASAN_REPORT_ERROR(store, true, 8)
210ASAN_REPORT_ERROR(store, true, 16)
211
212// Force the linker to keep the symbols for various ASan interface functions.
213// We want to keep those in the executable in order to let the instrumented
214// dynamic libraries access the symbol even if it is not used by the executable
215// itself. This should help if the build system is removing dead code at link
216// time.
217static NOINLINE void force_interface_symbols() {
218  volatile int fake_condition = 0;  // prevent dead condition elimination.
219  // __asan_report_* functions are noreturn, so we need a switch to prevent
220  // the compiler from removing any of them.
221  switch (fake_condition) {
222    case 1: __asan_report_load1(0); break;
223    case 2: __asan_report_load2(0); break;
224    case 3: __asan_report_load4(0); break;
225    case 4: __asan_report_load8(0); break;
226    case 5: __asan_report_load16(0); break;
227    case 6: __asan_report_store1(0); break;
228    case 7: __asan_report_store2(0); break;
229    case 8: __asan_report_store4(0); break;
230    case 9: __asan_report_store8(0); break;
231    case 10: __asan_report_store16(0); break;
232    case 11: __asan_register_global(0, 0, 0); break;
233    case 12: __asan_register_globals(0, 0); break;
234    case 13: __asan_unregister_globals(0, 0); break;
235    case 14: __asan_set_death_callback(0); break;
236    case 15: __asan_set_error_report_callback(0); break;
237    case 16: __asan_handle_no_return(); break;
238    case 17: __asan_address_is_poisoned(0); break;
239    case 18: __asan_get_allocated_size(0); break;
240    case 19: __asan_get_current_allocated_bytes(); break;
241    case 20: __asan_get_estimated_allocated_size(0); break;
242    case 21: __asan_get_free_bytes(); break;
243    case 22: __asan_get_heap_size(); break;
244    case 23: __asan_get_ownership(0); break;
245    case 24: __asan_get_unmapped_bytes(); break;
246    case 25: __asan_poison_memory_region(0, 0); break;
247    case 26: __asan_unpoison_memory_region(0, 0); break;
248    case 27: __asan_set_error_exit_code(0); break;
249    case 28: __asan_stack_free(0, 0, 0); break;
250    case 29: __asan_stack_malloc(0, 0); break;
251    case 30: __asan_before_dynamic_init(0, 0); break;
252    case 31: __asan_after_dynamic_init(); break;
253    case 32: __asan_poison_stack_memory(0, 0); break;
254    case 33: __asan_unpoison_stack_memory(0, 0); break;
255    case 34: __asan_region_is_poisoned(0, 0); break;
256    case 35: __asan_describe_address(0); break;
257  }
258}
259
260static void asan_atexit() {
261  Printf("AddressSanitizer exit stats:\n");
262  __asan_print_accumulated_stats();
263}
264
265}  // namespace __asan
266
267// ---------------------- Interface ---------------- {{{1
268using namespace __asan;  // NOLINT
269
270#if !SANITIZER_SUPPORTS_WEAK_HOOKS
271extern "C" {
272SANITIZER_WEAK_ATTRIBUTE SANITIZER_INTERFACE_ATTRIBUTE
273const char* __asan_default_options() { return ""; }
274}  // extern "C"
275#endif
276
277int NOINLINE __asan_set_error_exit_code(int exit_code) {
278  int old = flags()->exitcode;
279  flags()->exitcode = exit_code;
280  return old;
281}
282
283void NOINLINE __asan_handle_no_return() {
284  int local_stack;
285  AsanThread *curr_thread = asanThreadRegistry().GetCurrent();
286  CHECK(curr_thread);
287  uptr PageSize = GetPageSizeCached();
288  uptr top = curr_thread->stack_top();
289  uptr bottom = ((uptr)&local_stack - PageSize) & ~(PageSize-1);
290  PoisonShadow(bottom, top - bottom, 0);
291}
292
293void NOINLINE __asan_set_death_callback(void (*callback)(void)) {
294  death_callback = callback;
295}
296
297void __asan_init() {
298  if (asan_inited) return;
299  CHECK(!asan_init_is_running && "ASan init calls itself!");
300  asan_init_is_running = true;
301
302  // Make sure we are not statically linked.
303  AsanDoesNotSupportStaticLinkage();
304
305  // Install tool-specific callbacks in sanitizer_common.
306  SetDieCallback(AsanDie);
307  SetCheckFailedCallback(AsanCheckFailed);
308  SetPrintfAndReportCallback(AppendToErrorMessageBuffer);
309
310  // Initialize flags. This must be done early, because most of the
311  // initialization steps look at flags().
312  const char *options = GetEnv("ASAN_OPTIONS");
313  InitializeFlags(flags(), options);
314  __sanitizer_set_report_path(flags()->log_path);
315
316  if (flags()->verbosity && options) {
317    Report("Parsed ASAN_OPTIONS: %s\n", options);
318  }
319
320  // Re-exec ourselves if we need to set additional env or command line args.
321  MaybeReexec();
322
323  // Setup internal allocator callback.
324  SetLowLevelAllocateCallback(OnLowLevelAllocate);
325
326  if (flags()->atexit) {
327    Atexit(asan_atexit);
328  }
329
330  // interceptors
331  InitializeAsanInterceptors();
332
333  ReplaceSystemMalloc();
334  ReplaceOperatorsNewAndDelete();
335
336  if (flags()->verbosity) {
337    Printf("|| `[%p, %p]` || HighMem    ||\n",
338           (void*)kHighMemBeg, (void*)kHighMemEnd);
339    Printf("|| `[%p, %p]` || HighShadow ||\n",
340           (void*)kHighShadowBeg, (void*)kHighShadowEnd);
341    Printf("|| `[%p, %p]` || ShadowGap  ||\n",
342           (void*)kShadowGapBeg, (void*)kShadowGapEnd);
343    Printf("|| `[%p, %p]` || LowShadow  ||\n",
344           (void*)kLowShadowBeg, (void*)kLowShadowEnd);
345    Printf("|| `[%p, %p]` || LowMem     ||\n",
346           (void*)kLowMemBeg, (void*)kLowMemEnd);
347    Printf("MemToShadow(shadow): %p %p %p %p\n",
348           (void*)MEM_TO_SHADOW(kLowShadowBeg),
349           (void*)MEM_TO_SHADOW(kLowShadowEnd),
350           (void*)MEM_TO_SHADOW(kHighShadowBeg),
351           (void*)MEM_TO_SHADOW(kHighShadowEnd));
352    Printf("red_zone=%zu\n", (uptr)flags()->redzone);
353    Printf("malloc_context_size=%zu\n", (uptr)flags()->malloc_context_size);
354
355    Printf("SHADOW_SCALE: %zx\n", (uptr)SHADOW_SCALE);
356    Printf("SHADOW_GRANULARITY: %zx\n", (uptr)SHADOW_GRANULARITY);
357    Printf("SHADOW_OFFSET: %zx\n", (uptr)SHADOW_OFFSET);
358    CHECK(SHADOW_SCALE >= 3 && SHADOW_SCALE <= 7);
359  }
360
361  if (flags()->disable_core) {
362    DisableCoreDumper();
363  }
364
365  uptr shadow_start = kLowShadowBeg;
366  if (kLowShadowBeg > 0) shadow_start -= GetMmapGranularity();
367  uptr shadow_end = kHighShadowEnd;
368  if (MemoryRangeIsAvailable(shadow_start, shadow_end)) {
369    if (kLowShadowBeg != kLowShadowEnd) {
370      // mmap the low shadow plus at least one page.
371      ReserveShadowMemoryRange(kLowShadowBeg - GetMmapGranularity(),
372                               kLowShadowEnd);
373    }
374    // mmap the high shadow.
375    ReserveShadowMemoryRange(kHighShadowBeg, kHighShadowEnd);
376    // protect the gap
377    void *prot = Mprotect(kShadowGapBeg, kShadowGapEnd - kShadowGapBeg + 1);
378    CHECK(prot == (void*)kShadowGapBeg);
379  } else {
380    Report("Shadow memory range interleaves with an existing memory mapping. "
381           "ASan cannot proceed correctly. ABORTING.\n");
382    DumpProcessMap();
383    Die();
384  }
385
386  InstallSignalHandlers();
387  // Start symbolizer process if necessary.
388  if (flags()->symbolize) {
389    const char *external_symbolizer = GetEnv("ASAN_SYMBOLIZER_PATH");
390    if (external_symbolizer) {
391      InitializeExternalSymbolizer(external_symbolizer);
392    }
393  }
394
395  // On Linux AsanThread::ThreadStart() calls malloc() that's why asan_inited
396  // should be set to 1 prior to initializing the threads.
397  asan_inited = 1;
398  asan_init_is_running = false;
399
400  asanThreadRegistry().Init();
401  asanThreadRegistry().GetMain()->ThreadStart();
402  force_interface_symbols();  // no-op.
403
404  if (flags()->verbosity) {
405    Report("AddressSanitizer Init done\n");
406  }
407}
408
409#if defined(ASAN_USE_PREINIT_ARRAY)
410  // On Linux, we force __asan_init to be called before anyone else
411  // by placing it into .preinit_array section.
412  // FIXME: do we have anything like this on Mac?
413  __attribute__((section(".preinit_array")))
414    typeof(__asan_init) *__asan_preinit =__asan_init;
415#elif defined(_WIN32) && defined(_DLL)
416  // On Windows, when using dynamic CRT (/MD), we can put a pointer
417  // to __asan_init into the global list of C initializers.
418  // See crt0dat.c in the CRT sources for the details.
419  #pragma section(".CRT$XIB", long, read)  // NOLINT
420  __declspec(allocate(".CRT$XIB")) void (*__asan_preinit)() = __asan_init;
421#endif
422