add: conditional request handling

4 years ago · bd1d20fd12
parent 308e492fe7
commit bd1d20fd12
1 changed files with 26 additions and 6 deletions
--- a/src/main.rs
+++ b/src/main.rs
@ -15,15 +15,35 @@ fn handle_connection(mut stream: TcpStream) {
    // make it 8k?
    // https://stackoverflow.com/questions/686217/maximum-on-http-header-values
    let mut buffer = [0; 1024];
-
    stream.read(&mut buffer).unwrap();

-    let ip = stream.peer_addr().unwrap().ip().to_string();
-
+    let get = b"GET / HTTP/1.1\r\n";
+
+    let (status_line, contents) = if buffer.starts_with(get) {
+        (
+            "HTTP/1.1 200 OK",
+            stream.peer_addr().unwrap().ip().to_string(),
+        )
+    } else {
+        (
+            "HTTP/1.1 404 NOT FOUND",
+            "<html>\n<head><title>404 Not Found</title></head>\n\
+            <body bgcolor=\"white\">\n<center><h1>404 Not Found</h1></center>\n\
+            <hr>\n</body>\n</html>"
+                .to_string(),
+        )
+    };
+
+    // add date header formatted to rfc2822?
    let response = format!(
-        "HTTP/1.1 200 OK\r\nContent-Lengh: {}\r\n\r\n{}",
-        ip.len(),
-        ip
+        "{}\r\n\
+        Access-control-allow-origin: *\r\n\
+        Content-type: text/plain; charset=utf-8\r\n\
+        Content-Lengh: {}\r\n\r\n\
+        {}",
+        status_line,
+        contents.len(),
+        contents
    );

    stream.write(response.as_bytes()).unwrap();