Add check for blobs that are too large.

scpwiki · Oct 3, 2024 · a3b3be7 · a3b3be7
1 parent e79f766
commit a3b3be7
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/deepwell/src/services/blob/service.rs b/deepwell/src/services/blob/service.rs
@@ -72,8 +72,16 @@ impl BlobService {
         let config = ctx.config();
         let txn = ctx.transaction();
 
-        // Convert expected length integer type
+        // Convert expected length integer type, then check it
         let blob_size = i64::try_from(blob_size).map_err(|_| Error::BlobTooBig)?;
+        if blob_size > config.maximum_blob_size {
+            error!(
+                "Blob proposed to upload is too big ({} > {})",
+                blob_size, config.maximum_blob_size,
+            );
+
+            return Err(Error::BlobTooBig);
+        }
 
         // Generate primary key and random S3 path
         let pending_blob_id = cuid();