contrib/compiler-rt/lib/sanitizer_common/sanitizer_flags.cc

   1 //===-- sanitizer_flags.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 ThreadSanitizer/AddressSanitizer runtime.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "sanitizer_flags.h"
  15
  16 #include "sanitizer_common.h"
  17 #include "sanitizer_libc.h"
  18 #include "sanitizer_list.h"
  19 #include "sanitizer_flag_parser.h"
  20
  21 namespace __sanitizer {
  22
  23 CommonFlags common_flags_dont_use;
  24
  25 struct FlagDescription {
  26   const char *name;
  27   const char *description;
  28   FlagDescription *next;
  29 };
  30
  31 IntrusiveList<FlagDescription> flag_descriptions;
  32
  33 // If set, the tool will install its own SEGV signal handler by default.
  34 #ifndef SANITIZER_NEEDS_SEGV
  35 # define SANITIZER_NEEDS_SEGV 1
  36 #endif
  37
  38 void CommonFlags::SetDefaults() {
  39 #define COMMON_FLAG(Type, Name, DefaultValue, Description) Name = DefaultValue;
  40 #include "sanitizer_flags.inc"
  41 #undef COMMON_FLAG
  42 }
  43
  44 void CommonFlags::CopyFrom(const CommonFlags &other) {
  45   internal_memcpy(this, &other, sizeof(*this));
  46 }
  47
  48 class FlagHandlerInclude : public FlagHandlerBase {
  49   static const uptr kMaxIncludeSize = 1 << 15;
  50   FlagParser *parser_;
  51
  52  public:
  53   explicit FlagHandlerInclude(FlagParser *parser) : parser_(parser) {}
  54   bool Parse(const char *value) {
  55     char *data;
  56     uptr data_mapped_size;
  57     int err;
  58     uptr len =
  59       ReadFileToBuffer(value, &data, &data_mapped_size,
  60                        Max(kMaxIncludeSize, GetPageSizeCached()), &err);
  61     if (!len) {
  62       Printf("Failed to read options from '%s': error %d\n", value, err);
  63       return false;
  64     }
  65     parser_->ParseString(data);
  66     UnmapOrDie(data, data_mapped_size);
  67     return true;
  68   }
  69 };
  70
  71 void RegisterIncludeFlag(FlagParser *parser, CommonFlags *cf) {
  72   FlagHandlerInclude *fh_include =
  73       new (FlagParser::Alloc) FlagHandlerInclude(parser);  // NOLINT
  74   parser->RegisterHandler("include", fh_include,
  75                           "read more options from the given file");
  76 }
  77
  78 void RegisterCommonFlags(FlagParser *parser, CommonFlags *cf) {
  79 #define COMMON_FLAG(Type, Name, DefaultValue, Description) \
  80   RegisterFlag(parser, #Name, Description, &cf->Name);
  81 #include "sanitizer_flags.inc"
  82 #undef COMMON_FLAG
  83
  84   RegisterIncludeFlag(parser, cf);
  85 }
  86
  87 }  // namespace __sanitizer