]>
Commit | Line | Data |
---|---|---|
1815bff5 A |
1 | /* |
2 | * Copyright (c) 1980, 1986, 1993 | |
20e66415 | 3 | * The Regents of the University of California. All rights reserved. |
1815bff5 A |
4 | * |
5 | * Redistribution and use in source and binary forms, with or without | |
6 | * modification, are permitted provided that the following conditions | |
7 | * are met: | |
8 | * 1. Redistributions of source code must retain the above copyright | |
9 | * notice, this list of conditions and the following disclaimer. | |
10 | * 2. Redistributions in binary form must reproduce the above copyright | |
11 | * notice, this list of conditions and the following disclaimer in the | |
12 | * documentation and/or other materials provided with the distribution. | |
13 | * 3. All advertising materials mentioning features or use of this software | |
14 | * must display the following acknowledgement: | |
20e66415 A |
15 | * This product includes software developed by the University of |
16 | * California, Berkeley and its contributors. | |
1815bff5 A |
17 | * 4. Neither the name of the University nor the names of its contributors |
18 | * may be used to endorse or promote products derived from this software | |
19 | * without specific prior written permission. | |
20 | * | |
21 | * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND | |
22 | * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
23 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | |
24 | * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE | |
25 | * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | |
26 | * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS | |
27 | * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) | |
28 | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT | |
29 | * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY | |
30 | * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF | |
31 | * SUCH DAMAGE. | |
32 | */ | |
33 | ||
20e66415 A |
34 | #ifndef lint |
35 | static const char copyright[] = | |
36 | "@(#) Copyright (c) 1980, 1986, 1993\n\ | |
37 | The Regents of the University of California. All rights reserved.\n"; | |
38 | #endif /* not lint */ | |
39 | ||
40 | #ifndef lint | |
41 | #if 0 | |
42 | static char sccsid[] = "@(#)reboot.c 8.1 (Berkeley) 6/5/93"; | |
43 | #endif | |
44 | static const char rcsid[] = | |
45 | "$FreeBSD: src/sbin/reboot/reboot.c,v 1.17 2002/10/06 16:24:36 thomas Exp $"; | |
46 | #endif /* not lint */ | |
47 | ||
1815bff5 | 48 | #include <sys/reboot.h> |
20e66415 A |
49 | #include <sys/types.h> |
50 | #include <sys/sysctl.h> | |
1815bff5 | 51 | #include <signal.h> |
20e66415 | 52 | #include <err.h> |
1815bff5 | 53 | #include <errno.h> |
20e66415 A |
54 | #include <fcntl.h> |
55 | #include <util.h> | |
56 | #include <pwd.h> | |
1815bff5 | 57 | #include <syslog.h> |
1815bff5 A |
58 | #include <stdio.h> |
59 | #include <stdlib.h> | |
60 | #include <string.h> | |
20e66415 | 61 | #include <unistd.h> |
1815bff5 | 62 | |
b180c247 A |
63 | #ifdef __APPLE__ |
64 | #include "kextmanager.h" | |
65 | #include <IOKit/kext/kextmanager_types.h> | |
66 | #include <mach/mach_port.h> // allocate | |
67 | #include <mach/mach.h> // task_self, etc | |
68 | #include <servers/bootstrap.h> // bootstrap | |
69 | #endif | |
70 | ||
20e66415 A |
71 | void usage(void); |
72 | u_int get_pageins(void); | |
b180c247 A |
73 | #ifdef __APPLE__ |
74 | int reserve_reboot(void); | |
75 | #endif | |
1815bff5 A |
76 | |
77 | int dohalt; | |
78 | ||
79 | int | |
20e66415 | 80 | main(int argc, char *argv[]) |
1815bff5 | 81 | { |
1815bff5 | 82 | struct passwd *pw; |
b180c247 | 83 | int ch, howto, i, fd, kflag, lflag, nflag, qflag, pflag, uflag, sverrno; |
20e66415 A |
84 | u_int pageins; |
85 | char *kernel, *p; | |
86 | const char *user; | |
1815bff5 | 87 | |
20e66415 | 88 | if (strstr((p = rindex(*argv, '/')) ? p + 1 : *argv, "halt")) { |
1815bff5 A |
89 | dohalt = 1; |
90 | howto = RB_HALT; | |
91 | } else | |
92 | howto = 0; | |
20e66415 A |
93 | kflag = lflag = nflag = qflag = 0; |
94 | #ifndef __APPLE__ | |
95 | while ((ch = getopt(argc, argv, "dk:lnpq")) != -1) | |
96 | #else | |
b180c247 | 97 | while ((ch = getopt(argc, argv, "lnqu")) != -1) |
20e66415 | 98 | #endif |
1815bff5 | 99 | switch(ch) { |
20e66415 A |
100 | #ifndef __APPLE__ |
101 | case 'd': | |
102 | howto |= RB_DUMP; | |
103 | break; | |
20e66415 A |
104 | case 'k': |
105 | kflag = 1; | |
106 | kernel = optarg; | |
107 | break; | |
2fc1e207 | 108 | #endif |
20e66415 | 109 | case 'l': |
1815bff5 A |
110 | lflag = 1; |
111 | break; | |
112 | case 'n': | |
113 | nflag = 1; | |
114 | howto |= RB_NOSYNC; | |
115 | break; | |
20e66415 A |
116 | /* -p is irrelevant on OS X. It does that anyway. */ |
117 | #ifndef __APPLE__ | |
118 | case 'p': | |
119 | pflag = 1; | |
120 | howto |= RB_POWEROFF; | |
121 | break; | |
122 | #endif | |
b180c247 A |
123 | case 'u': |
124 | uflag = 1; | |
125 | howto |= RB_UPSDELAY; | |
126 | break; | |
1815bff5 A |
127 | case 'q': |
128 | qflag = 1; | |
129 | break; | |
130 | case '?': | |
131 | default: | |
132 | usage(); | |
133 | } | |
134 | argc -= optind; | |
135 | argv += optind; | |
136 | ||
20e66415 A |
137 | #ifndef __APPLE__ |
138 | if ((howto & (RB_DUMP | RB_HALT)) == (RB_DUMP | RB_HALT)) | |
139 | errx(1, "cannot dump (-d) when halting; must reboot instead"); | |
140 | #endif | |
141 | if (geteuid()) { | |
142 | errno = EPERM; | |
143 | err(1, NULL); | |
144 | } | |
1815bff5 | 145 | |
b180c247 A |
146 | #ifdef __APPLE__ |
147 | if (!lflag) { // shutdown(8) has already checked w/kextd | |
148 | if ((errno = reserve_reboot()) && !qflag) | |
149 | err(1, "couldn't lock for reboot"); | |
150 | } | |
151 | #endif | |
152 | ||
1815bff5 A |
153 | if (qflag) { |
154 | reboot(howto); | |
20e66415 A |
155 | err(1, NULL); |
156 | } | |
157 | ||
2fc1e207 | 158 | #ifndef __APPLE__ |
20e66415 A |
159 | if (kflag) { |
160 | fd = open("/boot/nextboot.conf", O_WRONLY | O_CREAT, 0444); | |
161 | if (fd > -1) { | |
162 | (void)write(fd, "nextboot_enable=\"YES\"\n", 22); | |
163 | (void)write(fd, "kernel=\"", 8L); | |
164 | (void)write(fd, kernel, strlen(kernel)); | |
165 | (void)write(fd, "\"\n", 2); | |
166 | close(fd); | |
167 | } | |
1815bff5 | 168 | } |
2fc1e207 | 169 | #endif |
1815bff5 A |
170 | |
171 | /* Log the reboot. */ | |
172 | if (!lflag) { | |
173 | if ((user = getlogin()) == NULL) | |
174 | user = (pw = getpwuid(getuid())) ? | |
175 | pw->pw_name : "???"; | |
176 | if (dohalt) { | |
177 | openlog("halt", 0, LOG_AUTH | LOG_CONS); | |
b180c247 A |
178 | syslog(LOG_CRIT, "halted by %s%s", user, |
179 | (howto & RB_UPSDELAY) ? " with UPS delay":""); | |
1815bff5 A |
180 | } else { |
181 | openlog("reboot", 0, LOG_AUTH | LOG_CONS); | |
182 | syslog(LOG_CRIT, "rebooted by %s", user); | |
183 | } | |
184 | } | |
185 | logwtmp("~", "shutdown", ""); | |
186 | ||
187 | /* | |
188 | * Do a sync early on, so disks start transfers while we're off | |
189 | * killing processes. Don't worry about writes done before the | |
190 | * processes die, the reboot system call syncs the disks. | |
191 | */ | |
192 | if (!nflag) | |
193 | sync(); | |
194 | ||
195 | /* Just stop init -- if we fail, we'll restart it. */ | |
196 | if (kill(1, SIGTSTP) == -1) | |
20e66415 | 197 | err(1, "SIGTSTP init"); |
1815bff5 A |
198 | |
199 | /* Ignore the SIGHUP we get when our parent shell dies. */ | |
200 | (void)signal(SIGHUP, SIG_IGN); | |
201 | ||
20e66415 | 202 | #ifndef __APPLE__ |
1815bff5 A |
203 | /* Send a SIGTERM first, a chance to save the buffers. */ |
204 | if (kill(-1, SIGTERM) == -1) | |
20e66415 | 205 | err(1, "SIGTERM processes"); |
1815bff5 A |
206 | |
207 | /* | |
208 | * After the processes receive the signal, start the rest of the | |
209 | * buffers on their way. Wait 5 seconds between the SIGTERM and | |
20e66415 A |
210 | * the SIGKILL to give everybody a chance. If there is a lot of |
211 | * paging activity then wait longer, up to a maximum of approx | |
212 | * 60 seconds. | |
1815bff5 A |
213 | */ |
214 | sleep(2); | |
20e66415 A |
215 | for (i = 0; i < 20; i++) { |
216 | pageins = get_pageins(); | |
217 | if (!nflag) | |
218 | sync(); | |
219 | sleep(3); | |
220 | if (get_pageins() == pageins) | |
221 | break; | |
222 | } | |
1815bff5 A |
223 | |
224 | for (i = 1;; ++i) { | |
225 | if (kill(-1, SIGKILL) == -1) { | |
226 | if (errno == ESRCH) | |
227 | break; | |
228 | goto restart; | |
229 | } | |
230 | if (i > 5) { | |
231 | (void)fprintf(stderr, | |
232 | "WARNING: some process(es) wouldn't die\n"); | |
233 | break; | |
234 | } | |
235 | (void)sleep(2 * i); | |
236 | } | |
237 | #endif | |
20e66415 | 238 | |
1815bff5 A |
239 | reboot(howto); |
240 | /* FALLTHROUGH */ | |
241 | ||
b180c247 | 242 | #ifndef __APPLE__ |
1815bff5 | 243 | restart: |
b180c247 | 244 | #endif |
1815bff5 | 245 | sverrno = errno; |
20e66415 | 246 | errx(1, "%s%s", kill(1, SIGHUP) == -1 ? "(can't restart init): " : "", |
1815bff5 A |
247 | strerror(sverrno)); |
248 | /* NOTREACHED */ | |
249 | } | |
250 | ||
251 | void | |
252 | usage() | |
253 | { | |
2fc1e207 | 254 | #ifndef __APPLE__ |
20e66415 | 255 | (void)fprintf(stderr, "usage: %s [-dnpq] [-k kernel]\n", |
2fc1e207 A |
256 | #else |
257 | (void)fprintf(stderr, "usage: %s [-lnq]\n", | |
258 | #endif | |
20e66415 | 259 | dohalt ? "halt" : "reboot"); |
1815bff5 A |
260 | exit(1); |
261 | } | |
262 | ||
20e66415 A |
263 | u_int |
264 | get_pageins() | |
1815bff5 | 265 | { |
20e66415 A |
266 | u_int pageins; |
267 | size_t len; | |
268 | ||
269 | len = sizeof(pageins); | |
270 | if (sysctlbyname("vm.stats.vm.v_swappgsin", &pageins, &len, NULL, 0) | |
271 | != 0) { | |
272 | warnx("v_swappgsin"); | |
273 | return (0); | |
274 | } | |
275 | return pageins; | |
1815bff5 | 276 | } |
b180c247 A |
277 | |
278 | #ifdef __APPLE__ | |
279 | // XX another copy of this routine is in shutdown.c; it would be nice to share | |
280 | ||
281 | #define LCK_MAXTRIES 10 | |
282 | #define LCK_DELAY 30 | |
283 | /* | |
284 | * contact kextd to lock for reboot | |
285 | */ | |
286 | int | |
287 | reserve_reboot() | |
288 | { | |
289 | int rval = ELAST+1; | |
290 | kern_return_t macherr = KERN_FAILURE; | |
291 | mach_port_t tport, bsport, kxport, myport = MACH_PORT_NULL; | |
292 | int busyStatus, nretries = LCK_MAXTRIES; | |
293 | dev_path_t busyDev = "<unknown>"; | |
294 | ||
295 | // find kextd | |
296 | tport = mach_task_self(); | |
297 | if (tport == MACH_PORT_NULL) goto finish; | |
298 | macherr = task_get_bootstrap_port(tport, &bsport); | |
299 | if (macherr) goto finish; | |
300 | macherr = bootstrap_look_up(bsport, KEXTD_SERVER_NAME, &kxport); | |
301 | if (macherr) goto finish; | |
302 | ||
303 | // allocate a port to pass to kextd (in case we die) | |
304 | macherr = mach_port_allocate(tport, MACH_PORT_RIGHT_RECEIVE, &myport); | |
305 | if (macherr) goto finish; | |
306 | ||
307 | // loop trying to lock for reboot (i.e. no volumes are busy) | |
308 | do { | |
309 | nretries--; | |
310 | macherr = kextmanager_lock_reboot(kxport, myport, busyDev, &busyStatus); | |
311 | if (macherr) goto finish; | |
312 | ||
313 | if (busyStatus == EBUSY) { | |
314 | if (*busyDev) { | |
315 | warnx("%s is busy updating; delaying reboot (%d retries left)", | |
316 | busyDev, nretries); | |
317 | } else | |
318 | warnx("kextd still starting up"); | |
319 | if (nretries) sleep(LCK_DELAY); // don't sleep the last time | |
320 | } | |
321 | } while (busyStatus == EBUSY && nretries > 0); | |
322 | ||
323 | rval = busyStatus; | |
324 | ||
325 | finish: | |
326 | if (macherr == BOOTSTRAP_UNKNOWN_SERVICE) { | |
327 | mach_port_mod_refs(tport, myport, MACH_PORT_RIGHT_RECEIVE, -1); | |
328 | rval = 0; | |
329 | } else if (macherr) { | |
330 | warnx("couldn't lock kext manager for reboot: %s", | |
331 | mach_error_string(macherr)); | |
332 | rval = ELAST + 1; | |
333 | } | |
334 | if (rval && myport != MACH_PORT_NULL) { | |
335 | mach_port_mod_refs(tport, myport, MACH_PORT_RIGHT_RECEIVE, -1); | |
336 | } | |
337 | ||
338 | return rval; | |
339 | } | |
340 | #endif |