contrib/compiler-rt/lib/lsan/lsan.cc

   1 //=-- lsan.cc -------------------------------------------------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file is a part of LeakSanitizer.
  11 // Standalone LSan RTL.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "lsan.h"
  16
  17 #include "sanitizer_common/sanitizer_flags.h"
  18 #include "sanitizer_common/sanitizer_flag_parser.h"
  19 #include "sanitizer_common/sanitizer_stacktrace.h"
  20 #include "lsan_allocator.h"
  21 #include "lsan_common.h"
  22 #include "lsan_thread.h"
  23
  24 bool lsan_inited;
  25 bool lsan_init_is_running;
  26
  27 namespace __lsan {
  28
  29 ///// Interface to the common LSan module. /////
  30 bool WordIsPoisoned(uptr addr) {
  31   return false;
  32 }
  33
  34 }  // namespace __lsan
  35
  36 using namespace __lsan;  // NOLINT
  37
  38 static void InitializeFlags() {
  39   // Set all the default values.
  40   SetCommonFlagsDefaults();
  41   {
  42     CommonFlags cf;
  43     cf.CopyFrom(*common_flags());
  44     cf.external_symbolizer_path = GetEnv("LSAN_SYMBOLIZER_PATH");
  45     cf.malloc_context_size = 30;
  46     cf.intercept_tls_get_addr = true;
  47     cf.detect_leaks = true;
  48     cf.exitcode = 23;
  49     OverrideCommonFlags(cf);
  50   }
  51
  52   Flags *f = flags();
  53   f->SetDefaults();
  54
  55   FlagParser parser;
  56   RegisterLsanFlags(&parser, f);
  57   RegisterCommonFlags(&parser);
  58
  59   // Override from user-specified string.
  60   const char *lsan_default_options = MaybeCallLsanDefaultOptions();
  61   parser.ParseString(lsan_default_options);
  62   parser.ParseString(GetEnv("LSAN_OPTIONS"));
  63
  64   SetVerbosity(common_flags()->verbosity);
  65
  66   if (Verbosity()) ReportUnrecognizedFlags();
  67
  68   if (common_flags()->help) parser.PrintFlagDescriptions();
  69 }
  70
  71 static void OnStackUnwind(const SignalContext &sig, const void *,
  72                           BufferedStackTrace *stack) {
  73   GetStackTrace(stack, kStackTraceMax, sig.pc, sig.bp, sig.context,
  74                 common_flags()->fast_unwind_on_fatal);
  75 }
  76
  77 static void LsanOnDeadlySignal(int signo, void *siginfo, void *context) {
  78   HandleDeadlySignal(siginfo, context, GetCurrentThread(), &OnStackUnwind,
  79                      nullptr);
  80 }
  81
  82 extern "C" void __lsan_init() {
  83   CHECK(!lsan_init_is_running);
  84   if (lsan_inited)
  85     return;
  86   lsan_init_is_running = true;
  87   SanitizerToolName = "LeakSanitizer";
  88   CacheBinaryName();
  89   AvoidCVE_2016_2143();
  90   InitializeFlags();
  91   InitCommonLsan();
  92   InitializeAllocator();
  93   ReplaceSystemMalloc();
  94   InitTlsSize();
  95   InitializeInterceptors();
  96   InitializeThreadRegistry();
  97   InstallDeadlySignalHandlers(LsanOnDeadlySignal);
  98   u32 tid = ThreadCreate(0, 0, true);
  99   CHECK_EQ(tid, 0);
 100   ThreadStart(tid, GetTid());
 101   SetCurrentThread(tid);
 102
 103   if (common_flags()->detect_leaks && common_flags()->leak_check_at_exit)
 104     Atexit(DoLeakCheck);
 105
 106   InitializeCoverage(common_flags()->coverage, common_flags()->coverage_dir);
 107
 108   lsan_inited = true;
 109   lsan_init_is_running = false;
 110 }
 111
 112 extern "C" SANITIZER_INTERFACE_ATTRIBUTE
 113 void __sanitizer_print_stack_trace() {
 114   GET_STACK_TRACE_FATAL;
 115   stack.Print();
 116 }