diff options
Diffstat (limited to 'lib/StaticAnalyzer/Core/ProgramState.cpp')
-rw-r--r-- | lib/StaticAnalyzer/Core/ProgramState.cpp | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/lib/StaticAnalyzer/Core/ProgramState.cpp b/lib/StaticAnalyzer/Core/ProgramState.cpp index af038c6f0f..9ea8abd952 100644 --- a/lib/StaticAnalyzer/Core/ProgramState.cpp +++ b/lib/StaticAnalyzer/Core/ProgramState.cpp @@ -658,13 +658,19 @@ const ProgramState* ProgramState::addTaint(const Stmt *S, return addTaint(Sym, Kind); const MemRegion *R = getSVal(S).getAsRegion(); - if (const SymbolicRegion *SR = dyn_cast_or_null<SymbolicRegion>(R)) - return addTaint(SR->getSymbol(), Kind); + addTaint(R, Kind); // Cannot add taint, so just return the state. return this; } +const ProgramState* ProgramState::addTaint(const MemRegion *R, + TaintTagType Kind) const { + if (const SymbolicRegion *SR = dyn_cast_or_null<SymbolicRegion>(R)) + return addTaint(SR->getSymbol(), Kind); + return this; +} + const ProgramState* ProgramState::addTaint(SymbolRef Sym, TaintTagType Kind) const { const ProgramState *NewState = set<TaintMap>(Sym, Kind); |