]> granicus.if.org Git - python/commitdiff
For ZlibError and ZLIB_VERSION, only attempt to add entry to the
authorFred Drake <fdrake@acm.org>
Wed, 22 Dec 1999 16:13:54 +0000 (16:13 +0000)
committerFred Drake <fdrake@acm.org>
Wed, 22 Dec 1999 16:13:54 +0000 (16:13 +0000)
module dict if the inserted object isn't NULL (basic defensive
programming!).

Modules/zlibmodule.c

index 7dda00f15202865349635595ec4f40c61a2ef377..82863399e5c95987f11fce842382faf33ae00a62 100644 (file)
@@ -870,7 +870,8 @@ PyInit_zlib()
                           (PyObject*)NULL,PYTHON_API_VERSION);
         d = PyModule_GetDict(m);
         ZlibError = PyErr_NewException("zlib.error", NULL, NULL);
-        PyDict_SetItemString(d, "error", ZlibError);
+        if (ZlibError != NULL)
+                PyDict_SetItemString(d, "error", ZlibError);
 
        insint(d, "MAX_WBITS", MAX_WBITS);
        insint(d, "DEFLATED", DEFLATED);
@@ -888,6 +889,8 @@ PyInit_zlib()
        insint(d, "Z_FULL_FLUSH", Z_FULL_FLUSH);
 
        ver = PyString_FromString(ZLIB_VERSION);
-       PyDict_SetItemString(d, "ZLIB_VERSION", ver);
-       Py_DECREF(ver);
+        if (ver != NULL) {
+                PyDict_SetItemString(d, "ZLIB_VERSION", ver);
+                Py_DECREF(ver);
+        }
 }