MAINT: Add a check of the return value of PyMem_Calloc().

octachrome · Jun 16, 2022 · 570e59e · 570e59e
1 parent 2d44524
commit 570e59e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/numpy/core/src/multiarray/textreading/readtext.c b/numpy/core/src/multiarray/textreading/readtext.c
@@ -270,6 +270,10 @@ _load_from_filelike(PyObject *NPY_UNUSED(mod),
         }
         /* Calloc just to not worry about overflow */
         usecols = PyMem_Calloc(num_usecols, sizeof(Py_ssize_t));
+        if (usecols == NULL) {
+            PyErr_NoMemory();
+            return NULL;
+        }
         for (Py_ssize_t i = 0; i < num_usecols; i++) {
             PyObject *tmp = PySequence_GetItem(usecols_obj, i);
             if (tmp == NULL) {