check frozenset return

WolframAlph · WolframAlph · commit b3a2aeb18b9f · 2025-02-12T13:29:42.000+01:00
diff --git a/Python/flowgraph.c b/Python/flowgraph.c
@@ -1580,6 +1580,10 @@ optimize_if_const_list_or_set_iter(basicblock *bb, int n,
     }
     if (instr->i_opcode == BUILD_SET) {
         PyObject *frozenset = PyFrozenSet_New(newconst);
+        if (frozenset == NULL) {
+            Py_DECREF(newconst);
+            return ERROR;
+        }
         Py_SETREF(newconst, frozenset);
     }
     int index = add_const(newconst, consts, const_cache);

Original file line number	Diff line number	Diff line change
`@@ -1580,6 +1580,10 @@ optimize_if_const_list_or_set_iter(basicblock *bb, int n,`
`1580`	`1580`	`}`
`1581`	`1581`	`if (instr->i_opcode == BUILD_SET) {`
`1582`	`1582`	`PyObject *frozenset = PyFrozenSet_New(newconst);`
	`1583`	`+ if (frozenset == NULL) {`
	`1584`	`+ Py_DECREF(newconst);`
	`1585`	`+ return ERROR;`
	`1586`	`+ }`
`1583`	`1587`	`Py_SETREF(newconst, frozenset);`
`1584`	`1588`	`}`
`1585`	`1589`	`int index = add_const(newconst, consts, const_cache);`