1 //=== InnerPointerChecker.cpp -------------------------------------*- C++ -*--//
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
7 //===----------------------------------------------------------------------===//
9 // This file defines a check that marks a raw pointer to a C++ container's
10 // inner buffer released when the object is destroyed. This information can
11 // be used by MallocChecker to detect use-after-free problems.
13 //===----------------------------------------------------------------------===//
15 #include "AllocationState.h"
16 #include "clang/StaticAnalyzer/Checkers/BuiltinCheckerRegistration.h"
17 #include "InterCheckerAPI.h"
18 #include "clang/StaticAnalyzer/Core/BugReporter/BugType.h"
19 #include "clang/StaticAnalyzer/Core/BugReporter/CommonBugCategories.h"
20 #include "clang/StaticAnalyzer/Core/Checker.h"
21 #include "clang/StaticAnalyzer/Core/PathSensitive/CallEvent.h"
22 #include "clang/StaticAnalyzer/Core/PathSensitive/CheckerContext.h"
24 using namespace clang;
27 // Associate container objects with a set of raw pointer symbols.
28 REGISTER_SET_FACTORY_WITH_PROGRAMSTATE(PtrSet, SymbolRef)
29 REGISTER_MAP_WITH_PROGRAMSTATE(RawPtrMap, const MemRegion *, PtrSet)
34 class InnerPointerChecker
35 : public Checker<check::DeadSymbols, check::PostCall> {
37 CallDescription AppendFn, AssignFn, ClearFn, CStrFn, DataFn, EraseFn,
38 InsertFn, PopBackFn, PushBackFn, ReplaceFn, ReserveFn, ResizeFn,
39 ShrinkToFitFn, SwapFn;
42 class InnerPointerBRVisitor : public BugReporterVisitor {
46 InnerPointerBRVisitor(SymbolRef Sym) : PtrToBuf(Sym) {}
48 static void *getTag() {
53 void Profile(llvm::FoldingSetNodeID &ID) const override {
54 ID.AddPointer(getTag());
57 virtual std::shared_ptr<PathDiagnosticPiece> VisitNode(const ExplodedNode *N,
58 BugReporterContext &BRC,
59 BugReport &BR) override;
61 // FIXME: Scan the map once in the visitor's constructor and do a direct
63 bool isSymbolTracked(ProgramStateRef State, SymbolRef Sym) {
64 RawPtrMapTy Map = State->get<RawPtrMap>();
65 for (const auto Entry : Map) {
66 if (Entry.second.contains(Sym))
74 : AppendFn({"std", "basic_string", "append"}),
75 AssignFn({"std", "basic_string", "assign"}),
76 ClearFn({"std", "basic_string", "clear"}),
77 CStrFn({"std", "basic_string", "c_str"}),
78 DataFn({"std", "basic_string", "data"}),
79 EraseFn({"std", "basic_string", "erase"}),
80 InsertFn({"std", "basic_string", "insert"}),
81 PopBackFn({"std", "basic_string", "pop_back"}),
82 PushBackFn({"std", "basic_string", "push_back"}),
83 ReplaceFn({"std", "basic_string", "replace"}),
84 ReserveFn({"std", "basic_string", "reserve"}),
85 ResizeFn({"std", "basic_string", "resize"}),
86 ShrinkToFitFn({"std", "basic_string", "shrink_to_fit"}),
87 SwapFn({"std", "basic_string", "swap"}) {}
89 /// Check whether the called member function potentially invalidates
90 /// pointers referring to the container object's inner buffer.
91 bool isInvalidatingMemberFunction(const CallEvent &Call) const;
93 /// Mark pointer symbols associated with the given memory region released
94 /// in the program state.
95 void markPtrSymbolsReleased(const CallEvent &Call, ProgramStateRef State,
96 const MemRegion *ObjRegion,
97 CheckerContext &C) const;
99 /// Standard library functions that take a non-const `basic_string` argument by
100 /// reference may invalidate its inner pointers. Check for these cases and
101 /// mark the pointers released.
102 void checkFunctionArguments(const CallEvent &Call, ProgramStateRef State,
103 CheckerContext &C) const;
105 /// Record the connection between raw pointers referring to a container
106 /// object's inner buffer and the object's memory region in the program state.
107 /// Mark potentially invalidated pointers released.
108 void checkPostCall(const CallEvent &Call, CheckerContext &C) const;
110 /// Clean up the program state map.
111 void checkDeadSymbols(SymbolReaper &SymReaper, CheckerContext &C) const;
114 } // end anonymous namespace
116 bool InnerPointerChecker::isInvalidatingMemberFunction(
117 const CallEvent &Call) const {
118 if (const auto *MemOpCall = dyn_cast<CXXMemberOperatorCall>(&Call)) {
119 OverloadedOperatorKind Opc = MemOpCall->getOriginExpr()->getOperator();
120 if (Opc == OO_Equal || Opc == OO_PlusEqual)
124 return (isa<CXXDestructorCall>(Call) || Call.isCalled(AppendFn) ||
125 Call.isCalled(AssignFn) || Call.isCalled(ClearFn) ||
126 Call.isCalled(EraseFn) || Call.isCalled(InsertFn) ||
127 Call.isCalled(PopBackFn) || Call.isCalled(PushBackFn) ||
128 Call.isCalled(ReplaceFn) || Call.isCalled(ReserveFn) ||
129 Call.isCalled(ResizeFn) || Call.isCalled(ShrinkToFitFn) ||
130 Call.isCalled(SwapFn));
133 void InnerPointerChecker::markPtrSymbolsReleased(const CallEvent &Call,
134 ProgramStateRef State,
136 CheckerContext &C) const {
137 if (const PtrSet *PS = State->get<RawPtrMap>(MR)) {
138 const Expr *Origin = Call.getOriginExpr();
139 for (const auto Symbol : *PS) {
140 // NOTE: `Origin` may be null, and will be stored so in the symbol's
141 // `RefState` in MallocChecker's `RegionState` program state map.
142 State = allocation_state::markReleased(State, Symbol, Origin);
144 State = State->remove<RawPtrMap>(MR);
145 C.addTransition(State);
150 void InnerPointerChecker::checkFunctionArguments(const CallEvent &Call,
151 ProgramStateRef State,
152 CheckerContext &C) const {
153 if (const auto *FC = dyn_cast<AnyFunctionCall>(&Call)) {
154 const FunctionDecl *FD = FC->getDecl();
155 if (!FD || !FD->isInStdNamespace())
158 for (unsigned I = 0, E = FD->getNumParams(); I != E; ++I) {
159 QualType ParamTy = FD->getParamDecl(I)->getType();
160 if (!ParamTy->isReferenceType() ||
161 ParamTy->getPointeeType().isConstQualified())
164 // In case of member operator calls, `this` is counted as an
165 // argument but not as a parameter.
166 bool isaMemberOpCall = isa<CXXMemberOperatorCall>(FC);
167 unsigned ArgI = isaMemberOpCall ? I+1 : I;
169 SVal Arg = FC->getArgSVal(ArgI);
170 const auto *ArgRegion =
171 dyn_cast_or_null<TypedValueRegion>(Arg.getAsRegion());
175 markPtrSymbolsReleased(Call, State, ArgRegion, C);
182 // "References, pointers, and iterators referring to the elements of a
183 // basic_string sequence may be invalidated by the following uses of that
184 // basic_string object:
186 // -- As an argument to any standard library function taking a reference
187 // to non-const basic_string as an argument. For example, as an argument to
188 // non-member functions swap(), operator>>(), and getline(), or as an argument
189 // to basic_string::swap().
191 // -- Calling non-const member functions, except operator[], at, front, back,
192 // begin, rbegin, end, and rend."
194 void InnerPointerChecker::checkPostCall(const CallEvent &Call,
195 CheckerContext &C) const {
196 ProgramStateRef State = C.getState();
198 if (const auto *ICall = dyn_cast<CXXInstanceCall>(&Call)) {
199 // TODO: Do we need these to be typed?
200 const auto *ObjRegion = dyn_cast_or_null<TypedValueRegion>(
201 ICall->getCXXThisVal().getAsRegion());
205 if (Call.isCalled(CStrFn) || Call.isCalled(DataFn)) {
206 SVal RawPtr = Call.getReturnValue();
207 if (SymbolRef Sym = RawPtr.getAsSymbol(/*IncludeBaseRegions=*/true)) {
208 // Start tracking this raw pointer by adding it to the set of symbols
209 // associated with this container object in the program state map.
211 PtrSet::Factory &F = State->getStateManager().get_context<PtrSet>();
212 const PtrSet *SetPtr = State->get<RawPtrMap>(ObjRegion);
213 PtrSet Set = SetPtr ? *SetPtr : F.getEmptySet();
214 assert(C.wasInlined || !Set.contains(Sym));
215 Set = F.add(Set, Sym);
217 State = State->set<RawPtrMap>(ObjRegion, Set);
218 C.addTransition(State);
223 // Check [string.require] / second point.
224 if (isInvalidatingMemberFunction(Call)) {
225 markPtrSymbolsReleased(Call, State, ObjRegion, C);
230 // Check [string.require] / first point.
231 checkFunctionArguments(Call, State, C);
234 void InnerPointerChecker::checkDeadSymbols(SymbolReaper &SymReaper,
235 CheckerContext &C) const {
236 ProgramStateRef State = C.getState();
237 PtrSet::Factory &F = State->getStateManager().get_context<PtrSet>();
238 RawPtrMapTy RPM = State->get<RawPtrMap>();
239 for (const auto Entry : RPM) {
240 if (!SymReaper.isLiveRegion(Entry.first)) {
241 // Due to incomplete destructor support, some dead regions might
242 // remain in the program state map. Clean them up.
243 State = State->remove<RawPtrMap>(Entry.first);
245 if (const PtrSet *OldSet = State->get<RawPtrMap>(Entry.first)) {
246 PtrSet CleanedUpSet = *OldSet;
247 for (const auto Symbol : Entry.second) {
248 if (!SymReaper.isLive(Symbol))
249 CleanedUpSet = F.remove(CleanedUpSet, Symbol);
251 State = CleanedUpSet.isEmpty()
252 ? State->remove<RawPtrMap>(Entry.first)
253 : State->set<RawPtrMap>(Entry.first, CleanedUpSet);
256 C.addTransition(State);
261 namespace allocation_state {
263 std::unique_ptr<BugReporterVisitor> getInnerPointerBRVisitor(SymbolRef Sym) {
264 return llvm::make_unique<InnerPointerChecker::InnerPointerBRVisitor>(Sym);
267 const MemRegion *getContainerObjRegion(ProgramStateRef State, SymbolRef Sym) {
268 RawPtrMapTy Map = State->get<RawPtrMap>();
269 for (const auto Entry : Map) {
270 if (Entry.second.contains(Sym)) {
277 } // end namespace allocation_state
278 } // end namespace ento
279 } // end namespace clang
281 std::shared_ptr<PathDiagnosticPiece>
282 InnerPointerChecker::InnerPointerBRVisitor::VisitNode(const ExplodedNode *N,
283 BugReporterContext &BRC,
285 if (!isSymbolTracked(N->getState(), PtrToBuf) ||
286 isSymbolTracked(N->getFirstPred()->getState(), PtrToBuf))
289 const Stmt *S = PathDiagnosticLocation::getStmt(N);
293 const MemRegion *ObjRegion =
294 allocation_state::getContainerObjRegion(N->getState(), PtrToBuf);
295 const auto *TypedRegion = cast<TypedValueRegion>(ObjRegion);
296 QualType ObjTy = TypedRegion->getValueType();
298 SmallString<256> Buf;
299 llvm::raw_svector_ostream OS(Buf);
300 OS << "Pointer to inner buffer of '" << ObjTy.getAsString()
301 << "' obtained here";
302 PathDiagnosticLocation Pos(S, BRC.getSourceManager(),
303 N->getLocationContext());
304 return std::make_shared<PathDiagnosticEventPiece>(Pos, OS.str(), true,
308 void ento::registerInnerPointerChecker(CheckerManager &Mgr) {
309 registerInnerPointerCheckerAux(Mgr);
310 Mgr.registerChecker<InnerPointerChecker>();
313 bool ento::shouldRegisterInnerPointerChecker(const LangOptions &LO) {