Teach oneit to reboot on exit more reliably.

1 files changed, 6 insertions, 1 deletions

diff --git a/toys/oneit.c b/toys/oneit.c
index 341bb080..45935918 100644
--- a/toys/oneit.c
+++ b/toys/oneit.c
@@ -58,7 +58,12 @@ void oneit_main(void)
     // pid 1 just reaps zombies until it gets its child, then halts the system.
     while (pid!=wait(&i));
     sync();
-    reboot(toys.optflags ? RB_POWER_OFF : RB_AUTOBOOT);
+
+	// PID 1 can't call reboot() because it kills the task that calls it,
+	// which causes the kernel to panic before the actual reboot happens.
+	if (!vfork()) reboot((toys.optflags&1) ? RB_POWER_OFF : RB_AUTOBOOT);
+	sleep(5);
+	_exit(1);
   }
 
   // Redirect stdio to /dev/tty0, with new session ID, so ctrl-c works.