From e3812907a8f17fb06fbc5338d8dca404c8b681ac Mon Sep 17 00:00:00 2001 From: wzl12356 Date: Fri, 25 Mar 2022 06:00:59 +0800 Subject: [PATCH] kafka: fix zstd decompress bug --- src/protocol/KafkaMessage.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/protocol/KafkaMessage.cc b/src/protocol/KafkaMessage.cc index 27113230..e4de3cfe 100644 --- a/src/protocol/KafkaMessage.cc +++ b/src/protocol/KafkaMessage.cc @@ -774,10 +774,14 @@ static int zstd_decompress(void *buf, size_t n, KafkaBlock *block) return 0; if (ZSTD_getErrorCode(ret) == ZSTD_error_dstSize_tooSmall) + { out_bufsize += out_bufsize * 2; + } else + { errno = EBADMSG; return -1; + } } }