Fix CFG block splitting on labels.

whitequark · May 6, 2012 · 2bd7bff · 2bd7bff
1 parent b48d66b
commit 2bd7bff
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lib/furnace-avm2/transform/cfg_build.rb b/lib/furnace-avm2/transform/cfg_build.rb
@@ -32,6 +32,10 @@ def transform(ast, body)
         @pending_queue = []
 
         ast.children.each_with_index do |node, index|
+          if node.type == :label
+            cutoff(nil, [ node.metadata[:label] ])
+          end
+
           unless @pending_label
             @pending_label = node.metadata[:label]
 
@@ -48,7 +52,6 @@ def transform(ast, body)
 
           case node.type
           when :label
-            @jumps.add node.children.first
             node.update :nop
 
           when :return_value, :return_void