From ecba9a266d6711150e5b651c6ef39b562af599e4 Mon Sep 17 00:00:00 2001
From: int 80h <int@80h.dev>
Date: Mon, 8 Jun 2020 13:21:45 -0400
Subject: [PATCH] Url encoding and decoding

---
 src/main.rs | 18 ++++++++++++++----
 src/util.rs |  8 ++++++++
 2 files changed, 22 insertions(+), 4 deletions(-)
 create mode 100644 src/util.rs

diff --git a/src/main.rs b/src/main.rs
index b458ba5..813cde7 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -24,6 +24,7 @@ mod conn;
 mod logger;
 mod revproxy;
 mod tls;
+mod util;
 
 fn get_mime(path: &PathBuf) -> String {
     let mut mime = "text/gemini".to_string();
@@ -86,10 +87,18 @@ async fn get_content(path: PathBuf, u: url::Url) -> Result<String, io::Error> {
             }
             let file = file.path();
             let p = file.strip_prefix(&path).unwrap();
+            let ps = match p.to_str() {
+                Some(s) => s,
+                None => continue,
+            };
+            let ep = match u.join(ps) {
+                Ok(p) => p,
+                _ => continue,
+            };
             if m.is_dir() {
-                list.push_str(&format!("=> {}/ {}/\r\n", p.display(), p.display()));
+                list.push_str(&format!("=> {}/ {}/\r\n", ep, p.display()));
             } else {
-                list.push_str(&format!("=> {} {}\r\n", p.display(), p.display()));
+                list.push_str(&format!("=> {} {}\r\n", ep, p.display()));
             }
         }
     }
@@ -215,14 +224,15 @@ async fn handle_connection(
         let usr: Vec<&str> = usr.splitn(2, "/").collect();
         path.push("/home/");
         if usr.len() == 2 {
-            path.push(format!("{}/{}/{}", usr[0], "public_gemini", usr[1]));
+            path.push(format!("{}/{}/{}", usr[0], "public_gemini", util::url_decode(usr[1].as_bytes())));
         } else {
             path.push(format!("{}/{}/", usr[0], "public_gemini"));
         }
     } else {
         path.push(&srv.server.dir);
         if url.path() != "" || url.path() != "/" {
-            path.push(url.path().trim_start_matches("/"));
+            let decoded = util::url_decode(url.path().trim_start_matches("/").as_bytes());
+            path.push(decoded);
         }
     }
 
diff --git a/src/util.rs b/src/util.rs
new file mode 100644
index 0000000..c215ad1
--- /dev/null
+++ b/src/util.rs
@@ -0,0 +1,8 @@
+use url::form_urlencoded;
+
+pub fn url_decode(url: &[u8]) -> String {
+    let decoded: String = form_urlencoded::parse(url)
+        .map(|(key, val)| [key, val].concat())
+        .collect();
+    return decoded
+}