From 047746d2560babd0bca7848bb630433230b04481 Mon Sep 17 00:00:00 2001
From: Irit Katriel <1055913+iritkatriel@users.noreply.github.com>
Date: Fri, 22 Dec 2023 01:50:26 +0000
Subject: [PATCH] gh-113297: Fix segfault in compiler for with statement with
 19 context managers (#113327)

---
 Lib/test/test_syntax.py                       | 26 +++++++++++++++++++
 ...-12-20-18-27-11.gh-issue-113297.BZyAI_.rst |  1 +
 Python/flowgraph.c                            |  3 ++-
 3 files changed, 29 insertions(+), 1 deletion(-)
 create mode 100644 Misc/NEWS.d/next/Core and Builtins/2023-12-20-18-27-11.gh-issue-113297.BZyAI_.rst

diff --git a/Lib/test/test_syntax.py b/Lib/test/test_syntax.py
index ece1366076798c9..8b3ca69c9fe1550 100644
--- a/Lib/test/test_syntax.py
+++ b/Lib/test/test_syntax.py
@@ -2017,6 +2017,7 @@ def f(x: *b)
 
 import re
 import doctest
+import textwrap
 import unittest
 
 from test import support
@@ -2279,6 +2280,31 @@ def test_nested_named_except_blocks(self):
         code += f"{' '*4*12}pass"
         self._check_error(code, "too many statically nested blocks")
 
+    @support.cpython_only
+    def test_with_statement_many_context_managers(self):
+        # See gh-113297
+
+        def get_code(n):
+            code = textwrap.dedent("""
+                def bug():
+                    with (
+                    a
+                """)
+            for i in range(n):
+                code += f"    as a{i}, a\n"
+            code += "): yield a"
+            return code
+
+        CO_MAXBLOCKS = 20  # static nesting limit of the compiler
+
+        for n in range(CO_MAXBLOCKS):
+            with self.subTest(f"within range: {n=}"):
+                compile(get_code(n), "<string>", "exec")
+
+        for n in range(CO_MAXBLOCKS, CO_MAXBLOCKS + 5):
+            with self.subTest(f"out of range: {n=}"):
+                self._check_error(get_code(n), "too many statically nested blocks")
+
     def test_barry_as_flufl_with_syntax_errors(self):
         # The "barry_as_flufl" rule can produce some "bugs-at-a-distance" if
         # is reading the wrong token in the presence of syntax errors later
diff --git a/Misc/NEWS.d/next/Core and Builtins/2023-12-20-18-27-11.gh-issue-113297.BZyAI_.rst b/Misc/NEWS.d/next/Core and Builtins/2023-12-20-18-27-11.gh-issue-113297.BZyAI_.rst
new file mode 100644
index 000000000000000..b6aee1f241fd23a
--- /dev/null
+++ b/Misc/NEWS.d/next/Core and Builtins/2023-12-20-18-27-11.gh-issue-113297.BZyAI_.rst	
@@ -0,0 +1 @@
+Fix segfault in the compiler on with statement with 19 context managers.
diff --git a/Python/flowgraph.c b/Python/flowgraph.c
index e6c824a85ef51e0..0e6ffbc32e15269 100644
--- a/Python/flowgraph.c
+++ b/Python/flowgraph.c
@@ -648,7 +648,7 @@ mark_except_handlers(basicblock *entryblock) {
 
 
 struct _PyCfgExceptStack {
-    basicblock *handlers[CO_MAXBLOCKS+1];
+    basicblock *handlers[CO_MAXBLOCKS+2];
     int depth;
 };
 
@@ -661,6 +661,7 @@ push_except_block(struct _PyCfgExceptStack *stack, cfg_instr *setup) {
     if (opcode == SETUP_WITH || opcode == SETUP_CLEANUP) {
         target->b_preserve_lasti = 1;
     }
+    assert(stack->depth <= CO_MAXBLOCKS);
     stack->handlers[++stack->depth] = target;
     return target;
 }