/* 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. */ #include "libbb.h" #include <sys/reboot.h> #if ENABLE_FEATURE_WTMP #include <sys/utsname.h> #include <utmp.h> static void write_wtmp(void) { struct utmp utmp; struct utsname uts; 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); } #else #define write_wtmp() ((void)0) #endif #ifndef RB_HALT_SYSTEM #define RB_HALT_SYSTEM RB_HALT #endif #ifndef RB_POWER_OFF #define RB_POWER_OFF RB_POWERDOWN #endif int halt_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; int halt_main(int argc UNUSED_PARAM, char **argv) { static const int magic[] = { RB_HALT_SYSTEM, RB_POWER_OFF, RB_AUTOBOOT }; static const smallint signals[] = { SIGUSR1, SIGUSR2, SIGTERM }; int delay = 0; int which, flags, rc; /* Figure out which applet we're running */ for (which = 0; "hpr"[which] != applet_name[0]; which++) continue; /* Parse and handle arguments */ opt_complementary = "d+"; /* -d N */ /* We support -w even if !ENABLE_FEATURE_WTMP, in order * to not break scripts */ flags = getopt32(argv, "d:nfw", &delay); sleep(delay); write_wtmp(); if (flags & 8) /* -w */ return EXIT_SUCCESS; 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_perror_nomsg_and_die(); return rc; }