python · priyanshu2282-cyber · Feb 15, 2026
@@ -1955,6 +1955,21 @@ def test_exec_set_nomemory_hang(self):
         self.assertGreater(len(output), 0)  # At minimum, should not hang
         self.assertIn(b"MemoryError", output)
 
+    def test_setstate_thread_safety(self):
+        import threading
+        import random
+        exc = Exception()
+        def worker():
+            for _ in range(100):
+                setattr(exc, "x", random.randint(0, 1000))
+                copy.copy(exc)
+
+        threads = [threading.Thread(target=worker) for _ in range(4)]
+        for t in threads:
+            t.start()
+        for t in threads:
+            t.join()
+        self.assertTrue(True)
 
 class NameErrorTests(unittest.TestCase):
     def test_name_error_has_name(self):

@@ -242,16 +242,21 @@
             PyErr_SetString(PyExc_TypeError, "state is not a dictionary");
             return NULL;
         }
+        PyCriticalSection cs;
+        PyThreadState *tstate = _PyThreadState_GET();
+        _PyCriticalSection_Begin(tstate, &cs, state);
         while (PyDict_Next(state, &i, &d_key, &d_value)) {
             Py_INCREF(d_key);
             Py_INCREF(d_value);
             int res = PyObject_SetAttr((PyObject *)self, d_key, d_value);
             Py_DECREF(d_value);
             Py_DECREF(d_key);
             if (res < 0) {
+                _PyCriticalSection_End(tstate, &cs);
                 return NULL;
             }
         }
+        _PyCriticalSection_End(tstate, &cs);
     }
     Py_RETURN_NONE;
 }