From abea62caba879d83839b965ca3e34219ba6e2a1b Mon Sep 17 00:00:00 2001 From: Petr Viktorin Date: Mon, 1 Apr 2024 16:28:54 +0200 Subject: [PATCH] gh-94808: Fix refcounting in PyObject_Print tests --- Modules/_testcapi/object.c | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/Modules/_testcapi/object.c b/Modules/_testcapi/object.c index ce5b574ec5ce8f..8dd34cf4fc47d4 100644 --- a/Modules/_testcapi/object.c +++ b/Modules/_testcapi/object.c @@ -76,6 +76,8 @@ pyobject_print_noref_object(PyObject *self, PyObject *args) if (PyObject_Print(test_string, fp, 0) < 0){ fclose(fp); + Py_SET_REFCNT(test_string, 1); + Py_DECREF(test_string); return NULL; } @@ -105,10 +107,12 @@ pyobject_print_os_error(PyObject *self, PyObject *args) if (PyObject_Print(test_string, fp, 0) < 0) { fclose(fp); + Py_DECREF(test_string); return NULL; } fclose(fp); + Py_DECREF(test_string); Py_RETURN_NONE; }