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