From 17dc610f8a2e8e8d15b427d0c4f38dca498f798f Mon Sep 17 00:00:00 2001 From: Quentin Date: Sun, 22 Nov 2020 11:14:46 +0100 Subject: [PATCH] Also check hash for < 3KB files --- src/api/s3_put.rs | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/src/api/s3_put.rs b/src/api/s3_put.rs index 54fa320..a528720 100644 --- a/src/api/s3_put.rs +++ b/src/api/s3_put.rs @@ -51,6 +51,20 @@ pub async fn handle_put( let md5sum_arr = md5sum.finalize(); let md5sum_hex = hex::encode(md5sum_arr); + let mut sha256sum = Sha256::new(); + sha256sum.input(&first_block[..]); + let sha256sum_arr = sha256sum.result(); + let mut hash = [0u8; 32]; + hash.copy_from_slice(&sha256sum_arr[..]); + let sha256sum_hash = Hash::from(hash); + + ensure_checksum_matches( + md5sum_arr.as_slice(), + sha256sum_hash, + content_md5.as_deref(), + content_sha256, + )?; + object_version.state = ObjectVersionState::Complete(ObjectVersionData::Inline( ObjectVersionMeta { headers,