Add zero PyObject pointer null-check

Licht-T · Licht-T · commit 41cd4e123dd7 · 2017-12-28T09:32:36.000+09:00
diff --git a/numpy/core/src/multiarray/ctors.c b/numpy/core/src/multiarray/ctors.c
@@ -3051,11 +3051,15 @@ static npy_intp
 _calc_length(PyObject *start, PyObject *stop, PyObject *step, PyObject **next, int cmplx)
 {
     npy_intp len, tmp;
-    PyObject *zero = PyInt_FromLong(0);
-    PyObject *val;
+    PyObject *zero, *val;
     int next_is_nonzero, val_is_zero;
     double value;
 
+    zero = PyInt_FromLong(0);
+    if (!zero) {
+        return -1;
+    }
+
     *next = PyNumber_Subtract(stop, start);
     if (!(*next)) {
         Py_DECREF(zero);
