add check on response length

Signed-off-by: zombee0 <ewang2027@gmail.com>
StarRocks · Aug 14, 2024 · 22d9a34 · 22d9a34
1 parent c25637c
commit 22d9a34
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/be/src/io/s3_input_stream.cpp b/be/src/io/s3_input_stream.cpp
@@ -57,6 +57,9 @@ StatusOr<int64_t> S3InputStream::read(void* out, int64_t count) {
 
     Aws::S3::Model::GetObjectOutcome outcome = _s3client->GetObject(request);
     if (outcome.IsSuccess()) {
+        if (UNLIKELY(outcome.GetResult().GetContentLength() != real_length)) {
+            return Status::InternalError("The response length is different from request length for io stream!");
+        }
         _offset += real_length;
         return real_length;
     } else {