From cac7e394be426388b4ec66d7bcda3dfb073f6f75 Mon Sep 17 00:00:00 2001 From: Matt Brubeck Date: Fri, 22 May 2020 15:26:16 -0700 Subject: [PATCH] Minor cleanup in parse_request --- src/main.rs | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/main.rs b/src/main.rs index 75311a7..4243437 100644 --- a/src/main.rs +++ b/src/main.rs @@ -89,14 +89,13 @@ async fn parse_request(stream: &mut TlsStream) -> Result { while !buf.is_empty() { let n = stream.read(buf).await?; len += n; - if n == 0 || request[..len].ends_with(b"\r\n") { + if request[..len].ends_with(b"\r\n") { break; + } else if n == 0 { + Err("Request ended unexpectedly")? } buf = &mut request[len..]; } - if !request[..len].ends_with(b"\r\n") { - Err("Missing CRLF")? - } let request = str::from_utf8(&request[..len - 2])?; let url = if request.starts_with("//") {