forked from didirus/AstralRinth
fb30c0ba2b
* fix(labrinth): return version artifact size exceeded error eagerly Now we don't wait until the result memory buffer has grown to a size greater than the maximum allowed, and instead we return such an error before the buffer is grown with the current chunk, which should reduce memory usage. * fix(labrinth): proper supported game versions range for datapacks * feat(labrinth): allow protected resource and data packs to pass validation
45 lines
1.2 KiB
Rust
45 lines
1.2 KiB
Rust
use crate::routes::ApiError;
|
|
use crate::routes::v3::project_creation::CreateError;
|
|
use actix_multipart::Field;
|
|
use actix_web::web::Payload;
|
|
use bytes::BytesMut;
|
|
use futures::StreamExt;
|
|
|
|
pub async fn read_from_payload(
|
|
payload: &mut Payload,
|
|
cap: usize,
|
|
err_msg: &'static str,
|
|
) -> Result<BytesMut, ApiError> {
|
|
let mut bytes = BytesMut::new();
|
|
while let Some(item) = payload.next().await {
|
|
if bytes.len() >= cap {
|
|
return Err(ApiError::InvalidInput(String::from(err_msg)));
|
|
} else {
|
|
bytes.extend_from_slice(&item.map_err(|_| {
|
|
ApiError::InvalidInput(
|
|
"Unable to parse bytes in payload sent!".to_string(),
|
|
)
|
|
})?);
|
|
}
|
|
}
|
|
Ok(bytes)
|
|
}
|
|
|
|
pub async fn read_from_field(
|
|
field: &mut Field,
|
|
cap: usize,
|
|
err_msg: &'static str,
|
|
) -> Result<BytesMut, CreateError> {
|
|
let mut bytes = BytesMut::new();
|
|
while let Some(chunk) = field.next().await {
|
|
let chunk = chunk?;
|
|
|
|
if bytes.len().saturating_add(chunk.len()) > cap {
|
|
return Err(CreateError::InvalidInput(String::from(err_msg)));
|
|
}
|
|
|
|
bytes.extend_from_slice(&chunk);
|
|
}
|
|
Ok(bytes)
|
|
}
|