aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--AST/CFG.cpp8
1 files changed, 5 insertions, 3 deletions
diff --git a/AST/CFG.cpp b/AST/CFG.cpp
index a2ce8a283b..f2a1caa373 100644
--- a/AST/CFG.cpp
+++ b/AST/CFG.cpp
@@ -822,9 +822,11 @@ CFGBlock* CFGBuilder::VisitSwitchCase(SwitchCase* S) {
// A SwitchCase is either a "default" or "case" statement. We handle
// both in the same way. They are essentially labels, so they are the
// first statement in a block.
- CFGBlock* CaseBlock = Visit(S->getSubStmt());
- assert (CaseBlock);
-
+
+ if (S->getSubStmt()) Visit(S->getSubStmt());
+ CFGBlock* CaseBlock = Block;
+ if (!CaseBlock) CaseBlock = createBlock();
+
// Cases/Default statements partition block, so this is the top of
// the basic block we were processing (the case/default is the label).
CaseBlock->setLabel(S);