From: Petr Viktorin <encukou@gmail.com>
Date: Mon, 1 Apr 2024 15:01:22 +0000 (+0200)
Subject: gh-94808: Fix refcounting in PyObject_Print tests (GH-117421)
X-Git-Tag: v3.13.0a6~86
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=179869af922252a0c1cef65fd2923856895e7d1b;p=thirdparty%2FPython%2Fcpython.git

gh-94808: Fix refcounting in PyObject_Print tests (GH-117421)
---

diff --git a/Modules/_testcapi/object.c b/Modules/_testcapi/object.c
index ce5b574ec5ce..8dd34cf4fc47 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;
 }