]> git.ipfire.org Git - thirdparty/Python/cpython.git/commitdiff
gh-111926: Simplify weakref creation logic (#116843)
authormpage <mpage@meta.com>
Fri, 15 Mar 2024 13:56:13 +0000 (06:56 -0700)
committerGitHub <noreply@github.com>
Fri, 15 Mar 2024 13:56:13 +0000 (09:56 -0400)
Since 3.12, allocating a GC object cannot immediately trigger GC. This
allows us to simplify the logic for creating the canonical callback-less
weakref.

Objects/weakrefobject.c

index df74be6aba72445eafa083d6572f4a3dc890810b..6264b1825eda73141f4f830bb4589e53273ccc51 100644 (file)
@@ -801,24 +801,14 @@ PyWeakref_NewRef(PyObject *ob, PyObject *callback)
     if (result != NULL)
         Py_INCREF(result);
     else {
-        /* Note: new_weakref() can trigger cyclic GC, so the weakref
-           list on ob can be mutated.  This means that the ref and
-           proxy pointers we got back earlier may have been collected,
-           so we need to compute these values again before we use
-           them. */
+        /* We do not need to recompute ref/proxy; new_weakref() cannot
+           trigger GC.
+        */
         result = new_weakref(ob, callback);
         if (result != NULL) {
-            get_basic_refs(*list, &ref, &proxy);
             if (callback == NULL) {
-                if (ref == NULL)
-                    insert_head(result, list);
-                else {
-                    /* Someone else added a ref without a callback
-                       during GC.  Return that one instead of this one
-                       to avoid violating the invariants of the list
-                       of weakrefs for ob. */
-                    Py_SETREF(result, (PyWeakReference*)Py_NewRef(ref));
-                }
+                assert(ref == NULL);
+                insert_head(result, list);
             }
             else {
                 PyWeakReference *prev;