From 3d0cb505699369eccdbdfdfac727f9b9ae53bade Mon Sep 17 00:00:00 2001
From: Andrew Steinborn <git@steinborn.me>
Date: Tue, 17 Nov 2020 04:48:20 -0500
Subject: [PATCH] Better check for unfinished decompression

---
 .../natives/compression/Java11VelocityCompressor.java       | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/native/src/main/java/com/velocitypowered/natives/compression/Java11VelocityCompressor.java b/native/src/main/java/com/velocitypowered/natives/compression/Java11VelocityCompressor.java
index 845c3adcd..49ac1da26 100644
--- a/native/src/main/java/com/velocitypowered/natives/compression/Java11VelocityCompressor.java
+++ b/native/src/main/java/com/velocitypowered/natives/compression/Java11VelocityCompressor.java
@@ -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) {