[PRISM] Rescue should set correct end_label

In order for a break inside the rescue to have the correct jump target
ruby · Dec 7, 2023 · c4b9695 · c4b9695
1 parent 071df40
commit c4b9695
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/prism_compile.c b/prism_compile.c
@@ -3851,6 +3851,8 @@ pm_compile_node(rb_iseq_t *iseq, const pm_node_t *node, LINK_ANCHOR *const ret,
         LABEL *excep_match = NEW_LABEL(lineno);
         LABEL *rescue_end = NEW_LABEL(lineno);
 
+        ISEQ_COMPILE_DATA(iseq)->end_label = rescue_end;
+
         pm_rescue_node_t *rescue_node = (pm_rescue_node_t *)node;
         iseq_set_exception_local_table(iseq);
 

diff --git a/test/ruby/test_compile_prism.rb b/test/ruby/test_compile_prism.rb
@@ -950,6 +950,14 @@ def test_RescueNode
           end
         end
       CODE
+      assert_prism_eval(<<~CODE)
+        10.times do
+          begin
+          rescue
+            break
+          end
+        end
+      CODE
     end
 
     def test_RescueModiferNode