Been struggling with this for several days. I'm wanting to stream at most n bytes to a file from the HTTP body and return an error if the user sends more than the negotiated size. A stub of what I have so far assuming a 10MB limit:

pub async fn handle_chunk_upload(
    ...
    body: Body,
) -> Response {
    ...
    let body_with_io_error = body
        .into_data_stream()
        .take(10_000_001) // <- was expecting this to limit the size, but it limits # of items.
        .map_err(io::Error::other);
    let mut body_reader = pin!(StreamReader::new(body_with_io_error));

    let bytes_written = tokio::io::copy(&mut body_reader, &mut file)
        .await
        .expect("couldn't copy bytes");

    if bytes_written > 10_000_000 {
        remove_file(full_path)
             .await
             .expect("could not delete oversized file");

        return resp
            .fail("bytes received exceeded negotiated size")
            .status_code(StatusCode::BAD_REQUEST)
            .into_response();   
    }
    ...
...

I'd like to be able to do this from within the request scope so I can programmatically set this limit, do validation, and send back the appropriate error message.

Thanks in advance!

3 posts - 2 participants

Read full topic