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
15#include "asan_activation.h"
16#include "asan_allocator.h"
17#include "asan_interceptors.h"
18#include "asan_interface_internal.h"
19#include "asan_internal.h"
20#include "asan_mapping.h"
21#include "asan_poisoning.h"
22#include "asan_report.h"
23#include "asan_stack.h"
24#include "asan_stats.h"
25#include "asan_suppressions.h"
26#include "asan_thread.h"
27#include "sanitizer_common/sanitizer_atomic.h"
28#include "sanitizer_common/sanitizer_flags.h"
29#include "sanitizer_common/sanitizer_libc.h"
30#include "sanitizer_common/sanitizer_symbolizer.h"
31#include "lsan/lsan_common.h"
32#include "ubsan/ubsan_init.h"
33#include "ubsan/ubsan_platform.h"
34
35int __asan_option_detect_stack_use_after_return;  // Global interface symbol.
36uptr *__asan_test_only_reported_buggy_pointer;  // Used only for testing asan.
37
38namespace __asan {
39
40uptr AsanMappingProfile[kAsanMappingProfileSize];
41
42static void AsanDie() {
43  static atomic_uint32_t num_calls;
44  if (atomic_fetch_add(&num_calls, 1, memory_order_relaxed) != 0) {
45    // Don't die twice - run a busy loop.
46    while (1) { }
47  }
48  if (flags()->sleep_before_dying) {
49    Report("Sleeping for %d second(s)\n", flags()->sleep_before_dying);
50    SleepForSeconds(flags()->sleep_before_dying);
51  }
52  if (flags()->unmap_shadow_on_exit) {
53    if (kMidMemBeg) {
54      UnmapOrDie((void*)kLowShadowBeg, kMidMemBeg - kLowShadowBeg);
55      UnmapOrDie((void*)kMidMemEnd, kHighShadowEnd - kMidMemEnd);
56    } else {
57      UnmapOrDie((void*)kLowShadowBeg, kHighShadowEnd - kLowShadowBeg);
58    }
59  }
60}
61
62static void AsanCheckFailed(const char *file, int line, const char *cond,
63                            u64 v1, u64 v2) {
64  Report("AddressSanitizer CHECK failed: %s:%d \"%s\" (0x%zx, 0x%zx)\n", file,
65         line, cond, (uptr)v1, (uptr)v2);
66  // FIXME: check for infinite recursion without a thread-local counter here.
67  PRINT_CURRENT_STACK_CHECK();
68  Die();
69}
70
71// -------------------------- Globals --------------------- {{{1
72int asan_inited;
73bool asan_init_is_running;
74
75#if !ASAN_FIXED_MAPPING
76uptr kHighMemEnd, kMidMemBeg, kMidMemEnd;
77#endif
78
79// -------------------------- Misc ---------------- {{{1
80void ShowStatsAndAbort() {
81  __asan_print_accumulated_stats();
82  Die();
83}
84
85// ---------------------- mmap -------------------- {{{1
86// Reserve memory range [beg, end].
87// We need to use inclusive range because end+1 may not be representable.
88void ReserveShadowMemoryRange(uptr beg, uptr end, const char *name) {
89  CHECK_EQ((beg % GetPageSizeCached()), 0);
90  CHECK_EQ(((end + 1) % GetPageSizeCached()), 0);
91  uptr size = end - beg + 1;
92  DecreaseTotalMmap(size);  // Don't count the shadow against mmap_limit_mb.
93  void *res = MmapFixedNoReserve(beg, size, name);
94  if (res != (void*)beg) {
95    Report("ReserveShadowMemoryRange failed while trying to map 0x%zx bytes. "
96           "Perhaps you're using ulimit -v\n", size);
97    Abort();
98  }
99  if (common_flags()->no_huge_pages_for_shadow)
100    NoHugePagesInRegion(beg, size);
101  if (common_flags()->use_madv_dontdump)
102    DontDumpShadowMemory(beg, size);
103}
104
105// --------------- LowLevelAllocateCallbac ---------- {{{1
106static void OnLowLevelAllocate(uptr ptr, uptr size) {
107  PoisonShadow(ptr, size, kAsanInternalHeapMagic);
108}
109
110// -------------------------- Run-time entry ------------------- {{{1
111// exported functions
112#define ASAN_REPORT_ERROR(type, is_write, size)                     \
113extern "C" NOINLINE INTERFACE_ATTRIBUTE                             \
114void __asan_report_ ## type ## size(uptr addr) {                    \
115  GET_CALLER_PC_BP_SP;                                              \
116  ReportGenericError(pc, bp, sp, addr, is_write, size, 0, true);    \
117}                                                                   \
118extern "C" NOINLINE INTERFACE_ATTRIBUTE                             \
119void __asan_report_exp_ ## type ## size(uptr addr, u32 exp) {       \
120  GET_CALLER_PC_BP_SP;                                              \
121  ReportGenericError(pc, bp, sp, addr, is_write, size, exp, true);  \
122}                                                                   \
123extern "C" NOINLINE INTERFACE_ATTRIBUTE                             \
124void __asan_report_ ## type ## size ## _noabort(uptr addr) {        \
125  GET_CALLER_PC_BP_SP;                                              \
126  ReportGenericError(pc, bp, sp, addr, is_write, size, 0, false);   \
127}                                                                   \
128
129ASAN_REPORT_ERROR(load, false, 1)
130ASAN_REPORT_ERROR(load, false, 2)
131ASAN_REPORT_ERROR(load, false, 4)
132ASAN_REPORT_ERROR(load, false, 8)
133ASAN_REPORT_ERROR(load, false, 16)
134ASAN_REPORT_ERROR(store, true, 1)
135ASAN_REPORT_ERROR(store, true, 2)
136ASAN_REPORT_ERROR(store, true, 4)
137ASAN_REPORT_ERROR(store, true, 8)
138ASAN_REPORT_ERROR(store, true, 16)
139
140#define ASAN_REPORT_ERROR_N(type, is_write)                                 \
141extern "C" NOINLINE INTERFACE_ATTRIBUTE                                     \
142void __asan_report_ ## type ## _n(uptr addr, uptr size) {                   \
143  GET_CALLER_PC_BP_SP;                                                      \
144  ReportGenericError(pc, bp, sp, addr, is_write, size, 0, true);            \
145}                                                                           \
146extern "C" NOINLINE INTERFACE_ATTRIBUTE                                     \
147void __asan_report_exp_ ## type ## _n(uptr addr, uptr size, u32 exp) {      \
148  GET_CALLER_PC_BP_SP;                                                      \
149  ReportGenericError(pc, bp, sp, addr, is_write, size, exp, true);          \
150}                                                                           \
151extern "C" NOINLINE INTERFACE_ATTRIBUTE                                     \
152void __asan_report_ ## type ## _n_noabort(uptr addr, uptr size) {           \
153  GET_CALLER_PC_BP_SP;                                                      \
154  ReportGenericError(pc, bp, sp, addr, is_write, size, 0, false);           \
155}                                                                           \
156
157ASAN_REPORT_ERROR_N(load, false)
158ASAN_REPORT_ERROR_N(store, true)
159
160#define ASAN_MEMORY_ACCESS_CALLBACK_BODY(type, is_write, size, exp_arg, fatal) \
161    uptr sp = MEM_TO_SHADOW(addr);                                             \
162    uptr s = size <= SHADOW_GRANULARITY ? *reinterpret_cast<u8 *>(sp)          \
163                                        : *reinterpret_cast<u16 *>(sp);        \
164    if (UNLIKELY(s)) {                                                         \
165      if (UNLIKELY(size >= SHADOW_GRANULARITY ||                               \
166                   ((s8)((addr & (SHADOW_GRANULARITY - 1)) + size - 1)) >=     \
167                       (s8)s)) {                                               \
168        if (__asan_test_only_reported_buggy_pointer) {                         \
169          *__asan_test_only_reported_buggy_pointer = addr;                     \
170        } else {                                                               \
171          GET_CALLER_PC_BP_SP;                                                 \
172          ReportGenericError(pc, bp, sp, addr, is_write, size, exp_arg,        \
173                              fatal);                                          \
174        }                                                                      \
175      }                                                                        \
176    }
177
178#define ASAN_MEMORY_ACCESS_CALLBACK(type, is_write, size)                      \
179  extern "C" NOINLINE INTERFACE_ATTRIBUTE                                      \
180  void __asan_##type##size(uptr addr) {                                        \
181    ASAN_MEMORY_ACCESS_CALLBACK_BODY(type, is_write, size, 0, true)            \
182  }                                                                            \
183  extern "C" NOINLINE INTERFACE_ATTRIBUTE                                      \
184  void __asan_exp_##type##size(uptr addr, u32 exp) {                           \
185    ASAN_MEMORY_ACCESS_CALLBACK_BODY(type, is_write, size, exp, true)          \
186  }                                                                            \
187  extern "C" NOINLINE INTERFACE_ATTRIBUTE                                      \
188  void __asan_##type##size ## _noabort(uptr addr) {                            \
189    ASAN_MEMORY_ACCESS_CALLBACK_BODY(type, is_write, size, 0, false)           \
190  }                                                                            \
191
192ASAN_MEMORY_ACCESS_CALLBACK(load, false, 1)
193ASAN_MEMORY_ACCESS_CALLBACK(load, false, 2)
194ASAN_MEMORY_ACCESS_CALLBACK(load, false, 4)
195ASAN_MEMORY_ACCESS_CALLBACK(load, false, 8)
196ASAN_MEMORY_ACCESS_CALLBACK(load, false, 16)
197ASAN_MEMORY_ACCESS_CALLBACK(store, true, 1)
198ASAN_MEMORY_ACCESS_CALLBACK(store, true, 2)
199ASAN_MEMORY_ACCESS_CALLBACK(store, true, 4)
200ASAN_MEMORY_ACCESS_CALLBACK(store, true, 8)
201ASAN_MEMORY_ACCESS_CALLBACK(store, true, 16)
202
203extern "C"
204NOINLINE INTERFACE_ATTRIBUTE
205void __asan_loadN(uptr addr, uptr size) {
206  if (__asan_region_is_poisoned(addr, size)) {
207    GET_CALLER_PC_BP_SP;
208    ReportGenericError(pc, bp, sp, addr, false, size, 0, true);
209  }
210}
211
212extern "C"
213NOINLINE INTERFACE_ATTRIBUTE
214void __asan_exp_loadN(uptr addr, uptr size, u32 exp) {
215  if (__asan_region_is_poisoned(addr, size)) {
216    GET_CALLER_PC_BP_SP;
217    ReportGenericError(pc, bp, sp, addr, false, size, exp, true);
218  }
219}
220
221extern "C"
222NOINLINE INTERFACE_ATTRIBUTE
223void __asan_loadN_noabort(uptr addr, uptr size) {
224  if (__asan_region_is_poisoned(addr, size)) {
225    GET_CALLER_PC_BP_SP;
226    ReportGenericError(pc, bp, sp, addr, false, size, 0, false);
227  }
228}
229
230extern "C"
231NOINLINE INTERFACE_ATTRIBUTE
232void __asan_storeN(uptr addr, uptr size) {
233  if (__asan_region_is_poisoned(addr, size)) {
234    GET_CALLER_PC_BP_SP;
235    ReportGenericError(pc, bp, sp, addr, true, size, 0, true);
236  }
237}
238
239extern "C"
240NOINLINE INTERFACE_ATTRIBUTE
241void __asan_exp_storeN(uptr addr, uptr size, u32 exp) {
242  if (__asan_region_is_poisoned(addr, size)) {
243    GET_CALLER_PC_BP_SP;
244    ReportGenericError(pc, bp, sp, addr, true, size, exp, true);
245  }
246}
247
248extern "C"
249NOINLINE INTERFACE_ATTRIBUTE
250void __asan_storeN_noabort(uptr addr, uptr size) {
251  if (__asan_region_is_poisoned(addr, size)) {
252    GET_CALLER_PC_BP_SP;
253    ReportGenericError(pc, bp, sp, addr, true, size, 0, false);
254  }
255}
256
257// Force the linker to keep the symbols for various ASan interface functions.
258// We want to keep those in the executable in order to let the instrumented
259// dynamic libraries access the symbol even if it is not used by the executable
260// itself. This should help if the build system is removing dead code at link
261// time.
262static NOINLINE void force_interface_symbols() {
263  volatile int fake_condition = 0;  // prevent dead condition elimination.
264  // __asan_report_* functions are noreturn, so we need a switch to prevent
265  // the compiler from removing any of them.
266  switch (fake_condition) {
267    case 1: __asan_report_load1(0); break;
268    case 2: __asan_report_load2(0); break;
269    case 3: __asan_report_load4(0); break;
270    case 4: __asan_report_load8(0); break;
271    case 5: __asan_report_load16(0); break;
272    case 6: __asan_report_load_n(0, 0); break;
273    case 7: __asan_report_store1(0); break;
274    case 8: __asan_report_store2(0); break;
275    case 9: __asan_report_store4(0); break;
276    case 10: __asan_report_store8(0); break;
277    case 11: __asan_report_store16(0); break;
278    case 12: __asan_report_store_n(0, 0); break;
279    case 13: __asan_report_exp_load1(0, 0); break;
280    case 14: __asan_report_exp_load2(0, 0); break;
281    case 15: __asan_report_exp_load4(0, 0); break;
282    case 16: __asan_report_exp_load8(0, 0); break;
283    case 17: __asan_report_exp_load16(0, 0); break;
284    case 18: __asan_report_exp_load_n(0, 0, 0); break;
285    case 19: __asan_report_exp_store1(0, 0); break;
286    case 20: __asan_report_exp_store2(0, 0); break;
287    case 21: __asan_report_exp_store4(0, 0); break;
288    case 22: __asan_report_exp_store8(0, 0); break;
289    case 23: __asan_report_exp_store16(0, 0); break;
290    case 24: __asan_report_exp_store_n(0, 0, 0); break;
291    case 25: __asan_register_globals(nullptr, 0); break;
292    case 26: __asan_unregister_globals(nullptr, 0); break;
293    case 27: __asan_set_death_callback(nullptr); break;
294    case 28: __asan_set_error_report_callback(nullptr); break;
295    case 29: __asan_handle_no_return(); break;
296    case 30: __asan_address_is_poisoned(nullptr); break;
297    case 31: __asan_poison_memory_region(nullptr, 0); break;
298    case 32: __asan_unpoison_memory_region(nullptr, 0); break;
299    case 34: __asan_before_dynamic_init(nullptr); break;
300    case 35: __asan_after_dynamic_init(); break;
301    case 36: __asan_poison_stack_memory(0, 0); break;
302    case 37: __asan_unpoison_stack_memory(0, 0); break;
303    case 38: __asan_region_is_poisoned(0, 0); break;
304    case 39: __asan_describe_address(0); break;
305  }
306}
307
308static void asan_atexit() {
309  Printf("AddressSanitizer exit stats:\n");
310  __asan_print_accumulated_stats();
311  // Print AsanMappingProfile.
312  for (uptr i = 0; i < kAsanMappingProfileSize; i++) {
313    if (AsanMappingProfile[i] == 0) continue;
314    Printf("asan_mapping.h:%zd -- %zd\n", i, AsanMappingProfile[i]);
315  }
316}
317
318static void InitializeHighMemEnd() {
319#if !ASAN_FIXED_MAPPING
320  kHighMemEnd = GetMaxVirtualAddress();
321  // Increase kHighMemEnd to make sure it's properly
322  // aligned together with kHighMemBeg:
323  kHighMemEnd |= SHADOW_GRANULARITY * GetPageSizeCached() - 1;
324#endif  // !ASAN_FIXED_MAPPING
325  CHECK_EQ((kHighMemBeg % GetPageSizeCached()), 0);
326}
327
328static void ProtectGap(uptr addr, uptr size) {
329  if (!flags()->protect_shadow_gap)
330    return;
331  void *res = MmapNoAccess(addr, size, "shadow gap");
332  if (addr == (uptr)res)
333    return;
334  // A few pages at the start of the address space can not be protected.
335  // But we really want to protect as much as possible, to prevent this memory
336  // being returned as a result of a non-FIXED mmap().
337  if (addr == kZeroBaseShadowStart) {
338    uptr step = GetPageSizeCached();
339    while (size > step && addr < kZeroBaseMaxShadowStart) {
340      addr += step;
341      size -= step;
342      void *res = MmapNoAccess(addr, size, "shadow gap");
343      if (addr == (uptr)res)
344        return;
345    }
346  }
347
348  Report("ERROR: Failed to protect the shadow gap. "
349         "ASan cannot proceed correctly. ABORTING.\n");
350  DumpProcessMap();
351  Die();
352}
353
354static void PrintAddressSpaceLayout() {
355  Printf("|| `[%p, %p]` || HighMem    ||\n",
356         (void*)kHighMemBeg, (void*)kHighMemEnd);
357  Printf("|| `[%p, %p]` || HighShadow ||\n",
358         (void*)kHighShadowBeg, (void*)kHighShadowEnd);
359  if (kMidMemBeg) {
360    Printf("|| `[%p, %p]` || ShadowGap3 ||\n",
361           (void*)kShadowGap3Beg, (void*)kShadowGap3End);
362    Printf("|| `[%p, %p]` || MidMem     ||\n",
363           (void*)kMidMemBeg, (void*)kMidMemEnd);
364    Printf("|| `[%p, %p]` || ShadowGap2 ||\n",
365           (void*)kShadowGap2Beg, (void*)kShadowGap2End);
366    Printf("|| `[%p, %p]` || MidShadow  ||\n",
367           (void*)kMidShadowBeg, (void*)kMidShadowEnd);
368  }
369  Printf("|| `[%p, %p]` || ShadowGap  ||\n",
370         (void*)kShadowGapBeg, (void*)kShadowGapEnd);
371  if (kLowShadowBeg) {
372    Printf("|| `[%p, %p]` || LowShadow  ||\n",
373           (void*)kLowShadowBeg, (void*)kLowShadowEnd);
374    Printf("|| `[%p, %p]` || LowMem     ||\n",
375           (void*)kLowMemBeg, (void*)kLowMemEnd);
376  }
377  Printf("MemToShadow(shadow): %p %p %p %p",
378         (void*)MEM_TO_SHADOW(kLowShadowBeg),
379         (void*)MEM_TO_SHADOW(kLowShadowEnd),
380         (void*)MEM_TO_SHADOW(kHighShadowBeg),
381         (void*)MEM_TO_SHADOW(kHighShadowEnd));
382  if (kMidMemBeg) {
383    Printf(" %p %p",
384           (void*)MEM_TO_SHADOW(kMidShadowBeg),
385           (void*)MEM_TO_SHADOW(kMidShadowEnd));
386  }
387  Printf("\n");
388  Printf("redzone=%zu\n", (uptr)flags()->redzone);
389  Printf("max_redzone=%zu\n", (uptr)flags()->max_redzone);
390  Printf("quarantine_size_mb=%zuM\n", (uptr)flags()->quarantine_size_mb);
391  Printf("malloc_context_size=%zu\n",
392         (uptr)common_flags()->malloc_context_size);
393
394  Printf("SHADOW_SCALE: %d\n", (int)SHADOW_SCALE);
395  Printf("SHADOW_GRANULARITY: %d\n", (int)SHADOW_GRANULARITY);
396  Printf("SHADOW_OFFSET: 0x%zx\n", (uptr)SHADOW_OFFSET);
397  CHECK(SHADOW_SCALE >= 3 && SHADOW_SCALE <= 7);
398  if (kMidMemBeg)
399    CHECK(kMidShadowBeg > kLowShadowEnd &&
400          kMidMemBeg > kMidShadowEnd &&
401          kHighShadowBeg > kMidMemEnd);
402}
403
404static void AsanInitInternal() {
405  if (LIKELY(asan_inited)) return;
406  SanitizerToolName = "AddressSanitizer";
407  CHECK(!asan_init_is_running && "ASan init calls itself!");
408  asan_init_is_running = true;
409
410  CacheBinaryName();
411
412  // Initialize flags. This must be done early, because most of the
413  // initialization steps look at flags().
414  InitializeFlags();
415
416  AsanCheckIncompatibleRT();
417  AsanCheckDynamicRTPrereqs();
418
419  SetCanPoisonMemory(flags()->poison_heap);
420  SetMallocContextSize(common_flags()->malloc_context_size);
421
422  InitializeHighMemEnd();
423
424  // Make sure we are not statically linked.
425  AsanDoesNotSupportStaticLinkage();
426
427  // Install tool-specific callbacks in sanitizer_common.
428  AddDieCallback(AsanDie);
429  SetCheckFailedCallback(AsanCheckFailed);
430  SetPrintfAndReportCallback(AppendToErrorMessageBuffer);
431
432  __sanitizer_set_report_path(common_flags()->log_path);
433
434  // Enable UAR detection, if required.
435  __asan_option_detect_stack_use_after_return =
436      flags()->detect_stack_use_after_return;
437
438  // Re-exec ourselves if we need to set additional env or command line args.
439  MaybeReexec();
440
441  // Setup internal allocator callback.
442  SetLowLevelAllocateCallback(OnLowLevelAllocate);
443
444  InitializeAsanInterceptors();
445
446  // Enable system log ("adb logcat") on Android.
447  // Doing this before interceptors are initialized crashes in:
448  // AsanInitInternal -> android_log_write -> __interceptor_strcmp
449  AndroidLogInit();
450
451  ReplaceSystemMalloc();
452
453  uptr shadow_start = kLowShadowBeg;
454  if (kLowShadowBeg)
455    shadow_start -= GetMmapGranularity();
456  bool full_shadow_is_available =
457      MemoryRangeIsAvailable(shadow_start, kHighShadowEnd);
458
459#if SANITIZER_LINUX && defined(__x86_64__) && defined(_LP64) &&                \
460    !ASAN_FIXED_MAPPING
461  if (!full_shadow_is_available) {
462    kMidMemBeg = kLowMemEnd < 0x3000000000ULL ? 0x3000000000ULL : 0;
463    kMidMemEnd = kLowMemEnd < 0x3000000000ULL ? 0x4fffffffffULL : 0;
464  }
465#endif
466
467  if (Verbosity()) PrintAddressSpaceLayout();
468
469  DisableCoreDumperIfNecessary();
470
471  if (full_shadow_is_available) {
472    // mmap the low shadow plus at least one page at the left.
473    if (kLowShadowBeg)
474      ReserveShadowMemoryRange(shadow_start, kLowShadowEnd, "low shadow");
475    // mmap the high shadow.
476    ReserveShadowMemoryRange(kHighShadowBeg, kHighShadowEnd, "high shadow");
477    // protect the gap.
478    ProtectGap(kShadowGapBeg, kShadowGapEnd - kShadowGapBeg + 1);
479    CHECK_EQ(kShadowGapEnd, kHighShadowBeg - 1);
480  } else if (kMidMemBeg &&
481      MemoryRangeIsAvailable(shadow_start, kMidMemBeg - 1) &&
482      MemoryRangeIsAvailable(kMidMemEnd + 1, kHighShadowEnd)) {
483    CHECK(kLowShadowBeg != kLowShadowEnd);
484    // mmap the low shadow plus at least one page at the left.
485    ReserveShadowMemoryRange(shadow_start, kLowShadowEnd, "low shadow");
486    // mmap the mid shadow.
487    ReserveShadowMemoryRange(kMidShadowBeg, kMidShadowEnd, "mid shadow");
488    // mmap the high shadow.
489    ReserveShadowMemoryRange(kHighShadowBeg, kHighShadowEnd, "high shadow");
490    // protect the gaps.
491    ProtectGap(kShadowGapBeg, kShadowGapEnd - kShadowGapBeg + 1);
492    ProtectGap(kShadowGap2Beg, kShadowGap2End - kShadowGap2Beg + 1);
493    ProtectGap(kShadowGap3Beg, kShadowGap3End - kShadowGap3Beg + 1);
494  } else {
495    Report("Shadow memory range interleaves with an existing memory mapping. "
496           "ASan cannot proceed correctly. ABORTING.\n");
497    Report("ASan shadow was supposed to be located in the [%p-%p] range.\n",
498           shadow_start, kHighShadowEnd);
499    DumpProcessMap();
500    Die();
501  }
502
503  AsanTSDInit(PlatformTSDDtor);
504  InstallDeadlySignalHandlers(AsanOnDeadlySignal);
505
506  AllocatorOptions allocator_options;
507  allocator_options.SetFrom(flags(), common_flags());
508  InitializeAllocator(allocator_options);
509
510  MaybeStartBackgroudThread();
511  SetSoftRssLimitExceededCallback(AsanSoftRssLimitExceededCallback);
512
513  // On Linux AsanThread::ThreadStart() calls malloc() that's why asan_inited
514  // should be set to 1 prior to initializing the threads.
515  asan_inited = 1;
516  asan_init_is_running = false;
517
518  if (flags()->atexit)
519    Atexit(asan_atexit);
520
521  InitializeCoverage(common_flags()->coverage, common_flags()->coverage_dir);
522
523  // Now that ASan runtime is (mostly) initialized, deactivate it if
524  // necessary, so that it can be re-activated when requested.
525  if (flags()->start_deactivated)
526    AsanDeactivate();
527
528  // interceptors
529  InitTlsSize();
530
531  // Create main thread.
532  AsanThread *main_thread = AsanThread::Create(
533      /* start_routine */ nullptr, /* arg */ nullptr, /* parent_tid */ 0,
534      /* stack */ nullptr, /* detached */ true);
535  CHECK_EQ(0, main_thread->tid());
536  SetCurrentThread(main_thread);
537  main_thread->ThreadStart(internal_getpid(),
538                           /* signal_thread_is_registered */ nullptr);
539  force_interface_symbols();  // no-op.
540  SanitizerInitializeUnwinder();
541
542#if CAN_SANITIZE_LEAKS
543  __lsan::InitCommonLsan();
544  if (common_flags()->detect_leaks && common_flags()->leak_check_at_exit) {
545    Atexit(__lsan::DoLeakCheck);
546  }
547#endif  // CAN_SANITIZE_LEAKS
548
549#if CAN_SANITIZE_UB
550  __ubsan::InitAsPlugin();
551#endif
552
553  InitializeSuppressions();
554
555  VReport(1, "AddressSanitizer Init done\n");
556}
557
558// Initialize as requested from some part of ASan runtime library (interceptors,
559// allocator, etc).
560void AsanInitFromRtl() {
561  AsanInitInternal();
562}
563
564#if ASAN_DYNAMIC
565// Initialize runtime in case it's LD_PRELOAD-ed into unsanitized executable
566// (and thus normal initializers from .preinit_array or modules haven't run).
567
568class AsanInitializer {
569public:  // NOLINT
570  AsanInitializer() {
571    AsanInitFromRtl();
572  }
573};
574
575static AsanInitializer asan_initializer;
576#endif  // ASAN_DYNAMIC
577
578} // namespace __asan
579
580// ---------------------- Interface ---------------- {{{1
581using namespace __asan;  // NOLINT
582
583void NOINLINE __asan_handle_no_return() {
584  int local_stack;
585  AsanThread *curr_thread = GetCurrentThread();
586  uptr PageSize = GetPageSizeCached();
587  uptr top, bottom;
588  if (curr_thread) {
589    top = curr_thread->stack_top();
590    bottom = ((uptr)&local_stack - PageSize) & ~(PageSize - 1);
591  } else {
592    // If we haven't seen this thread, try asking the OS for stack bounds.
593    uptr tls_addr, tls_size, stack_size;
594    GetThreadStackAndTls(/*main=*/false, &bottom, &stack_size, &tls_addr,
595                         &tls_size);
596    top = bottom + stack_size;
597  }
598  static const uptr kMaxExpectedCleanupSize = 64 << 20;  // 64M
599  if (top - bottom > kMaxExpectedCleanupSize) {
600    static bool reported_warning = false;
601    if (reported_warning)
602      return;
603    reported_warning = true;
604    Report("WARNING: ASan is ignoring requested __asan_handle_no_return: "
605           "stack top: %p; bottom %p; size: %p (%zd)\n"
606           "False positive error reports may follow\n"
607           "For details see "
608           "https://github.com/google/sanitizers/issues/189\n",
609           top, bottom, top - bottom, top - bottom);
610    return;
611  }
612  PoisonShadow(bottom, top - bottom, 0);
613  if (curr_thread && curr_thread->has_fake_stack())
614    curr_thread->fake_stack()->HandleNoReturn();
615}
616
617void NOINLINE __asan_set_death_callback(void (*callback)(void)) {
618  SetUserDieCallback(callback);
619}
620
621// Initialize as requested from instrumented application code.
622// We use this call as a trigger to wake up ASan from deactivated state.
623void __asan_init() {
624  AsanActivate();
625  AsanInitInternal();
626}
627
628void __asan_version_mismatch_check() {
629  // Do nothing.
630}
631