63 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			63 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| //== TaintTesterChecker.cpp ----------------------------------- -*- C++ -*--=//
 | |
| //
 | |
| //                     The LLVM Compiler Infrastructure
 | |
| //
 | |
| // This file is distributed under the University of Illinois Open Source
 | |
| // License. See LICENSE.TXT for details.
 | |
| //
 | |
| //===----------------------------------------------------------------------===//
 | |
| //
 | |
| // This checker can be used for testing how taint data is propagated.
 | |
| //
 | |
| //===----------------------------------------------------------------------===//
 | |
| #include "ClangSACheckers.h"
 | |
| #include "clang/StaticAnalyzer/Core/BugReporter/BugType.h"
 | |
| #include "clang/StaticAnalyzer/Core/Checker.h"
 | |
| #include "clang/StaticAnalyzer/Core/CheckerManager.h"
 | |
| #include "clang/StaticAnalyzer/Core/PathSensitive/CheckerContext.h"
 | |
| 
 | |
| using namespace clang;
 | |
| using namespace ento;
 | |
| 
 | |
| namespace {
 | |
| class TaintTesterChecker : public Checker< check::PostStmt<Expr> > {
 | |
| 
 | |
|   mutable OwningPtr<BugType> BT;
 | |
|   void initBugType() const;
 | |
| 
 | |
|   /// Given a pointer argument, get the symbol of the value it contains
 | |
|   /// (points to).
 | |
|   SymbolRef getPointedToSymbol(CheckerContext &C,
 | |
|                                const Expr* Arg,
 | |
|                                bool IssueWarning = true) const;
 | |
| 
 | |
| public:
 | |
|   void checkPostStmt(const Expr *E, CheckerContext &C) const;
 | |
| };
 | |
| }
 | |
| 
 | |
| inline void TaintTesterChecker::initBugType() const {
 | |
|   if (!BT)
 | |
|     BT.reset(new BugType("Tainted data", "General"));
 | |
| }
 | |
| 
 | |
| void TaintTesterChecker::checkPostStmt(const Expr *E,
 | |
|                                        CheckerContext &C) const {
 | |
|   ProgramStateRef State = C.getState();
 | |
|   if (!State)
 | |
|     return;
 | |
| 
 | |
|   if (State->isTainted(E, C.getLocationContext())) {
 | |
|     if (ExplodedNode *N = C.addTransition()) {
 | |
|       initBugType();
 | |
|       BugReport *report = new BugReport(*BT, "tainted",N);
 | |
|       report->addRange(E->getSourceRange());
 | |
|       C.emitReport(report);
 | |
|     }
 | |
|   }
 | |
| }
 | |
| 
 | |
| void ento::registerTaintTesterChecker(CheckerManager &mgr) {
 | |
|   mgr.registerChecker<TaintTesterChecker>();
 | |
| }
 | 
