gh-126366: Fix crash if __iter__ raises an exception during yield from (#126369)

2026-05-06 20:51:17 -04:00 · 2024-11-05 04:56:36 -05:00
parent 407c0366d9
commit 1371295e67
6 changed files with 25 additions and 6 deletions
@@ -2811,11 +2811,12 @@ dummy_func(
            }
            else {
                /* `iterable` is not a generator. */
-                iter = PyStackRef_FromPyObjectSteal(PyObject_GetIter(iterable_o));
+                PyObject *iter_o = PyObject_GetIter(iterable_o);
                DEAD(iterable);
-                if (PyStackRef_IsNull(iter)) {
+                if (iter_o == NULL) {
                    ERROR_NO_POP();
                }
+                iter = PyStackRef_FromPyObjectSteal(iter_o);
                DECREF_INPUTS();
            }
        }