amigan/rustypaste-pretty
1
0
Fork 0
mirror of https://github.com/amigan/rustypaste-pretty.git synced 2024-11-21 20:09:48 -05:00
Code Issues Projects Releases Packages Wiki Activity

fix(upload): prevent sending zero bytes

Browse source
This commit is contained in:
orhun 2021-07-28 00:25:18 +03:00
parent f4a3c07def
commit 315a649777
No known key found for this signature in database
GPG key ID: F83424824B3E4B90
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/server.rs
View file

@ -56,6 +56,10 @@ async fn upload(
return Err(error::ErrorPayloadTooLarge("upload limit exceeded")); return Err(error::ErrorPayloadTooLarge("upload limit exceeded"));
} }
} }
if bytes.is_empty() {
log::warn!("{} sent zero bytes", host);
return Err(error::ErrorBadRequest("invalid file size"));
}
let bytes_unit = Byte::from_bytes(bytes.len() as u128).get_appropriate_unit(false); 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);