/* vi: set sw=4 ts=4: */
/*
- * Mini halt implementation for busybox
+ * Poweroff reboot and halt, oh my.
*
+ * Copyright 2006 by Rob Landley <rob@landley.net>
*
- * Copyright (C) 1995, 1996 by Bruce Perens <bruce@pixar.com>.
- *
- * 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
- *
+ * Licensed under GPL version 2, see file LICENSE in this tarball for details.
*/
-#include "busybox.h"
-#include <signal.h>
+#include "libbb.h"
+#include <sys/reboot.h>
-extern int halt_main(int argc, char **argv)
+#if ENABLE_FEATURE_WTMP
+#include <sys/utsname.h>
+#include <utmp.h>
+#endif
+
+int halt_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int halt_main(int argc UNUSED_PARAM, char **argv)
{
-#ifdef BB_FEATURE_LINUXRC
- /* don't assume init's pid == 1 */
- return(kill(*(find_pid_by_name("init")), SIGUSR1));
-#else
- return(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 smallint signals[] = { SIGUSR1, SIGUSR2, SIGTERM };
+
+ int delay = 0;
+ int which, flags, rc;
+#if ENABLE_FEATURE_WTMP
+ struct utmp utmp;
+ struct utsname uts;
#endif
+
+ /* Figure out which applet we're running */
+ for (which = 0; "hpr"[which] != *applet_name; which++)
+ continue;
+
+ /* Parse and handle arguments */
+ opt_complementary = "d+"; /* -d N */
+ flags = getopt32(argv, "d:nf" USE_FEATURE_WTMP("w"), &delay);
+
+ sleep(delay);
+
+#if ENABLE_FEATURE_WTMP
+ if (access(bb_path_wtmp_file, R_OK|W_OK) == -1) {
+ close(creat(bb_path_wtmp_file, 0664));
+ }
+ memset(&utmp, 0, sizeof(utmp));
+ utmp.ut_tv.tv_sec = time(NULL);
+ safe_strncpy(utmp.ut_user, "shutdown", UT_NAMESIZE);
+ utmp.ut_type = RUN_LVL;
+ safe_strncpy(utmp.ut_id, "~~", sizeof(utmp.ut_id));
+ safe_strncpy(utmp.ut_line, "~~", UT_LINESIZE);
+ if (uname(&uts) == 0)
+ safe_strncpy(utmp.ut_host, uts.release, sizeof(utmp.ut_host));
+ updwtmp(bb_path_wtmp_file, &utmp);
+
+ if (flags & 8) /* -w */
+ return EXIT_SUCCESS;
+#endif /* !ENABLE_FEATURE_WTMP */
+
+ if (!(flags & 2)) /* no -n */
+ sync();
+
+ /* Perform action. */
+ rc = 1;
+ if (!(flags & 4)) { /* no -f */
+//TODO: I tend to think that signalling linuxrc is wrong
+// pity original author didn't comment on it...
+ 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;
}