diff options
| -rw-r--r-- | lib/AST/CFG.cpp | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/lib/AST/CFG.cpp b/lib/AST/CFG.cpp index ea29259f9e..5658386a08 100644 --- a/lib/AST/CFG.cpp +++ b/lib/AST/CFG.cpp @@ -1069,8 +1069,18 @@ CFGBlock* CFGBuilder::VisitDoStmt(DoStmt* D) { else if (Block) FinishBlock(BodyBlock); + // Add an intermediate block between the BodyBlock and the + // ExitConditionBlock to represent the "loop back" transition. + // Create an empty block to represent the transition block for looping + // back to the head of the loop. + // FIXME: Can we do this more efficiently without adding another block? + Block = NULL; + Succ = BodyBlock; + CFGBlock *LoopBackBlock = createBlock(); + LoopBackBlock->setLoopTarget(D); + // Add the loop body entry as a successor to the condition. - ExitConditionBlock->addSuccessor(BodyBlock); + ExitConditionBlock->addSuccessor(LoopBackBlock); } // Link up the condition block with the code that follows the loop. |