file: patch process stdin to /dev/null

[project/rpcd.git] / main.c

diff --git a/main.c b/main.c
index 0bb06cea61089ebbbc02215e7d7887d035d80517..1db3241958f0d61ba01c59a5f6710b7d6839e58b 100644 (file)
--- a/main.c
+++ b/main.c
@@ -2,7 +2,7 @@
  * rpcd - UBUS RPC server
  *
  *   Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
- *   Copyright (C) 2013 Jo-Philipp Wich <jow@openwrt.org>
+ *   Copyright (C) 2013-2014 Jo-Philipp Wich <jow@openwrt.org>
  *
  * Permission to use, copy, modify, and/or distribute this software for any
  * purpose with or without fee is hereby granted, provided that the above
@@ -18,6 +18,7 @@
  */
 
 #include <unistd.h>
+#include <stdlib.h>
 
 #include <libubox/blobmsg_json.h>
 #include <libubus.h>
@@ -32,6 +33,8 @@
 static struct ubus_context *ctx;
 static bool respawn = false;
 
+int rpc_exec_timeout = RPC_EXEC_DEFAULT_TIMEOUT;
+
 static void
 handle_signal(int sig)
 {
@@ -64,18 +67,30 @@ int main(int argc, char **argv)
        const char *ubus_socket = NULL;
        int ch;
 
-       while ((ch = getopt(argc, argv, "s:")) != -1) {
+       while ((ch = getopt(argc, argv, "s:t:")) != -1) {
                switch (ch) {
                case 's':
                        ubus_socket = optarg;
                        break;
+
+               case 't':
+                       rpc_exec_timeout = 1000 * strtol(optarg, NULL, 0);
+                       break;
+
                default:
                        break;
                }
        }
 
-       if (stat("/var/run/rpcd", &s))
-               mkdir("/var/run/rpcd", 0700);
+       if (rpc_exec_timeout < 1000 || rpc_exec_timeout > 600000) {
+               fprintf(stderr, "Invalid execution timeout specified\n");
+               return -1;
+       }
+
+       if (stat(RPC_UCI_DIR_PREFIX, &s))
+               mkdir(RPC_UCI_DIR_PREFIX, 0700);
+
+       umask(0077);
 
        signal(SIGPIPE, SIG_IGN);
        signal(SIGHUP,  handle_signal);