contrib/llvm/tools/clang/include/clang/Driver/SanitizerArgs.h

   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   int CoverageFeatures;
  31   int MsanTrackOrigins;
  32   int AsanFieldPadding;
  33   bool AsanZeroBaseShadow;
  34   bool AsanSharedRuntime;
  35   bool LinkCXXRuntimes;
  36
  37  public:
  38   /// Parses the sanitizer arguments from an argument list.
  39   SanitizerArgs(const ToolChain &TC, const llvm::opt::ArgList &Args);
  40
  41   bool needsAsanRt() const { return Sanitizers.has(SanitizerKind::Address); }
  42   bool needsSharedAsanRt() const { return AsanSharedRuntime; }
  43   bool needsTsanRt() const { return Sanitizers.has(SanitizerKind::Thread); }
  44   bool needsMsanRt() const { return Sanitizers.has(SanitizerKind::Memory); }
  45   bool needsLsanRt() const {
  46     return Sanitizers.has(SanitizerKind::Leak) &&
  47            !Sanitizers.has(SanitizerKind::Address);
  48   }
  49   bool needsUbsanRt() const;
  50   bool needsDfsanRt() const { return Sanitizers.has(SanitizerKind::DataFlow); }
  51   bool needsSafeStackRt() const {
  52     return Sanitizers.has(SanitizerKind::SafeStack);
  53   }
  54
  55   bool requiresPIE() const;
  56   bool needsUnwindTables() const;
  57   bool linkCXXRuntimes() const { return LinkCXXRuntimes; }
  58   void addArgs(const ToolChain &TC, const llvm::opt::ArgList &Args,
  59                llvm::opt::ArgStringList &CmdArgs, types::ID InputType) const;
  60
  61  private:
  62   void clear();
  63 };
  64
  65 }  // namespace driver
  66 }  // namespace clang
  67
  68 #endif