blob: e852b870ec17f69244fb493c85630e8fb8ee4bd3 [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>
Denis Vlasenko680b86a2008-01-24 02:28:00 +000016
Denis Vlasenko9725daa2008-09-11 09:54:23 +000017static void write_wtmp(void)
Eric Andersencc8ed391999-10-05 16:24:54 +000018{
Denis Vlasenko680b86a2008-01-24 02:28:00 +000019 struct utmp utmp;
Denis Vlasenko680b86a2008-01-24 02:28:00 +000020 struct utsname uts;
Denis Vlasenko680b86a2008-01-24 02:28:00 +000021 if (access(bb_path_wtmp_file, R_OK|W_OK) == -1) {
22 close(creat(bb_path_wtmp_file, 0664));
23 }
24 memset(&utmp, 0, sizeof(utmp));
Bernhard Reutner-Fischer62d85032008-06-01 10:10:22 +000025 utmp.ut_tv.tv_sec = time(NULL);
Denis Vlasenko680b86a2008-01-24 02:28:00 +000026 safe_strncpy(utmp.ut_user, "shutdown", UT_NAMESIZE);
27 utmp.ut_type = RUN_LVL;
28 safe_strncpy(utmp.ut_id, "~~", sizeof(utmp.ut_id));
29 safe_strncpy(utmp.ut_line, "~~", UT_LINESIZE);
30 if (uname(&uts) == 0)
31 safe_strncpy(utmp.ut_host, uts.release, sizeof(utmp.ut_host));
32 updwtmp(bb_path_wtmp_file, &utmp);
Denis Vlasenko680b86a2008-01-24 02:28:00 +000033
Denis Vlasenko9725daa2008-09-11 09:54:23 +000034}
35#else
36#define write_wtmp() ((void)0)
37#endif
38
39#ifndef RB_HALT_SYSTEM
40#define RB_HALT_SYSTEM RB_HALT
41#endif
42
43#ifndef RB_POWER_OFF
44#define RB_POWER_OFF RB_POWERDOWN
45#endif
46
47int halt_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
48int halt_main(int argc UNUSED_PARAM, char **argv)
49{
50 static const int magic[] = {
51 RB_HALT_SYSTEM,
52 RB_POWER_OFF,
Denis Vlasenkoea7c9b32008-09-25 10:39:10 +000053 RB_AUTOBOOT
Denis Vlasenko9725daa2008-09-11 09:54:23 +000054 };
55 static const smallint signals[] = { SIGUSR1, SIGUSR2, SIGTERM };
56
57 int delay = 0;
58 int which, flags, rc;
59
60 /* Figure out which applet we're running */
61 for (which = 0; "hpr"[which] != applet_name[0]; which++)
62 continue;
63
64 /* Parse and handle arguments */
65 opt_complementary = "d+"; /* -d N */
66 /* We support -w even if !ENABLE_FEATURE_WTMP, in order
67 * to not break scripts */
68 flags = getopt32(argv, "d:nfw", &delay);
69
70 sleep(delay);
71
72 write_wtmp();
73
Denis Vlasenko680b86a2008-01-24 02:28:00 +000074 if (flags & 8) /* -w */
Bernhard Reutner-Fischeref9876a2008-07-21 11:30:51 +000075 return EXIT_SUCCESS;
Bernhard Reutner-Fischerd93179f2008-09-01 15:24:52 +000076
Denis Vlasenko680b86a2008-01-24 02:28:00 +000077 if (!(flags & 2)) /* no -n */
78 sync();
Bernhard Reutner-Fischere15d7572006-06-02 20:56:16 +000079
Rob Landley64612912006-01-30 08:31:37 +000080 /* Perform action. */
Denis Vlasenko1e28f612008-08-03 18:43:45 +000081 rc = 1;
82 if (!(flags & 4)) { /* no -f */
83//TODO: I tend to think that signalling linuxrc is wrong
84// pity original author didn't comment on it...
Rob Landley64612912006-01-30 08:31:37 +000085 if (ENABLE_FEATURE_INITRD) {
Denis Vlasenko35fb5122006-11-01 09:16:49 +000086 pid_t *pidlist = find_pid_by_name("linuxrc");
87 if (pidlist[0] > 0)
88 rc = kill(pidlist[0], signals[which]);
89 if (ENABLE_FEATURE_CLEAN_UP)
90 free(pidlist);
Rob Landley64612912006-01-30 08:31:37 +000091 }
Denis Vlasenko35fb5122006-11-01 09:16:49 +000092 if (rc)
93 rc = kill(1, signals[which]);
94 } else
95 rc = reboot(magic[which]);
Rob Landley64612912006-01-30 08:31:37 +000096
Denis Vlasenko35fb5122006-11-01 09:16:49 +000097 if (rc)
98 bb_error_msg("no");
Rob Landley64612912006-01-30 08:31:37 +000099 return rc;
Eric Andersencc8ed391999-10-05 16:24:54 +0000100}