From e5a9abefed0dfcbeed464de56339aeaed7b966b1 Mon Sep 17 00:00:00 2001
From: Davies Liu <davies@databricks.com>
Date: Tue, 25 Oct 2016 16:53:49 -0700
Subject: [PATCH] fix buffer overflow in lz4

---
 core/src/main/java/org/apache/spark/io/LZ4BlockInputStream.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/core/src/main/java/org/apache/spark/io/LZ4BlockInputStream.java b/core/src/main/java/org/apache/spark/io/LZ4BlockInputStream.java
index 9d6f06ed2888f..e108a19510efe 100644
--- a/core/src/main/java/org/apache/spark/io/LZ4BlockInputStream.java
+++ b/core/src/main/java/org/apache/spark/io/LZ4BlockInputStream.java
@@ -197,7 +197,7 @@ private void refill() throws IOException {
         readFully(buffer, originalLen);
         break;
       case COMPRESSION_METHOD_LZ4:
-        if (compressedBuffer.length < originalLen) {
+        if (compressedBuffer.length < compressedLen) {
           compressedBuffer = new byte[Math.max(compressedLen, compressedBuffer.length * 3 / 2)];
         }
         readFully(compressedBuffer, compressedLen);