YGGverse/agate
1
0
Fork 0
mirror of https://github.com/YGGverse/agate.git synced 2026-04-08 20:45:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Check URL schemes

Browse source
This commit is contained in:
Matt Brubeck 2020-05-21 10:25:43 -07:00
parent 3a41f40775
commit 6f490876cd
2 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/main.rs
View file

@ -45,6 +45,7 @@ lazy_static! {
static ref ARGS: Args = args()
.expect("usage: agate <addr:port> <dir> <cert> <key>");
static ref ACCEPTOR: TlsAcceptor = acceptor().unwrap();
static ref BASE: Url = Url::parse(&format!("gemini://{}", ARGS.sock_addr)).unwrap();
}
fn args() -> Option<Args> {
@ -87,8 +88,17 @@ async fn parse_request(stream: &mut TlsStream<TcpStream>) -> Result<Url> {
let mut stream = async_std::io::BufReader::new(stream);
let mut request = String::new();
stream.read_line(&mut request).await?;
let url = Url::parse(request.trim())?;
eprintln!("Got request for {:?}", url);
let request = request.trim();
eprintln!("Got request for {:?}", request);
let url = if request.starts_with("//") {
BASE.join(request.trim())?
} else {
Url::parse(request)?
};
if url.scheme() != "gemini" {
Err("unsupported URL scheme")?
}
Ok(url)
}