Handle no memory case

python · May 1, 2023 · 2a09314 · 2a09314
1 parent 6977273
commit 2a09314
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/Modules/gcmodule.c b/Modules/gcmodule.c
@@ -1390,6 +1390,12 @@ invoke_gc_callback(PyThreadState *tstate, const char *phase,
     }
 
     PyObject *phase_obj = PyUnicode_FromString(phase);
+    if (phase_obj == NULL) {
+        Py_XDECREF(info);
+        PyErr_WriteUnraisable(NULL);
+        return;
+    }
+
     PyObject *stack[] = {phase_obj, info};
     for (Py_ssize_t i=0; i<PyList_GET_SIZE(gcstate->callbacks); i++) {
         PyObject *r, *cb = PyList_GET_ITEM(gcstate->callbacks, i);