summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorJohn Crispin <blogic@openwrt.org>2013-03-08 23:48:01 +0100
committerJohn Crispin <blogic@openwrt.org>2013-03-13 18:58:13 +0100
commitf1bf99b09f664dce09f55e7102ce8e6ef77ef10a (patch)
treea38df7e2f561e5ea9cddb8c8efa798e987041fd6
parenta89b1a3883b2ee3552334b70e4465ff9426bab2e (diff)
downloadunitd-f1bf99b09f664dce09f55e7102ce8e6ef77ef10a.tar
unitd-f1bf99b09f664dce09f55e7102ce8e6ef77ef10a.zip
add signal handler
-rw-r--r--signal.c88
1 files changed, 88 insertions, 0 deletions
diff --git a/signal.c b/signal.c
new file mode 100644
index 0000000..a4b08c3
--- /dev/null
+++ b/signal.c
@@ -0,0 +1,88 @@
+/*
+ * Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
+ * Copyright (C) 2013 John Crispin <blogic@openwrt.org>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License version 2.1
+ * as published by the Free Software Foundation
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ */
+
+#include <sys/reboot.h>
+#include <sys/types.h>
+
+#include <unistd.h>
+
+#include "procd.h"
+
+static void signal_shutdown(int signal, siginfo_t *siginfo, void *data)
+{
+ int event = 0;
+ char *msg = NULL;
+
+ switch(signal) {
+ case SIGTERM:
+ event = RB_AUTOBOOT;
+ msg = "reboot";
+ break;
+ case SIGUSR1:
+ case SIGUSR2:
+ event = RB_POWER_OFF;
+ msg = "poweroff";
+ break;
+ }
+
+ DEBUG(1, "Triggering %s\n", msg);
+ if (event)
+ procd_shutdown(event);
+}
+
+struct sigaction sa_shutdown = {
+ .sa_sigaction = signal_shutdown,
+ .sa_flags = SA_SIGINFO
+};
+
+static void signal_crash(int signal, siginfo_t *siginfo, void *data)
+{
+ ERROR("Rebooting as procd has crashed\n");
+ fflush(stderr);
+ sync();
+ sleep(1);
+ reboot(RB_AUTOBOOT);
+ while (1)
+ ;
+}
+
+struct sigaction sa_crash = {
+ .sa_sigaction = signal_crash,
+ .sa_flags = SA_SIGINFO
+};
+
+static void signal_dummy(int signal, siginfo_t *siginfo, void *data)
+{
+ ERROR("Got unexpected signal %d\n", signal);
+}
+
+struct sigaction sa_dummy = {
+ .sa_sigaction = signal_dummy,
+ .sa_flags = SA_SIGINFO
+};
+
+void procd_signal(void)
+{
+ sigaction(SIGPIPE, &sa_dummy, NULL);
+ if (getpid() != 1)
+ return;
+ sigaction(SIGTERM, &sa_shutdown, NULL);
+ sigaction(SIGUSR1, &sa_shutdown, NULL);
+ sigaction(SIGUSR2, &sa_shutdown, NULL);
+ sigaction(SIGSEGV, &sa_crash, NULL);
+ sigaction(SIGBUS, &sa_crash, NULL);
+ sigaction(SIGHUP, &sa_dummy, NULL);
+ sigaction(SIGKILL, &sa_dummy, NULL);
+ sigaction(SIGSTOP, &sa_dummy, NULL);
+}