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 
SetDefaults()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 
CopyFrom(const CommonFlags & other)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:
FlagHandlerInclude(FlagParser * parser)53   explicit FlagHandlerInclude(FlagParser *parser) : parser_(parser) {}
Parse(const char * value)54   bool Parse(const char *value) final {
55     char *data;
56     uptr data_mapped_size;
57     error_t 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 
RegisterIncludeFlag(FlagParser * parser,CommonFlags * cf)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 
RegisterCommonFlags(FlagParser * parser,CommonFlags * cf)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
88