GH-123185: Check for NULL after calling _PyEvalFramePushAndInit (GH-123194)

2024-08-21 12:44:56 +01:00
parent 90c892efea
commit 1eba8bae92
5 changed files with 29 additions and 4 deletions
--- a/Python/bytecodes.c
+++ b/Python/bytecodes.c
@@ -3584,8 +3584,12 @@ dummy_func(
            args[-1] = self;
            init_frame = _PyEvalFramePushAndInit(
                tstate, init_func, NULL, args-1, oparg+1, NULL, shim);
-            frame->return_offset = 1 + INLINE_CACHE_ENTRIES_CALL;
            SYNC_SP();
+            if (init_frame == NULL) {
+                _PyEval_FrameClearAndPop(tstate, shim);
+                ERROR_NO_POP();
+            }
+            frame->return_offset = 1 + INLINE_CACHE_ENTRIES_CALL;
            /* Account for pushing the extra frame.
             * We don't check recursion depth here,
             * as it will be checked after start_frame */