Add/improve comment, shorten some usage texts. No code changes.

[oweals/busybox.git] / init / halt.c

diff --git a/init/halt.c b/init/halt.c
index 0dbdc55f1439a3a2ce5c2342ee3fc99987408121..d5c22a8ac5f72572b5d7e2f13505cdea9f0e16cd 100644 (file)
--- a/init/halt.c
+++ b/init/halt.c
@@ -1,9 +1,59 @@
-#include "internal.h"
-#include <signal.h>
+/* vi: set sw=4 ts=4: */
+/*
+ * Poweroff reboot and halt, oh my.
+ *
+ * Copyright 2006 by Rob Landley <rob@landley.net>
+ *
+ * Licensed under GPL version 2, see file LICENSE in this tarball for details.
+ */
 
-extern int
-halt_main(int argc, char ** argv)
+#include "busybox.h"
+#include <sys/reboot.h>
+
+int halt_main(int argc, char *argv[]);
+int halt_main(int argc, char *argv[])
 {
-    exit( kill(1, SIGUSR1));
-}
+       static const int magic[] = {
+#ifdef RB_HALT_SYSTEM
+RB_HALT_SYSTEM,
+#elif defined RB_HALT
+RB_HALT,
+#endif
+#ifdef RB_POWER_OFF
+RB_POWER_OFF,
+#elif defined RB_POWERDOWN
+RB_POWERDOWN,
+#endif
+RB_AUTOBOOT
+       };
+       static const int signals[] = { SIGUSR1, SIGUSR2, SIGTERM };
+
+       char *delay;
+       int which, flags, rc = 1;
+
+       /* Figure out which applet we're running */
+       for (which = 0; "hpr"[which] != *applet_name; which++);
 
+       /* Parse and handle arguments */
+       flags = getopt32(argc, argv, "d:nf", &delay);
+       if (flags & 1) sleep(xatou(delay));
+       if (!(flags & 2)) sync();
+
+       /* Perform action. */
+       if (ENABLE_INIT && !(flags & 4)) {
+               if (ENABLE_FEATURE_INITRD) {
+                       pid_t *pidlist = find_pid_by_name("linuxrc");
+                       if (pidlist[0] > 0)
+                               rc = kill(pidlist[0], signals[which]);
+                       if (ENABLE_FEATURE_CLEAN_UP)
+                               free(pidlist);
+               }
+               if (rc)
+                       rc = kill(1, signals[which]);
+       } else
+               rc = reboot(magic[which]);
+
+       if (rc)
+               bb_error_msg("no");
+       return rc;
+}