contrib/llvm-project/clang/lib/StaticAnalyzer/Checkers/TaintTesterChecker.cpp

   1 //== TaintTesterChecker.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 checker can be used for testing how taint data is propagated.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #include "Taint.h"
  14 #include "clang/StaticAnalyzer/Checkers/BuiltinCheckerRegistration.h"
  15 #include "clang/StaticAnalyzer/Core/BugReporter/BugType.h"
  16 #include "clang/StaticAnalyzer/Core/Checker.h"
  17 #include "clang/StaticAnalyzer/Core/CheckerManager.h"
  18 #include "clang/StaticAnalyzer/Core/PathSensitive/CheckerContext.h"
  19
  20 using namespace clang;
  21 using namespace ento;
  22 using namespace taint;
  23
  24 namespace {
  25 class TaintTesterChecker : public Checker< check::PostStmt<Expr> > {
  26
  27   mutable std::unique_ptr<BugType> BT;
  28   void initBugType() const;
  29
  30   /// Given a pointer argument, get the symbol of the value it contains
  31   /// (points to).
  32   SymbolRef getPointedToSymbol(CheckerContext &C,
  33                                const Expr* Arg,
  34                                bool IssueWarning = true) const;
  35
  36 public:
  37   void checkPostStmt(const Expr *E, CheckerContext &C) const;
  38 };
  39 }
  40
  41 inline void TaintTesterChecker::initBugType() const {
  42   if (!BT)
  43     BT.reset(new BugType(this, "Tainted data", "General"));
  44 }
  45
  46 void TaintTesterChecker::checkPostStmt(const Expr *E,
  47                                        CheckerContext &C) const {
  48   ProgramStateRef State = C.getState();
  49   if (!State)
  50     return;
  51
  52   if (isTainted(State, E, C.getLocationContext())) {
  53     if (ExplodedNode *N = C.generateNonFatalErrorNode()) {
  54       initBugType();
  55       auto report = llvm::make_unique<BugReport>(*BT, "tainted",N);
  56       report->addRange(E->getSourceRange());
  57       C.emitReport(std::move(report));
  58     }
  59   }
  60 }
  61
  62 void ento::registerTaintTesterChecker(CheckerManager &mgr) {
  63   mgr.registerChecker<TaintTesterChecker>();
  64 }
  65
  66 bool ento::shouldRegisterTaintTesterChecker(const LangOptions &LO) {
  67   return true;
  68 }