Update Modules/atexitmodule.c

Co-authored-by: Christian Heimes <[email protected]>

Author: pablogsal

Lib/test/_test_atexit.py

@@ -123,7 +123,11 @@ def func():
             1/0
         atexit.register(func)
         try:
-            atexit._run_exitfuncs()
+            with support.catch_unraisable_exception() as cm:
+                atexit._run_exitfuncs()
+                self.assertEqual(cm.unraisable.object, func)
+                self.assertEqual(cm.unraisable.exc_type, ZeroDivisionError)
+                self.assertEqual(type(cm.unraisable.exc_value), ZeroDivisionError)
         finally:
             atexit.unregister(func)
 

Modules/atexitmodule.c

@@ -94,8 +94,7 @@ atexit_callfuncs(struct atexit_state *state)
         }
 
         // bpo-46025: Increment the refcount of cb->func as the call itself may unregister it
-        PyObject* the_func = cb->func;
-        Py_INCREF(the_func);
+        PyObject* the_func = Py_NewRef(cb->func);
         PyObject *res = PyObject_Call(cb->func, cb->args, cb->kwargs);
         if (res == NULL) {
             _PyErr_WriteUnraisableMsg("in atexit callback", the_func);