amigan/rustypaste-pretty
1
0
Fork 0
mirror of https://github.com/amigan/rustypaste-pretty.git synced 2024-11-22 04:19:47 -05:00
Code Issues Projects Releases Packages Wiki Activity

fix(upload): check the content length while reading bytes

Browse source 
closes #1
This commit is contained in:
orhun 2021-07-27 23:49:35 +03:00
parent d02865b956
commit 60c25e2fbc
No known key found for this signature in database
GPG key ID: F83424824B3E4B90
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/server.rs
View file

@ -51,12 +51,12 @@ async fn upload(
let mut bytes = Vec::<u8>::new(); let mut bytes = Vec::<u8>::new();
while let Some(chunk) = field.next().await { while let Some(chunk) = field.next().await {
bytes.append(&mut chunk?.to_vec()); bytes.append(&mut chunk?.to_vec());
}
let bytes_unit = Byte::from_bytes(bytes.len() as u128).get_appropriate_unit(false);
if bytes.len() as u128 > config.server.max_content_length.get_bytes() { if bytes.len() as u128 > config.server.max_content_length.get_bytes() {
log::warn!("upload rejected for {} ({})", host, bytes_unit); log::warn!("upload rejected for {}", host);
return Err(error::ErrorPayloadTooLarge("upload limit exceeded")); return Err(error::ErrorPayloadTooLarge("upload limit exceeded"));
} }
}
let bytes_unit = Byte::from_bytes(bytes.len() as u128).get_appropriate_unit(false);
let file_name = &file::save(content.get_file_name()?, &bytes, &config)?; let file_name = &file::save(content.get_file_name()?, &bytes, &config)?;
log::info!("{} ({}) is uploaded from {}", file_name, bytes_unit, host); log::info!("{} ({}) is uploaded from {}", file_name, bytes_unit, host);
urls.push(format!( urls.push(format!(