[3.13] Free arena on _PyCompile_AstOptimize failure in Py_CompileStringObject (GH...

Free arena on _PyCompile_AstOptimize failure in Py_CompileStringObject (GH-127910)

After commit 10a91d7e9 introduced arena cleanup, commit 2dfbd4f36
removed the free call when _PyCompile_AstOptimize fails.

(cherry picked from commit cfeaa992ba9bad9be2687afcafd85156703d74e8)

Co-authored-by: Berker Peksag <berker.peksag@gmail.com>

diff --git a/Python/pythonrun.c b/Python/pythonrun.c
index 5891d50cd3f598f7cf02784fc7712cc54b7a1b9e..f9daf8168f1c5208e85ff0c2ccfa7d13e23d1319 100644 (file)
--- a/Python/pythonrun.c
+++ b/Python/pythonrun.c
@@ -1485,6 +1485,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;
             }
         }