From 0ba03ac3f0f9f7791e033a99b4fcc8d25da0f171 Mon Sep 17 00:00:00 2001
From: Berker Peksag <berker.peksag@gmail.com>
Date: Fri, 13 Dec 2024 13:44:57 +0200
Subject: [PATCH] Free arena on _PyCompile_AstOptimize failure in
 Py_CompileStringObject

After commit 10a91d7e9 introduced arena cleanup, commit 2dfbd4f36
removed the free call when _PyCompile_AstOptimize fails.
---
 Python/pythonrun.c | 1 +
 1 file changed, 1 insertion(+)

diff --git a/Python/pythonrun.c b/Python/pythonrun.c
index 8b57018321c070..31e065ff00d59a 100644
--- a/Python/pythonrun.c
+++ b/Python/pythonrun.c
@@ -1486,6 +1486,7 @@ Py_CompileStringObject(const char *str, PyObject *filename, int start,
     if (flags && (flags->cf_flags & PyCF_ONLY_AST)) {
         if ((flags->cf_flags & PyCF_OPTIMIZED_AST) == PyCF_OPTIMIZED_AST) {
             if (_PyCompile_AstOptimize(mod, filename, flags, optimize, arena) < 0) {
+                _PyArena_Free(arena);
                 return NULL;
             }
         }