[3.7] bpo-43660: Fix crash when displaying exceptions with custom values for sys.stderr (GH-25075). (GH-25085)

(cherry picked from commit 09b90a0)

Co-authored-by: Pablo Galindo <[email protected]>

Author: pablogsal

Lib/test/test_sys.py

@@ -1292,6 +1292,21 @@ def test_asyncgen_hooks(self):
         self.assertIsNone(cur.firstiter)
         self.assertIsNone(cur.finalizer)
 
+    def test_changing_sys_stderr_and_removing_reference(self):
+        # If the default displayhook doesn't take a strong reference
+        # to sys.stderr the following code can crash. See bpo-43660
+        # for more details.
+        code = textwrap.dedent('''
+            import sys
+            class MyStderr:
+                def write(self, s):
+                    sys.stderr = None
+            sys.stderr = MyStderr()
+            1/0
+        ''')
+        rc, out, err = assert_python_failure('-c', code)
+        self.assertEqual(out, b"")
+        self.assertEqual(err, b"")
 
 def test_main():
     test.support.run_unittest(SysModuleTest, SizeofTest)

Misc/NEWS.d/next/Core and Builtins/2021-03-29-19-50-34.bpo-43660.scTgag.rst

@@ -0,0 +1,3 @@
+Fix crash that happens when replacing ``sys.stderr`` with a callable that
+can remove the object while an exception is being printed. Patch by Pablo
+Galindo.

Python/pythonrun.c

@@ -932,7 +932,9 @@ PyErr_Display(PyObject *exception, PyObject *value, PyObject *tb)
         seen = PySet_New(NULL);
         if (seen == NULL)
             PyErr_Clear();
+        Py_INCREF(f);
         print_exception_recursive(f, value, seen);
+        Py_DECREF(f);
         Py_XDECREF(seen);
     }
 }