traverseda
/
cpython
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix thread locks in zlib module may go wrong in rare case 

Setting `next_in` before acquiring the thread lock may mix up compress/decompress state in other threads.
This commit is contained in:
animalize 2020-09-07 21:52:02 +08:00
parent 4e581d64b8
commit 378e9ad737
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Misc/NEWS.d/next/Library/2020-09-07-21-51-17.bpo-41735.NKqGKy.rst Normal file
View File

@ -0,0 +1 @@
Fix thread locks in zlib module may go wrong in rare case. Patch by Ma Lin.

8
Modules/zlibmodule.c
View File

@ -653,11 +653,11 @@ zlib_Compress_compress_impl(compobject *self, Py_buffer *data)
Py_ssize_t ibuflen, obuflen = DEF_BUF_SIZE; Py_ssize_t ibuflen, obuflen = DEF_BUF_SIZE;
int err; int err;
ENTER_ZLIB(self);
self->zst.next_in = data->buf; self->zst.next_in = data->buf;
ibuflen = data->len; ibuflen = data->len;
ENTER_ZLIB(self);
do { do {
arrange_input_buffer(&self->zst, &ibuflen); arrange_input_buffer(&self->zst, &ibuflen);
@ -771,6 +771,8 @@ zlib_Decompress_decompress_impl(compobject *self, Py_buffer *data,
else else
hard_limit = max_length; hard_limit = max_length;
ENTER_ZLIB(self);
self->zst.next_in = data->buf; self->zst.next_in = data->buf;
ibuflen = data->len; ibuflen = data->len;
@ -778,8 +780,6 @@ zlib_Decompress_decompress_impl(compobject *self, Py_buffer *data,
if (max_length && obuflen > max_length) if (max_length && obuflen > max_length)
obuflen = max_length; obuflen = max_length;
ENTER_ZLIB(self);
do { do {
arrange_input_buffer(&self->zst, &ibuflen); arrange_input_buffer(&self->zst, &ibuflen);