ExprEngineCallAndReturn.cpp revision 0849ade4bb3e90c2fc0ce01ccd330f76f91da732
1//=-- ExprEngineCallAndReturn.cpp - Support for call/return -----*- C++ -*-===// 2// 3// The LLVM Compiler Infrastructure 4// 5// This file is distributed under the University of Illinois Open Source 6// License. See LICENSE.TXT for details. 7// 8//===----------------------------------------------------------------------===// 9// 10// This file defines ExprEngine's support for calls and returns. 11// 12//===----------------------------------------------------------------------===// 13 14#include "clang/StaticAnalyzer/Core/CheckerManager.h" 15#include "clang/StaticAnalyzer/Core/PathSensitive/ExprEngine.h" 16#include "clang/StaticAnalyzer/Core/PathSensitive/ObjCMessage.h" 17#include "clang/Analysis/Support/SaveAndRestore.h" 18#include "clang/AST/DeclCXX.h" 19 20using namespace clang; 21using namespace ento; 22 23void ExprEngine::processCallEnter(CallEnter CE, ExplodedNode *Pred) { 24 // Get the entry block in the CFG of the callee. 25 const StackFrameContext *calleeCtx = CE.getCalleeContext(); 26 const CFG *CalleeCFG = calleeCtx->getCFG(); 27 const CFGBlock *Entry = &(CalleeCFG->getEntry()); 28 29 // Validate the CFG. 30 assert(Entry->empty()); 31 assert(Entry->succ_size() == 1); 32 33 // Get the solitary sucessor. 34 const CFGBlock *Succ = *(Entry->succ_begin()); 35 36 // Construct an edge representing the starting location in the callee. 37 BlockEdge Loc(Entry, Succ, calleeCtx); 38 39 // Construct a new state which contains the mapping from actual to 40 // formal arguments. 41 const LocationContext *callerCtx = Pred->getLocationContext(); 42 const ProgramState *state = Pred->getState()->enterStackFrame(callerCtx, 43 calleeCtx); 44 45 // Construct a new node and add it to the worklist. 46 bool isNew; 47 ExplodedNode *Node = G.getNode(Loc, state, false, &isNew); 48 Node->addPredecessor(Pred, G); 49 if (isNew) 50 Engine.getWorkList()->enqueue(Node); 51} 52 53static const ReturnStmt *getReturnStmt(const ExplodedNode *Node) { 54 while (Node) { 55 const ProgramPoint &PP = Node->getLocation(); 56 // Skip any BlockEdges. 57 if (isa<BlockEdge>(PP) || isa<CallExit>(PP)) { 58 assert(Node->pred_size() == 1); 59 Node = *Node->pred_begin(); 60 continue; 61 } 62 if (const StmtPoint *SP = dyn_cast<StmtPoint>(&PP)) { 63 const Stmt *S = SP->getStmt(); 64 return dyn_cast<ReturnStmt>(S); 65 } 66 break; 67 } 68 return 0; 69} 70 71void ExprEngine::processCallExit(ExplodedNode *Pred) { 72 const ProgramState *state = Pred->getState(); 73 const StackFrameContext *calleeCtx = 74 Pred->getLocationContext()->getCurrentStackFrame(); 75 const LocationContext *callerCtx = calleeCtx->getParent(); 76 const Stmt *CE = calleeCtx->getCallSite(); 77 78 // If the callee returns an expression, bind its value to CallExpr. 79 if (const ReturnStmt *RS = getReturnStmt(Pred)) { 80 const LocationContext *LCtx = Pred->getLocationContext(); 81 SVal V = state->getSVal(RS, LCtx); 82 state = state->BindExpr(CE, callerCtx, V); 83 } 84 85 // Bind the constructed object value to CXXConstructExpr. 86 if (const CXXConstructExpr *CCE = dyn_cast<CXXConstructExpr>(CE)) { 87 const CXXThisRegion *ThisR = 88 getCXXThisRegion(CCE->getConstructor()->getParent(), calleeCtx); 89 90 SVal ThisV = state->getSVal(ThisR); 91 // Always bind the region to the CXXConstructExpr. 92 state = state->BindExpr(CCE, Pred->getLocationContext(), ThisV); 93 } 94 95 static SimpleProgramPointTag returnTag("ExprEngine : Call Return"); 96 PostStmt Loc(CE, callerCtx, &returnTag); 97 bool isNew; 98 ExplodedNode *N = G.getNode(Loc, state, false, &isNew); 99 N->addPredecessor(Pred, G); 100 if (!isNew) 101 return; 102 103 // Perform the post-condition check of the CallExpr. 104 ExplodedNodeSet Dst; 105 NodeBuilderContext Ctx(Engine, calleeCtx->getCallSiteBlock(), N); 106 SaveAndRestore<const NodeBuilderContext*> NBCSave(currentBuilderContext, 107 &Ctx); 108 SaveAndRestore<unsigned> CBISave(currentStmtIdx, calleeCtx->getIndex()); 109 110 getCheckerManager().runCheckersForPostStmt(Dst, N, CE, *this); 111 112 // Enqueue the next element in the block. 113 for (ExplodedNodeSet::iterator I = Dst.begin(), E = Dst.end(); I != E; ++I) { 114 Engine.getWorkList()->enqueue(*I, 115 calleeCtx->getCallSiteBlock(), 116 calleeCtx->getIndex()+1); 117 } 118} 119 120static unsigned getNumberStackFrames(const LocationContext *LCtx) { 121 unsigned count = 0; 122 while (LCtx) { 123 if (isa<StackFrameContext>(LCtx)) 124 ++count; 125 LCtx = LCtx->getParent(); 126 } 127 return count; 128} 129 130bool ExprEngine::InlineCall(ExplodedNodeSet &Dst, 131 const CallExpr *CE, 132 ExplodedNode *Pred) { 133 const ProgramState *state = Pred->getState(); 134 const Expr *Callee = CE->getCallee(); 135 const FunctionDecl *FD = 136 state->getSVal(Callee, Pred->getLocationContext()).getAsFunctionDecl(); 137 if (!FD || !FD->hasBody(FD)) 138 return false; 139 140 switch (CE->getStmtClass()) { 141 default: 142 // FIXME: Handle C++. 143 break; 144 case Stmt::CallExprClass: { 145 // Cap the stack depth at 4 calls (5 stack frames, base + 4 calls). 146 // These heuristics are a WIP. 147 if (getNumberStackFrames(Pred->getLocationContext()) == 5) 148 return false; 149 150 // Construct a new stack frame for the callee. 151 AnalysisDeclContext *CalleeADC = AMgr.getAnalysisDeclContext(FD); 152 const StackFrameContext *CallerSFC = 153 Pred->getLocationContext()->getCurrentStackFrame(); 154 const StackFrameContext *CalleeSFC = 155 CalleeADC->getStackFrame(CallerSFC, CE, 156 currentBuilderContext->getBlock(), 157 currentStmtIdx); 158 159 CallEnter Loc(CE, CalleeSFC, Pred->getLocationContext()); 160 bool isNew; 161 ExplodedNode *N = G.getNode(Loc, state, false, &isNew); 162 N->addPredecessor(Pred, G); 163 if (isNew) 164 Engine.getWorkList()->enqueue(N); 165 return true; 166 } 167 } 168 return false; 169} 170 171static bool isPointerToConst(const ParmVarDecl *ParamDecl) { 172 QualType PointeeTy = ParamDecl->getOriginalType()->getPointeeType(); 173 if (PointeeTy != QualType() && PointeeTy.isConstQualified() && 174 !PointeeTy->isAnyPointerType() && !PointeeTy->isReferenceType()) { 175 return true; 176 } 177 return false; 178} 179 180// Try to retrieve the function declaration and find the function parameter 181// types which are pointers/references to a non-pointer const. 182// We do not invalidate the corresponding argument regions. 183static void findPtrToConstParams(llvm::SmallSet<unsigned, 1> &PreserveArgs, 184 const CallOrObjCMessage &Call) { 185 const Decl *CallDecl = Call.getDecl(); 186 if (!CallDecl) 187 return; 188 189 if (const FunctionDecl *FDecl = dyn_cast<FunctionDecl>(CallDecl)) { 190 for (unsigned Idx = 0, E = Call.getNumArgs(); Idx != E; ++Idx) { 191 if (FDecl && Idx < FDecl->getNumParams()) { 192 if (isPointerToConst(FDecl->getParamDecl(Idx))) 193 PreserveArgs.insert(Idx); 194 } 195 } 196 return; 197 } 198 199 if (const ObjCMethodDecl *MDecl = dyn_cast<ObjCMethodDecl>(CallDecl)) { 200 assert(MDecl->param_size() <= Call.getNumArgs()); 201 unsigned Idx = 0; 202 for (clang::ObjCMethodDecl::param_const_iterator 203 I = MDecl->param_begin(), E = MDecl->param_end(); I != E; ++I, ++Idx) { 204 if (isPointerToConst(*I)) 205 PreserveArgs.insert(Idx); 206 } 207 return; 208 } 209} 210 211const ProgramState * 212ExprEngine::invalidateArguments(const ProgramState *State, 213 const CallOrObjCMessage &Call, 214 const LocationContext *LC) { 215 SmallVector<const MemRegion *, 8> RegionsToInvalidate; 216 217 if (Call.isObjCMessage()) { 218 // Invalidate all instance variables of the receiver of an ObjC message. 219 // FIXME: We should be able to do better with inter-procedural analysis. 220 if (const MemRegion *MR = Call.getInstanceMessageReceiver(LC).getAsRegion()) 221 RegionsToInvalidate.push_back(MR); 222 223 } else if (Call.isCXXCall()) { 224 // Invalidate all instance variables for the callee of a C++ method call. 225 // FIXME: We should be able to do better with inter-procedural analysis. 226 // FIXME: We can probably do better for const versus non-const methods. 227 if (const MemRegion *Callee = Call.getCXXCallee().getAsRegion()) 228 RegionsToInvalidate.push_back(Callee); 229 230 } else if (Call.isFunctionCall()) { 231 // Block calls invalidate all captured-by-reference values. 232 SVal CalleeVal = Call.getFunctionCallee(); 233 if (const MemRegion *Callee = CalleeVal.getAsRegion()) { 234 if (isa<BlockDataRegion>(Callee)) 235 RegionsToInvalidate.push_back(Callee); 236 } 237 } 238 239 // Indexes of arguments whose values will be preserved by the call. 240 llvm::SmallSet<unsigned, 1> PreserveArgs; 241 findPtrToConstParams(PreserveArgs, Call); 242 243 for (unsigned idx = 0, e = Call.getNumArgs(); idx != e; ++idx) { 244 if (PreserveArgs.count(idx)) 245 continue; 246 247 SVal V = Call.getArgSVal(idx); 248 249 // If we are passing a location wrapped as an integer, unwrap it and 250 // invalidate the values referred by the location. 251 if (nonloc::LocAsInteger *Wrapped = dyn_cast<nonloc::LocAsInteger>(&V)) 252 V = Wrapped->getLoc(); 253 else if (!isa<Loc>(V)) 254 continue; 255 256 if (const MemRegion *R = V.getAsRegion()) { 257 // Invalidate the value of the variable passed by reference. 258 259 // Are we dealing with an ElementRegion? If the element type is 260 // a basic integer type (e.g., char, int) and the underlying region 261 // is a variable region then strip off the ElementRegion. 262 // FIXME: We really need to think about this for the general case 263 // as sometimes we are reasoning about arrays and other times 264 // about (char*), etc., is just a form of passing raw bytes. 265 // e.g., void *p = alloca(); foo((char*)p); 266 if (const ElementRegion *ER = dyn_cast<ElementRegion>(R)) { 267 // Checking for 'integral type' is probably too promiscuous, but 268 // we'll leave it in for now until we have a systematic way of 269 // handling all of these cases. Eventually we need to come up 270 // with an interface to StoreManager so that this logic can be 271 // appropriately delegated to the respective StoreManagers while 272 // still allowing us to do checker-specific logic (e.g., 273 // invalidating reference counts), probably via callbacks. 274 if (ER->getElementType()->isIntegralOrEnumerationType()) { 275 const MemRegion *superReg = ER->getSuperRegion(); 276 if (isa<VarRegion>(superReg) || isa<FieldRegion>(superReg) || 277 isa<ObjCIvarRegion>(superReg)) 278 R = cast<TypedRegion>(superReg); 279 } 280 // FIXME: What about layers of ElementRegions? 281 } 282 283 // Mark this region for invalidation. We batch invalidate regions 284 // below for efficiency. 285 RegionsToInvalidate.push_back(R); 286 } else { 287 // Nuke all other arguments passed by reference. 288 // FIXME: is this necessary or correct? This handles the non-Region 289 // cases. Is it ever valid to store to these? 290 State = State->unbindLoc(cast<Loc>(V)); 291 } 292 } 293 294 // Invalidate designated regions using the batch invalidation API. 295 296 // FIXME: We can have collisions on the conjured symbol if the 297 // expression *I also creates conjured symbols. We probably want 298 // to identify conjured symbols by an expression pair: the enclosing 299 // expression (the context) and the expression itself. This should 300 // disambiguate conjured symbols. 301 unsigned Count = currentBuilderContext->getCurrentBlockCount(); 302 StoreManager::InvalidatedSymbols IS; 303 304 // NOTE: Even if RegionsToInvalidate is empty, we may still invalidate 305 // global variables. 306 return State->invalidateRegions(RegionsToInvalidate, 307 Call.getOriginExpr(), Count, 308 &IS, &Call); 309 310} 311 312void ExprEngine::VisitCallExpr(const CallExpr *CE, ExplodedNode *Pred, 313 ExplodedNodeSet &dst) { 314 // Perform the previsit of the CallExpr. 315 ExplodedNodeSet dstPreVisit; 316 getCheckerManager().runCheckersForPreStmt(dstPreVisit, Pred, CE, *this); 317 318 // Now evaluate the call itself. 319 class DefaultEval : public GraphExpander { 320 ExprEngine &Eng; 321 const CallExpr *CE; 322 public: 323 324 DefaultEval(ExprEngine &eng, const CallExpr *ce) 325 : Eng(eng), CE(ce) {} 326 virtual void expandGraph(ExplodedNodeSet &Dst, ExplodedNode *Pred) { 327 // Should we inline the call? 328 if (Eng.getAnalysisManager().shouldInlineCall() && 329 Eng.InlineCall(Dst, CE, Pred)) { 330 return; 331 } 332 333 // First handle the return value. 334 StmtNodeBuilder Bldr(Pred, Dst, *Eng.currentBuilderContext); 335 336 // Get the callee. 337 const Expr *Callee = CE->getCallee()->IgnoreParens(); 338 const ProgramState *state = Pred->getState(); 339 SVal L = state->getSVal(Callee, Pred->getLocationContext()); 340 341 // Figure out the result type. We do this dance to handle references. 342 QualType ResultTy; 343 if (const FunctionDecl *FD = L.getAsFunctionDecl()) 344 ResultTy = FD->getResultType(); 345 else 346 ResultTy = CE->getType(); 347 348 if (CE->isLValue()) 349 ResultTy = Eng.getContext().getPointerType(ResultTy); 350 351 // Conjure a symbol value to use as the result. 352 SValBuilder &SVB = Eng.getSValBuilder(); 353 unsigned Count = Eng.currentBuilderContext->getCurrentBlockCount(); 354 SVal RetVal = SVB.getConjuredSymbolVal(0, CE, ResultTy, Count); 355 356 // Generate a new state with the return value set. 357 const LocationContext *LCtx = Pred->getLocationContext(); 358 state = state->BindExpr(CE, LCtx, RetVal); 359 360 // Invalidate the arguments. 361 state = Eng.invalidateArguments(state, CallOrObjCMessage(CE, state, LCtx), 362 LCtx); 363 364 // And make the result node. 365 Bldr.generateNode(CE, Pred, state); 366 } 367 }; 368 369 // Finally, evaluate the function call. We try each of the checkers 370 // to see if the can evaluate the function call. 371 ExplodedNodeSet dstCallEvaluated; 372 DefaultEval defEval(*this, CE); 373 getCheckerManager().runCheckersForEvalCall(dstCallEvaluated, 374 dstPreVisit, 375 CE, *this, &defEval); 376 377 // Finally, perform the post-condition check of the CallExpr and store 378 // the created nodes in 'Dst'. 379 getCheckerManager().runCheckersForPostStmt(dst, dstCallEvaluated, CE, 380 *this); 381} 382 383void ExprEngine::VisitReturnStmt(const ReturnStmt *RS, ExplodedNode *Pred, 384 ExplodedNodeSet &Dst) { 385 386 ExplodedNodeSet dstPreVisit; 387 getCheckerManager().runCheckersForPreStmt(dstPreVisit, Pred, RS, *this); 388 389 StmtNodeBuilder B(dstPreVisit, Dst, *currentBuilderContext); 390 391 if (RS->getRetValue()) { 392 for (ExplodedNodeSet::iterator it = dstPreVisit.begin(), 393 ei = dstPreVisit.end(); it != ei; ++it) { 394 B.generateNode(RS, *it, (*it)->getState()); 395 } 396 } 397 else { 398 B.takeNodes(dstPreVisit); 399 } 400} 401