From 4a298c8b1ece254e88320965ba42cc55f73a3544 Mon Sep 17 00:00:00 2001
From: YK <yk@qcl.one>
Date: Sat, 11 May 2024 01:09:54 +0300
Subject: [PATCH] stage 7 attempt 3 (add 404)

---
 src/main.rs | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index f836c02..d8f05f3 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -70,12 +70,12 @@ async fn process (mut stream: TcpStream, args: A) -> Result<()> {
         p if p.starts_with("/files/") => {
             let path = args.directory.join(p.trim_start_matches("/files/"));
             let mut buf = vec![];
-            let mut f = File::open(path).await?;
-
-
-            let _ = f.read_to_end(&mut buf).await;
-
-            Response::OctetStream(buf)
+            if let Ok(mut f) = File::open(path).await {
+                let _ = f.read_to_end(&mut buf).await;
+                Response::OctetStream(buf)
+            } else {
+                Response::_404
+            }
         },
         _ => Response::_404,
     };