///|
priv struct ZlibStreamInflater {
decoder : @zlib.Decoder
pending : Array[Byte]
// Reused across messages; grows to the largest message seen.
mut out : FixedArray[Byte]
}
///|
fn ZlibStreamInflater::ZlibStreamInflater() -> ZlibStreamInflater {
{ decoder: @zlib.Decoder(), pending: [], out: FixedArray::make(16384, 0), }
}
///|
fn has_sync_flush_suffix(bytes : Array[Byte]) -> Bool {
let len = bytes.length()
len >= 4 &&
bytes[len - 4] == 0x00 &&
bytes[len - 3] == 0x00 &&
bytes[len - 2] == 0xff &&
bytes[len - 1] == 0xff
}
///|
/// Inflate one sync-flushed message. The decoder emits every byte up to the
/// flush point, so `NeedMoreInput` means the message is complete.
fn ZlibStreamInflater::inflate(
self : ZlibStreamInflater,
compressed : Bytes,
) -> Bytes raise InflateError {
let mut input = compressed[:]
let mut len = 0
while true {
if len == self.out.length() {
let grown = FixedArray::make(len * 2, b'\x00')
self.out.blit_to(grown, len~)
self.out = grown
}
let status = self.decoder.step(input, self.out.mut_view(start=len)) catch {
ZlibError(_, message) => raise Malformed(message~)
}
len += self.decoder.last_produced()
input = input[self.decoder.last_consumed():]
match status {
NeedMoreInput => break
NeedMoreOutput => continue
Done => raise Malformed(message="zlib-stream ended")
}
}
Bytes::from_array(self.out[:len])
}
///|
impl Inflater for ZlibStreamInflater with fn push(self, bytes) {
for byte in bytes {
self.pending.push(byte)
}
if !has_sync_flush_suffix(self.pending) {
return None
}
let compressed = Bytes::from_array(self.pending)
self.pending.clear()
let plain = self.inflate(compressed)
Some(@utf8.decode(plain) catch { _ => raise InflateError::InvalidUtf8 })
}