danvics/oxipng
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Cleanup on error (#132)

Browse source 
Closes #126
This commit is contained in:
Kornel 2018-08-05 04:02:56 +01:00 committed by Josh Holmer
parent 947ccaad04
commit 1322416ba4
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/deflate/mod.rs
View file

@ -81,8 +81,14 @@ pub fn cfzlib_deflate(
stream.avail_out = out.capacity() as uInt; stream.avail_out = out.capacity() as uInt;
match deflate(&mut stream, Z_FINISH) { match deflate(&mut stream, Z_FINISH) {
Z_STREAM_END => {} Z_STREAM_END => {}
Z_OK => return Err(PngError::DeflatedDataTooLong(stream.total_out as usize)), Z_OK | Z_BUF_ERROR => {
_ => return Err(PngError::new("deflate")), deflateEnd(&mut stream);
return Err(PngError::DeflatedDataTooLong(stream.total_out as usize));
},
_ => {
deflateEnd(&mut stream);
return Err(PngError::new("deflate"));
},
} }
if Z_OK != deflateEnd(&mut stream) { if Z_OK != deflateEnd(&mut stream) {
return Err(PngError::new("deflateEnd")); return Err(PngError::new("deflateEnd"));