1 //===-- memprof_flags.cpp --------------------------------------*- C++ -*-===//
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 // This file is a part of MemProfiler, a memory profiler.
10 //
11 // MemProf flag parsing logic.
12 //===----------------------------------------------------------------------===//
13 
14 #include "memprof_flags.h"
15 #include "memprof_interface_internal.h"
16 #include "memprof_stack.h"
17 #include "sanitizer_common/sanitizer_common.h"
18 #include "sanitizer_common/sanitizer_flag_parser.h"
19 #include "sanitizer_common/sanitizer_flags.h"
20 
21 namespace __memprof {
22 
23 Flags memprof_flags_dont_use_directly; // use via flags().
24 
25 static const char *MaybeUseMemprofDefaultOptionsCompileDefinition() {
26 #ifdef MEMPROF_DEFAULT_OPTIONS
27   return SANITIZER_STRINGIFY(MEMPROF_DEFAULT_OPTIONS);
28 #else
29   return "";
30 #endif
31 }
32 
33 void Flags::SetDefaults() {
34 #define MEMPROF_FLAG(Type, Name, DefaultValue, Description) Name = DefaultValue;
35 #include "memprof_flags.inc"
36 #undef MEMPROF_FLAG
37 }
38 
39 static void RegisterMemprofFlags(FlagParser *parser, Flags *f) {
40 #define MEMPROF_FLAG(Type, Name, DefaultValue, Description)                    \
41   RegisterFlag(parser, #Name, Description, &f->Name);
42 #include "memprof_flags.inc"
43 #undef MEMPROF_FLAG
44 }
45 
46 void InitializeFlags() {
47   // Set the default values and prepare for parsing MemProf and common flags.
48   SetCommonFlagsDefaults();
49   {
50     CommonFlags cf;
51     cf.CopyFrom(*common_flags());
52     cf.external_symbolizer_path = GetEnv("MEMPROF_SYMBOLIZER_PATH");
53     cf.malloc_context_size = kDefaultMallocContextSize;
54     cf.intercept_tls_get_addr = true;
55     cf.exitcode = 1;
56     OverrideCommonFlags(cf);
57   }
58   Flags *f = flags();
59   f->SetDefaults();
60 
61   FlagParser memprof_parser;
62   RegisterMemprofFlags(&memprof_parser, f);
63   RegisterCommonFlags(&memprof_parser);
64 
65   // Override from MemProf compile definition.
66   const char *memprof_compile_def =
67       MaybeUseMemprofDefaultOptionsCompileDefinition();
68   memprof_parser.ParseString(memprof_compile_def);
69 
70   // Override from user-specified string.
71   const char *memprof_default_options = __memprof_default_options();
72   memprof_parser.ParseString(memprof_default_options);
73 
74   // Override from command line.
75   memprof_parser.ParseStringFromEnv("MEMPROF_OPTIONS");
76 
77   InitializeCommonFlags();
78 
79   if (Verbosity())
80     ReportUnrecognizedFlags();
81 
82   if (common_flags()->help) {
83     memprof_parser.PrintFlagDescriptions();
84   }
85 
86   CHECK_LE((uptr)common_flags()->malloc_context_size, kStackTraceMax);
87 }
88 
89 } // namespace __memprof
90 
91 SANITIZER_INTERFACE_WEAK_DEF(const char *, __memprof_default_options, void) {
92   return "";
93 }
94