1 //===--- SanitizerArgs.h - Arguments for sanitizer tools  -------*- 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 #ifndef LLVM_CLANG_DRIVER_SANITIZERARGS_H
9 #define LLVM_CLANG_DRIVER_SANITIZERARGS_H
10 
11 #include "clang/Basic/Sanitizers.h"
12 #include "clang/Driver/Types.h"
13 #include "llvm/Option/Arg.h"
14 #include "llvm/Option/ArgList.h"
15 #include "llvm/Transforms/Instrumentation/AddressSanitizerOptions.h"
16 #include <string>
17 #include <vector>
18 
19 namespace clang {
20 namespace driver {
21 
22 class ToolChain;
23 
24 class SanitizerArgs {
25   SanitizerSet Sanitizers;
26   SanitizerSet RecoverableSanitizers;
27   SanitizerSet TrapSanitizers;
28 
29   std::vector<std::string> UserIgnorelistFiles;
30   std::vector<std::string> SystemIgnorelistFiles;
31   std::vector<std::string> CoverageAllowlistFiles;
32   std::vector<std::string> CoverageIgnorelistFiles;
33   int CoverageFeatures = 0;
34   int MsanTrackOrigins = 0;
35   bool MsanUseAfterDtor = true;
36   bool MsanParamRetval = false;
37   bool CfiCrossDso = false;
38   bool CfiICallGeneralizePointers = false;
39   bool CfiCanonicalJumpTables = false;
40   int AsanFieldPadding = 0;
41   bool SharedRuntime = false;
42   bool AsanUseAfterScope = true;
43   bool AsanPoisonCustomArrayCookie = false;
44   bool AsanGlobalsDeadStripping = false;
45   bool AsanUseOdrIndicator = false;
46   bool AsanInvalidPointerCmp = false;
47   bool AsanInvalidPointerSub = false;
48   bool AsanOutlineInstrumentation = false;
49   llvm::AsanDtorKind AsanDtorKind = llvm::AsanDtorKind::Invalid;
50   std::string HwasanAbi;
51   bool LinkRuntimes = true;
52   bool LinkCXXRuntimes = false;
53   bool NeedPIE = false;
54   bool SafeStackRuntime = false;
55   bool Stats = false;
56   bool TsanMemoryAccess = true;
57   bool TsanFuncEntryExit = true;
58   bool TsanAtomics = true;
59   bool MinimalRuntime = false;
60   // True if cross-dso CFI support if provided by the system (i.e. Android).
61   bool ImplicitCfiRuntime = false;
62   bool NeedsMemProfRt = false;
63   bool HwasanUseAliases = false;
64   llvm::AsanDetectStackUseAfterReturnMode AsanUseAfterReturn =
65       llvm::AsanDetectStackUseAfterReturnMode::Invalid;
66 
67 public:
68   /// Parses the sanitizer arguments from an argument list.
69   SanitizerArgs(const ToolChain &TC, const llvm::opt::ArgList &Args,
70                 bool DiagnoseErrors = true);
71 
72   bool needsSharedRt() const { return SharedRuntime; }
73 
74   bool needsMemProfRt() const { return NeedsMemProfRt; }
75   bool needsAsanRt() const { return Sanitizers.has(SanitizerKind::Address); }
76   bool needsHwasanRt() const {
77     return Sanitizers.has(SanitizerKind::HWAddress);
78   }
79   bool needsHwasanAliasesRt() const {
80     return needsHwasanRt() && HwasanUseAliases;
81   }
82   bool needsTsanRt() const { return Sanitizers.has(SanitizerKind::Thread); }
83   bool needsMsanRt() const { return Sanitizers.has(SanitizerKind::Memory); }
84   bool needsFuzzer() const { return Sanitizers.has(SanitizerKind::Fuzzer); }
85   bool needsLsanRt() const {
86     return Sanitizers.has(SanitizerKind::Leak) &&
87            !Sanitizers.has(SanitizerKind::Address) &&
88            !Sanitizers.has(SanitizerKind::HWAddress);
89   }
90   bool needsFuzzerInterceptors() const;
91   bool needsUbsanRt() const;
92   bool requiresMinimalRuntime() const { return MinimalRuntime; }
93   bool needsDfsanRt() const { return Sanitizers.has(SanitizerKind::DataFlow); }
94   bool needsSafeStackRt() const { return SafeStackRuntime; }
95   bool needsCfiRt() const;
96   bool needsCfiDiagRt() const;
97   bool needsStatsRt() const { return Stats; }
98   bool needsScudoRt() const { return Sanitizers.has(SanitizerKind::Scudo); }
99 
100   bool requiresPIE() const;
101   bool needsUnwindTables() const;
102   bool needsLTO() const;
103   bool linkRuntimes() const { return LinkRuntimes; }
104   bool linkCXXRuntimes() const { return LinkCXXRuntimes; }
105   bool hasCrossDsoCfi() const { return CfiCrossDso; }
106   bool hasAnySanitizer() const { return !Sanitizers.empty(); }
107   void addArgs(const ToolChain &TC, const llvm::opt::ArgList &Args,
108                llvm::opt::ArgStringList &CmdArgs, types::ID InputType) const;
109 };
110 
111 }  // namespace driver
112 }  // namespace clang
113 
114 #endif
115