From 179869af922252a0c1cef65fd2923856895e7d1b Mon Sep 17 00:00:00 2001 From: Petr Viktorin Date: Mon, 1 Apr 2024 17:01:22 +0200 Subject: [PATCH] gh-94808: Fix refcounting in PyObject_Print tests (GH-117421) --- Modules/_testcapi/object.c | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/Modules/_testcapi/object.c b/Modules/_testcapi/object.c index ce5b574ec5c..8dd34cf4fc4 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; }