Files
gemserv/src/cgi.rs

43 lines
1.3 KiB
Rust

use std::collections::HashMap;
use std::io::{self, BufReader};
use std::path::{Path, PathBuf};
use std::process::Command;
use std::os::unix::fs::PermissionsExt;
use tokio::net::TcpStream;
use tokio_rustls::server::TlsStream;
use url::Url;
use crate::config;
use crate::status;
use crate::conn;
pub async fn cgi(mut con: conn::Connection, path: PathBuf, url: Url) -> Result<(), io::Error> {
let mut envs = HashMap::new();
envs.insert("GEMINI_URL", url.as_str());
envs.insert("SERVER_NAME", url.host_str().unwrap());
envs.insert("SCRIPT_NAME", path.file_name().unwrap().to_str().unwrap());
envs.insert("SERVER_PROTOCOL", "GEMINI");
let addr = con.peer_addr.ip().to_string();
envs.insert("REMOTE_ADDR", &addr);
envs.insert("REMOTE_HOST", &addr);
let port = con.peer_addr.port().to_string();
envs.insert("REMOTE_PORT", &port);
if let Some(q) = url.query() {
envs.insert("QUERY_STRING", q);
}
let cmd = Command::new(path.to_str().unwrap())
.env_clear()
.envs(&envs)
.output()
.unwrap();
if !cmd.status.success() {
con.send_status(status::Status::CGIError, "CGI Error!").await?;
return Ok(());
}
let cmd = String::from_utf8(cmd.stdout).unwrap();
con.send_raw(cmd.as_bytes()).await?;
return Ok(());
}