Fix Gzip::Compress error for gzip files with extra field

crystal-lang · May 1, 2024 · 85456e0 · 85456e0
1 parent 8b9e299
commit 85456e0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/compress/gzip/header.cr b/src/compress/gzip/header.cr
@@ -41,7 +41,7 @@ class Compress::Gzip::Header
  @os = header[9]
 
  if flg.extra?
- xlen = io.read_byte.not_nil!
+ xlen = io.read_bytes(UInt16, IO::ByteFormat::LittleEndian)
  @extra = Bytes.new(xlen)
  io.read_fully(@extra)
  end
@@ -86,7 +86,7 @@ class Compress::Gzip::Header
  io.write_byte os
 
  unless @extra.empty?
- io.write_byte @extra.size.to_u8
+ io.write_bytes(@extra.size.to_u16, IO::ByteFormat::LittleEndian)
  io.write(@extra)
  end