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

feat(server): use logging for erroneous situations

Browse source
This commit is contained in:
orhun 2021-07-23 23:07:01 +03:00
parent 1413335519
commit 53ba3dcac9
No known key found for this signature in database
GPG key ID: F83424824B3E4B90
1 changed files with 15 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
src/server.rs
View file

@ -37,14 +37,20 @@ async fn upload(
mut payload: Multipart, mut payload: Multipart,
config: web::Data<Config>, config: web::Data<Config>,
) -> Result<HttpResponse, Error> { ) -> Result<HttpResponse, Error> {
let connection = request.connection_info();
let host = connection.remote_addr().unwrap_or("unknown host");
if let Some(token) = &config.server.auth_token { if let Some(token) = &config.server.auth_token {
if request let auth_header = request
.headers() .headers()
.get(AUTHORIZATION) .get(AUTHORIZATION)
.map(|v| v.to_str().unwrap_or_default()) .map(|v| v.to_str().unwrap_or_default())
.map(|v| v.split_whitespace().last().unwrap_or_default()) .map(|v| v.split_whitespace().last().unwrap_or_default());
!= Some(token) if auth_header != Some(token) {
{ log::warn!(
"authorization failure for {} (header: {})",
host,
auth_header.unwrap_or("none"),
);
return Err(error::ErrorUnauthorized("unauthorized")); return Err(error::ErrorUnauthorized("unauthorized"));
} }
} }
@ -57,17 +63,13 @@ async fn upload(
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);
return Err(error::ErrorPayloadTooLarge("upload limit exceeded")); return Err(error::ErrorPayloadTooLarge("upload limit exceeded"));
} }
let file_name = &file::save(content.get_file_name()?, &bytes, &config)?; let file_name = &file::save(content.get_file_name()?, &bytes, &config)?;
let connection = request.connection_info(); log::info!("{} ({}) is uploaded from {}", file_name, bytes_unit, host);
log::info!(
"{} ({}) is uploaded from {}",
file_name,
Byte::from_bytes(bytes.len() as u128).get_appropriate_unit(false),
connection.remote_addr().unwrap_or("unknown host")
);
urls.push(format!( urls.push(format!(
"{}://{}/{}\n", "{}://{}/{}\n",
connection.scheme(), connection.scheme(),
@ -75,7 +77,8 @@ async fn upload(
file_name file_name
)); ));
} else { } else {
return Err(error::ErrorUnprocessableEntity("invalid form parameters")); log::warn!("{} sent an invalid form field", host);
return Err(error::ErrorBadRequest("invalid form field"));
} }
} }
Ok(HttpResponse::Ok().body(urls.join(""))) Ok(HttpResponse::Ok().body(urls.join("")))