1 //===-- sanitizer_flags.cpp -----------------------------------------------===//
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
7 //===----------------------------------------------------------------------===//
9 // This file is a part of ThreadSanitizer/AddressSanitizer runtime.
11 //===----------------------------------------------------------------------===//
13 #include "sanitizer_flags.h"
15 #include "sanitizer_common.h"
16 #include "sanitizer_flag_parser.h"
17 #include "sanitizer_libc.h"
18 #include "sanitizer_linux.h"
19 #include "sanitizer_list.h"
21 namespace __sanitizer
{
23 CommonFlags common_flags_dont_use
;
25 void CommonFlags::SetDefaults() {
26 #define COMMON_FLAG(Type, Name, DefaultValue, Description) Name = DefaultValue;
27 #include "sanitizer_flags.inc"
31 void CommonFlags::CopyFrom(const CommonFlags
&other
) {
32 internal_memcpy(this, &other
, sizeof(*this));
35 // Copy the string from "s" to "out", making the following substitutions:
36 // %b = binary basename
38 void SubstituteForFlagValue(const char *s
, char *out
, uptr out_size
) {
39 char *out_end
= out
+ out_size
;
40 while (*s
&& out
< out_end
- 1) {
47 const char *base
= GetProcessName();
49 while (*base
&& out
< out_end
- 1)
55 int pid
= internal_getpid();
57 char *buf_pos
= buf
+ 32;
59 *--buf_pos
= (pid
% 10) + '0';
62 while (buf_pos
< buf
+ 32 && out
< out_end
- 1)
72 CHECK(out
< out_end
- 1);
76 class FlagHandlerInclude final
: public FlagHandlerBase
{
79 const char *original_path_
;
82 explicit FlagHandlerInclude(FlagParser
*parser
, bool ignore_missing
)
83 : parser_(parser
), ignore_missing_(ignore_missing
), original_path_("") {}
84 bool Parse(const char *value
) final
{
85 original_path_
= value
;
86 if (internal_strchr(value
, '%')) {
87 char *buf
= (char *)MmapOrDie(kMaxPathLength
, "FlagHandlerInclude");
88 SubstituteForFlagValue(value
, buf
, kMaxPathLength
);
89 bool res
= parser_
->ParseFile(buf
, ignore_missing_
);
90 UnmapOrDie(buf
, kMaxPathLength
);
93 return parser_
->ParseFile(value
, ignore_missing_
);
95 bool Format(char *buffer
, uptr size
) override
{
96 // Note `original_path_` isn't actually what's parsed due to `%`
97 // substitutions. Printing the substituted path would require holding onto
99 return FormatString(buffer
, size
, original_path_
);
103 void RegisterIncludeFlags(FlagParser
*parser
, CommonFlags
*cf
) {
104 FlagHandlerInclude
*fh_include
= new (FlagParser::Alloc
)
105 FlagHandlerInclude(parser
, /*ignore_missing*/ false);
106 parser
->RegisterHandler("include", fh_include
,
107 "read more options from the given file");
108 FlagHandlerInclude
*fh_include_if_exists
= new (FlagParser::Alloc
)
109 FlagHandlerInclude(parser
, /*ignore_missing*/ true);
110 parser
->RegisterHandler(
111 "include_if_exists", fh_include_if_exists
,
112 "read more options from the given file (if it exists)");
115 void RegisterCommonFlags(FlagParser
*parser
, CommonFlags
*cf
) {
116 #define COMMON_FLAG(Type, Name, DefaultValue, Description) \
117 RegisterFlag(parser, #Name, Description, &cf->Name);
118 #include "sanitizer_flags.inc"
121 RegisterIncludeFlags(parser
, cf
);
124 void InitializeCommonFlags(CommonFlags
*cf
) {
125 // need to record coverage to generate coverage report.
126 cf
->coverage
|= cf
->html_cov_report
;
127 SetVerbosity(cf
->verbosity
);
129 InitializePlatformCommonFlags(cf
);
132 } // namespace __sanitizer