From 17dc610f8a2e8e8d15b427d0c4f38dca498f798f Mon Sep 17 00:00:00 2001
From: Quentin <quentin@deuxfleurs.fr>
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,