blob: 7f1bebb9930a263cf7b3118790c43491f061c314
1 | /* vi: set sw=4 ts=4: */ |
2 | /* |
3 | * Poweroff reboot and halt, oh my. |
4 | * |
5 | * Copyright 2006 by Rob Landley <rob@landley.net> |
6 | * |
7 | * Licensed under GPLv2, see file LICENSE in this source tree. |
8 | */ |
9 | |
10 | //applet:IF_HALT(APPLET(halt, BB_DIR_SBIN, BB_SUID_DROP)) |
11 | //applet:IF_HALT(APPLET_ODDNAME(poweroff, halt, BB_DIR_SBIN, BB_SUID_DROP, poweroff)) |
12 | //applet:IF_HALT(APPLET_ODDNAME(reboot, halt, BB_DIR_SBIN, BB_SUID_DROP, reboot)) |
13 | |
14 | //kbuild:lib-$(CONFIG_HALT) += halt.o |
15 | |
16 | //config:config HALT |
17 | //config: bool "poweroff, halt, and reboot" |
18 | //config: default y |
19 | //config: help |
20 | //config: Stop all processes and either halt, reboot, or power off the system. |
21 | //config: |
22 | //config:config FEATURE_CALL_TELINIT |
23 | //config: bool "Call telinit on shutdown and reboot" |
24 | //config: default y |
25 | //config: depends on HALT && !INIT |
26 | //config: help |
27 | //config: Call an external program (normally telinit) to facilitate |
28 | //config: a switch to a proper runlevel. |
29 | //config: |
30 | //config: This option is only available if you selected halt and friends, |
31 | //config: but did not select init. |
32 | //config: |
33 | //config:config TELINIT_PATH |
34 | //config: string "Path to telinit executable" |
35 | //config: default "/sbin/telinit" |
36 | //config: depends on FEATURE_CALL_TELINIT |
37 | //config: help |
38 | //config: When busybox halt and friends have to call external telinit |
39 | //config: to facilitate proper shutdown, this path is to be used when |
40 | //config: locating telinit executable. |
41 | |
42 | //usage:#define halt_trivial_usage |
43 | //usage: "[-d DELAY] [-n] [-f]" IF_FEATURE_WTMP(" [-w]") |
44 | //usage:#define halt_full_usage "\n\n" |
45 | //usage: "Halt the system\n" |
46 | //usage: "\n -d SEC Delay interval" |
47 | //usage: "\n -n Do not sync" |
48 | //usage: "\n -f Force (don't go through init)" |
49 | //usage: IF_FEATURE_WTMP( |
50 | //usage: "\n -w Only write a wtmp record" |
51 | //usage: ) |
52 | //usage: |
53 | //usage:#define poweroff_trivial_usage |
54 | //usage: "[-d DELAY] [-n] [-f]" |
55 | //usage:#define poweroff_full_usage "\n\n" |
56 | //usage: "Halt and shut off power\n" |
57 | //usage: "\n -d SEC Delay interval" |
58 | //usage: "\n -n Do not sync" |
59 | //usage: "\n -f Force (don't go through init)" |
60 | //usage: |
61 | //usage:#define reboot_trivial_usage |
62 | //usage: "[-d DELAY] [-n] [-f]" |
63 | //usage:#define reboot_full_usage "\n\n" |
64 | //usage: "Reboot the system\n" |
65 | //usage: "\n -d SEC Delay interval" |
66 | //usage: "\n -n Do not sync" |
67 | //usage: "\n -f Force (don't go through init)" |
68 | |
69 | #include "libbb.h" |
70 | #include "reboot.h" |
71 | |
72 | #ifdef __BIONIC__ |
73 | # include "android/reboot.c" |
74 | #endif |
75 | |
76 | #if ENABLE_FEATURE_WTMP |
77 | #include <sys/utsname.h> |
78 | |
79 | static void write_wtmp(void) |
80 | { |
81 | struct utmp utmp; |
82 | struct utsname uts; |
83 | /* "man utmp" says wtmp file should *not* be created automagically */ |
84 | /*if (access(bb_path_wtmp_file, R_OK|W_OK) == -1) { |
85 | close(creat(bb_path_wtmp_file, 0664)); |
86 | }*/ |
87 | memset(&utmp, 0, sizeof(utmp)); |
88 | utmp.ut_tv.tv_sec = time(NULL); |
89 | strcpy(utmp.ut_user, "shutdown"); /* it is wide enough */ |
90 | utmp.ut_type = RUN_LVL; |
91 | utmp.ut_id[0] = '~'; utmp.ut_id[1] = '~'; /* = strcpy(utmp.ut_id, "~~"); */ |
92 | utmp.ut_line[0] = '~'; utmp.ut_line[1] = '~'; /* = strcpy(utmp.ut_line, "~~"); */ |
93 | uname(&uts); |
94 | safe_strncpy(utmp.ut_host, uts.release, sizeof(utmp.ut_host)); |
95 | updwtmp(bb_path_wtmp_file, &utmp); |
96 | } |
97 | #else |
98 | #define write_wtmp() ((void)0) |
99 | #endif |
100 | |
101 | |
102 | int halt_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
103 | int halt_main(int argc UNUSED_PARAM, char **argv) |
104 | { |
105 | static const int magic[] = { |
106 | RB_HALT_SYSTEM, |
107 | RB_POWER_OFF, |
108 | RB_AUTOBOOT |
109 | }; |
110 | static const smallint signals[] = { SIGUSR1, SIGUSR2, SIGTERM }; |
111 | |
112 | int delay = 0; |
113 | int which, flags, rc; |
114 | |
115 | /* Figure out which applet we're running */ |
116 | for (which = 0; "hpr"[which] != applet_name[0]; which++) |
117 | continue; |
118 | |
119 | /* Parse and handle arguments */ |
120 | opt_complementary = "d+"; /* -d N */ |
121 | /* We support -w even if !ENABLE_FEATURE_WTMP, |
122 | * in order to not break scripts. |
123 | * -i (shut down network interfaces) is ignored. |
124 | */ |
125 | flags = getopt32(argv, "d:nfwi", &delay); |
126 | |
127 | sleep(delay); |
128 | |
129 | write_wtmp(); |
130 | |
131 | if (flags & 8) /* -w */ |
132 | return EXIT_SUCCESS; |
133 | |
134 | if (!(flags & 2)) /* no -n */ |
135 | sync(); |
136 | |
137 | #ifdef __BIONIC__ |
138 | char *mode[4]; |
139 | int c = 1; |
140 | mode[0] = strdup("reboot"); |
141 | mode[1] = mode[2] = mode[3] = NULL; |
142 | switch (which) { |
143 | case 0: |
144 | case 1: |
145 | //-p for halt |
146 | mode[1] = strdup("-p"); |
147 | c=2; |
148 | break; |
149 | case 2: |
150 | //reboot |
151 | #ifdef CYANOGEN_LIBREBOOT |
152 | if (argc > 1) { |
153 | mode[1] = strdup(argv[1]); |
154 | c = 2; |
155 | } |
156 | #endif |
157 | break; |
158 | } |
159 | return reboot_main(c, mode); |
160 | #endif |
161 | |
162 | /* Perform action. */ |
163 | rc = 1; |
164 | if (!(flags & 4)) { /* no -f */ |
165 | //TODO: I tend to think that signalling linuxrc is wrong |
166 | // pity original author didn't comment on it... |
167 | if (ENABLE_FEATURE_INITRD) { |
168 | /* talk to linuxrc */ |
169 | /* bbox init/linuxrc assumed */ |
170 | pid_t *pidlist = find_pid_by_name("linuxrc"); |
171 | if (pidlist[0] > 0) |
172 | rc = kill(pidlist[0], signals[which]); |
173 | if (ENABLE_FEATURE_CLEAN_UP) |
174 | free(pidlist); |
175 | } |
176 | if (rc) { |
177 | /* talk to init */ |
178 | if (!ENABLE_FEATURE_CALL_TELINIT) { |
179 | /* bbox init assumed */ |
180 | rc = kill(1, signals[which]); |
181 | } else { |
182 | /* SysV style init assumed */ |
183 | /* runlevels: |
184 | * 0 == shutdown |
185 | * 6 == reboot */ |
186 | execlp(CONFIG_TELINIT_PATH, |
187 | CONFIG_TELINIT_PATH, |
188 | which == 2 ? "6" : "0", |
189 | (char *)NULL |
190 | ); |
191 | bb_perror_msg_and_die("can't execute '%s'", |
192 | CONFIG_TELINIT_PATH); |
193 | } |
194 | } |
195 | } else { |
196 | rc = reboot(magic[which]); |
197 | } |
198 | |
199 | if (rc) |
200 | bb_perror_nomsg_and_die(); |
201 | return rc; |
202 | } |
203 |