X-Git-Url: https://git.nihav.org/?p=nihav.git;a=blobdiff_plain;f=nihav-core%2Fsrc%2Fcompr%2Fdeflate.rs;h=da37fde37df255faa698b82bd66597a8347f8d05;hp=fa318a473e84647e9ca281cfa6fc868c243c812a;hb=c96ad969bab50f17ffe6c421375e68fe0cc9648f;hpb=b0e12d42963a2476e54b8f9dce15b339ff7abba5 diff --git a/nihav-core/src/compr/deflate.rs b/nihav-core/src/compr/deflate.rs index fa318a4..da37fde 100644 --- a/nihav-core/src/compr/deflate.rs +++ b/nihav-core/src/compr/deflate.rs @@ -320,19 +320,19 @@ impl Inflate { } fn put_literal(&mut self, val: u8) { self.buf[self.bpos] = val; - self.bpos += 1; + self.bpos = (self.bpos + 1) & (self.buf.len() - 1); } fn lz_copy(&mut self, offset: usize, len: usize, dst: &mut [u8]) -> DecompressResult<()> { let mask = self.buf.len() - 1; - if self.bpos < offset { + if offset > self.output_idx { return Err(DecompressError::InvalidData); } - let cstart = (self.bpos - offset) & mask; + let cstart = (self.bpos.wrapping_sub(offset)) & mask; for i in 0..len { self.buf[(self.bpos + i) & mask] = self.buf[(cstart + i) & mask]; dst[i] = self.buf[(cstart + i) & mask]; } - self.bpos += len; + self.bpos = (self.bpos + len) & mask; Ok(()) } ///! Reports whether decoder has finished decoding the input.