blob: 73879214f3434a59089ecc518deca5af3fdb916d [file] [log] [blame]
Ted Kremenekfddd5182007-08-21 21:42:03 +00001//===--- CFG.cpp - Classes for representing and building CFGs----*- C++ -*-===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner0bc735f2007-12-29 19:59:25 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Ted Kremenekfddd5182007-08-21 21:42:03 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This file defines the CFG and CFGBuilder classes for representing and
11// building Control-Flow Graphs (CFGs) from ASTs.
12//
13//===----------------------------------------------------------------------===//
14
Ted Kremenekbd048782009-07-22 21:45:16 +000015#include "clang/Analysis/Support/SaveAndRestore.h"
Ted Kremeneke41611a2009-07-16 18:13:04 +000016#include "clang/Analysis/CFG.h"
Mike Stumpb978a442010-01-21 02:21:40 +000017#include "clang/AST/DeclCXX.h"
Ted Kremenekc310e932007-08-21 22:06:14 +000018#include "clang/AST/StmtVisitor.h"
Ted Kremenek42a509f2007-08-31 21:30:12 +000019#include "clang/AST/PrettyPrinter.h"
Ted Kremenekc56c0042011-02-23 05:11:46 +000020#include "clang/AST/CharUnits.h"
Benjamin Kramer6cb7c1a2009-08-23 12:08:50 +000021#include "llvm/Support/GraphWriter.h"
Benjamin Kramer6cb7c1a2009-08-23 12:08:50 +000022#include "llvm/Support/Allocator.h"
23#include "llvm/Support/Format.h"
Ted Kremenek0cebe3e2007-08-21 23:26:17 +000024#include "llvm/ADT/DenseMap.h"
Ted Kremenek19bb3562007-08-28 19:26:49 +000025#include "llvm/ADT/SmallPtrSet.h"
Ted Kremenek0ba497b2009-10-20 23:46:25 +000026#include "llvm/ADT/OwningPtr.h"
Ted Kremenek83c01da2008-01-11 00:40:29 +000027
Ted Kremenekfddd5182007-08-21 21:42:03 +000028using namespace clang;
29
30namespace {
31
Ted Kremenek9c378f72011-08-12 23:37:29 +000032static SourceLocation GetEndLoc(Decl *D) {
33 if (VarDecl *VD = dyn_cast<VarDecl>(D))
34 if (Expr *Ex = VD->getInit())
Ted Kremenekc7eb9032008-08-06 23:20:50 +000035 return Ex->getSourceRange().getEnd();
Mike Stump6d9828c2009-07-17 01:31:16 +000036 return D->getLocation();
Ted Kremenekc7eb9032008-08-06 23:20:50 +000037}
Ted Kremenekad5a8942010-08-02 23:46:59 +000038
Ted Kremenek3179a452011-03-10 01:14:11 +000039class CFGBuilder;
40
Zhanyong Wan94a3dcf2010-11-24 03:28:53 +000041/// The CFG builder uses a recursive algorithm to build the CFG. When
42/// we process an expression, sometimes we know that we must add the
43/// subexpressions as block-level expressions. For example:
44///
45/// exp1 || exp2
46///
47/// When processing the '||' expression, we know that exp1 and exp2
48/// need to be added as block-level expressions, even though they
49/// might not normally need to be. AddStmtChoice records this
50/// contextual information. If AddStmtChoice is 'NotAlwaysAdd', then
51/// the builder has an option not to add a subexpression as a
52/// block-level expression.
53///
Ted Kremenek852274d2009-12-16 03:18:58 +000054class AddStmtChoice {
55public:
Ted Kremenek892697d2010-12-16 07:46:53 +000056 enum Kind { NotAlwaysAdd = 0, AlwaysAdd = 1 };
Ted Kremenek5ba290a2010-03-02 21:43:54 +000057
Zhanyong Wan94a3dcf2010-11-24 03:28:53 +000058 AddStmtChoice(Kind a_kind = NotAlwaysAdd) : kind(a_kind) {}
Ted Kremenek5ba290a2010-03-02 21:43:54 +000059
Ted Kremenek3179a452011-03-10 01:14:11 +000060 bool alwaysAdd(CFGBuilder &builder,
61 const Stmt *stmt) const;
Zhanyong Wan94a3dcf2010-11-24 03:28:53 +000062
63 /// Return a copy of this object, except with the 'always-add' bit
64 /// set as specified.
65 AddStmtChoice withAlwaysAdd(bool alwaysAdd) const {
Ted Kremenek3179a452011-03-10 01:14:11 +000066 return AddStmtChoice(alwaysAdd ? AlwaysAdd : NotAlwaysAdd);
Zhanyong Wan94a3dcf2010-11-24 03:28:53 +000067 }
68
Ted Kremenek852274d2009-12-16 03:18:58 +000069private:
Zhanyong Wan94a3dcf2010-11-24 03:28:53 +000070 Kind kind;
Ted Kremenek852274d2009-12-16 03:18:58 +000071};
Mike Stump6d9828c2009-07-17 01:31:16 +000072
Marcin Swiderskif1308c72010-09-25 11:05:21 +000073/// LocalScope - Node in tree of local scopes created for C++ implicit
74/// destructor calls generation. It contains list of automatic variables
75/// declared in the scope and link to position in previous scope this scope
76/// began in.
77///
78/// The process of creating local scopes is as follows:
79/// - Init CFGBuilder::ScopePos with invalid position (equivalent for null),
80/// - Before processing statements in scope (e.g. CompoundStmt) create
81/// LocalScope object using CFGBuilder::ScopePos as link to previous scope
82/// and set CFGBuilder::ScopePos to the end of new scope,
Marcin Swiderski35387a02010-09-30 22:42:32 +000083/// - On every occurrence of VarDecl increase CFGBuilder::ScopePos if it points
Marcin Swiderskif1308c72010-09-25 11:05:21 +000084/// at this VarDecl,
85/// - For every normal (without jump) end of scope add to CFGBlock destructors
86/// for objects in the current scope,
87/// - For every jump add to CFGBlock destructors for objects
88/// between CFGBuilder::ScopePos and local scope position saved for jump
89/// target. Thanks to C++ restrictions on goto jumps we can be sure that
90/// jump target position will be on the path to root from CFGBuilder::ScopePos
91/// (adding any variable that doesn't need constructor to be called to
92/// LocalScope can break this assumption),
93///
94class LocalScope {
95public:
Ted Kremenekfe59b742011-02-15 02:47:45 +000096 typedef BumpVector<VarDecl*> AutomaticVarsTy;
Marcin Swiderskif1308c72010-09-25 11:05:21 +000097
98 /// const_iterator - Iterates local scope backwards and jumps to previous
Marcin Swiderski35387a02010-09-30 22:42:32 +000099 /// scope on reaching the beginning of currently iterated scope.
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000100 class const_iterator {
101 const LocalScope* Scope;
102
103 /// VarIter is guaranteed to be greater then 0 for every valid iterator.
104 /// Invalid iterator (with null Scope) has VarIter equal to 0.
105 unsigned VarIter;
106
107 public:
108 /// Create invalid iterator. Dereferencing invalid iterator is not allowed.
109 /// Incrementing invalid iterator is allowed and will result in invalid
110 /// iterator.
111 const_iterator()
112 : Scope(NULL), VarIter(0) {}
113
114 /// Create valid iterator. In case when S.Prev is an invalid iterator and
115 /// I is equal to 0, this will create invalid iterator.
116 const_iterator(const LocalScope& S, unsigned I)
117 : Scope(&S), VarIter(I) {
118 // Iterator to "end" of scope is not allowed. Handle it by going up
119 // in scopes tree possibly up to invalid iterator in the root.
120 if (VarIter == 0 && Scope)
121 *this = Scope->Prev;
122 }
123
Ted Kremenek9c378f72011-08-12 23:37:29 +0000124 VarDecl *const* operator->() const {
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000125 assert (Scope && "Dereferencing invalid iterator is not allowed");
126 assert (VarIter != 0 && "Iterator has invalid value of VarIter member");
127 return &Scope->Vars[VarIter - 1];
128 }
Ted Kremenek9c378f72011-08-12 23:37:29 +0000129 VarDecl *operator*() const {
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000130 return *this->operator->();
131 }
132
Ted Kremenek9c378f72011-08-12 23:37:29 +0000133 const_iterator &operator++() {
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000134 if (!Scope)
135 return *this;
136
137 assert (VarIter != 0 && "Iterator has invalid value of VarIter member");
138 --VarIter;
139 if (VarIter == 0)
140 *this = Scope->Prev;
141 return *this;
142 }
Marcin Swiderski35387a02010-09-30 22:42:32 +0000143 const_iterator operator++(int) {
144 const_iterator P = *this;
145 ++*this;
146 return P;
147 }
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000148
Ted Kremenek9c378f72011-08-12 23:37:29 +0000149 bool operator==(const const_iterator &rhs) const {
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000150 return Scope == rhs.Scope && VarIter == rhs.VarIter;
151 }
Ted Kremenek9c378f72011-08-12 23:37:29 +0000152 bool operator!=(const const_iterator &rhs) const {
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000153 return !(*this == rhs);
154 }
Marcin Swiderski35387a02010-09-30 22:42:32 +0000155
156 operator bool() const {
157 return *this != const_iterator();
158 }
159
160 int distance(const_iterator L);
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000161 };
162
163 friend class const_iterator;
164
165private:
Ted Kremenekfe59b742011-02-15 02:47:45 +0000166 BumpVectorContext ctx;
167
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000168 /// Automatic variables in order of declaration.
169 AutomaticVarsTy Vars;
170 /// Iterator to variable in previous scope that was declared just before
171 /// begin of this scope.
172 const_iterator Prev;
173
174public:
175 /// Constructs empty scope linked to previous scope in specified place.
Ted Kremenekfe59b742011-02-15 02:47:45 +0000176 LocalScope(BumpVectorContext &ctx, const_iterator P)
177 : ctx(ctx), Vars(ctx, 4), Prev(P) {}
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000178
179 /// Begin of scope in direction of CFG building (backwards).
180 const_iterator begin() const { return const_iterator(*this, Vars.size()); }
Marcin Swiderski35387a02010-09-30 22:42:32 +0000181
Ted Kremenek9c378f72011-08-12 23:37:29 +0000182 void addVar(VarDecl *VD) {
Ted Kremenekfe59b742011-02-15 02:47:45 +0000183 Vars.push_back(VD, ctx);
Marcin Swiderski35387a02010-09-30 22:42:32 +0000184 }
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000185};
186
Marcin Swiderski35387a02010-09-30 22:42:32 +0000187/// distance - Calculates distance from this to L. L must be reachable from this
188/// (with use of ++ operator). Cost of calculating the distance is linear w.r.t.
189/// number of scopes between this and L.
190int LocalScope::const_iterator::distance(LocalScope::const_iterator L) {
191 int D = 0;
192 const_iterator F = *this;
193 while (F.Scope != L.Scope) {
Ted Kremenek5290c802011-08-12 14:41:23 +0000194 assert (F != const_iterator()
195 && "L iterator is not reachable from F iterator.");
Marcin Swiderski35387a02010-09-30 22:42:32 +0000196 D += F.VarIter;
197 F = F.Scope->Prev;
198 }
199 D += F.VarIter - L.VarIter;
200 return D;
201}
202
203/// BlockScopePosPair - Structure for specifying position in CFG during its
204/// build process. It consists of CFGBlock that specifies position in CFG graph
205/// and LocalScope::const_iterator that specifies position in LocalScope graph.
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000206struct BlockScopePosPair {
Ted Kremenek9ce52702011-01-07 19:37:16 +0000207 BlockScopePosPair() : block(0) {}
Ted Kremenek9c378f72011-08-12 23:37:29 +0000208 BlockScopePosPair(CFGBlock *b, LocalScope::const_iterator scopePos)
Ted Kremenek9ce52702011-01-07 19:37:16 +0000209 : block(b), scopePosition(scopePos) {}
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000210
Ted Kremenek9ce52702011-01-07 19:37:16 +0000211 CFGBlock *block;
212 LocalScope::const_iterator scopePosition;
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000213};
214
Ted Kremeneke71f3d52011-03-01 23:12:55 +0000215/// TryResult - a class representing a variant over the values
216/// 'true', 'false', or 'unknown'. This is returned by tryEvaluateBool,
217/// and is used by the CFGBuilder to decide if a branch condition
218/// can be decided up front during CFG construction.
219class TryResult {
220 int X;
221public:
222 TryResult(bool b) : X(b ? 1 : 0) {}
223 TryResult() : X(-1) {}
224
225 bool isTrue() const { return X == 1; }
226 bool isFalse() const { return X == 0; }
227 bool isKnown() const { return X >= 0; }
228 void negate() {
229 assert(isKnown());
230 X ^= 0x1;
231 }
232};
233
Ted Kremeneka34ea072008-08-04 22:51:42 +0000234/// CFGBuilder - This class implements CFG construction from an AST.
Ted Kremenekfddd5182007-08-21 21:42:03 +0000235/// The builder is stateful: an instance of the builder should be used to only
236/// construct a single CFG.
237///
238/// Example usage:
239///
240/// CFGBuilder builder;
241/// CFG* cfg = builder.BuildAST(stmt1);
242///
Mike Stump6d9828c2009-07-17 01:31:16 +0000243/// CFG construction is done via a recursive walk of an AST. We actually parse
244/// the AST in reverse order so that the successor of a basic block is
245/// constructed prior to its predecessor. This allows us to nicely capture
246/// implicit fall-throughs without extra basic blocks.
Ted Kremenekc310e932007-08-21 22:06:14 +0000247///
Kovarththanan Rajaratnamba5fb5a2009-11-28 06:07:30 +0000248class CFGBuilder {
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000249 typedef BlockScopePosPair JumpTarget;
250 typedef BlockScopePosPair JumpSource;
251
Mike Stumpe5af3ce2009-07-20 23:24:15 +0000252 ASTContext *Context;
Ted Kremenek0ba497b2009-10-20 23:46:25 +0000253 llvm::OwningPtr<CFG> cfg;
Ted Kremenekee82d9b2009-10-12 20:55:07 +0000254
Ted Kremenek9c378f72011-08-12 23:37:29 +0000255 CFGBlock *Block;
256 CFGBlock *Succ;
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000257 JumpTarget ContinueJumpTarget;
258 JumpTarget BreakJumpTarget;
Ted Kremenek9c378f72011-08-12 23:37:29 +0000259 CFGBlock *SwitchTerminatedBlock;
260 CFGBlock *DefaultCaseBlock;
261 CFGBlock *TryTerminatedBlock;
Ted Kremeneke71f3d52011-03-01 23:12:55 +0000262
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000263 // Current position in local scope.
264 LocalScope::const_iterator ScopePos;
265
266 // LabelMap records the mapping from Label expressions to their jump targets.
Chris Lattnerad8dcf42011-02-17 07:39:24 +0000267 typedef llvm::DenseMap<LabelDecl*, JumpTarget> LabelMapTy;
Ted Kremenek0cebe3e2007-08-21 23:26:17 +0000268 LabelMapTy LabelMap;
Mike Stump6d9828c2009-07-17 01:31:16 +0000269
270 // A list of blocks that end with a "goto" that must be backpatched to their
271 // resolved targets upon completion of CFG construction.
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000272 typedef std::vector<JumpSource> BackpatchBlocksTy;
Ted Kremenek0cebe3e2007-08-21 23:26:17 +0000273 BackpatchBlocksTy BackpatchBlocks;
Mike Stump6d9828c2009-07-17 01:31:16 +0000274
Ted Kremenek19bb3562007-08-28 19:26:49 +0000275 // A list of labels whose address has been taken (for indirect gotos).
Chris Lattnerad8dcf42011-02-17 07:39:24 +0000276 typedef llvm::SmallPtrSet<LabelDecl*, 5> LabelSetTy;
Ted Kremenek19bb3562007-08-28 19:26:49 +0000277 LabelSetTy AddressTakenLabels;
Mike Stump6d9828c2009-07-17 01:31:16 +0000278
Zhongxing Xu49b4ef32010-09-16 03:28:18 +0000279 bool badCFG;
Ted Kremenekb8ad5ee2011-03-10 01:14:05 +0000280 const CFG::BuildOptions &BuildOpts;
Ted Kremeneke71f3d52011-03-01 23:12:55 +0000281
282 // State to track for building switch statements.
283 bool switchExclusivelyCovered;
Ted Kremenek04982472011-03-04 01:03:41 +0000284 Expr::EvalResult *switchCond;
Ted Kremenek0d28d362011-03-10 03:50:34 +0000285
286 CFG::BuildOptions::ForcedBlkExprs::value_type *cachedEntry;
Ted Kremeneka8d459e2011-03-23 21:33:21 +0000287 const Stmt *lastLookup;
Zhongxing Xu49b4ef32010-09-16 03:28:18 +0000288
Mike Stump6d9828c2009-07-17 01:31:16 +0000289public:
Ted Kremenekb8ad5ee2011-03-10 01:14:05 +0000290 explicit CFGBuilder(ASTContext *astContext,
291 const CFG::BuildOptions &buildOpts)
292 : Context(astContext), cfg(new CFG()), // crew a new CFG
293 Block(NULL), Succ(NULL),
294 SwitchTerminatedBlock(NULL), DefaultCaseBlock(NULL),
295 TryTerminatedBlock(NULL), badCFG(false), BuildOpts(buildOpts),
Ted Kremenek0d28d362011-03-10 03:50:34 +0000296 switchExclusivelyCovered(false), switchCond(0),
Ted Kremeneka8d459e2011-03-23 21:33:21 +0000297 cachedEntry(0), lastLookup(0) {}
Mike Stump6d9828c2009-07-17 01:31:16 +0000298
Ted Kremenekd4fdee32007-08-23 21:42:29 +0000299 // buildCFG - Used by external clients to construct the CFG.
Ted Kremenekb8ad5ee2011-03-10 01:14:05 +0000300 CFG* buildCFG(const Decl *D, Stmt *Statement);
Mike Stump6d9828c2009-07-17 01:31:16 +0000301
Ted Kremenek0d28d362011-03-10 03:50:34 +0000302 bool alwaysAdd(const Stmt *stmt);
303
Ted Kremenek4f880632009-07-17 22:18:43 +0000304private:
305 // Visitors to walk an AST and construct the CFG.
Ted Kremenek852274d2009-12-16 03:18:58 +0000306 CFGBlock *VisitAddrLabelExpr(AddrLabelExpr *A, AddStmtChoice asc);
307 CFGBlock *VisitBinaryOperator(BinaryOperator *B, AddStmtChoice asc);
Ted Kremenek9c378f72011-08-12 23:37:29 +0000308 CFGBlock *VisitBlockExpr(BlockExpr *E, AddStmtChoice asc);
Ted Kremenek4f880632009-07-17 22:18:43 +0000309 CFGBlock *VisitBreakStmt(BreakStmt *B);
Ted Kremenek7ea21362010-04-11 17:01:59 +0000310 CFGBlock *VisitCXXCatchStmt(CXXCatchStmt *S);
John McCall4765fa02010-12-06 08:20:24 +0000311 CFGBlock *VisitExprWithCleanups(ExprWithCleanups *E,
Marcin Swiderski8599e762010-11-03 06:19:35 +0000312 AddStmtChoice asc);
Ted Kremenek7ea21362010-04-11 17:01:59 +0000313 CFGBlock *VisitCXXThrowExpr(CXXThrowExpr *T);
314 CFGBlock *VisitCXXTryStmt(CXXTryStmt *S);
Richard Smithad762fc2011-04-14 22:09:26 +0000315 CFGBlock *VisitCXXForRangeStmt(CXXForRangeStmt *S);
Zhongxing Xua725ed42010-11-01 13:04:58 +0000316 CFGBlock *VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E,
317 AddStmtChoice asc);
Zhongxing Xu81bc7d02010-11-01 06:46:05 +0000318 CFGBlock *VisitCXXConstructExpr(CXXConstructExpr *C, AddStmtChoice asc);
Zhongxing Xua725ed42010-11-01 13:04:58 +0000319 CFGBlock *VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E,
320 AddStmtChoice asc);
Zhongxing Xu81bc7d02010-11-01 06:46:05 +0000321 CFGBlock *VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *C,
322 AddStmtChoice asc);
Ted Kremenek852274d2009-12-16 03:18:58 +0000323 CFGBlock *VisitCallExpr(CallExpr *C, AddStmtChoice asc);
Ted Kremenek4f880632009-07-17 22:18:43 +0000324 CFGBlock *VisitCaseStmt(CaseStmt *C);
Ted Kremenek852274d2009-12-16 03:18:58 +0000325 CFGBlock *VisitChooseExpr(ChooseExpr *C, AddStmtChoice asc);
Ted Kremenek3fc8ef52009-07-17 18:20:32 +0000326 CFGBlock *VisitCompoundStmt(CompoundStmt *C);
John McCall56ca35d2011-02-17 10:25:35 +0000327 CFGBlock *VisitConditionalOperator(AbstractConditionalOperator *C,
328 AddStmtChoice asc);
Ted Kremenek3fc8ef52009-07-17 18:20:32 +0000329 CFGBlock *VisitContinueStmt(ContinueStmt *C);
Ted Kremenek4f880632009-07-17 22:18:43 +0000330 CFGBlock *VisitDeclStmt(DeclStmt *DS);
Ted Kremenek9c378f72011-08-12 23:37:29 +0000331 CFGBlock *VisitDeclSubExpr(DeclStmt *DS);
Ted Kremenek3fc8ef52009-07-17 18:20:32 +0000332 CFGBlock *VisitDefaultStmt(DefaultStmt *D);
333 CFGBlock *VisitDoStmt(DoStmt *D);
334 CFGBlock *VisitForStmt(ForStmt *F);
Ted Kremenek9c378f72011-08-12 23:37:29 +0000335 CFGBlock *VisitGotoStmt(GotoStmt *G);
Ted Kremenek4f880632009-07-17 22:18:43 +0000336 CFGBlock *VisitIfStmt(IfStmt *I);
Zhongxing Xua725ed42010-11-01 13:04:58 +0000337 CFGBlock *VisitImplicitCastExpr(ImplicitCastExpr *E, AddStmtChoice asc);
Ted Kremenek4f880632009-07-17 22:18:43 +0000338 CFGBlock *VisitIndirectGotoStmt(IndirectGotoStmt *I);
339 CFGBlock *VisitLabelStmt(LabelStmt *L);
Ted Kremenek115c1b92010-04-11 17:02:10 +0000340 CFGBlock *VisitMemberExpr(MemberExpr *M, AddStmtChoice asc);
Ted Kremenek4f880632009-07-17 22:18:43 +0000341 CFGBlock *VisitObjCAtCatchStmt(ObjCAtCatchStmt *S);
342 CFGBlock *VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S);
343 CFGBlock *VisitObjCAtThrowStmt(ObjCAtThrowStmt *S);
344 CFGBlock *VisitObjCAtTryStmt(ObjCAtTryStmt *S);
345 CFGBlock *VisitObjCForCollectionStmt(ObjCForCollectionStmt *S);
Ted Kremenek9c378f72011-08-12 23:37:29 +0000346 CFGBlock *VisitReturnStmt(ReturnStmt *R);
John McCall4b9c2d22011-11-06 09:01:30 +0000347 CFGBlock *VisitPseudoObjectExpr(PseudoObjectExpr *E);
Peter Collingbournef4e3cfb2011-03-11 19:24:49 +0000348 CFGBlock *VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E,
349 AddStmtChoice asc);
Ted Kremenek852274d2009-12-16 03:18:58 +0000350 CFGBlock *VisitStmtExpr(StmtExpr *S, AddStmtChoice asc);
Ted Kremenek4f880632009-07-17 22:18:43 +0000351 CFGBlock *VisitSwitchStmt(SwitchStmt *S);
Zhanyong Wan99cae5b2010-11-22 08:45:56 +0000352 CFGBlock *VisitUnaryOperator(UnaryOperator *U, AddStmtChoice asc);
Ted Kremenek4f880632009-07-17 22:18:43 +0000353 CFGBlock *VisitWhileStmt(WhileStmt *W);
Mike Stumpcd7bf232009-07-17 01:04:31 +0000354
Ted Kremenek852274d2009-12-16 03:18:58 +0000355 CFGBlock *Visit(Stmt *S, AddStmtChoice asc = AddStmtChoice::NotAlwaysAdd);
356 CFGBlock *VisitStmt(Stmt *S, AddStmtChoice asc);
Ted Kremenek9c378f72011-08-12 23:37:29 +0000357 CFGBlock *VisitChildren(Stmt *S);
Mike Stumpcd7bf232009-07-17 01:04:31 +0000358
Marcin Swiderski8599e762010-11-03 06:19:35 +0000359 // Visitors to walk an AST and generate destructors of temporaries in
360 // full expression.
361 CFGBlock *VisitForTemporaryDtors(Stmt *E, bool BindToTemporary = false);
362 CFGBlock *VisitChildrenForTemporaryDtors(Stmt *E);
363 CFGBlock *VisitBinaryOperatorForTemporaryDtors(BinaryOperator *E);
364 CFGBlock *VisitCXXBindTemporaryExprForTemporaryDtors(CXXBindTemporaryExpr *E,
365 bool BindToTemporary);
John McCall56ca35d2011-02-17 10:25:35 +0000366 CFGBlock *
367 VisitConditionalOperatorForTemporaryDtors(AbstractConditionalOperator *E,
368 bool BindToTemporary);
Marcin Swiderski8599e762010-11-03 06:19:35 +0000369
Ted Kremenek274f4332008-04-28 18:00:46 +0000370 // NYS == Not Yet Supported
Ted Kremenek9c378f72011-08-12 23:37:29 +0000371 CFGBlock *NYS() {
Ted Kremenek4102af92008-03-13 03:04:22 +0000372 badCFG = true;
373 return Block;
374 }
Mike Stump6d9828c2009-07-17 01:31:16 +0000375
Ted Kremenek4f880632009-07-17 22:18:43 +0000376 void autoCreateBlock() { if (!Block) Block = createBlock(); }
377 CFGBlock *createBlock(bool add_successor = true);
Chandler Carruthdba3fb52011-09-13 09:13:49 +0000378 CFGBlock *createNoReturnBlock();
Zhongxing Xud438b3d2010-09-06 07:32:31 +0000379
Zhongxing Xudf119892010-06-03 06:43:23 +0000380 CFGBlock *addStmt(Stmt *S) {
381 return Visit(S, AddStmtChoice::AlwaysAdd);
Ted Kremenek852274d2009-12-16 03:18:58 +0000382 }
Sean Huntcbb67482011-01-08 20:30:50 +0000383 CFGBlock *addInitializer(CXXCtorInitializer *I);
Zhongxing Xu6a16a302010-10-01 03:22:39 +0000384 void addAutomaticObjDtors(LocalScope::const_iterator B,
Ted Kremenek9c378f72011-08-12 23:37:29 +0000385 LocalScope::const_iterator E, Stmt *S);
Marcin Swiderski7c625d82010-10-05 05:37:00 +0000386 void addImplicitDtorsForDestructor(const CXXDestructorDecl *DD);
Ted Kremenekad5a8942010-08-02 23:46:59 +0000387
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000388 // Local scopes creation.
389 LocalScope* createOrReuseLocalScope(LocalScope* Scope);
390
Ted Kremenek9c378f72011-08-12 23:37:29 +0000391 void addLocalScopeForStmt(Stmt *S);
392 LocalScope* addLocalScopeForDeclStmt(DeclStmt *DS, LocalScope* Scope = NULL);
393 LocalScope* addLocalScopeForVarDecl(VarDecl *VD, LocalScope* Scope = NULL);
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000394
Ted Kremenek9c378f72011-08-12 23:37:29 +0000395 void addLocalScopeAndDtors(Stmt *S);
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000396
397 // Interface to CFGBlock - adding CFGElements.
Ted Kremenekd40066b2011-04-04 23:29:12 +0000398 void appendStmt(CFGBlock *B, const Stmt *S) {
Ted Kremenek74fb1a42011-07-19 14:18:43 +0000399 if (alwaysAdd(S) && cachedEntry)
Ted Kremenek0d28d362011-03-10 03:50:34 +0000400 cachedEntry->second = B;
Ted Kremenek0d28d362011-03-10 03:50:34 +0000401
Jordy Roseac73ea82011-06-10 08:49:37 +0000402 // All block-level expressions should have already been IgnoreParens()ed.
403 assert(!isa<Expr>(S) || cast<Expr>(S)->IgnoreParens() == S);
Ted Kremenekd40066b2011-04-04 23:29:12 +0000404 B->appendStmt(const_cast<Stmt*>(S), cfg->getBumpVectorContext());
Ted Kremenekee82d9b2009-10-12 20:55:07 +0000405 }
Sean Huntcbb67482011-01-08 20:30:50 +0000406 void appendInitializer(CFGBlock *B, CXXCtorInitializer *I) {
Marcin Swiderski82bc3fd2010-10-04 03:38:22 +0000407 B->appendInitializer(I, cfg->getBumpVectorContext());
408 }
Marcin Swiderski7c625d82010-10-05 05:37:00 +0000409 void appendBaseDtor(CFGBlock *B, const CXXBaseSpecifier *BS) {
410 B->appendBaseDtor(BS, cfg->getBumpVectorContext());
411 }
412 void appendMemberDtor(CFGBlock *B, FieldDecl *FD) {
413 B->appendMemberDtor(FD, cfg->getBumpVectorContext());
414 }
Marcin Swiderski8599e762010-11-03 06:19:35 +0000415 void appendTemporaryDtor(CFGBlock *B, CXXBindTemporaryExpr *E) {
416 B->appendTemporaryDtor(E, cfg->getBumpVectorContext());
417 }
Chandler Carruthc8cfc742011-09-13 06:09:01 +0000418 void appendAutomaticObjDtor(CFGBlock *B, VarDecl *VD, Stmt *S) {
419 B->appendAutomaticObjDtor(VD, S, cfg->getBumpVectorContext());
420 }
Ted Kremenekad5a8942010-08-02 23:46:59 +0000421
Ted Kremenek9c378f72011-08-12 23:37:29 +0000422 void prependAutomaticObjDtorsWithTerminator(CFGBlock *Blk,
Marcin Swiderski53de1342010-09-30 22:54:37 +0000423 LocalScope::const_iterator B, LocalScope::const_iterator E);
424
Ted Kremenek0a3ed312010-12-17 04:44:39 +0000425 void addSuccessor(CFGBlock *B, CFGBlock *S) {
Ted Kremenekee82d9b2009-10-12 20:55:07 +0000426 B->addSuccessor(S, cfg->getBumpVectorContext());
427 }
Mike Stump1eb44332009-09-09 15:08:12 +0000428
Ted Kremeneke71f3d52011-03-01 23:12:55 +0000429 /// Try and evaluate an expression to an integer constant.
430 bool tryEvaluate(Expr *S, Expr::EvalResult &outResult) {
431 if (!BuildOpts.PruneTriviallyFalseEdges)
432 return false;
433 return !S->isTypeDependent() &&
Ted Kremenekf8adeef2011-04-04 20:30:58 +0000434 !S->isValueDependent() &&
Richard Smith51f47082011-10-29 00:50:52 +0000435 S->EvaluateAsRValue(outResult, *Context);
Ted Kremeneke71f3d52011-03-01 23:12:55 +0000436 }
Mike Stump1eb44332009-09-09 15:08:12 +0000437
Ted Kremenek0a3ed312010-12-17 04:44:39 +0000438 /// tryEvaluateBool - Try and evaluate the Stmt and return 0 or 1
Mike Stump00998a02009-07-23 23:25:26 +0000439 /// if we can evaluate to a known value, otherwise return -1.
Ted Kremenek0a3ed312010-12-17 04:44:39 +0000440 TryResult tryEvaluateBool(Expr *S) {
Richard Smith85df96c2011-10-14 20:22:00 +0000441 bool Result;
442 if (!BuildOpts.PruneTriviallyFalseEdges ||
443 S->isTypeDependent() || S->isValueDependent() ||
444 !S->EvaluateAsBooleanCondition(Result, *Context))
Ted Kremeneke71f3d52011-03-01 23:12:55 +0000445 return TryResult();
Richard Smith85df96c2011-10-14 20:22:00 +0000446 return Result;
Mike Stump00998a02009-07-23 23:25:26 +0000447 }
Ted Kremeneke71f3d52011-03-01 23:12:55 +0000448
Ted Kremenekfddd5182007-08-21 21:42:03 +0000449};
Mike Stump6d9828c2009-07-17 01:31:16 +0000450
Ted Kremenek0d28d362011-03-10 03:50:34 +0000451inline bool AddStmtChoice::alwaysAdd(CFGBuilder &builder,
452 const Stmt *stmt) const {
453 return builder.alwaysAdd(stmt) || kind == AlwaysAdd;
454}
Ted Kremeneka8d459e2011-03-23 21:33:21 +0000455
Ted Kremenek0d28d362011-03-10 03:50:34 +0000456bool CFGBuilder::alwaysAdd(const Stmt *stmt) {
Ted Kremenek74fb1a42011-07-19 14:18:43 +0000457 bool shouldAdd = BuildOpts.alwaysAdd(stmt);
458
Ted Kremenek0d28d362011-03-10 03:50:34 +0000459 if (!BuildOpts.forcedBlkExprs)
Ted Kremenek74fb1a42011-07-19 14:18:43 +0000460 return shouldAdd;
Ted Kremeneka8d459e2011-03-23 21:33:21 +0000461
462 if (lastLookup == stmt) {
463 if (cachedEntry) {
464 assert(cachedEntry->first == stmt);
465 return true;
466 }
Ted Kremenek74fb1a42011-07-19 14:18:43 +0000467 return shouldAdd;
Ted Kremeneka8d459e2011-03-23 21:33:21 +0000468 }
Ted Kremenek0d28d362011-03-10 03:50:34 +0000469
Ted Kremeneka8d459e2011-03-23 21:33:21 +0000470 lastLookup = stmt;
471
472 // Perform the lookup!
Ted Kremenek0d28d362011-03-10 03:50:34 +0000473 CFG::BuildOptions::ForcedBlkExprs *fb = *BuildOpts.forcedBlkExprs;
474
Ted Kremeneka8d459e2011-03-23 21:33:21 +0000475 if (!fb) {
476 // No need to update 'cachedEntry', since it will always be null.
477 assert(cachedEntry == 0);
Ted Kremenek74fb1a42011-07-19 14:18:43 +0000478 return shouldAdd;
Ted Kremeneka8d459e2011-03-23 21:33:21 +0000479 }
Ted Kremenek0d28d362011-03-10 03:50:34 +0000480
481 CFG::BuildOptions::ForcedBlkExprs::iterator itr = fb->find(stmt);
Ted Kremeneka8d459e2011-03-23 21:33:21 +0000482 if (itr == fb->end()) {
483 cachedEntry = 0;
Ted Kremenek74fb1a42011-07-19 14:18:43 +0000484 return shouldAdd;
Ted Kremeneka8d459e2011-03-23 21:33:21 +0000485 }
486
Ted Kremenek0d28d362011-03-10 03:50:34 +0000487 cachedEntry = &*itr;
488 return true;
Ted Kremenek3179a452011-03-10 01:14:11 +0000489}
490
Douglas Gregor898574e2008-12-05 23:32:09 +0000491// FIXME: Add support for dependent-sized array types in C++?
492// Does it even make sense to build a CFG for an uninstantiated template?
John McCallf4c73712011-01-19 06:33:43 +0000493static const VariableArrayType *FindVA(const Type *t) {
494 while (const ArrayType *vt = dyn_cast<ArrayType>(t)) {
495 if (const VariableArrayType *vat = dyn_cast<VariableArrayType>(vt))
Ted Kremenek610a09e2008-09-26 22:58:57 +0000496 if (vat->getSizeExpr())
497 return vat;
Mike Stump6d9828c2009-07-17 01:31:16 +0000498
Ted Kremenek610a09e2008-09-26 22:58:57 +0000499 t = vt->getElementType().getTypePtr();
500 }
Mike Stump6d9828c2009-07-17 01:31:16 +0000501
Ted Kremenek610a09e2008-09-26 22:58:57 +0000502 return 0;
503}
Mike Stump6d9828c2009-07-17 01:31:16 +0000504
505/// BuildCFG - Constructs a CFG from an AST (a Stmt*). The AST can represent an
506/// arbitrary statement. Examples include a single expression or a function
507/// body (compound statement). The ownership of the returned CFG is
508/// transferred to the caller. If CFG construction fails, this method returns
509/// NULL.
Ted Kremenek9c378f72011-08-12 23:37:29 +0000510CFG* CFGBuilder::buildCFG(const Decl *D, Stmt *Statement) {
Ted Kremenek0ba497b2009-10-20 23:46:25 +0000511 assert(cfg.get());
Ted Kremenek4f880632009-07-17 22:18:43 +0000512 if (!Statement)
513 return NULL;
Ted Kremenekd4fdee32007-08-23 21:42:29 +0000514
Mike Stump6d9828c2009-07-17 01:31:16 +0000515 // Create an empty block that will serve as the exit block for the CFG. Since
516 // this is the first block added to the CFG, it will be implicitly registered
517 // as the exit block.
Ted Kremenek49af7cb2007-08-27 19:46:09 +0000518 Succ = createBlock();
Ted Kremenekee82d9b2009-10-12 20:55:07 +0000519 assert(Succ == &cfg->getExit());
Ted Kremenek49af7cb2007-08-27 19:46:09 +0000520 Block = NULL; // the EXIT block is empty. Create all other blocks lazily.
Mike Stump6d9828c2009-07-17 01:31:16 +0000521
Marcin Swiderski7c625d82010-10-05 05:37:00 +0000522 if (BuildOpts.AddImplicitDtors)
523 if (const CXXDestructorDecl *DD = dyn_cast_or_null<CXXDestructorDecl>(D))
524 addImplicitDtorsForDestructor(DD);
525
Ted Kremenekd4fdee32007-08-23 21:42:29 +0000526 // Visit the statements and create the CFG.
Zhongxing Xu1b3b7cb2010-09-06 07:04:06 +0000527 CFGBlock *B = addStmt(Statement);
528
529 if (badCFG)
530 return NULL;
531
Marcin Swiderski82bc3fd2010-10-04 03:38:22 +0000532 // For C++ constructor add initializers to CFG.
533 if (const CXXConstructorDecl *CD = dyn_cast_or_null<CXXConstructorDecl>(D)) {
534 for (CXXConstructorDecl::init_const_reverse_iterator I = CD->init_rbegin(),
535 E = CD->init_rend(); I != E; ++I) {
536 B = addInitializer(*I);
537 if (badCFG)
538 return NULL;
539 }
540 }
541
Zhongxing Xu1b3b7cb2010-09-06 07:04:06 +0000542 if (B)
543 Succ = B;
Mike Stumpb978a442010-01-21 02:21:40 +0000544
Zhongxing Xu1b3b7cb2010-09-06 07:04:06 +0000545 // Backpatch the gotos whose label -> block mappings we didn't know when we
546 // encountered them.
547 for (BackpatchBlocksTy::iterator I = BackpatchBlocks.begin(),
548 E = BackpatchBlocks.end(); I != E; ++I ) {
Mike Stump6d9828c2009-07-17 01:31:16 +0000549
Ted Kremenek9c378f72011-08-12 23:37:29 +0000550 CFGBlock *B = I->block;
551 GotoStmt *G = cast<GotoStmt>(B->getTerminator());
Zhongxing Xu1b3b7cb2010-09-06 07:04:06 +0000552 LabelMapTy::iterator LI = LabelMap.find(G->getLabel());
Mike Stump6d9828c2009-07-17 01:31:16 +0000553
Zhongxing Xu1b3b7cb2010-09-06 07:04:06 +0000554 // If there is no target for the goto, then we are looking at an
555 // incomplete AST. Handle this by not registering a successor.
556 if (LI == LabelMap.end()) continue;
Ted Kremenekd4fdee32007-08-23 21:42:29 +0000557
Marcin Swiderskif1308c72010-09-25 11:05:21 +0000558 JumpTarget JT = LI->second;
Ted Kremenek9ce52702011-01-07 19:37:16 +0000559 prependAutomaticObjDtorsWithTerminator(B, I->scopePosition,
560 JT.scopePosition);
561 addSuccessor(B, JT.block);
Zhongxing Xu1b3b7cb2010-09-06 07:04:06 +0000562 }
563
564 // Add successors to the Indirect Goto Dispatch block (if we have one).
Ted Kremenek9c378f72011-08-12 23:37:29 +0000565 if (CFGBlock *B = cfg->getIndirectGotoBlock())
Zhongxing Xu1b3b7cb2010-09-06 07:04:06 +0000566 for (LabelSetTy::iterator I = AddressTakenLabels.begin(),
567 E = AddressTakenLabels.end(); I != E; ++I ) {
568
569 // Lookup the target block.
570 LabelMapTy::iterator LI = LabelMap.find(*I);
571
572 // If there is no target block that contains label, then we are looking
573 // at an incomplete AST. Handle this by not registering a successor.
Ted Kremenekd4fdee32007-08-23 21:42:29 +0000574 if (LI == LabelMap.end()) continue;
Zhongxing Xu1b3b7cb2010-09-06 07:04:06 +0000575
Ted Kremenek9ce52702011-01-07 19:37:16 +0000576 addSuccessor(B, LI->second.block);
Ted Kremenek19bb3562007-08-28 19:26:49 +0000577 }
Mike Stump6d9828c2009-07-17 01:31:16 +0000578
Mike Stump6d9828c2009-07-17 01:31:16 +0000579 // Create an empty entry block that has no predecessors.
Ted Kremenek322f58d2007-09-26 21:23:31 +0000580 cfg->setEntry(createBlock());
Mike Stump6d9828c2009-07-17 01:31:16 +0000581
Zhongxing Xu1b3b7cb2010-09-06 07:04:06 +0000582 return cfg.take();
Ted Kremenekd4fdee32007-08-23 21:42:29 +0000583}
Mike Stump6d9828c2009-07-17 01:31:16 +0000584
Ted Kremenekd4fdee32007-08-23 21:42:29 +0000585/// createBlock - Used to lazily create blocks that are connected
586/// to the current (global) succcessor.
Ted Kremenek9c378f72011-08-12 23:37:29 +0000587CFGBlock *CFGBuilder::createBlock(bool add_successor) {
588 CFGBlock *B = cfg->createBlock();
Ted Kremenek4f880632009-07-17 22:18:43 +0000589 if (add_successor && Succ)
Ted Kremenek0a3ed312010-12-17 04:44:39 +0000590 addSuccessor(B, Succ);
Ted Kremenekd4fdee32007-08-23 21:42:29 +0000591 return B;
592}
Mike Stump6d9828c2009-07-17 01:31:16 +0000593
Chandler Carruthdba3fb52011-09-13 09:13:49 +0000594/// createNoReturnBlock - Used to create a block is a 'noreturn' point in the
595/// CFG. It is *not* connected to the current (global) successor, and instead
596/// directly tied to the exit block in order to be reachable.
597CFGBlock *CFGBuilder::createNoReturnBlock() {
598 CFGBlock *B = createBlock(false);
Chandler Carruth83754162011-09-13 09:53:55 +0000599 B->setHasNoReturnElement();
Chandler Carruthdba3fb52011-09-13 09:13:49 +0000600 addSuccessor(B, &cfg->getExit());
601 return B;
602}
603
Marcin Swiderski82bc3fd2010-10-04 03:38:22 +0000604/// addInitializer - Add C++ base or member initializer element to CFG.
Sean Huntcbb67482011-01-08 20:30:50 +0000605CFGBlock *CFGBuilder::addInitializer(CXXCtorInitializer *I) {
Marcin Swiderski82bc3fd2010-10-04 03:38:22 +0000606 if (!BuildOpts.AddInitializers)
607 return Block;
608
Marcin Swiderski8599e762010-11-03 06:19:35 +0000609 bool IsReference = false;
610 bool HasTemporaries = false;
611
612 // Destructors of temporaries in initialization expression should be called
613 // after initialization finishes.
614 Expr *Init = I->getInit();
615 if (Init) {
Francois Pichet00eb3f92010-12-04 09:14:42 +0000616 if (FieldDecl *FD = I->getAnyMember())
Marcin Swiderski8599e762010-11-03 06:19:35 +0000617 IsReference = FD->getType()->isReferenceType();
John McCall4765fa02010-12-06 08:20:24 +0000618 HasTemporaries = isa<ExprWithCleanups>(Init);
Marcin Swiderski8599e762010-11-03 06:19:35 +0000619
620 if (BuildOpts.AddImplicitDtors && HasTemporaries) {
621 // Generate destructors for temporaries in initialization expression.
John McCall4765fa02010-12-06 08:20:24 +0000622 VisitForTemporaryDtors(cast<ExprWithCleanups>(Init)->getSubExpr(),
Marcin Swiderski8599e762010-11-03 06:19:35 +0000623 IsReference);
624 }
625 }
626
Marcin Swiderski82bc3fd2010-10-04 03:38:22 +0000627 autoCreateBlock();
628 appendInitializer(Block, I);
629
Marcin Swiderski8599e762010-11-03 06:19:35 +0000630 if (Init) {
Ted Kremenek892697d2010-12-16 07:46:53 +0000631 if (HasTemporaries) {
Marcin Swiderski8599e762010-11-03 06:19:35 +0000632 // For expression with temporaries go directly to subexpression to omit
633 // generating destructors for the second time.
Ted Kremenek892697d2010-12-16 07:46:53 +0000634 return Visit(cast<ExprWithCleanups>(Init)->getSubExpr());
635 }
636 return Visit(Init);
Marcin Swiderski82bc3fd2010-10-04 03:38:22 +0000637 }
Marcin Swiderski8599e762010-11-03 06:19:35 +0000638
Marcin Swiderski82bc3fd2010-10-04 03:38:22 +0000639 return Block;
640}
641
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000642/// addAutomaticObjDtors - Add to current block automatic objects destructors
643/// for objects in range of local scope positions. Use S as trigger statement
644/// for destructors.
Zhongxing Xu6a16a302010-10-01 03:22:39 +0000645void CFGBuilder::addAutomaticObjDtors(LocalScope::const_iterator B,
Ted Kremenek9c378f72011-08-12 23:37:29 +0000646 LocalScope::const_iterator E, Stmt *S) {
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000647 if (!BuildOpts.AddImplicitDtors)
Zhongxing Xu6a16a302010-10-01 03:22:39 +0000648 return;
649
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000650 if (B == E)
Zhongxing Xu6a16a302010-10-01 03:22:39 +0000651 return;
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000652
Chandler Carruthc8cfc742011-09-13 06:09:01 +0000653 CFGBlock::iterator InsertPos;
654
655 // We need to append the destructors in reverse order, but any one of them
656 // may be a no-return destructor which changes the CFG. As a result, buffer
657 // this sequence up and replay them in reverse order when appending onto the
658 // CFGBlock(s).
659 SmallVector<VarDecl*, 10> Decls;
660 Decls.reserve(B.distance(E));
661 for (LocalScope::const_iterator I = B; I != E; ++I)
662 Decls.push_back(*I);
663
664 for (SmallVectorImpl<VarDecl*>::reverse_iterator I = Decls.rbegin(),
665 E = Decls.rend();
666 I != E; ++I) {
667 // If this destructor is marked as a no-return destructor, we need to
668 // create a new block for the destructor which does not have as a successor
669 // anything built thus far: control won't flow out of this block.
670 QualType Ty = (*I)->getType().getNonReferenceType();
671 if (const ArrayType *AT = Context->getAsArrayType(Ty))
672 Ty = AT->getElementType();
673 const CXXDestructorDecl *Dtor = Ty->getAsCXXRecordDecl()->getDestructor();
Chandler Carruthdba3fb52011-09-13 09:13:49 +0000674 if (cast<FunctionType>(Dtor->getType())->getNoReturnAttr())
675 Block = createNoReturnBlock();
676 else
Chandler Carruthc8cfc742011-09-13 06:09:01 +0000677 autoCreateBlock();
Chandler Carruthc8cfc742011-09-13 06:09:01 +0000678
679 appendAutomaticObjDtor(Block, *I, S);
680 }
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000681}
682
Marcin Swiderski7c625d82010-10-05 05:37:00 +0000683/// addImplicitDtorsForDestructor - Add implicit destructors generated for
684/// base and member objects in destructor.
685void CFGBuilder::addImplicitDtorsForDestructor(const CXXDestructorDecl *DD) {
686 assert (BuildOpts.AddImplicitDtors
687 && "Can be called only when dtors should be added");
688 const CXXRecordDecl *RD = DD->getParent();
689
690 // At the end destroy virtual base objects.
691 for (CXXRecordDecl::base_class_const_iterator VI = RD->vbases_begin(),
692 VE = RD->vbases_end(); VI != VE; ++VI) {
693 const CXXRecordDecl *CD = VI->getType()->getAsCXXRecordDecl();
694 if (!CD->hasTrivialDestructor()) {
695 autoCreateBlock();
696 appendBaseDtor(Block, VI);
697 }
698 }
699
700 // Before virtual bases destroy direct base objects.
701 for (CXXRecordDecl::base_class_const_iterator BI = RD->bases_begin(),
702 BE = RD->bases_end(); BI != BE; ++BI) {
703 if (!BI->isVirtual()) {
704 const CXXRecordDecl *CD = BI->getType()->getAsCXXRecordDecl();
705 if (!CD->hasTrivialDestructor()) {
706 autoCreateBlock();
707 appendBaseDtor(Block, BI);
708 }
709 }
710 }
711
712 // First destroy member objects.
713 for (CXXRecordDecl::field_iterator FI = RD->field_begin(),
714 FE = RD->field_end(); FI != FE; ++FI) {
Marcin Swiderski8c5e5d62010-10-25 07:05:54 +0000715 // Check for constant size array. Set type to array element type.
716 QualType QT = FI->getType();
717 if (const ConstantArrayType *AT = Context->getAsConstantArrayType(QT)) {
718 if (AT->getSize() == 0)
719 continue;
720 QT = AT->getElementType();
721 }
722
723 if (const CXXRecordDecl *CD = QT->getAsCXXRecordDecl())
Marcin Swiderski7c625d82010-10-05 05:37:00 +0000724 if (!CD->hasTrivialDestructor()) {
725 autoCreateBlock();
726 appendMemberDtor(Block, *FI);
727 }
728 }
729}
730
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000731/// createOrReuseLocalScope - If Scope is NULL create new LocalScope. Either
732/// way return valid LocalScope object.
733LocalScope* CFGBuilder::createOrReuseLocalScope(LocalScope* Scope) {
734 if (!Scope) {
Ted Kremenekfe59b742011-02-15 02:47:45 +0000735 llvm::BumpPtrAllocator &alloc = cfg->getAllocator();
736 Scope = alloc.Allocate<LocalScope>();
737 BumpVectorContext ctx(alloc);
738 new (Scope) LocalScope(ctx, ScopePos);
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000739 }
740 return Scope;
741}
742
743/// addLocalScopeForStmt - Add LocalScope to local scopes tree for statement
Zhongxing Xu02acdfa2010-10-01 03:00:16 +0000744/// that should create implicit scope (e.g. if/else substatements).
Ted Kremenek9c378f72011-08-12 23:37:29 +0000745void CFGBuilder::addLocalScopeForStmt(Stmt *S) {
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000746 if (!BuildOpts.AddImplicitDtors)
Zhongxing Xu02acdfa2010-10-01 03:00:16 +0000747 return;
748
749 LocalScope *Scope = 0;
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000750
751 // For compound statement we will be creating explicit scope.
Chris Lattnerad8dcf42011-02-17 07:39:24 +0000752 if (CompoundStmt *CS = dyn_cast<CompoundStmt>(S)) {
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000753 for (CompoundStmt::body_iterator BI = CS->body_begin(), BE = CS->body_end()
754 ; BI != BE; ++BI) {
Chandler Carrutha1364be2011-09-10 00:02:34 +0000755 Stmt *SI = (*BI)->stripLabelLikeStatements();
Chris Lattnerad8dcf42011-02-17 07:39:24 +0000756 if (DeclStmt *DS = dyn_cast<DeclStmt>(SI))
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000757 Scope = addLocalScopeForDeclStmt(DS, Scope);
758 }
Zhongxing Xu02acdfa2010-10-01 03:00:16 +0000759 return;
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000760 }
761
762 // For any other statement scope will be implicit and as such will be
763 // interesting only for DeclStmt.
Chandler Carrutha1364be2011-09-10 00:02:34 +0000764 if (DeclStmt *DS = dyn_cast<DeclStmt>(S->stripLabelLikeStatements()))
Zhongxing Xub6edff52010-10-01 03:09:09 +0000765 addLocalScopeForDeclStmt(DS);
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000766}
767
768/// addLocalScopeForDeclStmt - Add LocalScope for declaration statement. Will
769/// reuse Scope if not NULL.
Ted Kremenek9c378f72011-08-12 23:37:29 +0000770LocalScope* CFGBuilder::addLocalScopeForDeclStmt(DeclStmt *DS,
Zhongxing Xub6edff52010-10-01 03:09:09 +0000771 LocalScope* Scope) {
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000772 if (!BuildOpts.AddImplicitDtors)
773 return Scope;
774
775 for (DeclStmt::decl_iterator DI = DS->decl_begin(), DE = DS->decl_end()
776 ; DI != DE; ++DI) {
Ted Kremenek9c378f72011-08-12 23:37:29 +0000777 if (VarDecl *VD = dyn_cast<VarDecl>(*DI))
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000778 Scope = addLocalScopeForVarDecl(VD, Scope);
779 }
780 return Scope;
781}
782
783/// addLocalScopeForVarDecl - Add LocalScope for variable declaration. It will
784/// create add scope for automatic objects and temporary objects bound to
785/// const reference. Will reuse Scope if not NULL.
Ted Kremenek9c378f72011-08-12 23:37:29 +0000786LocalScope* CFGBuilder::addLocalScopeForVarDecl(VarDecl *VD,
Zhongxing Xub6edff52010-10-01 03:09:09 +0000787 LocalScope* Scope) {
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000788 if (!BuildOpts.AddImplicitDtors)
789 return Scope;
790
791 // Check if variable is local.
792 switch (VD->getStorageClass()) {
793 case SC_None:
794 case SC_Auto:
795 case SC_Register:
796 break;
797 default: return Scope;
798 }
799
800 // Check for const references bound to temporary. Set type to pointee.
801 QualType QT = VD->getType();
802 if (const ReferenceType* RT = QT.getTypePtr()->getAs<ReferenceType>()) {
803 QT = RT->getPointeeType();
804 if (!QT.isConstQualified())
805 return Scope;
Douglas Gregor03e80032011-06-21 17:03:29 +0000806 if (!VD->extendsLifetimeOfTemporary())
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000807 return Scope;
808 }
809
Marcin Swiderskib1c52872010-10-25 07:00:40 +0000810 // Check for constant size array. Set type to array element type.
811 if (const ConstantArrayType *AT = Context->getAsConstantArrayType(QT)) {
812 if (AT->getSize() == 0)
813 return Scope;
814 QT = AT->getElementType();
815 }
Zhongxing Xu4e493e02010-10-05 08:38:06 +0000816
Marcin Swiderskib1c52872010-10-25 07:00:40 +0000817 // Check if type is a C++ class with non-trivial destructor.
Ted Kremenek9c378f72011-08-12 23:37:29 +0000818 if (const CXXRecordDecl *CD = QT->getAsCXXRecordDecl())
Zhongxing Xu4e493e02010-10-05 08:38:06 +0000819 if (!CD->hasTrivialDestructor()) {
820 // Add the variable to scope
821 Scope = createOrReuseLocalScope(Scope);
822 Scope->addVar(VD);
823 ScopePos = Scope->begin();
824 }
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000825 return Scope;
826}
827
828/// addLocalScopeAndDtors - For given statement add local scope for it and
829/// add destructors that will cleanup the scope. Will reuse Scope if not NULL.
Ted Kremenek9c378f72011-08-12 23:37:29 +0000830void CFGBuilder::addLocalScopeAndDtors(Stmt *S) {
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000831 if (!BuildOpts.AddImplicitDtors)
832 return;
833
834 LocalScope::const_iterator scopeBeginPos = ScopePos;
Zhongxing Xu02acdfa2010-10-01 03:00:16 +0000835 addLocalScopeForStmt(S);
Marcin Swiderski239a7c42010-09-30 23:05:00 +0000836 addAutomaticObjDtors(ScopePos, scopeBeginPos, S);
837}
838
Marcin Swiderski53de1342010-09-30 22:54:37 +0000839/// prependAutomaticObjDtorsWithTerminator - Prepend destructor CFGElements for
840/// variables with automatic storage duration to CFGBlock's elements vector.
841/// Elements will be prepended to physical beginning of the vector which
842/// happens to be logical end. Use blocks terminator as statement that specifies
843/// destructors call site.
Chandler Carruthc8cfc742011-09-13 06:09:01 +0000844/// FIXME: This mechanism for adding automatic destructors doesn't handle
845/// no-return destructors properly.
Ted Kremenek9c378f72011-08-12 23:37:29 +0000846void CFGBuilder::prependAutomaticObjDtorsWithTerminator(CFGBlock *Blk,
Marcin Swiderski53de1342010-09-30 22:54:37 +0000847 LocalScope::const_iterator B, LocalScope::const_iterator E) {
Chandler Carruthc8cfc742011-09-13 06:09:01 +0000848 BumpVectorContext &C = cfg->getBumpVectorContext();
849 CFGBlock::iterator InsertPos
850 = Blk->beginAutomaticObjDtorsInsert(Blk->end(), B.distance(E), C);
851 for (LocalScope::const_iterator I = B; I != E; ++I)
852 InsertPos = Blk->insertAutomaticObjDtor(InsertPos, *I,
853 Blk->getTerminator());
Marcin Swiderski53de1342010-09-30 22:54:37 +0000854}
855
Ted Kremenek4f880632009-07-17 22:18:43 +0000856/// Visit - Walk the subtree of a statement and add extra
Mike Stump6d9828c2009-07-17 01:31:16 +0000857/// blocks for ternary operators, &&, and ||. We also process "," and
858/// DeclStmts (which may contain nested control-flow).
Ted Kremenek9c378f72011-08-12 23:37:29 +0000859CFGBlock *CFGBuilder::Visit(Stmt * S, AddStmtChoice asc) {
Ted Kremenekf42e3372010-04-30 22:25:53 +0000860 if (!S) {
861 badCFG = true;
862 return 0;
863 }
Jordy Roseac73ea82011-06-10 08:49:37 +0000864
865 if (Expr *E = dyn_cast<Expr>(S))
866 S = E->IgnoreParens();
867
Ted Kremenek4f880632009-07-17 22:18:43 +0000868 switch (S->getStmtClass()) {
869 default:
Ted Kremenek852274d2009-12-16 03:18:58 +0000870 return VisitStmt(S, asc);
Ted Kremenek4f880632009-07-17 22:18:43 +0000871
872 case Stmt::AddrLabelExprClass:
Ted Kremenek852274d2009-12-16 03:18:58 +0000873 return VisitAddrLabelExpr(cast<AddrLabelExpr>(S), asc);
Mike Stump1eb44332009-09-09 15:08:12 +0000874
John McCall56ca35d2011-02-17 10:25:35 +0000875 case Stmt::BinaryConditionalOperatorClass:
876 return VisitConditionalOperator(cast<BinaryConditionalOperator>(S), asc);
877
Ted Kremenek4f880632009-07-17 22:18:43 +0000878 case Stmt::BinaryOperatorClass:
Ted Kremenek852274d2009-12-16 03:18:58 +0000879 return VisitBinaryOperator(cast<BinaryOperator>(S), asc);
Mike Stump1eb44332009-09-09 15:08:12 +0000880
Ted Kremenek4f880632009-07-17 22:18:43 +0000881 case Stmt::BlockExprClass:
Ted Kremenek852274d2009-12-16 03:18:58 +0000882 return VisitBlockExpr(cast<BlockExpr>(S), asc);
Ted Kremenek4f880632009-07-17 22:18:43 +0000883
Ted Kremenek4f880632009-07-17 22:18:43 +0000884 case Stmt::BreakStmtClass:
885 return VisitBreakStmt(cast<BreakStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000886
Ted Kremenek4f880632009-07-17 22:18:43 +0000887 case Stmt::CallExprClass:
Ted Kremeneka427f1d2010-08-31 18:47:34 +0000888 case Stmt::CXXOperatorCallExprClass:
John McCall1de85332011-05-11 07:19:11 +0000889 case Stmt::CXXMemberCallExprClass:
Ted Kremenek852274d2009-12-16 03:18:58 +0000890 return VisitCallExpr(cast<CallExpr>(S), asc);
Mike Stump1eb44332009-09-09 15:08:12 +0000891
Ted Kremenek4f880632009-07-17 22:18:43 +0000892 case Stmt::CaseStmtClass:
893 return VisitCaseStmt(cast<CaseStmt>(S));
894
895 case Stmt::ChooseExprClass:
Ted Kremenek852274d2009-12-16 03:18:58 +0000896 return VisitChooseExpr(cast<ChooseExpr>(S), asc);
Mike Stump1eb44332009-09-09 15:08:12 +0000897
Ted Kremenek4f880632009-07-17 22:18:43 +0000898 case Stmt::CompoundStmtClass:
899 return VisitCompoundStmt(cast<CompoundStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000900
Ted Kremenek4f880632009-07-17 22:18:43 +0000901 case Stmt::ConditionalOperatorClass:
Ted Kremenek852274d2009-12-16 03:18:58 +0000902 return VisitConditionalOperator(cast<ConditionalOperator>(S), asc);
Mike Stump1eb44332009-09-09 15:08:12 +0000903
Ted Kremenek4f880632009-07-17 22:18:43 +0000904 case Stmt::ContinueStmtClass:
905 return VisitContinueStmt(cast<ContinueStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000906
Ted Kremenek021c8af2010-01-19 20:40:33 +0000907 case Stmt::CXXCatchStmtClass:
908 return VisitCXXCatchStmt(cast<CXXCatchStmt>(S));
909
John McCall4765fa02010-12-06 08:20:24 +0000910 case Stmt::ExprWithCleanupsClass:
911 return VisitExprWithCleanups(cast<ExprWithCleanups>(S), asc);
Ted Kremenek47e331e2010-08-28 00:19:02 +0000912
Zhongxing Xua725ed42010-11-01 13:04:58 +0000913 case Stmt::CXXBindTemporaryExprClass:
914 return VisitCXXBindTemporaryExpr(cast<CXXBindTemporaryExpr>(S), asc);
915
Zhongxing Xu81bc7d02010-11-01 06:46:05 +0000916 case Stmt::CXXConstructExprClass:
917 return VisitCXXConstructExpr(cast<CXXConstructExpr>(S), asc);
918
Zhongxing Xua725ed42010-11-01 13:04:58 +0000919 case Stmt::CXXFunctionalCastExprClass:
920 return VisitCXXFunctionalCastExpr(cast<CXXFunctionalCastExpr>(S), asc);
921
Zhongxing Xu81bc7d02010-11-01 06:46:05 +0000922 case Stmt::CXXTemporaryObjectExprClass:
923 return VisitCXXTemporaryObjectExpr(cast<CXXTemporaryObjectExpr>(S), asc);
924
Ted Kremenek021c8af2010-01-19 20:40:33 +0000925 case Stmt::CXXThrowExprClass:
926 return VisitCXXThrowExpr(cast<CXXThrowExpr>(S));
Ted Kremenekad5a8942010-08-02 23:46:59 +0000927
Ted Kremenek021c8af2010-01-19 20:40:33 +0000928 case Stmt::CXXTryStmtClass:
929 return VisitCXXTryStmt(cast<CXXTryStmt>(S));
Ted Kremenekad5a8942010-08-02 23:46:59 +0000930
Richard Smithad762fc2011-04-14 22:09:26 +0000931 case Stmt::CXXForRangeStmtClass:
932 return VisitCXXForRangeStmt(cast<CXXForRangeStmt>(S));
933
Ted Kremenek4f880632009-07-17 22:18:43 +0000934 case Stmt::DeclStmtClass:
935 return VisitDeclStmt(cast<DeclStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000936
Ted Kremenek4f880632009-07-17 22:18:43 +0000937 case Stmt::DefaultStmtClass:
938 return VisitDefaultStmt(cast<DefaultStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000939
Ted Kremenek4f880632009-07-17 22:18:43 +0000940 case Stmt::DoStmtClass:
941 return VisitDoStmt(cast<DoStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000942
Ted Kremenek4f880632009-07-17 22:18:43 +0000943 case Stmt::ForStmtClass:
944 return VisitForStmt(cast<ForStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000945
Ted Kremenek4f880632009-07-17 22:18:43 +0000946 case Stmt::GotoStmtClass:
947 return VisitGotoStmt(cast<GotoStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000948
Ted Kremenek4f880632009-07-17 22:18:43 +0000949 case Stmt::IfStmtClass:
950 return VisitIfStmt(cast<IfStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000951
Ted Kremenek892697d2010-12-16 07:46:53 +0000952 case Stmt::ImplicitCastExprClass:
953 return VisitImplicitCastExpr(cast<ImplicitCastExpr>(S), asc);
Zhongxing Xua725ed42010-11-01 13:04:58 +0000954
Ted Kremenek4f880632009-07-17 22:18:43 +0000955 case Stmt::IndirectGotoStmtClass:
956 return VisitIndirectGotoStmt(cast<IndirectGotoStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000957
Ted Kremenek4f880632009-07-17 22:18:43 +0000958 case Stmt::LabelStmtClass:
959 return VisitLabelStmt(cast<LabelStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000960
Ted Kremenek115c1b92010-04-11 17:02:10 +0000961 case Stmt::MemberExprClass:
962 return VisitMemberExpr(cast<MemberExpr>(S), asc);
963
Ted Kremenek6a9065a2011-11-05 00:10:15 +0000964 case Stmt::NullStmtClass:
965 return Block;
966
Ted Kremenek4f880632009-07-17 22:18:43 +0000967 case Stmt::ObjCAtCatchStmtClass:
Mike Stump1eb44332009-09-09 15:08:12 +0000968 return VisitObjCAtCatchStmt(cast<ObjCAtCatchStmt>(S));
969
Ted Kremenek4f880632009-07-17 22:18:43 +0000970 case Stmt::ObjCAtSynchronizedStmtClass:
971 return VisitObjCAtSynchronizedStmt(cast<ObjCAtSynchronizedStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000972
Ted Kremenek4f880632009-07-17 22:18:43 +0000973 case Stmt::ObjCAtThrowStmtClass:
974 return VisitObjCAtThrowStmt(cast<ObjCAtThrowStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000975
Ted Kremenek4f880632009-07-17 22:18:43 +0000976 case Stmt::ObjCAtTryStmtClass:
977 return VisitObjCAtTryStmt(cast<ObjCAtTryStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000978
Ted Kremenek4f880632009-07-17 22:18:43 +0000979 case Stmt::ObjCForCollectionStmtClass:
980 return VisitObjCForCollectionStmt(cast<ObjCForCollectionStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000981
Ted Kremenek6a9065a2011-11-05 00:10:15 +0000982 case Stmt::OpaqueValueExprClass:
Ted Kremenek4f880632009-07-17 22:18:43 +0000983 return Block;
Mike Stump1eb44332009-09-09 15:08:12 +0000984
John McCall4b9c2d22011-11-06 09:01:30 +0000985 case Stmt::PseudoObjectExprClass:
986 return VisitPseudoObjectExpr(cast<PseudoObjectExpr>(S));
987
Ted Kremenek4f880632009-07-17 22:18:43 +0000988 case Stmt::ReturnStmtClass:
989 return VisitReturnStmt(cast<ReturnStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +0000990
Peter Collingbournef4e3cfb2011-03-11 19:24:49 +0000991 case Stmt::UnaryExprOrTypeTraitExprClass:
992 return VisitUnaryExprOrTypeTraitExpr(cast<UnaryExprOrTypeTraitExpr>(S),
993 asc);
Mike Stump1eb44332009-09-09 15:08:12 +0000994
Ted Kremenek4f880632009-07-17 22:18:43 +0000995 case Stmt::StmtExprClass:
Ted Kremenek852274d2009-12-16 03:18:58 +0000996 return VisitStmtExpr(cast<StmtExpr>(S), asc);
Mike Stump1eb44332009-09-09 15:08:12 +0000997
Ted Kremenek4f880632009-07-17 22:18:43 +0000998 case Stmt::SwitchStmtClass:
999 return VisitSwitchStmt(cast<SwitchStmt>(S));
Mike Stump1eb44332009-09-09 15:08:12 +00001000
Zhanyong Wan99cae5b2010-11-22 08:45:56 +00001001 case Stmt::UnaryOperatorClass:
1002 return VisitUnaryOperator(cast<UnaryOperator>(S), asc);
1003
Ted Kremenek4f880632009-07-17 22:18:43 +00001004 case Stmt::WhileStmtClass:
1005 return VisitWhileStmt(cast<WhileStmt>(S));
1006 }
1007}
Mike Stump1eb44332009-09-09 15:08:12 +00001008
Ted Kremenek852274d2009-12-16 03:18:58 +00001009CFGBlock *CFGBuilder::VisitStmt(Stmt *S, AddStmtChoice asc) {
Ted Kremenek3179a452011-03-10 01:14:11 +00001010 if (asc.alwaysAdd(*this, S)) {
Ted Kremenek4f880632009-07-17 22:18:43 +00001011 autoCreateBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00001012 appendStmt(Block, S);
Mike Stump6d9828c2009-07-17 01:31:16 +00001013 }
Mike Stump1eb44332009-09-09 15:08:12 +00001014
Ted Kremenek4f880632009-07-17 22:18:43 +00001015 return VisitChildren(S);
Ted Kremenek9da2fb72007-08-27 21:27:44 +00001016}
Mike Stump6d9828c2009-07-17 01:31:16 +00001017
Ted Kremenek4f880632009-07-17 22:18:43 +00001018/// VisitChildren - Visit the children of a Stmt.
Ted Kremenek9c378f72011-08-12 23:37:29 +00001019CFGBlock *CFGBuilder::VisitChildren(Stmt *Terminator) {
Ted Kremenek6b12da92011-02-21 22:11:26 +00001020 CFGBlock *lastBlock = Block;
1021 for (Stmt::child_range I = Terminator->children(); I; ++I)
1022 if (Stmt *child = *I)
1023 if (CFGBlock *b = Visit(child))
1024 lastBlock = b;
1025
1026 return lastBlock;
Ted Kremenek9da2fb72007-08-27 21:27:44 +00001027}
Mike Stump1eb44332009-09-09 15:08:12 +00001028
Ted Kremenek852274d2009-12-16 03:18:58 +00001029CFGBlock *CFGBuilder::VisitAddrLabelExpr(AddrLabelExpr *A,
1030 AddStmtChoice asc) {
Ted Kremenek4f880632009-07-17 22:18:43 +00001031 AddressTakenLabels.insert(A->getLabel());
Ted Kremenek9da2fb72007-08-27 21:27:44 +00001032
Ted Kremenek3179a452011-03-10 01:14:11 +00001033 if (asc.alwaysAdd(*this, A)) {
Ted Kremenek4f880632009-07-17 22:18:43 +00001034 autoCreateBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00001035 appendStmt(Block, A);
Ted Kremenek4f880632009-07-17 22:18:43 +00001036 }
Ted Kremenek49af7cb2007-08-27 19:46:09 +00001037
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001038 return Block;
1039}
Mike Stump1eb44332009-09-09 15:08:12 +00001040
Zhanyong Wan99cae5b2010-11-22 08:45:56 +00001041CFGBlock *CFGBuilder::VisitUnaryOperator(UnaryOperator *U,
Ted Kremenek892697d2010-12-16 07:46:53 +00001042 AddStmtChoice asc) {
Ted Kremenek3179a452011-03-10 01:14:11 +00001043 if (asc.alwaysAdd(*this, U)) {
Zhanyong Wan99cae5b2010-11-22 08:45:56 +00001044 autoCreateBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00001045 appendStmt(Block, U);
Zhanyong Wan99cae5b2010-11-22 08:45:56 +00001046 }
1047
Ted Kremenek892697d2010-12-16 07:46:53 +00001048 return Visit(U->getSubExpr(), AddStmtChoice());
Zhanyong Wan99cae5b2010-11-22 08:45:56 +00001049}
1050
Ted Kremenek852274d2009-12-16 03:18:58 +00001051CFGBlock *CFGBuilder::VisitBinaryOperator(BinaryOperator *B,
1052 AddStmtChoice asc) {
Ted Kremenek4f880632009-07-17 22:18:43 +00001053 if (B->isLogicalOp()) { // && or ||
Ted Kremenek9c378f72011-08-12 23:37:29 +00001054 CFGBlock *ConfluenceBlock = Block ? Block : createBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00001055 appendStmt(ConfluenceBlock, B);
Mike Stump1eb44332009-09-09 15:08:12 +00001056
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001057 if (badCFG)
Ted Kremenek4f880632009-07-17 22:18:43 +00001058 return 0;
Mike Stump1eb44332009-09-09 15:08:12 +00001059
Ted Kremenek4f880632009-07-17 22:18:43 +00001060 // create the block evaluating the LHS
Ted Kremenek9c378f72011-08-12 23:37:29 +00001061 CFGBlock *LHSBlock = createBlock(false);
Ted Kremenek4f880632009-07-17 22:18:43 +00001062 LHSBlock->setTerminator(B);
Mike Stump1eb44332009-09-09 15:08:12 +00001063
Ted Kremenek4f880632009-07-17 22:18:43 +00001064 // create the block evaluating the RHS
1065 Succ = ConfluenceBlock;
1066 Block = NULL;
Ted Kremenek9c378f72011-08-12 23:37:29 +00001067 CFGBlock *RHSBlock = addStmt(B->getRHS());
Ted Kremenek862b24f2010-04-29 01:10:26 +00001068
1069 if (RHSBlock) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001070 if (badCFG)
Ted Kremenek862b24f2010-04-29 01:10:26 +00001071 return 0;
Zhanyong Wan36f327c2010-11-22 19:32:14 +00001072 } else {
Ted Kremenek862b24f2010-04-29 01:10:26 +00001073 // Create an empty block for cases where the RHS doesn't require
1074 // any explicit statements in the CFG.
1075 RHSBlock = createBlock();
1076 }
Mike Stump1eb44332009-09-09 15:08:12 +00001077
Mike Stump00998a02009-07-23 23:25:26 +00001078 // See if this is a known constant.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001079 TryResult KnownVal = tryEvaluateBool(B->getLHS());
John McCall2de56d12010-08-25 11:45:40 +00001080 if (KnownVal.isKnown() && (B->getOpcode() == BO_LOr))
Ted Kremenek941fde82009-07-24 04:47:11 +00001081 KnownVal.negate();
Mike Stump00998a02009-07-23 23:25:26 +00001082
Ted Kremenek4f880632009-07-17 22:18:43 +00001083 // Now link the LHSBlock with RHSBlock.
John McCall2de56d12010-08-25 11:45:40 +00001084 if (B->getOpcode() == BO_LOr) {
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001085 addSuccessor(LHSBlock, KnownVal.isTrue() ? NULL : ConfluenceBlock);
1086 addSuccessor(LHSBlock, KnownVal.isFalse() ? NULL : RHSBlock);
Mike Stump1eb44332009-09-09 15:08:12 +00001087 } else {
John McCall2de56d12010-08-25 11:45:40 +00001088 assert(B->getOpcode() == BO_LAnd);
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001089 addSuccessor(LHSBlock, KnownVal.isFalse() ? NULL : RHSBlock);
1090 addSuccessor(LHSBlock, KnownVal.isTrue() ? NULL : ConfluenceBlock);
Ted Kremenek4f880632009-07-17 22:18:43 +00001091 }
Mike Stump1eb44332009-09-09 15:08:12 +00001092
Ted Kremenek4f880632009-07-17 22:18:43 +00001093 // Generate the blocks for evaluating the LHS.
1094 Block = LHSBlock;
1095 return addStmt(B->getLHS());
Mike Stump1eb44332009-09-09 15:08:12 +00001096 }
Zhanyong Wan36f327c2010-11-22 19:32:14 +00001097
1098 if (B->getOpcode() == BO_Comma) { // ,
Ted Kremenek6dc534e2009-07-17 22:57:50 +00001099 autoCreateBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00001100 appendStmt(Block, B);
Ted Kremenek4f880632009-07-17 22:18:43 +00001101 addStmt(B->getRHS());
1102 return addStmt(B->getLHS());
1103 }
Zhanyong Wan36f327c2010-11-22 19:32:14 +00001104
1105 if (B->isAssignmentOp()) {
Ted Kremenek3179a452011-03-10 01:14:11 +00001106 if (asc.alwaysAdd(*this, B)) {
Zhongxing Xufc61d942010-06-03 06:23:18 +00001107 autoCreateBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00001108 appendStmt(Block, B);
Zhongxing Xufc61d942010-06-03 06:23:18 +00001109 }
Ted Kremenek892697d2010-12-16 07:46:53 +00001110 Visit(B->getLHS());
Marcin Swiderskie1667192010-10-24 08:21:40 +00001111 return Visit(B->getRHS());
Zhongxing Xufc61d942010-06-03 06:23:18 +00001112 }
Mike Stump1eb44332009-09-09 15:08:12 +00001113
Ted Kremenek3179a452011-03-10 01:14:11 +00001114 if (asc.alwaysAdd(*this, B)) {
Marcin Swiderskie1667192010-10-24 08:21:40 +00001115 autoCreateBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00001116 appendStmt(Block, B);
Marcin Swiderskie1667192010-10-24 08:21:40 +00001117 }
1118
Zhongxing Xua1898dd2010-10-27 03:23:10 +00001119 CFGBlock *RBlock = Visit(B->getRHS());
1120 CFGBlock *LBlock = Visit(B->getLHS());
1121 // If visiting RHS causes us to finish 'Block', e.g. the RHS is a StmtExpr
1122 // containing a DoStmt, and the LHS doesn't create a new block, then we should
1123 // return RBlock. Otherwise we'll incorrectly return NULL.
1124 return (LBlock ? LBlock : RBlock);
Ted Kremenek4f880632009-07-17 22:18:43 +00001125}
1126
Ted Kremenek852274d2009-12-16 03:18:58 +00001127CFGBlock *CFGBuilder::VisitBlockExpr(BlockExpr *E, AddStmtChoice asc) {
Ted Kremenek3179a452011-03-10 01:14:11 +00001128 if (asc.alwaysAdd(*this, E)) {
Ted Kremenek721903e2009-11-25 01:34:30 +00001129 autoCreateBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00001130 appendStmt(Block, E);
Ted Kremenek721903e2009-11-25 01:34:30 +00001131 }
1132 return Block;
Ted Kremenek4f880632009-07-17 22:18:43 +00001133}
1134
Ted Kremenek4f880632009-07-17 22:18:43 +00001135CFGBlock *CFGBuilder::VisitBreakStmt(BreakStmt *B) {
1136 // "break" is a control-flow statement. Thus we stop processing the current
1137 // block.
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001138 if (badCFG)
1139 return 0;
Mike Stump1eb44332009-09-09 15:08:12 +00001140
Ted Kremenek4f880632009-07-17 22:18:43 +00001141 // Now create a new block that ends with the break statement.
1142 Block = createBlock(false);
1143 Block->setTerminator(B);
Mike Stump1eb44332009-09-09 15:08:12 +00001144
Ted Kremenek4f880632009-07-17 22:18:43 +00001145 // If there is no target for the break, then we are looking at an incomplete
1146 // AST. This means that the CFG cannot be constructed.
Ted Kremenek9ce52702011-01-07 19:37:16 +00001147 if (BreakJumpTarget.block) {
1148 addAutomaticObjDtors(ScopePos, BreakJumpTarget.scopePosition, B);
1149 addSuccessor(Block, BreakJumpTarget.block);
Marcin Swiderskif1308c72010-09-25 11:05:21 +00001150 } else
Ted Kremenek4f880632009-07-17 22:18:43 +00001151 badCFG = true;
Mike Stump1eb44332009-09-09 15:08:12 +00001152
1153
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001154 return Block;
1155}
Mike Stump1eb44332009-09-09 15:08:12 +00001156
Sebastian Redl8026f6d2011-03-13 17:09:40 +00001157static bool CanThrow(Expr *E, ASTContext &Ctx) {
Mike Stump4c45aa12010-01-21 15:20:48 +00001158 QualType Ty = E->getType();
1159 if (Ty->isFunctionPointerType())
1160 Ty = Ty->getAs<PointerType>()->getPointeeType();
1161 else if (Ty->isBlockPointerType())
1162 Ty = Ty->getAs<BlockPointerType>()->getPointeeType();
Ted Kremenekad5a8942010-08-02 23:46:59 +00001163
Mike Stump4c45aa12010-01-21 15:20:48 +00001164 const FunctionType *FT = Ty->getAs<FunctionType>();
1165 if (FT) {
1166 if (const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(FT))
Sebastian Redl8026f6d2011-03-13 17:09:40 +00001167 if (Proto->isNothrow(Ctx))
Mike Stump4c45aa12010-01-21 15:20:48 +00001168 return false;
1169 }
1170 return true;
1171}
1172
Ted Kremenek852274d2009-12-16 03:18:58 +00001173CFGBlock *CFGBuilder::VisitCallExpr(CallExpr *C, AddStmtChoice asc) {
John McCall1de85332011-05-11 07:19:11 +00001174 // Compute the callee type.
1175 QualType calleeType = C->getCallee()->getType();
1176 if (calleeType == Context->BoundMemberTy) {
1177 QualType boundType = Expr::findBoundMemberType(C->getCallee());
1178
1179 // We should only get a null bound type if processing a dependent
1180 // CFG. Recover by assuming nothing.
1181 if (!boundType.isNull()) calleeType = boundType;
Ted Kremenek4f880632009-07-17 22:18:43 +00001182 }
Mike Stump24556362009-07-25 21:26:53 +00001183
John McCall1de85332011-05-11 07:19:11 +00001184 // If this is a call to a no-return function, this stops the block here.
1185 bool NoReturn = getFunctionExtInfo(*calleeType).getNoReturn();
1186
Mike Stump4c45aa12010-01-21 15:20:48 +00001187 bool AddEHEdge = false;
Mike Stump079bd722010-01-19 22:00:14 +00001188
1189 // Languages without exceptions are assumed to not throw.
Anders Carlsson7a178512011-02-28 00:33:03 +00001190 if (Context->getLangOptions().Exceptions) {
Ted Kremenek6c52c782010-09-14 23:41:16 +00001191 if (BuildOpts.AddEHEdges)
Mike Stump4c45aa12010-01-21 15:20:48 +00001192 AddEHEdge = true;
Mike Stump079bd722010-01-19 22:00:14 +00001193 }
1194
1195 if (FunctionDecl *FD = C->getDirectCallee()) {
Mike Stump24556362009-07-25 21:26:53 +00001196 if (FD->hasAttr<NoReturnAttr>())
1197 NoReturn = true;
Mike Stump079bd722010-01-19 22:00:14 +00001198 if (FD->hasAttr<NoThrowAttr>())
Mike Stump4c45aa12010-01-21 15:20:48 +00001199 AddEHEdge = false;
Mike Stump079bd722010-01-19 22:00:14 +00001200 }
Mike Stump24556362009-07-25 21:26:53 +00001201
Sebastian Redl8026f6d2011-03-13 17:09:40 +00001202 if (!CanThrow(C->getCallee(), *Context))
Mike Stump4c45aa12010-01-21 15:20:48 +00001203 AddEHEdge = false;
1204
Zhanyong Wan94a3dcf2010-11-24 03:28:53 +00001205 if (!NoReturn && !AddEHEdge)
1206 return VisitStmt(C, asc.withAlwaysAdd(true));
Mike Stump1eb44332009-09-09 15:08:12 +00001207
Mike Stump079bd722010-01-19 22:00:14 +00001208 if (Block) {
1209 Succ = Block;
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001210 if (badCFG)
Mike Stump079bd722010-01-19 22:00:14 +00001211 return 0;
1212 }
Mike Stump1eb44332009-09-09 15:08:12 +00001213
Chandler Carruthdba3fb52011-09-13 09:13:49 +00001214 if (NoReturn)
1215 Block = createNoReturnBlock();
1216 else
1217 Block = createBlock();
1218
Ted Kremenek247e9662011-03-10 01:14:08 +00001219 appendStmt(Block, C);
Mike Stump24556362009-07-25 21:26:53 +00001220
Mike Stump4c45aa12010-01-21 15:20:48 +00001221 if (AddEHEdge) {
Mike Stump079bd722010-01-19 22:00:14 +00001222 // Add exceptional edges.
1223 if (TryTerminatedBlock)
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001224 addSuccessor(Block, TryTerminatedBlock);
Mike Stump079bd722010-01-19 22:00:14 +00001225 else
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001226 addSuccessor(Block, &cfg->getExit());
Mike Stump079bd722010-01-19 22:00:14 +00001227 }
Mike Stump1eb44332009-09-09 15:08:12 +00001228
Mike Stump24556362009-07-25 21:26:53 +00001229 return VisitChildren(C);
Ted Kremenek4f880632009-07-17 22:18:43 +00001230}
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001231
Ted Kremenek852274d2009-12-16 03:18:58 +00001232CFGBlock *CFGBuilder::VisitChooseExpr(ChooseExpr *C,
1233 AddStmtChoice asc) {
Ted Kremenek9c378f72011-08-12 23:37:29 +00001234 CFGBlock *ConfluenceBlock = Block ? Block : createBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00001235 appendStmt(ConfluenceBlock, C);
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001236 if (badCFG)
Ted Kremenek3fc8ef52009-07-17 18:20:32 +00001237 return 0;
Mike Stump1eb44332009-09-09 15:08:12 +00001238
Zhanyong Wan94a3dcf2010-11-24 03:28:53 +00001239 AddStmtChoice alwaysAdd = asc.withAlwaysAdd(true);
Ted Kremenek3fc8ef52009-07-17 18:20:32 +00001240 Succ = ConfluenceBlock;
1241 Block = NULL;
Ted Kremenek9c378f72011-08-12 23:37:29 +00001242 CFGBlock *LHSBlock = Visit(C->getLHS(), alwaysAdd);
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001243 if (badCFG)
Ted Kremenek3fc8ef52009-07-17 18:20:32 +00001244 return 0;
Mike Stump1eb44332009-09-09 15:08:12 +00001245
Ted Kremenek3fc8ef52009-07-17 18:20:32 +00001246 Succ = ConfluenceBlock;
1247 Block = NULL;
Ted Kremenek9c378f72011-08-12 23:37:29 +00001248 CFGBlock *RHSBlock = Visit(C->getRHS(), alwaysAdd);
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001249 if (badCFG)
Ted Kremenek3fc8ef52009-07-17 18:20:32 +00001250 return 0;
Mike Stump1eb44332009-09-09 15:08:12 +00001251
Ted Kremenek3fc8ef52009-07-17 18:20:32 +00001252 Block = createBlock(false);
Mike Stump00998a02009-07-23 23:25:26 +00001253 // See if this is a known constant.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001254 const TryResult& KnownVal = tryEvaluateBool(C->getCond());
1255 addSuccessor(Block, KnownVal.isFalse() ? NULL : LHSBlock);
1256 addSuccessor(Block, KnownVal.isTrue() ? NULL : RHSBlock);
Ted Kremenek3fc8ef52009-07-17 18:20:32 +00001257 Block->setTerminator(C);
Mike Stump1eb44332009-09-09 15:08:12 +00001258 return addStmt(C->getCond());
Ted Kremenek3fc8ef52009-07-17 18:20:32 +00001259}
Mike Stump1eb44332009-09-09 15:08:12 +00001260
1261
Ted Kremenek9c378f72011-08-12 23:37:29 +00001262CFGBlock *CFGBuilder::VisitCompoundStmt(CompoundStmt *C) {
Marcin Swiderskifcb72ac2010-10-01 00:23:17 +00001263 addLocalScopeAndDtors(C);
Ted Kremenek9c378f72011-08-12 23:37:29 +00001264 CFGBlock *LastBlock = Block;
Ted Kremenek4f880632009-07-17 22:18:43 +00001265
1266 for (CompoundStmt::reverse_body_iterator I=C->body_rbegin(), E=C->body_rend();
1267 I != E; ++I ) {
Ted Kremenek334c1952010-08-17 21:00:06 +00001268 // If we hit a segment of code just containing ';' (NullStmts), we can
1269 // get a null block back. In such cases, just use the LastBlock
1270 if (CFGBlock *newBlock = addStmt(*I))
1271 LastBlock = newBlock;
Mike Stump1eb44332009-09-09 15:08:12 +00001272
Ted Kremeneke8d6d2b2009-08-27 23:16:26 +00001273 if (badCFG)
1274 return NULL;
Mike Stump1eb44332009-09-09 15:08:12 +00001275 }
Mike Stump079bd722010-01-19 22:00:14 +00001276
Ted Kremenek4f880632009-07-17 22:18:43 +00001277 return LastBlock;
1278}
Mike Stump1eb44332009-09-09 15:08:12 +00001279
John McCall56ca35d2011-02-17 10:25:35 +00001280CFGBlock *CFGBuilder::VisitConditionalOperator(AbstractConditionalOperator *C,
Ted Kremenek852274d2009-12-16 03:18:58 +00001281 AddStmtChoice asc) {
John McCall56ca35d2011-02-17 10:25:35 +00001282 const BinaryConditionalOperator *BCO = dyn_cast<BinaryConditionalOperator>(C);
1283 const OpaqueValueExpr *opaqueValue = (BCO ? BCO->getOpaqueValue() : NULL);
1284
Ted Kremenekf34bb2e2009-07-17 18:15:54 +00001285 // Create the confluence block that will "merge" the results of the ternary
1286 // expression.
Ted Kremenek9c378f72011-08-12 23:37:29 +00001287 CFGBlock *ConfluenceBlock = Block ? Block : createBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00001288 appendStmt(ConfluenceBlock, C);
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001289 if (badCFG)
Ted Kremenekf34bb2e2009-07-17 18:15:54 +00001290 return 0;
Mike Stump1eb44332009-09-09 15:08:12 +00001291
Zhanyong Wan94a3dcf2010-11-24 03:28:53 +00001292 AddStmtChoice alwaysAdd = asc.withAlwaysAdd(true);
Ted Kremenek115c1b92010-04-11 17:02:10 +00001293
Ted Kremenekf34bb2e2009-07-17 18:15:54 +00001294 // Create a block for the LHS expression if there is an LHS expression. A
1295 // GCC extension allows LHS to be NULL, causing the condition to be the
1296 // value that is returned instead.
1297 // e.g: x ?: y is shorthand for: x ? x : y;
1298 Succ = ConfluenceBlock;
1299 Block = NULL;
Ted Kremenek9c378f72011-08-12 23:37:29 +00001300 CFGBlock *LHSBlock = 0;
John McCall56ca35d2011-02-17 10:25:35 +00001301 const Expr *trueExpr = C->getTrueExpr();
1302 if (trueExpr != opaqueValue) {
1303 LHSBlock = Visit(C->getTrueExpr(), alwaysAdd);
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001304 if (badCFG)
Ted Kremenekf34bb2e2009-07-17 18:15:54 +00001305 return 0;
1306 Block = NULL;
1307 }
Ted Kremenekf226d182011-02-24 03:09:15 +00001308 else
1309 LHSBlock = ConfluenceBlock;
Mike Stump1eb44332009-09-09 15:08:12 +00001310
Ted Kremenekf34bb2e2009-07-17 18:15:54 +00001311 // Create the block for the RHS expression.
1312 Succ = ConfluenceBlock;
Ted Kremenek9c378f72011-08-12 23:37:29 +00001313 CFGBlock *RHSBlock = Visit(C->getFalseExpr(), alwaysAdd);
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001314 if (badCFG)
Ted Kremenekf34bb2e2009-07-17 18:15:54 +00001315 return 0;
Mike Stump1eb44332009-09-09 15:08:12 +00001316
Ted Kremenekf34bb2e2009-07-17 18:15:54 +00001317 // Create the block that will contain the condition.
1318 Block = createBlock(false);
Mike Stump1eb44332009-09-09 15:08:12 +00001319
Mike Stump00998a02009-07-23 23:25:26 +00001320 // See if this is a known constant.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001321 const TryResult& KnownVal = tryEvaluateBool(C->getCond());
Ted Kremenekf226d182011-02-24 03:09:15 +00001322 addSuccessor(Block, KnownVal.isFalse() ? NULL : LHSBlock);
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001323 addSuccessor(Block, KnownVal.isTrue() ? NULL : RHSBlock);
Ted Kremenekf34bb2e2009-07-17 18:15:54 +00001324 Block->setTerminator(C);
John McCall56ca35d2011-02-17 10:25:35 +00001325 Expr *condExpr = C->getCond();
John McCalld40baf62011-02-19 03:13:26 +00001326
Ted Kremenekf226d182011-02-24 03:09:15 +00001327 if (opaqueValue) {
1328 // Run the condition expression if it's not trivially expressed in
1329 // terms of the opaque value (or if there is no opaque value).
1330 if (condExpr != opaqueValue)
1331 addStmt(condExpr);
John McCalld40baf62011-02-19 03:13:26 +00001332
Ted Kremenekf226d182011-02-24 03:09:15 +00001333 // Before that, run the common subexpression if there was one.
1334 // At least one of this or the above will be run.
1335 return addStmt(BCO->getCommon());
1336 }
1337
1338 return addStmt(condExpr);
Ted Kremenekf34bb2e2009-07-17 18:15:54 +00001339}
1340
Ted Kremenek4f880632009-07-17 22:18:43 +00001341CFGBlock *CFGBuilder::VisitDeclStmt(DeclStmt *DS) {
Ted Kremenekbc869de2011-05-10 18:42:15 +00001342 // Check if the Decl is for an __label__. If so, elide it from the
1343 // CFG entirely.
1344 if (isa<LabelDecl>(*DS->decl_begin()))
1345 return Block;
1346
Ted Kremenek29c9e622011-05-24 20:41:31 +00001347 // This case also handles static_asserts.
Marcin Swiderski8599e762010-11-03 06:19:35 +00001348 if (DS->isSingleDecl())
1349 return VisitDeclSubExpr(DS);
Mike Stump1eb44332009-09-09 15:08:12 +00001350
Ted Kremenek4f880632009-07-17 22:18:43 +00001351 CFGBlock *B = 0;
Mike Stump1eb44332009-09-09 15:08:12 +00001352
Ted Kremenek4f880632009-07-17 22:18:43 +00001353 // FIXME: Add a reverse iterator for DeclStmt to avoid this extra copy.
Chris Lattner5f9e2722011-07-23 10:55:15 +00001354 typedef SmallVector<Decl*,10> BufTy;
Ted Kremenek4f880632009-07-17 22:18:43 +00001355 BufTy Buf(DS->decl_begin(), DS->decl_end());
Mike Stump1eb44332009-09-09 15:08:12 +00001356
Ted Kremenek4f880632009-07-17 22:18:43 +00001357 for (BufTy::reverse_iterator I = Buf.rbegin(), E = Buf.rend(); I != E; ++I) {
1358 // Get the alignment of the new DeclStmt, padding out to >=8 bytes.
1359 unsigned A = llvm::AlignOf<DeclStmt>::Alignment < 8
1360 ? 8 : llvm::AlignOf<DeclStmt>::Alignment;
Mike Stump1eb44332009-09-09 15:08:12 +00001361
Ted Kremenek4f880632009-07-17 22:18:43 +00001362 // Allocate the DeclStmt using the BumpPtrAllocator. It will get
1363 // automatically freed with the CFG.
1364 DeclGroupRef DG(*I);
1365 Decl *D = *I;
Mike Stump1eb44332009-09-09 15:08:12 +00001366 void *Mem = cfg->getAllocator().Allocate(sizeof(DeclStmt), A);
Ted Kremenek4f880632009-07-17 22:18:43 +00001367 DeclStmt *DSNew = new (Mem) DeclStmt(DG, D->getLocation(), GetEndLoc(D));
Mike Stump1eb44332009-09-09 15:08:12 +00001368
Ted Kremenek4f880632009-07-17 22:18:43 +00001369 // Append the fake DeclStmt to block.
Marcin Swiderski8599e762010-11-03 06:19:35 +00001370 B = VisitDeclSubExpr(DSNew);
Ted Kremenek4f880632009-07-17 22:18:43 +00001371 }
Mike Stump1eb44332009-09-09 15:08:12 +00001372
1373 return B;
Ted Kremenek4f880632009-07-17 22:18:43 +00001374}
Mike Stump1eb44332009-09-09 15:08:12 +00001375
Ted Kremenek4f880632009-07-17 22:18:43 +00001376/// VisitDeclSubExpr - Utility method to add block-level expressions for
Marcin Swiderski8599e762010-11-03 06:19:35 +00001377/// DeclStmts and initializers in them.
Ted Kremenek9c378f72011-08-12 23:37:29 +00001378CFGBlock *CFGBuilder::VisitDeclSubExpr(DeclStmt *DS) {
Marcin Swiderski8599e762010-11-03 06:19:35 +00001379 assert(DS->isSingleDecl() && "Can handle single declarations only.");
Ted Kremenek29c9e622011-05-24 20:41:31 +00001380 Decl *D = DS->getSingleDecl();
1381
1382 if (isa<StaticAssertDecl>(D)) {
1383 // static_asserts aren't added to the CFG because they do not impact
1384 // runtime semantics.
1385 return Block;
1386 }
1387
Marcin Swiderski8599e762010-11-03 06:19:35 +00001388 VarDecl *VD = dyn_cast<VarDecl>(DS->getSingleDecl());
Mike Stump1eb44332009-09-09 15:08:12 +00001389
Marcin Swiderski8599e762010-11-03 06:19:35 +00001390 if (!VD) {
1391 autoCreateBlock();
Ted Kremenek892697d2010-12-16 07:46:53 +00001392 appendStmt(Block, DS);
Ted Kremenek4f880632009-07-17 22:18:43 +00001393 return Block;
Marcin Swiderski8599e762010-11-03 06:19:35 +00001394 }
Mike Stump1eb44332009-09-09 15:08:12 +00001395
Marcin Swiderski8599e762010-11-03 06:19:35 +00001396 bool IsReference = false;
1397 bool HasTemporaries = false;
1398
1399 // Destructors of temporaries in initialization expression should be called
1400 // after initialization finishes.
Ted Kremenek4f880632009-07-17 22:18:43 +00001401 Expr *Init = VD->getInit();
Marcin Swiderski8599e762010-11-03 06:19:35 +00001402 if (Init) {
1403 IsReference = VD->getType()->isReferenceType();
John McCall4765fa02010-12-06 08:20:24 +00001404 HasTemporaries = isa<ExprWithCleanups>(Init);
Marcin Swiderski8599e762010-11-03 06:19:35 +00001405
1406 if (BuildOpts.AddImplicitDtors && HasTemporaries) {
1407 // Generate destructors for temporaries in initialization expression.
John McCall4765fa02010-12-06 08:20:24 +00001408 VisitForTemporaryDtors(cast<ExprWithCleanups>(Init)->getSubExpr(),
Marcin Swiderski8599e762010-11-03 06:19:35 +00001409 IsReference);
1410 }
1411 }
1412
1413 autoCreateBlock();
Ted Kremenek892697d2010-12-16 07:46:53 +00001414 appendStmt(Block, DS);
Mike Stump1eb44332009-09-09 15:08:12 +00001415
Ted Kremenek4f880632009-07-17 22:18:43 +00001416 if (Init) {
Marcin Swiderski8599e762010-11-03 06:19:35 +00001417 if (HasTemporaries)
1418 // For expression with temporaries go directly to subexpression to omit
1419 // generating destructors for the second time.
Ted Kremenek892697d2010-12-16 07:46:53 +00001420 Visit(cast<ExprWithCleanups>(Init)->getSubExpr());
Marcin Swiderski8599e762010-11-03 06:19:35 +00001421 else
Ted Kremenek892697d2010-12-16 07:46:53 +00001422 Visit(Init);
Ted Kremenek4f880632009-07-17 22:18:43 +00001423 }
Mike Stump1eb44332009-09-09 15:08:12 +00001424
Ted Kremenek4f880632009-07-17 22:18:43 +00001425 // If the type of VD is a VLA, then we must process its size expressions.
John McCallf4c73712011-01-19 06:33:43 +00001426 for (const VariableArrayType* VA = FindVA(VD->getType().getTypePtr());
1427 VA != 0; VA = FindVA(VA->getElementType().getTypePtr()))
Ted Kremenek4f880632009-07-17 22:18:43 +00001428 Block = addStmt(VA->getSizeExpr());
Mike Stump1eb44332009-09-09 15:08:12 +00001429
Marcin Swiderskifcb72ac2010-10-01 00:23:17 +00001430 // Remove variable from local scope.
1431 if (ScopePos && VD == *ScopePos)
1432 ++ScopePos;
1433
Ted Kremenek4f880632009-07-17 22:18:43 +00001434 return Block;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001435}
1436
Ted Kremenek9c378f72011-08-12 23:37:29 +00001437CFGBlock *CFGBuilder::VisitIfStmt(IfStmt *I) {
Mike Stump6d9828c2009-07-17 01:31:16 +00001438 // We may see an if statement in the middle of a basic block, or it may be the
1439 // first statement we are processing. In either case, we create a new basic
1440 // block. First, we create the blocks for the then...else statements, and
1441 // then we create the block containing the if statement. If we were in the
Ted Kremenek6c249722009-09-24 18:45:41 +00001442 // middle of a block, we stop processing that block. That block is then the
1443 // implicit successor for the "then" and "else" clauses.
Mike Stump6d9828c2009-07-17 01:31:16 +00001444
Marcin Swiderski04e046c2010-10-01 00:52:17 +00001445 // Save local scope position because in case of condition variable ScopePos
1446 // won't be restored when traversing AST.
1447 SaveAndRestore<LocalScope::const_iterator> save_scope_pos(ScopePos);
1448
1449 // Create local scope for possible condition variable.
1450 // Store scope position. Add implicit destructor.
Ted Kremenek9c378f72011-08-12 23:37:29 +00001451 if (VarDecl *VD = I->getConditionVariable()) {
Marcin Swiderski04e046c2010-10-01 00:52:17 +00001452 LocalScope::const_iterator BeginScopePos = ScopePos;
1453 addLocalScopeForVarDecl(VD);
1454 addAutomaticObjDtors(ScopePos, BeginScopePos, I);
1455 }
1456
Chris Lattnerfc8f0e12011-04-15 05:22:18 +00001457 // The block we were processing is now finished. Make it the successor
Mike Stump6d9828c2009-07-17 01:31:16 +00001458 // block.
1459 if (Block) {
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001460 Succ = Block;
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001461 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001462 return 0;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001463 }
Mike Stump6d9828c2009-07-17 01:31:16 +00001464
Ted Kremenekb6f1d782009-07-17 18:04:55 +00001465 // Process the false branch.
Ted Kremenek9c378f72011-08-12 23:37:29 +00001466 CFGBlock *ElseBlock = Succ;
Mike Stump6d9828c2009-07-17 01:31:16 +00001467
Ted Kremenek9c378f72011-08-12 23:37:29 +00001468 if (Stmt *Else = I->getElse()) {
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001469 SaveAndRestore<CFGBlock*> sv(Succ);
Mike Stump6d9828c2009-07-17 01:31:16 +00001470
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001471 // NULL out Block so that the recursive call to Visit will
Mike Stump6d9828c2009-07-17 01:31:16 +00001472 // create a new basic block.
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001473 Block = NULL;
Marcin Swiderski04e046c2010-10-01 00:52:17 +00001474
1475 // If branch is not a compound statement create implicit scope
1476 // and add destructors.
1477 if (!isa<CompoundStmt>(Else))
1478 addLocalScopeAndDtors(Else);
1479
Ted Kremenek4f880632009-07-17 22:18:43 +00001480 ElseBlock = addStmt(Else);
Mike Stump6d9828c2009-07-17 01:31:16 +00001481
Ted Kremenekb6f7b722007-08-30 18:13:31 +00001482 if (!ElseBlock) // Can occur when the Else body has all NullStmts.
1483 ElseBlock = sv.get();
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001484 else if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001485 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001486 return 0;
1487 }
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001488 }
Mike Stump6d9828c2009-07-17 01:31:16 +00001489
Ted Kremenekb6f1d782009-07-17 18:04:55 +00001490 // Process the true branch.
Ted Kremenek9c378f72011-08-12 23:37:29 +00001491 CFGBlock *ThenBlock;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001492 {
Ted Kremenek9c378f72011-08-12 23:37:29 +00001493 Stmt *Then = I->getThen();
Ted Kremenek6db0ad32010-01-19 20:46:35 +00001494 assert(Then);
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001495 SaveAndRestore<CFGBlock*> sv(Succ);
Mike Stump6d9828c2009-07-17 01:31:16 +00001496 Block = NULL;
Marcin Swiderski04e046c2010-10-01 00:52:17 +00001497
1498 // If branch is not a compound statement create implicit scope
1499 // and add destructors.
1500 if (!isa<CompoundStmt>(Then))
1501 addLocalScopeAndDtors(Then);
1502
Ted Kremenek4f880632009-07-17 22:18:43 +00001503 ThenBlock = addStmt(Then);
Mike Stump6d9828c2009-07-17 01:31:16 +00001504
Ted Kremenekdbdf7942009-04-01 03:52:47 +00001505 if (!ThenBlock) {
1506 // We can reach here if the "then" body has all NullStmts.
1507 // Create an empty block so we can distinguish between true and false
1508 // branches in path-sensitive analyses.
1509 ThenBlock = createBlock(false);
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001510 addSuccessor(ThenBlock, sv.get());
Mike Stump6d9828c2009-07-17 01:31:16 +00001511 } else if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001512 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001513 return 0;
Mike Stump6d9828c2009-07-17 01:31:16 +00001514 }
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001515 }
1516
Mike Stump6d9828c2009-07-17 01:31:16 +00001517 // Now create a new block containing the if statement.
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001518 Block = createBlock(false);
Mike Stump6d9828c2009-07-17 01:31:16 +00001519
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001520 // Set the terminator of the new block to the If statement.
1521 Block->setTerminator(I);
Mike Stump6d9828c2009-07-17 01:31:16 +00001522
Mike Stump00998a02009-07-23 23:25:26 +00001523 // See if this is a known constant.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001524 const TryResult &KnownVal = tryEvaluateBool(I->getCond());
Mike Stump00998a02009-07-23 23:25:26 +00001525
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001526 // Now add the successors.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001527 addSuccessor(Block, KnownVal.isFalse() ? NULL : ThenBlock);
1528 addSuccessor(Block, KnownVal.isTrue()? NULL : ElseBlock);
Mike Stump6d9828c2009-07-17 01:31:16 +00001529
1530 // Add the condition as the last statement in the new block. This may create
1531 // new blocks as the condition may contain control-flow. Any newly created
1532 // blocks will be pointed to be "Block".
Ted Kremenek61dfbec2009-12-23 04:49:01 +00001533 Block = addStmt(I->getCond());
Ted Kremenekad5a8942010-08-02 23:46:59 +00001534
Ted Kremenek61dfbec2009-12-23 04:49:01 +00001535 // Finally, if the IfStmt contains a condition variable, add both the IfStmt
1536 // and the condition variable initialization to the CFG.
1537 if (VarDecl *VD = I->getConditionVariable()) {
1538 if (Expr *Init = VD->getInit()) {
1539 autoCreateBlock();
Ted Kremenekd40066b2011-04-04 23:29:12 +00001540 appendStmt(Block, I->getConditionVariableDeclStmt());
Ted Kremenek61dfbec2009-12-23 04:49:01 +00001541 addStmt(Init);
1542 }
1543 }
Ted Kremenekad5a8942010-08-02 23:46:59 +00001544
Ted Kremenek61dfbec2009-12-23 04:49:01 +00001545 return Block;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001546}
Mike Stump6d9828c2009-07-17 01:31:16 +00001547
1548
Ted Kremenek9c378f72011-08-12 23:37:29 +00001549CFGBlock *CFGBuilder::VisitReturnStmt(ReturnStmt *R) {
Ted Kremenek6c249722009-09-24 18:45:41 +00001550 // If we were in the middle of a block we stop processing that block.
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001551 //
Mike Stump6d9828c2009-07-17 01:31:16 +00001552 // NOTE: If a "return" appears in the middle of a block, this means that the
1553 // code afterwards is DEAD (unreachable). We still keep a basic block
1554 // for that code; a simple "mark-and-sweep" from the entry block will be
1555 // able to report such dead blocks.
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001556
1557 // Create the new block.
1558 Block = createBlock(false);
Mike Stump6d9828c2009-07-17 01:31:16 +00001559
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001560 // The Exit block is the only successor.
Marcin Swiderskifcb72ac2010-10-01 00:23:17 +00001561 addAutomaticObjDtors(ScopePos, LocalScope::const_iterator(), R);
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001562 addSuccessor(Block, &cfg->getExit());
Mike Stump6d9828c2009-07-17 01:31:16 +00001563
1564 // Add the return statement to the block. This may create new blocks if R
1565 // contains control-flow (short-circuit operations).
Ted Kremenek852274d2009-12-16 03:18:58 +00001566 return VisitStmt(R, AddStmtChoice::AlwaysAdd);
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001567}
1568
Ted Kremenek9c378f72011-08-12 23:37:29 +00001569CFGBlock *CFGBuilder::VisitLabelStmt(LabelStmt *L) {
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001570 // Get the block of the labeled statement. Add it to our map.
Ted Kremenek4f880632009-07-17 22:18:43 +00001571 addStmt(L->getSubStmt());
Chris Lattnerad8dcf42011-02-17 07:39:24 +00001572 CFGBlock *LabelBlock = Block;
Mike Stump6d9828c2009-07-17 01:31:16 +00001573
Ted Kremenek4f880632009-07-17 22:18:43 +00001574 if (!LabelBlock) // This can happen when the body is empty, i.e.
1575 LabelBlock = createBlock(); // scopes that only contains NullStmts.
Mike Stump6d9828c2009-07-17 01:31:16 +00001576
Chris Lattnerad8dcf42011-02-17 07:39:24 +00001577 assert(LabelMap.find(L->getDecl()) == LabelMap.end() &&
1578 "label already in map");
1579 LabelMap[L->getDecl()] = JumpTarget(LabelBlock, ScopePos);
Mike Stump6d9828c2009-07-17 01:31:16 +00001580
1581 // Labels partition blocks, so this is the end of the basic block we were
1582 // processing (L is the block's label). Because this is label (and we have
1583 // already processed the substatement) there is no extra control-flow to worry
1584 // about.
Ted Kremenek9cffe732007-08-29 23:20:49 +00001585 LabelBlock->setLabel(L);
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001586 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001587 return 0;
Mike Stump6d9828c2009-07-17 01:31:16 +00001588
1589 // We set Block to NULL to allow lazy creation of a new block (if necessary);
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001590 Block = NULL;
Mike Stump6d9828c2009-07-17 01:31:16 +00001591
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001592 // This block is now the implicit successor of other blocks.
1593 Succ = LabelBlock;
Mike Stump6d9828c2009-07-17 01:31:16 +00001594
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001595 return LabelBlock;
1596}
1597
Ted Kremenek9c378f72011-08-12 23:37:29 +00001598CFGBlock *CFGBuilder::VisitGotoStmt(GotoStmt *G) {
Mike Stump6d9828c2009-07-17 01:31:16 +00001599 // Goto is a control-flow statement. Thus we stop processing the current
1600 // block and create a new one.
Ted Kremenek4f880632009-07-17 22:18:43 +00001601
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001602 Block = createBlock(false);
1603 Block->setTerminator(G);
Mike Stump6d9828c2009-07-17 01:31:16 +00001604
1605 // If we already know the mapping to the label block add the successor now.
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001606 LabelMapTy::iterator I = LabelMap.find(G->getLabel());
Mike Stump6d9828c2009-07-17 01:31:16 +00001607
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001608 if (I == LabelMap.end())
1609 // We will need to backpatch this block later.
Marcin Swiderskif1308c72010-09-25 11:05:21 +00001610 BackpatchBlocks.push_back(JumpSource(Block, ScopePos));
1611 else {
1612 JumpTarget JT = I->second;
Ted Kremenek9ce52702011-01-07 19:37:16 +00001613 addAutomaticObjDtors(ScopePos, JT.scopePosition, G);
1614 addSuccessor(Block, JT.block);
Marcin Swiderskif1308c72010-09-25 11:05:21 +00001615 }
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001616
Mike Stump6d9828c2009-07-17 01:31:16 +00001617 return Block;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001618}
1619
Ted Kremenek9c378f72011-08-12 23:37:29 +00001620CFGBlock *CFGBuilder::VisitForStmt(ForStmt *F) {
1621 CFGBlock *LoopSuccessor = NULL;
Mike Stump6d9828c2009-07-17 01:31:16 +00001622
Marcin Swiderski47575f12010-10-01 01:38:14 +00001623 // Save local scope position because in case of condition variable ScopePos
1624 // won't be restored when traversing AST.
1625 SaveAndRestore<LocalScope::const_iterator> save_scope_pos(ScopePos);
1626
1627 // Create local scope for init statement and possible condition variable.
1628 // Add destructor for init statement and condition variable.
1629 // Store scope position for continue statement.
Ted Kremenek9c378f72011-08-12 23:37:29 +00001630 if (Stmt *Init = F->getInit())
Marcin Swiderski47575f12010-10-01 01:38:14 +00001631 addLocalScopeForStmt(Init);
Marcin Swiderskif1308c72010-09-25 11:05:21 +00001632 LocalScope::const_iterator LoopBeginScopePos = ScopePos;
1633
Ted Kremenek9c378f72011-08-12 23:37:29 +00001634 if (VarDecl *VD = F->getConditionVariable())
Marcin Swiderski47575f12010-10-01 01:38:14 +00001635 addLocalScopeForVarDecl(VD);
1636 LocalScope::const_iterator ContinueScopePos = ScopePos;
1637
1638 addAutomaticObjDtors(ScopePos, save_scope_pos.get(), F);
1639
Mike Stumpfefb9f72009-07-21 01:12:51 +00001640 // "for" is a control-flow statement. Thus we stop processing the current
1641 // block.
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001642 if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001643 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001644 return 0;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001645 LoopSuccessor = Block;
Ted Kremenek4f880632009-07-17 22:18:43 +00001646 } else
1647 LoopSuccessor = Succ;
Mike Stump6d9828c2009-07-17 01:31:16 +00001648
Ted Kremenek3f64a0e2010-05-21 20:30:15 +00001649 // Save the current value for the break targets.
1650 // All breaks should go to the code following the loop.
Marcin Swiderskif1308c72010-09-25 11:05:21 +00001651 SaveAndRestore<JumpTarget> save_break(BreakJumpTarget);
Marcin Swiderski47575f12010-10-01 01:38:14 +00001652 BreakJumpTarget = JumpTarget(LoopSuccessor, ScopePos);
Ted Kremenek3f64a0e2010-05-21 20:30:15 +00001653
Mike Stump6d9828c2009-07-17 01:31:16 +00001654 // Because of short-circuit evaluation, the condition of the loop can span
1655 // multiple basic blocks. Thus we need the "Entry" and "Exit" blocks that
1656 // evaluate the condition.
Ted Kremenek9c378f72011-08-12 23:37:29 +00001657 CFGBlock *ExitConditionBlock = createBlock(false);
1658 CFGBlock *EntryConditionBlock = ExitConditionBlock;
Mike Stump6d9828c2009-07-17 01:31:16 +00001659
Ted Kremenek49af7cb2007-08-27 19:46:09 +00001660 // Set the terminator for the "exit" condition block.
Mike Stump6d9828c2009-07-17 01:31:16 +00001661 ExitConditionBlock->setTerminator(F);
1662
1663 // Now add the actual condition to the condition block. Because the condition
1664 // itself may contain control-flow, new blocks may be created.
Ted Kremenek9c378f72011-08-12 23:37:29 +00001665 if (Stmt *C = F->getCond()) {
Ted Kremenek49af7cb2007-08-27 19:46:09 +00001666 Block = ExitConditionBlock;
1667 EntryConditionBlock = addStmt(C);
Ted Kremenek9ce52702011-01-07 19:37:16 +00001668 if (badCFG)
1669 return 0;
Ted Kremenek8f3b8342010-09-15 07:01:20 +00001670 assert(Block == EntryConditionBlock ||
1671 (Block == 0 && EntryConditionBlock == Succ));
Ted Kremenek58b87fe2009-12-24 01:49:06 +00001672
1673 // If this block contains a condition variable, add both the condition
1674 // variable and initializer to the CFG.
1675 if (VarDecl *VD = F->getConditionVariable()) {
1676 if (Expr *Init = VD->getInit()) {
1677 autoCreateBlock();
Ted Kremenekd40066b2011-04-04 23:29:12 +00001678 appendStmt(Block, F->getConditionVariableDeclStmt());
Ted Kremenek58b87fe2009-12-24 01:49:06 +00001679 EntryConditionBlock = addStmt(Init);
1680 assert(Block == EntryConditionBlock);
1681 }
1682 }
Ted Kremenekad5a8942010-08-02 23:46:59 +00001683
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001684 if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001685 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001686 return 0;
1687 }
Ted Kremenek49af7cb2007-08-27 19:46:09 +00001688 }
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001689
Mike Stump6d9828c2009-07-17 01:31:16 +00001690 // The condition block is the implicit successor for the loop body as well as
1691 // any code above the loop.
Ted Kremenek49af7cb2007-08-27 19:46:09 +00001692 Succ = EntryConditionBlock;
Mike Stump6d9828c2009-07-17 01:31:16 +00001693
Mike Stump00998a02009-07-23 23:25:26 +00001694 // See if this is a known constant.
Ted Kremenek941fde82009-07-24 04:47:11 +00001695 TryResult KnownVal(true);
Mike Stump1eb44332009-09-09 15:08:12 +00001696
Mike Stump00998a02009-07-23 23:25:26 +00001697 if (F->getCond())
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001698 KnownVal = tryEvaluateBool(F->getCond());
Mike Stump00998a02009-07-23 23:25:26 +00001699
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001700 // Now create the loop body.
1701 {
Ted Kremenek6db0ad32010-01-19 20:46:35 +00001702 assert(F->getBody());
Mike Stump6d9828c2009-07-17 01:31:16 +00001703
Ted Kremenek3f64a0e2010-05-21 20:30:15 +00001704 // Save the current values for Block, Succ, and continue targets.
Marcin Swiderskif1308c72010-09-25 11:05:21 +00001705 SaveAndRestore<CFGBlock*> save_Block(Block), save_Succ(Succ);
1706 SaveAndRestore<JumpTarget> save_continue(ContinueJumpTarget);
Mike Stump6d9828c2009-07-17 01:31:16 +00001707
Ted Kremenekaf603f72007-08-30 18:39:40 +00001708 // Create a new block to contain the (bottom) of the loop body.
1709 Block = NULL;
Marcin Swiderski47575f12010-10-01 01:38:14 +00001710
1711 // Loop body should end with destructor of Condition variable (if any).
1712 addAutomaticObjDtors(ScopePos, LoopBeginScopePos, F);
Mike Stump6d9828c2009-07-17 01:31:16 +00001713
Ted Kremenek9c378f72011-08-12 23:37:29 +00001714 if (Stmt *I = F->getInc()) {
Mike Stump6d9828c2009-07-17 01:31:16 +00001715 // Generate increment code in its own basic block. This is the target of
1716 // continue statements.
Ted Kremenek4f880632009-07-17 22:18:43 +00001717 Succ = addStmt(I);
Mike Stump6d9828c2009-07-17 01:31:16 +00001718 } else {
1719 // No increment code. Create a special, empty, block that is used as the
1720 // target block for "looping back" to the start of the loop.
Ted Kremenek3575f842009-04-28 00:51:56 +00001721 assert(Succ == EntryConditionBlock);
Marcin Swiderski47575f12010-10-01 01:38:14 +00001722 Succ = Block ? Block : createBlock();
Ted Kremeneke9334502008-09-04 21:48:47 +00001723 }
Mike Stump6d9828c2009-07-17 01:31:16 +00001724
Ted Kremenek3575f842009-04-28 00:51:56 +00001725 // Finish up the increment (or empty) block if it hasn't been already.
1726 if (Block) {
1727 assert(Block == Succ);
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001728 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001729 return 0;
Ted Kremenek3575f842009-04-28 00:51:56 +00001730 Block = 0;
1731 }
Mike Stump6d9828c2009-07-17 01:31:16 +00001732
Marcin Swiderski47575f12010-10-01 01:38:14 +00001733 ContinueJumpTarget = JumpTarget(Succ, ContinueScopePos);
Mike Stump6d9828c2009-07-17 01:31:16 +00001734
Ted Kremenek3575f842009-04-28 00:51:56 +00001735 // The starting block for the loop increment is the block that should
1736 // represent the 'loop target' for looping back to the start of the loop.
Ted Kremenek9ce52702011-01-07 19:37:16 +00001737 ContinueJumpTarget.block->setLoopTarget(F);
Ted Kremenek3575f842009-04-28 00:51:56 +00001738
Marcin Swiderski47575f12010-10-01 01:38:14 +00001739 // If body is not a compound statement create implicit scope
1740 // and add destructors.
1741 if (!isa<CompoundStmt>(F->getBody()))
1742 addLocalScopeAndDtors(F->getBody());
1743
Mike Stump6d9828c2009-07-17 01:31:16 +00001744 // Now populate the body block, and in the process create new blocks as we
1745 // walk the body of the loop.
Ted Kremenek9c378f72011-08-12 23:37:29 +00001746 CFGBlock *BodyBlock = addStmt(F->getBody());
Ted Kremenekaf603f72007-08-30 18:39:40 +00001747
1748 if (!BodyBlock)
Ted Kremenek9ce52702011-01-07 19:37:16 +00001749 BodyBlock = ContinueJumpTarget.block;//can happen for "for (...;...;...);"
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001750 else if (badCFG)
Ted Kremenek941fde82009-07-24 04:47:11 +00001751 return 0;
Mike Stump6d9828c2009-07-17 01:31:16 +00001752
Ted Kremenek941fde82009-07-24 04:47:11 +00001753 // This new body block is a successor to our "exit" condition block.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001754 addSuccessor(ExitConditionBlock, KnownVal.isFalse() ? NULL : BodyBlock);
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001755 }
Mike Stump6d9828c2009-07-17 01:31:16 +00001756
Ted Kremenek941fde82009-07-24 04:47:11 +00001757 // Link up the condition block with the code that follows the loop. (the
1758 // false branch).
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001759 addSuccessor(ExitConditionBlock, KnownVal.isTrue() ? NULL : LoopSuccessor);
Mike Stump6d9828c2009-07-17 01:31:16 +00001760
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001761 // If the loop contains initialization, create a new block for those
Mike Stump6d9828c2009-07-17 01:31:16 +00001762 // statements. This block can also contain statements that precede the loop.
Ted Kremenek9c378f72011-08-12 23:37:29 +00001763 if (Stmt *I = F->getInit()) {
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001764 Block = createBlock();
Ted Kremenek49af7cb2007-08-27 19:46:09 +00001765 return addStmt(I);
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001766 }
Zhanyong Wan36f327c2010-11-22 19:32:14 +00001767
1768 // There is no loop initialization. We are thus basically a while loop.
1769 // NULL out Block to force lazy block construction.
1770 Block = NULL;
1771 Succ = EntryConditionBlock;
1772 return EntryConditionBlock;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001773}
1774
Ted Kremenek115c1b92010-04-11 17:02:10 +00001775CFGBlock *CFGBuilder::VisitMemberExpr(MemberExpr *M, AddStmtChoice asc) {
Ted Kremenek3179a452011-03-10 01:14:11 +00001776 if (asc.alwaysAdd(*this, M)) {
Ted Kremenek115c1b92010-04-11 17:02:10 +00001777 autoCreateBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00001778 appendStmt(Block, M);
Ted Kremenek115c1b92010-04-11 17:02:10 +00001779 }
Ted Kremenek892697d2010-12-16 07:46:53 +00001780 return Visit(M->getBase());
Ted Kremenek115c1b92010-04-11 17:02:10 +00001781}
1782
Ted Kremenek9c378f72011-08-12 23:37:29 +00001783CFGBlock *CFGBuilder::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
Ted Kremenek514de5a2008-11-11 17:10:00 +00001784 // Objective-C fast enumeration 'for' statements:
1785 // http://developer.apple.com/documentation/Cocoa/Conceptual/ObjectiveC
1786 //
1787 // for ( Type newVariable in collection_expression ) { statements }
1788 //
1789 // becomes:
1790 //
1791 // prologue:
1792 // 1. collection_expression
1793 // T. jump to loop_entry
1794 // loop_entry:
Ted Kremenek4cb3a852008-11-14 01:57:41 +00001795 // 1. side-effects of element expression
Ted Kremenek514de5a2008-11-11 17:10:00 +00001796 // 1. ObjCForCollectionStmt [performs binding to newVariable]
1797 // T. ObjCForCollectionStmt TB, FB [jumps to TB if newVariable != nil]
1798 // TB:
1799 // statements
1800 // T. jump to loop_entry
1801 // FB:
1802 // what comes after
1803 //
1804 // and
1805 //
1806 // Type existingItem;
1807 // for ( existingItem in expression ) { statements }
1808 //
1809 // becomes:
1810 //
Mike Stump6d9828c2009-07-17 01:31:16 +00001811 // the same with newVariable replaced with existingItem; the binding works
1812 // the same except that for one ObjCForCollectionStmt::getElement() returns
1813 // a DeclStmt and the other returns a DeclRefExpr.
Ted Kremenek514de5a2008-11-11 17:10:00 +00001814 //
Mike Stump6d9828c2009-07-17 01:31:16 +00001815
Ted Kremenek9c378f72011-08-12 23:37:29 +00001816 CFGBlock *LoopSuccessor = 0;
Mike Stump6d9828c2009-07-17 01:31:16 +00001817
Ted Kremenek514de5a2008-11-11 17:10:00 +00001818 if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001819 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001820 return 0;
Ted Kremenek514de5a2008-11-11 17:10:00 +00001821 LoopSuccessor = Block;
1822 Block = 0;
Ted Kremenek4f880632009-07-17 22:18:43 +00001823 } else
1824 LoopSuccessor = Succ;
Mike Stump6d9828c2009-07-17 01:31:16 +00001825
Ted Kremenek4cb3a852008-11-14 01:57:41 +00001826 // Build the condition blocks.
Ted Kremenek9c378f72011-08-12 23:37:29 +00001827 CFGBlock *ExitConditionBlock = createBlock(false);
Mike Stump6d9828c2009-07-17 01:31:16 +00001828
Ted Kremenek4cb3a852008-11-14 01:57:41 +00001829 // Set the terminator for the "exit" condition block.
Mike Stump6d9828c2009-07-17 01:31:16 +00001830 ExitConditionBlock->setTerminator(S);
1831
1832 // The last statement in the block should be the ObjCForCollectionStmt, which
1833 // performs the actual binding to 'element' and determines if there are any
1834 // more items in the collection.
Ted Kremenek892697d2010-12-16 07:46:53 +00001835 appendStmt(ExitConditionBlock, S);
Ted Kremenek4cb3a852008-11-14 01:57:41 +00001836 Block = ExitConditionBlock;
Mike Stump6d9828c2009-07-17 01:31:16 +00001837
Ted Kremenek4cb3a852008-11-14 01:57:41 +00001838 // Walk the 'element' expression to see if there are any side-effects. We
Chris Lattnerfc8f0e12011-04-15 05:22:18 +00001839 // generate new blocks as necessary. We DON'T add the statement by default to
Mike Stump6d9828c2009-07-17 01:31:16 +00001840 // the CFG unless it contains control-flow.
Ted Kremenek012614e2011-08-17 21:04:19 +00001841 CFGBlock *EntryConditionBlock = Visit(S->getElement(),
1842 AddStmtChoice::NotAlwaysAdd);
Mike Stump6d9828c2009-07-17 01:31:16 +00001843 if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001844 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001845 return 0;
1846 Block = 0;
1847 }
Mike Stump6d9828c2009-07-17 01:31:16 +00001848
1849 // The condition block is the implicit successor for the loop body as well as
1850 // any code above the loop.
Ted Kremenek4cb3a852008-11-14 01:57:41 +00001851 Succ = EntryConditionBlock;
Mike Stump6d9828c2009-07-17 01:31:16 +00001852
Ted Kremenek514de5a2008-11-11 17:10:00 +00001853 // Now create the true branch.
Mike Stump6d9828c2009-07-17 01:31:16 +00001854 {
Ted Kremenek4cb3a852008-11-14 01:57:41 +00001855 // Save the current values for Succ, continue and break targets.
Marcin Swiderskif1308c72010-09-25 11:05:21 +00001856 SaveAndRestore<CFGBlock*> save_Succ(Succ);
1857 SaveAndRestore<JumpTarget> save_continue(ContinueJumpTarget),
1858 save_break(BreakJumpTarget);
Mike Stump6d9828c2009-07-17 01:31:16 +00001859
Marcin Swiderskif1308c72010-09-25 11:05:21 +00001860 BreakJumpTarget = JumpTarget(LoopSuccessor, ScopePos);
1861 ContinueJumpTarget = JumpTarget(EntryConditionBlock, ScopePos);
Mike Stump6d9828c2009-07-17 01:31:16 +00001862
Ted Kremenek9c378f72011-08-12 23:37:29 +00001863 CFGBlock *BodyBlock = addStmt(S->getBody());
Mike Stump6d9828c2009-07-17 01:31:16 +00001864
Ted Kremenek4cb3a852008-11-14 01:57:41 +00001865 if (!BodyBlock)
1866 BodyBlock = EntryConditionBlock; // can happen for "for (X in Y) ;"
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001867 else if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001868 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001869 return 0;
1870 }
Mike Stump6d9828c2009-07-17 01:31:16 +00001871
Ted Kremenek4cb3a852008-11-14 01:57:41 +00001872 // This new body block is a successor to our "exit" condition block.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001873 addSuccessor(ExitConditionBlock, BodyBlock);
Ted Kremenek4cb3a852008-11-14 01:57:41 +00001874 }
Mike Stump6d9828c2009-07-17 01:31:16 +00001875
Ted Kremenek4cb3a852008-11-14 01:57:41 +00001876 // Link up the condition block with the code that follows the loop.
1877 // (the false branch).
Ted Kremenek0a3ed312010-12-17 04:44:39 +00001878 addSuccessor(ExitConditionBlock, LoopSuccessor);
Ted Kremenek4cb3a852008-11-14 01:57:41 +00001879
Ted Kremenek514de5a2008-11-11 17:10:00 +00001880 // Now create a prologue block to contain the collection expression.
Ted Kremenek4cb3a852008-11-14 01:57:41 +00001881 Block = createBlock();
Ted Kremenek514de5a2008-11-11 17:10:00 +00001882 return addStmt(S->getCollection());
Mike Stump6d9828c2009-07-17 01:31:16 +00001883}
1884
Ted Kremenek9c378f72011-08-12 23:37:29 +00001885CFGBlock *CFGBuilder::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
Ted Kremenekb3b0b362009-05-02 01:49:13 +00001886 // FIXME: Add locking 'primitives' to CFG for @synchronized.
Mike Stump6d9828c2009-07-17 01:31:16 +00001887
Ted Kremenekb3b0b362009-05-02 01:49:13 +00001888 // Inline the body.
Ted Kremenek4f880632009-07-17 22:18:43 +00001889 CFGBlock *SyncBlock = addStmt(S->getSynchBody());
Mike Stump6d9828c2009-07-17 01:31:16 +00001890
Ted Kremenekda5348e2009-05-05 23:11:51 +00001891 // The sync body starts its own basic block. This makes it a little easier
1892 // for diagnostic clients.
1893 if (SyncBlock) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001894 if (badCFG)
Ted Kremenekda5348e2009-05-05 23:11:51 +00001895 return 0;
Mike Stump6d9828c2009-07-17 01:31:16 +00001896
Ted Kremenekda5348e2009-05-05 23:11:51 +00001897 Block = 0;
Ted Kremenekfadebba2010-05-13 16:38:08 +00001898 Succ = SyncBlock;
Ted Kremenekda5348e2009-05-05 23:11:51 +00001899 }
Mike Stump6d9828c2009-07-17 01:31:16 +00001900
Ted Kremenek4beaa9f2010-09-10 03:05:33 +00001901 // Add the @synchronized to the CFG.
1902 autoCreateBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00001903 appendStmt(Block, S);
Ted Kremenek4beaa9f2010-09-10 03:05:33 +00001904
Ted Kremenekb3b0b362009-05-02 01:49:13 +00001905 // Inline the sync expression.
Ted Kremenek4f880632009-07-17 22:18:43 +00001906 return addStmt(S->getSynchExpr());
Ted Kremenekb3b0b362009-05-02 01:49:13 +00001907}
Mike Stump6d9828c2009-07-17 01:31:16 +00001908
Ted Kremenek9c378f72011-08-12 23:37:29 +00001909CFGBlock *CFGBuilder::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
Ted Kremenek4f880632009-07-17 22:18:43 +00001910 // FIXME
Ted Kremenek90658ec2009-04-07 04:26:02 +00001911 return NYS();
Ted Kremeneke31c0d22009-03-30 22:29:21 +00001912}
Ted Kremenek514de5a2008-11-11 17:10:00 +00001913
John McCall4b9c2d22011-11-06 09:01:30 +00001914CFGBlock *CFGBuilder::VisitPseudoObjectExpr(PseudoObjectExpr *E) {
1915 autoCreateBlock();
1916
1917 // Add the PseudoObject as the last thing.
1918 appendStmt(Block, E);
1919
1920 CFGBlock *lastBlock = Block;
1921
1922 // Before that, evaluate all of the semantics in order. In
1923 // CFG-land, that means appending them in reverse order.
1924 for (unsigned i = E->getNumSemanticExprs(); i != 0; ) {
1925 Expr *Semantic = E->getSemanticExpr(--i);
1926
1927 // If the semantic is an opaque value, we're being asked to bind
1928 // it to its source expression.
1929 if (OpaqueValueExpr *OVE = dyn_cast<OpaqueValueExpr>(Semantic))
1930 Semantic = OVE->getSourceExpr();
1931
1932 if (CFGBlock *B = Visit(Semantic))
1933 lastBlock = B;
1934 }
1935
1936 return lastBlock;
1937}
1938
Ted Kremenek9c378f72011-08-12 23:37:29 +00001939CFGBlock *CFGBuilder::VisitWhileStmt(WhileStmt *W) {
1940 CFGBlock *LoopSuccessor = NULL;
Mike Stump6d9828c2009-07-17 01:31:16 +00001941
Marcin Swiderski05adedc2010-10-01 01:14:17 +00001942 // Save local scope position because in case of condition variable ScopePos
1943 // won't be restored when traversing AST.
1944 SaveAndRestore<LocalScope::const_iterator> save_scope_pos(ScopePos);
1945
1946 // Create local scope for possible condition variable.
1947 // Store scope position for continue statement.
Marcin Swiderskif1308c72010-09-25 11:05:21 +00001948 LocalScope::const_iterator LoopBeginScopePos = ScopePos;
Ted Kremenek9c378f72011-08-12 23:37:29 +00001949 if (VarDecl *VD = W->getConditionVariable()) {
Marcin Swiderski05adedc2010-10-01 01:14:17 +00001950 addLocalScopeForVarDecl(VD);
1951 addAutomaticObjDtors(ScopePos, LoopBeginScopePos, W);
1952 }
Marcin Swiderskif1308c72010-09-25 11:05:21 +00001953
Mike Stumpfefb9f72009-07-21 01:12:51 +00001954 // "while" is a control-flow statement. Thus we stop processing the current
1955 // block.
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001956 if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001957 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001958 return 0;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00001959 LoopSuccessor = Block;
Ted Kremenek6b12da92011-02-21 22:11:26 +00001960 Block = 0;
Ted Kremenek4f880632009-07-17 22:18:43 +00001961 } else
1962 LoopSuccessor = Succ;
Mike Stump6d9828c2009-07-17 01:31:16 +00001963
1964 // Because of short-circuit evaluation, the condition of the loop can span
1965 // multiple basic blocks. Thus we need the "Entry" and "Exit" blocks that
1966 // evaluate the condition.
Ted Kremenek9c378f72011-08-12 23:37:29 +00001967 CFGBlock *ExitConditionBlock = createBlock(false);
1968 CFGBlock *EntryConditionBlock = ExitConditionBlock;
Mike Stump6d9828c2009-07-17 01:31:16 +00001969
Ted Kremenek49af7cb2007-08-27 19:46:09 +00001970 // Set the terminator for the "exit" condition block.
1971 ExitConditionBlock->setTerminator(W);
Mike Stump6d9828c2009-07-17 01:31:16 +00001972
1973 // Now add the actual condition to the condition block. Because the condition
1974 // itself may contain control-flow, new blocks may be created. Thus we update
1975 // "Succ" after adding the condition.
Ted Kremenek9c378f72011-08-12 23:37:29 +00001976 if (Stmt *C = W->getCond()) {
Ted Kremenek49af7cb2007-08-27 19:46:09 +00001977 Block = ExitConditionBlock;
1978 EntryConditionBlock = addStmt(C);
Zhongxing Xua1898dd2010-10-27 03:23:10 +00001979 // The condition might finish the current 'Block'.
1980 Block = EntryConditionBlock;
Ted Kremenekad5a8942010-08-02 23:46:59 +00001981
Ted Kremenek4ec010a2009-12-24 01:34:10 +00001982 // If this block contains a condition variable, add both the condition
1983 // variable and initializer to the CFG.
1984 if (VarDecl *VD = W->getConditionVariable()) {
1985 if (Expr *Init = VD->getInit()) {
1986 autoCreateBlock();
Ted Kremenekd40066b2011-04-04 23:29:12 +00001987 appendStmt(Block, W->getConditionVariableDeclStmt());
Ted Kremenek4ec010a2009-12-24 01:34:10 +00001988 EntryConditionBlock = addStmt(Init);
1989 assert(Block == EntryConditionBlock);
1990 }
1991 }
1992
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001993 if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00001994 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00001995 return 0;
1996 }
Ted Kremenek49af7cb2007-08-27 19:46:09 +00001997 }
Mike Stump6d9828c2009-07-17 01:31:16 +00001998
1999 // The condition block is the implicit successor for the loop body as well as
2000 // any code above the loop.
Ted Kremenek49af7cb2007-08-27 19:46:09 +00002001 Succ = EntryConditionBlock;
Mike Stump6d9828c2009-07-17 01:31:16 +00002002
Mike Stump00998a02009-07-23 23:25:26 +00002003 // See if this is a known constant.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002004 const TryResult& KnownVal = tryEvaluateBool(W->getCond());
Mike Stump00998a02009-07-23 23:25:26 +00002005
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002006 // Process the loop body.
2007 {
Ted Kremenekf6e85412009-04-28 03:09:44 +00002008 assert(W->getBody());
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002009
2010 // Save the current values for Block, Succ, and continue and break targets
Marcin Swiderskif1308c72010-09-25 11:05:21 +00002011 SaveAndRestore<CFGBlock*> save_Block(Block), save_Succ(Succ);
2012 SaveAndRestore<JumpTarget> save_continue(ContinueJumpTarget),
2013 save_break(BreakJumpTarget);
Ted Kremenekf6e85412009-04-28 03:09:44 +00002014
Mike Stump6d9828c2009-07-17 01:31:16 +00002015 // Create an empty block to represent the transition block for looping back
2016 // to the head of the loop.
Ted Kremenekf6e85412009-04-28 03:09:44 +00002017 Block = 0;
2018 assert(Succ == EntryConditionBlock);
2019 Succ = createBlock();
2020 Succ->setLoopTarget(W);
Marcin Swiderskif1308c72010-09-25 11:05:21 +00002021 ContinueJumpTarget = JumpTarget(Succ, LoopBeginScopePos);
Mike Stump6d9828c2009-07-17 01:31:16 +00002022
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002023 // All breaks should go to the code following the loop.
Marcin Swiderski05adedc2010-10-01 01:14:17 +00002024 BreakJumpTarget = JumpTarget(LoopSuccessor, ScopePos);
Mike Stump6d9828c2009-07-17 01:31:16 +00002025
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002026 // NULL out Block to force lazy instantiation of blocks for the body.
2027 Block = NULL;
Mike Stump6d9828c2009-07-17 01:31:16 +00002028
Marcin Swiderski05adedc2010-10-01 01:14:17 +00002029 // Loop body should end with destructor of Condition variable (if any).
2030 addAutomaticObjDtors(ScopePos, LoopBeginScopePos, W);
2031
2032 // If body is not a compound statement create implicit scope
2033 // and add destructors.
2034 if (!isa<CompoundStmt>(W->getBody()))
2035 addLocalScopeAndDtors(W->getBody());
2036
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002037 // Create the body. The returned block is the entry to the loop body.
Ted Kremenek9c378f72011-08-12 23:37:29 +00002038 CFGBlock *BodyBlock = addStmt(W->getBody());
Mike Stump6d9828c2009-07-17 01:31:16 +00002039
Ted Kremenekaf603f72007-08-30 18:39:40 +00002040 if (!BodyBlock)
Ted Kremenek9ce52702011-01-07 19:37:16 +00002041 BodyBlock = ContinueJumpTarget.block; // can happen for "while(...) ;"
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00002042 else if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00002043 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00002044 return 0;
2045 }
Mike Stump6d9828c2009-07-17 01:31:16 +00002046
Ted Kremenek941fde82009-07-24 04:47:11 +00002047 // Add the loop body entry as a successor to the condition.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002048 addSuccessor(ExitConditionBlock, KnownVal.isFalse() ? NULL : BodyBlock);
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002049 }
Mike Stump6d9828c2009-07-17 01:31:16 +00002050
Ted Kremenek941fde82009-07-24 04:47:11 +00002051 // Link up the condition block with the code that follows the loop. (the
2052 // false branch).
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002053 addSuccessor(ExitConditionBlock, KnownVal.isTrue() ? NULL : LoopSuccessor);
Mike Stump6d9828c2009-07-17 01:31:16 +00002054
2055 // There can be no more statements in the condition block since we loop back
2056 // to this block. NULL out Block to force lazy creation of another block.
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002057 Block = NULL;
Mike Stump6d9828c2009-07-17 01:31:16 +00002058
Ted Kremenek4ec010a2009-12-24 01:34:10 +00002059 // Return the condition block, which is the dominating block for the loop.
Ted Kremenek54827132008-02-27 07:20:00 +00002060 Succ = EntryConditionBlock;
Ted Kremenek49af7cb2007-08-27 19:46:09 +00002061 return EntryConditionBlock;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002062}
Mike Stump1eb44332009-09-09 15:08:12 +00002063
2064
Ted Kremenek9c378f72011-08-12 23:37:29 +00002065CFGBlock *CFGBuilder::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
Ted Kremenek4f880632009-07-17 22:18:43 +00002066 // FIXME: For now we pretend that @catch and the code it contains does not
2067 // exit.
2068 return Block;
2069}
Mike Stump6d9828c2009-07-17 01:31:16 +00002070
Ted Kremenek9c378f72011-08-12 23:37:29 +00002071CFGBlock *CFGBuilder::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
Ted Kremenek2fda5042008-12-09 20:20:09 +00002072 // FIXME: This isn't complete. We basically treat @throw like a return
2073 // statement.
Mike Stump6d9828c2009-07-17 01:31:16 +00002074
Ted Kremenek6c249722009-09-24 18:45:41 +00002075 // If we were in the middle of a block we stop processing that block.
Zhongxing Xud438b3d2010-09-06 07:32:31 +00002076 if (badCFG)
Ted Kremenek4f880632009-07-17 22:18:43 +00002077 return 0;
Mike Stump6d9828c2009-07-17 01:31:16 +00002078
Ted Kremenek2fda5042008-12-09 20:20:09 +00002079 // Create the new block.
2080 Block = createBlock(false);
Mike Stump6d9828c2009-07-17 01:31:16 +00002081
Ted Kremenek2fda5042008-12-09 20:20:09 +00002082 // The Exit block is the only successor.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002083 addSuccessor(Block, &cfg->getExit());
Mike Stump6d9828c2009-07-17 01:31:16 +00002084
2085 // Add the statement to the block. This may create new blocks if S contains
2086 // control-flow (short-circuit operations).
Ted Kremenek852274d2009-12-16 03:18:58 +00002087 return VisitStmt(S, AddStmtChoice::AlwaysAdd);
Ted Kremenek2fda5042008-12-09 20:20:09 +00002088}
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002089
Ted Kremenek9c378f72011-08-12 23:37:29 +00002090CFGBlock *CFGBuilder::VisitCXXThrowExpr(CXXThrowExpr *T) {
Ted Kremenek6c249722009-09-24 18:45:41 +00002091 // If we were in the middle of a block we stop processing that block.
Zhongxing Xud438b3d2010-09-06 07:32:31 +00002092 if (badCFG)
Mike Stump0979d802009-07-22 22:56:04 +00002093 return 0;
2094
2095 // Create the new block.
2096 Block = createBlock(false);
2097
Mike Stump5d1d2022010-01-19 02:20:09 +00002098 if (TryTerminatedBlock)
2099 // The current try statement is the only successor.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002100 addSuccessor(Block, TryTerminatedBlock);
Ted Kremenekad5a8942010-08-02 23:46:59 +00002101 else
Mike Stump5d1d2022010-01-19 02:20:09 +00002102 // otherwise the Exit block is the only successor.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002103 addSuccessor(Block, &cfg->getExit());
Mike Stump0979d802009-07-22 22:56:04 +00002104
2105 // Add the statement to the block. This may create new blocks if S contains
2106 // control-flow (short-circuit operations).
Ted Kremenek852274d2009-12-16 03:18:58 +00002107 return VisitStmt(T, AddStmtChoice::AlwaysAdd);
Mike Stump0979d802009-07-22 22:56:04 +00002108}
2109
Ted Kremenek9c378f72011-08-12 23:37:29 +00002110CFGBlock *CFGBuilder::VisitDoStmt(DoStmt *D) {
2111 CFGBlock *LoopSuccessor = NULL;
Mike Stump6d9828c2009-07-17 01:31:16 +00002112
Mike Stump8f9893a2009-07-21 01:27:50 +00002113 // "do...while" is a control-flow statement. Thus we stop processing the
2114 // current block.
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002115 if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00002116 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00002117 return 0;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002118 LoopSuccessor = Block;
Ted Kremenek4f880632009-07-17 22:18:43 +00002119 } else
2120 LoopSuccessor = Succ;
Mike Stump6d9828c2009-07-17 01:31:16 +00002121
2122 // Because of short-circuit evaluation, the condition of the loop can span
2123 // multiple basic blocks. Thus we need the "Entry" and "Exit" blocks that
2124 // evaluate the condition.
Ted Kremenek9c378f72011-08-12 23:37:29 +00002125 CFGBlock *ExitConditionBlock = createBlock(false);
2126 CFGBlock *EntryConditionBlock = ExitConditionBlock;
Mike Stump6d9828c2009-07-17 01:31:16 +00002127
Ted Kremenek49af7cb2007-08-27 19:46:09 +00002128 // Set the terminator for the "exit" condition block.
Mike Stump6d9828c2009-07-17 01:31:16 +00002129 ExitConditionBlock->setTerminator(D);
2130
2131 // Now add the actual condition to the condition block. Because the condition
2132 // itself may contain control-flow, new blocks may be created.
Ted Kremenek9c378f72011-08-12 23:37:29 +00002133 if (Stmt *C = D->getCond()) {
Ted Kremenek49af7cb2007-08-27 19:46:09 +00002134 Block = ExitConditionBlock;
2135 EntryConditionBlock = addStmt(C);
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00002136 if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00002137 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00002138 return 0;
2139 }
Ted Kremenek49af7cb2007-08-27 19:46:09 +00002140 }
Mike Stump6d9828c2009-07-17 01:31:16 +00002141
Ted Kremenek54827132008-02-27 07:20:00 +00002142 // The condition block is the implicit successor for the loop body.
Ted Kremenek49af7cb2007-08-27 19:46:09 +00002143 Succ = EntryConditionBlock;
2144
Mike Stump00998a02009-07-23 23:25:26 +00002145 // See if this is a known constant.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002146 const TryResult &KnownVal = tryEvaluateBool(D->getCond());
Mike Stump00998a02009-07-23 23:25:26 +00002147
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002148 // Process the loop body.
Ted Kremenek9c378f72011-08-12 23:37:29 +00002149 CFGBlock *BodyBlock = NULL;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002150 {
Ted Kremenek6db0ad32010-01-19 20:46:35 +00002151 assert(D->getBody());
Mike Stump6d9828c2009-07-17 01:31:16 +00002152
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002153 // Save the current values for Block, Succ, and continue and break targets
Marcin Swiderskif1308c72010-09-25 11:05:21 +00002154 SaveAndRestore<CFGBlock*> save_Block(Block), save_Succ(Succ);
2155 SaveAndRestore<JumpTarget> save_continue(ContinueJumpTarget),
2156 save_break(BreakJumpTarget);
Mike Stump6d9828c2009-07-17 01:31:16 +00002157
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002158 // All continues within this loop should go to the condition block
Marcin Swiderskif1308c72010-09-25 11:05:21 +00002159 ContinueJumpTarget = JumpTarget(EntryConditionBlock, ScopePos);
Mike Stump6d9828c2009-07-17 01:31:16 +00002160
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002161 // All breaks should go to the code following the loop.
Marcin Swiderskif1308c72010-09-25 11:05:21 +00002162 BreakJumpTarget = JumpTarget(LoopSuccessor, ScopePos);
Mike Stump6d9828c2009-07-17 01:31:16 +00002163
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002164 // NULL out Block to force lazy instantiation of blocks for the body.
2165 Block = NULL;
Mike Stump6d9828c2009-07-17 01:31:16 +00002166
Marcin Swiderski05adedc2010-10-01 01:14:17 +00002167 // If body is not a compound statement create implicit scope
2168 // and add destructors.
2169 if (!isa<CompoundStmt>(D->getBody()))
2170 addLocalScopeAndDtors(D->getBody());
2171
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002172 // Create the body. The returned block is the entry to the loop body.
Ted Kremenek4f880632009-07-17 22:18:43 +00002173 BodyBlock = addStmt(D->getBody());
Mike Stump6d9828c2009-07-17 01:31:16 +00002174
Ted Kremenekaf603f72007-08-30 18:39:40 +00002175 if (!BodyBlock)
Ted Kremeneka9d996d2008-02-27 00:28:17 +00002176 BodyBlock = EntryConditionBlock; // can happen for "do ; while(...)"
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00002177 else if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00002178 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00002179 return 0;
2180 }
Mike Stump6d9828c2009-07-17 01:31:16 +00002181
Ted Kremenekd173dc72010-08-17 20:59:56 +00002182 if (!KnownVal.isFalse()) {
2183 // Add an intermediate block between the BodyBlock and the
2184 // ExitConditionBlock to represent the "loop back" transition. Create an
2185 // empty block to represent the transition block for looping back to the
2186 // head of the loop.
2187 // FIXME: Can we do this more efficiently without adding another block?
2188 Block = NULL;
2189 Succ = BodyBlock;
2190 CFGBlock *LoopBackBlock = createBlock();
2191 LoopBackBlock->setLoopTarget(D);
Mike Stump6d9828c2009-07-17 01:31:16 +00002192
Ted Kremenekd173dc72010-08-17 20:59:56 +00002193 // Add the loop body entry as a successor to the condition.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002194 addSuccessor(ExitConditionBlock, LoopBackBlock);
Ted Kremenekd173dc72010-08-17 20:59:56 +00002195 }
2196 else
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002197 addSuccessor(ExitConditionBlock, NULL);
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002198 }
Mike Stump6d9828c2009-07-17 01:31:16 +00002199
Ted Kremenek941fde82009-07-24 04:47:11 +00002200 // Link up the condition block with the code that follows the loop.
2201 // (the false branch).
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002202 addSuccessor(ExitConditionBlock, KnownVal.isTrue() ? NULL : LoopSuccessor);
Mike Stump6d9828c2009-07-17 01:31:16 +00002203
2204 // There can be no more statements in the body block(s) since we loop back to
2205 // the body. NULL out Block to force lazy creation of another block.
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002206 Block = NULL;
Mike Stump6d9828c2009-07-17 01:31:16 +00002207
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002208 // Return the loop body, which is the dominating block for the loop.
Ted Kremenek54827132008-02-27 07:20:00 +00002209 Succ = BodyBlock;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002210 return BodyBlock;
2211}
2212
Ted Kremenek9c378f72011-08-12 23:37:29 +00002213CFGBlock *CFGBuilder::VisitContinueStmt(ContinueStmt *C) {
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002214 // "continue" is a control-flow statement. Thus we stop processing the
2215 // current block.
Zhongxing Xud438b3d2010-09-06 07:32:31 +00002216 if (badCFG)
2217 return 0;
Mike Stump6d9828c2009-07-17 01:31:16 +00002218
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002219 // Now create a new block that ends with the continue statement.
2220 Block = createBlock(false);
2221 Block->setTerminator(C);
Mike Stump6d9828c2009-07-17 01:31:16 +00002222
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002223 // If there is no target for the continue, then we are looking at an
Ted Kremenek235c5ed2009-04-07 18:53:24 +00002224 // incomplete AST. This means the CFG cannot be constructed.
Ted Kremenek9ce52702011-01-07 19:37:16 +00002225 if (ContinueJumpTarget.block) {
2226 addAutomaticObjDtors(ScopePos, ContinueJumpTarget.scopePosition, C);
2227 addSuccessor(Block, ContinueJumpTarget.block);
Marcin Swiderskif1308c72010-09-25 11:05:21 +00002228 } else
Ted Kremenek235c5ed2009-04-07 18:53:24 +00002229 badCFG = true;
Mike Stump6d9828c2009-07-17 01:31:16 +00002230
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002231 return Block;
2232}
Mike Stump1eb44332009-09-09 15:08:12 +00002233
Peter Collingbournef4e3cfb2011-03-11 19:24:49 +00002234CFGBlock *CFGBuilder::VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E,
2235 AddStmtChoice asc) {
Ted Kremenek13fc08a2009-07-18 00:47:21 +00002236
Ted Kremenek3179a452011-03-10 01:14:11 +00002237 if (asc.alwaysAdd(*this, E)) {
Ted Kremenek13fc08a2009-07-18 00:47:21 +00002238 autoCreateBlock();
Ted Kremenek892697d2010-12-16 07:46:53 +00002239 appendStmt(Block, E);
Ted Kremenek13fc08a2009-07-18 00:47:21 +00002240 }
Mike Stump1eb44332009-09-09 15:08:12 +00002241
Ted Kremenek4f880632009-07-17 22:18:43 +00002242 // VLA types have expressions that must be evaluated.
Ted Kremenek97e50712011-04-14 01:50:50 +00002243 CFGBlock *lastBlock = Block;
2244
Ted Kremenek4f880632009-07-17 22:18:43 +00002245 if (E->isArgumentType()) {
John McCallf4c73712011-01-19 06:33:43 +00002246 for (const VariableArrayType *VA =FindVA(E->getArgumentType().getTypePtr());
Ted Kremenek4f880632009-07-17 22:18:43 +00002247 VA != 0; VA = FindVA(VA->getElementType().getTypePtr()))
Ted Kremenek97e50712011-04-14 01:50:50 +00002248 lastBlock = addStmt(VA->getSizeExpr());
Ted Kremenekf91a5b02011-08-06 00:30:00 +00002249 }
Ted Kremenek97e50712011-04-14 01:50:50 +00002250 return lastBlock;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002251}
Mike Stump1eb44332009-09-09 15:08:12 +00002252
Ted Kremenek4f880632009-07-17 22:18:43 +00002253/// VisitStmtExpr - Utility method to handle (nested) statement
2254/// expressions (a GCC extension).
Ted Kremenek9c378f72011-08-12 23:37:29 +00002255CFGBlock *CFGBuilder::VisitStmtExpr(StmtExpr *SE, AddStmtChoice asc) {
Ted Kremenek3179a452011-03-10 01:14:11 +00002256 if (asc.alwaysAdd(*this, SE)) {
Ted Kremenek13fc08a2009-07-18 00:47:21 +00002257 autoCreateBlock();
Ted Kremenek892697d2010-12-16 07:46:53 +00002258 appendStmt(Block, SE);
Ted Kremenek13fc08a2009-07-18 00:47:21 +00002259 }
Ted Kremenek4f880632009-07-17 22:18:43 +00002260 return VisitCompoundStmt(SE->getSubStmt());
2261}
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002262
Ted Kremenek9c378f72011-08-12 23:37:29 +00002263CFGBlock *CFGBuilder::VisitSwitchStmt(SwitchStmt *Terminator) {
Mike Stump6d9828c2009-07-17 01:31:16 +00002264 // "switch" is a control-flow statement. Thus we stop processing the current
2265 // block.
Ted Kremenek9c378f72011-08-12 23:37:29 +00002266 CFGBlock *SwitchSuccessor = NULL;
Mike Stump6d9828c2009-07-17 01:31:16 +00002267
Marcin Swiderski8ae60582010-10-01 01:24:41 +00002268 // Save local scope position because in case of condition variable ScopePos
2269 // won't be restored when traversing AST.
2270 SaveAndRestore<LocalScope::const_iterator> save_scope_pos(ScopePos);
2271
2272 // Create local scope for possible condition variable.
2273 // Store scope position. Add implicit destructor.
Ted Kremenek9c378f72011-08-12 23:37:29 +00002274 if (VarDecl *VD = Terminator->getConditionVariable()) {
Marcin Swiderski8ae60582010-10-01 01:24:41 +00002275 LocalScope::const_iterator SwitchBeginScopePos = ScopePos;
2276 addLocalScopeForVarDecl(VD);
2277 addAutomaticObjDtors(ScopePos, SwitchBeginScopePos, Terminator);
2278 }
2279
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002280 if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00002281 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00002282 return 0;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002283 SwitchSuccessor = Block;
Mike Stump6d9828c2009-07-17 01:31:16 +00002284 } else SwitchSuccessor = Succ;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002285
2286 // Save the current "switch" context.
2287 SaveAndRestore<CFGBlock*> save_switch(SwitchTerminatedBlock),
Ted Kremenekeef5a9a2008-02-13 22:05:39 +00002288 save_default(DefaultCaseBlock);
Marcin Swiderskif1308c72010-09-25 11:05:21 +00002289 SaveAndRestore<JumpTarget> save_break(BreakJumpTarget);
Ted Kremenekeef5a9a2008-02-13 22:05:39 +00002290
Mike Stump6d9828c2009-07-17 01:31:16 +00002291 // Set the "default" case to be the block after the switch statement. If the
2292 // switch statement contains a "default:", this value will be overwritten with
2293 // the block for that code.
Ted Kremenekeef5a9a2008-02-13 22:05:39 +00002294 DefaultCaseBlock = SwitchSuccessor;
Mike Stump6d9828c2009-07-17 01:31:16 +00002295
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002296 // Create a new block that will contain the switch statement.
2297 SwitchTerminatedBlock = createBlock(false);
Mike Stump6d9828c2009-07-17 01:31:16 +00002298
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002299 // Now process the switch body. The code after the switch is the implicit
2300 // successor.
2301 Succ = SwitchSuccessor;
Marcin Swiderskif1308c72010-09-25 11:05:21 +00002302 BreakJumpTarget = JumpTarget(SwitchSuccessor, ScopePos);
Mike Stump6d9828c2009-07-17 01:31:16 +00002303
2304 // When visiting the body, the case statements should automatically get linked
2305 // up to the switch. We also don't keep a pointer to the body, since all
2306 // control-flow from the switch goes to case/default statements.
Ted Kremenek6db0ad32010-01-19 20:46:35 +00002307 assert(Terminator->getBody() && "switch must contain a non-NULL body");
Ted Kremenek49af7cb2007-08-27 19:46:09 +00002308 Block = NULL;
Marcin Swiderski8ae60582010-10-01 01:24:41 +00002309
Ted Kremeneke71f3d52011-03-01 23:12:55 +00002310 // For pruning unreachable case statements, save the current state
2311 // for tracking the condition value.
2312 SaveAndRestore<bool> save_switchExclusivelyCovered(switchExclusivelyCovered,
2313 false);
Ted Kremenek04982472011-03-04 01:03:41 +00002314
Ted Kremeneke71f3d52011-03-01 23:12:55 +00002315 // Determine if the switch condition can be explicitly evaluated.
2316 assert(Terminator->getCond() && "switch condition must be non-NULL");
Ted Kremenek04982472011-03-04 01:03:41 +00002317 Expr::EvalResult result;
Ted Kremeneke9cd9c02011-03-13 03:48:04 +00002318 bool b = tryEvaluate(Terminator->getCond(), result);
2319 SaveAndRestore<Expr::EvalResult*> save_switchCond(switchCond,
2320 b ? &result : 0);
Ted Kremenek04982472011-03-04 01:03:41 +00002321
Marcin Swiderski8ae60582010-10-01 01:24:41 +00002322 // If body is not a compound statement create implicit scope
2323 // and add destructors.
2324 if (!isa<CompoundStmt>(Terminator->getBody()))
2325 addLocalScopeAndDtors(Terminator->getBody());
2326
Zhongxing Xud438b3d2010-09-06 07:32:31 +00002327 addStmt(Terminator->getBody());
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00002328 if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00002329 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00002330 return 0;
2331 }
Ted Kremenek49af7cb2007-08-27 19:46:09 +00002332
Mike Stump6d9828c2009-07-17 01:31:16 +00002333 // If we have no "default:" case, the default transition is to the code
Ted Kremenek432c4782011-03-16 04:32:01 +00002334 // following the switch body. Moreover, take into account if all the
2335 // cases of a switch are covered (e.g., switching on an enum value).
Ted Kremeneke71f3d52011-03-01 23:12:55 +00002336 addSuccessor(SwitchTerminatedBlock,
Ted Kremenek432c4782011-03-16 04:32:01 +00002337 switchExclusivelyCovered || Terminator->isAllEnumCasesCovered()
2338 ? 0 : DefaultCaseBlock);
Mike Stump6d9828c2009-07-17 01:31:16 +00002339
Ted Kremenek49af7cb2007-08-27 19:46:09 +00002340 // Add the terminator and condition in the switch block.
Ted Kremenek411cdee2008-04-16 21:10:48 +00002341 SwitchTerminatedBlock->setTerminator(Terminator);
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002342 Block = SwitchTerminatedBlock;
Ted Kremenek6b501eb2009-12-24 00:39:26 +00002343 Block = addStmt(Terminator->getCond());
Ted Kremenekad5a8942010-08-02 23:46:59 +00002344
Ted Kremenek6b501eb2009-12-24 00:39:26 +00002345 // Finally, if the SwitchStmt contains a condition variable, add both the
2346 // SwitchStmt and the condition variable initialization to the CFG.
2347 if (VarDecl *VD = Terminator->getConditionVariable()) {
2348 if (Expr *Init = VD->getInit()) {
2349 autoCreateBlock();
Ted Kremenekd40066b2011-04-04 23:29:12 +00002350 appendStmt(Block, Terminator->getConditionVariableDeclStmt());
Ted Kremenek6b501eb2009-12-24 00:39:26 +00002351 addStmt(Init);
2352 }
2353 }
Ted Kremenekad5a8942010-08-02 23:46:59 +00002354
Ted Kremenek6b501eb2009-12-24 00:39:26 +00002355 return Block;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002356}
Ted Kremeneke71f3d52011-03-01 23:12:55 +00002357
2358static bool shouldAddCase(bool &switchExclusivelyCovered,
Ted Kremeneke9cd9c02011-03-13 03:48:04 +00002359 const Expr::EvalResult *switchCond,
Ted Kremeneke71f3d52011-03-01 23:12:55 +00002360 const CaseStmt *CS,
2361 ASTContext &Ctx) {
Ted Kremeneke9cd9c02011-03-13 03:48:04 +00002362 if (!switchCond)
2363 return true;
2364
Ted Kremeneke71f3d52011-03-01 23:12:55 +00002365 bool addCase = false;
Ted Kremenek04982472011-03-04 01:03:41 +00002366
Ted Kremeneke71f3d52011-03-01 23:12:55 +00002367 if (!switchExclusivelyCovered) {
Ted Kremeneke9cd9c02011-03-13 03:48:04 +00002368 if (switchCond->Val.isInt()) {
Ted Kremeneke71f3d52011-03-01 23:12:55 +00002369 // Evaluate the LHS of the case value.
Richard Smith85df96c2011-10-14 20:22:00 +00002370 const llvm::APSInt &lhsInt = CS->getLHS()->EvaluateKnownConstInt(Ctx);
Ted Kremeneke9cd9c02011-03-13 03:48:04 +00002371 const llvm::APSInt &condInt = switchCond->Val.getInt();
Ted Kremeneke71f3d52011-03-01 23:12:55 +00002372
2373 if (condInt == lhsInt) {
2374 addCase = true;
2375 switchExclusivelyCovered = true;
2376 }
2377 else if (condInt < lhsInt) {
2378 if (const Expr *RHS = CS->getRHS()) {
2379 // Evaluate the RHS of the case value.
Richard Smith85df96c2011-10-14 20:22:00 +00002380 const llvm::APSInt &V2 = RHS->EvaluateKnownConstInt(Ctx);
2381 if (V2 <= condInt) {
Ted Kremeneke71f3d52011-03-01 23:12:55 +00002382 addCase = true;
2383 switchExclusivelyCovered = true;
2384 }
2385 }
2386 }
2387 }
2388 else
2389 addCase = true;
2390 }
2391 return addCase;
2392}
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002393
Ted Kremenek9c378f72011-08-12 23:37:29 +00002394CFGBlock *CFGBuilder::VisitCaseStmt(CaseStmt *CS) {
Mike Stump6d9828c2009-07-17 01:31:16 +00002395 // CaseStmts are essentially labels, so they are the first statement in a
2396 // block.
Ted Kremenek0fc67e22010-08-04 23:54:30 +00002397 CFGBlock *TopBlock = 0, *LastBlock = 0;
Ted Kremenek04982472011-03-04 01:03:41 +00002398
Ted Kremenek0fc67e22010-08-04 23:54:30 +00002399 if (Stmt *Sub = CS->getSubStmt()) {
2400 // For deeply nested chains of CaseStmts, instead of doing a recursion
2401 // (which can blow out the stack), manually unroll and create blocks
2402 // along the way.
2403 while (isa<CaseStmt>(Sub)) {
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002404 CFGBlock *currentBlock = createBlock(false);
2405 currentBlock->setLabel(CS);
Ted Kremenek29ccaa12007-08-30 18:48:11 +00002406
Ted Kremenek0fc67e22010-08-04 23:54:30 +00002407 if (TopBlock)
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002408 addSuccessor(LastBlock, currentBlock);
Ted Kremenek0fc67e22010-08-04 23:54:30 +00002409 else
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002410 TopBlock = currentBlock;
Ted Kremenek0fc67e22010-08-04 23:54:30 +00002411
Ted Kremeneke71f3d52011-03-01 23:12:55 +00002412 addSuccessor(SwitchTerminatedBlock,
Ted Kremeneke9cd9c02011-03-13 03:48:04 +00002413 shouldAddCase(switchExclusivelyCovered, switchCond,
Ted Kremeneke71f3d52011-03-01 23:12:55 +00002414 CS, *Context)
2415 ? currentBlock : 0);
Ted Kremenek0fc67e22010-08-04 23:54:30 +00002416
Ted Kremeneke71f3d52011-03-01 23:12:55 +00002417 LastBlock = currentBlock;
Ted Kremenek0fc67e22010-08-04 23:54:30 +00002418 CS = cast<CaseStmt>(Sub);
2419 Sub = CS->getSubStmt();
2420 }
2421
2422 addStmt(Sub);
2423 }
Mike Stump1eb44332009-09-09 15:08:12 +00002424
Ted Kremenek9c378f72011-08-12 23:37:29 +00002425 CFGBlock *CaseBlock = Block;
Ted Kremenek4f880632009-07-17 22:18:43 +00002426 if (!CaseBlock)
2427 CaseBlock = createBlock();
Mike Stump6d9828c2009-07-17 01:31:16 +00002428
2429 // Cases statements partition blocks, so this is the top of the basic block we
2430 // were processing (the "case XXX:" is the label).
Ted Kremenek4f880632009-07-17 22:18:43 +00002431 CaseBlock->setLabel(CS);
2432
Zhongxing Xud438b3d2010-09-06 07:32:31 +00002433 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00002434 return 0;
Mike Stump6d9828c2009-07-17 01:31:16 +00002435
2436 // Add this block to the list of successors for the block with the switch
2437 // statement.
Ted Kremenek4f880632009-07-17 22:18:43 +00002438 assert(SwitchTerminatedBlock);
Ted Kremeneke71f3d52011-03-01 23:12:55 +00002439 addSuccessor(SwitchTerminatedBlock,
Ted Kremeneke9cd9c02011-03-13 03:48:04 +00002440 shouldAddCase(switchExclusivelyCovered, switchCond,
Ted Kremeneke71f3d52011-03-01 23:12:55 +00002441 CS, *Context)
2442 ? CaseBlock : 0);
Mike Stump6d9828c2009-07-17 01:31:16 +00002443
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002444 // We set Block to NULL to allow lazy creation of a new block (if necessary)
2445 Block = NULL;
Mike Stump6d9828c2009-07-17 01:31:16 +00002446
Ted Kremenek0fc67e22010-08-04 23:54:30 +00002447 if (TopBlock) {
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002448 addSuccessor(LastBlock, CaseBlock);
Ted Kremenek0fc67e22010-08-04 23:54:30 +00002449 Succ = TopBlock;
Zhanyong Wan36f327c2010-11-22 19:32:14 +00002450 } else {
Ted Kremenek0fc67e22010-08-04 23:54:30 +00002451 // This block is now the implicit successor of other blocks.
2452 Succ = CaseBlock;
2453 }
Mike Stump6d9828c2009-07-17 01:31:16 +00002454
Ted Kremenek0fc67e22010-08-04 23:54:30 +00002455 return Succ;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002456}
Mike Stump6d9828c2009-07-17 01:31:16 +00002457
Ted Kremenek9c378f72011-08-12 23:37:29 +00002458CFGBlock *CFGBuilder::VisitDefaultStmt(DefaultStmt *Terminator) {
Ted Kremenek4f880632009-07-17 22:18:43 +00002459 if (Terminator->getSubStmt())
2460 addStmt(Terminator->getSubStmt());
Mike Stump1eb44332009-09-09 15:08:12 +00002461
Ted Kremenekeef5a9a2008-02-13 22:05:39 +00002462 DefaultCaseBlock = Block;
Ted Kremenek4f880632009-07-17 22:18:43 +00002463
2464 if (!DefaultCaseBlock)
2465 DefaultCaseBlock = createBlock();
Mike Stump6d9828c2009-07-17 01:31:16 +00002466
2467 // Default statements partition blocks, so this is the top of the basic block
2468 // we were processing (the "default:" is the label).
Ted Kremenek411cdee2008-04-16 21:10:48 +00002469 DefaultCaseBlock->setLabel(Terminator);
Mike Stump1eb44332009-09-09 15:08:12 +00002470
Zhongxing Xud438b3d2010-09-06 07:32:31 +00002471 if (badCFG)
Ted Kremenek4e8df2e2009-05-02 00:13:27 +00002472 return 0;
Ted Kremenekeef5a9a2008-02-13 22:05:39 +00002473
Mike Stump6d9828c2009-07-17 01:31:16 +00002474 // Unlike case statements, we don't add the default block to the successors
2475 // for the switch statement immediately. This is done when we finish
2476 // processing the switch statement. This allows for the default case
2477 // (including a fall-through to the code after the switch statement) to always
2478 // be the last successor of a switch-terminated block.
2479
Ted Kremenekeef5a9a2008-02-13 22:05:39 +00002480 // We set Block to NULL to allow lazy creation of a new block (if necessary)
2481 Block = NULL;
Mike Stump6d9828c2009-07-17 01:31:16 +00002482
Ted Kremenekeef5a9a2008-02-13 22:05:39 +00002483 // This block is now the implicit successor of other blocks.
2484 Succ = DefaultCaseBlock;
Mike Stump6d9828c2009-07-17 01:31:16 +00002485
2486 return DefaultCaseBlock;
Ted Kremenek295222c2008-02-13 21:46:34 +00002487}
Ted Kremenekd4fdee32007-08-23 21:42:29 +00002488
Mike Stump5d1d2022010-01-19 02:20:09 +00002489CFGBlock *CFGBuilder::VisitCXXTryStmt(CXXTryStmt *Terminator) {
2490 // "try"/"catch" is a control-flow statement. Thus we stop processing the
2491 // current block.
Ted Kremenek9c378f72011-08-12 23:37:29 +00002492 CFGBlock *TrySuccessor = NULL;
Mike Stump5d1d2022010-01-19 02:20:09 +00002493
2494 if (Block) {
Zhongxing Xud438b3d2010-09-06 07:32:31 +00002495 if (badCFG)
Mike Stump5d1d2022010-01-19 02:20:09 +00002496 return 0;
2497 TrySuccessor = Block;
2498 } else TrySuccessor = Succ;
2499
Mike Stumpa1f93632010-01-20 01:15:34 +00002500 CFGBlock *PrevTryTerminatedBlock = TryTerminatedBlock;
Mike Stump5d1d2022010-01-19 02:20:09 +00002501
2502 // Create a new block that will contain the try statement.
Mike Stumpf00cca52010-01-20 01:30:58 +00002503 CFGBlock *NewTryTerminatedBlock = createBlock(false);
Mike Stump5d1d2022010-01-19 02:20:09 +00002504 // Add the terminator in the try block.
Mike Stumpf00cca52010-01-20 01:30:58 +00002505 NewTryTerminatedBlock->setTerminator(Terminator);
Mike Stump5d1d2022010-01-19 02:20:09 +00002506
Mike Stumpa1f93632010-01-20 01:15:34 +00002507 bool HasCatchAll = false;
Mike Stump5d1d2022010-01-19 02:20:09 +00002508 for (unsigned h = 0; h <Terminator->getNumHandlers(); ++h) {
2509 // The code after the try is the implicit successor.
2510 Succ = TrySuccessor;
2511 CXXCatchStmt *CS = Terminator->getHandler(h);
Mike Stumpa1f93632010-01-20 01:15:34 +00002512 if (CS->getExceptionDecl() == 0) {
2513 HasCatchAll = true;
2514 }
Mike Stump5d1d2022010-01-19 02:20:09 +00002515 Block = NULL;
2516 CFGBlock *CatchBlock = VisitCXXCatchStmt(CS);
2517 if (CatchBlock == 0)
2518 return 0;
2519 // Add this block to the list of successors for the block with the try
2520 // statement.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002521 addSuccessor(NewTryTerminatedBlock, CatchBlock);
Mike Stump5d1d2022010-01-19 02:20:09 +00002522 }
Mike Stumpa1f93632010-01-20 01:15:34 +00002523 if (!HasCatchAll) {
2524 if (PrevTryTerminatedBlock)
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002525 addSuccessor(NewTryTerminatedBlock, PrevTryTerminatedBlock);
Mike Stumpa1f93632010-01-20 01:15:34 +00002526 else
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002527 addSuccessor(NewTryTerminatedBlock, &cfg->getExit());
Mike Stumpa1f93632010-01-20 01:15:34 +00002528 }
Mike Stump5d1d2022010-01-19 02:20:09 +00002529
2530 // The code after the try is the implicit successor.
2531 Succ = TrySuccessor;
2532
Mike Stumpf00cca52010-01-20 01:30:58 +00002533 // Save the current "try" context.
Ted Kremenekf0e71ae2011-08-23 23:05:07 +00002534 SaveAndRestore<CFGBlock*> save_try(TryTerminatedBlock, NewTryTerminatedBlock);
2535 cfg->addTryDispatchBlock(TryTerminatedBlock);
Mike Stumpf00cca52010-01-20 01:30:58 +00002536
Ted Kremenek6db0ad32010-01-19 20:46:35 +00002537 assert(Terminator->getTryBlock() && "try must contain a non-NULL body");
Mike Stump5d1d2022010-01-19 02:20:09 +00002538 Block = NULL;
Ted Kremenek3fa1e4b2010-01-19 20:52:05 +00002539 Block = addStmt(Terminator->getTryBlock());
Mike Stump5d1d2022010-01-19 02:20:09 +00002540 return Block;
2541}
2542
Ted Kremenek9c378f72011-08-12 23:37:29 +00002543CFGBlock *CFGBuilder::VisitCXXCatchStmt(CXXCatchStmt *CS) {
Mike Stump5d1d2022010-01-19 02:20:09 +00002544 // CXXCatchStmt are treated like labels, so they are the first statement in a
2545 // block.
2546
Marcin Swiderski0e97bcb2010-10-01 01:46:52 +00002547 // Save local scope position because in case of exception variable ScopePos
2548 // won't be restored when traversing AST.
2549 SaveAndRestore<LocalScope::const_iterator> save_scope_pos(ScopePos);
2550
2551 // Create local scope for possible exception variable.
2552 // Store scope position. Add implicit destructor.
Ted Kremenek9c378f72011-08-12 23:37:29 +00002553 if (VarDecl *VD = CS->getExceptionDecl()) {
Marcin Swiderski0e97bcb2010-10-01 01:46:52 +00002554 LocalScope::const_iterator BeginScopePos = ScopePos;
2555 addLocalScopeForVarDecl(VD);
2556 addAutomaticObjDtors(ScopePos, BeginScopePos, CS);
2557 }
2558
Mike Stump5d1d2022010-01-19 02:20:09 +00002559 if (CS->getHandlerBlock())
2560 addStmt(CS->getHandlerBlock());
2561
Ted Kremenek9c378f72011-08-12 23:37:29 +00002562 CFGBlock *CatchBlock = Block;
Mike Stump5d1d2022010-01-19 02:20:09 +00002563 if (!CatchBlock)
2564 CatchBlock = createBlock();
2565
2566 CatchBlock->setLabel(CS);
2567
Zhongxing Xud438b3d2010-09-06 07:32:31 +00002568 if (badCFG)
Mike Stump5d1d2022010-01-19 02:20:09 +00002569 return 0;
2570
2571 // We set Block to NULL to allow lazy creation of a new block (if necessary)
2572 Block = NULL;
2573
2574 return CatchBlock;
2575}
2576
Ted Kremenek9c378f72011-08-12 23:37:29 +00002577CFGBlock *CFGBuilder::VisitCXXForRangeStmt(CXXForRangeStmt *S) {
Richard Smithad762fc2011-04-14 22:09:26 +00002578 // C++0x for-range statements are specified as [stmt.ranged]:
2579 //
2580 // {
2581 // auto && __range = range-init;
2582 // for ( auto __begin = begin-expr,
2583 // __end = end-expr;
2584 // __begin != __end;
2585 // ++__begin ) {
2586 // for-range-declaration = *__begin;
2587 // statement
2588 // }
2589 // }
2590
2591 // Save local scope position before the addition of the implicit variables.
2592 SaveAndRestore<LocalScope::const_iterator> save_scope_pos(ScopePos);
2593
2594 // Create local scopes and destructors for range, begin and end variables.
2595 if (Stmt *Range = S->getRangeStmt())
2596 addLocalScopeForStmt(Range);
2597 if (Stmt *BeginEnd = S->getBeginEndStmt())
2598 addLocalScopeForStmt(BeginEnd);
2599 addAutomaticObjDtors(ScopePos, save_scope_pos.get(), S);
2600
2601 LocalScope::const_iterator ContinueScopePos = ScopePos;
2602
2603 // "for" is a control-flow statement. Thus we stop processing the current
2604 // block.
Ted Kremenek9c378f72011-08-12 23:37:29 +00002605 CFGBlock *LoopSuccessor = NULL;
Richard Smithad762fc2011-04-14 22:09:26 +00002606 if (Block) {
2607 if (badCFG)
2608 return 0;
2609 LoopSuccessor = Block;
2610 } else
2611 LoopSuccessor = Succ;
2612
2613 // Save the current value for the break targets.
2614 // All breaks should go to the code following the loop.
2615 SaveAndRestore<JumpTarget> save_break(BreakJumpTarget);
2616 BreakJumpTarget = JumpTarget(LoopSuccessor, ScopePos);
2617
2618 // The block for the __begin != __end expression.
Ted Kremenek9c378f72011-08-12 23:37:29 +00002619 CFGBlock *ConditionBlock = createBlock(false);
Richard Smithad762fc2011-04-14 22:09:26 +00002620 ConditionBlock->setTerminator(S);
2621
2622 // Now add the actual condition to the condition block.
2623 if (Expr *C = S->getCond()) {
2624 Block = ConditionBlock;
2625 CFGBlock *BeginConditionBlock = addStmt(C);
2626 if (badCFG)
2627 return 0;
2628 assert(BeginConditionBlock == ConditionBlock &&
2629 "condition block in for-range was unexpectedly complex");
2630 (void)BeginConditionBlock;
2631 }
2632
2633 // The condition block is the implicit successor for the loop body as well as
2634 // any code above the loop.
2635 Succ = ConditionBlock;
2636
2637 // See if this is a known constant.
2638 TryResult KnownVal(true);
2639
2640 if (S->getCond())
2641 KnownVal = tryEvaluateBool(S->getCond());
2642
2643 // Now create the loop body.
2644 {
2645 assert(S->getBody());
2646
2647 // Save the current values for Block, Succ, and continue targets.
2648 SaveAndRestore<CFGBlock*> save_Block(Block), save_Succ(Succ);
2649 SaveAndRestore<JumpTarget> save_continue(ContinueJumpTarget);
2650
2651 // Generate increment code in its own basic block. This is the target of
2652 // continue statements.
2653 Block = 0;
2654 Succ = addStmt(S->getInc());
2655 ContinueJumpTarget = JumpTarget(Succ, ContinueScopePos);
2656
2657 // The starting block for the loop increment is the block that should
2658 // represent the 'loop target' for looping back to the start of the loop.
2659 ContinueJumpTarget.block->setLoopTarget(S);
2660
2661 // Finish up the increment block and prepare to start the loop body.
2662 assert(Block);
2663 if (badCFG)
2664 return 0;
2665 Block = 0;
2666
2667
2668 // Add implicit scope and dtors for loop variable.
2669 addLocalScopeAndDtors(S->getLoopVarStmt());
2670
2671 // Populate a new block to contain the loop body and loop variable.
2672 Block = addStmt(S->getBody());
2673 if (badCFG)
2674 return 0;
2675 Block = addStmt(S->getLoopVarStmt());
2676 if (badCFG)
2677 return 0;
2678
2679 // This new body block is a successor to our condition block.
2680 addSuccessor(ConditionBlock, KnownVal.isFalse() ? 0 : Block);
2681 }
2682
2683 // Link up the condition block with the code that follows the loop (the
2684 // false branch).
2685 addSuccessor(ConditionBlock, KnownVal.isTrue() ? 0 : LoopSuccessor);
2686
2687 // Add the initialization statements.
2688 Block = createBlock();
Richard Smithb403d6d2011-04-18 15:49:25 +00002689 addStmt(S->getBeginEndStmt());
2690 return addStmt(S->getRangeStmt());
Richard Smithad762fc2011-04-14 22:09:26 +00002691}
2692
John McCall4765fa02010-12-06 08:20:24 +00002693CFGBlock *CFGBuilder::VisitExprWithCleanups(ExprWithCleanups *E,
Marcin Swiderski8599e762010-11-03 06:19:35 +00002694 AddStmtChoice asc) {
2695 if (BuildOpts.AddImplicitDtors) {
2696 // If adding implicit destructors visit the full expression for adding
2697 // destructors of temporaries.
2698 VisitForTemporaryDtors(E->getSubExpr());
2699
2700 // Full expression has to be added as CFGStmt so it will be sequenced
2701 // before destructors of it's temporaries.
Zhanyong Wan94a3dcf2010-11-24 03:28:53 +00002702 asc = asc.withAlwaysAdd(true);
Marcin Swiderski8599e762010-11-03 06:19:35 +00002703 }
2704 return Visit(E->getSubExpr(), asc);
2705}
2706
Zhongxing Xua725ed42010-11-01 13:04:58 +00002707CFGBlock *CFGBuilder::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E,
2708 AddStmtChoice asc) {
Ted Kremenek3179a452011-03-10 01:14:11 +00002709 if (asc.alwaysAdd(*this, E)) {
Zhongxing Xua725ed42010-11-01 13:04:58 +00002710 autoCreateBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00002711 appendStmt(Block, E);
Zhongxing Xua725ed42010-11-01 13:04:58 +00002712
2713 // We do not want to propagate the AlwaysAdd property.
Zhanyong Wan94a3dcf2010-11-24 03:28:53 +00002714 asc = asc.withAlwaysAdd(false);
Zhongxing Xua725ed42010-11-01 13:04:58 +00002715 }
2716 return Visit(E->getSubExpr(), asc);
2717}
2718
Zhongxing Xu81bc7d02010-11-01 06:46:05 +00002719CFGBlock *CFGBuilder::VisitCXXConstructExpr(CXXConstructExpr *C,
2720 AddStmtChoice asc) {
Zhongxing Xu81bc7d02010-11-01 06:46:05 +00002721 autoCreateBlock();
Zhongxing Xu3ff5b262010-11-03 11:14:06 +00002722 if (!C->isElidable())
Ted Kremenek247e9662011-03-10 01:14:08 +00002723 appendStmt(Block, C);
Zhanyong Wan94a3dcf2010-11-24 03:28:53 +00002724
Zhongxing Xu81bc7d02010-11-01 06:46:05 +00002725 return VisitChildren(C);
2726}
2727
Zhongxing Xua725ed42010-11-01 13:04:58 +00002728CFGBlock *CFGBuilder::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E,
2729 AddStmtChoice asc) {
Ted Kremenek3179a452011-03-10 01:14:11 +00002730 if (asc.alwaysAdd(*this, E)) {
Zhongxing Xua725ed42010-11-01 13:04:58 +00002731 autoCreateBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00002732 appendStmt(Block, E);
Zhongxing Xua725ed42010-11-01 13:04:58 +00002733 // We do not want to propagate the AlwaysAdd property.
Zhanyong Wan94a3dcf2010-11-24 03:28:53 +00002734 asc = asc.withAlwaysAdd(false);
Zhongxing Xua725ed42010-11-01 13:04:58 +00002735 }
2736 return Visit(E->getSubExpr(), asc);
2737}
2738
Zhongxing Xu81bc7d02010-11-01 06:46:05 +00002739CFGBlock *CFGBuilder::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *C,
2740 AddStmtChoice asc) {
Zhongxing Xu81bc7d02010-11-01 06:46:05 +00002741 autoCreateBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00002742 appendStmt(Block, C);
Zhongxing Xu81bc7d02010-11-01 06:46:05 +00002743 return VisitChildren(C);
2744}
2745
Zhongxing Xua725ed42010-11-01 13:04:58 +00002746CFGBlock *CFGBuilder::VisitImplicitCastExpr(ImplicitCastExpr *E,
2747 AddStmtChoice asc) {
Ted Kremenek3179a452011-03-10 01:14:11 +00002748 if (asc.alwaysAdd(*this, E)) {
Zhongxing Xua725ed42010-11-01 13:04:58 +00002749 autoCreateBlock();
Ted Kremenek247e9662011-03-10 01:14:08 +00002750 appendStmt(Block, E);
Zhongxing Xua725ed42010-11-01 13:04:58 +00002751 }
Ted Kremenek892697d2010-12-16 07:46:53 +00002752 return Visit(E->getSubExpr(), AddStmtChoice());
Zhongxing Xua725ed42010-11-01 13:04:58 +00002753}
2754
Ted Kremenek9c378f72011-08-12 23:37:29 +00002755CFGBlock *CFGBuilder::VisitIndirectGotoStmt(IndirectGotoStmt *I) {
Mike Stump6d9828c2009-07-17 01:31:16 +00002756 // Lazily create the indirect-goto dispatch block if there isn't one already.
Ted Kremenek9c378f72011-08-12 23:37:29 +00002757 CFGBlock *IBlock = cfg->getIndirectGotoBlock();
Mike Stump6d9828c2009-07-17 01:31:16 +00002758
Ted Kremenek19bb3562007-08-28 19:26:49 +00002759 if (!IBlock) {
2760 IBlock = createBlock(false);
2761 cfg->setIndirectGotoBlock(IBlock);
2762 }
Mike Stump6d9828c2009-07-17 01:31:16 +00002763
Ted Kremenek19bb3562007-08-28 19:26:49 +00002764 // IndirectGoto is a control-flow statement. Thus we stop processing the
2765 // current block and create a new one.
Zhongxing Xud438b3d2010-09-06 07:32:31 +00002766 if (badCFG)
Ted Kremenek4f880632009-07-17 22:18:43 +00002767 return 0;
2768
Ted Kremenek19bb3562007-08-28 19:26:49 +00002769 Block = createBlock(false);
2770 Block->setTerminator(I);
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002771 addSuccessor(Block, IBlock);
Ted Kremenek19bb3562007-08-28 19:26:49 +00002772 return addStmt(I->getTarget());
2773}
2774
Marcin Swiderski8599e762010-11-03 06:19:35 +00002775CFGBlock *CFGBuilder::VisitForTemporaryDtors(Stmt *E, bool BindToTemporary) {
2776tryAgain:
2777 if (!E) {
2778 badCFG = true;
2779 return NULL;
2780 }
2781 switch (E->getStmtClass()) {
2782 default:
2783 return VisitChildrenForTemporaryDtors(E);
2784
2785 case Stmt::BinaryOperatorClass:
2786 return VisitBinaryOperatorForTemporaryDtors(cast<BinaryOperator>(E));
2787
2788 case Stmt::CXXBindTemporaryExprClass:
2789 return VisitCXXBindTemporaryExprForTemporaryDtors(
2790 cast<CXXBindTemporaryExpr>(E), BindToTemporary);
2791
John McCall56ca35d2011-02-17 10:25:35 +00002792 case Stmt::BinaryConditionalOperatorClass:
Marcin Swiderski8599e762010-11-03 06:19:35 +00002793 case Stmt::ConditionalOperatorClass:
2794 return VisitConditionalOperatorForTemporaryDtors(
John McCall56ca35d2011-02-17 10:25:35 +00002795 cast<AbstractConditionalOperator>(E), BindToTemporary);
Marcin Swiderski8599e762010-11-03 06:19:35 +00002796
2797 case Stmt::ImplicitCastExprClass:
2798 // For implicit cast we want BindToTemporary to be passed further.
2799 E = cast<CastExpr>(E)->getSubExpr();
2800 goto tryAgain;
2801
2802 case Stmt::ParenExprClass:
2803 E = cast<ParenExpr>(E)->getSubExpr();
2804 goto tryAgain;
Douglas Gregor03e80032011-06-21 17:03:29 +00002805
2806 case Stmt::MaterializeTemporaryExprClass:
2807 E = cast<MaterializeTemporaryExpr>(E)->GetTemporaryExpr();
2808 goto tryAgain;
Marcin Swiderski8599e762010-11-03 06:19:35 +00002809 }
2810}
2811
2812CFGBlock *CFGBuilder::VisitChildrenForTemporaryDtors(Stmt *E) {
2813 // When visiting children for destructors we want to visit them in reverse
2814 // order. Because there's no reverse iterator for children must to reverse
2815 // them in helper vector.
Chris Lattner5f9e2722011-07-23 10:55:15 +00002816 typedef SmallVector<Stmt *, 4> ChildrenVect;
Marcin Swiderski8599e762010-11-03 06:19:35 +00002817 ChildrenVect ChildrenRev;
John McCall7502c1d2011-02-13 04:07:26 +00002818 for (Stmt::child_range I = E->children(); I; ++I) {
Marcin Swiderski8599e762010-11-03 06:19:35 +00002819 if (*I) ChildrenRev.push_back(*I);
2820 }
2821
2822 CFGBlock *B = Block;
2823 for (ChildrenVect::reverse_iterator I = ChildrenRev.rbegin(),
2824 L = ChildrenRev.rend(); I != L; ++I) {
2825 if (CFGBlock *R = VisitForTemporaryDtors(*I))
2826 B = R;
2827 }
2828 return B;
2829}
2830
2831CFGBlock *CFGBuilder::VisitBinaryOperatorForTemporaryDtors(BinaryOperator *E) {
2832 if (E->isLogicalOp()) {
2833 // Destructors for temporaries in LHS expression should be called after
2834 // those for RHS expression. Even if this will unnecessarily create a block,
2835 // this block will be used at least by the full expression.
2836 autoCreateBlock();
2837 CFGBlock *ConfluenceBlock = VisitForTemporaryDtors(E->getLHS());
2838 if (badCFG)
2839 return NULL;
2840
2841 Succ = ConfluenceBlock;
2842 Block = NULL;
2843 CFGBlock *RHSBlock = VisitForTemporaryDtors(E->getRHS());
2844
2845 if (RHSBlock) {
2846 if (badCFG)
2847 return NULL;
2848
2849 // If RHS expression did produce destructors we need to connect created
2850 // blocks to CFG in same manner as for binary operator itself.
2851 CFGBlock *LHSBlock = createBlock(false);
2852 LHSBlock->setTerminator(CFGTerminator(E, true));
2853
2854 // For binary operator LHS block is before RHS in list of predecessors
2855 // of ConfluenceBlock.
2856 std::reverse(ConfluenceBlock->pred_begin(),
2857 ConfluenceBlock->pred_end());
2858
2859 // See if this is a known constant.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002860 TryResult KnownVal = tryEvaluateBool(E->getLHS());
Marcin Swiderski8599e762010-11-03 06:19:35 +00002861 if (KnownVal.isKnown() && (E->getOpcode() == BO_LOr))
2862 KnownVal.negate();
2863
2864 // Link LHSBlock with RHSBlock exactly the same way as for binary operator
2865 // itself.
2866 if (E->getOpcode() == BO_LOr) {
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002867 addSuccessor(LHSBlock, KnownVal.isTrue() ? NULL : ConfluenceBlock);
2868 addSuccessor(LHSBlock, KnownVal.isFalse() ? NULL : RHSBlock);
Marcin Swiderski8599e762010-11-03 06:19:35 +00002869 } else {
2870 assert (E->getOpcode() == BO_LAnd);
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002871 addSuccessor(LHSBlock, KnownVal.isFalse() ? NULL : RHSBlock);
2872 addSuccessor(LHSBlock, KnownVal.isTrue() ? NULL : ConfluenceBlock);
Marcin Swiderski8599e762010-11-03 06:19:35 +00002873 }
2874
2875 Block = LHSBlock;
2876 return LHSBlock;
2877 }
2878
2879 Block = ConfluenceBlock;
2880 return ConfluenceBlock;
2881 }
2882
Zhanyong Wan36f327c2010-11-22 19:32:14 +00002883 if (E->isAssignmentOp()) {
Marcin Swiderski8599e762010-11-03 06:19:35 +00002884 // For assignment operator (=) LHS expression is visited
2885 // before RHS expression. For destructors visit them in reverse order.
2886 CFGBlock *RHSBlock = VisitForTemporaryDtors(E->getRHS());
2887 CFGBlock *LHSBlock = VisitForTemporaryDtors(E->getLHS());
2888 return LHSBlock ? LHSBlock : RHSBlock;
2889 }
2890
2891 // For any other binary operator RHS expression is visited before
2892 // LHS expression (order of children). For destructors visit them in reverse
2893 // order.
2894 CFGBlock *LHSBlock = VisitForTemporaryDtors(E->getLHS());
2895 CFGBlock *RHSBlock = VisitForTemporaryDtors(E->getRHS());
2896 return RHSBlock ? RHSBlock : LHSBlock;
2897}
2898
2899CFGBlock *CFGBuilder::VisitCXXBindTemporaryExprForTemporaryDtors(
2900 CXXBindTemporaryExpr *E, bool BindToTemporary) {
2901 // First add destructors for temporaries in subexpression.
2902 CFGBlock *B = VisitForTemporaryDtors(E->getSubExpr());
Zhongxing Xu249c9452010-11-14 15:23:50 +00002903 if (!BindToTemporary) {
Marcin Swiderski8599e762010-11-03 06:19:35 +00002904 // If lifetime of temporary is not prolonged (by assigning to constant
2905 // reference) add destructor for it.
Chandler Carruthc8cfc742011-09-13 06:09:01 +00002906
2907 // If the destructor is marked as a no-return destructor, we need to create
2908 // a new block for the destructor which does not have as a successor
2909 // anything built thus far. Control won't flow out of this block.
2910 const CXXDestructorDecl *Dtor = E->getTemporary()->getDestructor();
Chandler Carruthdba3fb52011-09-13 09:13:49 +00002911 if (cast<FunctionType>(Dtor->getType())->getNoReturnAttr())
2912 Block = createNoReturnBlock();
2913 else
Chandler Carruthc8cfc742011-09-13 06:09:01 +00002914 autoCreateBlock();
Chandler Carruthc8cfc742011-09-13 06:09:01 +00002915
Marcin Swiderski8599e762010-11-03 06:19:35 +00002916 appendTemporaryDtor(Block, E);
2917 B = Block;
2918 }
2919 return B;
2920}
2921
2922CFGBlock *CFGBuilder::VisitConditionalOperatorForTemporaryDtors(
John McCall56ca35d2011-02-17 10:25:35 +00002923 AbstractConditionalOperator *E, bool BindToTemporary) {
Marcin Swiderski8599e762010-11-03 06:19:35 +00002924 // First add destructors for condition expression. Even if this will
2925 // unnecessarily create a block, this block will be used at least by the full
2926 // expression.
2927 autoCreateBlock();
2928 CFGBlock *ConfluenceBlock = VisitForTemporaryDtors(E->getCond());
2929 if (badCFG)
2930 return NULL;
John McCall56ca35d2011-02-17 10:25:35 +00002931 if (BinaryConditionalOperator *BCO
2932 = dyn_cast<BinaryConditionalOperator>(E)) {
2933 ConfluenceBlock = VisitForTemporaryDtors(BCO->getCommon());
Marcin Swiderski8599e762010-11-03 06:19:35 +00002934 if (badCFG)
2935 return NULL;
2936 }
2937
John McCall56ca35d2011-02-17 10:25:35 +00002938 // Try to add block with destructors for LHS expression.
2939 CFGBlock *LHSBlock = NULL;
2940 Succ = ConfluenceBlock;
2941 Block = NULL;
2942 LHSBlock = VisitForTemporaryDtors(E->getTrueExpr(), BindToTemporary);
2943 if (badCFG)
2944 return NULL;
2945
Marcin Swiderski8599e762010-11-03 06:19:35 +00002946 // Try to add block with destructors for RHS expression;
2947 Succ = ConfluenceBlock;
2948 Block = NULL;
John McCall56ca35d2011-02-17 10:25:35 +00002949 CFGBlock *RHSBlock = VisitForTemporaryDtors(E->getFalseExpr(),
2950 BindToTemporary);
Marcin Swiderski8599e762010-11-03 06:19:35 +00002951 if (badCFG)
2952 return NULL;
2953
2954 if (!RHSBlock && !LHSBlock) {
2955 // If neither LHS nor RHS expression had temporaries to destroy don't create
2956 // more blocks.
2957 Block = ConfluenceBlock;
2958 return Block;
2959 }
2960
2961 Block = createBlock(false);
2962 Block->setTerminator(CFGTerminator(E, true));
2963
2964 // See if this is a known constant.
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002965 const TryResult &KnownVal = tryEvaluateBool(E->getCond());
Marcin Swiderski8599e762010-11-03 06:19:35 +00002966
2967 if (LHSBlock) {
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002968 addSuccessor(Block, KnownVal.isFalse() ? NULL : LHSBlock);
Marcin Swiderski8599e762010-11-03 06:19:35 +00002969 } else if (KnownVal.isFalse()) {
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002970 addSuccessor(Block, NULL);
Marcin Swiderski8599e762010-11-03 06:19:35 +00002971 } else {
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002972 addSuccessor(Block, ConfluenceBlock);
Marcin Swiderski8599e762010-11-03 06:19:35 +00002973 std::reverse(ConfluenceBlock->pred_begin(), ConfluenceBlock->pred_end());
2974 }
2975
2976 if (!RHSBlock)
2977 RHSBlock = ConfluenceBlock;
Ted Kremenek0a3ed312010-12-17 04:44:39 +00002978 addSuccessor(Block, KnownVal.isTrue() ? NULL : RHSBlock);
Marcin Swiderski8599e762010-11-03 06:19:35 +00002979
2980 return Block;
2981}
2982
Ted Kremenekbefef2f2007-08-23 21:26:19 +00002983} // end anonymous namespace
Ted Kremenek026473c2007-08-23 16:51:22 +00002984
Mike Stump6d9828c2009-07-17 01:31:16 +00002985/// createBlock - Constructs and adds a new CFGBlock to the CFG. The block has
2986/// no successors or predecessors. If this is the first block created in the
2987/// CFG, it is automatically set to be the Entry and Exit of the CFG.
Ted Kremenek9c378f72011-08-12 23:37:29 +00002988CFGBlock *CFG::createBlock() {
Ted Kremenek026473c2007-08-23 16:51:22 +00002989 bool first_block = begin() == end();
2990
2991 // Create the block.
Ted Kremenekee82d9b2009-10-12 20:55:07 +00002992 CFGBlock *Mem = getAllocator().Allocate<CFGBlock>();
2993 new (Mem) CFGBlock(NumBlockIDs++, BlkBVC);
2994 Blocks.push_back(Mem, BlkBVC);
Ted Kremenek026473c2007-08-23 16:51:22 +00002995
2996 // If this is the first block, set it as the Entry and Exit.
Ted Kremenekee82d9b2009-10-12 20:55:07 +00002997 if (first_block)
2998 Entry = Exit = &back();
Ted Kremenek026473c2007-08-23 16:51:22 +00002999
3000 // Return the block.
Ted Kremenekee82d9b2009-10-12 20:55:07 +00003001 return &back();
Ted Kremenekfddd5182007-08-21 21:42:03 +00003002}
3003
Ted Kremenek026473c2007-08-23 16:51:22 +00003004/// buildCFG - Constructs a CFG from an AST. Ownership of the returned
3005/// CFG is returned to the caller.
Ted Kremenek9c378f72011-08-12 23:37:29 +00003006CFG* CFG::buildCFG(const Decl *D, Stmt *Statement, ASTContext *C,
Ted Kremenekb8ad5ee2011-03-10 01:14:05 +00003007 const BuildOptions &BO) {
3008 CFGBuilder Builder(C, BO);
3009 return Builder.buildCFG(D, Statement);
Ted Kremenek026473c2007-08-23 16:51:22 +00003010}
3011
Ted Kremenekc5aff442011-03-03 01:21:32 +00003012const CXXDestructorDecl *
3013CFGImplicitDtor::getDestructorDecl(ASTContext &astContext) const {
Ted Kremenekc9f8f5a2011-03-02 20:32:29 +00003014 switch (getKind()) {
3015 case CFGElement::Invalid:
3016 case CFGElement::Statement:
3017 case CFGElement::Initializer:
3018 llvm_unreachable("getDestructorDecl should only be used with "
3019 "ImplicitDtors");
3020 case CFGElement::AutomaticObjectDtor: {
3021 const VarDecl *var = cast<CFGAutomaticObjDtor>(this)->getVarDecl();
3022 QualType ty = var->getType();
Ted Kremenek697d42d2011-03-03 01:01:03 +00003023 ty = ty.getNonReferenceType();
Ted Kremenekc5aff442011-03-03 01:21:32 +00003024 if (const ArrayType *arrayType = astContext.getAsArrayType(ty)) {
3025 ty = arrayType->getElementType();
3026 }
Ted Kremenekc9f8f5a2011-03-02 20:32:29 +00003027 const RecordType *recordType = ty->getAs<RecordType>();
3028 const CXXRecordDecl *classDecl =
Ted Kremenek697d42d2011-03-03 01:01:03 +00003029 cast<CXXRecordDecl>(recordType->getDecl());
Ted Kremenekc9f8f5a2011-03-02 20:32:29 +00003030 return classDecl->getDestructor();
3031 }
3032 case CFGElement::TemporaryDtor: {
3033 const CXXBindTemporaryExpr *bindExpr =
3034 cast<CFGTemporaryDtor>(this)->getBindTemporaryExpr();
3035 const CXXTemporary *temp = bindExpr->getTemporary();
3036 return temp->getDestructor();
3037 }
3038 case CFGElement::BaseDtor:
3039 case CFGElement::MemberDtor:
3040
3041 // Not yet supported.
3042 return 0;
3043 }
Ted Kremenek697d42d2011-03-03 01:01:03 +00003044 llvm_unreachable("getKind() returned bogus value");
Matt Beaumont-Gay2ae00542011-03-03 00:48:05 +00003045 return 0;
Ted Kremenekc9f8f5a2011-03-02 20:32:29 +00003046}
3047
Ted Kremenekc5aff442011-03-03 01:21:32 +00003048bool CFGImplicitDtor::isNoReturn(ASTContext &astContext) const {
3049 if (const CXXDestructorDecl *cdecl = getDestructorDecl(astContext)) {
Ted Kremenekc9f8f5a2011-03-02 20:32:29 +00003050 QualType ty = cdecl->getType();
3051 return cast<FunctionType>(ty)->getNoReturnAttr();
3052 }
3053 return false;
Ted Kremenek3c0349e2011-03-01 03:15:10 +00003054}
3055
Ted Kremenek63f58872007-10-01 19:33:33 +00003056//===----------------------------------------------------------------------===//
3057// CFG: Queries for BlkExprs.
3058//===----------------------------------------------------------------------===//
Ted Kremenek7dba8602007-08-29 21:56:09 +00003059
Ted Kremenek63f58872007-10-01 19:33:33 +00003060namespace {
Ted Kremenek86946742008-01-17 20:48:37 +00003061 typedef llvm::DenseMap<const Stmt*,unsigned> BlkExprMapTy;
Ted Kremenek63f58872007-10-01 19:33:33 +00003062}
3063
Ted Kremenekf1d10d92011-08-23 23:05:04 +00003064static void FindSubExprAssignments(const Stmt *S,
3065 llvm::SmallPtrSet<const Expr*,50>& Set) {
Ted Kremenek8a693662009-12-23 23:37:10 +00003066 if (!S)
Ted Kremenek33d4aab2008-01-26 00:03:27 +00003067 return;
Mike Stump6d9828c2009-07-17 01:31:16 +00003068
Ted Kremenekf1d10d92011-08-23 23:05:04 +00003069 for (Stmt::const_child_range I = S->children(); I; ++I) {
3070 const Stmt *child = *I;
Ted Kremenek8a693662009-12-23 23:37:10 +00003071 if (!child)
3072 continue;
Ted Kremenekad5a8942010-08-02 23:46:59 +00003073
Ted Kremenekf1d10d92011-08-23 23:05:04 +00003074 if (const BinaryOperator* B = dyn_cast<BinaryOperator>(child))
Ted Kremenek33d4aab2008-01-26 00:03:27 +00003075 if (B->isAssignmentOp()) Set.insert(B);
Mike Stump6d9828c2009-07-17 01:31:16 +00003076
Ted Kremenek8a693662009-12-23 23:37:10 +00003077 FindSubExprAssignments(child, Set);
Ted Kremenek33d4aab2008-01-26 00:03:27 +00003078 }
3079}
3080
Ted Kremenek63f58872007-10-01 19:33:33 +00003081static BlkExprMapTy* PopulateBlkExprMap(CFG& cfg) {
3082 BlkExprMapTy* M = new BlkExprMapTy();
Mike Stump6d9828c2009-07-17 01:31:16 +00003083
3084 // Look for assignments that are used as subexpressions. These are the only
3085 // assignments that we want to *possibly* register as a block-level
3086 // expression. Basically, if an assignment occurs both in a subexpression and
3087 // at the block-level, it is a block-level expression.
Ted Kremenekf1d10d92011-08-23 23:05:04 +00003088 llvm::SmallPtrSet<const Expr*,50> SubExprAssignments;
Mike Stump6d9828c2009-07-17 01:31:16 +00003089
Ted Kremenek63f58872007-10-01 19:33:33 +00003090 for (CFG::iterator I=cfg.begin(), E=cfg.end(); I != E; ++I)
Ted Kremenekee82d9b2009-10-12 20:55:07 +00003091 for (CFGBlock::iterator BI=(*I)->begin(), EI=(*I)->end(); BI != EI; ++BI)
Ted Kremenek3c0349e2011-03-01 03:15:10 +00003092 if (const CFGStmt *S = BI->getAs<CFGStmt>())
3093 FindSubExprAssignments(S->getStmt(), SubExprAssignments);
Ted Kremenek86946742008-01-17 20:48:37 +00003094
Ted Kremenek411cdee2008-04-16 21:10:48 +00003095 for (CFG::iterator I=cfg.begin(), E=cfg.end(); I != E; ++I) {
Mike Stump6d9828c2009-07-17 01:31:16 +00003096
3097 // Iterate over the statements again on identify the Expr* and Stmt* at the
3098 // block-level that are block-level expressions.
Ted Kremenek411cdee2008-04-16 21:10:48 +00003099
Zhongxing Xub36cd3e2010-09-16 01:25:47 +00003100 for (CFGBlock::iterator BI=(*I)->begin(), EI=(*I)->end(); BI != EI; ++BI) {
Ted Kremenek3c0349e2011-03-01 03:15:10 +00003101 const CFGStmt *CS = BI->getAs<CFGStmt>();
3102 if (!CS)
Zhongxing Xub36cd3e2010-09-16 01:25:47 +00003103 continue;
Ted Kremenekf1d10d92011-08-23 23:05:04 +00003104 if (const Expr *Exp = dyn_cast<Expr>(CS->getStmt())) {
Jordy Roseac73ea82011-06-10 08:49:37 +00003105 assert((Exp->IgnoreParens() == Exp) && "No parens on block-level exps");
Mike Stump6d9828c2009-07-17 01:31:16 +00003106
Ted Kremenekf1d10d92011-08-23 23:05:04 +00003107 if (const BinaryOperator* B = dyn_cast<BinaryOperator>(Exp)) {
Ted Kremenek33d4aab2008-01-26 00:03:27 +00003108 // Assignment expressions that are not nested within another
Mike Stump6d9828c2009-07-17 01:31:16 +00003109 // expression are really "statements" whose value is never used by
3110 // another expression.
Ted Kremenek411cdee2008-04-16 21:10:48 +00003111 if (B->isAssignmentOp() && !SubExprAssignments.count(Exp))
Ted Kremenek33d4aab2008-01-26 00:03:27 +00003112 continue;
Ted Kremenek9c378f72011-08-12 23:37:29 +00003113 } else if (const StmtExpr *SE = dyn_cast<StmtExpr>(Exp)) {
Mike Stump6d9828c2009-07-17 01:31:16 +00003114 // Special handling for statement expressions. The last statement in
3115 // the statement expression is also a block-level expr.
Ted Kremenek9c378f72011-08-12 23:37:29 +00003116 const CompoundStmt *C = SE->getSubStmt();
Ted Kremenek86946742008-01-17 20:48:37 +00003117 if (!C->body_empty()) {
Jordy Roseac73ea82011-06-10 08:49:37 +00003118 const Stmt *Last = C->body_back();
3119 if (const Expr *LastEx = dyn_cast<Expr>(Last))
3120 Last = LastEx->IgnoreParens();
Ted Kremenek33d4aab2008-01-26 00:03:27 +00003121 unsigned x = M->size();
Jordy Roseac73ea82011-06-10 08:49:37 +00003122 (*M)[Last] = x;
Ted Kremenek86946742008-01-17 20:48:37 +00003123 }
3124 }
Ted Kremeneke2dcd782008-01-25 23:22:27 +00003125
Ted Kremenek33d4aab2008-01-26 00:03:27 +00003126 unsigned x = M->size();
Ted Kremenek411cdee2008-04-16 21:10:48 +00003127 (*M)[Exp] = x;
Ted Kremenek33d4aab2008-01-26 00:03:27 +00003128 }
Zhongxing Xub36cd3e2010-09-16 01:25:47 +00003129 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003130
Ted Kremenek411cdee2008-04-16 21:10:48 +00003131 // Look at terminators. The condition is a block-level expression.
Mike Stump6d9828c2009-07-17 01:31:16 +00003132
Ted Kremenek9c378f72011-08-12 23:37:29 +00003133 Stmt *S = (*I)->getTerminatorCondition();
Mike Stump6d9828c2009-07-17 01:31:16 +00003134
Ted Kremenek390e48b2008-11-12 21:11:49 +00003135 if (S && M->find(S) == M->end()) {
Jordy Roseac73ea82011-06-10 08:49:37 +00003136 unsigned x = M->size();
3137 (*M)[S] = x;
Ted Kremenek411cdee2008-04-16 21:10:48 +00003138 }
3139 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003140
Ted Kremenek63f58872007-10-01 19:33:33 +00003141 return M;
3142}
3143
Ted Kremenek9c378f72011-08-12 23:37:29 +00003144CFG::BlkExprNumTy CFG::getBlkExprNum(const Stmt *S) {
Ted Kremenek86946742008-01-17 20:48:37 +00003145 assert(S != NULL);
Ted Kremenek63f58872007-10-01 19:33:33 +00003146 if (!BlkExprMap) { BlkExprMap = (void*) PopulateBlkExprMap(*this); }
Mike Stump6d9828c2009-07-17 01:31:16 +00003147
Ted Kremenek63f58872007-10-01 19:33:33 +00003148 BlkExprMapTy* M = reinterpret_cast<BlkExprMapTy*>(BlkExprMap);
Ted Kremenek86946742008-01-17 20:48:37 +00003149 BlkExprMapTy::iterator I = M->find(S);
Ted Kremenek3fa1e4b2010-01-19 20:52:05 +00003150 return (I == M->end()) ? CFG::BlkExprNumTy() : CFG::BlkExprNumTy(I->second);
Ted Kremenek63f58872007-10-01 19:33:33 +00003151}
3152
3153unsigned CFG::getNumBlkExprs() {
3154 if (const BlkExprMapTy* M = reinterpret_cast<const BlkExprMapTy*>(BlkExprMap))
3155 return M->size();
Zhanyong Wan36f327c2010-11-22 19:32:14 +00003156
3157 // We assume callers interested in the number of BlkExprs will want
3158 // the map constructed if it doesn't already exist.
3159 BlkExprMap = (void*) PopulateBlkExprMap(*this);
3160 return reinterpret_cast<BlkExprMapTy*>(BlkExprMap)->size();
Ted Kremenek63f58872007-10-01 19:33:33 +00003161}
3162
Ted Kremenek274f4332008-04-28 18:00:46 +00003163//===----------------------------------------------------------------------===//
Ted Kremenekee7f84d2010-09-09 00:06:04 +00003164// Filtered walking of the CFG.
3165//===----------------------------------------------------------------------===//
3166
3167bool CFGBlock::FilterEdge(const CFGBlock::FilterOptions &F,
Ted Kremenekbe39a562010-09-09 02:57:48 +00003168 const CFGBlock *From, const CFGBlock *To) {
Ted Kremenekee7f84d2010-09-09 00:06:04 +00003169
Ted Kremenek6e400352011-03-07 22:04:39 +00003170 if (To && F.IgnoreDefaultsWithCoveredEnums) {
Ted Kremenekee7f84d2010-09-09 00:06:04 +00003171 // If the 'To' has no label or is labeled but the label isn't a
3172 // CaseStmt then filter this edge.
3173 if (const SwitchStmt *S =
Ted Kremenek6e400352011-03-07 22:04:39 +00003174 dyn_cast_or_null<SwitchStmt>(From->getTerminator().getStmt())) {
Ted Kremenekee7f84d2010-09-09 00:06:04 +00003175 if (S->isAllEnumCasesCovered()) {
Ted Kremenek6e400352011-03-07 22:04:39 +00003176 const Stmt *L = To->getLabel();
3177 if (!L || !isa<CaseStmt>(L))
3178 return true;
Ted Kremenekee7f84d2010-09-09 00:06:04 +00003179 }
3180 }
3181 }
3182
3183 return false;
3184}
3185
3186//===----------------------------------------------------------------------===//
Ted Kremenek274f4332008-04-28 18:00:46 +00003187// Cleanup: CFG dstor.
3188//===----------------------------------------------------------------------===//
3189
Ted Kremenek63f58872007-10-01 19:33:33 +00003190CFG::~CFG() {
3191 delete reinterpret_cast<const BlkExprMapTy*>(BlkExprMap);
3192}
Mike Stump6d9828c2009-07-17 01:31:16 +00003193
Ted Kremenek7dba8602007-08-29 21:56:09 +00003194//===----------------------------------------------------------------------===//
3195// CFG pretty printing
3196//===----------------------------------------------------------------------===//
3197
Ted Kremeneke8ee26b2007-08-22 18:22:34 +00003198namespace {
3199
Kovarththanan Rajaratnamba5fb5a2009-11-28 06:07:30 +00003200class StmtPrinterHelper : public PrinterHelper {
Ted Kremenek3c0349e2011-03-01 03:15:10 +00003201 typedef llvm::DenseMap<const Stmt*,std::pair<unsigned,unsigned> > StmtMapTy;
3202 typedef llvm::DenseMap<const Decl*,std::pair<unsigned,unsigned> > DeclMapTy;
Ted Kremenek42a509f2007-08-31 21:30:12 +00003203 StmtMapTy StmtMap;
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003204 DeclMapTy DeclMap;
Ted Kremenek0a3ed312010-12-17 04:44:39 +00003205 signed currentBlock;
3206 unsigned currentStmt;
Chris Lattnere4f21422009-06-30 01:26:17 +00003207 const LangOptions &LangOpts;
Ted Kremenekd4fdee32007-08-23 21:42:29 +00003208public:
Ted Kremenek1c29bba2007-08-31 22:26:13 +00003209
Chris Lattnere4f21422009-06-30 01:26:17 +00003210 StmtPrinterHelper(const CFG* cfg, const LangOptions &LO)
Ted Kremenek3c0349e2011-03-01 03:15:10 +00003211 : currentBlock(0), currentStmt(0), LangOpts(LO)
3212 {
Ted Kremenek42a509f2007-08-31 21:30:12 +00003213 for (CFG::const_iterator I = cfg->begin(), E = cfg->end(); I != E; ++I ) {
3214 unsigned j = 1;
Ted Kremenekee82d9b2009-10-12 20:55:07 +00003215 for (CFGBlock::const_iterator BI = (*I)->begin(), BEnd = (*I)->end() ;
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003216 BI != BEnd; ++BI, ++j ) {
Ted Kremenek3c0349e2011-03-01 03:15:10 +00003217 if (const CFGStmt *SE = BI->getAs<CFGStmt>()) {
3218 const Stmt *stmt= SE->getStmt();
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003219 std::pair<unsigned, unsigned> P((*I)->getBlockID(), j);
Ted Kremenek3c0349e2011-03-01 03:15:10 +00003220 StmtMap[stmt] = P;
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003221
Ted Kremenek3c0349e2011-03-01 03:15:10 +00003222 switch (stmt->getStmtClass()) {
3223 case Stmt::DeclStmtClass:
3224 DeclMap[cast<DeclStmt>(stmt)->getSingleDecl()] = P;
3225 break;
3226 case Stmt::IfStmtClass: {
3227 const VarDecl *var = cast<IfStmt>(stmt)->getConditionVariable();
3228 if (var)
3229 DeclMap[var] = P;
3230 break;
3231 }
3232 case Stmt::ForStmtClass: {
3233 const VarDecl *var = cast<ForStmt>(stmt)->getConditionVariable();
3234 if (var)
3235 DeclMap[var] = P;
3236 break;
3237 }
3238 case Stmt::WhileStmtClass: {
3239 const VarDecl *var =
3240 cast<WhileStmt>(stmt)->getConditionVariable();
3241 if (var)
3242 DeclMap[var] = P;
3243 break;
3244 }
3245 case Stmt::SwitchStmtClass: {
3246 const VarDecl *var =
3247 cast<SwitchStmt>(stmt)->getConditionVariable();
3248 if (var)
3249 DeclMap[var] = P;
3250 break;
3251 }
3252 case Stmt::CXXCatchStmtClass: {
3253 const VarDecl *var =
3254 cast<CXXCatchStmt>(stmt)->getExceptionDecl();
3255 if (var)
3256 DeclMap[var] = P;
3257 break;
3258 }
3259 default:
3260 break;
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003261 }
3262 }
Ted Kremenek42a509f2007-08-31 21:30:12 +00003263 }
Zhongxing Xub36cd3e2010-09-16 01:25:47 +00003264 }
Ted Kremenek42a509f2007-08-31 21:30:12 +00003265 }
Ted Kremenek3c0349e2011-03-01 03:15:10 +00003266
Mike Stump6d9828c2009-07-17 01:31:16 +00003267
Ted Kremenek42a509f2007-08-31 21:30:12 +00003268 virtual ~StmtPrinterHelper() {}
Mike Stump6d9828c2009-07-17 01:31:16 +00003269
Chris Lattnere4f21422009-06-30 01:26:17 +00003270 const LangOptions &getLangOpts() const { return LangOpts; }
Ted Kremenek0a3ed312010-12-17 04:44:39 +00003271 void setBlockID(signed i) { currentBlock = i; }
3272 void setStmtID(unsigned i) { currentStmt = i; }
Mike Stump6d9828c2009-07-17 01:31:16 +00003273
Ted Kremenek9c378f72011-08-12 23:37:29 +00003274 virtual bool handledStmt(Stmt *S, raw_ostream &OS) {
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003275 StmtMapTy::iterator I = StmtMap.find(S);
Ted Kremenek42a509f2007-08-31 21:30:12 +00003276
3277 if (I == StmtMap.end())
3278 return false;
Mike Stump6d9828c2009-07-17 01:31:16 +00003279
Ted Kremenek0a3ed312010-12-17 04:44:39 +00003280 if (currentBlock >= 0 && I->second.first == (unsigned) currentBlock
3281 && I->second.second == currentStmt) {
Ted Kremenek42a509f2007-08-31 21:30:12 +00003282 return false;
Ted Kremenek3fa1e4b2010-01-19 20:52:05 +00003283 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003284
Ted Kremenek3fa1e4b2010-01-19 20:52:05 +00003285 OS << "[B" << I->second.first << "." << I->second.second << "]";
Ted Kremenek1c29bba2007-08-31 22:26:13 +00003286 return true;
Ted Kremenek42a509f2007-08-31 21:30:12 +00003287 }
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003288
Ted Kremenek9c378f72011-08-12 23:37:29 +00003289 bool handleDecl(const Decl *D, raw_ostream &OS) {
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003290 DeclMapTy::iterator I = DeclMap.find(D);
3291
3292 if (I == DeclMap.end())
3293 return false;
3294
Ted Kremenek0a3ed312010-12-17 04:44:39 +00003295 if (currentBlock >= 0 && I->second.first == (unsigned) currentBlock
3296 && I->second.second == currentStmt) {
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003297 return false;
3298 }
3299
3300 OS << "[B" << I->second.first << "." << I->second.second << "]";
3301 return true;
3302 }
Ted Kremenek42a509f2007-08-31 21:30:12 +00003303};
Chris Lattnere4f21422009-06-30 01:26:17 +00003304} // end anonymous namespace
Ted Kremenek42a509f2007-08-31 21:30:12 +00003305
Chris Lattnere4f21422009-06-30 01:26:17 +00003306
3307namespace {
Kovarththanan Rajaratnamba5fb5a2009-11-28 06:07:30 +00003308class CFGBlockTerminatorPrint
Ted Kremenek6fa9b882008-01-08 18:15:10 +00003309 : public StmtVisitor<CFGBlockTerminatorPrint,void> {
Mike Stump6d9828c2009-07-17 01:31:16 +00003310
Ted Kremenek9c378f72011-08-12 23:37:29 +00003311 raw_ostream &OS;
Ted Kremenek42a509f2007-08-31 21:30:12 +00003312 StmtPrinterHelper* Helper;
Douglas Gregord249e1d1f2009-05-29 20:38:28 +00003313 PrintingPolicy Policy;
Ted Kremenek42a509f2007-08-31 21:30:12 +00003314public:
Ted Kremenek9c378f72011-08-12 23:37:29 +00003315 CFGBlockTerminatorPrint(raw_ostream &os, StmtPrinterHelper* helper,
Chris Lattnere4f21422009-06-30 01:26:17 +00003316 const PrintingPolicy &Policy)
Douglas Gregord249e1d1f2009-05-29 20:38:28 +00003317 : OS(os), Helper(helper), Policy(Policy) {}
Mike Stump6d9828c2009-07-17 01:31:16 +00003318
Ted Kremenek9c378f72011-08-12 23:37:29 +00003319 void VisitIfStmt(IfStmt *I) {
Ted Kremenekd4fdee32007-08-23 21:42:29 +00003320 OS << "if ";
Douglas Gregord249e1d1f2009-05-29 20:38:28 +00003321 I->getCond()->printPretty(OS,Helper,Policy);
Ted Kremenekd4fdee32007-08-23 21:42:29 +00003322 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003323
Ted Kremenekd4fdee32007-08-23 21:42:29 +00003324 // Default case.
Ted Kremenek9c378f72011-08-12 23:37:29 +00003325 void VisitStmt(Stmt *Terminator) {
Mike Stump6d9828c2009-07-17 01:31:16 +00003326 Terminator->printPretty(OS, Helper, Policy);
3327 }
3328
Ted Kremenek9c378f72011-08-12 23:37:29 +00003329 void VisitForStmt(ForStmt *F) {
Ted Kremenekd4fdee32007-08-23 21:42:29 +00003330 OS << "for (" ;
Ted Kremenek3fa1e4b2010-01-19 20:52:05 +00003331 if (F->getInit())
3332 OS << "...";
Ted Kremenek535bb202007-08-30 21:28:02 +00003333 OS << "; ";
Ted Kremenek9c378f72011-08-12 23:37:29 +00003334 if (Stmt *C = F->getCond())
Ted Kremenek3fa1e4b2010-01-19 20:52:05 +00003335 C->printPretty(OS, Helper, Policy);
Ted Kremenek535bb202007-08-30 21:28:02 +00003336 OS << "; ";
Ted Kremenek3fa1e4b2010-01-19 20:52:05 +00003337 if (F->getInc())
3338 OS << "...";
Ted Kremeneka2925852008-01-30 23:02:42 +00003339 OS << ")";
Ted Kremenekd4fdee32007-08-23 21:42:29 +00003340 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003341
Ted Kremenek9c378f72011-08-12 23:37:29 +00003342 void VisitWhileStmt(WhileStmt *W) {
Ted Kremenekd4fdee32007-08-23 21:42:29 +00003343 OS << "while " ;
Ted Kremenek9c378f72011-08-12 23:37:29 +00003344 if (Stmt *C = W->getCond())
Ted Kremenek3fa1e4b2010-01-19 20:52:05 +00003345 C->printPretty(OS, Helper, Policy);
Ted Kremenekd4fdee32007-08-23 21:42:29 +00003346 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003347
Ted Kremenek9c378f72011-08-12 23:37:29 +00003348 void VisitDoStmt(DoStmt *D) {
Ted Kremenekd4fdee32007-08-23 21:42:29 +00003349 OS << "do ... while ";
Ted Kremenek9c378f72011-08-12 23:37:29 +00003350 if (Stmt *C = D->getCond())
Ted Kremenek3fa1e4b2010-01-19 20:52:05 +00003351 C->printPretty(OS, Helper, Policy);
Ted Kremenek9da2fb72007-08-27 21:27:44 +00003352 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003353
Ted Kremenek9c378f72011-08-12 23:37:29 +00003354 void VisitSwitchStmt(SwitchStmt *Terminator) {
Ted Kremenek9da2fb72007-08-27 21:27:44 +00003355 OS << "switch ";
Douglas Gregord249e1d1f2009-05-29 20:38:28 +00003356 Terminator->getCond()->printPretty(OS, Helper, Policy);
Ted Kremenek9da2fb72007-08-27 21:27:44 +00003357 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003358
Ted Kremenek9c378f72011-08-12 23:37:29 +00003359 void VisitCXXTryStmt(CXXTryStmt *CS) {
Mike Stump5d1d2022010-01-19 02:20:09 +00003360 OS << "try ...";
3361 }
3362
John McCall56ca35d2011-02-17 10:25:35 +00003363 void VisitAbstractConditionalOperator(AbstractConditionalOperator* C) {
Douglas Gregord249e1d1f2009-05-29 20:38:28 +00003364 C->getCond()->printPretty(OS, Helper, Policy);
Mike Stump6d9828c2009-07-17 01:31:16 +00003365 OS << " ? ... : ...";
Ted Kremenek805e9a82007-08-31 21:49:40 +00003366 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003367
Ted Kremenek9c378f72011-08-12 23:37:29 +00003368 void VisitChooseExpr(ChooseExpr *C) {
Ted Kremenekaeddbf62007-08-31 22:29:13 +00003369 OS << "__builtin_choose_expr( ";
Douglas Gregord249e1d1f2009-05-29 20:38:28 +00003370 C->getCond()->printPretty(OS, Helper, Policy);
Ted Kremeneka2925852008-01-30 23:02:42 +00003371 OS << " )";
Ted Kremenekaeddbf62007-08-31 22:29:13 +00003372 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003373
Ted Kremenek9c378f72011-08-12 23:37:29 +00003374 void VisitIndirectGotoStmt(IndirectGotoStmt *I) {
Ted Kremenek1c29bba2007-08-31 22:26:13 +00003375 OS << "goto *";
Douglas Gregord249e1d1f2009-05-29 20:38:28 +00003376 I->getTarget()->printPretty(OS, Helper, Policy);
Ted Kremenek1c29bba2007-08-31 22:26:13 +00003377 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003378
Ted Kremenek805e9a82007-08-31 21:49:40 +00003379 void VisitBinaryOperator(BinaryOperator* B) {
3380 if (!B->isLogicalOp()) {
3381 VisitExpr(B);
3382 return;
3383 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003384
Douglas Gregord249e1d1f2009-05-29 20:38:28 +00003385 B->getLHS()->printPretty(OS, Helper, Policy);
Mike Stump6d9828c2009-07-17 01:31:16 +00003386
Ted Kremenek805e9a82007-08-31 21:49:40 +00003387 switch (B->getOpcode()) {
John McCall2de56d12010-08-25 11:45:40 +00003388 case BO_LOr:
Ted Kremeneka2925852008-01-30 23:02:42 +00003389 OS << " || ...";
Ted Kremenek805e9a82007-08-31 21:49:40 +00003390 return;
John McCall2de56d12010-08-25 11:45:40 +00003391 case BO_LAnd:
Ted Kremeneka2925852008-01-30 23:02:42 +00003392 OS << " && ...";
Ted Kremenek805e9a82007-08-31 21:49:40 +00003393 return;
3394 default:
David Blaikieb219cfc2011-09-23 05:06:16 +00003395 llvm_unreachable("Invalid logical operator.");
Mike Stump6d9828c2009-07-17 01:31:16 +00003396 }
Ted Kremenek805e9a82007-08-31 21:49:40 +00003397 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003398
Ted Kremenek9c378f72011-08-12 23:37:29 +00003399 void VisitExpr(Expr *E) {
Douglas Gregord249e1d1f2009-05-29 20:38:28 +00003400 E->printPretty(OS, Helper, Policy);
Mike Stump6d9828c2009-07-17 01:31:16 +00003401 }
Ted Kremenekd4fdee32007-08-23 21:42:29 +00003402};
Chris Lattnere4f21422009-06-30 01:26:17 +00003403} // end anonymous namespace
3404
Chris Lattner5f9e2722011-07-23 10:55:15 +00003405static void print_elem(raw_ostream &OS, StmtPrinterHelper* Helper,
Mike Stump079bd722010-01-19 22:00:14 +00003406 const CFGElement &E) {
Ted Kremenek3c0349e2011-03-01 03:15:10 +00003407 if (const CFGStmt *CS = E.getAs<CFGStmt>()) {
Ted Kremenekf1d10d92011-08-23 23:05:04 +00003408 const Stmt *S = CS->getStmt();
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003409
3410 if (Helper) {
Mike Stump6d9828c2009-07-17 01:31:16 +00003411
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003412 // special printing for statement-expressions.
Ted Kremenekf1d10d92011-08-23 23:05:04 +00003413 if (const StmtExpr *SE = dyn_cast<StmtExpr>(S)) {
3414 const CompoundStmt *Sub = SE->getSubStmt();
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003415
John McCall7502c1d2011-02-13 04:07:26 +00003416 if (Sub->children()) {
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003417 OS << "({ ... ; ";
3418 Helper->handledStmt(*SE->getSubStmt()->body_rbegin(),OS);
3419 OS << " })\n";
3420 return;
3421 }
3422 }
3423 // special printing for comma expressions.
Ted Kremenekf1d10d92011-08-23 23:05:04 +00003424 if (const BinaryOperator* B = dyn_cast<BinaryOperator>(S)) {
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003425 if (B->getOpcode() == BO_Comma) {
3426 OS << "... , ";
3427 Helper->handledStmt(B->getRHS(),OS);
3428 OS << '\n';
3429 return;
3430 }
Ted Kremenek1c29bba2007-08-31 22:26:13 +00003431 }
3432 }
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003433 S->printPretty(OS, Helper, PrintingPolicy(Helper->getLangOpts()));
Mike Stump6d9828c2009-07-17 01:31:16 +00003434
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003435 if (isa<CXXOperatorCallExpr>(S)) {
Zhanyong Wan36f327c2010-11-22 19:32:14 +00003436 OS << " (OperatorCall)";
3437 } else if (isa<CXXBindTemporaryExpr>(S)) {
3438 OS << " (BindTemporary)";
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003439 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003440
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003441 // Expressions need a newline.
3442 if (isa<Expr>(S))
3443 OS << '\n';
Ted Kremenek4e0cfa82010-08-31 18:47:37 +00003444
Ted Kremenek3c0349e2011-03-01 03:15:10 +00003445 } else if (const CFGInitializer *IE = E.getAs<CFGInitializer>()) {
3446 const CXXCtorInitializer *I = IE->getInitializer();
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003447 if (I->isBaseInitializer())
3448 OS << I->getBaseClass()->getAsCXXRecordDecl()->getName();
Francois Pichet00eb3f92010-12-04 09:14:42 +00003449 else OS << I->getAnyMember()->getName();
Mike Stump6d9828c2009-07-17 01:31:16 +00003450
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003451 OS << "(";
Ted Kremenek9c378f72011-08-12 23:37:29 +00003452 if (Expr *IE = I->getInit())
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003453 IE->printPretty(OS, Helper, PrintingPolicy(Helper->getLangOpts()));
3454 OS << ")";
3455
3456 if (I->isBaseInitializer())
3457 OS << " (Base initializer)\n";
3458 else OS << " (Member initializer)\n";
3459
Ted Kremenek3c0349e2011-03-01 03:15:10 +00003460 } else if (const CFGAutomaticObjDtor *DE = E.getAs<CFGAutomaticObjDtor>()){
Ted Kremenek9c378f72011-08-12 23:37:29 +00003461 const VarDecl *VD = DE->getVarDecl();
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003462 Helper->handleDecl(VD, OS);
3463
Marcin Swiderskib1c52872010-10-25 07:00:40 +00003464 const Type* T = VD->getType().getTypePtr();
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003465 if (const ReferenceType* RT = T->getAs<ReferenceType>())
3466 T = RT->getPointeeType().getTypePtr();
Marcin Swiderskib1c52872010-10-25 07:00:40 +00003467 else if (const Type *ET = T->getArrayElementTypeNoTypeQual())
3468 T = ET;
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003469
3470 OS << ".~" << T->getAsCXXRecordDecl()->getName().str() << "()";
3471 OS << " (Implicit destructor)\n";
Marcin Swiderski7c625d82010-10-05 05:37:00 +00003472
Ted Kremenek3c0349e2011-03-01 03:15:10 +00003473 } else if (const CFGBaseDtor *BE = E.getAs<CFGBaseDtor>()) {
3474 const CXXBaseSpecifier *BS = BE->getBaseSpecifier();
Marcin Swiderski7c625d82010-10-05 05:37:00 +00003475 OS << "~" << BS->getType()->getAsCXXRecordDecl()->getName() << "()";
Zhongxing Xu4e493e02010-10-05 08:38:06 +00003476 OS << " (Base object destructor)\n";
Marcin Swiderski7c625d82010-10-05 05:37:00 +00003477
Ted Kremenek3c0349e2011-03-01 03:15:10 +00003478 } else if (const CFGMemberDtor *ME = E.getAs<CFGMemberDtor>()) {
3479 const FieldDecl *FD = ME->getFieldDecl();
Marcin Swiderski8c5e5d62010-10-25 07:05:54 +00003480
3481 const Type *T = FD->getType().getTypePtr();
3482 if (const Type *ET = T->getArrayElementTypeNoTypeQual())
3483 T = ET;
3484
Marcin Swiderski7c625d82010-10-05 05:37:00 +00003485 OS << "this->" << FD->getName();
Marcin Swiderski8c5e5d62010-10-25 07:05:54 +00003486 OS << ".~" << T->getAsCXXRecordDecl()->getName() << "()";
Zhongxing Xu4e493e02010-10-05 08:38:06 +00003487 OS << " (Member object destructor)\n";
Marcin Swiderski8599e762010-11-03 06:19:35 +00003488
Ted Kremenek3c0349e2011-03-01 03:15:10 +00003489 } else if (const CFGTemporaryDtor *TE = E.getAs<CFGTemporaryDtor>()) {
3490 const CXXBindTemporaryExpr *BT = TE->getBindTemporaryExpr();
Marcin Swiderski8599e762010-11-03 06:19:35 +00003491 OS << "~" << BT->getType()->getAsCXXRecordDecl()->getName() << "()";
3492 OS << " (Temporary object destructor)\n";
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003493 }
Zhongxing Xu81bc7d02010-11-01 06:46:05 +00003494}
Mike Stump6d9828c2009-07-17 01:31:16 +00003495
Ted Kremenek9c378f72011-08-12 23:37:29 +00003496static void print_block(raw_ostream &OS, const CFG* cfg,
3497 const CFGBlock &B,
Chris Lattnere4f21422009-06-30 01:26:17 +00003498 StmtPrinterHelper* Helper, bool print_edges) {
Mike Stump6d9828c2009-07-17 01:31:16 +00003499
Ted Kremenek42a509f2007-08-31 21:30:12 +00003500 if (Helper) Helper->setBlockID(B.getBlockID());
Mike Stump6d9828c2009-07-17 01:31:16 +00003501
Ted Kremenek7dba8602007-08-29 21:56:09 +00003502 // Print the header.
Mike Stump6d9828c2009-07-17 01:31:16 +00003503 OS << "\n [ B" << B.getBlockID();
3504
Ted Kremenek42a509f2007-08-31 21:30:12 +00003505 if (&B == &cfg->getEntry())
3506 OS << " (ENTRY) ]\n";
3507 else if (&B == &cfg->getExit())
3508 OS << " (EXIT) ]\n";
3509 else if (&B == cfg->getIndirectGotoBlock())
Ted Kremenek7dba8602007-08-29 21:56:09 +00003510 OS << " (INDIRECT GOTO DISPATCH) ]\n";
Ted Kremenek42a509f2007-08-31 21:30:12 +00003511 else
3512 OS << " ]\n";
Mike Stump6d9828c2009-07-17 01:31:16 +00003513
Ted Kremenek9cffe732007-08-29 23:20:49 +00003514 // Print the label of this block.
Ted Kremenek9c378f72011-08-12 23:37:29 +00003515 if (Stmt *Label = const_cast<Stmt*>(B.getLabel())) {
Ted Kremenek42a509f2007-08-31 21:30:12 +00003516
3517 if (print_edges)
3518 OS << " ";
Mike Stump6d9828c2009-07-17 01:31:16 +00003519
Ted Kremenek9c378f72011-08-12 23:37:29 +00003520 if (LabelStmt *L = dyn_cast<LabelStmt>(Label))
Ted Kremenek9cffe732007-08-29 23:20:49 +00003521 OS << L->getName();
Ted Kremenek9c378f72011-08-12 23:37:29 +00003522 else if (CaseStmt *C = dyn_cast<CaseStmt>(Label)) {
Ted Kremenek9cffe732007-08-29 23:20:49 +00003523 OS << "case ";
Chris Lattnere4f21422009-06-30 01:26:17 +00003524 C->getLHS()->printPretty(OS, Helper,
3525 PrintingPolicy(Helper->getLangOpts()));
Ted Kremenek9cffe732007-08-29 23:20:49 +00003526 if (C->getRHS()) {
3527 OS << " ... ";
Chris Lattnere4f21422009-06-30 01:26:17 +00003528 C->getRHS()->printPretty(OS, Helper,
3529 PrintingPolicy(Helper->getLangOpts()));
Ted Kremenek9cffe732007-08-29 23:20:49 +00003530 }
Mike Stump079bd722010-01-19 22:00:14 +00003531 } else if (isa<DefaultStmt>(Label))
Ted Kremenek9cffe732007-08-29 23:20:49 +00003532 OS << "default";
Mike Stump079bd722010-01-19 22:00:14 +00003533 else if (CXXCatchStmt *CS = dyn_cast<CXXCatchStmt>(Label)) {
Mike Stump5d1d2022010-01-19 02:20:09 +00003534 OS << "catch (";
Mike Stumpa1f93632010-01-20 01:15:34 +00003535 if (CS->getExceptionDecl())
3536 CS->getExceptionDecl()->print(OS, PrintingPolicy(Helper->getLangOpts()),
3537 0);
3538 else
3539 OS << "...";
Mike Stump5d1d2022010-01-19 02:20:09 +00003540 OS << ")";
3541
3542 } else
David Blaikieb219cfc2011-09-23 05:06:16 +00003543 llvm_unreachable("Invalid label statement in CFGBlock.");
Mike Stump6d9828c2009-07-17 01:31:16 +00003544
Ted Kremenek9cffe732007-08-29 23:20:49 +00003545 OS << ":\n";
3546 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003547
Ted Kremenekfddd5182007-08-21 21:42:03 +00003548 // Iterate through the statements in the block and print them.
Ted Kremenekfddd5182007-08-21 21:42:03 +00003549 unsigned j = 1;
Mike Stump6d9828c2009-07-17 01:31:16 +00003550
Ted Kremenek42a509f2007-08-31 21:30:12 +00003551 for (CFGBlock::const_iterator I = B.begin(), E = B.end() ;
3552 I != E ; ++I, ++j ) {
Mike Stump6d9828c2009-07-17 01:31:16 +00003553
Ted Kremenek9cffe732007-08-29 23:20:49 +00003554 // Print the statement # in the basic block and the statement itself.
Ted Kremenek42a509f2007-08-31 21:30:12 +00003555 if (print_edges)
3556 OS << " ";
Mike Stump6d9828c2009-07-17 01:31:16 +00003557
Ted Kremeneka95d3752008-09-13 05:16:45 +00003558 OS << llvm::format("%3d", j) << ": ";
Mike Stump6d9828c2009-07-17 01:31:16 +00003559
Ted Kremenek42a509f2007-08-31 21:30:12 +00003560 if (Helper)
3561 Helper->setStmtID(j);
Mike Stump6d9828c2009-07-17 01:31:16 +00003562
Marcin Swiderski1cff1322010-09-21 05:58:15 +00003563 print_elem(OS,Helper,*I);
Ted Kremenekfddd5182007-08-21 21:42:03 +00003564 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003565
Ted Kremenek9cffe732007-08-29 23:20:49 +00003566 // Print the terminator of this block.
Ted Kremenek42a509f2007-08-31 21:30:12 +00003567 if (B.getTerminator()) {
3568 if (print_edges)
3569 OS << " ";
Mike Stump6d9828c2009-07-17 01:31:16 +00003570
Ted Kremenek9cffe732007-08-29 23:20:49 +00003571 OS << " T: ";
Mike Stump6d9828c2009-07-17 01:31:16 +00003572
Ted Kremenek42a509f2007-08-31 21:30:12 +00003573 if (Helper) Helper->setBlockID(-1);
Mike Stump6d9828c2009-07-17 01:31:16 +00003574
Chris Lattnere4f21422009-06-30 01:26:17 +00003575 CFGBlockTerminatorPrint TPrinter(OS, Helper,
3576 PrintingPolicy(Helper->getLangOpts()));
Marcin Swiderski4ba72a02010-10-29 05:21:47 +00003577 TPrinter.Visit(const_cast<Stmt*>(B.getTerminator().getStmt()));
Ted Kremeneka2925852008-01-30 23:02:42 +00003578 OS << '\n';
Ted Kremenekfddd5182007-08-21 21:42:03 +00003579 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003580
Ted Kremenek9cffe732007-08-29 23:20:49 +00003581 if (print_edges) {
3582 // Print the predecessors of this block.
Ted Kremenek42a509f2007-08-31 21:30:12 +00003583 OS << " Predecessors (" << B.pred_size() << "):";
Ted Kremenek9cffe732007-08-29 23:20:49 +00003584 unsigned i = 0;
Ted Kremenek9cffe732007-08-29 23:20:49 +00003585
Ted Kremenek42a509f2007-08-31 21:30:12 +00003586 for (CFGBlock::const_pred_iterator I = B.pred_begin(), E = B.pred_end();
3587 I != E; ++I, ++i) {
Mike Stump6d9828c2009-07-17 01:31:16 +00003588
Ted Kremenek42a509f2007-08-31 21:30:12 +00003589 if (i == 8 || (i-8) == 0)
3590 OS << "\n ";
Mike Stump6d9828c2009-07-17 01:31:16 +00003591
Ted Kremenek9cffe732007-08-29 23:20:49 +00003592 OS << " B" << (*I)->getBlockID();
3593 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003594
Ted Kremenek42a509f2007-08-31 21:30:12 +00003595 OS << '\n';
Mike Stump6d9828c2009-07-17 01:31:16 +00003596
Ted Kremenek42a509f2007-08-31 21:30:12 +00003597 // Print the successors of this block.
3598 OS << " Successors (" << B.succ_size() << "):";
3599 i = 0;
3600
3601 for (CFGBlock::const_succ_iterator I = B.succ_begin(), E = B.succ_end();
3602 I != E; ++I, ++i) {
Mike Stump6d9828c2009-07-17 01:31:16 +00003603
Ted Kremenek42a509f2007-08-31 21:30:12 +00003604 if (i == 8 || (i-8) % 10 == 0)
3605 OS << "\n ";
3606
Mike Stumpe5af3ce2009-07-20 23:24:15 +00003607 if (*I)
3608 OS << " B" << (*I)->getBlockID();
3609 else
3610 OS << " NULL";
Ted Kremenek42a509f2007-08-31 21:30:12 +00003611 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003612
Ted Kremenek9cffe732007-08-29 23:20:49 +00003613 OS << '\n';
Ted Kremenekfddd5182007-08-21 21:42:03 +00003614 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003615}
Ted Kremenek42a509f2007-08-31 21:30:12 +00003616
Ted Kremenek42a509f2007-08-31 21:30:12 +00003617
3618/// dump - A simple pretty printer of a CFG that outputs to stderr.
Chris Lattnere4f21422009-06-30 01:26:17 +00003619void CFG::dump(const LangOptions &LO) const { print(llvm::errs(), LO); }
Ted Kremenek42a509f2007-08-31 21:30:12 +00003620
3621/// print - A simple pretty printer of a CFG that outputs to an ostream.
Chris Lattner5f9e2722011-07-23 10:55:15 +00003622void CFG::print(raw_ostream &OS, const LangOptions &LO) const {
Chris Lattnere4f21422009-06-30 01:26:17 +00003623 StmtPrinterHelper Helper(this, LO);
Mike Stump6d9828c2009-07-17 01:31:16 +00003624
Ted Kremenek42a509f2007-08-31 21:30:12 +00003625 // Print the entry block.
3626 print_block(OS, this, getEntry(), &Helper, true);
Mike Stump6d9828c2009-07-17 01:31:16 +00003627
Ted Kremenek42a509f2007-08-31 21:30:12 +00003628 // Iterate through the CFGBlocks and print them one by one.
3629 for (const_iterator I = Blocks.begin(), E = Blocks.end() ; I != E ; ++I) {
3630 // Skip the entry block, because we already printed it.
Ted Kremenekee82d9b2009-10-12 20:55:07 +00003631 if (&(**I) == &getEntry() || &(**I) == &getExit())
Ted Kremenek42a509f2007-08-31 21:30:12 +00003632 continue;
Mike Stump6d9828c2009-07-17 01:31:16 +00003633
Ted Kremenekee82d9b2009-10-12 20:55:07 +00003634 print_block(OS, this, **I, &Helper, true);
Ted Kremenek42a509f2007-08-31 21:30:12 +00003635 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003636
Ted Kremenek42a509f2007-08-31 21:30:12 +00003637 // Print the exit block.
3638 print_block(OS, this, getExit(), &Helper, true);
Ted Kremenekd0172432008-11-24 20:50:24 +00003639 OS.flush();
Mike Stump6d9828c2009-07-17 01:31:16 +00003640}
Ted Kremenek42a509f2007-08-31 21:30:12 +00003641
3642/// dump - A simply pretty printer of a CFGBlock that outputs to stderr.
Chris Lattnere4f21422009-06-30 01:26:17 +00003643void CFGBlock::dump(const CFG* cfg, const LangOptions &LO) const {
3644 print(llvm::errs(), cfg, LO);
3645}
Ted Kremenek42a509f2007-08-31 21:30:12 +00003646
3647/// print - A simple pretty printer of a CFGBlock that outputs to an ostream.
3648/// Generally this will only be called from CFG::print.
Ted Kremenek9c378f72011-08-12 23:37:29 +00003649void CFGBlock::print(raw_ostream &OS, const CFG* cfg,
Chris Lattnere4f21422009-06-30 01:26:17 +00003650 const LangOptions &LO) const {
3651 StmtPrinterHelper Helper(cfg, LO);
Ted Kremenek42a509f2007-08-31 21:30:12 +00003652 print_block(OS, cfg, *this, &Helper, true);
Ted Kremenek026473c2007-08-23 16:51:22 +00003653}
Ted Kremenek7dba8602007-08-29 21:56:09 +00003654
Ted Kremeneka2925852008-01-30 23:02:42 +00003655/// printTerminator - A simple pretty printer of the terminator of a CFGBlock.
Chris Lattner5f9e2722011-07-23 10:55:15 +00003656void CFGBlock::printTerminator(raw_ostream &OS,
Mike Stump6d9828c2009-07-17 01:31:16 +00003657 const LangOptions &LO) const {
Chris Lattnere4f21422009-06-30 01:26:17 +00003658 CFGBlockTerminatorPrint TPrinter(OS, NULL, PrintingPolicy(LO));
Marcin Swiderski4ba72a02010-10-29 05:21:47 +00003659 TPrinter.Visit(const_cast<Stmt*>(getTerminator().getStmt()));
Ted Kremeneka2925852008-01-30 23:02:42 +00003660}
3661
Ted Kremenek9c378f72011-08-12 23:37:29 +00003662Stmt *CFGBlock::getTerminatorCondition() {
Marcin Swiderski4ba72a02010-10-29 05:21:47 +00003663 Stmt *Terminator = this->Terminator;
Ted Kremenek411cdee2008-04-16 21:10:48 +00003664 if (!Terminator)
3665 return NULL;
Mike Stump6d9828c2009-07-17 01:31:16 +00003666
Ted Kremenek9c378f72011-08-12 23:37:29 +00003667 Expr *E = NULL;
Mike Stump6d9828c2009-07-17 01:31:16 +00003668
Ted Kremenek411cdee2008-04-16 21:10:48 +00003669 switch (Terminator->getStmtClass()) {
3670 default:
3671 break;
Mike Stump6d9828c2009-07-17 01:31:16 +00003672
Ted Kremenek411cdee2008-04-16 21:10:48 +00003673 case Stmt::ForStmtClass:
3674 E = cast<ForStmt>(Terminator)->getCond();
3675 break;
Mike Stump6d9828c2009-07-17 01:31:16 +00003676
Ted Kremenek411cdee2008-04-16 21:10:48 +00003677 case Stmt::WhileStmtClass:
3678 E = cast<WhileStmt>(Terminator)->getCond();
3679 break;
Mike Stump6d9828c2009-07-17 01:31:16 +00003680
Ted Kremenek411cdee2008-04-16 21:10:48 +00003681 case Stmt::DoStmtClass:
3682 E = cast<DoStmt>(Terminator)->getCond();
3683 break;
Mike Stump6d9828c2009-07-17 01:31:16 +00003684
Ted Kremenek411cdee2008-04-16 21:10:48 +00003685 case Stmt::IfStmtClass:
3686 E = cast<IfStmt>(Terminator)->getCond();
3687 break;
Mike Stump6d9828c2009-07-17 01:31:16 +00003688
Ted Kremenek411cdee2008-04-16 21:10:48 +00003689 case Stmt::ChooseExprClass:
3690 E = cast<ChooseExpr>(Terminator)->getCond();
3691 break;
Mike Stump6d9828c2009-07-17 01:31:16 +00003692
Ted Kremenek411cdee2008-04-16 21:10:48 +00003693 case Stmt::IndirectGotoStmtClass:
3694 E = cast<IndirectGotoStmt>(Terminator)->getTarget();
3695 break;
Mike Stump6d9828c2009-07-17 01:31:16 +00003696
Ted Kremenek411cdee2008-04-16 21:10:48 +00003697 case Stmt::SwitchStmtClass:
3698 E = cast<SwitchStmt>(Terminator)->getCond();
3699 break;
Mike Stump6d9828c2009-07-17 01:31:16 +00003700
John McCall56ca35d2011-02-17 10:25:35 +00003701 case Stmt::BinaryConditionalOperatorClass:
3702 E = cast<BinaryConditionalOperator>(Terminator)->getCond();
3703 break;
3704
Ted Kremenek411cdee2008-04-16 21:10:48 +00003705 case Stmt::ConditionalOperatorClass:
3706 E = cast<ConditionalOperator>(Terminator)->getCond();
3707 break;
Mike Stump6d9828c2009-07-17 01:31:16 +00003708
Ted Kremenek411cdee2008-04-16 21:10:48 +00003709 case Stmt::BinaryOperatorClass: // '&&' and '||'
3710 E = cast<BinaryOperator>(Terminator)->getLHS();
Ted Kremenek390e48b2008-11-12 21:11:49 +00003711 break;
Mike Stump6d9828c2009-07-17 01:31:16 +00003712
Ted Kremenek390e48b2008-11-12 21:11:49 +00003713 case Stmt::ObjCForCollectionStmtClass:
Mike Stump6d9828c2009-07-17 01:31:16 +00003714 return Terminator;
Ted Kremenek411cdee2008-04-16 21:10:48 +00003715 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003716
Ted Kremenek411cdee2008-04-16 21:10:48 +00003717 return E ? E->IgnoreParens() : NULL;
3718}
3719
Ted Kremenek7dba8602007-08-29 21:56:09 +00003720//===----------------------------------------------------------------------===//
3721// CFG Graphviz Visualization
3722//===----------------------------------------------------------------------===//
3723
Ted Kremenek42a509f2007-08-31 21:30:12 +00003724
3725#ifndef NDEBUG
Mike Stump6d9828c2009-07-17 01:31:16 +00003726static StmtPrinterHelper* GraphHelper;
Ted Kremenek42a509f2007-08-31 21:30:12 +00003727#endif
3728
Chris Lattnere4f21422009-06-30 01:26:17 +00003729void CFG::viewCFG(const LangOptions &LO) const {
Ted Kremenek42a509f2007-08-31 21:30:12 +00003730#ifndef NDEBUG
Chris Lattnere4f21422009-06-30 01:26:17 +00003731 StmtPrinterHelper H(this, LO);
Ted Kremenek42a509f2007-08-31 21:30:12 +00003732 GraphHelper = &H;
3733 llvm::ViewGraph(this,"CFG");
3734 GraphHelper = NULL;
Ted Kremenek42a509f2007-08-31 21:30:12 +00003735#endif
3736}
3737
Ted Kremenek7dba8602007-08-29 21:56:09 +00003738namespace llvm {
3739template<>
3740struct DOTGraphTraits<const CFG*> : public DefaultDOTGraphTraits {
Tobias Grosser006b0eb2009-11-30 14:16:05 +00003741
3742 DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {}
3743
Ted Kremenek9c378f72011-08-12 23:37:29 +00003744 static std::string getNodeLabel(const CFGBlock *Node, const CFG* Graph) {
Ted Kremenek7dba8602007-08-29 21:56:09 +00003745
Hartmut Kaiserbd250b42007-09-16 00:28:28 +00003746#ifndef NDEBUG
Ted Kremeneka95d3752008-09-13 05:16:45 +00003747 std::string OutSStr;
3748 llvm::raw_string_ostream Out(OutSStr);
Ted Kremenek42a509f2007-08-31 21:30:12 +00003749 print_block(Out,Graph, *Node, GraphHelper, false);
Ted Kremeneka95d3752008-09-13 05:16:45 +00003750 std::string& OutStr = Out.str();
Ted Kremenek7dba8602007-08-29 21:56:09 +00003751
3752 if (OutStr[0] == '\n') OutStr.erase(OutStr.begin());
3753
3754 // Process string output to make it nicer...
3755 for (unsigned i = 0; i != OutStr.length(); ++i)
3756 if (OutStr[i] == '\n') { // Left justify
3757 OutStr[i] = '\\';
3758 OutStr.insert(OutStr.begin()+i+1, 'l');
3759 }
Mike Stump6d9828c2009-07-17 01:31:16 +00003760
Ted Kremenek7dba8602007-08-29 21:56:09 +00003761 return OutStr;
Hartmut Kaiserbd250b42007-09-16 00:28:28 +00003762#else
3763 return "";
3764#endif
Ted Kremenek7dba8602007-08-29 21:56:09 +00003765 }
3766};
3767} // end namespace llvm