1fa0734ecSArgyrios Kyrtzidis //=-- ExplodedGraph.cpp - Local, Path-Sens. "Exploded Graph" -*- C++ -*------=// 2fa0734ecSArgyrios Kyrtzidis // 3fa0734ecSArgyrios Kyrtzidis // The LLVM Compiler Infrastructure 4fa0734ecSArgyrios Kyrtzidis // 5fa0734ecSArgyrios Kyrtzidis // This file is distributed under the University of Illinois Open Source 6fa0734ecSArgyrios Kyrtzidis // License. See LICENSE.TXT for details. 7fa0734ecSArgyrios Kyrtzidis // 8fa0734ecSArgyrios Kyrtzidis //===----------------------------------------------------------------------===// 9fa0734ecSArgyrios Kyrtzidis // 10fa0734ecSArgyrios Kyrtzidis // This file defines the template classes ExplodedNode and ExplodedGraph, 11fa0734ecSArgyrios Kyrtzidis // which represent a path-sensitive, intra-procedural "exploded graph." 12fa0734ecSArgyrios Kyrtzidis // 13fa0734ecSArgyrios Kyrtzidis //===----------------------------------------------------------------------===// 14fa0734ecSArgyrios Kyrtzidis 15f8cbac4bSTed Kremenek #include "clang/StaticAnalyzer/Core/PathSensitive/ExplodedGraph.h" 163a02247dSChandler Carruth #include "clang/AST/ParentMap.h" 173a02247dSChandler Carruth #include "clang/AST/Stmt.h" 184f7df9beSJordan Rose #include "clang/StaticAnalyzer/Core/PathSensitive/CallEvent.h" 19001fd5b4STed Kremenek #include "clang/StaticAnalyzer/Core/PathSensitive/ProgramState.h" 20fa0734ecSArgyrios Kyrtzidis #include "llvm/ADT/DenseMap.h" 213a02247dSChandler Carruth #include "llvm/ADT/DenseSet.h" 22fa0734ecSArgyrios Kyrtzidis #include "llvm/ADT/SmallVector.h" 237aa3687bSAnna Zaks #include "llvm/ADT/Statistic.h" 24fa0734ecSArgyrios Kyrtzidis #include <vector> 25fa0734ecSArgyrios Kyrtzidis 26fa0734ecSArgyrios Kyrtzidis using namespace clang; 27fa0734ecSArgyrios Kyrtzidis using namespace ento; 28fa0734ecSArgyrios Kyrtzidis 29fa0734ecSArgyrios Kyrtzidis //===----------------------------------------------------------------------===// 30fa0734ecSArgyrios Kyrtzidis // Node auditing. 31fa0734ecSArgyrios Kyrtzidis //===----------------------------------------------------------------------===// 32fa0734ecSArgyrios Kyrtzidis 33fa0734ecSArgyrios Kyrtzidis // An out of line virtual method to provide a home for the class vtable. 34fa0734ecSArgyrios Kyrtzidis ExplodedNode::Auditor::~Auditor() {} 35fa0734ecSArgyrios Kyrtzidis 36fa0734ecSArgyrios Kyrtzidis #ifndef NDEBUG 37fa0734ecSArgyrios Kyrtzidis static ExplodedNode::Auditor* NodeAuditor = 0; 38fa0734ecSArgyrios Kyrtzidis #endif 39fa0734ecSArgyrios Kyrtzidis 40fa0734ecSArgyrios Kyrtzidis void ExplodedNode::SetAuditor(ExplodedNode::Auditor* A) { 41fa0734ecSArgyrios Kyrtzidis #ifndef NDEBUG 42fa0734ecSArgyrios Kyrtzidis NodeAuditor = A; 43fa0734ecSArgyrios Kyrtzidis #endif 44fa0734ecSArgyrios Kyrtzidis } 45fa0734ecSArgyrios Kyrtzidis 46fa0734ecSArgyrios Kyrtzidis //===----------------------------------------------------------------------===// 47a40f8ebcSTed Kremenek // Cleanup. 48a40f8ebcSTed Kremenek //===----------------------------------------------------------------------===// 49a40f8ebcSTed Kremenek 5044d2973bSTed Kremenek ExplodedGraph::ExplodedGraph() 51746c06d0SJordan Rose : NumNodes(0), ReclaimNodeInterval(0) {} 5244d2973bSTed Kremenek 53a2aa929eSTed Kremenek ExplodedGraph::~ExplodedGraph() {} 54a40f8ebcSTed Kremenek 55a40f8ebcSTed Kremenek //===----------------------------------------------------------------------===// 56a40f8ebcSTed Kremenek // Node reclamation. 57a40f8ebcSTed Kremenek //===----------------------------------------------------------------------===// 58a40f8ebcSTed Kremenek 5904fa9e3dSTed Kremenek bool ExplodedGraph::isInterestingLValueExpr(const Expr *Ex) { 6004fa9e3dSTed Kremenek if (!Ex->isLValue()) 6104fa9e3dSTed Kremenek return false; 6204fa9e3dSTed Kremenek return isa<DeclRefExpr>(Ex) || 6304fa9e3dSTed Kremenek isa<MemberExpr>(Ex) || 6404fa9e3dSTed Kremenek isa<ObjCIvarRefExpr>(Ex); 6504fa9e3dSTed Kremenek } 6604fa9e3dSTed Kremenek 671dd7fd71STed Kremenek bool ExplodedGraph::shouldCollect(const ExplodedNode *node) { 688f564058STed Kremenek // First, we only consider nodes for reclamation of the following 698f564058STed Kremenek // conditions apply: 701dd7fd71STed Kremenek // 711dd7fd71STed Kremenek // (1) 1 predecessor (that has one successor) 721dd7fd71STed Kremenek // (2) 1 successor (that has one predecessor) 738f564058STed Kremenek // 748f564058STed Kremenek // If a node has no successor it is on the "frontier", while a node 758f564058STed Kremenek // with no predecessor is a root. 768f564058STed Kremenek // 778f564058STed Kremenek // After these prerequisites, we discard all "filler" nodes that 788f564058STed Kremenek // are used only for intermediate processing, and are not essential 798f564058STed Kremenek // for analyzer history: 808f564058STed Kremenek // 818f564058STed Kremenek // (a) PreStmtPurgeDeadSymbols 828f564058STed Kremenek // 838f564058STed Kremenek // We then discard all other nodes where *all* of the following conditions 848f564058STed Kremenek // apply: 858f564058STed Kremenek // 86199fdd82SJordan Rose // (3) The ProgramPoint is for a PostStmt, but not a PostStore. 871dd7fd71STed Kremenek // (4) There is no 'tag' for the ProgramPoint. 881dd7fd71STed Kremenek // (5) The 'store' is the same as the predecessor. 891dd7fd71STed Kremenek // (6) The 'GDM' is the same as the predecessor. 901dd7fd71STed Kremenek // (7) The LocationContext is the same as the predecessor. 9196250482STed Kremenek // (8) Expressions that are *not* lvalue expressions. 9296250482STed Kremenek // (9) The PostStmt isn't for a non-consumed Stmt or Expr. 9396250482STed Kremenek // (10) The successor is not a CallExpr StmtPoint (so that we would 9496250482STed Kremenek // be able to find it when retrying a call with no inlining). 95681cce99SJordan Rose // FIXME: It may be safe to reclaim PreCall and PostCall nodes as well. 961dd7fd71STed Kremenek 971dd7fd71STed Kremenek // Conditions 1 and 2. 981dd7fd71STed Kremenek if (node->pred_size() != 1 || node->succ_size() != 1) 991dd7fd71STed Kremenek return false; 1001dd7fd71STed Kremenek 1011dd7fd71STed Kremenek const ExplodedNode *pred = *(node->pred_begin()); 1021dd7fd71STed Kremenek if (pred->succ_size() != 1) 1031dd7fd71STed Kremenek return false; 1041dd7fd71STed Kremenek 1051dd7fd71STed Kremenek const ExplodedNode *succ = *(node->succ_begin()); 1061dd7fd71STed Kremenek if (succ->pred_size() != 1) 1071dd7fd71STed Kremenek return false; 1081dd7fd71STed Kremenek 1098f564058STed Kremenek // Now reclaim any nodes that are (by definition) not essential to 1108f564058STed Kremenek // analysis history and are not consulted by any client code. 1111dd7fd71STed Kremenek ProgramPoint progPoint = node->getLocation(); 1128f564058STed Kremenek if (progPoint.getAs<PreStmtPurgeDeadSymbols>()) 113*f352d8c7STed Kremenek return !progPoint.getTag(); 1148f564058STed Kremenek 1158f564058STed Kremenek // Condition 3. 11687396b9bSDavid Blaikie if (!progPoint.getAs<PostStmt>() || progPoint.getAs<PostStore>()) 1171dd7fd71STed Kremenek return false; 1181dd7fd71STed Kremenek 1191dd7fd71STed Kremenek // Condition 4. 12087396b9bSDavid Blaikie PostStmt ps = progPoint.castAs<PostStmt>(); 1211dd7fd71STed Kremenek if (ps.getTag()) 1221dd7fd71STed Kremenek return false; 1231dd7fd71STed Kremenek 1241dd7fd71STed Kremenek // Conditions 5, 6, and 7. 1251dd7fd71STed Kremenek ProgramStateRef state = node->getState(); 1261dd7fd71STed Kremenek ProgramStateRef pred_state = pred->getState(); 1271dd7fd71STed Kremenek if (state->store != pred_state->store || state->GDM != pred_state->GDM || 1281dd7fd71STed Kremenek progPoint.getLocationContext() != pred->getLocationContext()) 1291dd7fd71STed Kremenek return false; 1301dd7fd71STed Kremenek 13104fa9e3dSTed Kremenek // All further checks require expressions. 13296250482STed Kremenek const Expr *Ex = dyn_cast<Expr>(ps.getStmt()); 13304fa9e3dSTed Kremenek if (!Ex) 13404fa9e3dSTed Kremenek return false; 13504fa9e3dSTed Kremenek 13604fa9e3dSTed Kremenek // Condition 8. 13704fa9e3dSTed Kremenek // Do not collect nodes for "interesting" lvalue expressions since they are 13804fa9e3dSTed Kremenek // used extensively for generating path diagnostics. 13904fa9e3dSTed Kremenek if (isInterestingLValueExpr(Ex)) 14096250482STed Kremenek return false; 14196250482STed Kremenek 14296250482STed Kremenek // Condition 9. 14367e0062bSAnna Zaks // Do not collect nodes for non-consumed Stmt or Expr to ensure precise 14467e0062bSAnna Zaks // diagnostic generation; specifically, so that we could anchor arrows 14567e0062bSAnna Zaks // pointing to the beginning of statements (as written in code). 1461dd7fd71STed Kremenek ParentMap &PM = progPoint.getLocationContext()->getParentMap(); 1471dd7fd71STed Kremenek if (!PM.isConsumedExpr(Ex)) 1481dd7fd71STed Kremenek return false; 1491dd7fd71STed Kremenek 15096250482STed Kremenek // Condition 10. 151bec49efdSAnna Zaks const ProgramPoint SuccLoc = succ->getLocation(); 15287396b9bSDavid Blaikie if (Optional<StmtPoint> SP = SuccLoc.getAs<StmtPoint>()) 153e537cc05SJordan Rose if (CallEvent::isCallStmt(SP->getStmt())) 154bec49efdSAnna Zaks return false; 155bec49efdSAnna Zaks 1561dd7fd71STed Kremenek return true; 1571dd7fd71STed Kremenek } 1581dd7fd71STed Kremenek 1591dd7fd71STed Kremenek void ExplodedGraph::collectNode(ExplodedNode *node) { 1601dd7fd71STed Kremenek // Removing a node means: 1611dd7fd71STed Kremenek // (a) changing the predecessors successor to the successor of this node 1621dd7fd71STed Kremenek // (b) changing the successors predecessor to the predecessor of this node 1631dd7fd71STed Kremenek // (c) Putting 'node' onto freeNodes. 1641dd7fd71STed Kremenek assert(node->pred_size() == 1 || node->succ_size() == 1); 1651dd7fd71STed Kremenek ExplodedNode *pred = *(node->pred_begin()); 1661dd7fd71STed Kremenek ExplodedNode *succ = *(node->succ_begin()); 1671dd7fd71STed Kremenek pred->replaceSuccessor(succ); 1681dd7fd71STed Kremenek succ->replacePredecessor(pred); 169a2aa929eSTed Kremenek FreeNodes.push_back(node); 1701dd7fd71STed Kremenek Nodes.RemoveNode(node); 1711dd7fd71STed Kremenek --NumNodes; 1721dd7fd71STed Kremenek node->~ExplodedNode(); 1731dd7fd71STed Kremenek } 1741dd7fd71STed Kremenek 17535e55fe4STed Kremenek void ExplodedGraph::reclaimRecentlyAllocatedNodes() { 176a2aa929eSTed Kremenek if (ChangedNodes.empty()) 177a40f8ebcSTed Kremenek return; 17844d2973bSTed Kremenek 179746c06d0SJordan Rose // Only periodically reclaim nodes so that we can build up a set of 18035e55fe4STed Kremenek // nodes that meet the reclamation criteria. Freshly created nodes 18135e55fe4STed Kremenek // by definition have no successor, and thus cannot be reclaimed (see below). 182746c06d0SJordan Rose assert(ReclaimCounter > 0); 183746c06d0SJordan Rose if (--ReclaimCounter != 0) 18435e55fe4STed Kremenek return; 185746c06d0SJordan Rose ReclaimCounter = ReclaimNodeInterval; 18635e55fe4STed Kremenek 18735e55fe4STed Kremenek for (NodeVector::iterator it = ChangedNodes.begin(), et = ChangedNodes.end(); 188a2aa929eSTed Kremenek it != et; ++it) { 189a2aa929eSTed Kremenek ExplodedNode *node = *it; 1901dd7fd71STed Kremenek if (shouldCollect(node)) 1911dd7fd71STed Kremenek collectNode(node); 192a40f8ebcSTed Kremenek } 193a2aa929eSTed Kremenek ChangedNodes.clear(); 194a40f8ebcSTed Kremenek } 195a40f8ebcSTed Kremenek 196a40f8ebcSTed Kremenek //===----------------------------------------------------------------------===// 197fa0734ecSArgyrios Kyrtzidis // ExplodedNode. 198fa0734ecSArgyrios Kyrtzidis //===----------------------------------------------------------------------===// 199fa0734ecSArgyrios Kyrtzidis 2002b10f3f8SJordan Rose // An NodeGroup's storage type is actually very much like a TinyPtrVector: 2012b10f3f8SJordan Rose // it can be either a pointer to a single ExplodedNode, or a pointer to a 2022b10f3f8SJordan Rose // BumpVector allocated with the ExplodedGraph's allocator. This allows the 2032b10f3f8SJordan Rose // common case of single-node NodeGroups to be implemented with no extra memory. 2042b10f3f8SJordan Rose // 2052b10f3f8SJordan Rose // Consequently, each of the NodeGroup methods have up to four cases to handle: 2062b10f3f8SJordan Rose // 1. The flag is set and this group does not actually contain any nodes. 2072b10f3f8SJordan Rose // 2. The group is empty, in which case the storage value is null. 2082b10f3f8SJordan Rose // 3. The group contains a single node. 2092b10f3f8SJordan Rose // 4. The group contains more than one node. 21080547386SJordan Rose typedef BumpVector<ExplodedNode *> ExplodedNodeVector; 21180547386SJordan Rose typedef llvm::PointerUnion<ExplodedNode *, ExplodedNodeVector *> GroupStorage; 212fa0734ecSArgyrios Kyrtzidis 213fa0734ecSArgyrios Kyrtzidis void ExplodedNode::addPredecessor(ExplodedNode *V, ExplodedGraph &G) { 214fa0734ecSArgyrios Kyrtzidis assert (!V->isSink()); 215fa0734ecSArgyrios Kyrtzidis Preds.addNode(V, G); 216fa0734ecSArgyrios Kyrtzidis V->Succs.addNode(this, G); 217fa0734ecSArgyrios Kyrtzidis #ifndef NDEBUG 218fa0734ecSArgyrios Kyrtzidis if (NodeAuditor) NodeAuditor->AddEdge(V, this); 219fa0734ecSArgyrios Kyrtzidis #endif 220fa0734ecSArgyrios Kyrtzidis } 221fa0734ecSArgyrios Kyrtzidis 222a40f8ebcSTed Kremenek void ExplodedNode::NodeGroup::replaceNode(ExplodedNode *node) { 2232b10f3f8SJordan Rose assert(!getFlag()); 2242b10f3f8SJordan Rose 22580547386SJordan Rose GroupStorage &Storage = reinterpret_cast<GroupStorage&>(P); 22680547386SJordan Rose assert(Storage.is<ExplodedNode *>()); 22780547386SJordan Rose Storage = node; 22880547386SJordan Rose assert(Storage.is<ExplodedNode *>()); 229a40f8ebcSTed Kremenek } 230a40f8ebcSTed Kremenek 231fa0734ecSArgyrios Kyrtzidis void ExplodedNode::NodeGroup::addNode(ExplodedNode *N, ExplodedGraph &G) { 232fa0734ecSArgyrios Kyrtzidis assert(!getFlag()); 233fa0734ecSArgyrios Kyrtzidis 23480547386SJordan Rose GroupStorage &Storage = reinterpret_cast<GroupStorage&>(P); 23580547386SJordan Rose if (Storage.isNull()) { 23680547386SJordan Rose Storage = N; 23780547386SJordan Rose assert(Storage.is<ExplodedNode *>()); 23880547386SJordan Rose return; 23980547386SJordan Rose } 240fa0734ecSArgyrios Kyrtzidis 24180547386SJordan Rose ExplodedNodeVector *V = Storage.dyn_cast<ExplodedNodeVector *>(); 24280547386SJordan Rose 24380547386SJordan Rose if (!V) { 24480547386SJordan Rose // Switch from single-node to multi-node representation. 24580547386SJordan Rose ExplodedNode *Old = Storage.get<ExplodedNode *>(); 24680547386SJordan Rose 24780547386SJordan Rose BumpVectorContext &Ctx = G.getNodeAllocator(); 24880547386SJordan Rose V = G.getAllocator().Allocate<ExplodedNodeVector>(); 24980547386SJordan Rose new (V) ExplodedNodeVector(Ctx, 4); 25080547386SJordan Rose V->push_back(Old, Ctx); 25180547386SJordan Rose 25280547386SJordan Rose Storage = V; 25380547386SJordan Rose assert(!getFlag()); 25480547386SJordan Rose assert(Storage.is<ExplodedNodeVector *>()); 255fa0734ecSArgyrios Kyrtzidis } 25680547386SJordan Rose 25780547386SJordan Rose V->push_back(N, G.getNodeAllocator()); 258fa0734ecSArgyrios Kyrtzidis } 259fa0734ecSArgyrios Kyrtzidis 260fa0734ecSArgyrios Kyrtzidis unsigned ExplodedNode::NodeGroup::size() const { 261fa0734ecSArgyrios Kyrtzidis if (getFlag()) 262fa0734ecSArgyrios Kyrtzidis return 0; 263fa0734ecSArgyrios Kyrtzidis 26480547386SJordan Rose const GroupStorage &Storage = reinterpret_cast<const GroupStorage &>(P); 26580547386SJordan Rose if (Storage.isNull()) 26680547386SJordan Rose return 0; 26780547386SJordan Rose if (ExplodedNodeVector *V = Storage.dyn_cast<ExplodedNodeVector *>()) 26880547386SJordan Rose return V->size(); 26980547386SJordan Rose return 1; 270fa0734ecSArgyrios Kyrtzidis } 271fa0734ecSArgyrios Kyrtzidis 27280547386SJordan Rose ExplodedNode * const *ExplodedNode::NodeGroup::begin() const { 273fa0734ecSArgyrios Kyrtzidis if (getFlag()) 27480547386SJordan Rose return 0; 275fa0734ecSArgyrios Kyrtzidis 27680547386SJordan Rose const GroupStorage &Storage = reinterpret_cast<const GroupStorage &>(P); 27780547386SJordan Rose if (Storage.isNull()) 27880547386SJordan Rose return 0; 27980547386SJordan Rose if (ExplodedNodeVector *V = Storage.dyn_cast<ExplodedNodeVector *>()) 28080547386SJordan Rose return V->begin(); 28180547386SJordan Rose return Storage.getAddrOfPtr1(); 282fa0734ecSArgyrios Kyrtzidis } 283fa0734ecSArgyrios Kyrtzidis 28480547386SJordan Rose ExplodedNode * const *ExplodedNode::NodeGroup::end() const { 285fa0734ecSArgyrios Kyrtzidis if (getFlag()) 28680547386SJordan Rose return 0; 287fa0734ecSArgyrios Kyrtzidis 28880547386SJordan Rose const GroupStorage &Storage = reinterpret_cast<const GroupStorage &>(P); 28980547386SJordan Rose if (Storage.isNull()) 29080547386SJordan Rose return 0; 29180547386SJordan Rose if (ExplodedNodeVector *V = Storage.dyn_cast<ExplodedNodeVector *>()) 29280547386SJordan Rose return V->end(); 29380547386SJordan Rose return Storage.getAddrOfPtr1() + 1; 294fa0734ecSArgyrios Kyrtzidis } 295fa0734ecSArgyrios Kyrtzidis 296fa0734ecSArgyrios Kyrtzidis ExplodedNode *ExplodedGraph::getNode(const ProgramPoint &L, 29749b1e38eSTed Kremenek ProgramStateRef State, 29849ea5bf5SAnna Zaks bool IsSink, 29949ea5bf5SAnna Zaks bool* IsNew) { 300fa0734ecSArgyrios Kyrtzidis // Profile 'State' to determine if we already have an existing node. 301fa0734ecSArgyrios Kyrtzidis llvm::FoldingSetNodeID profile; 302fa0734ecSArgyrios Kyrtzidis void *InsertPos = 0; 303fa0734ecSArgyrios Kyrtzidis 30449ea5bf5SAnna Zaks NodeTy::Profile(profile, L, State, IsSink); 305fa0734ecSArgyrios Kyrtzidis NodeTy* V = Nodes.FindNodeOrInsertPos(profile, InsertPos); 306fa0734ecSArgyrios Kyrtzidis 307fa0734ecSArgyrios Kyrtzidis if (!V) { 308a2aa929eSTed Kremenek if (!FreeNodes.empty()) { 309a2aa929eSTed Kremenek V = FreeNodes.back(); 310a2aa929eSTed Kremenek FreeNodes.pop_back(); 311a40f8ebcSTed Kremenek } 312a40f8ebcSTed Kremenek else { 313fa0734ecSArgyrios Kyrtzidis // Allocate a new node. 314fa0734ecSArgyrios Kyrtzidis V = (NodeTy*) getAllocator().Allocate<NodeTy>(); 315a40f8ebcSTed Kremenek } 316a40f8ebcSTed Kremenek 31749ea5bf5SAnna Zaks new (V) NodeTy(L, State, IsSink); 318fa0734ecSArgyrios Kyrtzidis 319746c06d0SJordan Rose if (ReclaimNodeInterval) 32035e55fe4STed Kremenek ChangedNodes.push_back(V); 32135e55fe4STed Kremenek 322fa0734ecSArgyrios Kyrtzidis // Insert the node into the node set and return it. 323fa0734ecSArgyrios Kyrtzidis Nodes.InsertNode(V, InsertPos); 324fa0734ecSArgyrios Kyrtzidis ++NumNodes; 325fa0734ecSArgyrios Kyrtzidis 326fa0734ecSArgyrios Kyrtzidis if (IsNew) *IsNew = true; 327fa0734ecSArgyrios Kyrtzidis } 328fa0734ecSArgyrios Kyrtzidis else 329fa0734ecSArgyrios Kyrtzidis if (IsNew) *IsNew = false; 330fa0734ecSArgyrios Kyrtzidis 331fa0734ecSArgyrios Kyrtzidis return V; 332fa0734ecSArgyrios Kyrtzidis } 333fa0734ecSArgyrios Kyrtzidis 334fa0734ecSArgyrios Kyrtzidis std::pair<ExplodedGraph*, InterExplodedGraphMap*> 335fa0734ecSArgyrios Kyrtzidis ExplodedGraph::Trim(const NodeTy* const* NBeg, const NodeTy* const* NEnd, 336fa0734ecSArgyrios Kyrtzidis llvm::DenseMap<const void*, const void*> *InverseMap) const { 337fa0734ecSArgyrios Kyrtzidis 338fa0734ecSArgyrios Kyrtzidis if (NBeg == NEnd) 339fa0734ecSArgyrios Kyrtzidis return std::make_pair((ExplodedGraph*) 0, 340fa0734ecSArgyrios Kyrtzidis (InterExplodedGraphMap*) 0); 341fa0734ecSArgyrios Kyrtzidis 342fa0734ecSArgyrios Kyrtzidis assert (NBeg < NEnd); 343fa0734ecSArgyrios Kyrtzidis 344e2778999SDylan Noblesmith OwningPtr<InterExplodedGraphMap> M(new InterExplodedGraphMap()); 345fa0734ecSArgyrios Kyrtzidis 346fa0734ecSArgyrios Kyrtzidis ExplodedGraph* G = TrimInternal(NBeg, NEnd, M.get(), InverseMap); 347fa0734ecSArgyrios Kyrtzidis 348fa0734ecSArgyrios Kyrtzidis return std::make_pair(static_cast<ExplodedGraph*>(G), M.take()); 349fa0734ecSArgyrios Kyrtzidis } 350fa0734ecSArgyrios Kyrtzidis 351fa0734ecSArgyrios Kyrtzidis ExplodedGraph* 352fa0734ecSArgyrios Kyrtzidis ExplodedGraph::TrimInternal(const ExplodedNode* const* BeginSources, 353fa0734ecSArgyrios Kyrtzidis const ExplodedNode* const* EndSources, 354fa0734ecSArgyrios Kyrtzidis InterExplodedGraphMap* M, 355fa0734ecSArgyrios Kyrtzidis llvm::DenseMap<const void*, const void*> *InverseMap) const { 356fa0734ecSArgyrios Kyrtzidis 357fa0734ecSArgyrios Kyrtzidis typedef llvm::DenseSet<const ExplodedNode*> Pass1Ty; 358fa0734ecSArgyrios Kyrtzidis Pass1Ty Pass1; 359fa0734ecSArgyrios Kyrtzidis 360fa0734ecSArgyrios Kyrtzidis typedef llvm::DenseMap<const ExplodedNode*, ExplodedNode*> Pass2Ty; 361fa0734ecSArgyrios Kyrtzidis Pass2Ty& Pass2 = M->M; 362fa0734ecSArgyrios Kyrtzidis 3630e62c1ccSChris Lattner SmallVector<const ExplodedNode*, 10> WL1, WL2; 364fa0734ecSArgyrios Kyrtzidis 365fa0734ecSArgyrios Kyrtzidis // ===- Pass 1 (reverse DFS) -=== 366fa0734ecSArgyrios Kyrtzidis for (const ExplodedNode* const* I = BeginSources; I != EndSources; ++I) { 3675a751b99SJordan Rose if (*I) 368fa0734ecSArgyrios Kyrtzidis WL1.push_back(*I); 369fa0734ecSArgyrios Kyrtzidis } 370fa0734ecSArgyrios Kyrtzidis 371fa0734ecSArgyrios Kyrtzidis // Process the first worklist until it is empty. Because it is a std::list 372fa0734ecSArgyrios Kyrtzidis // it acts like a FIFO queue. 373fa0734ecSArgyrios Kyrtzidis while (!WL1.empty()) { 374fa0734ecSArgyrios Kyrtzidis const ExplodedNode *N = WL1.back(); 375fa0734ecSArgyrios Kyrtzidis WL1.pop_back(); 376fa0734ecSArgyrios Kyrtzidis 377fa0734ecSArgyrios Kyrtzidis // Have we already visited this node? If so, continue to the next one. 378fa0734ecSArgyrios Kyrtzidis if (Pass1.count(N)) 379fa0734ecSArgyrios Kyrtzidis continue; 380fa0734ecSArgyrios Kyrtzidis 381fa0734ecSArgyrios Kyrtzidis // Otherwise, mark this node as visited. 382fa0734ecSArgyrios Kyrtzidis Pass1.insert(N); 383fa0734ecSArgyrios Kyrtzidis 384fa0734ecSArgyrios Kyrtzidis // If this is a root enqueue it to the second worklist. 385fa0734ecSArgyrios Kyrtzidis if (N->Preds.empty()) { 386fa0734ecSArgyrios Kyrtzidis WL2.push_back(N); 387fa0734ecSArgyrios Kyrtzidis continue; 388fa0734ecSArgyrios Kyrtzidis } 389fa0734ecSArgyrios Kyrtzidis 390fa0734ecSArgyrios Kyrtzidis // Visit our predecessors and enqueue them. 39180547386SJordan Rose for (ExplodedNode::pred_iterator I = N->Preds.begin(), E = N->Preds.end(); 39280547386SJordan Rose I != E; ++I) 393fa0734ecSArgyrios Kyrtzidis WL1.push_back(*I); 394fa0734ecSArgyrios Kyrtzidis } 395fa0734ecSArgyrios Kyrtzidis 396fa0734ecSArgyrios Kyrtzidis // We didn't hit a root? Return with a null pointer for the new graph. 397fa0734ecSArgyrios Kyrtzidis if (WL2.empty()) 398fa0734ecSArgyrios Kyrtzidis return 0; 399fa0734ecSArgyrios Kyrtzidis 400fa0734ecSArgyrios Kyrtzidis // Create an empty graph. 401fa0734ecSArgyrios Kyrtzidis ExplodedGraph* G = MakeEmptyGraph(); 402fa0734ecSArgyrios Kyrtzidis 403fa0734ecSArgyrios Kyrtzidis // ===- Pass 2 (forward DFS to construct the new graph) -=== 404fa0734ecSArgyrios Kyrtzidis while (!WL2.empty()) { 405fa0734ecSArgyrios Kyrtzidis const ExplodedNode *N = WL2.back(); 406fa0734ecSArgyrios Kyrtzidis WL2.pop_back(); 407fa0734ecSArgyrios Kyrtzidis 408fa0734ecSArgyrios Kyrtzidis // Skip this node if we have already processed it. 409fa0734ecSArgyrios Kyrtzidis if (Pass2.find(N) != Pass2.end()) 410fa0734ecSArgyrios Kyrtzidis continue; 411fa0734ecSArgyrios Kyrtzidis 412fa0734ecSArgyrios Kyrtzidis // Create the corresponding node in the new graph and record the mapping 413fa0734ecSArgyrios Kyrtzidis // from the old node to the new node. 41449ea5bf5SAnna Zaks ExplodedNode *NewN = G->getNode(N->getLocation(), N->State, N->isSink(), 0); 415fa0734ecSArgyrios Kyrtzidis Pass2[N] = NewN; 416fa0734ecSArgyrios Kyrtzidis 417fa0734ecSArgyrios Kyrtzidis // Also record the reverse mapping from the new node to the old node. 418fa0734ecSArgyrios Kyrtzidis if (InverseMap) (*InverseMap)[NewN] = N; 419fa0734ecSArgyrios Kyrtzidis 420fa0734ecSArgyrios Kyrtzidis // If this node is a root, designate it as such in the graph. 421fa0734ecSArgyrios Kyrtzidis if (N->Preds.empty()) 422fa0734ecSArgyrios Kyrtzidis G->addRoot(NewN); 423fa0734ecSArgyrios Kyrtzidis 424fa0734ecSArgyrios Kyrtzidis // In the case that some of the intended predecessors of NewN have already 425fa0734ecSArgyrios Kyrtzidis // been created, we should hook them up as predecessors. 426fa0734ecSArgyrios Kyrtzidis 427fa0734ecSArgyrios Kyrtzidis // Walk through the predecessors of 'N' and hook up their corresponding 428fa0734ecSArgyrios Kyrtzidis // nodes in the new graph (if any) to the freshly created node. 42980547386SJordan Rose for (ExplodedNode::pred_iterator I = N->Preds.begin(), E = N->Preds.end(); 43080547386SJordan Rose I != E; ++I) { 431fa0734ecSArgyrios Kyrtzidis Pass2Ty::iterator PI = Pass2.find(*I); 432fa0734ecSArgyrios Kyrtzidis if (PI == Pass2.end()) 433fa0734ecSArgyrios Kyrtzidis continue; 434fa0734ecSArgyrios Kyrtzidis 435fa0734ecSArgyrios Kyrtzidis NewN->addPredecessor(PI->second, *G); 436fa0734ecSArgyrios Kyrtzidis } 437fa0734ecSArgyrios Kyrtzidis 438fa0734ecSArgyrios Kyrtzidis // In the case that some of the intended successors of NewN have already 439fa0734ecSArgyrios Kyrtzidis // been created, we should hook them up as successors. Otherwise, enqueue 440fa0734ecSArgyrios Kyrtzidis // the new nodes from the original graph that should have nodes created 441fa0734ecSArgyrios Kyrtzidis // in the new graph. 44280547386SJordan Rose for (ExplodedNode::succ_iterator I = N->Succs.begin(), E = N->Succs.end(); 44380547386SJordan Rose I != E; ++I) { 444fa0734ecSArgyrios Kyrtzidis Pass2Ty::iterator PI = Pass2.find(*I); 445fa0734ecSArgyrios Kyrtzidis if (PI != Pass2.end()) { 446fa0734ecSArgyrios Kyrtzidis PI->second->addPredecessor(NewN, *G); 447fa0734ecSArgyrios Kyrtzidis continue; 448fa0734ecSArgyrios Kyrtzidis } 449fa0734ecSArgyrios Kyrtzidis 450fa0734ecSArgyrios Kyrtzidis // Enqueue nodes to the worklist that were marked during pass 1. 451fa0734ecSArgyrios Kyrtzidis if (Pass1.count(*I)) 452fa0734ecSArgyrios Kyrtzidis WL2.push_back(*I); 453fa0734ecSArgyrios Kyrtzidis } 454fa0734ecSArgyrios Kyrtzidis } 455fa0734ecSArgyrios Kyrtzidis 456fa0734ecSArgyrios Kyrtzidis return G; 457fa0734ecSArgyrios Kyrtzidis } 458fa0734ecSArgyrios Kyrtzidis 45968e081d6SDavid Blaikie void InterExplodedGraphMap::anchor() { } 46068e081d6SDavid Blaikie 461fa0734ecSArgyrios Kyrtzidis ExplodedNode* 462fa0734ecSArgyrios Kyrtzidis InterExplodedGraphMap::getMappedNode(const ExplodedNode *N) const { 463fa0734ecSArgyrios Kyrtzidis llvm::DenseMap<const ExplodedNode*, ExplodedNode*>::const_iterator I = 464fa0734ecSArgyrios Kyrtzidis M.find(N); 465fa0734ecSArgyrios Kyrtzidis 466fa0734ecSArgyrios Kyrtzidis return I == M.end() ? 0 : I->second; 467fa0734ecSArgyrios Kyrtzidis } 468fa0734ecSArgyrios Kyrtzidis 469