Fix possible NULL pointer dereference in PyCurses_Start_Color()

tiran · tiran · commit ef86368ea6f9 · 2013-07-26T14:46:02.000+02:00
CID 1058276
diff --git a/Modules/_cursesmodule.c b/Modules/_cursesmodule.c
@@ -2926,9 +2926,13 @@ PyCurses_Start_Color(PyObject *self)
     if (code != ERR) {
         initialisedcolors = TRUE;
         c = PyLong_FromLong((long) COLORS);
+        if (c == NULL)
+            return NULL;
         PyDict_SetItemString(ModDict, "COLORS", c);
         Py_DECREF(c);
         cp = PyLong_FromLong((long) COLOR_PAIRS);
+        if (cp == NULL)
+            return NULL;
         PyDict_SetItemString(ModDict, "COLOR_PAIRS", cp);
         Py_DECREF(cp);
         Py_INCREF(Py_None);
