trigger: fix memory leak in script calls
[project/procd.git] / signal.c
1 /*
2  * Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
3  * Copyright (C) 2013 John Crispin <blogic@openwrt.org>
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU Lesser General Public License version 2.1
7  * as published by the Free Software Foundation
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  */
14
15 #include <sys/reboot.h>
16 #include <sys/types.h>
17
18 #include <unistd.h>
19
20 #include "procd.h"
21
22 static void do_reboot(void)
23 {
24         LOG("reboot\n");
25         fflush(stderr);
26         sync();
27         sleep(2);
28         reboot(RB_AUTOBOOT);
29         while (1)
30         ;
31 }
32
33 static void signal_shutdown(int signal, siginfo_t *siginfo, void *data)
34 {
35         int event = 0;
36         char *msg = NULL;
37
38         switch(signal) {
39         case SIGINT:
40         case SIGTERM:
41                 event = RB_AUTOBOOT;
42                 msg = "reboot";
43                 break;
44         case SIGUSR1:
45         case SIGUSR2:
46                 event = RB_POWER_OFF;
47                 msg = "poweroff";
48                 break;
49         }
50
51         DEBUG(1, "Triggering %s\n", msg);
52         if (event)
53                 procd_shutdown(event);
54 }
55
56 struct sigaction sa_shutdown = {
57         .sa_sigaction = signal_shutdown,
58         .sa_flags = SA_SIGINFO
59 };
60
61 static void signal_crash(int signal, siginfo_t *siginfo, void *data)
62 {
63         ERROR("Rebooting as procd has crashed\n");
64         do_reboot();
65 }
66
67 struct sigaction sa_crash = {
68         .sa_sigaction = signal_crash,
69         .sa_flags = SA_SIGINFO
70 };
71
72 static void signal_dummy(int signal, siginfo_t *siginfo, void *data)
73 {
74         ERROR("Got unexpected signal %d\n", signal);
75 }
76
77 struct sigaction sa_dummy = {
78         .sa_sigaction = signal_dummy,
79         .sa_flags = SA_SIGINFO
80 };
81
82 void procd_signal(void)
83 {
84         signal(SIGPIPE, SIG_IGN);
85         if (getpid() != 1)
86                 return;
87         sigaction(SIGTERM, &sa_shutdown, NULL);
88         sigaction(SIGINT, &sa_shutdown, NULL);
89         sigaction(SIGUSR1, &sa_shutdown, NULL);
90         sigaction(SIGUSR2, &sa_shutdown, NULL);
91         sigaction(SIGSEGV, &sa_crash, NULL);
92         sigaction(SIGBUS, &sa_crash, NULL);
93         sigaction(SIGHUP, &sa_dummy, NULL);
94         sigaction(SIGKILL, &sa_dummy, NULL);
95         sigaction(SIGSTOP, &sa_dummy, NULL);
96         reboot(RB_DISABLE_CAD);
97 }