procd: stop service using SIGKILL if SIGTERM failed to do so

[project/procd.git] / state.c

diff --git a/state.c b/state.c
index cf0c81d..3b56bd0 100644 (file)
--- a/state.c
+++ b/state.c
@@ -12,18 +12,25 @@
  * GNU General Public License for more details.
  */
 
+#include <fcntl.h>
 #include <sys/reboot.h>
+#include <stdio.h>
 #include <stdlib.h>
 #include <unistd.h>
+#include <sys/types.h>
+#include <signal.h>
 
 #include "procd.h"
 #include "syslog.h"
-#include "hotplug.h"
+#include "plug/hotplug.h"
 #include "watchdog.h"
+#include "service/service.h"
+#include "utils/utils.h"
 
 enum {
        STATE_NONE = 0,
        STATE_EARLY,
+       STATE_UBUS,
        STATE_INIT,
        STATE_RUNNING,
        STATE_SHUTDOWN,
@@ -34,41 +41,137 @@ enum {
 static int state = STATE_NONE;
 static int reboot_event;
 
+static void set_stdio(const char* tty)
+{
+       if (chdir("/dev") ||
+           !freopen(tty, "r", stdin) ||
+           !freopen(tty, "w", stdout) ||
+           !freopen(tty, "w", stderr) ||
+           chdir("/"))
+               ERROR("failed to set stdio\n");
+       else
+               fcntl(STDERR_FILENO, F_SETFL, fcntl(STDERR_FILENO, F_GETFL) | O_NONBLOCK);
+}
+
+static void set_console(void)
+{
+       const char* tty;
+       char* split;
+       char line[ 20 ];
+       const char* try[] = { "tty0", "console", NULL }; /* Try the most common outputs */
+       int f, i = 0;
+
+       tty = get_cmdline_val("console",line,sizeof(line));
+       if (tty != NULL) {
+               split = strchr(tty, ',');
+               if ( split != NULL )
+                       *split = '\0';
+       } else {
+               // Try a default
+               tty=try[i];
+               i++;
+       }
+
+       if (chdir("/dev")) {
+               ERROR("failed to change dir to /dev\n");
+               return;
+       }
+       while (tty!=NULL) {
+               f = open(tty, O_RDONLY);
+               if (f >= 0) {
+                       close(f);
+                       break;
+               }
+
+               tty=try[i];
+               i++;
+       }
+       if (chdir("/"))
+               ERROR("failed to change dir to /\n");
+
+       if (tty != NULL)
+               set_stdio(tty);
+}
+
 static void state_enter(void)
 {
+       char ubus_cmd[] = "/sbin/ubusd";
 
        switch (state) {
        case STATE_EARLY:
                LOG("- early -\n");
-               watchdog_init();
+               watchdog_init(0);
                hotplug("/etc/hotplug.json");
                procd_coldplug();
                break;
 
+       case STATE_UBUS:
+               // try to reopen incase the wdt was not available before coldplug
+               watchdog_init(0);
+               set_stdio("console");
+               LOG("- ubus -\n");
+               procd_connect_ubus();
+               service_start_early("ubus", ubus_cmd);
+               break;
+
        case STATE_INIT:
                LOG("- init -\n");
-               log_init();
-               procd_connect_ubus();
                procd_inittab();
                procd_inittab_run("respawn");
                procd_inittab_run("askconsole");
                procd_inittab_run("askfirst");
                procd_inittab_run("sysinit");
+
+               // switch to syslog log channel
+               ulog_open(ULOG_SYSLOG, LOG_DAEMON, "procd");
                break;
 
        case STATE_RUNNING:
                LOG("- init complete -\n");
+               procd_inittab_run("respawnlate");
+               procd_inittab_run("askconsolelate");
                break;
 
        case STATE_SHUTDOWN:
+               /* Redirect output to the console for the users' benefit */
+               set_console();
                LOG("- shutdown -\n");
                procd_inittab_run("shutdown");
                sync();
                break;
 
        case STATE_HALT:
-               LOG("- reboot -\n");
-               reboot(reboot_event);
+               // To prevent killed processes from interrupting the sleep
+               signal(SIGCHLD, SIG_IGN);
+               LOG("- SIGTERM processes -\n");
+               kill(-1, SIGTERM);
+               sync();
+               sleep(1);
+               LOG("- SIGKILL processes -\n");
+               kill(-1, SIGKILL);
+               sync();
+               sleep(1);
+#ifndef DISABLE_INIT
+               if (reboot_event == RB_POWER_OFF)
+                       LOG("- power down -\n");
+               else
+                       LOG("- reboot -\n");
+
+               /* Allow time for last message to reach serial console, etc */
+               sleep(1);
+
+               /* We have to fork here, since the kernel calls do_exit(EXIT_SUCCESS)
+                * in linux/kernel/sys.c, which can cause the machine to panic when
+                * the init process exits... */
+               if (!vfork( )) { /* child */
+                       reboot(reboot_event);
+                       _exit(EXIT_SUCCESS);
+               }
+               while (1)
+                       sleep(1);
+#else
+               exit(0);
+#endif
                break;
 
        default:
@@ -79,14 +182,22 @@ static void state_enter(void)
 
 void procd_state_next(void)
 {
-       DEBUG(2, "Change state %d -> %d\n", state, state + 1);
+       DEBUG(4, "Change state %d -> %d\n", state, state + 1);
        state++;
        state_enter();
 }
 
+void procd_state_ubus_connect(void)
+{
+       if (state == STATE_UBUS)
+               procd_state_next();
+}
+
 void procd_shutdown(int event)
 {
-       DEBUG(1, "Shutting down system with event %x\n", reboot_event);
+       if (state >= STATE_SHUTDOWN)
+               return;
+       DEBUG(2, "Shutting down system with event %x\n", event);
        reboot_event = event;
        state = STATE_SHUTDOWN;
        state_enter();