gh-132713: Fix typing.Union[index] race condition (#132802)

vstinner · web-flow · commit dc3e9638c22f · 2025-04-23T15:45:42.000+02:00
Add union_init_parameters() helper function. Use a critical section
to initialize the 'parameters' member.
diff --git a/Objects/unionobject.c b/Objects/unionobject.c
@@ -322,17 +322,29 @@ static PyMemberDef union_members[] = {
         {0}
 };
 
-static PyObject *
-union_getitem(PyObject *self, PyObject *item)
+// Populate __parameters__ if needed.
+static int
+union_init_parameters(unionobject *alias)
 {
-    unionobject *alias = (unionobject *)self;
-    // Populate __parameters__ if needed.
+    int result = 0;
+    Py_BEGIN_CRITICAL_SECTION(alias);
     if (alias->parameters == NULL) {
         alias->parameters = _Py_make_parameters(alias->args);
         if (alias->parameters == NULL) {
-            return NULL;
+            result = -1;
         }
     }
+    Py_END_CRITICAL_SECTION();
+    return result;
+}
+
+static PyObject *
+union_getitem(PyObject *self, PyObject *item)
+{
+    unionobject *alias = (unionobject *)self;
+    if (union_init_parameters(alias) < 0) {
+        return NULL;
+    }
 
     PyObject *newargs = _Py_subs_parameters(self, alias->args, alias->parameters, item);
     if (newargs == NULL) {
@@ -352,11 +364,8 @@ static PyObject *
 union_parameters(PyObject *self, void *Py_UNUSED(unused))
 {
     unionobject *alias = (unionobject *)self;
-    if (alias->parameters == NULL) {
-        alias->parameters = _Py_make_parameters(alias->args);
-        if (alias->parameters == NULL) {
-            return NULL;
-        }
+    if (union_init_parameters(alias) < 0) {
+        return NULL;
     }
     return Py_NewRef(alias->parameters);
 }
