From 6fa0e14469588bdaf07e43afe00331d2978d4dd2 Mon Sep 17 00:00:00 2001
From: p11 <Krapohl.f@gmx.de>
Date: Sun, 5 May 2024 23:57:41 +0200
Subject: [PATCH] Changed how binary replies work

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

diff --git a/src/main.rs b/src/main.rs
index 7aefa4b..1d8b531 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -325,8 +325,10 @@ fn handle_connection(
         let path = path.to_path_buf();
         let ending = ending.to_lowercase();
         match ending.as_ref() {
-            "jpg" => reply_image(stream, &relative_path, "jpeg", path),
-            "png" | "tiff" | "gif" | "jpeg" => reply_image(stream, &relative_path, &ending, path),
+            "jpg" => reply_binary(stream, &relative_path, "image", "jpeg", path),
+            "png" | "tiff" | "gif" | "jpeg" => {
+                reply_binary(stream, &relative_path, "image", &ending, path)
+            }
             _ => fail(stream),
         };
         return;
@@ -378,7 +380,13 @@ fn handle_connection(
     )
 }
 
-fn reply_image(mut stream: TcpStream, relative_path: &str, image_type: &str, mut path: PathBuf) {
+fn reply_binary(
+    mut stream: TcpStream,
+    relative_path: &str,
+    data_type: &str,
+    file_type: &str,
+    mut path: PathBuf,
+) {
     path.push(relative_path);
     let Ok(mut file) = File::open(path) else {
         fail(stream);
@@ -386,7 +394,7 @@ fn reply_image(mut stream: TcpStream, relative_path: &str, image_type: &str, mut
     };
 
     let _ = write!(stream, "HTTP/1.1 200 OK\r\n");
-    let _ = write!(stream, "Content-Type: image/{image_type}\r\n");
+    let _ = write!(stream, "Content-Type: {data_type}/{file_type}\r\n");
     let _ = write!(stream, "\r\n");
 
     let mut buf = [0; 0x400];