WIP.

.gitignore

@@ -2,3 +2,5 @@
 *~
 .*.swp
 Cargo.lock
+/*.wasm
+wasm_tests/*.wasm

src/backend/mod.rs

@@ -25,7 +25,7 @@ impl<'a> WasmBackend<'a> {
         log::trace!("RPO:\n{:?}\n", rpo);
         let trees = Trees::compute(body);
         log::trace!("Trees:\n{:?}\n", trees);
-        let ctrl = StackifyContext::new(body, &cfg, &rpo).compute();
+        let ctrl = StackifyContext::new(body, &cfg, &rpo)?.compute();
         log::trace!("Ctrl:\n{:?}\n", ctrl);
         Ok(WasmBackend {
             body,

src/backend/stackify.rs

@@ -96,17 +96,17 @@ impl CtrlEntry {
 }
 
 impl<'a, 'b> Context<'a, 'b> {
-    pub fn new(body: &'a FunctionBody, cfg: &'b CFGInfo, rpo: &'b RPO) -> Self {
+    pub fn new(body: &'a FunctionBody, cfg: &'b CFGInfo, rpo: &'b RPO) -> anyhow::Result<Self> {
         let (merge_nodes, loop_headers) =
-            Self::compute_merge_nodes_and_loop_headers(body, cfg, rpo);
-        Self {
+            Self::compute_merge_nodes_and_loop_headers(body, cfg, rpo)?;
+        Ok(Self {
             body,
             cfg,
             rpo,
             merge_nodes,
             loop_headers,
             ctrl_stack: vec![],
-        }
+        })
     }
 
     pub fn compute(mut self) -> Vec<WasmBlock<'a>> {
@@ -119,7 +119,7 @@ impl<'a, 'b> Context<'a, 'b> {
         body: &FunctionBody,
         cfg: &CFGInfo,
         rpo: &RPO,
-    ) -> (HashSet<Block>, HashSet<Block>) {
+    ) -> anyhow::Result<(HashSet<Block>, HashSet<Block>)> {
         let mut loop_headers = HashSet::new();
         let mut branched_once = HashSet::new();
         let mut merge_nodes = HashSet::new();
@@ -128,6 +128,9 @@ impl<'a, 'b> Context<'a, 'b> {
             for &succ in cfg.succs(block) {
                 let succ_rpo = rpo.rev[succ].unwrap();
                 if succ_rpo <= block_rpo {
+                    if !cfg.dominates(succ, block) {
+                        anyhow::bail!("Irreducible control flow: edge from {} to {}", block, succ);
+                    }
                     // Backward branch.
                     loop_headers.insert(succ);
                 } else {
@@ -155,7 +158,7 @@ impl<'a, 'b> Context<'a, 'b> {
             }
         }
 
-        (merge_nodes, loop_headers)
+        Ok((merge_nodes, loop_headers))
     }
 
     fn handle_dom_subtree(&mut self, block: Block, into: &mut Vec<WasmBlock<'a>>) {

wasm_tests/test.wat

@@ -0,0 +1,6 @@
+(module
+  (func (result i32)
+        i32.const 2
+        i32.const 2
+        i32.add
+        ))

wasm_tests/test2.wat

@@ -0,0 +1,12 @@
+(module
+  (func (param i32) (result i32)
+        local.get 0
+        if (result i32)
+          i32.const 1
+          local.get 0
+          i32.add
+        else
+          i32.const 2
+          local.get 0
+          i32.add
+        end))

wasm_tests/test3.wat

@@ -0,0 +1,13 @@
+(module
+  (func (param i32) (result i32)
+        (local $i i32)
+        (local.set $i (local.get 0))
+
+        (local.get $i)
+        (block $loop-break (param i32) (result i32)
+            (loop $l (param i32) (result i32)
+                  (if (i32.eq (local.get $i) (i32.const 100))
+                    (br $loop-break
+                      (local.get $i)))
+                  (local.set $i (i32.add (i32.const 10) (local.get $i)))
+                  (br $l (local.get $i))))))