getopt_ulflags -> getopt32.

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

diff --git a/init/halt.c b/init/halt.c
index 10f16c75da5002b18d65f9f009d6ef33a335b04e..d796a4aea6f0053e32c40c091e2d0c40c807463f 100644 (file)
--- a/init/halt.c
+++ b/init/halt.c
@@ -1,32 +1,53 @@
 /* vi: set sw=4 ts=4: */
 /*
- * Mini halt implementation for busybox
+ * Poweroff reboot and halt, oh my.
  *
- * Copyright (C) 1995, 1996 by Bruce Perens <bruce@pixar.com>.
- * Copyright (C) 1999-2003 by Erik Andersen <andersen@codepoet.org>
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * 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.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ * Copyright 2006 by Rob Landley <rob@landley.net>
  *
+ * Licensed under GPL version 2, see file LICENSE in this tarball for details.
  */
 
 #include "busybox.h"
-#include <signal.h>
-#include "init_shared.h"
+#include <sys/reboot.h>
 
-
-extern int halt_main(int argc, char **argv)
+int halt_main(int argc, char *argv[])
 {
-       return kill_init(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 = "hpr";
+       int which, flags, rc = 1;
+
+       /* Figure out which applet we're running */
+       for(which=0;delay[which]!=*bb_applet_name;which++);
+
+       /* Parse and handle arguments */
+       flags = getopt32(argc, argv, "d:nf", &delay);
+       if (flags&1) sleep(atoi(delay));
+       if (!(flags&2)) sync();
+
+       /* Perform action. */
+       if (ENABLE_INIT && !(flags & 4)) {
+               if (ENABLE_FEATURE_INITRD) {
+                       long *pidlist=find_pid_by_name("linuxrc");
+                       if (*pidlist>0) rc = kill(*pidlist,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;
 }