ddidderr/zstd-rs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Passing in dummy dst buffer of compressbound(srcSize)

Browse Source
This commit is contained in:
Bimba Shrestha
2019-09-20 15:50:58 -07:00
parent ae6d0e64ae
commit f3c4fd17e3
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/compress/zstd_compress.c
+5 -2
View File
@@ -2319,6 +2319,9 @@ static void ZSTD_copyBlockSequences(ZSTD_CCtx* zc)
size_t ZSTD_getSequences(ZSTD_CCtx* zc, ZSTD_Sequence* outSeqs, size_t ZSTD_getSequences(ZSTD_CCtx* zc, ZSTD_Sequence* outSeqs,
size_t outSeqsSize, const void* src, size_t srcSize) size_t outSeqsSize, const void* src, size_t srcSize)
{ {
const size_t dstCapacity = ZSTD_compressBound(srcSize);
void* dst = ZSTD_malloc(dstCapacity, ZSTD_defaultCMem);
SeqCollector seqCollector; SeqCollector seqCollector;
seqCollector.collectSequences = 1; seqCollector.collectSequences = 1;
seqCollector.seqStart = outSeqs; seqCollector.seqStart = outSeqs;
@@ -2326,8 +2329,8 @@ size_t ZSTD_getSequences(ZSTD_CCtx* zc, ZSTD_Sequence* outSeqs,
seqCollector.maxSequences = outSeqsSize; seqCollector.maxSequences = outSeqsSize;
zc->seqCollector = seqCollector; zc->seqCollector = seqCollector;
/* We never write to dst when collecing sequences so setting dst = src is harmless */ ZSTD_compress2(zc, dst, dstCapacity, src, srcSize);
ZSTD_compress2(zc, (void*)src, srcSize, src, srcSize); ZSTD_free(dst, ZSTD_defaultCMem);
return zc->seqCollector.seqIndex; return zc->seqCollector.seqIndex;
} }