inflating can also take streams larger than typemax(UInt32)

JuliaIO · Jun 26, 2023 · 27782ca · 27782ca
1 parent f9fddaa
commit 27782ca
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/decompression.jl b/src/decompression.jl
@@ -157,12 +157,15 @@ end
 function TranscodingStreams.process(codec::DecompressorCodec, input::Memory, output::Memory, error::Error)
     zstream = codec.zstream
     zstream.next_in = input.ptr
-    zstream.avail_in = input.size
+
+    avail_in = min(input.size, typemax(UInt32))
+    zstream.avail_in = avail_in
     zstream.next_out = output.ptr
-    zstream.avail_out = output.size
+    avail_out = min(output.size, typemax(UInt32))
+    zstream.avail_out = avail_out
     code = inflate!(zstream, Z_NO_FLUSH)
-    Δin = Int(input.size - zstream.avail_in)
-    Δout = Int(output.size - zstream.avail_out)
+    Δin = Int(avail_in - zstream.avail_in)
+    Δout = Int(avail_out - zstream.avail_out)
     if code == Z_OK
         return Δin, Δout, :ok
     elseif code == Z_STREAM_END