bpo-40696: Fix a hang that can arise after gen.throw() (GH-20287)
This updates _PyErr_ChainStackItem() to use _PyErr_SetObject()
instead of _PyErr_ChainExceptions(). This prevents a hang in
certain circumstances because _PyErr_SetObject() performs checks
to prevent cycles in the exception context chain while
_PyErr_ChainExceptions() doesn't.
(cherry picked from commit 7c30d12bd5359b0f66c4fbc98aa055398bcc8a7e)
Co-authored-by: Chris Jerdonek <chris.jerdonek@gmail.com>
diff --git a/Modules/_asynciomodule.c b/Modules/_asynciomodule.c
index 1b6a579..0608c40 100644
--- a/Modules/_asynciomodule.c
+++ b/Modules/_asynciomodule.c
@@ -612,19 +612,20 @@
}
static void
-set_cancelled_error(PyObject *msg)
+future_set_cancelled_error(FutureObj *fut)
{
- PyObject *exc = create_cancelled_error(msg);
+ PyObject *exc = create_cancelled_error(fut->fut_cancel_msg);
PyErr_SetObject(asyncio_CancelledError, exc);
Py_DECREF(exc);
+
+ _PyErr_ChainStackItem(&fut->fut_cancelled_exc_state);
}
static int
future_get_result(FutureObj *fut, PyObject **result)
{
if (fut->fut_state == STATE_CANCELLED) {
- set_cancelled_error(fut->fut_cancel_msg);
- _PyErr_ChainStackItem(&fut->fut_cancelled_exc_state);
+ future_set_cancelled_error(fut);
return -1;
}
@@ -866,8 +867,7 @@
}
if (self->fut_state == STATE_CANCELLED) {
- set_cancelled_error(self->fut_cancel_msg);
- _PyErr_ChainStackItem(&self->fut_cancelled_exc_state);
+ future_set_cancelled_error(self);
return NULL;
}