Skip to content

Commit

Permalink
[3.12] gh-108295: Fix crashes with TypeVar weakrefs (GH-108517) (#108527
Browse files Browse the repository at this point in the history
)

gh-108295: Fix crashes with TypeVar weakrefs (GH-108517)
(cherry picked from commit 482fad7)

Co-authored-by: Jelle Zijlstra <jelle.zijlstra@gmail.com>
  • Loading branch information
miss-islington and JelleZijlstra committed Aug 27, 2023
1 parent 3e20303 commit 09877a1
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 0 deletions.
10 changes: 10 additions & 0 deletions Lib/test/test_typing.py
Expand Up @@ -544,6 +544,16 @@ def test_bad_var_substitution(self):
with self.assertRaises(TypeError):
list[T][arg]

def test_many_weakrefs(self):
# gh-108295: this used to segfault
for cls in (ParamSpec, TypeVarTuple, TypeVar):
with self.subTest(cls=cls):
vals = weakref.WeakValueDictionary()

for x in range(100000):
vals[x] = cls(str(x))
del vals


def template_replace(templates: list[str], replacements: dict[str, list[str]]) -> list[tuple[str]]:
"""Renders templates with possible combinations of replacements.
Expand Down
@@ -0,0 +1 @@
Fix crashes related to use of weakrefs on :data:`typing.TypeVar`.
3 changes: 3 additions & 0 deletions Objects/typevarobject.c
Expand Up @@ -201,6 +201,7 @@ typevar_dealloc(PyObject *self)
Py_XDECREF(tv->constraints);
Py_XDECREF(tv->evaluate_constraints);
_PyObject_ClearManagedDict(self);
PyObject_ClearWeakRefs(self);

Py_TYPE(self)->tp_free(self);
Py_DECREF(tp);
Expand Down Expand Up @@ -743,6 +744,7 @@ paramspec_dealloc(PyObject *self)
Py_DECREF(ps->name);
Py_XDECREF(ps->bound);
_PyObject_ClearManagedDict(self);
PyObject_ClearWeakRefs(self);

Py_TYPE(self)->tp_free(self);
Py_DECREF(tp);
Expand Down Expand Up @@ -1022,6 +1024,7 @@ typevartuple_dealloc(PyObject *self)

Py_DECREF(tvt->name);
_PyObject_ClearManagedDict(self);
PyObject_ClearWeakRefs(self);

Py_TYPE(self)->tp_free(self);
Py_DECREF(tp);
Expand Down

0 comments on commit 09877a1

Please sign in to comment.