v0.2.4

Cargo.lock

@@ -246,7 +246,7 @@ checksum = "cf4b9d6a944f767f8e5e0db018570623c85f3d925ac718db4e06d0187adb21c1"
 
 [[package]]
 name = "breeze"
-version = "0.2.3"
+version = "0.2.4"
 dependencies = [
  "anyhow",
  "async-recursion",

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "breeze"
-version = "0.2.3"
+version = "0.2.4"
 edition = "2021"
 
 [dependencies]

src/config.rs

@@ -33,6 +33,10 @@ pub struct EngineConfig {
     /// Configuration for cache system
     pub cache: CacheConfig,
 
+    /// Maximum size of an upload that will be accepted.
+    /// Files above this size can not be uploaded.
+    pub max_upload_len: Option<usize>,
+
     /// Maximum lifetime of a temporary upload
     #[serde_as(as = "DurationSeconds")]
     pub max_temp_lifetime: Duration,

src/disk.rs

@@ -60,7 +60,7 @@ impl Disk {
     /// Create a background I/O task
     pub async fn start_save(&self, saved_name: &str) -> Sender<Bytes> {
         // start a task that handles saving files to disk (we can save to cache/disk in parallel that way)
-        let (tx, mut rx): (Sender<Bytes>, Receiver<Bytes>) = mpsc::channel(1);
+        let (tx, mut rx): (Sender<Bytes>, Receiver<Bytes>) = mpsc::channel(256);
 
         let p = self.path_for(saved_name);
 

src/engine.rs

@@ -31,6 +31,9 @@ pub enum ProcessOutcome {
     /// We give the user their file's URL
     Success(String),
 
+    /// Occurs when an upload exceeds the chosen maximum file size.
+    UploadTooLarge,
+
     /// Occurs when a temporary upload is too big to fit in the cache.
     TemporaryUploadTooLarge,
 
@@ -285,6 +288,11 @@ impl Engine {
         lifetime: Option<Duration>,
         keep_exif: bool,
     ) -> Result<ProcessOutcome, anyhow::Error> {
+        // if the upload size is greater than our max file size, deny it now
+        if self.cfg.max_upload_len.is_some_and(|l| provided_len > l) {
+            return Ok(ProcessOutcome::UploadTooLarge);
+        }
+
         // if the upload size is smaller than the specified maximum, we use the cache!
         let use_cache: bool = self.cache.will_use(provided_len);
 

src/main.rs

@@ -68,6 +68,7 @@ async fn main() {
         .await
         .expect("failed to bind to given `http.listen_on` address! make sure it's valid, and the port isn't already bound");
 
+    info!("starting server.");
     axum::serve(listener, app)
         .with_graceful_shutdown(shutdown_signal())
         .await

src/new.rs

@@ -81,7 +81,9 @@ pub async fn new(
             ProcessOutcome::Success(url) => Ok(url),
 
             // 413 Payload Too Large
-            ProcessOutcome::TemporaryUploadTooLarge => Err(StatusCode::PAYLOAD_TOO_LARGE),
+            ProcessOutcome::UploadTooLarge | ProcessOutcome::TemporaryUploadTooLarge => {
+                Err(StatusCode::PAYLOAD_TOO_LARGE)
+            }
 
             // 400 Bad Request
             ProcessOutcome::TemporaryUploadLifetimeTooLong => Err(StatusCode::BAD_REQUEST),