libsanitizer/ubsan/ubsan_init.cpp

   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