diff options
-rw-r--r-- | prism_compile.c | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/prism_compile.c b/prism_compile.c index 9be075275d..b60ca0ac68 100644 --- a/prism_compile.c +++ b/prism_compile.c @@ -6153,11 +6153,16 @@ pm_compile_node(rb_iseq_t *iseq, const pm_node_t *node, LINK_ANCHOR *const ret, } case PM_FOR_NODE: { pm_for_node_t *for_node = (pm_for_node_t *)scope_node->ast_node; + LABEL *target = NEW_LABEL(lineno); + LABEL *old_start = ISEQ_COMPILE_DATA(iseq)->start_label; ADD_GETLOCAL(ret, &dummy_line_node, 1, 0); PM_COMPILE(for_node->index); PM_NOP; + ADD_LABEL(ret, target); + ISEQ_COMPILE_DATA(iseq)->start_label = target; pm_compile_node(iseq, (pm_node_t *)(scope_node->body), ret, src, popped, scope_node); + ISEQ_COMPILE_DATA(iseq)->start_label = old_start; break; } case PM_INTERPOLATED_REGULAR_EXPRESSION_NODE: { |