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 
GetSanititizerToolName()25 const char *__ubsan::GetSanititizerToolName() {
26   return "UndefinedBehaviorSanitizer";
27 }
28 
29 static bool ubsan_initialized;
30 static StaticSpinMutex ubsan_init_mu;
31 
CommonInit()32 static void CommonInit() {
33   InitializeSuppressions();
34 }
35 
UbsanDie()36 static void UbsanDie() {
37   if (common_flags()->print_module_map >= 1)
38     DumpProcessMap();
39 }
40 
CommonStandaloneInit()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 
InitAsStandalone()57 void __ubsan::InitAsStandalone() {
58   SpinMutexLock l(&ubsan_init_mu);
59   if (!ubsan_initialized) {
60     CommonStandaloneInit();
61     ubsan_initialized = true;
62   }
63 }
64 
InitAsStandaloneIfNecessary()65 void __ubsan::InitAsStandaloneIfNecessary() { return InitAsStandalone(); }
66 
InitAsPlugin()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