load files or something, this is really old code

This commit is contained in:
Crispy 2024-03-23 18:03:05 +01:00
parent 3683f37943
commit 83d8ffa0ff

View file

@ -1,7 +1,6 @@
use std::io::prelude::*; use std::io::prelude::*;
use std::{fs, env}; use std::net::{TcpListener, TcpStream};
use std::net::{TcpStream, TcpListener}; use std::{env, fs};
fn main() { fn main() {
let args: Vec<String> = env::args().collect(); let args: Vec<String> = env::args().collect();
@ -16,41 +15,83 @@ fn main() {
let listener = TcpListener::bind(host).expect("Could not bind to address"); let listener = TcpListener::bind(host).expect("Could not bind to address");
for stream in listener.incoming() { for stream in listener.incoming() {
if let Ok(stream) = stream { match stream {
handle_connection(stream); Ok(stream) => handle_connection(stream),
} Err(err) => println!("Error with incoming stream: {}", err),
else {
println!("Error with incoming stream: {}", stream.err().unwrap());
} }
} }
} }
fn handle_connection(mut stream: TcpStream) { fn handle_connection(mut stream: TcpStream) {
let mut buffer = vec![0; 2048]; let mut buffer = vec![0; 2048];
let size = stream.read(&mut buffer).unwrap(); let size = if let Ok(size) = stream.read(&mut buffer) {
size
} else {
return;
};
buffer.resize(size, 0); buffer.resize(size, 0);
send_file(&mut stream, "src/main.rs"); let request = String::from_utf8_lossy(&buffer);
let peer_addr = stream.peer_addr().unwrap(); let peer_addr = stream.peer_addr().ok();
println!("Received {} bytes from {}\n\n[{}]\n", size, peer_addr,
String::from_utf8_lossy(&buffer) println!(
.escape_debug() "Received {} bytes from {:?}\n\n[{}]",
.collect::<String>() size,
.replace("\\r\\n", "\n") peer_addr,
.replace("\\n", "\n") request
.escape_debug()
.collect::<String>()
.replace("\\r\\n", "\n")
.replace("\\n", "\n")
); );
let mut request = request.into_owned();
while request.contains("..") {
request = request.replace("..", "");
}
if request.starts_with("GET") {
if let Some(file) = request.split_whitespace().nth(1) {
let path = format!("./{}", file);
send_file(&mut stream, &path);
return;
}
}
let response = "HTTP/1.1 400 BAD REQUEST\r\n\r\n";
stream
.write_all(response.as_bytes())
.unwrap_or_else(|_| println!("failed to respond"));
stream
.flush()
.unwrap_or_else(|_| println!("failed to respond"));
} }
fn send_file(stream: &mut TcpStream, path: &str) { fn send_file(stream: &mut TcpStream, path: &str) {
let contents = fs::read_to_string(path).unwrap() + "\n\n"; if let Ok(text) = fs::read_to_string(path) {
let contents = text + "\n\n";
let response = format!( let response = format!(
"HTTP/1.1 200 OK\nContent-Length: {}\n\n{}", "HTTP/1.1 200 OK\r\nContent-Type: {}; charset=UTF-8\r\nContent-Length: {}\r\n\r\n{}",
contents.len(), if path.ends_with(".html") {
contents "text/html"
); } else {
stream.write_all(response.as_bytes()).unwrap(); "text/plain"
stream.flush().unwrap(); },
contents.len(),
contents
);
stream
.write_all(response.as_bytes())
.unwrap_or_else(|_| println!("failed to respond"));
} else {
eprintln!("File does not exist: {}", path);
let response = format!("HTTP/1.1 404 NOT FOUND\r\nContent-Type: text/plain; charset=UTF-8\r\nContent-Length: {}\r\n\r\n{}", path.len(), path);
stream
.write_all(response.as_bytes())
.unwrap_or_else(|_| println!("failed to respond with 404"));
};
stream
.flush()
.unwrap_or_else(|_| println!("failed to respond"));
} }