load files or something, this is really old code
This commit is contained in:
parent
3683f37943
commit
83d8ffa0ff
1 changed files with 69 additions and 28 deletions
97
src/main.rs
97
src/main.rs
|
@ -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"));
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue