]> git.saurik.com Git - apple/system_cmds.git/blame - reboot.tproj/reboot.c
system_cmds-336.23.tar.gz
[apple/system_cmds.git] / reboot.tproj / reboot.c
CommitLineData
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
35static 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
42static char sccsid[] = "@(#)reboot.c 8.1 (Berkeley) 6/5/93";
43#endif
44static 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
71void usage(void);
72u_int get_pageins(void);
b180c247
A
73#ifdef __APPLE__
74int reserve_reboot(void);
75#endif
1815bff5
A
76
77int dohalt;
78
79int
20e66415 80main(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 243restart:
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
251void
252usage()
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
263u_int
264get_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 */
286int
287reserve_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
325finish:
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