Fix lzma decompression

1 files changed, 33 insertions, 9 deletions

diff --git a/src/zip/file/read.rs b/src/zip/file/read.rs
index 56f42da..c5c7e99 100644
--- a/src/zip/file/read.rs
+++ b/src/zip/file/read.rs
@@ -2,16 +2,17 @@ use crate::driver::FileDriver;
 use crate::zip::{CompressionMethod, ZipError, ZipFileInfo, ZipResult};
 use bzip2::read::BzDecoder;
 use flate2::read::DeflateDecoder;
+use liblzma::read::XzDecoder;
+use liblzma::stream::{Filters, LzmaOptions, Stream};
 use std::io::{
     Error as IoError, ErrorKind as IoErrorKind, Read, Result as IoResult, Seek, SeekFrom,
 };
-use xz2::read::XzDecoder;
 
 enum IoProxy<Io: Read> {
     Store(Io),
     Deflate(DeflateDecoder<Io>),
     BZip2(BzDecoder<Io>),
-    XZ(XzDecoder<Io>),
+    Xz(XzDecoder<Io>),
 }
 
 pub struct ZipFileReader<'d, Io: Read> {
@@ -39,6 +40,7 @@ impl<'d, Io: Read + Seek> ZipFileReader<'d, Io> {
             io.read(&mut buf)?;
             buf
         };
+
         if u32::from_le_bytes(buf[..4].try_into().unwrap()) != 0x04034b50 {
             return Err(ZipError::InvalidFileHeaderSignature.into());
         }
@@ -46,20 +48,42 @@ impl<'d, Io: Read + Seek> ZipFileReader<'d, Io> {
             + 30
             + u16::from_le_bytes(buf[26..28].try_into().unwrap()) as u64
             + u16::from_le_bytes(buf[28..30].try_into().unwrap()) as u64;
-        io.seek(SeekFrom::Start(data_pointer))?;
+        let mut cursor = io.seek(SeekFrom::Start(data_pointer))?;
 
         Ok(Self {
             io: match info.compression_method {
                 CompressionMethod::Store => IoProxy::Store(io),
                 CompressionMethod::Deflate => IoProxy::Deflate(DeflateDecoder::new(io)),
                 CompressionMethod::BZip2 => IoProxy::BZip2(BzDecoder::new(io)),
-                CompressionMethod::LZMA => IoProxy::XZ(XzDecoder::new(io)),
-                CompressionMethod::XZ => IoProxy::XZ(XzDecoder::new(io)),
+                CompressionMethod::Lzma => {
+                    let buf = {
+                        let mut buf = [0; 9];
+                        io.read(&mut buf)?;
+                        cursor += 9;
+                        buf
+                    };
+                    IoProxy::Xz(XzDecoder::new_stream(
+                        io,
+                        Stream::new_raw_decoder(
+                            Filters::new().lzma1(
+                                LzmaOptions::new()
+                                    .literal_context_bits((buf[4] % 9) as u32)
+                                    .literal_position_bits((buf[4] / 9 % 5) as u32)
+                                    .position_bits((buf[4] / 45) as u32)
+                                    .dict_size(
+                                        u32::from_le_bytes(buf[5..9].try_into().unwrap()).min(4096),
+                                    ),
+                            ),
+                        )
+                        .unwrap(),
+                    ))
+                }
+                CompressionMethod::Xz => IoProxy::Xz(XzDecoder::new(io)),
             },
             info,
 
-            bounds: (data_pointer, data_pointer + info.compressed_size),
-            cursor: data_pointer,
+            bounds: (cursor, data_pointer + info.compressed_size),
+            cursor: cursor,
         })
     }
 
@@ -78,7 +102,7 @@ impl<'d, Io: Read> Read for ZipFileReader<'d, Io> {
             IoProxy::Store(io) => io.read(&mut buf[..upper]),
             IoProxy::Deflate(io) => io.read(&mut buf[..upper]),
             IoProxy::BZip2(io) => io.read(&mut buf[..upper]),
-            IoProxy::XZ(io) => io.read(&mut buf[..upper]),
+            IoProxy::Xz(io) => io.read(&mut buf[..upper]),
         }?;
         self.cursor += upper as u64;
         Ok(bytes)
@@ -118,7 +142,7 @@ impl<'d, Io: Read + Seek> Seek for ZipFileReader<'d, Io> {
             }
             _ => Err(IoError::new(
                 IoErrorKind::Unsupported,
-                ZipError::CompressionDataIsUnseekable,
+                ZipError::CompressedDataIsUnseekable,
             )),
         }
     }