Add tests and fix bugs

3 files changed, 14 insertions, 14 deletions

diff --git a/src/zip/driver.rs b/src/zip/driver.rs
index 62da39f..631c4ed 100644
--- a/src/zip/driver.rs
+++ b/src/zip/driver.rs
@@ -68,23 +68,23 @@ impl<Io: Read + Seek> ArchiveRead for Zip<Io> {
         let limit = 65557.min(io.seek(SeekFrom::End(0))?) as i64;
         let start = io.seek(SeekFrom::End(-limit))?;
         let pos = start
-            + io.read_vec(limit as usize - 18)?
-                .windows(4)
-                .rposition(|v| u32::from_le_bytes(v.try_into().unwrap()) == 0x06054b50)
-                .ok_or(ZipError::EOCDRNotFound)? as u64;
+            + io.read_vec(
+                (limit as usize)
+                    .checked_sub(18)
+                    .ok_or(ZipError::EocdrNotFound)?,
+            )?
+            .windows(4)
+            .rposition(|v| u32::from_le_bytes(v.try_into().unwrap()) == 0x06054b50)
+            .ok_or(ZipError::EocdrNotFound)? as u64;
 
         // Read eocdr
         io.seek(SeekFrom::Start(pos + 4))?;
         let buf = io.read_arr::<18>()?;
         let eocdr: Eocdr = deserialize(&buf).unwrap();
-        let comment = {
-            let mut buf: Vec<u8> = vec![0; eocdr.comment_len as usize];
-            io.read(&mut buf)?;
-            String::from_cp437(buf)
-        };
+        let comment = String::from_cp437(io.read_vec(eocdr.comment_len as usize)?);
 
         // Try to find eocdr64locator
-        io.seek(SeekFrom::Start(pos - 20))?;
+        io.seek(SeekFrom::Start(pos.saturating_sub(20)))?;
         let buf = io.read_arr::<20>()?;
         let (cd_pointer, cd_size, cd_records) =
             // If locator found then read eocdr64
diff --git a/src/zip/encryption.rs b/src/zip/encryption.rs
index 28a6bdb..76824a1 100644
--- a/src/zip/encryption.rs
+++ b/src/zip/encryption.rs
@@ -72,7 +72,7 @@ impl<Io: Read> WeakDecoder<Io> {
 
     fn decode_byte(&mut self, byte: u8) -> u8 {
         let key = self.key2 | 2;
-        let byte = byte ^ ((key * (key ^ 1)) >> 8) as u8;
+        let byte = byte ^ ((key.wrapping_mul(key ^ 1)) >> 8) as u8;
         self.update_keys(byte);
         byte
     }
diff --git a/src/zip/error.rs b/src/zip/error.rs
index a4b8c2b..525a67b 100644
--- a/src/zip/error.rs
+++ b/src/zip/error.rs
@@ -4,9 +4,9 @@ use std::fmt::Display;
 
 pub type ZipResult<T> = ArchiveResult<T, ZipError>;
 
-#[derive(Debug)]
+#[derive(Debug, PartialEq, Eq)]
 pub enum ZipError {
-    EOCDRNotFound,
+    EocdrNotFound,
     InvalidEOCDR64Signature,
     InvalidFileHeaderSignature,
     InvalidCDRSignature,
@@ -38,7 +38,7 @@ impl From<ZipError> for ArchiveError<ZipError> {
 impl Display for ZipError {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         match self {
-            Self::EOCDRNotFound => write!(f, "End of central directory record not found"),
+            Self::EocdrNotFound => write!(f, "End of central directory record not found"),
             Self::InvalidEOCDR64Signature => {
                 write!(
                     f,