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

(MemoryError).

diff --git a/Modules/zlibmodule.c b/Modules/zlibmodule.c
index f21184a78fd56236f0f273bdfb153c2b3adebef6..da0d3db5c66b7021151b7282adb5537f85685998 100644 (file)
--- a/Modules/zlibmodule.c
+++ b/Modules/zlibmodule.c
@@ -132,6 +132,10 @@ newcompobject(PyTypeObject *type)
     }
 #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;
 }