From: Kostya Shishkov <kostya.shishkov@gmail.com>
Date: Sat, 27 Jun 2020 12:26:31 +0000 (+0200)
Subject: nihav_core/deflate: fix handling large buffers
X-Git-Url: https://git.nihav.org/?p=nihav.git;a=commitdiff_plain;h=c96ad969bab50f17ffe6c421375e68fe0cc9648f

nihav_core/deflate: fix handling large buffers
---

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.