Fix a corner case of throw.

whitequark · Aug 19, 2012 · a7d5133 · a7d5133
1 parent aebfc78
commit a7d5133
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/lib/furnace-avm2/transform/liveness_analysis.rb b/lib/furnace-avm2/transform/liveness_analysis.rb
@@ -17,19 +17,21 @@ def transform(cfg)
         dead_ends = Set[ cfg.exit ]
 
         # Search from the entry node, mark live variables
-        worklist = Set[ cfg.entry ]
+        worklist = Set[ cfg.entry, cfg.exit ]
         while worklist.any?
           block = worklist.first
           worklist.delete block
 
-          if block.cti && block.cti.type == :throw
-            dead_ends.add block
-          elsif loops.include? block
+          if loops.include? block
             back_edged, sources = block.sources.partition do |source|
               dom[source].include? block
             end
             dead_ends.merge back_edged
           else
+            if block.cti && block.cti.type == :throw
+              dead_ends.add block
+            end
+
             sources = block.sources
           end