commit a956e645c10051cada87abc90248b112df7e1974
author Christian Heimes <christian@cheimes.de> Fri Jul 26 14:45:37 2013 +0200
committer Christian Heimes <christian@cheimes.de> Fri Jul 26 14:45:37 2013 +0200
tree 6226355902e0cd2411f9560697d3c276a37f0362
parent 845f7845aa6493b59a3b3c427ae85fcd42f58d16

Fix possible NULL pointer dereference in PyCurses_Start_Color()
CID 1058276

Modules/_cursesmodule.c

diff --git a/Modules/_cursesmodule.c b/Modules/_cursesmodule.c
index 132670d..f5df80c 100644
--- a/Modules/_cursesmodule.c
+++ b/Modules/_cursesmodule.c
@@ -2930,9 +2930,13 @@
     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);