commit bf2e2f9bdf3366abf6004cad2cc46782d8b31367
author Victor Stinner <victor.stinner@gmail.com> Tue Jul 09 00:29:03 2013 +0200
committer Victor Stinner <victor.stinner@gmail.com> Tue Jul 09 00:29:03 2013 +0200
tree 6f2f09289e997519178d630d1ab6db8198f97933
parent 3f15cf0961946702c87e09bc312b2bfec39654b6

Issue #18408: Fix zlib.compressobj() to handle PyThread_allocate_lock() failure
(MemoryError).

Modules/zlibmodule.c

diff --git a/Modules/zlibmodule.c b/Modules/zlibmodule.c
index f21184a..da0d3db 100644
--- a/Modules/zlibmodule.c
+++ b/Modules/zlibmodule.c
@@ -132,6 +132,10 @@
     }
 #ifdef WITH_THREAD
     self->lock = PyThread_allocate_lock();
+    if (self->lock == NULL) {
+        PyErr_SetString(PyExc_MemoryError, "Unable to allocate lock");
+        return NULL;
+    }
 #endif
     return self;
 }