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

vstinner · vstinner · commit bf2e2f9bdf33 · 2013-07-09T00:29:03.000+02:00
(MemoryError).
diff --git 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;
 }

Original file line number	Diff line number	Diff line change
`@@ -132,6 +132,10 @@ newcompobject(PyTypeObject *type)`
`132`	`132`	`}`
`133`	`133`	`#ifdef WITH_THREAD`
`134`	`134`	`self->lock = PyThread_allocate_lock();`
	`135`	`+ if (self->lock == NULL) {`
	`136`	`+ PyErr_SetString(PyExc_MemoryError, "Unable to allocate lock");`
	`137`	`+ return NULL;`
	`138`	`+ }`
`135`	`139`	`#endif`
`136`	`140`	`return self;`
`137`	`141`	`}`