fixed zbuff error 775-3 (internal buffer management)

diff --git a/lib/zbuff.c b/lib/zbuff.c
index 1027c6973276454a5dcb5f167677277378f07e4a..06a7ec4e81fc23e8c178bcfb46ba09ccbd77fd19 100644 (file)
--- 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 */