add assert size * itemisze < size_t in buffer alloc func

tom-pytel · tom-pytel · commit b8bdb13ac516 · 2025-07-23T07:51:00.000-04:00
diff --git a/Modules/arraymodule.c b/Modules/arraymodule.c
@@ -154,6 +154,7 @@ enum machine_format_code {
 static arraydata *
 arraydata_alloc(Py_ssize_t size, int itemsize)
 {
+    assert (size <= PY_SSIZE_T_MAX / itemsize);
     Py_ssize_t bufsize = sizeof(arraydata) + size * itemsize;
     arraydata *data = (arraydata *)PyMem_Malloc(bufsize);
     if (data == NULL) {

Original file line number	Diff line number	Diff line change
`@@ -154,6 +154,7 @@ enum machine_format_code {`
`154`	`154`	`static arraydata *`
`155`	`155`	`arraydata_alloc(Py_ssize_t size, int itemsize)`
`156`	`156`	`{`
	`157`	`+ assert (size <= PY_SSIZE_T_MAX / itemsize);`
`157`	`158`	`Py_ssize_t bufsize = sizeof(arraydata) + size * itemsize;`
`158`	`159`	`arraydata data = (arraydata )PyMem_Malloc(bufsize);`
`159`	`160`	`if (data == NULL) {`