Generators

Author: mpage

Python/bytecodes.c

@@ -1201,14 +1201,16 @@ dummy_func(
 
             PyObject *retval_o;
             assert(frame->owner != FRAME_OWNED_BY_INTERPRETER);
+            _PyStackRef tmp = _PyStackRef_StealIfUnborrowed(v);
+            DEAD(v);
             if ((tstate->interp->eval_frame == NULL) &&
                 (Py_TYPE(receiver_o) == &PyGen_Type || Py_TYPE(receiver_o) == &PyCoro_Type) &&
                 ((PyGenObject *)receiver_o)->gi_frame_state < FRAME_EXECUTING)
             {
                 PyGenObject *gen = (PyGenObject *)receiver_o;
                 _PyInterpreterFrame *gen_frame = &gen->gi_iframe;
                 STACK_SHRINK(1);
-                _PyFrame_StackPush(gen_frame, v);
+                _PyFrame_StackPush(gen_frame, tmp);
                 gen->gi_frame_state = FRAME_EXECUTING;
                 gen->gi_exc_state.previous_item = tstate->exc_info;
                 tstate->exc_info = &gen->gi_exc_state;
@@ -1224,7 +1226,7 @@ dummy_func(
             else {
                 retval_o = PyObject_CallMethodOneArg(receiver_o,
                                                      &_Py_ID(send),
-                                                     PyStackRef_AsPyObjectBorrow(v));
+                                                     PyStackRef_AsPyObjectBorrow(tmp));
             }
             if (retval_o == NULL) {
                 int matches = _PyErr_ExceptionMatches(tstate, PyExc_StopIteration);
@@ -1237,11 +1239,11 @@ dummy_func(
                     JUMPBY(oparg);
                 }
                 else {
-                    PyStackRef_CLOSE(v);
+                    PyStackRef_CLOSE(tmp);
                     ERROR_IF(true, error);
                 }
             }
-            PyStackRef_CLOSE(v);
+            PyStackRef_CLOSE(tmp);
             retval = PyStackRef_FromPyObjectSteal(retval_o);
         }
 
@@ -1253,7 +1255,7 @@ dummy_func(
             DEOPT_IF(gen->gi_frame_state >= FRAME_EXECUTING);
             STAT_INC(SEND, hit);
             gen_frame = &gen->gi_iframe;
-            _PyFrame_StackPush(gen_frame, v);
+            _PyFrame_StackPush(gen_frame, _PyStackRef_StealIfUnborrowed(v));
             DEAD(v);
             gen->gi_frame_state = FRAME_EXECUTING;
             gen->gi_exc_state.previous_item = tstate->exc_info;
@@ -1300,7 +1302,7 @@ dummy_func(
             #endif
             RELOAD_STACK();
             LOAD_IP(1 + INLINE_CACHE_ENTRIES_SEND);
-            value = temp;
+            value = _PyStackRef_StealIfUnborrowed(temp);
             LLTRACE_RESUME_FRAME();
         }