ti-nuage/gemserv
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Avoid logging 0 characters

Browse Source
This commit is contained in:
Remco van 't Veer
2020-10-23 10:30:50 +02:00
committed by int 80h
parent ddecdbccb3
commit 6fceccd2cd
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/main.rs
View File

@@ -182,12 +182,15 @@ async fn handle_connection(
None => "index.gemini".to_string(), None => "index.gemini".to_string(),
}; };
let mut buffer = [0; 1024]; let mut buffer = [0; 1024];
if let Err(_) = tokio::time::timeout(tokio::time::Duration::from_secs(5), con.stream.read(&mut buffer)).await { let len = match tokio::time::timeout(tokio::time::Duration::from_secs(5), con.stream.read(&mut buffer)).await {
Ok(result) => result.unwrap(),
Err(_) => {
logger::logger(con.peer_addr, Status::BadRequest, ""); logger::logger(con.peer_addr, Status::BadRequest, "");
con.send_status(Status::BadRequest, None).await?; con.send_status(Status::BadRequest, None).await?;
return Ok(()); return Ok(());
} }
let mut request = match String::from_utf8(buffer[..].to_vec()) { };
let mut request = match String::from_utf8(buffer[..len].to_vec()) {
Ok(request) => request, Ok(request) => request,
Err(_) => { Err(_) => {
logger::logger(con.peer_addr, Status::BadRequest, ""); logger::logger(con.peer_addr, Status::BadRequest, "");