Spaces:
Runtime error
Runtime error
| // CGI handler code inspired by SVR.JS's RedBrick mod, translated from JavaScript to Rust. | |
| use std::collections::HashMap; | |
| use std::error::Error; | |
| use std::path::{Path, PathBuf}; | |
| use std::process::Stdio; | |
| use std::sync::Arc; | |
| use std::time::Duration; | |
| use crate::ferron_common::{ | |
| ErrorLogger, HyperRequest, HyperResponse, RequestData, ResponseData, ServerConfig, ServerModule, | |
| ServerModuleHandlers, SocketData, | |
| }; | |
| use crate::ferron_common::{HyperUpgraded, WithRuntime}; | |
| use async_trait::async_trait; | |
| use futures_util::TryStreamExt; | |
| use hashlink::LinkedHashMap; | |
| use http_body_util::{BodyExt, StreamBody}; | |
| use httparse::EMPTY_HEADER; | |
| use hyper::body::Frame; | |
| use hyper::{header, Response, StatusCode}; | |
| use hyper_tungstenite::HyperWebsocket; | |
| use tokio::fs; | |
| use tokio::io::AsyncReadExt; | |
| use tokio::process::Command; | |
| use tokio::runtime::Handle; | |
| use tokio::sync::RwLock; | |
| use tokio_util::io::{ReaderStream, StreamReader}; | |
| use crate::ferron_res::server_software::SERVER_SOFTWARE; | |
| use crate::ferron_util::cgi_response::CgiResponse; | |
| use crate::ferron_util::copy_move::Copier; | |
| use crate::ferron_util::ttl_cache::TtlCache; | |
| pub fn server_module_init( | |
| _config: &ServerConfig, | |
| ) -> Result<Box<dyn ServerModule + Send + Sync>, Box<dyn Error + Send + Sync>> { | |
| let cache = Arc::new(RwLock::new(TtlCache::new(Duration::from_millis(100)))); | |
| Ok(Box::new(CgiModule::new(cache))) | |
| } | |
| struct CgiModule { | |
| path_cache: Arc<RwLock<TtlCache<String, (Option<PathBuf>, Option<String>)>>>, | |
| } | |
| impl CgiModule { | |
| fn new(path_cache: Arc<RwLock<TtlCache<String, (Option<PathBuf>, Option<String>)>>>) -> Self { | |
| Self { path_cache } | |
| } | |
| } | |
| impl ServerModule for CgiModule { | |
| fn get_handlers(&self, handle: Handle) -> Box<dyn ServerModuleHandlers + Send> { | |
| Box::new(CgiModuleHandlers { | |
| path_cache: self.path_cache.clone(), | |
| handle, | |
| }) | |
| } | |
| } | |
| struct CgiModuleHandlers { | |
| handle: Handle, | |
| path_cache: Arc<RwLock<TtlCache<String, (Option<PathBuf>, Option<String>)>>>, | |
| } | |
| impl ServerModuleHandlers for CgiModuleHandlers { | |
| async fn request_handler( | |
| &mut self, | |
| request: RequestData, | |
| config: &ServerConfig, | |
| socket_data: &SocketData, | |
| error_logger: &ErrorLogger, | |
| ) -> Result<ResponseData, Box<dyn Error + Send + Sync>> { | |
| WithRuntime::new(self.handle.clone(), async move { | |
| let mut cgi_script_exts = Vec::new(); | |
| let cgi_script_exts_yaml = &config["cgiScriptExtensions"]; | |
| if let Some(cgi_script_exts_obtained) = cgi_script_exts_yaml.as_vec() { | |
| for cgi_script_ext_yaml in cgi_script_exts_obtained.iter() { | |
| if let Some(cgi_script_ext) = cgi_script_ext_yaml.as_str() { | |
| cgi_script_exts.push(cgi_script_ext); | |
| } | |
| } | |
| } | |
| if let Some(wwwroot) = config["wwwroot"].as_str() { | |
| let hyper_request = request.get_hyper_request(); | |
| let request_path = hyper_request.uri().path(); | |
| let mut request_path_bytes = request_path.bytes(); | |
| if request_path_bytes.len() < 1 || request_path_bytes.nth(0) != Some(b'/') { | |
| return Ok( | |
| ResponseData::builder(request) | |
| .status(StatusCode::BAD_REQUEST) | |
| .build(), | |
| ); | |
| } | |
| let cache_key = format!( | |
| "{}{}{}", | |
| match config["ip"].as_str() { | |
| Some(ip) => format!("{}-", ip), | |
| None => String::from(""), | |
| }, | |
| match config["domain"].as_str() { | |
| Some(domain) => format!("{}-", domain), | |
| None => String::from(""), | |
| }, | |
| request_path | |
| ); | |
| let wwwroot_unknown = PathBuf::from(wwwroot); | |
| let wwwroot_pathbuf = match wwwroot_unknown.as_path().is_absolute() { | |
| true => wwwroot_unknown, | |
| false => match fs::canonicalize(&wwwroot_unknown).await { | |
| Ok(pathbuf) => pathbuf, | |
| Err(_) => wwwroot_unknown, | |
| }, | |
| }; | |
| let wwwroot = wwwroot_pathbuf.as_path(); | |
| let read_rwlock = self.path_cache.read().await; | |
| let (execute_pathbuf, execute_path_info) = match read_rwlock.get(&cache_key) { | |
| Some(data) => { | |
| drop(read_rwlock); | |
| data | |
| } | |
| None => { | |
| drop(read_rwlock); | |
| let mut relative_path = &request_path[1..]; | |
| while relative_path.as_bytes().first().copied() == Some(b'/') { | |
| relative_path = &relative_path[1..]; | |
| } | |
| let decoded_relative_path = match urlencoding::decode(relative_path) { | |
| Ok(path) => path.to_string(), | |
| Err(_) => { | |
| return Ok( | |
| ResponseData::builder(request) | |
| .status(StatusCode::BAD_REQUEST) | |
| .build(), | |
| ); | |
| } | |
| }; | |
| let joined_pathbuf = wwwroot.join(decoded_relative_path); | |
| let mut execute_pathbuf: Option<PathBuf> = None; | |
| let mut execute_path_info: Option<String> = None; | |
| match fs::metadata(&joined_pathbuf).await { | |
| Ok(metadata) => { | |
| if metadata.is_file() { | |
| let mut request_path_normalized = match cfg!(windows) { | |
| true => request_path.to_lowercase(), | |
| false => request_path.to_string(), | |
| }; | |
| while request_path_normalized.contains("//") { | |
| request_path_normalized = request_path_normalized.replace("//", "/"); | |
| } | |
| if request_path_normalized == "/cgi-bin" | |
| || request_path_normalized.starts_with("/cgi-bin/") | |
| { | |
| execute_pathbuf = Some(joined_pathbuf); | |
| } else { | |
| let contained_extension = joined_pathbuf | |
| .extension() | |
| .map(|a| format!(".{}", a.to_string_lossy())); | |
| if let Some(contained_extension) = contained_extension { | |
| if cgi_script_exts.contains(&(&contained_extension as &str)) { | |
| execute_pathbuf = Some(joined_pathbuf); | |
| } | |
| } | |
| } | |
| } else if metadata.is_dir() { | |
| let indexes = vec!["index.php", "index.cgi"]; | |
| for index in indexes { | |
| let temp_joined_pathbuf = joined_pathbuf.join(index); | |
| match fs::metadata(&temp_joined_pathbuf).await { | |
| Ok(temp_metadata) => { | |
| if temp_metadata.is_file() { | |
| let request_path_normalized = match cfg!(windows) { | |
| true => request_path.to_lowercase(), | |
| false => request_path.to_string(), | |
| }; | |
| if request_path_normalized == "/cgi-bin" | |
| || request_path_normalized.starts_with("/cgi-bin/") | |
| { | |
| execute_pathbuf = Some(temp_joined_pathbuf); | |
| break; | |
| } else { | |
| let contained_extension = temp_joined_pathbuf | |
| .extension() | |
| .map(|a| format!(".{}", a.to_string_lossy())); | |
| if let Some(contained_extension) = contained_extension { | |
| if cgi_script_exts.contains(&(&contained_extension as &str)) { | |
| execute_pathbuf = Some(temp_joined_pathbuf); | |
| break; | |
| } | |
| } | |
| } | |
| } | |
| } | |
| Err(_) => continue, | |
| }; | |
| } | |
| } | |
| } | |
| Err(err) => { | |
| if err.kind() == tokio::io::ErrorKind::NotADirectory { | |
| // TODO: find a file | |
| let mut temp_pathbuf = joined_pathbuf.clone(); | |
| loop { | |
| if !temp_pathbuf.pop() { | |
| break; | |
| } | |
| match fs::metadata(&temp_pathbuf).await { | |
| Ok(metadata) => { | |
| if metadata.is_file() { | |
| let temp_path = temp_pathbuf.as_path(); | |
| if !temp_path.starts_with(wwwroot) { | |
| // Traversed above the webroot, so ignore that. | |
| break; | |
| } | |
| let path_info = match joined_pathbuf.as_path().strip_prefix(temp_path) { | |
| Ok(path) => { | |
| let path = path.to_string_lossy().to_string(); | |
| Some(match cfg!(windows) { | |
| true => path.replace("\\", "/"), | |
| false => path, | |
| }) | |
| } | |
| Err(_) => None, | |
| }; | |
| let mut request_path_normalized = match cfg!(windows) { | |
| true => request_path.to_lowercase(), | |
| false => request_path.to_string(), | |
| }; | |
| while request_path_normalized.contains("//") { | |
| request_path_normalized = request_path_normalized.replace("//", "/"); | |
| } | |
| if request_path_normalized == "/cgi-bin" | |
| || request_path_normalized.starts_with("/cgi-bin/") | |
| { | |
| execute_pathbuf = Some(temp_pathbuf); | |
| execute_path_info = path_info; | |
| break; | |
| } else { | |
| let contained_extension = temp_pathbuf | |
| .extension() | |
| .map(|a| format!(".{}", a.to_string_lossy())); | |
| if let Some(contained_extension) = contained_extension { | |
| if cgi_script_exts.contains(&(&contained_extension as &str)) { | |
| execute_pathbuf = Some(temp_pathbuf); | |
| execute_path_info = path_info; | |
| break; | |
| } | |
| } | |
| } | |
| } else { | |
| break; | |
| } | |
| } | |
| Err(err) => match err.kind() { | |
| tokio::io::ErrorKind::NotADirectory => (), | |
| _ => break, | |
| }, | |
| }; | |
| } | |
| } | |
| } | |
| }; | |
| let data = (execute_pathbuf, execute_path_info); | |
| let mut write_rwlock = self.path_cache.write().await; | |
| write_rwlock.cleanup(); | |
| write_rwlock.insert(cache_key, data.clone()); | |
| drop(write_rwlock); | |
| data | |
| } | |
| }; | |
| if let Some(execute_pathbuf) = execute_pathbuf { | |
| let mut cgi_interpreters = HashMap::new(); | |
| cgi_interpreters.insert(".pl".to_string(), vec!["perl".to_string()]); | |
| cgi_interpreters.insert(".py".to_string(), vec!["python".to_string()]); | |
| cgi_interpreters.insert(".sh".to_string(), vec!["bash".to_string()]); | |
| cgi_interpreters.insert(".ksh".to_string(), vec!["ksh".to_string()]); | |
| cgi_interpreters.insert(".csh".to_string(), vec!["csh".to_string()]); | |
| cgi_interpreters.insert(".rb".to_string(), vec!["ruby".to_string()]); | |
| cgi_interpreters.insert(".php".to_string(), vec!["php-cgi".to_string()]); | |
| if cfg!(windows) { | |
| cgi_interpreters.insert(".exe".to_string(), vec![]); | |
| cgi_interpreters.insert( | |
| ".bat".to_string(), | |
| vec!["cmd".to_string(), "/c".to_string()], | |
| ); | |
| cgi_interpreters.insert(".vbs".to_string(), vec!["cscript".to_string()]); | |
| } | |
| let cgi_interpreters_yaml = &config["cgiScriptInterpreters"]; | |
| if let Some(cgi_interpreters_hashmap) = cgi_interpreters_yaml.as_hash() { | |
| for (key_yaml, value_yaml) in cgi_interpreters_hashmap.iter() { | |
| if let Some(key) = key_yaml.as_str() { | |
| if value_yaml.is_null() { | |
| cgi_interpreters.remove(key); | |
| } else if let Some(value) = value_yaml.as_vec() { | |
| let mut params = Vec::new(); | |
| for param_yaml in value.iter() { | |
| if let Some(param) = param_yaml.as_str() { | |
| params.push(param.to_string()); | |
| } | |
| } | |
| cgi_interpreters.insert(key.to_string(), params); | |
| } | |
| } | |
| } | |
| } | |
| return execute_cgi_with_environment_variables( | |
| request, | |
| socket_data, | |
| error_logger, | |
| wwwroot, | |
| execute_pathbuf, | |
| execute_path_info, | |
| config["serverAdministratorEmail"].as_str(), | |
| cgi_interpreters, | |
| ) | |
| .await; | |
| } | |
| } | |
| Ok(ResponseData::builder(request).build()) | |
| }) | |
| .await | |
| } | |
| async fn proxy_request_handler( | |
| &mut self, | |
| request: RequestData, | |
| _config: &ServerConfig, | |
| _socket_data: &SocketData, | |
| _error_logger: &ErrorLogger, | |
| ) -> Result<ResponseData, Box<dyn Error + Send + Sync>> { | |
| Ok(ResponseData::builder(request).build()) | |
| } | |
| async fn response_modifying_handler( | |
| &mut self, | |
| response: HyperResponse, | |
| ) -> Result<HyperResponse, Box<dyn Error + Send + Sync>> { | |
| Ok(response) | |
| } | |
| async fn proxy_response_modifying_handler( | |
| &mut self, | |
| response: HyperResponse, | |
| ) -> Result<HyperResponse, Box<dyn Error + Send + Sync>> { | |
| Ok(response) | |
| } | |
| async fn connect_proxy_request_handler( | |
| &mut self, | |
| _upgraded_request: HyperUpgraded, | |
| _connect_address: &str, | |
| _config: &ServerConfig, | |
| _socket_data: &SocketData, | |
| _error_logger: &ErrorLogger, | |
| ) -> Result<(), Box<dyn Error + Send + Sync>> { | |
| Ok(()) | |
| } | |
| fn does_connect_proxy_requests(&mut self) -> bool { | |
| false | |
| } | |
| async fn websocket_request_handler( | |
| &mut self, | |
| _websocket: HyperWebsocket, | |
| _uri: &hyper::Uri, | |
| _config: &ServerConfig, | |
| _socket_data: &SocketData, | |
| _error_logger: &ErrorLogger, | |
| ) -> Result<(), Box<dyn Error + Send + Sync>> { | |
| Ok(()) | |
| } | |
| fn does_websocket_requests(&mut self, _config: &ServerConfig, _socket_data: &SocketData) -> bool { | |
| false | |
| } | |
| } | |
| async fn execute_cgi_with_environment_variables( | |
| request: RequestData, | |
| socket_data: &SocketData, | |
| error_logger: &ErrorLogger, | |
| wwwroot: &Path, | |
| execute_pathbuf: PathBuf, | |
| path_info: Option<String>, | |
| server_administrator_email: Option<&str>, | |
| cgi_interpreters: HashMap<String, Vec<String>>, | |
| ) -> Result<ResponseData, Box<dyn Error + Send + Sync>> { | |
| let mut environment_variables: LinkedHashMap<String, String> = LinkedHashMap::new(); | |
| let hyper_request = request.get_hyper_request(); | |
| let original_request_uri = request.get_original_url().unwrap_or(hyper_request.uri()); | |
| if let Some(auth_user) = request.get_auth_user() { | |
| if let Some(authorization) = hyper_request.headers().get(header::AUTHORIZATION) { | |
| let authorization_value = String::from_utf8_lossy(authorization.as_bytes()).to_string(); | |
| let mut authorization_value_split = authorization_value.split(" "); | |
| if let Some(authorization_type) = authorization_value_split.next() { | |
| environment_variables.insert("AUTH_TYPE".to_string(), authorization_type.to_string()); | |
| } | |
| } | |
| environment_variables.insert("REMOTE_USER".to_string(), auth_user.to_string()); | |
| } | |
| environment_variables.insert( | |
| "QUERY_STRING".to_string(), | |
| match hyper_request.uri().query() { | |
| Some(query) => query.to_string(), | |
| None => "".to_string(), | |
| }, | |
| ); | |
| environment_variables.insert("SERVER_SOFTWARE".to_string(), SERVER_SOFTWARE.to_string()); | |
| environment_variables.insert( | |
| "SERVER_PROTOCOL".to_string(), | |
| match hyper_request.version() { | |
| hyper::Version::HTTP_09 => "HTTP/0.9".to_string(), | |
| hyper::Version::HTTP_10 => "HTTP/1.0".to_string(), | |
| hyper::Version::HTTP_11 => "HTTP/1.1".to_string(), | |
| hyper::Version::HTTP_2 => "HTTP/2.0".to_string(), | |
| hyper::Version::HTTP_3 => "HTTP/3.0".to_string(), | |
| _ => "HTTP/Unknown".to_string(), | |
| }, | |
| ); | |
| environment_variables.insert( | |
| "SERVER_PORT".to_string(), | |
| socket_data.local_addr.port().to_string(), | |
| ); | |
| environment_variables.insert( | |
| "SERVER_ADDR".to_string(), | |
| socket_data.local_addr.ip().to_canonical().to_string(), | |
| ); | |
| if let Some(server_administrator_email) = server_administrator_email { | |
| environment_variables.insert( | |
| "SERVER_ADMIN".to_string(), | |
| server_administrator_email.to_string(), | |
| ); | |
| } | |
| if let Some(host) = hyper_request.headers().get(header::HOST) { | |
| environment_variables.insert( | |
| "SERVER_NAME".to_string(), | |
| String::from_utf8_lossy(host.as_bytes()).to_string(), | |
| ); | |
| } | |
| environment_variables.insert( | |
| "DOCUMENT_ROOT".to_string(), | |
| wwwroot.to_string_lossy().to_string(), | |
| ); | |
| environment_variables.insert( | |
| "PATH_INFO".to_string(), | |
| match &path_info { | |
| Some(path_info) => format!("/{}", path_info), | |
| None => "".to_string(), | |
| }, | |
| ); | |
| environment_variables.insert( | |
| "PATH_TRANSLATED".to_string(), | |
| match &path_info { | |
| Some(path_info) => { | |
| let mut path_translated = execute_pathbuf.clone(); | |
| path_translated.push(path_info); | |
| path_translated.to_string_lossy().to_string() | |
| } | |
| None => "".to_string(), | |
| }, | |
| ); | |
| environment_variables.insert( | |
| "REQUEST_METHOD".to_string(), | |
| hyper_request.method().to_string(), | |
| ); | |
| environment_variables.insert("GATEWAY_INTERFACE".to_string(), "CGI/1.1".to_string()); | |
| environment_variables.insert( | |
| "REQUEST_URI".to_string(), | |
| format!( | |
| "{}{}", | |
| original_request_uri.path(), | |
| match original_request_uri.query() { | |
| Some(query) => format!("?{}", query), | |
| None => String::from(""), | |
| } | |
| ), | |
| ); | |
| environment_variables.insert( | |
| "REMOTE_PORT".to_string(), | |
| socket_data.remote_addr.port().to_string(), | |
| ); | |
| environment_variables.insert( | |
| "REMOTE_ADDR".to_string(), | |
| socket_data.remote_addr.ip().to_canonical().to_string(), | |
| ); | |
| environment_variables.insert( | |
| "SCRIPT_FILENAME".to_string(), | |
| execute_pathbuf.to_string_lossy().to_string(), | |
| ); | |
| if let Ok(script_path) = execute_pathbuf.as_path().strip_prefix(wwwroot) { | |
| environment_variables.insert( | |
| "SCRIPT_NAME".to_string(), | |
| format!( | |
| "/{}", | |
| match cfg!(windows) { | |
| true => script_path.to_string_lossy().to_string().replace("\\", "/"), | |
| false => script_path.to_string_lossy().to_string(), | |
| } | |
| ), | |
| ); | |
| } | |
| if socket_data.encrypted { | |
| environment_variables.insert("HTTPS".to_string(), "ON".to_string()); | |
| } | |
| for (header_name, header_value) in hyper_request.headers().iter() { | |
| let env_header_name = match *header_name { | |
| header::CONTENT_LENGTH => "CONTENT_LENGTH".to_string(), | |
| header::CONTENT_TYPE => "CONTENT_TYPE".to_string(), | |
| _ => { | |
| let mut result = String::new(); | |
| result.push_str("HTTP_"); | |
| for c in header_name.as_str().to_uppercase().chars() { | |
| if c.is_alphanumeric() { | |
| result.push(c); | |
| } else { | |
| result.push('_'); | |
| } | |
| } | |
| result | |
| } | |
| }; | |
| if environment_variables.contains_key(&env_header_name) { | |
| let value = environment_variables.get_mut(&env_header_name); | |
| if let Some(value) = value { | |
| if env_header_name == "HTTP_COOKIE" { | |
| value.push_str("; "); | |
| } else { | |
| // See https://stackoverflow.com/a/1801191 | |
| value.push_str(", "); | |
| } | |
| value.push_str(String::from_utf8_lossy(header_value.as_bytes()).as_ref()); | |
| } else { | |
| environment_variables.insert( | |
| env_header_name, | |
| String::from_utf8_lossy(header_value.as_bytes()).to_string(), | |
| ); | |
| } | |
| } else { | |
| environment_variables.insert( | |
| env_header_name, | |
| String::from_utf8_lossy(header_value.as_bytes()).to_string(), | |
| ); | |
| } | |
| } | |
| let (hyper_request, _, _) = request.into_parts(); | |
| execute_cgi( | |
| hyper_request, | |
| error_logger, | |
| execute_pathbuf, | |
| cgi_interpreters, | |
| environment_variables, | |
| ) | |
| .await | |
| } | |
| async fn execute_cgi( | |
| hyper_request: HyperRequest, | |
| error_logger: &ErrorLogger, | |
| execute_pathbuf: PathBuf, | |
| cgi_interpreters: HashMap<String, Vec<String>>, | |
| environment_variables: LinkedHashMap<String, String>, | |
| ) -> Result<ResponseData, Box<dyn Error + Send + Sync>> { | |
| let (_, body) = hyper_request.into_parts(); | |
| let executable_params = match get_executable(&execute_pathbuf).await { | |
| Ok(params) => params, | |
| Err(err) => { | |
| let contained_extension = execute_pathbuf | |
| .extension() | |
| .map(|a| format!(".{}", a.to_string_lossy())); | |
| if let Some(contained_extension) = contained_extension { | |
| if let Some(params_init) = cgi_interpreters.get(&contained_extension) { | |
| let mut params: Vec<String> = params_init.iter().map(|s| s.to_owned()).collect(); | |
| params.push(execute_pathbuf.to_string_lossy().to_string()); | |
| params | |
| } else { | |
| Err(err)? | |
| } | |
| } else { | |
| Err(err)? | |
| } | |
| } | |
| }; | |
| let mut executable_params_iter = executable_params.iter(); | |
| let mut command = Command::new(match executable_params_iter.next() { | |
| Some(executable_name) => executable_name, | |
| None => Err(anyhow::anyhow!("Cannot determine the executable"))?, | |
| }); | |
| // Set standard I/O to be piped | |
| command.stdin(Stdio::piped()); | |
| command.stdout(Stdio::piped()); | |
| command.stderr(Stdio::piped()); | |
| for param in executable_params_iter { | |
| command.arg(param); | |
| } | |
| command.envs(environment_variables); | |
| let mut execute_dir_pathbuf = execute_pathbuf.clone(); | |
| execute_dir_pathbuf.pop(); | |
| command.current_dir(execute_dir_pathbuf); | |
| let mut child = command.spawn()?; | |
| let cgi_stdin_reader = StreamReader::new(body.into_data_stream().map_err(std::io::Error::other)); | |
| let stdin = match child.stdin.take() { | |
| Some(stdin) => stdin, | |
| None => Err(anyhow::anyhow!( | |
| "The CGI process doesn't have standard input" | |
| ))?, | |
| }; | |
| let stdout = match child.stdout.take() { | |
| Some(stdout) => stdout, | |
| None => Err(anyhow::anyhow!( | |
| "The CGI process doesn't have standard output" | |
| ))?, | |
| }; | |
| let stderr = child.stderr.take(); | |
| let mut cgi_response = CgiResponse::new(stdout); | |
| let stdin_copy_future = Copier::new(cgi_stdin_reader, stdin).copy(); | |
| let mut stdin_copy_future_pinned = Box::pin(stdin_copy_future); | |
| let mut headers = [EMPTY_HEADER; 128]; | |
| let mut early_stdin_copied = false; | |
| // Needed to wrap this in another scope to prevent errors with multiple mutable borrows. | |
| { | |
| let mut head_obtained = false; | |
| let stdout_parse_future = cgi_response.get_head(); | |
| tokio::pin!(stdout_parse_future); | |
| // Cannot use a loop with tokio::select, since stdin_copy_future_pinned being constantly ready will make the web server stop responding to HTTP requests | |
| tokio::select! { | |
| biased; | |
| obtained_head = &mut stdout_parse_future => { | |
| let obtained_head = obtained_head?; | |
| if !obtained_head.is_empty() { | |
| httparse::parse_headers(obtained_head, &mut headers)?; | |
| } | |
| head_obtained = true; | |
| }, | |
| result = &mut stdin_copy_future_pinned => { | |
| early_stdin_copied = true; | |
| result?; | |
| } | |
| } | |
| if !head_obtained { | |
| // Kept it same as in the tokio::select macro | |
| let obtained_head = stdout_parse_future.await?; | |
| if !obtained_head.is_empty() { | |
| httparse::parse_headers(obtained_head, &mut headers)?; | |
| } | |
| } | |
| } | |
| let mut response_builder = Response::builder(); | |
| let mut status_code = 200; | |
| for header in headers { | |
| if header == EMPTY_HEADER { | |
| break; | |
| } | |
| let mut is_status_header = false; | |
| match &header.name.to_lowercase() as &str { | |
| "location" => { | |
| if !(300..=399).contains(&status_code) { | |
| status_code = 302; | |
| } | |
| } | |
| "status" => { | |
| is_status_header = true; | |
| let header_value_cow = String::from_utf8_lossy(header.value); | |
| let mut split_status = header_value_cow.split(" "); | |
| let first_part = split_status.next(); | |
| if let Some(first_part) = first_part { | |
| if first_part.starts_with("HTTP/") { | |
| let second_part = split_status.next(); | |
| if let Some(second_part) = second_part { | |
| if let Ok(parsed_status_code) = second_part.parse::<u16>() { | |
| status_code = parsed_status_code; | |
| } | |
| } | |
| } else if let Ok(parsed_status_code) = first_part.parse::<u16>() { | |
| status_code = parsed_status_code; | |
| } | |
| } | |
| } | |
| _ => (), | |
| } | |
| if !is_status_header { | |
| response_builder = response_builder.header(header.name, header.value); | |
| } | |
| } | |
| response_builder = response_builder.status(status_code); | |
| let reader_stream = ReaderStream::new(cgi_response); | |
| let stream_body = StreamBody::new(reader_stream.map_ok(Frame::data)); | |
| let boxed_body = stream_body.boxed(); | |
| let response = response_builder.body(boxed_body)?; | |
| if let Some(exit_code) = child.try_wait()? { | |
| if !exit_code.success() { | |
| if let Some(mut stderr) = stderr { | |
| let mut stderr_string = String::new(); | |
| stderr | |
| .read_to_string(&mut stderr_string) | |
| .await | |
| .unwrap_or_default(); | |
| let stderr_string_trimmed = stderr_string.trim(); | |
| if !stderr_string_trimmed.is_empty() { | |
| error_logger | |
| .log(&format!("There were CGI errors: {}", stderr_string_trimmed)) | |
| .await; | |
| } | |
| } | |
| return Ok( | |
| ResponseData::builder_without_request() | |
| .status(StatusCode::INTERNAL_SERVER_ERROR) | |
| .build(), | |
| ); | |
| } | |
| } | |
| let error_logger = error_logger.clone(); | |
| Ok( | |
| ResponseData::builder_without_request() | |
| .response(response) | |
| .parallel_fn(async move { | |
| if !early_stdin_copied { | |
| stdin_copy_future_pinned.await.unwrap_or_default(); | |
| } | |
| if let Some(mut stderr) = stderr { | |
| let mut stderr_string = String::new(); | |
| stderr | |
| .read_to_string(&mut stderr_string) | |
| .await | |
| .unwrap_or_default(); | |
| let stderr_string_trimmed = stderr_string.trim(); | |
| if !stderr_string_trimmed.is_empty() { | |
| error_logger | |
| .log(&format!("There were CGI errors: {}", stderr_string_trimmed)) | |
| .await; | |
| } | |
| } | |
| }) | |
| .build(), | |
| ) | |
| } | |
| async fn get_executable( | |
| execute_pathbuf: &PathBuf, | |
| ) -> Result<Vec<String>, Box<dyn Error + Send + Sync>> { | |
| use std::os::unix::fs::PermissionsExt; | |
| let metadata = fs::metadata(&execute_pathbuf).await?; | |
| let permissions = metadata.permissions(); | |
| let is_executable = permissions.mode() & 0o111 != 0; | |
| if !is_executable { | |
| Err(anyhow::anyhow!("The CGI program is not executable"))? | |
| } | |
| let executable_params_vector = vec![execute_pathbuf.to_string_lossy().to_string()]; | |
| Ok(executable_params_vector) | |
| } | |
| async fn get_executable( | |
| execute_pathbuf: &PathBuf, | |
| ) -> Result<Vec<String>, Box<dyn Error + Send + Sync>> { | |
| use tokio::io::{AsyncBufReadExt, AsyncSeekExt, BufReader}; | |
| let mut magic_signature_buffer = [0u8; 2]; | |
| let mut open_file = fs::File::open(&execute_pathbuf).await?; | |
| if open_file | |
| .read_exact(&mut magic_signature_buffer) | |
| .await | |
| .is_err() | |
| { | |
| Err(anyhow::anyhow!("Failed to read the CGI program signature"))? | |
| } | |
| match &magic_signature_buffer { | |
| b"PE" => { | |
| // Windows executables | |
| let executable_params_vector = vec![execute_pathbuf.to_string_lossy().to_string()]; | |
| Ok(executable_params_vector) | |
| } | |
| b"#!" => { | |
| // Scripts with a shebang line | |
| open_file.rewind().await?; | |
| let mut buffered_file = BufReader::new(open_file); | |
| let mut shebang_line = String::new(); | |
| buffered_file.read_line(&mut shebang_line).await?; | |
| let mut command_begin: Vec<String> = (&shebang_line[2..]) | |
| .replace("\r", "") | |
| .replace("\n", "") | |
| .split(" ") | |
| .map(|s| s.to_owned()) | |
| .collect(); | |
| command_begin.push(execute_pathbuf.to_string_lossy().to_string()); | |
| Ok(command_begin) | |
| } | |
| _ => { | |
| // It's not executable | |
| Err(anyhow::anyhow!("The CGI program is not executable"))? | |
| } | |
| } | |
| } | |