ui-shared: prevent malicious filename from injecting headers

1 files changed, 26 insertions, 0 deletions

diff --git a/html.c b/html.c
index 959148c..d89df3a 100644
--- a/html.c
+++ b/html.c
@@ -239,6 +239,32 @@ void html_url_arg(const char *txt)
                 html(txt);
 }
 
+void html_header_arg_in_quotes(const char *txt)
+{
+        const char *t = txt;
+        while (t && *t) {
+                unsigned char c = *t;
+                const char *e = NULL;
+                if (c == '\\')
+                        e = "\\\\";
+                else if (c == '\r')
+                        e = "\\r";
+                else if (c == '\n')
+                        e = "\\n";
+                else if (c == '"')
+                        e = "\\\"";
+                if (e) {
+                        html_raw(txt, t - txt);
+                        html(e);
+                        txt = t + 1;
+                }
+                t++;
+        }
+        if (t != txt)
+                html(txt);
+
+}
+
 void html_hidden(const char *name, const char *value)
 {
         html("<input type='hidden' name='");