Skip to content

Commit f494440

Browse files
LindaSummerroot
authored andcommitted
fix PyTypeObject->tp_base race in free threading
1 parent 6d45cd8 commit f494440

File tree

3 files changed

+24
-0
lines changed

3 files changed

+24
-0
lines changed

Lib/test/test_free_threading/test_type.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,25 @@ def reader():
141141

142142
self.run_one(writer, reader)
143143

144+
def test_bases_change(self):
145+
class BaseA:
146+
pass
147+
148+
class Derived(BaseA):
149+
pass
150+
151+
def writer():
152+
for _ in range(1000):
153+
class BaseB:
154+
pass
155+
Derived.__bases__ = (BaseB,)
156+
157+
def reader():
158+
for _ in range(1000):
159+
Derived.__base__
160+
161+
self.run_one(writer, reader)
162+
144163
def run_one(self, writer_func, reader_func):
145164
barrier = threading.Barrier(NTHREADS)
146165

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Fix race when updating ``type.__bases__`` that could allow a read of ``type.__base__`` to observe an inconsistent value on the free threaded build.

Objects/typeobject.c

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1920,8 +1920,10 @@ type_set_bases_unlocked(PyTypeObject *type, PyObject *new_bases, PyTypeObject *b
19201920
assert(old_bases != NULL);
19211921
PyTypeObject *old_base = type->tp_base;
19221922

1923+
types_stop_world();
19231924
set_tp_bases(type, Py_NewRef(new_bases), 0);
19241925
type->tp_base = (PyTypeObject *)Py_NewRef(best_base);
1926+
types_start_world();
19251927

19261928
PyObject *temp = PyList_New(0);
19271929
if (temp == NULL) {
@@ -1982,8 +1984,10 @@ type_set_bases_unlocked(PyTypeObject *type, PyObject *new_bases, PyTypeObject *b
19821984
if (lookup_tp_bases(type) == new_bases) {
19831985
assert(type->tp_base == best_base);
19841986

1987+
types_stop_world();
19851988
set_tp_bases(type, old_bases, 0);
19861989
type->tp_base = old_base;
1990+
types_start_world();
19871991

19881992
Py_DECREF(new_bases);
19891993
Py_DECREF(best_base);

0 commit comments

Comments
 (0)