diff options
-rw-r--r-- | src/de/fernflower/modules/decompiler/DomHelper.java | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/de/fernflower/modules/decompiler/DomHelper.java b/src/de/fernflower/modules/decompiler/DomHelper.java index 9377b30..7b3d46f 100644 --- a/src/de/fernflower/modules/decompiler/DomHelper.java +++ b/src/de/fernflower/modules/decompiler/DomHelper.java @@ -60,17 +60,17 @@ public class DomHelper { stats.addWithKey(new BasicBlockStatement(block), block.id); } + BasicBlock firstblock = graph.getFirst(); // head statement - Statement firstst = stats.getWithKey(graph.getFirst().id); + Statement firstst = stats.getWithKey(firstblock.id); // dummy exit statement Statement dummyexit = new Statement(); dummyexit.type = Statement.TYPE_DUMMYEXIT; Statement general; - if(stats.size() > 1) { + if(stats.size() > 1 || firstblock.isSuccessor(firstblock)) { // multiple basic blocks or an infinite loop of one block general = new GeneralStatement(firstst, stats, null); - } else { - // special case: one basic block + } else { // one straightforward basic block RootStatement root = new RootStatement(firstst, dummyexit); firstst.addSuccessor(new StatEdge(StatEdge.TYPE_BREAK, firstst, dummyexit, root)); |