ubsan_init.cc revision 1.1.1.2 
1//===-- ubsan_init.cc -----------------------------------------------------===//
2//
3// This file is distributed under the University of Illinois Open Source
4// License. See LICENSE.TXT for details.
5//
6//===----------------------------------------------------------------------===//
7//
8// Initialization of UBSan runtime.
9//
10//===----------------------------------------------------------------------===//
11
12#include "ubsan_platform.h"
13#if CAN_SANITIZE_UB
14#include "ubsan_diag.h"
15#include "ubsan_init.h"
16#include "ubsan_flags.h"
17#include "sanitizer_common/sanitizer_common.h"
18#include "sanitizer_common/sanitizer_libc.h"
19#include "sanitizer_common/sanitizer_mutex.h"
20#include "sanitizer_common/sanitizer_symbolizer.h"
21
22using namespace __ubsan;
23
24static enum {
25  UBSAN_MODE_UNKNOWN = 0,
26  UBSAN_MODE_STANDALONE,
27  UBSAN_MODE_PLUGIN
28} ubsan_mode;
29static StaticSpinMutex ubsan_init_mu;
30
31static void CommonInit() {
32  InitializeSuppressions();
33}
34
35static void CommonStandaloneInit() {
36  SanitizerToolName = "UndefinedBehaviorSanitizer";
37  InitializeFlags();
38  CacheBinaryName();
39  __sanitizer_set_report_path(common_flags()->log_path);
40  InitializeCoverage(common_flags()->coverage, common_flags()->coverage_dir);
41  CommonInit();
42  ubsan_mode = UBSAN_MODE_STANDALONE;
43}
44
45void __ubsan::InitAsStandalone() {
46  if (SANITIZER_CAN_USE_PREINIT_ARRAY) {
47    CHECK_EQ(UBSAN_MODE_UNKNOWN, ubsan_mode);
48    CommonStandaloneInit();
49    return;
50  }
51  SpinMutexLock l(&ubsan_init_mu);
52  CHECK_NE(UBSAN_MODE_PLUGIN, ubsan_mode);
53  if (ubsan_mode == UBSAN_MODE_UNKNOWN)
54    CommonStandaloneInit();
55}
56
57void __ubsan::InitAsStandaloneIfNecessary() {
58  if (SANITIZER_CAN_USE_PREINIT_ARRAY) {
59    CHECK_NE(UBSAN_MODE_UNKNOWN, ubsan_mode);
60    return;
61  }
62  SpinMutexLock l(&ubsan_init_mu);
63  if (ubsan_mode == UBSAN_MODE_UNKNOWN)
64    CommonStandaloneInit();
65}
66
67void __ubsan::InitAsPlugin() {
68#if !SANITIZER_CAN_USE_PREINIT_ARRAY
69  SpinMutexLock l(&ubsan_init_mu);
70#endif
71  CHECK_EQ(UBSAN_MODE_UNKNOWN, ubsan_mode);
72  CommonInit();
73  ubsan_mode = UBSAN_MODE_PLUGIN;
74}
75
76#endif  // CAN_SANITIZE_UB
77