2021-09-02 18:13:37 +01:00
use std ::io ::{ Write , BufReader , BufRead } ;
2021-12-24 21:30:39 +00:00
use std ::thread ;
2021-08-29 16:07:05 +01:00
use std ::net ::{ TcpListener , TcpStream } ;
2021-08-30 16:17:11 +01:00
use std ::fs ;
use std ::string ::{ String } ;
2021-12-22 20:30:53 +00:00
use std ::process ::Command ;
2023-10-19 11:24:36 +01:00
use packeteer ::http1 ::* ;
use packeteer ::generate_kvheader ;
struct Resource {
contents : String ,
status_code : i32 ,
mime : String ,
}
2021-08-30 16:17:11 +01:00
2021-12-26 17:29:34 +00:00
fn process_cgi ( filename : String ) -> String {
// This is gonna be the boggiest implementation of CGI that anyone
// has ever seen in the history of the fucking world
let query ;
let script ;
if filename . contains ( " ? " ) {
let vars = filename . to_string ( ) . find ( " ? " ) . unwrap ( ) ;
let vars1 = vars + 1 ;
query = & filename [ vars1 .. ] ;
script = & filename [ .. vars ] ;
} else {
query = " " ;
script = & filename ;
}
2022-08-28 10:42:32 +01:00
println! ( " {} " , script ) ;
2021-12-26 17:29:34 +00:00
2022-09-17 09:37:42 +01:00
let output ;
if query ! = " " {
output = Command ::new ( format! ( " ./ {} " , script ) )
. env ( " GATEWAY_INTERFACE " , " CGI/1.1 " )
2023-10-19 11:24:36 +01:00
. env ( " SERVER_SOFTWARE " , format! ( " herb/ {} " , env! ( " CARGO_PKG_VERSION " ) ) )
2022-09-17 09:37:42 +01:00
. env ( " REQUEST_METHOD " , " GET " )
. env ( " QUERY_STRING " , query )
. env ( " SCRIPT_NAME " , script )
. output ( )
. expect ( " failed to execute process " ) ;
} else {
output = Command ::new ( format! ( " ./ {} " , script ) )
. env ( " GATEWAY_INTERFACE " , " CGI/1.1 " )
2023-10-19 11:24:36 +01:00
. env ( " SERVER_SOFTWARE " , format! ( " herb/ {} " , env! ( " CARGO_PKG_VERSION " ) ) )
2022-09-17 09:37:42 +01:00
. env ( " REQUEST_METHOD " , " GET " )
. env ( " SCRIPT_NAME " , script )
. output ( )
. expect ( " failed to execute process " ) ;
}
2021-12-26 17:29:34 +00:00
let outputd = String ::from_utf8_lossy ( & output . stdout ) . to_string ( ) ;
return outputd ;
}
2021-12-22 20:02:09 +00:00
fn detect_media_type ( filename : String ) -> String {
// The Lynx terminal browser made me do this.
2021-12-26 17:29:34 +00:00
let path = filename . to_string ( ) ;
let paths = check_if_path_exists ( path ) ;
let path = filename . to_string ( ) ;
let dir = check_if_dir ( path ) ;
if paths = = true & & dir = = false & & filename . contains ( " . " ) = = false {
return " text/html " . to_string ( ) ;
} else {
let ext_index = filename . to_string ( ) . find ( " . " ) . unwrap ( ) ;
let test = ext_index + 1 ;
let ext = & filename [ test .. ] ;
if filename [ test .. ] . contains ( " cgi " ) { return " text/html " . to_string ( ) ; }
match ext {
" aac " = > " audio/aac " . to_string ( ) ,
" avi " = > " video/x-msvideo " . to_string ( ) ,
" bmp " = > " image/bmp " . to_string ( ) ,
" bz2 " = > " application/x-bzip2 " . to_string ( ) ,
" css " = > " text/css " . to_string ( ) ,
" gz " = > " application/gzip " . to_string ( ) ,
" gif " = > " image/gif " . to_string ( ) ,
" png " = > " image/png " . to_string ( ) ,
" pdf " = > " application/pdf " . to_string ( ) ,
" jpeg " = > " image/jpeg " . to_string ( ) ,
" jpg " = > " image/jpeg " . to_string ( ) ,
" js " = > " text/javascript " . to_string ( ) ,
" mid " = > " audio/midi " . to_string ( ) ,
" midi " = > " audio/midi " . to_string ( ) ,
" mp3 " = > " audio/mpeg " . to_string ( ) ,
" mp4 " = > " video/mp4 " . to_string ( ) ,
" mpeg " = > " video/mpeg " . to_string ( ) ,
" ogg " = > " audio/ogg " . to_string ( ) ,
" oga " = > " audio/ogg " . to_string ( ) ,
" ogv " = > " video/ogg " . to_string ( ) ,
" opus " = > " audio/opus " . to_string ( ) ,
" txt " = > " text/plain " . to_string ( ) ,
" html " = > " text/html " . to_string ( ) ,
" htm " = > " text/html " . to_string ( ) ,
" cgi " = > " text/html " . to_string ( ) ,
" .error_server_404 " = > " text/html " . to_string ( ) ,
2023-10-19 11:24:36 +01:00
" .error_server_501 " = > " text/html " . to_string ( ) ,
2021-12-26 17:29:34 +00:00
_ = > " application/octet-stream " . to_string ( ) ,
}
2021-12-22 20:02:09 +00:00
}
}
2021-09-02 18:13:37 +01:00
fn check_if_path_exists ( path : String ) -> bool {
// This is probably not the best way of checking if a path
// exists but I don't care :)
let file = fs ::metadata ( path ) ;
match file {
Ok ( _ ) = > true ,
Err ( _ ) = > false ,
}
}
fn check_if_dir ( directory : String ) -> bool {
let path = directory + " /index.html " ;
let result = check_if_path_exists ( path ) ;
return result ;
}
2021-09-05 16:16:07 +01:00
fn generate_index ( directory : String ) -> String {
2021-12-26 18:27:56 +00:00
let mut index = format! ( " <!DOCTYPE HTML><html><body><h1>Directory of {} </h1><hr/> " , directory ) ;
2021-09-05 16:16:07 +01:00
for file in fs ::read_dir ( directory ) . unwrap ( ) {
2021-12-26 18:27:56 +00:00
index = format! ( " {} <br/><a href= {} > {:#?} </a> " , index , format! ( " \" ./ {} \" " , file . as_ref ( ) . unwrap ( ) . path ( ) . display ( ) . to_string ( ) ) , file . unwrap ( ) . file_name ( ) ) ;
2021-09-05 16:16:07 +01:00
}
2023-10-19 11:24:36 +01:00
return format! ( " {} <hr/>Generated by herb {} " , index , env! ( " CARGO_PKG_VERSION " ) ) . to_string ( ) ;
2021-09-05 16:16:07 +01:00
}
2021-09-02 17:25:53 +01:00
fn get_page ( filename : String ) -> String {
2021-09-02 16:16:38 +01:00
2021-08-30 16:17:11 +01:00
// The loaded page should be left immutable as it does
// not need to be modified by the server.
2021-09-05 16:16:07 +01:00
let path = filename . to_string ( ) ;
2023-10-19 11:24:36 +01:00
let checks = check_if_path_exists ( path . clone ( ) ) ;
let index = check_if_dir ( path . clone ( ) ) ;
2021-09-05 16:16:07 +01:00
println! ( " {} {} {} " , path , checks , index ) ;
if checks = = true & & index = = false {
2021-12-26 17:29:34 +00:00
if path . contains ( " . " ) ! = true & & path . contains ( " .cgi? " ) ! = true {
2021-09-05 16:16:07 +01:00
let result = generate_index ( filename ) ;
2021-12-26 17:29:34 +00:00
return result . to_string ( ) ;
} else if path . contains ( " .cgi? " ) {
let result = process_cgi ( filename ) ;
2021-09-05 16:16:07 +01:00
return result . to_string ( ) ;
}
}
2023-10-19 11:24:36 +01:00
if filename . contains ( " .cgi " ) {
2021-12-26 17:29:34 +00:00
let result = process_cgi ( filename ) ;
return result . to_string ( ) ;
2021-09-02 17:25:53 +01:00
} else {
let result = fs ::read_to_string ( filename ) ;
match result {
Ok ( i ) = > i . to_string ( ) ,
2022-08-28 10:54:00 +01:00
Err ( _e ) = > " <!DOCTYPE HTML><html><body><h1>500 Internal Server Error</h1><p>The resource you are trying to access cannot be read by the server.</p></body></html> " . to_string ( ) ,
2021-09-02 17:25:53 +01:00
}
2021-08-30 17:32:44 +01:00
}
2021-08-30 16:17:11 +01:00
}
2021-08-29 16:07:05 +01:00
2021-12-22 20:30:53 +00:00
fn grab_time ( ) -> String {
let output = Command ::new ( " date " )
. arg ( " +'%a, %d %b %Y %H:%m:%S %Z' " )
. output ( )
. expect ( " failed to execute process " ) ;
2023-10-19 11:24:36 +01:00
return String ::from_utf8_lossy ( & output . stdout ) . to_string ( ) . replace ( " ' " , " " ) . replace ( " \n " , " " ) ;
2021-12-22 20:30:53 +00:00
}
2023-10-19 11:24:36 +01:00
fn process_request ( request : Vec < u8 > ) -> Resource {
2021-09-02 16:16:38 +01:00
let mut input = String ::from_utf8_lossy ( & request ) . to_string ( ) ;
2021-09-02 18:13:37 +01:00
2021-12-22 20:02:09 +00:00
let debug = false ;
2021-09-02 18:13:37 +01:00
let mut index = String ::new ( ) ;
2021-09-02 16:18:23 +01:00
let output ;
2021-09-02 16:16:38 +01:00
if input . contains ( " GET " ) {
// To-do: find a less atrocious way to do this.
2021-09-02 17:28:07 +01:00
println! ( " Stream sent GET request. " ) ;
2021-12-22 20:02:09 +00:00
if debug { println! ( " {} " , input ) ; }
2021-09-02 16:16:38 +01:00
input = input . replace ( " GET " , " " ) ;
2021-09-02 17:25:53 +01:00
input = input . replace ( " HTTP/1.1 \r \n " , " " ) ;
2021-12-22 20:02:09 +00:00
// Lynx also made me do this
input = input . replace ( " HTTP/1.0 \r \n " , " " ) ;
2021-09-02 16:16:38 +01:00
// Theoretically by this point, the request
// will have been cut down to just the
// requested resource, but in my experience
// this code is gonna result in like 50k errors
// and I'm gonna have to rewrite it to get it
// to actually work the way I want it to.
2021-09-02 17:25:53 +01:00
if input ! = " / " {
2021-09-02 16:25:13 +01:00
let mut chars = input . chars ( ) ;
chars . next ( ) ;
2021-09-02 18:13:37 +01:00
let path = chars . as_str ( ) . to_string ( ) ;
let exists = check_if_path_exists ( path ) ;
2021-12-26 17:29:34 +00:00
let path = chars . as_str ( ) . to_string ( ) ;
if exists = = false & & path . contains ( " .cgi? " ) = = false {
2023-10-19 11:24:36 +01:00
let resource = Resource { contents : " <!DOCTYPE HTML><html><body><h1>404 Not Found</h1><p>The resource you are trying to locate cannot be accessed!</p></body></html> " . to_string ( ) , status_code : 404 , mime : " text/html " . to_string ( ) } ;
return resource ;
2021-09-02 18:13:37 +01:00
} else {
let path = chars . as_str ( ) . to_string ( ) ;
let dir = check_if_dir ( path ) ;
if dir = = true {
let path = chars . as_str ( ) . to_string ( ) ;
index + = & path ;
index + = " /index.html " ;
output = index . as_str ( ) . clone ( ) ;
} else {
output = chars . as_str ( ) ;
}
}
2021-09-02 17:25:53 +01:00
} else {
output = " index.html " ;
2021-09-02 16:16:38 +01:00
}
2021-09-02 17:25:53 +01:00
2021-09-02 16:16:38 +01:00
} else {
// It's get_page()'s problem now.
2023-10-19 11:24:36 +01:00
let resource = Resource { contents : " <!DOCTYPE HTML><html><body><h1>501 Not Implemented</h1><p>The request sent by your web browser cannot be handled by this server software.</p></body></html> " . to_string ( ) , status_code : 501 , mime : " text/html " . to_string ( ) } ;
2021-09-02 18:13:37 +01:00
println! ( " Stream sent unhandlable request. " ) ;
2023-10-19 11:24:36 +01:00
return resource ;
2021-09-02 16:16:38 +01:00
}
2021-09-02 16:25:13 +01:00
// Did you want to see chars.as_str().to_string()?
2023-10-19 11:24:36 +01:00
let mut resource = Resource { contents : get_page ( output . to_string ( ) ) , status_code : 200 , mime : detect_media_type ( output . to_string ( ) ) } ;
if resource . contents . contains ( " 500 Internal Server Error " ) {
resource . status_code = 500 ;
resource . mime = " text/html " . to_string ( ) ;
}
return resource ;
2021-08-30 17:32:44 +01:00
2021-08-30 17:18:22 +01:00
}
2021-08-29 16:07:05 +01:00
fn serve ( mut stream : TcpStream ) {
2021-12-24 21:30:39 +00:00
thread ::spawn ( move | | {
println! ( " Stream thread created. " ) ;
let mut request = Vec ::new ( ) ;
let mut reader = BufReader ::new ( & mut stream ) ;
reader
. read_until ( b '\n' , & mut request )
. expect ( " Failed to read from stream! " ) ;
2021-08-30 17:18:22 +01:00
2021-12-24 21:30:39 +00:00
let resource = process_request ( request ) ;
2021-12-22 20:02:09 +00:00
2023-10-19 11:24:36 +01:00
// let contents = get_page(resource.clone());
// let header = "HTTP/1.1 200 OK\r\n";
// let content_type = format!("Content-Type: {}\r\n", mime);
// let server = format!("Server: herb/{}\r\n", env!("CARGO_PKG_VERSION"));
// let extra_fields;
2021-12-22 20:02:09 +00:00
2023-10-19 11:24:36 +01:00
let mut response_constructed = generate_response ( resource . status_code , & resource . contents ) ;
response_constructed . headers . push ( generate_kvheader ( " Content-Type " , & resource . mime ) ) ;
response_constructed . headers . push ( generate_kvheader ( " Server " , & format! ( " herb/ {} " , env! ( " CARGO_PKG_VERSION " ) ) ) ) ;
2021-12-22 20:38:36 +00:00
2021-12-24 21:30:39 +00:00
if cfg! ( unix ) {
2023-10-19 11:24:36 +01:00
response_constructed . headers . push ( generate_kvheader ( " Date " , & grab_time ( ) ) )
2021-12-24 21:30:39 +00:00
} else {
// I don't have a Windows or macOS box to test anything on
// which means others are gonna have to deal with it :/
}
2021-08-30 16:17:11 +01:00
2023-10-19 11:24:36 +01:00
println! ( " {:#?} " , response_constructed ) ;
stream . write ( unpack_response ( response_constructed ) . as_bytes ( ) ) . unwrap ( ) ;
2021-12-24 21:30:39 +00:00
stream . flush ( ) . unwrap ( ) ;
} ) ;
2021-08-29 16:07:05 +01:00
}
fn main ( ) -> std ::io ::Result < ( ) > {
2021-12-22 20:30:53 +00:00
let listen = TcpListener ::bind ( " 0.0.0.0:8080 " ) ? ;
2021-08-29 16:07:05 +01:00
for stream in listen . incoming ( ) {
2021-08-30 17:18:22 +01:00
println! ( " Serving incoming stream. " ) ;
2021-08-29 16:07:05 +01:00
serve ( stream ? ) ;
}
Ok ( ( ) )
2021-08-29 15:18:20 +01:00
}