Fix possible NULL pointer dereference in PyCurses_Start_Color()

CID 1058276

Fix possible NULL pointer dereference in PyCurses_Start_Color()
CID 1058276
ef86368e · Christian Heimes · 5d7c1b1a · a956e645 · ef86368e
Commit ef86368e authored Jul 26, 2013 by Christian Heimes
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

Modules/_cursesmodule.c Modules/_cursesmodule.c +4 -0

No files found.
--- 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);