main.c

   1 /*
   2  * rpcd - UBUS RPC server
   3  *
   4  *   Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
   5  *   Copyright (C) 2013 Jo-Philipp Wich <jow@openwrt.org>
   6  *
   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.
  10  *
  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.
  18  */
  19
  20 #include <unistd.h>
  21
  22 #include <libubox/blobmsg_json.h>
  23 #include <libubus.h>
  24 #include <signal.h>
  25
  26 #include <rpcd/session.h>
  27 #include <rpcd/uci.h>
  28 #include <rpcd/plugin.h>
  29 #include <rpcd/exec.h>
  30
  31 static struct ubus_context *ctx;
  32 static bool respawn = false;
  33
  34 static void
  35 handle_signal(int sig)
  36 {
  37         rpc_session_freeze();
  38         uloop_cancelled = true;
  39         respawn = (sig == SIGHUP);
  40 }
  41
  42 static void
  43 exec_self(int argc, char **argv)
  44 {
  45         int i;
  46         const char *cmd = rpc_exec_lookup(argv[0]);
  47         char **args = calloc(argc + 1, sizeof(char *));
  48
  49         if (!cmd || !args)
  50                 return;
  51
  52         for (i = 0; i < argc; i++)
  53                 args[i] = argv[i];
  54
  55         setenv("RPC_HANGUP", "1", 1);
  56         execv(cmd, (char * const *)args);
  57 }
  58
  59 int main(int argc, char **argv)
  60 {
  61         const char *hangup;
  62         const char *ubus_socket = NULL;
  63         int ch;
  64
  65         while ((ch = getopt(argc, argv, "s:")) != -1) {
  66                 switch (ch) {
  67                 case 's':
  68                         ubus_socket = optarg;
  69                         break;
  70                 default:
  71                         break;
  72                 }
  73         }
  74
  75         signal(SIGPIPE, SIG_IGN);
  76         signal(SIGHUP,  handle_signal);
  77         signal(SIGUSR1, handle_signal);
  78
  79         uloop_init();
  80
  81         ctx = ubus_connect(ubus_socket);
  82         if (!ctx) {
  83                 fprintf(stderr, "Failed to connect to ubus\n");
  84                 return -1;
  85         }
  86
  87         ubus_add_uloop(ctx);
  88
  89         rpc_session_api_init(ctx);
  90         rpc_uci_api_init(ctx);
  91         rpc_plugin_api_init(ctx);
  92
  93         hangup = getenv("RPC_HANGUP");
  94
  95         if (!hangup || strcmp(hangup, "1"))
  96                 rpc_uci_purge_savedirs();
  97         else
  98                 rpc_session_thaw();
  99
 100         uloop_run();
 101         ubus_free(ctx);
 102         uloop_done();
 103
 104         if (respawn)
 105                 exec_self(argc, argv);
 106
 107         return 0;
 108 }