From 2e761e571f7b3d3df7ba7407a13b37c81ee4e9eb Mon Sep 17 00:00:00 2001
From: Alon Zakai <alonzakai@gmail.com>
Date: Fri, 4 Mar 2016 18:36:05 -0800
Subject: add a finalize() method for blocks

---
 src/s2wasm.h | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'src/s2wasm.h')

diff --git a/src/s2wasm.h b/src/s2wasm.h
index 0520d5c3b..2de6cd60f 100644
--- a/src/s2wasm.h
+++ b/src/s2wasm.h
@@ -528,6 +528,7 @@ class S2WasmBuilder {
         last = last->cast<Loop>()->body;
       }
       last->cast<Block>()->list.push_back(curr);
+      last->cast<Block>()->finalize();
     };
     bstack.push_back(func->body);
     std::vector<Expression*> estack;
@@ -1023,6 +1024,7 @@ class S2WasmBuilder {
     for (auto block : loopBlocks) {
       block->name = Name();
     }
+    func->body->dyn_cast<Block>()->finalize();
     wasm.addFunction(func);
     // XXX for now, export all functions
     auto exp = allocator.alloc<Export>();
@@ -1243,6 +1245,7 @@ class S2WasmBuilder {
           call->operands.push_back(param);
         }
         block->list.push_back(call);
+        block->finalize();
       }
     }
   }
-- 
cgit v1.2.3