Better check for unfinished decompression

This commit is contained in:
Andrew Steinborn 2020-11-17 04:48:20 -05:00
parent 563a96e624
commit 3d0cb50569

View File

@ -78,9 +78,9 @@ public class Java11VelocityCompressor implements VelocityCompressor {
destination.writerIndex(destination.writerIndex() + produced);
}
if (inflater.getBytesWritten() != uncompressedSize) {
throw new DataFormatException("Did not write the exact expected number of"
+ " uncompressed bytes, expected " + uncompressedSize);
if (!inflater.finished()) {
throw new DataFormatException("Received a deflate stream that was too large, wanted "
+ uncompressedSize);
}
source.readerIndex(origIdx + inflater.getTotalIn());
} catch (Throwable e) {