[ruby/zlib] Resume zstream if available [Bug #10961]

ruby · Mar 9, 2021 · 0c5f8c6 · 0c5f8c6
1 parent e85bffc
commit 0c5f8c6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ext/zlib/zlib.c b/ext/zlib/zlib.c
@@ -1095,6 +1095,12 @@ zstream_run(struct zstream *z, Bytef *src, long len, int flush)
                                RB_NOGVL_UBF_ASYNC_SAFE);
 #endif
 
+    /* retry if no exception is thrown */
+    if (err == Z_OK && args.interrupt) {
+       args.interrupt = 0;
+       goto loop;
+    }
+
     if (flush != Z_FINISH && err == Z_BUF_ERROR
 	    && z->stream.avail_out > 0) {
 	z->flags |= ZSTREAM_FLAG_IN_STREAM;