spoody/website
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

fucking webcrawler took down my site
All checks were successful
Restart Website Service / restart_service (push) Successful in 1m4s
Details

Browse source
This commit is contained in:
Snorre 2025-04-04 18:47:23 +02:00
parent 55e61bf421
commit 582f839977
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/main.rs
View file

@ -31,14 +31,14 @@ struct Request {
path: Option<String>, path: Option<String>,
} }
impl Request { impl Request {
pub fn from(string: &str) -> Self { pub fn from(string: &str) -> Option<Self> {
let mut parts = string.split(" "); let mut parts = string.split(" ");
Self { Some(Self {
method: match parts.next().expect("Invalid request") { method: match parts.next().expect("Invalid request") {
"GET" => RequestMethod::GET, "GET" => RequestMethod::GET,
"POST" => RequestMethod::POST, "POST" => RequestMethod::POST,
_ => panic!("Unknown method"), // TODO: fix PUT crashing entire website _ => None?
}, },
path: Some( path: Some(
parts parts
@ -49,7 +49,7 @@ impl Request {
.unwrap() .unwrap()
.to_string(), .to_string(),
), ),
} })
} }
pub fn file(&self) -> Option<PathBuf> { pub fn file(&self) -> Option<PathBuf> {
if self.path == None { if self.path == None {
@ -115,6 +115,12 @@ fn handle_connection(mut stream: TcpStream) {
let req = Request::from(http_request.get(0).expect("Empty request")); let req = Request::from(http_request.get(0).expect("Empty request"));
if req.is_none() {
return;
};
let req = req.unwrap();
if req.path == Some("/posts/index".to_string()) { if req.path == Some("/posts/index".to_string()) {
stream stream
.write_all( .write_all(