From eaa0fbf9b956fa25e73c3d55e2eba8887324e233 Mon Sep 17 00:00:00 2001
From: Nobuyoshi Nakada <nobu@ruby-lang.org>
Date: Fri, 29 Sep 2023 01:58:07 +0900
Subject: Fix `retry` in nested `rescue` blocks

Restore `rescue`-context from the outer context.
`retry` targets the next outer block except for between `rescue` and
`else` or `ensure`, otherwise, if there is no enclosing block, it
should be syntax error.
---
 parse.y | 33 +++++++++++++++++++++++----------
 1 file changed, 23 insertions(+), 10 deletions(-)

(limited to 'parse.y')

diff --git a/parse.y b/parse.y
index e9ebfebf61..b572ab488a 100644
--- a/parse.y
+++ b/parse.y
@@ -1571,6 +1571,12 @@ static NODE *allow_block_exit(struct parser_params *p);
 static void restore_block_exit(struct parser_params *p, NODE *exits);
 static void clear_block_exit(struct parser_params *p, bool error);
 
+static void
+next_rescue_context(struct lex_context *next, const struct lex_context *outer, enum rescue_context def)
+{
+    next->in_rescue = outer->in_rescue == after_rescue ? after_rescue : def;
+}
+
 static void
 restore_defun(struct parser_params *p, NODE *name)
 {
@@ -2116,29 +2122,37 @@ begin_block	: block_open top_compstmt '}'
                     }
                 ;
 
-bodystmt	: compstmt
+bodystmt	: compstmt[body]
+                  lex_ctxt[ctxt]
                   opt_rescue
                   k_else
                     {
-                        if (!$2) yyerror1(&@3, "else without rescue is useless");
-                        p->ctxt.in_rescue = after_else;
+                        if (!$opt_rescue) yyerror1(&@k_else, "else without rescue is useless");
+                        next_rescue_context(&p->ctxt, &$ctxt, after_else);
+                    }
+                  compstmt[elsebody]
+                    {
+                        next_rescue_context(&p->ctxt, &$ctxt, after_ensure);
                     }
-                  compstmt
                   opt_ensure
                     {
                     /*%%%*/
-                        $$ = new_bodystmt(p, $1, $2, $5, $6, &@$);
+                        $$ = new_bodystmt(p, $body, $opt_rescue, $elsebody, $opt_ensure, &@$);
                     /*% %*/
-                    /*% ripper: bodystmt!($1, $2, $5, $6) %*/
+                    /*% ripper: bodystmt!($body, $opt_rescue, $elsebody, $opt_ensure) %*/
                     }
-                | compstmt
+                | compstmt[body]
+                  lex_ctxt[ctxt]
                   opt_rescue
+                    {
+                        next_rescue_context(&p->ctxt, &$ctxt, after_ensure);
+                    }
                   opt_ensure
                     {
                     /*%%%*/
-                        $$ = new_bodystmt(p, $1, $2, 0, $3, &@$);
+                        $$ = new_bodystmt(p, $body, $opt_rescue, 0, $opt_ensure, &@$);
                     /*% %*/
-                    /*% ripper: bodystmt!($1, $2, Qnil, $3) %*/
+                    /*% ripper: bodystmt!($body, $opt_rescue, Qnil, $opt_ensure) %*/
                     }
                 ;
 
@@ -4242,7 +4256,6 @@ k_ensure	: keyword_ensure
                     {
                         token_info_warn(p, "ensure", p->token_info, 1, &@$);
                         $$ = p->ctxt;
-                        p->ctxt.in_rescue = after_ensure;
                     }
                 ;
 
-- 
cgit v1.2.3