From f3938024184df77965c2ab91c48facdb9cb0ad3c Mon Sep 17 00:00:00 2001
From: Volker Berlin <volker.berlin@googlemail.com>
Date: Sun, 29 May 2022 10:47:50 +0200
Subject: [PATCH] fix BranchNode order for some complex constructs. #43

---
 .../jwebassembly/module/BranchManager.java    | 24 ++++++++++++-------
 1 file changed, 15 insertions(+), 9 deletions(-)

diff --git a/src/de/inetsoftware/jwebassembly/module/BranchManager.java b/src/de/inetsoftware/jwebassembly/module/BranchManager.java
index 52aae044..e57bafd3 100644
--- a/src/de/inetsoftware/jwebassembly/module/BranchManager.java
+++ b/src/de/inetsoftware/jwebassembly/module/BranchManager.java
@@ -554,7 +554,7 @@ private void calculateIf( BranchNode parent, IfParsedBlock startBlock, List<Pars
         }
 
         if( createThenBlock ) {
-            // handle jumps outside the then but inside calling parent
+            // handle jumps outside the then but inside calling parent and create the needed blocks in the hierarchy. Typical this is the ELSE block.
             for( int i = 0; i < parsedOperations.size(); i++ ) {
                 ParsedBlock block = parsedOperations.get( i );
                 if( block.startPosition >= endPos ) {
@@ -566,17 +566,14 @@ private void calculateIf( BranchNode parent, IfParsedBlock startBlock, List<Pars
                         BranchNode nodeParent = branch;
                         BranchNode node;
                         do {
-                            node = nodeParent; 
+                            node = nodeParent;
                             nodeParent = nodeParent.parent;
-                        }
-                        while( nodeParent.endPos < endPosition );
+                        } while( nodeParent.endPos < endPosition );
                         if( nodeParent.endPos == endPosition ) {
+                            // there is already a block end on this position
                             continue;
                         }
-                        nodeParent.remove( node );
-                        BranchNode elseBranch = new BranchNode( branch.startPos, endPosition, WasmBlockOperator.BLOCK, WasmBlockOperator.END );
-                        elseBranch.add( node );
-                        nodeParent.add( elseBranch );
+                        addMiddleNode( nodeParent, branch.startPos, endPosition );
                     }
                 }
             }
@@ -1446,6 +1443,15 @@ private void calculateBreak( BreakBlock breakBlock ) {
             insertConstBeforePosition( 0, breakBlock.breakPos, -1 );
         }
         BranchNode breakNode = new BranchNode( breakBlock.breakPos, breakBlock.breakPos, op, null, deep + 1 );
+        // add the BranchNode on the right position
+        int childCount = branch.size();
+        for( int i = 0; i < childCount; i++ ) {
+            BranchNode child = branch.get( i );
+            if( child.endPos> breakBlock.breakPos ) {
+                branch.add( i, breakNode );
+                return;
+            }
+        }
         branch.add( breakNode );
     }
 
@@ -1720,7 +1726,7 @@ private static class BranchNode extends ArrayList<BranchNode> {
         public boolean add( BranchNode node ) {
             node.parent = this;
             assert node.startOp == null || (node.startPos >= startPos && node.endPos <= endPos): "Node outside parent: " + this + " + " + node;
-//            assert !overlapped( node.startPos ) : "Node on wrong level: " + node + "; parent: " + this + "; last: " + get( size() - 1 );
+            assert node.startOp == null || !overlapped( node.startPos ) : "Node on wrong level: " + node + "; parent: " + this + "; last: " + get( size() - 1 );
             return super.add( node );
         }