diff options
Diffstat (limited to 'Python')
| -rw-r--r-- | Python/ceval.c | 10 | ||||
| -rw-r--r-- | Python/sysmodule.c | 16 |
2 files changed, 16 insertions, 10 deletions
diff --git a/Python/ceval.c b/Python/ceval.c index 836457d1a5..afaa6ff1b3 100644 --- a/Python/ceval.c +++ b/Python/ceval.c @@ -4785,17 +4785,18 @@ _PyEval_GetCoroutineOriginTrackingDepth(void) return tstate->coroutine_origin_tracking_depth; } -void +int _PyEval_SetAsyncGenFirstiter(PyObject *firstiter) { PyThreadState *tstate = _PyThreadState_GET(); if (PySys_Audit("sys.set_asyncgen_hook_firstiter", NULL) < 0) { - return; + return -1; } Py_XINCREF(firstiter); Py_XSETREF(tstate->async_gen_firstiter, firstiter); + return 0; } PyObject * @@ -4805,17 +4806,18 @@ _PyEval_GetAsyncGenFirstiter(void) return tstate->async_gen_firstiter; } -void +int _PyEval_SetAsyncGenFinalizer(PyObject *finalizer) { PyThreadState *tstate = _PyThreadState_GET(); if (PySys_Audit("sys.set_asyncgen_hook_finalizer", NULL) < 0) { - return; + return -1; } Py_XINCREF(finalizer); Py_XSETREF(tstate->async_gen_finalizer, finalizer); + return 0; } PyObject * diff --git a/Python/sysmodule.c b/Python/sysmodule.c index c78a627380..c877fd76ac 100644 --- a/Python/sysmodule.c +++ b/Python/sysmodule.c @@ -1222,10 +1222,12 @@ sys_set_asyncgen_hooks(PyObject *self, PyObject *args, PyObject *kw) Py_TYPE(finalizer)->tp_name); return NULL; } - _PyEval_SetAsyncGenFinalizer(finalizer); + if (_PyEval_SetAsyncGenFinalizer(finalizer) < 0) { + return NULL; + } } - else if (finalizer == Py_None) { - _PyEval_SetAsyncGenFinalizer(NULL); + else if (finalizer == Py_None && _PyEval_SetAsyncGenFinalizer(NULL) < 0) { + return NULL; } if (firstiter && firstiter != Py_None) { @@ -1235,10 +1237,12 @@ sys_set_asyncgen_hooks(PyObject *self, PyObject *args, PyObject *kw) Py_TYPE(firstiter)->tp_name); return NULL; } - _PyEval_SetAsyncGenFirstiter(firstiter); + if (_PyEval_SetAsyncGenFirstiter(firstiter) < 0) { + return NULL; + } } - else if (firstiter == Py_None) { - _PyEval_SetAsyncGenFirstiter(NULL); + else if (firstiter == Py_None && _PyEval_SetAsyncGenFirstiter(NULL) < 0) { + return NULL; } Py_RETURN_NONE; |