fix OOB exception due to assumption about bufferSize

apache · Nov 18, 2024 · 09c3b48 · 09c3b48
1 parent 317db31
commit 09c3b48
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...ject/hadoop-common/src/main/java/org/apache/hadoop/io/compress/BlockCompressorStream.java b/...ject/hadoop-common/src/main/java/org/apache/hadoop/io/compress/BlockCompressorStream.java
@@ -54,7 +54,11 @@ public class BlockCompressorStream extends CompressorStream {
   public BlockCompressorStream(OutputStream out, Compressor compressor, 
                                int bufferSize, int compressionOverhead) {
     super(out, compressor, bufferSize);
-    MAX_INPUT_SIZE = bufferSize - compressionOverhead;
+    if (bufferSize - compressionOverhead >= 0) {
+      MAX_INPUT_SIZE = bufferSize - compressionOverhead;
+    } else {
+      throw new IllegalArgumentException("buffer size is less than compression overhead");
+    }
   }
 
   /**