From: Yann Collet <yann.collet.73@gmail.com>
Date: Fri, 1 Apr 2016 16:57:41 +0000 (+0200)
Subject: fixed zbuff error 775-3 (internal buffer management)
X-Git-Tag: v0.6.0^2~17^2~28^2
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=refs%2Fpull%2F158%2Fhead;p=thirdparty%2Fzstd.git

fixed zbuff error 775-3 (internal buffer management)
---

diff --git a/lib/zbuff.c b/lib/zbuff.c
index 1027c6973..06a7ec4e8 100644
--- a/lib/zbuff.c
+++ b/lib/zbuff.c
@@ -135,8 +135,9 @@ size_t ZBUFF_compressInit_advanced(ZBUFF_CCtx* zbc,
             free(zbc->inBuff);   /* should not be necessary */
             zbc->inBuff = (char*)malloc(neededInBuffSize);
             if (zbc->inBuff == NULL) return ERROR(memory_allocation);
-    }   }
-    zbc->blockSize = MIN(ZSTD_BLOCKSIZE_MAX, zbc->inBuffSize);
+        }
+        zbc->blockSize = MIN(ZSTD_BLOCKSIZE_MAX, neededInBuffSize);
+    }
     if (zbc->outBuffSize < ZSTD_compressBound(zbc->blockSize)+1) {
         zbc->outBuffSize = ZSTD_compressBound(zbc->blockSize)+1;
         free(zbc->outBuff);   /* should not be necessary */