1 //===--- SanitizerArgs.h - Arguments for sanitizer tools  -------*- C++ -*-===//
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 #ifndef LLVM_CLANG_DRIVER_SANITIZERARGS_H
10 #define LLVM_CLANG_DRIVER_SANITIZERARGS_H
11 
12 #include "clang/Basic/Sanitizers.h"
13 #include "clang/Driver/Types.h"
14 #include "llvm/Option/Arg.h"
15 #include "llvm/Option/ArgList.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> BlacklistFiles;
30   std::vector<std::string> ExtraDeps;
31   int CoverageFeatures;
32   int MsanTrackOrigins;
33   bool MsanUseAfterDtor;
34   bool CfiCrossDso;
35   int AsanFieldPadding;
36   bool AsanSharedRuntime;
37   bool LinkCXXRuntimes;
38   bool NeedPIE;
39 
40  public:
41   /// Parses the sanitizer arguments from an argument list.
42   SanitizerArgs(const ToolChain &TC, const llvm::opt::ArgList &Args);
43 
needsAsanRt()44   bool needsAsanRt() const { return Sanitizers.has(SanitizerKind::Address); }
needsSharedAsanRt()45   bool needsSharedAsanRt() const { return AsanSharedRuntime; }
needsTsanRt()46   bool needsTsanRt() const { return Sanitizers.has(SanitizerKind::Thread); }
needsMsanRt()47   bool needsMsanRt() const { return Sanitizers.has(SanitizerKind::Memory); }
needsLsanRt()48   bool needsLsanRt() const {
49     return Sanitizers.has(SanitizerKind::Leak) &&
50            !Sanitizers.has(SanitizerKind::Address);
51   }
52   bool needsUbsanRt() const;
needsDfsanRt()53   bool needsDfsanRt() const { return Sanitizers.has(SanitizerKind::DataFlow); }
needsSafeStackRt()54   bool needsSafeStackRt() const {
55     return Sanitizers.has(SanitizerKind::SafeStack);
56   }
57   bool needsCfiRt() const;
58   bool needsCfiDiagRt() const;
59 
60   bool requiresPIE() const;
61   bool needsUnwindTables() const;
linkCXXRuntimes()62   bool linkCXXRuntimes() const { return LinkCXXRuntimes; }
63   void addArgs(const ToolChain &TC, const llvm::opt::ArgList &Args,
64                llvm::opt::ArgStringList &CmdArgs, types::ID InputType) const;
65 
66  private:
67   void clear();
68 };
69 
70 }  // namespace driver
71 }  // namespace clang
72 
73 #endif
74