Assert we don't get -1 from hashing exact unicode, lock in _PyObject_StoreInstanceAttribute

DinoV · DinoV · commit 8812f5b4e362 · 2024-02-29T11:14:01.000-08:00
diff --git a/Objects/dictobject.c b/Objects/dictobject.c
@@ -1597,19 +1597,11 @@ insertion_resize(PyInterpreterState *interp, PyDictObject *mp, int unicode)
 }
 
 static Py_ssize_t
-insert_into_splitdictkeys_locked(PyDictKeysObject *keys, PyObject *name)
+insert_into_splitdictkeys(PyDictKeysObject *keys, PyObject *name, Py_hash_t hash)
 {
     assert(PyUnicode_CheckExact(name));
     ASSERT_KEYS_LOCKED(keys);
 
-    Py_hash_t hash = unicode_get_hash(name);
-    if (hash == -1) {
-        hash = PyUnicode_Type.tp_hash(name);
-        if (hash == -1) {
-            PyErr_Clear();
-            return DKIX_EMPTY;
-        }
-    }
     Py_ssize_t ix = unicodekeys_lookup_unicode(keys, name, hash);
     if (ix == DKIX_EMPTY) {
         if (keys->dk_usable <= 0) {
@@ -1629,15 +1621,6 @@ insert_into_splitdictkeys_locked(PyDictKeysObject *keys, PyObject *name)
     return ix;
 }
 
-static Py_ssize_t
-insert_into_splitdictkeys(PyDictKeysObject *keys, PyObject *name)
-{
-    LOCK_KEYS(keys);
-    Py_ssize_t ix = insert_into_splitdictkeys_locked(keys, name);
-    UNLOCK_KEYS(keys);
-    return ix;
-}
-
 static inline int
 insert_combined_dict(PyInterpreterState *interp, PyDictObject *mp,
                      Py_hash_t hash, PyObject *key, PyObject *value)
@@ -6701,24 +6684,23 @@ _PyObject_StoreInstanceAttribute(PyObject *obj, PyDictValues *values,
     assert(Py_TYPE(obj)->tp_flags & Py_TPFLAGS_MANAGED_DICT);
     Py_ssize_t ix = DKIX_EMPTY;
     if (PyUnicode_CheckExact(name)) {
-#ifdef Py_GIL_DISABLED
         Py_hash_t hash = unicode_get_hash(name);
         if (hash == -1) {
             hash = PyUnicode_Type.tp_hash(name);
-            if (hash == -1) {
-                PyErr_Clear();
-                return DKIX_EMPTY;
-            }
+            assert(hash != -1);
         }
 
+#ifdef Py_GIL_DISABLED
         // Try a thread-safe lookup to see if the index is already allocated
         ix = unicodekeys_lookup_unicode_threadsafe(keys, name, hash);
         if (ix == DKIX_EMPTY) {
-            // Fall back to a version that will lock and maybe insert
-            ix = insert_into_splitdictkeys(keys, name);
+            // Lock keys and do insert
+            LOCK_KEYS(keys);
+            ix = insert_into_splitdictkeys(keys, name, hash);
+            UNLOCK_KEYS(keys);
         }
 #else
-        ix = insert_into_splitdictkeys(keys, name);
+        ix = insert_into_splitdictkeys(keys, name, hash);
 #endif
 
 #ifdef Py_STATS
