llvm-project
643 строки · 22.4 Кб
1//===-- asan_rtl.cpp ------------------------------------------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8//9// This file is a part of AddressSanitizer, an address sanity checker.10//11// Main file of the ASan run-time library.12//===----------------------------------------------------------------------===//13}67}72// -------------------------- Globals --------------------- {{{174static StaticSpinMutex asan_inited_mutex;75static atomic_uint8_t asan_inited = {0};76}80}84// -------------------------- Misc ---------------- {{{192void ShowStatsAndAbort() {93__asan_print_accumulated_stats();94Die();95}96NOINLINE98static void ReportGenericErrorWrapper(uptr addr, bool is_write, int size,99int exp_arg, bool fatal) {100GET_CALLER_PC_BP_SP;101ReportGenericError(pc, bp, sp, addr, is_write, size, exp_arg, fatal);102}103// --------------- LowLevelAllocateCallbac ---------- {{{1105static void OnLowLevelAllocate(uptr ptr, uptr size) {106PoisonShadow(ptr, size, kAsanInternalHeapMagic);107}108// -------------------------- Run-time entry ------------------- {{{1110// exported functions111#define ASAN_REPORT_ERROR(type, is_write, size) \112extern "C" NOINLINE INTERFACE_ATTRIBUTE \113void __asan_report_ ## type ## size(uptr addr) { \114GET_CALLER_PC_BP_SP; \115ReportGenericError(pc, bp, sp, addr, is_write, size, 0, true); \116} \117extern "C" NOINLINE INTERFACE_ATTRIBUTE \118void __asan_report_exp_ ## type ## size(uptr addr, u32 exp) { \119GET_CALLER_PC_BP_SP; \120ReportGenericError(pc, bp, sp, addr, is_write, size, exp, true); \121} \122extern "C" NOINLINE INTERFACE_ATTRIBUTE \123void __asan_report_ ## type ## size ## _noabort(uptr addr) { \124GET_CALLER_PC_BP_SP; \125ReportGenericError(pc, bp, sp, addr, is_write, size, 0, false); \126} \127128ASAN_REPORT_ERROR(load, false, 1)129ASAN_REPORT_ERROR(load, false, 2)130ASAN_REPORT_ERROR(load, false, 4)131ASAN_REPORT_ERROR(load, false, 8)132ASAN_REPORT_ERROR(load, false, 16)133ASAN_REPORT_ERROR(store, true, 1)134ASAN_REPORT_ERROR(store, true, 2)135ASAN_REPORT_ERROR(store, true, 4)136ASAN_REPORT_ERROR(store, true, 8)137ASAN_REPORT_ERROR(store, true, 16)138NOINLINE INTERFACE_ATTRIBUTE198void __asan_loadN(uptr addr, uptr size) {199if ((addr = __asan_region_is_poisoned(addr, size))) {200GET_CALLER_PC_BP_SP;201ReportGenericError(pc, bp, sp, addr, false, size, 0, true);202}203}204NOINLINE INTERFACE_ATTRIBUTE207void __asan_exp_loadN(uptr addr, uptr size, u32 exp) {208if ((addr = __asan_region_is_poisoned(addr, size))) {209GET_CALLER_PC_BP_SP;210ReportGenericError(pc, bp, sp, addr, false, size, exp, true);211}212}213NOINLINE INTERFACE_ATTRIBUTE216void __asan_loadN_noabort(uptr addr, uptr size) {217if ((addr = __asan_region_is_poisoned(addr, size))) {218GET_CALLER_PC_BP_SP;219ReportGenericError(pc, bp, sp, addr, false, size, 0, false);220}221}222NOINLINE INTERFACE_ATTRIBUTE225void __asan_storeN(uptr addr, uptr size) {226if ((addr = __asan_region_is_poisoned(addr, size))) {227GET_CALLER_PC_BP_SP;228ReportGenericError(pc, bp, sp, addr, true, size, 0, true);229}230}231NOINLINE INTERFACE_ATTRIBUTE234void __asan_exp_storeN(uptr addr, uptr size, u32 exp) {235if ((addr = __asan_region_is_poisoned(addr, size))) {236GET_CALLER_PC_BP_SP;237ReportGenericError(pc, bp, sp, addr, true, size, exp, true);238}239}240NOINLINE INTERFACE_ATTRIBUTE243void __asan_storeN_noabort(uptr addr, uptr size) {244if ((addr = __asan_region_is_poisoned(addr, size))) {245GET_CALLER_PC_BP_SP;246ReportGenericError(pc, bp, sp, addr, true, size, 0, false);247}248}249// Force the linker to keep the symbols for various ASan interface functions.251// We want to keep those in the executable in order to let the instrumented252// dynamic libraries access the symbol even if it is not used by the executable253// itself. This should help if the build system is removing dead code at link254// time.255static NOINLINE void force_interface_symbols() {256volatile int fake_condition = 0; // prevent dead condition elimination.257// __asan_report_* functions are noreturn, so we need a switch to prevent258// the compiler from removing any of them.259// clang-format off260switch (fake_condition) {261case 1: __asan_report_load1(0); break;262case 2: __asan_report_load2(0); break;263case 3: __asan_report_load4(0); break;264case 4: __asan_report_load8(0); break;265case 5: __asan_report_load16(0); break;266case 6: __asan_report_load_n(0, 0); break;267case 7: __asan_report_store1(0); break;268case 8: __asan_report_store2(0); break;269case 9: __asan_report_store4(0); break;270case 10: __asan_report_store8(0); break;271case 11: __asan_report_store16(0); break;272case 12: __asan_report_store_n(0, 0); break;273case 13: __asan_report_exp_load1(0, 0); break;274case 14: __asan_report_exp_load2(0, 0); break;275case 15: __asan_report_exp_load4(0, 0); break;276case 16: __asan_report_exp_load8(0, 0); break;277case 17: __asan_report_exp_load16(0, 0); break;278case 18: __asan_report_exp_load_n(0, 0, 0); break;279case 19: __asan_report_exp_store1(0, 0); break;280case 20: __asan_report_exp_store2(0, 0); break;281case 21: __asan_report_exp_store4(0, 0); break;282case 22: __asan_report_exp_store8(0, 0); break;283case 23: __asan_report_exp_store16(0, 0); break;284case 24: __asan_report_exp_store_n(0, 0, 0); break;285case 25: __asan_register_globals(nullptr, 0); break;286case 26: __asan_unregister_globals(nullptr, 0); break;287case 27: __asan_set_death_callback(nullptr); break;288case 28: __asan_set_error_report_callback(nullptr); break;289case 29: __asan_handle_no_return(); break;290case 30: __asan_address_is_poisoned(nullptr); break;291case 31: __asan_poison_memory_region(nullptr, 0); break;292case 32: __asan_unpoison_memory_region(nullptr, 0); break;293case 34: __asan_before_dynamic_init(nullptr); break;294case 35: __asan_after_dynamic_init(); break;295case 36: __asan_poison_stack_memory(0, 0); break;296case 37: __asan_unpoison_stack_memory(0, 0); break;297case 38: __asan_region_is_poisoned(0, 0); break;298case 39: __asan_describe_address(0); break;299case 40: __asan_set_shadow_00(0, 0); break;300case 41: __asan_set_shadow_01(0, 0); break;301case 42: __asan_set_shadow_02(0, 0); break;302case 43: __asan_set_shadow_03(0, 0); break;303case 44: __asan_set_shadow_04(0, 0); break;304case 45: __asan_set_shadow_05(0, 0); break;305case 46: __asan_set_shadow_06(0, 0); break;306case 47: __asan_set_shadow_07(0, 0); break;307case 48: __asan_set_shadow_f1(0, 0); break;308case 49: __asan_set_shadow_f2(0, 0); break;309case 50: __asan_set_shadow_f3(0, 0); break;310case 51: __asan_set_shadow_f5(0, 0); break;311case 52: __asan_set_shadow_f8(0, 0); break;312}313// clang-format on314}315}325}335}392}518// Initialize as requested from some part of ASan runtime library (interceptors,520// allocator, etc).521void AsanInitFromRtl() {522if (LIKELY(AsanInited()))523return;524SpinMutexLock lock(&asan_inited_mutex);525AsanInitInternal();526}527}537// Initialize runtime in case it's LD_PRELOAD-ed into unsanitized executable540// (and thus normal initializers from .preinit_array or modules haven't run).541}571}591}601// ---------------------- Interface ---------------- {{{1605using namespace __asan;606}616}622}629}633// Initialize as requested from instrumented application code.635// We use this call as a trigger to wake up ASan from deactivated state.636void __asan_init() {637AsanActivate();638AsanInitFromRtl();639}640}644