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