From b8e2ff3fd8e15aa13f99ed0856c8b6e6412f3e7c Mon Sep 17 00:00:00 2001
From: int 80h <int@80h.dev>
Date: Tue, 5 May 2020 19:31:31 -0400
Subject: [PATCH] Handle some errors

---
 src/main.rs | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index a6279f4..504358c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -101,15 +101,27 @@ fn get_content(path: PathBuf, u: url::Url) -> Result<String, io::Error> {
 async fn handle_connection(mut con: conn::Connection) -> Result<(), io::Error> {
     let now: DateTime<Utc> = Utc::now();
     println!("{} New Connection: {}", now, con.peer_addr);
-    let mut buffer = [0; 512];
+    let mut buffer = [0; 1024];
     con.stream.read(&mut buffer).await?;
-    let request = String::from_utf8_lossy(&buffer[..]).to_owned();
+    let request = match String::from_utf8(buffer[..].to_vec()) {
+        Ok(request) => request,
+        Err(_) => {
+            println!("Bad Request");
+            con.send_status(status::Status::BadRequest, "Bad Request!\r\n").await?;
+            return Ok(())
+        }
+    };
     println!("Request: {}", request);
 
-    let url = Url::parse(&request).unwrap();
+    let url = match Url::parse(&request) {
+        Ok(url) => url,
+        Err(_) => { con.send_status(status::Status::BadRequest, "Bad Request!\r\n").await?;
+                return Ok(())
+        }
+    };
 
     if Some(con.hostname.as_str()) != url.host_str() {
-        con.send_status(status::Status::PermanentFailure, "Url doesn't match certificate!").await?;
+        con.send_status(status::Status::ProxyRequestRefused, "Url doesn't match certificate!").await?;
         return Ok(());
     }