uhttpd/file: fix string out of buffer range on uh_defer_script
[project/uhttpd.git] / main.c
diff --git a/main.c b/main.c
index 8dc352e4a5e2ca932b5c9db1a29ebedec1bf8dcb..de5f2c3a6bcb298385e8840606eae19f737c4c43 100644 (file)
--- a/main.c
+++ b/main.c
  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  */
 
+#ifndef _DEFAULT_SOURCE
+# define _DEFAULT_SOURCE
+#endif
+
 #define _BSD_SOURCE
 #define _GNU_SOURCE
 #define _XOPEN_SOURCE  700
@@ -30,6 +34,7 @@
 #include <signal.h>
 
 #include <libubox/usock.h>
+#include <libubox/utils.h>
 
 #include "uhttpd.h"
 #include "tls.h"
@@ -134,9 +139,12 @@ static int usage(const char *name)
                "       -s [addr:]port  Like -p but provide HTTPS on this port\n"
                "       -C file         ASN.1 server certificate file\n"
                "       -K file         ASN.1 server private key file\n"
+               "       -P ciphers      Colon separated list of allowed TLS ciphers\n"
+               "       -q              Redirect all HTTP requests to HTTPS\n"
 #endif
                "       -h directory    Specify the document root, default is '.'\n"
                "       -E string       Use given virtual URL as 404 error handler\n"
+               "       -b string       Use given charset for directory listings, default to UTF-8\n"
                "       -I string       Use given filename as index for directories, multiple allowed\n"
                "       -S              Do not follow symbolic links outside of the docroot\n"
                "       -D              Do not allow directory listings, send 403 instead\n"
@@ -144,19 +152,27 @@ static int usage(const char *name)
                "       -n count        Maximum allowed number of concurrent script requests\n"
                "       -N count        Maximum allowed number of concurrent connections\n"
 #ifdef HAVE_LUA
-               "       -l string       URL prefix for Lua handler, default is '/lua'\n"
-               "       -L file         Lua handler script, omit to disable Lua\n"
+               "       -l string       URL prefix for Lua handler\n"
+               "       -L file         Path to Lua handler script, -l and -L may be repeated in pairs\n"
+#endif
+#ifdef HAVE_UCODE
+               "       -o string       URL prefix for ucode handler\n"
+               "       -O file         Path to ucode handler script, -o and -O may be repeated in pairs\n"
 #endif
 #ifdef HAVE_UBUS
                "       -u string       URL prefix for UBUS via JSON-RPC handler\n"
                "       -U file         Override ubus socket path\n"
                "       -a              Do not authenticate JSON-RPC requests against UBUS session api\n"
+               "       -X              Enable CORS HTTP headers on JSON-RPC api\n"
+               "       -e              Events subscription reconnection time (retry value)\n"
 #endif
                "       -x string       URL prefix for CGI handler, default is '/cgi-bin'\n"
+               "       -y alias[=path] URL alias handle\n"
                "       -i .ext=path    Use interpreter at path for files with the given extension\n"
                "       -t seconds      CGI, Lua and UBUS script timeout in seconds, default is 60\n"
                "       -T seconds      Network timeout in seconds, default is 30\n"
                "       -k seconds      HTTP keepalive timeout\n"
+               "       -A seconds      TCP keepalive timeout, default is unset\n"
                "       -d string       URL decode given string\n"
                "       -r string       Specify basic auth realm\n"
                "       -m string       MD5 crypt given string\n"
@@ -175,6 +191,11 @@ static void init_defaults_pre(void)
        conf.realm = "Protected Area";
        conf.cgi_prefix = "/cgi-bin";
        conf.cgi_path = "/sbin:/usr/sbin:/bin:/usr/bin";
+       INIT_LIST_HEAD(&conf.cgi_alias);
+       INIT_LIST_HEAD(&conf.lua_prefix);
+#if HAVE_UCODE
+       INIT_LIST_HEAD(&conf.ucode_prefix);
+#endif
 }
 
 static void init_defaults_post(void)
@@ -183,6 +204,15 @@ static void init_defaults_post(void)
        uh_index_add("index.htm");
        uh_index_add("default.html");
        uh_index_add("default.htm");
+
+       if (conf.cgi_prefix) {
+               char *str = malloc(strlen(conf.docroot) + strlen(conf.cgi_prefix) + 1);
+
+               strcpy(str, conf.docroot);
+               strcat(str, conf.cgi_prefix);
+               conf.cgi_docroot_path = str;
+               conf.cgi_prefix_len = strlen(conf.cgi_prefix);
+       };
 }
 
 static void fixup_prefix(char *str)
@@ -194,23 +224,67 @@ static void fixup_prefix(char *str)
 
        len = strlen(str) - 1;
 
-       while (len > 0 && str[len] == '/')
+       while (len >= 0 && str[len] == '/')
                len--;
 
        str[len + 1] = 0;
 }
 
+#ifdef HAVE_LUA
+static void add_lua_prefix(const char *prefix, const char *handler) {
+       struct lua_prefix *p;
+       char *pprefix, *phandler;
+
+       p = calloc_a(sizeof(*p),
+                    &pprefix, strlen(prefix) + 1,
+                    &phandler, strlen(handler) + 1);
+
+       if (!p)
+               return;
+
+       p->prefix = strcpy(pprefix, prefix);
+       p->handler = strcpy(phandler, handler);
+
+       list_add_tail(&p->list, &conf.lua_prefix);
+}
+#endif
+
+#ifdef HAVE_UCODE
+static void add_ucode_prefix(const char *prefix, const char *handler) {
+       struct ucode_prefix *p;
+       char *pprefix, *phandler;
+
+       p = calloc_a(sizeof(*p),
+                    &pprefix, strlen(prefix) + 1,
+                    &phandler, strlen(handler) + 1);
+
+       if (!p)
+               return;
+
+       p->prefix = strcpy(pprefix, prefix);
+       p->handler = strcpy(phandler, handler);
+
+       list_add_tail(&p->list, &conf.ucode_prefix);
+}
+#endif
+
 int main(int argc, char **argv)
 {
+       struct alias *alias;
        bool nofork = false;
        char *port;
        int opt, ch;
        int cur_fd;
        int bound = 0;
-
 #ifdef HAVE_TLS
        int n_tls = 0;
-       const char *tls_key = NULL, *tls_crt = NULL;
+       const char *tls_key = NULL, *tls_crt = NULL, *tls_ciphers = NULL;
+#endif
+#ifdef HAVE_LUA
+       const char *lua_prefix = NULL, *lua_handler = NULL;
+#endif
+#ifdef HAVE_UCODE
+       const char *ucode_prefix = NULL, *ucode_handler = NULL;
 #endif
 
        BUILD_BUG_ON(sizeof(uh_buf) < PATH_MAX);
@@ -219,7 +293,7 @@ int main(int argc, char **argv)
        init_defaults_pre();
        signal(SIGPIPE, SIG_IGN);
 
-       while ((ch = getopt(argc, argv, "afSDRC:K:E:I:p:s:h:c:l:L:d:r:m:n:N:x:i:t:k:T:A:u:U:")) != -1) {
+       while ((ch = getopt(argc, argv, "A:ab:C:c:Dd:E:e:fh:H:I:i:K:k:L:l:m:N:n:O:o:P:p:qRr:Ss:T:t:U:u:Xx:y:")) != -1) {
                switch(ch) {
 #ifdef HAVE_TLS
                case 'C':
@@ -230,19 +304,31 @@ int main(int argc, char **argv)
                        tls_key = optarg;
                        break;
 
+               case 'P':
+                       tls_ciphers = optarg;
+                       break;
+
+               case 'q':
+                       conf.tls_redirect = 1;
+                       break;
+
                case 's':
                        n_tls++;
                        /* fall through */
 #else
                case 'C':
                case 'K':
+               case 'P':
+               case 'q':
                case 's':
                        fprintf(stderr, "uhttpd: TLS support not compiled, "
-                                       "ignoring -%c\n", opt);
+                                       "ignoring -%c\n", ch);
                        break;
 #endif
                case 'p':
+                       optarg = strdup(optarg);
                        bound += add_listener_arg(optarg, (ch == 's'));
+                       free(optarg);
                        break;
 
                case 'h':
@@ -254,6 +340,14 @@ int main(int argc, char **argv)
                        conf.docroot = strdup(uh_buf);
                        break;
 
+               case 'H':
+                       if (uh_handler_add(optarg)) {
+                               fprintf(stderr, "Error: Failed to load handler script %s\n",
+                                       optarg);
+                               exit(1);
+                       }
+                       break;
+
                case 'E':
                        if (optarg[0] != '/') {
                                fprintf(stderr, "Error: Invalid error handler: %s\n",
@@ -272,6 +366,10 @@ int main(int argc, char **argv)
                        uh_index_add(optarg);
                        break;
 
+               case 'b':
+                       conf.dirlist_charset = optarg;
+                       break;
+
                case 'S':
                        conf.no_symlinks = 1;
                        break;
@@ -297,7 +395,21 @@ int main(int argc, char **argv)
                        conf.cgi_prefix = optarg;
                        break;
 
+               case 'y':
+                       alias = calloc(1, sizeof(*alias));
+                       if (!alias) {
+                               fprintf(stderr, "Error: failed to allocate alias\n");
+                               exit(1);
+                       }
+                       alias->alias = strdup(optarg);
+                       alias->path = strchr(alias->alias, '=');
+                       if (alias->path)
+                               *alias->path++ = 0;
+                       list_add(&alias->list, &conf.cgi_alias);
+                       break;
+
                case 'i':
+                       optarg = strdup(optarg);
                        port = strchr(optarg, '=');
                        if (optarg[0] != '.' || !port) {
                                fprintf(stderr, "Error: Invalid interpreter: %s\n",
@@ -330,6 +442,7 @@ int main(int argc, char **argv)
                        break;
 
                case 'd':
+                       optarg = strdup(optarg);
                        port = alloca(strlen(optarg) + 1);
                        if (!port)
                                return -1;
@@ -346,6 +459,7 @@ int main(int argc, char **argv)
                        }
 
                        printf("%s", port);
+                       return 0;
                        break;
 
                /* basic auth realm */
@@ -366,17 +480,66 @@ int main(int argc, char **argv)
 
 #ifdef HAVE_LUA
                case 'l':
-                       conf.lua_prefix = optarg;
-                       break;
-
                case 'L':
-                       conf.lua_handler = optarg;
+                       if (ch == 'l') {
+                               if (lua_prefix)
+                                       fprintf(stderr, "uhttpd: Ignoring previous -%c %s\n",
+                                               ch, lua_prefix);
+
+                               lua_prefix = optarg;
+                       }
+                       else {
+                               if (lua_handler)
+                                       fprintf(stderr, "uhttpd: Ignoring previous -%c %s\n",
+                                               ch, lua_handler);
+
+                               lua_handler = optarg;
+                       }
+
+                       if (lua_prefix && lua_handler) {
+                               add_lua_prefix(lua_prefix, lua_handler);
+                               lua_prefix = NULL;
+                               lua_handler = NULL;
+                       }
+
                        break;
 #else
                case 'l':
                case 'L':
                        fprintf(stderr, "uhttpd: Lua support not compiled, "
-                                       "ignoring -%c\n", opt);
+                                       "ignoring -%c\n", ch);
+                       break;
+#endif
+#ifdef HAVE_UCODE
+               case 'o':
+               case 'O':
+                       if (ch == 'o') {
+                               if (ucode_prefix)
+                                       fprintf(stderr, "uhttpd: Ignoring previous -%c %s\n",
+                                               ch, ucode_prefix);
+
+                               ucode_prefix = optarg;
+                       }
+                       else {
+                               if (ucode_handler)
+                                       fprintf(stderr, "uhttpd: Ignoring previous -%c %s\n",
+                                               ch, ucode_handler);
+
+                               ucode_handler = optarg;
+                       }
+
+                       if (ucode_prefix && ucode_handler) {
+                               add_ucode_prefix(ucode_prefix, ucode_handler);
+                               ucode_prefix = NULL;
+                               ucode_handler = NULL;
+                       }
+
+                       break;
+#else
+               case 'o':
+               case 'O':
+                       fprintf(stderr, "uhttpd: ucode support not compiled, "
+                                       "ignoring -%c\n", ch);
                        break;
 #endif
 #ifdef HAVE_UBUS
@@ -391,12 +554,22 @@ int main(int argc, char **argv)
                case 'U':
                        conf.ubus_socket = optarg;
                        break;
+
+               case 'X':
+                       conf.ubus_cors = 1;
+                       break;
+
+               case 'e':
+                       conf.events_retry = atoi(optarg);
+                       break;
 #else
                case 'a':
                case 'u':
                case 'U':
+               case 'X':
+               case 'e':
                        fprintf(stderr, "uhttpd: UBUS support not compiled, "
-                                       "ignoring -%c\n", opt);
+                                       "ignoring -%c\n", ch);
                        break;
 #endif
                default:
@@ -405,12 +578,6 @@ int main(int argc, char **argv)
        }
 
        uh_config_parse();
-       init_defaults_post();
-
-       if (!bound) {
-               fprintf(stderr, "Error: No sockets bound, unable to continue\n");
-               return 1;
-       }
 
        if (!conf.docroot) {
                if (!realpath(".", uh_buf)) {
@@ -420,6 +587,13 @@ int main(int argc, char **argv)
                conf.docroot = strdup(uh_buf);
        }
 
+       init_defaults_post();
+
+       if (!bound) {
+               fprintf(stderr, "Error: No sockets bound, unable to continue\n");
+               return 1;
+       }
+
 #ifdef HAVE_TLS
        if (n_tls) {
                if (!tls_crt || !tls_key) {
@@ -428,20 +602,28 @@ int main(int argc, char **argv)
                        return 1;
                }
 
-               if (uh_tls_init(tls_key, tls_crt))
+               if (uh_tls_init(tls_key, tls_crt, tls_ciphers))
                    return 1;
        }
 #endif
 
 #ifdef HAVE_LUA
-       if (conf.lua_handler || conf.lua_prefix) {
-               if (!conf.lua_handler || !conf.lua_prefix) {
-                       fprintf(stderr, "Need handler and prefix to enable Lua support\n");
-                       return 1;
-               }
-               if (uh_plugin_init("uhttpd_lua.so"))
-                       return 1;
+       if (lua_handler || lua_prefix) {
+               fprintf(stderr, "Need handler and prefix to enable Lua support\n");
+               return 1;
        }
+
+       if (!list_empty(&conf.lua_prefix) && uh_plugin_init("uhttpd_lua.so"))
+               return 1;
+#endif
+#ifdef HAVE_UCODE
+       if (ucode_handler || ucode_prefix) {
+               fprintf(stderr, "Need handler and prefix to enable ucode support\n");
+               return 1;
+       }
+
+       if (!list_empty(&conf.ucode_prefix) && uh_plugin_init("uhttpd_ucode.so"))
+               return 1;
 #endif
 #ifdef HAVE_UBUS
        if (conf.ubus_prefix && uh_plugin_init("uhttpd_ubus.so"))