use std::error::Error;
use std::fmt;
use std::io;
use std::marker;
use std::slice;
use libc::{c_int, c_uint};
use Compression;
use ffi;
#[derive(Debug)]
pub struct Compress {
inner: Stream<DirCompress>,
}
#[derive(Debug)]
pub struct Decompress {
inner: Stream<DirDecompress>,
}
#[derive(Debug)]
struct Stream<D: Direction> {
stream_wrapper: ffi::StreamWrapper,
total_in: u64,
total_out: u64,
_marker: marker::PhantomData<D>,
}
unsafe impl<D: Direction> Send for Stream<D> {}
unsafe impl<D: Direction> Sync for Stream<D> {}
trait Direction {
unsafe fn destroy(stream: *mut ffi::mz_stream) -> c_int;
}
#[derive(Debug)]
enum DirCompress {}
#[derive(Debug)]
enum DirDecompress {}
#[derive(Copy, Clone, PartialEq, Eq, Debug)]
pub enum FlushCompress {
None = ffi::MZ_NO_FLUSH as isize,
Sync = ffi::MZ_SYNC_FLUSH as isize,
Partial = ffi::MZ_PARTIAL_FLUSH as isize,
Full = ffi::MZ_FULL_FLUSH as isize,
Finish = ffi::MZ_FINISH as isize,
#[doc(hidden)]
_Nonexhaustive
}
#[derive(Copy, Clone, PartialEq, Eq, Debug)]
pub enum FlushDecompress {
None = ffi::MZ_NO_FLUSH as isize,
Sync = ffi::MZ_SYNC_FLUSH as isize,
Finish = ffi::MZ_FINISH as isize,
#[doc(hidden)]
_Nonexhaustive
}
#[derive(Debug)]
pub struct DecompressError(());
#[derive(Debug)]
pub struct CompressError(());
#[derive(Copy, Clone, PartialEq, Eq, Debug)]
pub enum Status {
Ok,
BufError,
StreamEnd,
}
impl Compress {
pub fn new(level: Compression, zlib_header: bool) -> Compress {
unsafe {
let mut state = ffi::StreamWrapper::default();
let ret = ffi::mz_deflateInit2(&mut *state,
level.0 as c_int,
ffi::MZ_DEFLATED,
if zlib_header {
ffi::MZ_DEFAULT_WINDOW_BITS
} else {
-ffi::MZ_DEFAULT_WINDOW_BITS
},
9,
ffi::MZ_DEFAULT_STRATEGY);
debug_assert_eq!(ret, 0);
Compress {
inner: Stream {
stream_wrapper: state,
total_in: 0,
total_out: 0,
_marker: marker::PhantomData,
},
}
}
}
pub fn total_in(&self) -> u64 {
self.inner.total_in
}
pub fn total_out(&self) -> u64 {
self.inner.total_out
}
pub fn reset(&mut self) {
let rc = unsafe { ffi::mz_deflateReset(&mut *self.inner.stream_wrapper) };
assert_eq!(rc, ffi::MZ_OK);
self.inner.total_in = 0;
self.inner.total_out = 0;
}
pub fn compress(&mut self,
input: &[u8],
output: &mut [u8],
flush: FlushCompress)
-> Result<Status, CompressError> {
let raw = &mut *self.inner.stream_wrapper;
raw.next_in = input.as_ptr() as *mut _;
raw.avail_in = input.len() as c_uint;
raw.next_out = output.as_mut_ptr();
raw.avail_out = output.len() as c_uint;
let rc = unsafe { ffi::mz_deflate(raw, flush as c_int) };
self.inner.total_in += (raw.next_in as usize -
input.as_ptr() as usize) as u64;
self.inner.total_out += (raw.next_out as usize -
output.as_ptr() as usize) as u64;
match rc {
ffi::MZ_OK => Ok(Status::Ok),
ffi::MZ_BUF_ERROR => Ok(Status::BufError),
ffi::MZ_STREAM_END => Ok(Status::StreamEnd),
ffi::MZ_STREAM_ERROR => Err(CompressError(())),
c => panic!("unknown return code: {}", c),
}
}
pub fn compress_vec(&mut self,
input: &[u8],
output: &mut Vec<u8>,
flush: FlushCompress)
-> Result<Status, CompressError> {
let cap = output.capacity();
let len = output.len();
unsafe {
let before = self.total_out();
let ret = {
let ptr = output.as_mut_ptr().offset(len as isize);
let out = slice::from_raw_parts_mut(ptr, cap - len);
self.compress(input, out, flush)
};
output.set_len((self.total_out() - before) as usize + len);
return ret
}
}
}
impl Decompress {
pub fn new(zlib_header: bool) -> Decompress {
unsafe {
let mut state = ffi::StreamWrapper::default();
let ret = ffi::mz_inflateInit2(&mut *state,
if zlib_header {
ffi::MZ_DEFAULT_WINDOW_BITS
} else {
-ffi::MZ_DEFAULT_WINDOW_BITS
});
debug_assert_eq!(ret, 0);
Decompress {
inner: Stream {
stream_wrapper: state,
total_in: 0,
total_out: 0,
_marker: marker::PhantomData,
},
}
}
}
pub fn total_in(&self) -> u64 {
self.inner.total_in
}
pub fn total_out(&self) -> u64 {
self.inner.total_out
}
pub fn decompress(&mut self,
input: &[u8],
output: &mut [u8],
flush: FlushDecompress)
-> Result<Status, DecompressError> {
let raw = &mut *self.inner.stream_wrapper;
raw.next_in = input.as_ptr() as *mut u8;
raw.avail_in = input.len() as c_uint;
raw.next_out = output.as_mut_ptr();
raw.avail_out = output.len() as c_uint;
let rc = unsafe { ffi::mz_inflate(raw, flush as c_int) };
self.inner.total_in += (raw.next_in as usize -
input.as_ptr() as usize) as u64;
self.inner.total_out += (raw.next_out as usize -
output.as_ptr() as usize) as u64;
match rc {
ffi::MZ_DATA_ERROR |
ffi::MZ_STREAM_ERROR => Err(DecompressError(())),
ffi::MZ_OK => Ok(Status::Ok),
ffi::MZ_BUF_ERROR => Ok(Status::BufError),
ffi::MZ_STREAM_END => Ok(Status::StreamEnd),
c => panic!("unknown return code: {}", c),
}
}
pub fn decompress_vec(&mut self,
input: &[u8],
output: &mut Vec<u8>,
flush: FlushDecompress)
-> Result<Status, DecompressError> {
let cap = output.capacity();
let len = output.len();
unsafe {
let before = self.total_out();
let ret = {
let ptr = output.as_mut_ptr().offset(len as isize);
let out = slice::from_raw_parts_mut(ptr, cap - len);
self.decompress(input, out, flush)
};
output.set_len((self.total_out() - before) as usize + len);
return ret
}
}
pub fn reset(&mut self, zlib_header: bool) {
self._reset(zlib_header);
}
#[cfg(feature = "zlib")]
fn _reset(&mut self, zlib_header: bool) {
let bits = if zlib_header {
ffi::MZ_DEFAULT_WINDOW_BITS
} else {
-ffi::MZ_DEFAULT_WINDOW_BITS
};
unsafe {
ffi::inflateReset2(&mut *self.inner.stream_wrapper, bits);
}
self.inner.total_out = 0;
self.inner.total_in = 0;
}
#[cfg(not(feature = "zlib"))]
fn _reset(&mut self, zlib_header: bool) {
*self = Decompress::new(zlib_header);
}
}
impl Error for DecompressError {
fn description(&self) -> &str { "deflate decompression error" }
}
impl From<DecompressError> for io::Error {
fn from(data: DecompressError) -> io::Error {
io::Error::new(io::ErrorKind::Other, data)
}
}
impl fmt::Display for DecompressError {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
self.description().fmt(f)
}
}
impl Error for CompressError {
fn description(&self) -> &str { "deflate compression error" }
}
impl From<CompressError> for io::Error {
fn from(data: CompressError) -> io::Error {
io::Error::new(io::ErrorKind::Other, data)
}
}
impl fmt::Display for CompressError {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
self.description().fmt(f)
}
}
impl Direction for DirCompress {
unsafe fn destroy(stream: *mut ffi::mz_stream) -> c_int {
ffi::mz_deflateEnd(stream)
}
}
impl Direction for DirDecompress {
unsafe fn destroy(stream: *mut ffi::mz_stream) -> c_int {
ffi::mz_inflateEnd(stream)
}
}
impl<D: Direction> Drop for Stream<D> {
fn drop(&mut self) {
unsafe {
let _ = D::destroy(&mut *self.stream_wrapper);
}
}
}
#[cfg(test)]
mod tests {
use std::io::Write;
use write;
use {Compression, Decompress, FlushDecompress};
#[test]
fn issue51() {
let data = vec![
0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03, 0xb3, 0xc9,
0x28, 0xc9, 0xcd, 0xb1, 0xe3, 0xe5, 0xb2, 0xc9, 0x48, 0x4d, 0x4c, 0xb1,
0xb3, 0x29, 0xc9, 0x2c, 0xc9, 0x49, 0xb5, 0x33, 0x31, 0x30, 0x51, 0xf0,
0xcb, 0x2f, 0x51, 0x70, 0xcb, 0x2f, 0xcd, 0x4b, 0xb1, 0xd1, 0x87, 0x08,
0xda, 0xe8, 0x83, 0x95, 0x00, 0x95, 0x26, 0xe5, 0xa7, 0x54, 0x2a, 0x24,
0xa5, 0x27, 0xe7, 0xe7, 0xe4, 0x17, 0xd9, 0x2a, 0x95, 0x67, 0x64, 0x96,
0xa4, 0x2a, 0x81, 0x8c, 0x48, 0x4e, 0xcd, 0x2b, 0x49, 0x2d, 0xb2, 0xb3,
0xc9, 0x30, 0x44, 0x37, 0x01, 0x28, 0x62, 0xa3, 0x0f, 0x95, 0x06, 0xd9,
0x05, 0x54, 0x04, 0xe5, 0xe5, 0xa5, 0x67, 0xe6, 0x55, 0xe8, 0x1b, 0xea,
0x99, 0xe9, 0x19, 0x21, 0xab, 0xd0, 0x07, 0xd9, 0x01, 0x32, 0x53, 0x1f,
0xea, 0x3e, 0x00, 0x94, 0x85, 0xeb, 0xe4, 0xa8, 0x00, 0x00, 0x00
];
let mut decoded = Vec::with_capacity(data.len()*2);
let mut d = Decompress::new(false);
assert!(d.decompress_vec(&data[10..], &mut decoded, FlushDecompress::Finish).is_ok());
drop(d.decompress_vec(&[0], &mut decoded, FlushDecompress::None));
}
#[test]
fn reset() {
let string = "hello world".as_bytes();
let mut zlib = Vec::new();
let mut deflate = Vec::new();
let comp = Compression::default();
write::ZlibEncoder::new(&mut zlib, comp).write_all(string).unwrap();
write::DeflateEncoder::new(&mut deflate, comp).write_all(string).unwrap();
let mut dst = [0; 1024];
let mut decoder = Decompress::new(true);
decoder.decompress(&zlib, &mut dst, FlushDecompress::Finish).unwrap();
assert_eq!(decoder.total_out(), string.len() as u64);
assert!(dst.starts_with(string));
decoder.reset(false);
decoder.decompress(&deflate, &mut dst, FlushDecompress::Finish).unwrap();
assert_eq!(decoder.total_out(), string.len() as u64);
assert!(dst.starts_with(string));
}
}