blob: 8c1f30b08ca87ad72fee07564c33d765e25b7690 [file] [log] [blame]
Erik Andersene49d5ec2000-02-08 19:58:47 +00001/* vi: set sw=4 ts=4: */
Eric Andersenc4996011999-10-20 22:08:37 +00002/*
Rob Landley64612912006-01-30 08:31:37 +00003 * Poweroff reboot and halt, oh my.
Eric Andersenc4996011999-10-20 22:08:37 +00004 *
Rob Landley64612912006-01-30 08:31:37 +00005 * Copyright 2006 by Rob Landley <rob@landley.net>
Eric Andersenc4996011999-10-20 22:08:37 +00006 *
Rob Landleye9a7a622006-09-22 02:52:41 +00007 * Licensed under GPL version 2, see file LICENSE in this tarball for details.
Eric Andersenc4996011999-10-20 22:08:37 +00008 */
9
Denis Vlasenkob6adbf12007-05-26 19:00:18 +000010#include "libbb.h"
Eric Andersen02462222003-07-22 09:41:39 +000011#include <sys/reboot.h>
Eric Andersencc8ed391999-10-05 16:24:54 +000012
Denis Vlasenko680b86a2008-01-24 02:28:00 +000013#if ENABLE_FEATURE_WTMP
14#include <sys/utsname.h>
15#include <utmp.h>
16#endif
17
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +000018int halt_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +000019int halt_main(int argc UNUSED_PARAM, char **argv)
Eric Andersencc8ed391999-10-05 16:24:54 +000020{
Bernhard Reutner-Fischerd765ee52006-05-26 20:34:02 +000021 static const int magic[] = {
22#ifdef RB_HALT_SYSTEM
23RB_HALT_SYSTEM,
24#elif defined RB_HALT
25RB_HALT,
26#endif
27#ifdef RB_POWER_OFF
28RB_POWER_OFF,
29#elif defined RB_POWERDOWN
30RB_POWERDOWN,
31#endif
32RB_AUTOBOOT
33 };
Denis Vlasenko680b86a2008-01-24 02:28:00 +000034 static const smallint signals[] = { SIGUSR1, SIGUSR2, SIGTERM };
Rob Landley0a7c8ef2006-02-22 17:01:00 +000035
Denis Vlasenko1d426652008-03-17 09:09:09 +000036 int delay = 0;
Denis Vlasenko1e28f612008-08-03 18:43:45 +000037 int which, flags, rc;
Denis Vlasenko680b86a2008-01-24 02:28:00 +000038#if ENABLE_FEATURE_WTMP
39 struct utmp utmp;
Denis Vlasenko680b86a2008-01-24 02:28:00 +000040 struct utsname uts;
41#endif
Eric Andersen02462222003-07-22 09:41:39 +000042
Rob Landley64612912006-01-30 08:31:37 +000043 /* Figure out which applet we're running */
Denis Vlasenko680b86a2008-01-24 02:28:00 +000044 for (which = 0; "hpr"[which] != *applet_name; which++)
45 continue;
Eric Andersen02462222003-07-22 09:41:39 +000046
Rob Landley64612912006-01-30 08:31:37 +000047 /* Parse and handle arguments */
Denis Vlasenko1d426652008-03-17 09:09:09 +000048 opt_complementary = "d+"; /* -d N */
Denis Vlasenko680b86a2008-01-24 02:28:00 +000049 flags = getopt32(argv, "d:nfw", &delay);
Denis Vlasenko1d426652008-03-17 09:09:09 +000050
51 sleep(delay);
Denis Vlasenko680b86a2008-01-24 02:28:00 +000052
53#if ENABLE_FEATURE_WTMP
54 if (access(bb_path_wtmp_file, R_OK|W_OK) == -1) {
55 close(creat(bb_path_wtmp_file, 0664));
56 }
57 memset(&utmp, 0, sizeof(utmp));
Bernhard Reutner-Fischer62d85032008-06-01 10:10:22 +000058 utmp.ut_tv.tv_sec = time(NULL);
Denis Vlasenko680b86a2008-01-24 02:28:00 +000059 safe_strncpy(utmp.ut_user, "shutdown", UT_NAMESIZE);
60 utmp.ut_type = RUN_LVL;
61 safe_strncpy(utmp.ut_id, "~~", sizeof(utmp.ut_id));
62 safe_strncpy(utmp.ut_line, "~~", UT_LINESIZE);
63 if (uname(&uts) == 0)
64 safe_strncpy(utmp.ut_host, uts.release, sizeof(utmp.ut_host));
65 updwtmp(bb_path_wtmp_file, &utmp);
66#endif /* !ENABLE_FEATURE_WTMP */
67
68 if (flags & 8) /* -w */
Bernhard Reutner-Fischeref9876a2008-07-21 11:30:51 +000069 return EXIT_SUCCESS;
Denis Vlasenko680b86a2008-01-24 02:28:00 +000070 if (!(flags & 2)) /* no -n */
71 sync();
Bernhard Reutner-Fischere15d7572006-06-02 20:56:16 +000072
Rob Landley64612912006-01-30 08:31:37 +000073 /* Perform action. */
Denis Vlasenko1e28f612008-08-03 18:43:45 +000074 rc = 1;
75 if (!(flags & 4)) { /* no -f */
76//TODO: I tend to think that signalling linuxrc is wrong
77// pity original author didn't comment on it...
Rob Landley64612912006-01-30 08:31:37 +000078 if (ENABLE_FEATURE_INITRD) {
Denis Vlasenko35fb5122006-11-01 09:16:49 +000079 pid_t *pidlist = find_pid_by_name("linuxrc");
80 if (pidlist[0] > 0)
81 rc = kill(pidlist[0], signals[which]);
82 if (ENABLE_FEATURE_CLEAN_UP)
83 free(pidlist);
Rob Landley64612912006-01-30 08:31:37 +000084 }
Denis Vlasenko35fb5122006-11-01 09:16:49 +000085 if (rc)
86 rc = kill(1, signals[which]);
87 } else
88 rc = reboot(magic[which]);
Rob Landley64612912006-01-30 08:31:37 +000089
Denis Vlasenko35fb5122006-11-01 09:16:49 +000090 if (rc)
91 bb_error_msg("no");
Rob Landley64612912006-01-30 08:31:37 +000092 return rc;
Eric Andersencc8ed391999-10-05 16:24:54 +000093}