Fix rare bug with uncompressed block conversion

NeRdTheNed · Aug 21, 2023 · 80a56a8 · 80a56a8
1 parent 92e9b18
commit 80a56a8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/deft4j-base/src/main/java/com/github/NeRdTheNed/deft4j/deflate/DeflateStream.java b/deft4j-base/src/main/java/com/github/NeRdTheNed/deft4j/deflate/DeflateStream.java
@@ -371,7 +371,11 @@ private static DeflateBlock optimiseBlock(DeflateBlock toOptimise, long position
 
         if (toOptimise.getDeflateBlockType() != DeflateBlockType.STORED) {
             // Uncompressed
-            callback.accept(new Pair<>(toOptimise.asUncompressed(), "uncompressed"));
+            final DeflateBlockUncompressed stored = toOptimise.asUncompressed();
+
+            if (stored.getUncompressedData().length <= 65535) {
+                callback.accept(new Pair<>(stored, "uncompressed"));
+            }
         }
 
         DeflateBlockHuffman toOptimiseHuffman = null;