2 * uhttpd - Tiny single-threaded httpd
4 * Copyright (C) 2010-2013 Jo-Philipp Wich <xm@subsignal.org>
5 * Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
7 * Permission to use, copy, modify, and/or distribute this software for any
8 * purpose with or without fee is hereby granted, provided that the above
9 * copyright notice and this permission notice appear in all copies.
11 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
12 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
14 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
16 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
17 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
23 bool uh_use_chunked(struct client *cl)
25 if (cl->request.version != UH_HTTP_VER_1_1)
28 if (cl->request.method == UH_HTTP_MSG_HEAD || cl->request.method == UH_HTTP_MSG_OPTIONS)
31 /* RFC2616 10.2.5, 10.3.5 */
32 if (cl->http_code == 204 || cl->http_code == 304)
35 return !cl->request.disable_chunked;
38 void uh_chunk_write(struct client *cl, const void *data, int len)
40 bool chunked = uh_use_chunked(cl);
42 if (cl->state == CLIENT_STATE_CLEANUP)
45 uloop_timeout_set(&cl->timeout, conf.network_timeout * 1000);
47 ustream_printf(cl->us, "%X\r\n", len);
48 ustream_write(cl->us, data, len, true);
50 ustream_printf(cl->us, "\r\n", len);
53 void uh_chunk_vprintf(struct client *cl, const char *format, va_list arg)
59 if (cl->state == CLIENT_STATE_CLEANUP)
62 uloop_timeout_set(&cl->timeout, conf.network_timeout * 1000);
63 if (!uh_use_chunked(cl)) {
64 ustream_vprintf(cl->us, format, arg);
69 len = vsnprintf(buf, sizeof(buf), format, arg2);
72 ustream_printf(cl->us, "%X\r\n", len);
73 if (len < sizeof(buf))
74 ustream_write(cl->us, buf, len, true);
76 ustream_vprintf(cl->us, format, arg);
77 ustream_printf(cl->us, "\r\n", len);
80 void uh_chunk_printf(struct client *cl, const char *format, ...)
84 va_start(arg, format);
85 uh_chunk_vprintf(cl, format, arg);
89 void uh_chunk_eof(struct client *cl)
91 if (!uh_use_chunked(cl))
94 if (cl->state == CLIENT_STATE_CLEANUP)
97 ustream_printf(cl->us, "0\r\n\r\n");
100 /* blen is the size of buf; slen is the length of src. The input-string need
101 ** not be, and the output string will not be, null-terminated. Returns the
102 ** length of the decoded string, -1 on buffer overflow, -2 on malformed string. */
103 int uh_urldecode(char *buf, int blen, const char *src, int slen)
109 (((x) <= '9') ? ((x) - '0') : \
110 (((x) <= 'F') ? ((x) - 'A' + 10) : \
113 for (i = 0; (i < slen) && (len < blen); i++)
120 if (i + 2 >= slen || !isxdigit(src[i + 1]) || !isxdigit(src[i + 2]))
123 buf[len++] = (char)(16 * hex(src[i+1]) + hex(src[i+2]));
128 return (i == slen) ? len : -1;
131 /* blen is the size of buf; slen is the length of src. The input-string need
132 ** not be, and the output string will not be, null-terminated. Returns the
133 ** length of the encoded string, or -1 on error (buffer overflow) */
134 int uh_urlencode(char *buf, int blen, const char *src, int slen)
138 static const char hex[] = "0123456789abcdef";
140 for (i = 0; (i < slen) && (len < blen); i++)
142 if( isalnum(src[i]) || (src[i] == '-') || (src[i] == '_') ||
143 (src[i] == '.') || (src[i] == '~') )
147 else if ((len+3) <= blen)
150 buf[len++] = hex[(src[i] >> 4) & 15];
151 buf[len++] = hex[ src[i] & 15];
160 return (i == slen) ? len : -1;
163 int uh_b64decode(char *buf, int blen, const void *src, int slen)
165 const unsigned char *str = src;
166 unsigned int cout = 0;
167 unsigned int cin = 0;
171 for (i = 0; (i <= slen) && (str[i] != 0); i++)
175 if ((cin >= '0') && (cin <= '9'))
176 cin = cin - '0' + 52;
177 else if ((cin >= 'A') && (cin <= 'Z'))
179 else if ((cin >= 'a') && (cin <= 'z'))
180 cin = cin - 'a' + 26;
190 cout = (cout << 6) | cin;
195 if ((len + 3) >= blen)
198 buf[len++] = (char)(cout >> 16);
199 buf[len++] = (char)(cout >> 8);
200 buf[len++] = (char)(cout);
207 bool uh_path_match(const char *prefix, const char *url)
209 int len = strlen(prefix);
211 /* A prefix of "/" will - by definition - match any url */
212 if (prefix[0] == '/' && len == 1)
215 if (strncmp(url, prefix, len) != 0)
218 return url[len] == '/' || url[len] == 0;
221 char *uh_split_header(char *str)
225 val = strchr(str, ':');
232 while (isspace(*val))
238 bool uh_addr_rfc1918(struct uh_addr *addr)
242 if (addr->family != AF_INET)
245 a = htonl(addr->in.s_addr);
246 return ((a >= 0x0A000000) && (a <= 0x0AFFFFFF)) ||
247 ((a >= 0xAC100000) && (a <= 0xAC1FFFFF)) ||
248 ((a >= 0xC0A80000) && (a <= 0xC0A8FFFF));
254 static bool is_html_special_char(char c)
270 char *uh_htmlescape(const char *str)
275 for (p = str, len = 1; *p; p++)
276 if (is_html_special_char(*p))
277 len += 6; /* &#x??; */
281 copy = calloc(1, len);
286 for (p = copy; *str; str++)
287 if (is_html_special_char(*str))
288 p += sprintf(p, "&#x%02x;", (unsigned int)*str);