Add some checks
This commit is contained in:
parent
ebdd92c64c
commit
c8539f78a1
41
src/main.rs
41
src/main.rs
|
@ -17,14 +17,41 @@ fn handle_connection(mut stream: TcpStream) {
|
||||||
|
|
||||||
stream.read(&mut buffer).unwrap();
|
stream.read(&mut buffer).unwrap();
|
||||||
|
|
||||||
let contents = fs::read_to_string("index.gmi").unwrap();
|
let hostname = "localhost";
|
||||||
let response = format!(
|
let path = "/";
|
||||||
"2 text/gemini\r\n{}",
|
let post = "\r\n";
|
||||||
contents
|
|
||||||
);
|
/*
|
||||||
|
let get = format!(
|
||||||
|
"{} {} {} {}",
|
||||||
|
hostname,
|
||||||
|
path,
|
||||||
|
post.len(),
|
||||||
|
post
|
||||||
|
);*/
|
||||||
|
|
||||||
|
let get = b"127.0.0.1 / 0\r\n";
|
||||||
|
|
||||||
|
if buffer.starts_with(get) {
|
||||||
|
let contents = fs::read_to_string("index.gmi").unwrap();
|
||||||
|
let status = 2;
|
||||||
|
let meta = "text/gemini";
|
||||||
|
let response = format!(
|
||||||
|
"{} {}\r\n{}",
|
||||||
|
status,
|
||||||
|
meta,
|
||||||
|
contents
|
||||||
|
);
|
||||||
|
|
||||||
|
stream.write(response.as_bytes()).unwrap();
|
||||||
|
stream.flush().unwrap();
|
||||||
|
} else {
|
||||||
|
let status_line = "4 Not Found";
|
||||||
|
|
||||||
|
stream.write(status_line.as_bytes()).unwrap();
|
||||||
|
stream.flush().unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
stream.write(response.as_bytes()).unwrap();
|
|
||||||
stream.flush().unwrap();
|
|
||||||
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue