Skip to content

Commit

Permalink
Add tests for reading with an empty buffer, and fix bug in GzDecoder
Browse files Browse the repository at this point in the history
  • Loading branch information
mdsteele committed Mar 16, 2018
1 parent b769bb7 commit 31cbd0b
Show file tree
Hide file tree
Showing 2 changed files with 84 additions and 0 deletions.
3 changes: 3 additions & 0 deletions src/gz/bufread.rs
Original file line number Diff line number Diff line change
Expand Up @@ -358,6 +358,9 @@ impl<R: BufRead> Read for GzDecoder<R> {
let another_error = io::ErrorKind::Other.into();
return Err(mem::replace(e, another_error));
}
if into.is_empty() {
return Ok(0);
}
match try!(self.inner.read(into)) {
0 => {
try!(self.finish());
Expand Down
81 changes: 81 additions & 0 deletions tests/empty-read.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,81 @@
extern crate flate2;

use std::io::{Read, Write};

#[test]
fn deflate_decoder_empty_read() {
let original: &[u8] = b"Lorem ipsum dolor sit amet.";
let mut encoder = flate2::write::DeflateEncoder::new(Vec::new(), flate2::Compression::default());
encoder.write_all(original).unwrap();
let encoded: Vec<u8> = encoder.finish().unwrap();
let mut decoder = flate2::read::DeflateDecoder::new(encoded.as_slice());
assert_eq!(decoder.read(&mut []).unwrap(), 0);
let mut decoded = Vec::new();
decoder.read_to_end(&mut decoded).unwrap();
assert_eq!(decoded.as_slice(), original);
}

#[test]
fn deflate_encoder_empty_read() {
let original: &[u8] = b"Lorem ipsum dolor sit amet.";
let mut encoder = flate2::read::DeflateEncoder::new(original, flate2::Compression::default());
assert_eq!(encoder.read(&mut []).unwrap(), 0);
let mut encoded = Vec::new();
encoder.read_to_end(&mut encoded).unwrap();
let mut decoder = flate2::read::DeflateDecoder::new(encoded.as_slice());
let mut decoded = Vec::new();
decoder.read_to_end(&mut decoded).unwrap();
assert_eq!(decoded.as_slice(), original);
}

#[test]
fn gzip_decoder_empty_read() {
let original: &[u8] = b"Lorem ipsum dolor sit amet.";
let mut encoder = flate2::write::GzEncoder::new(Vec::new(), flate2::Compression::default());
encoder.write_all(original).unwrap();
let encoded: Vec<u8> = encoder.finish().unwrap();
let mut decoder = flate2::read::GzDecoder::new(encoded.as_slice());
assert_eq!(decoder.read(&mut []).unwrap(), 0);
let mut decoded = Vec::new();
decoder.read_to_end(&mut decoded).unwrap();
assert_eq!(decoded.as_slice(), original);
}

#[test]
fn gzip_encoder_empty_read() {
let original: &[u8] = b"Lorem ipsum dolor sit amet.";
let mut encoder = flate2::read::GzEncoder::new(original, flate2::Compression::default());
assert_eq!(encoder.read(&mut []).unwrap(), 0);
let mut encoded = Vec::new();
encoder.read_to_end(&mut encoded).unwrap();
let mut decoder = flate2::read::GzDecoder::new(encoded.as_slice());
let mut decoded = Vec::new();
decoder.read_to_end(&mut decoded).unwrap();
assert_eq!(decoded.as_slice(), original);
}

#[test]
fn zlib_decoder_empty_read() {
let original: &[u8] = b"Lorem ipsum dolor sit amet.";
let mut encoder = flate2::write::ZlibEncoder::new(Vec::new(), flate2::Compression::default());
encoder.write_all(original).unwrap();
let encoded: Vec<u8> = encoder.finish().unwrap();
let mut decoder = flate2::read::ZlibDecoder::new(encoded.as_slice());
assert_eq!(decoder.read(&mut []).unwrap(), 0);
let mut decoded = Vec::new();
decoder.read_to_end(&mut decoded).unwrap();
assert_eq!(decoded.as_slice(), original);
}

#[test]
fn zlib_encoder_empty_read() {
let original: &[u8] = b"Lorem ipsum dolor sit amet.";
let mut encoder = flate2::read::ZlibEncoder::new(original, flate2::Compression::default());
assert_eq!(encoder.read(&mut []).unwrap(), 0);
let mut encoded = Vec::new();
encoder.read_to_end(&mut encoded).unwrap();
let mut decoder = flate2::read::ZlibDecoder::new(encoded.as_slice());
let mut decoded = Vec::new();
decoder.read_to_end(&mut decoded).unwrap();
assert_eq!(decoded.as_slice(), original);
}

0 comments on commit 31cbd0b

Please sign in to comment.