zstd-ruby 2.0.7 → 2.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 743345c216c9251b2ec2bcec5231362bff5571308405ab237590acc6a8ef28c3
-  data.tar.gz: 83c867433d047e039f8455575a04c44f6c46081b272d1c7aff67fccfcdda900e
+  metadata.gz: 2747a2ccb21d362d5523b5c986fd5f45e7bc2007a40710c303e14e69b66dd626
+  data.tar.gz: 9b396596f0877e8da07fedf4a90e5d54af41414262dd12754faff6b71e56b771
 SHA512:
-  metadata.gz: 514f3b696028151ede41e5d56cd3fc83335e0e3714558b15177891c323162a7a49b77f717dc62adda812966edc83fa71c5a05aa5a668fa45b5a577194d66a937
-  data.tar.gz: f76066d3ba52a3545982c71e42b805798caab46fa93547f96c5321d4d57c225575abc5fe8b89253c7ec826c1fa45222b40c00cf8a3910c64861044d2a469ae6e
+  metadata.gz: 5cdeca9f57890066be8ad60fa7768b0920faac9bae169f6ca5069302a38e48a8c1e22d6b93a13abda415447dc7d750f38aef923527d055d04075d0d81d9a6556
+  data.tar.gz: a625f95323296768a68ccbc98fa4a24175d8a6d2643b3ef7ee4bf3b7cb1ceef2afc4ba4ba263b7bf2b7dbbc279ae89231f92d0d043aa4a1c6d96e8b8e03aa776

data/ext/zstdruby/zstdruby.c

@@ -40,7 +40,7 @@ static VALUE rb_compress(int argc, VALUE *argv, VALUE self)
   return output;
 }
 
-static VALUE decode_one_frame(ZSTD_DCtx* dctx, const unsigned char* src, size_t size, VALUE kwargs) {
+static VALUE decode_one_frame(ZSTD_DCtx* dctx, const unsigned char* src, size_t size, VALUE kwargs, size_t* consumed) {
   VALUE out = rb_str_buf_new(0);
   size_t cap = ZSTD_DStreamOutSize();
   char *buf = ALLOC_N(char, cap);
@@ -64,11 +64,14 @@ static VALUE decode_one_frame(ZSTD_DCtx* dctx, const unsigned char* src, size_t
     }
   }
   xfree(buf);
+  if (consumed) {
+    *consumed = in.pos;
+  }
   return out;
 }
 
 static VALUE decompress_buffered(ZSTD_DCtx* dctx, const char* data, size_t len) {
-  return decode_one_frame(dctx, (const unsigned char*)data, len, Qnil);
+  return decode_one_frame(dctx, (const unsigned char*)data, len, Qnil, NULL);
 }
 
 static VALUE rb_decompress(int argc, VALUE *argv, VALUE self)
@@ -84,6 +87,9 @@ static VALUE rb_decompress(int argc, VALUE *argv, VALUE self)
   const uint32_t ZSTD_MAGIC = 0xFD2FB528U;
   const uint32_t SKIP_LO    = 0x184D2A50U; /* ...5F */
 
+  VALUE result = Qnil;
+  ZSTD_DCtx *dctx = NULL;
+
   while (off + 4 <= in_size) {
     uint32_t magic = (uint32_t)in[off]
                    | ((uint32_t)in[off+1] << 8)
@@ -103,23 +109,43 @@ static VALUE rb_decompress(int argc, VALUE *argv, VALUE self)
     }
 
     if (magic == ZSTD_MAGIC) {
-      ZSTD_DCtx *dctx = ZSTD_createDCtx();
-      if (!dctx) {
-        rb_raise(rb_eRuntimeError, "ZSTD_createDCtx failed");
+      if (dctx == NULL) {
+        dctx = ZSTD_createDCtx();
+        if (!dctx) {
+          rb_raise(rb_eRuntimeError, "ZSTD_createDCtx failed");
+        }
       }
 
-      VALUE out = decode_one_frame(dctx, in + off, in_size - off, kwargs);
+      size_t consumed = 0;
+      VALUE out = decode_one_frame(dctx, in + off, in_size - off, kwargs, &consumed);
+      if (result == Qnil) {
+        /* First frame becomes the accumulator, avoiding a copy of its
+           (potentially large) output in the common single-frame case. */
+        result = out;
+      } else {
+        rb_str_cat(result, RSTRING_PTR(out), RSTRING_LEN(out));
+      }
 
-      ZSTD_freeDCtx(dctx);
-      RB_GC_GUARD(input_value);
-      return out;
+      if (consumed == 0) {
+        /* Guard against a non-advancing frame to avoid an infinite loop. */
+        break;
+      }
+      off += consumed;
+      continue;
     }
 
     off += 1;
   }
 
+  if (dctx != NULL) {
+    ZSTD_freeDCtx(dctx);
+  }
+
   RB_GC_GUARD(input_value);
-  rb_raise(rb_eRuntimeError, "not a zstd frame (magic not found)");
+  if (result == Qnil) {
+    rb_raise(rb_eRuntimeError, "not a zstd frame (magic not found)");
+  }
+  return result;
 }
 
 static void free_cdict(void *dict)

data/lib/zstd-ruby/version.rb

@@ -1,3 +1,3 @@
 module Zstd
-  VERSION = "2.0.7"
+  VERSION = "2.0.8"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: zstd-ruby
 version: !ruby/object:Gem::Version
-  version: 2.0.7
+  version: 2.0.8
 platform: ruby
 authors:
 - SpringMT
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2026-08-06 00:00:00.000000000 Z
+date: 2026-08-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler