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
initBugType() const41 inline void TaintTesterChecker::initBugType() const {
42 if (!BT)
43 BT.reset(new BugType(this, "Tainted data", "General"));
44 }
45
checkPostStmt(const Expr * E,CheckerContext & C) const46 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 = std::make_unique<PathSensitiveBugReport>(*BT, "tainted", N);
56 report->addRange(E->getSourceRange());
57 C.emitReport(std::move(report));
58 }
59 }
60 }
61
registerTaintTesterChecker(CheckerManager & mgr)62 void ento::registerTaintTesterChecker(CheckerManager &mgr) {
63 mgr.registerChecker<TaintTesterChecker>();
64 }
65
shouldRegisterTaintTesterChecker(const CheckerManager & mgr)66 bool ento::shouldRegisterTaintTesterChecker(const CheckerManager &mgr) {
67 return true;
68 }
69