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); destination.writerIndex(destination.writerIndex() + produced);
} }
if (inflater.getBytesWritten() != uncompressedSize) { if (!inflater.finished()) {
throw new DataFormatException("Did not write the exact expected number of" throw new DataFormatException("Received a deflate stream that was too large, wanted "
+ " uncompressed bytes, expected " + uncompressedSize); + uncompressedSize);
} }
source.readerIndex(origIdx + inflater.getTotalIn()); source.readerIndex(origIdx + inflater.getTotalIn());
} catch (Throwable e) { } catch (Throwable e) {