utils: add uh_htmlescape() helper
[project/uhttpd.git] / utils.c
diff --git a/utils.c b/utils.c
index ec0b3aa..1c61c41 100644 (file)
--- a/utils.c
+++ b/utils.c
@@ -1,20 +1,20 @@
 /*
  * uhttpd - Tiny single-threaded httpd
  *
- *   Copyright (C) 2010-2012 Jo-Philipp Wich <xm@subsignal.org>
- *   Copyright (C) 2012 Felix Fietkau <nbd@openwrt.org>
+ *   Copyright (C) 2010-2013 Jo-Philipp Wich <xm@subsignal.org>
+ *   Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
  *
- *  Licensed under the Apache License, Version 2.0 (the "License");
- *  you may not use this file except in compliance with the License.
- *  You may obtain a copy of the License at
+ * Permission to use, copy, modify, and/or distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
  *
- *      http://www.apache.org/licenses/LICENSE-2.0
- *
- *  Unless required by applicable law or agreed to in writing, software
- *  distributed under the License is distributed on an "AS IS" BASIS,
- *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- *  See the License for the specific language governing permissions and
- *  limitations under the License.
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  */
 
 #include <ctype.h>
@@ -25,16 +25,23 @@ bool uh_use_chunked(struct client *cl)
        if (cl->request.version != UH_HTTP_VER_1_1)
                return false;
 
-       if (cl->request.method == UH_HTTP_MSG_HEAD)
+       if (cl->request.method == UH_HTTP_MSG_HEAD || cl->request.method == UH_HTTP_MSG_OPTIONS)
+               return false;
+
+       /* RFC2616 10.2.5, 10.3.5 */
+       if (cl->http_code == 204 || cl->http_code == 304)
                return false;
 
-       return true;
+       return !cl->request.disable_chunked;
 }
 
 void uh_chunk_write(struct client *cl, const void *data, int len)
 {
        bool chunked = uh_use_chunked(cl);
 
+       if (cl->state == CLIENT_STATE_CLEANUP)
+               return;
+
        uloop_timeout_set(&cl->timeout, conf.network_timeout * 1000);
        if (chunked)
                ustream_printf(cl->us, "%X\r\n", len);
@@ -49,6 +56,9 @@ void uh_chunk_vprintf(struct client *cl, const char *format, va_list arg)
        va_list arg2;
        int len;
 
+       if (cl->state == CLIENT_STATE_CLEANUP)
+               return;
+
        uloop_timeout_set(&cl->timeout, conf.network_timeout * 1000);
        if (!uh_use_chunked(cl)) {
                ustream_vprintf(cl->us, format, arg);
@@ -81,6 +91,9 @@ void uh_chunk_eof(struct client *cl)
        if (!uh_use_chunked(cl))
                return;
 
+       if (cl->state == CLIENT_STATE_CLEANUP)
+               return;
+
        ustream_printf(cl->us, "0\r\n\r\n");
 }
 
@@ -99,31 +112,18 @@ int uh_urldecode(char *buf, int blen, const char *src, int slen)
 
        for (i = 0; (i < slen) && (len < blen); i++)
        {
-               if (src[i] == '%')
-               {
-                       if (((i+2) < slen) && isxdigit(src[i+1]) && isxdigit(src[i+2]))
-                       {
-                               buf[len++] = (char)(16 * hex(src[i+1]) + hex(src[i+2]));
-                               i += 2;
-                       }
-                       else
-                       {
-                               /* Encoding error: it's hard to think of a
-                               ** scenario in which returning an incorrect
-                               ** 'decoding' of the malformed string is
-                               ** preferable to signaling an error condition. */
-                               #if 0 /* WORSE_IS_BETTER */
-                                   buf[len++] = '%';
-                               #else
-                                   return -2;
-                               #endif
-                       }
-               }
-               else
-               {
+               if (src[i] != '%') {
                        buf[len++] = src[i];
+                       continue;
                }
+
+               if (i + 2 >= slen || !isxdigit(src[i + 1]) || !isxdigit(src[i + 2]))
+                       return -2;
+
+               buf[len++] = (char)(16 * hex(src[i+1]) + hex(src[i+2]));
+               i += 2;
        }
+       buf[len] = 0;
 
        return (i == slen) ? len : -1;
 }
@@ -135,7 +135,7 @@ int uh_urlencode(char *buf, int blen, const char *src, int slen)
 {
        int i;
        int len = 0;
-       const char hex[] = "0123456789abcdef";
+       static const char hex[] = "0123456789abcdef";
 
        for (i = 0; (i < slen) && (len < blen); i++)
        {
@@ -160,18 +160,17 @@ int uh_urlencode(char *buf, int blen, const char *src, int slen)
        return (i == slen) ? len : -1;
 }
 
-int uh_b64decode(char *buf, int blen, const unsigned char *src, int slen)
+int uh_b64decode(char *buf, int blen, const void *src, int slen)
 {
-       int i = 0;
-       int len = 0;
-
-       unsigned int cin  = 0;
+       const unsigned char *str = src;
        unsigned int cout = 0;
+       unsigned int cin  = 0;
+       int len = 0;
+       int i = 0;
 
-
-       for (i = 0; (i <= slen) && (src[i] != 0); i++)
+       for (i = 0; (i <= slen) && (str[i] != 0); i++)
        {
-               cin = src[i];
+               cin = str[i];
 
                if ((cin >= '0') && (cin <= '9'))
                        cin = cin - '0' + 52;
@@ -190,21 +189,105 @@ int uh_b64decode(char *buf, int blen, const unsigned char *src, int slen)
 
                cout = (cout << 6) | cin;
 
-               if ((i % 4) == 3)
-               {
-                       if ((len + 3) < blen)
-                       {
-                               buf[len++] = (char)(cout >> 16);
-                               buf[len++] = (char)(cout >> 8);
-                               buf[len++] = (char)(cout);
-                       }
-                       else
-                       {
-                               break;
-                       }
-               }
+               if ((i % 4) != 3)
+                       continue;
+
+               if ((len + 3) >= blen)
+                       break;
+
+               buf[len++] = (char)(cout >> 16);
+               buf[len++] = (char)(cout >> 8);
+               buf[len++] = (char)(cout);
        }
 
        buf[len++] = 0;
        return len;
 }
+
+bool uh_path_match(const char *prefix, const char *url)
+{
+       int len = strlen(prefix);
+
+       /* A prefix of "/" will - by definition - match any url */
+       if (prefix[0] == '/' && len == 1)
+               return true;
+
+       if (strncmp(url, prefix, len) != 0)
+               return false;
+
+       return url[len] == '/' || url[len] == 0;
+}
+
+char *uh_split_header(char *str)
+{
+       char *val;
+
+       val = strchr(str, ':');
+       if (!val)
+               return NULL;
+
+       *val = 0;
+       val++;
+
+       while (isspace(*val))
+               val++;
+
+       return val;
+}
+
+bool uh_addr_rfc1918(struct uh_addr *addr)
+{
+       uint32_t a;
+
+       if (addr->family != AF_INET)
+               return false;
+
+       a = htonl(addr->in.s_addr);
+       return ((a >= 0x0A000000) && (a <= 0x0AFFFFFF)) ||
+              ((a >= 0xAC100000) && (a <= 0xAC1FFFFF)) ||
+              ((a >= 0xC0A80000) && (a <= 0xC0A8FFFF));
+
+       return 0;
+}
+
+
+static bool is_html_special_char(char c)
+{
+       switch (c)
+       {
+       case 0x22:
+       case 0x26:
+       case 0x27:
+       case 0x3C:
+       case 0x3E:
+               return true;
+
+       default:
+               return false;
+       }
+}
+
+char *uh_htmlescape(const char *str)
+{
+       size_t len;
+       char *p, *copy;
+
+       for (p = str, len = 1; *p; p++)
+               if (is_html_special_char(*p))
+                       len += 6; /* &#x??; */
+               else
+                       len++;
+
+       copy = calloc(1, len);
+
+       if (!copy)
+               return NULL;
+
+       for (p = copy; *str; str++)
+               if (is_html_special_char(*str))
+                       p += sprintf(p, "&#x%02x;", (unsigned int)*str);
+               else
+                       *p++ = *str;
+
+       return copy;
+}