1 //===-- ubsan_init.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-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // Initialization of UBSan runtime. 10 // 11 //===----------------------------------------------------------------------===// 12 13 #include "ubsan_platform.h" 14 #if CAN_SANITIZE_UB 15 #include "ubsan_diag.h" 16 #include "ubsan_init.h" 17 #include "ubsan_flags.h" 18 #include "sanitizer_common/sanitizer_common.h" 19 #include "sanitizer_common/sanitizer_libc.h" 20 #include "sanitizer_common/sanitizer_mutex.h" 21 #include "sanitizer_common/sanitizer_symbolizer.h" 22 23 using namespace __ubsan; 24 25 const char *__ubsan::GetSanititizerToolName() { 26 return "UndefinedBehaviorSanitizer"; 27 } 28 29 static bool ubsan_initialized; 30 static StaticSpinMutex ubsan_init_mu; 31 32 static void CommonInit() { 33 InitializeSuppressions(); 34 } 35 36 static void UbsanDie() { 37 if (common_flags()->print_module_map >= 1) 38 DumpProcessMap(); 39 } 40 41 static void CommonStandaloneInit() { 42 SanitizerToolName = GetSanititizerToolName(); 43 CacheBinaryName(); 44 InitializeFlags(); 45 __sanitizer::InitializePlatformEarly(); 46 __sanitizer_set_report_path(common_flags()->log_path); 47 AndroidLogInit(); 48 InitializeCoverage(common_flags()->coverage, common_flags()->coverage_dir); 49 CommonInit(); 50 51 // Only add die callback when running in standalone mode to avoid printing 52 // the same information from multiple sanitizers' output 53 AddDieCallback(UbsanDie); 54 Symbolizer::LateInitialize(); 55 } 56 57 void __ubsan::InitAsStandalone() { 58 SpinMutexLock l(&ubsan_init_mu); 59 if (!ubsan_initialized) { 60 CommonStandaloneInit(); 61 ubsan_initialized = true; 62 } 63 } 64 65 void __ubsan::InitAsStandaloneIfNecessary() { return InitAsStandalone(); } 66 67 void __ubsan::InitAsPlugin() { 68 SpinMutexLock l(&ubsan_init_mu); 69 if (!ubsan_initialized) { 70 CommonInit(); 71 ubsan_initialized = true; 72 } 73 } 74 75 #endif // CAN_SANITIZE_UB 76