1 files changed, 10 insertions, 2 deletions

diff --git a/src/console.rs b/src/console.rs
index eb78c6a..6e2649f 100644
--- a/src/console.rs
+++ b/src/console.rs
@@ -36,7 +36,8 @@ impl ToString for LogMessage {
             Method::OPTIONS => "#423fe0",
             _ => "white",
         };
-        format!("<div><span class='ip'>{}</span> <span class='method' style='color: {};'>{}</span> <span class='path'>{}{}</span> <span class='body'>{}</span></div>", ip, color, self.method, self.path, self.uri, self.body)
+        format!("<div><span class='ip'>{}</span> <span class='method' style='color: {};'>{}</span> <span class='path'>{}{}</span> <span class='body'>{}</span></div>", 
+            ip, color, self.method, self.path, sanatize(self.uri.to_string()), self.body)
     }
 }
 
@@ -200,7 +201,14 @@ impl Formatter for HtmlFormatter {
     }
 }
 
-fn beautify(body: String) -> String {
+pub fn sanatize(input: String) -> String {
+    input.replace("&", "&amp;").replace("<", "&lt;").replace(">", "&gt;")
+}
+
+pub fn beautify(body: String) -> String {
+
+    let body = sanatize(body);
+
     if body.is_empty() {
         return String::new();
     }