Add rb_str_locktmp to prevent accidental modifications of z->buf.

ruby · Nov 4, 2024 · 668c1db · 668c1db
1 parent 2fe600e
commit 668c1db
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ext/zlib/zlib.c b/ext/zlib/zlib.c
@@ -1053,13 +1053,16 @@ zstream_run_func(struct zstream_run_args *args)
     while (!args->interrupt) {
         n = z->stream.avail_out;
 
+        rb_str_locktmp(z->buf);
+
 #ifndef RB_NOGVL_UBF_ASYNC_SAFE
         err = (int)(VALUE)rb_thread_call_without_gvl(zstream_run_once, (void *)args, zstream_unblock_func, (void *)args);
 #else
         err = (int)(VALUE)rb_nogvl(zstream_run_once, (void *)args, zstream_unblock_func, (void *)args, RB_NOGVL_UBF_ASYNC_SAFE);
 #endif
 
         rb_str_set_len(z->buf, ZSTREAM_BUF_FILLED(z) + (n - z->stream.avail_out));
+        rb_str_unlocktmp(z->buf);
 
         if (err == Z_STREAM_END) {
             z->flags &= ~ZSTREAM_FLAG_IN_STREAM;