kern_subr.c revision 1.78 1 1.78 thorpej /* $NetBSD: kern_subr.c,v 1.78 2002/02/08 00:58:58 thorpej Exp $ */
2 1.31 thorpej
3 1.31 thorpej /*-
4 1.78 thorpej * Copyright (c) 1997, 1998, 1999, 2002 The NetBSD Foundation, Inc.
5 1.31 thorpej * All rights reserved.
6 1.31 thorpej *
7 1.31 thorpej * This code is derived from software contributed to The NetBSD Foundation
8 1.31 thorpej * by Jason R. Thorpe of the Numerical Aerospace Simulation Facility,
9 1.48 lukem * NASA Ames Research Center, and by Luke Mewburn.
10 1.31 thorpej *
11 1.31 thorpej * Redistribution and use in source and binary forms, with or without
12 1.31 thorpej * modification, are permitted provided that the following conditions
13 1.31 thorpej * are met:
14 1.31 thorpej * 1. Redistributions of source code must retain the above copyright
15 1.31 thorpej * notice, this list of conditions and the following disclaimer.
16 1.31 thorpej * 2. Redistributions in binary form must reproduce the above copyright
17 1.31 thorpej * notice, this list of conditions and the following disclaimer in the
18 1.31 thorpej * documentation and/or other materials provided with the distribution.
19 1.31 thorpej * 3. All advertising materials mentioning features or use of this software
20 1.31 thorpej * must display the following acknowledgement:
21 1.31 thorpej * This product includes software developed by the NetBSD
22 1.31 thorpej * Foundation, Inc. and its contributors.
23 1.31 thorpej * 4. Neither the name of The NetBSD Foundation nor the names of its
24 1.31 thorpej * contributors may be used to endorse or promote products derived
25 1.31 thorpej * from this software without specific prior written permission.
26 1.31 thorpej *
27 1.31 thorpej * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
28 1.31 thorpej * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
29 1.31 thorpej * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
30 1.31 thorpej * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
31 1.31 thorpej * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
32 1.31 thorpej * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
33 1.31 thorpej * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
34 1.31 thorpej * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
35 1.31 thorpej * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
36 1.31 thorpej * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
37 1.31 thorpej * POSSIBILITY OF SUCH DAMAGE.
38 1.31 thorpej */
39 1.12 cgd
40 1.9 cgd /*
41 1.10 cgd * Copyright (c) 1982, 1986, 1991, 1993
42 1.10 cgd * The Regents of the University of California. All rights reserved.
43 1.9 cgd * (c) UNIX System Laboratories, Inc.
44 1.9 cgd * All or some portions of this file are derived from material licensed
45 1.9 cgd * to the University of California by American Telephone and Telegraph
46 1.9 cgd * Co. or Unix System Laboratories, Inc. and are reproduced herein with
47 1.9 cgd * the permission of UNIX System Laboratories, Inc.
48 1.9 cgd *
49 1.18 thorpej * Copyright (c) 1992, 1993
50 1.18 thorpej * The Regents of the University of California. All rights reserved.
51 1.18 thorpej *
52 1.18 thorpej * This software was developed by the Computer Systems Engineering group
53 1.18 thorpej * at Lawrence Berkeley Laboratory under DARPA contract BG 91-66 and
54 1.18 thorpej * contributed to Berkeley.
55 1.18 thorpej *
56 1.18 thorpej * All advertising materials mentioning features or use of this software
57 1.18 thorpej * must display the following acknowledgement:
58 1.18 thorpej * This product includes software developed by the University of
59 1.18 thorpej * California, Lawrence Berkeley Laboratory.
60 1.18 thorpej *
61 1.9 cgd * Redistribution and use in source and binary forms, with or without
62 1.9 cgd * modification, are permitted provided that the following conditions
63 1.9 cgd * are met:
64 1.9 cgd * 1. Redistributions of source code must retain the above copyright
65 1.9 cgd * notice, this list of conditions and the following disclaimer.
66 1.9 cgd * 2. Redistributions in binary form must reproduce the above copyright
67 1.9 cgd * notice, this list of conditions and the following disclaimer in the
68 1.9 cgd * documentation and/or other materials provided with the distribution.
69 1.9 cgd * 3. All advertising materials mentioning features or use of this software
70 1.9 cgd * must display the following acknowledgement:
71 1.9 cgd * This product includes software developed by the University of
72 1.9 cgd * California, Berkeley and its contributors.
73 1.9 cgd * 4. Neither the name of the University nor the names of its contributors
74 1.9 cgd * may be used to endorse or promote products derived from this software
75 1.9 cgd * without specific prior written permission.
76 1.9 cgd *
77 1.9 cgd * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
78 1.9 cgd * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
79 1.9 cgd * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
80 1.9 cgd * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
81 1.9 cgd * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
82 1.9 cgd * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
83 1.9 cgd * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
84 1.9 cgd * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
85 1.9 cgd * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
86 1.9 cgd * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
87 1.9 cgd * SUCH DAMAGE.
88 1.9 cgd *
89 1.38 fvdl * @(#)kern_subr.c 8.4 (Berkeley) 2/14/95
90 1.9 cgd */
91 1.77 lukem
92 1.77 lukem #include <sys/cdefs.h>
93 1.78 thorpej __KERNEL_RCSID(0, "$NetBSD: kern_subr.c,v 1.78 2002/02/08 00:58:58 thorpej Exp $");
94 1.34 mrg
95 1.78 thorpej #include "opt_ddb.h"
96 1.43 jonathan #include "opt_md.h"
97 1.9 cgd
98 1.9 cgd #include <sys/param.h>
99 1.9 cgd #include <sys/systm.h>
100 1.9 cgd #include <sys/proc.h>
101 1.10 cgd #include <sys/malloc.h>
102 1.18 thorpej #include <sys/mount.h>
103 1.18 thorpej #include <sys/device.h>
104 1.18 thorpej #include <sys/reboot.h>
105 1.18 thorpej #include <sys/conf.h>
106 1.18 thorpej #include <sys/disklabel.h>
107 1.10 cgd #include <sys/queue.h>
108 1.9 cgd
109 1.18 thorpej #include <dev/cons.h>
110 1.18 thorpej
111 1.18 thorpej #include <net/if.h>
112 1.18 thorpej
113 1.18 thorpej /* XXX these should eventually move to subr_autoconf.c */
114 1.51 thorpej static int findblkmajor __P((const char *));
115 1.74 fvdl const char *findblkname __P((int));
116 1.56 enami static struct device *finddevice __P((const char *));
117 1.51 thorpej static struct device *getdisk __P((char *, int, int, dev_t *, int));
118 1.51 thorpej static struct device *parsedisk __P((char *, int, int, dev_t *));
119 1.18 thorpej
120 1.11 mycroft int
121 1.16 christos uiomove(buf, n, uio)
122 1.63 augustss void *buf;
123 1.63 augustss int n;
124 1.63 augustss struct uio *uio;
125 1.9 cgd {
126 1.63 augustss struct iovec *iov;
127 1.9 cgd u_int cnt;
128 1.9 cgd int error = 0;
129 1.16 christos char *cp = buf;
130 1.70 enami struct proc *p = uio->uio_procp;
131 1.68 enami
132 1.68 enami #ifdef DIAGNOSTIC
133 1.9 cgd if (uio->uio_rw != UIO_READ && uio->uio_rw != UIO_WRITE)
134 1.9 cgd panic("uiomove: mode");
135 1.62 thorpej if (uio->uio_segflg == UIO_USERSPACE && p != curproc)
136 1.9 cgd panic("uiomove proc");
137 1.9 cgd #endif
138 1.9 cgd while (n > 0 && uio->uio_resid) {
139 1.9 cgd iov = uio->uio_iov;
140 1.9 cgd cnt = iov->iov_len;
141 1.9 cgd if (cnt == 0) {
142 1.9 cgd uio->uio_iov++;
143 1.9 cgd uio->uio_iovcnt--;
144 1.9 cgd continue;
145 1.9 cgd }
146 1.9 cgd if (cnt > n)
147 1.9 cgd cnt = n;
148 1.9 cgd switch (uio->uio_segflg) {
149 1.9 cgd
150 1.9 cgd case UIO_USERSPACE:
151 1.69 thorpej KDASSERT(p->p_cpu != NULL);
152 1.69 thorpej KDASSERT(p->p_cpu == curcpu());
153 1.69 thorpej if (p->p_cpu->ci_schedstate.spc_flags &
154 1.67 thorpej SPCF_SHOULDYIELD)
155 1.62 thorpej preempt(NULL);
156 1.9 cgd if (uio->uio_rw == UIO_READ)
157 1.9 cgd error = copyout(cp, iov->iov_base, cnt);
158 1.9 cgd else
159 1.9 cgd error = copyin(iov->iov_base, cp, cnt);
160 1.9 cgd if (error)
161 1.9 cgd return (error);
162 1.9 cgd break;
163 1.9 cgd
164 1.9 cgd case UIO_SYSSPACE:
165 1.32 mrg if (uio->uio_rw == UIO_READ)
166 1.32 mrg error = kcopy(cp, iov->iov_base, cnt);
167 1.32 mrg else
168 1.32 mrg error = kcopy(iov->iov_base, cp, cnt);
169 1.32 mrg if (error)
170 1.59 enami return (error);
171 1.9 cgd break;
172 1.9 cgd }
173 1.44 kleink iov->iov_base = (caddr_t)iov->iov_base + cnt;
174 1.9 cgd iov->iov_len -= cnt;
175 1.9 cgd uio->uio_resid -= cnt;
176 1.9 cgd uio->uio_offset += cnt;
177 1.9 cgd cp += cnt;
178 1.9 cgd n -= cnt;
179 1.9 cgd }
180 1.9 cgd return (error);
181 1.9 cgd }
182 1.9 cgd
183 1.9 cgd /*
184 1.9 cgd * Give next character to user as result of read.
185 1.9 cgd */
186 1.11 mycroft int
187 1.9 cgd ureadc(c, uio)
188 1.63 augustss int c;
189 1.63 augustss struct uio *uio;
190 1.9 cgd {
191 1.63 augustss struct iovec *iov;
192 1.9 cgd
193 1.11 mycroft if (uio->uio_resid <= 0)
194 1.11 mycroft panic("ureadc: non-positive resid");
195 1.9 cgd again:
196 1.11 mycroft if (uio->uio_iovcnt <= 0)
197 1.11 mycroft panic("ureadc: non-positive iovcnt");
198 1.9 cgd iov = uio->uio_iov;
199 1.11 mycroft if (iov->iov_len <= 0) {
200 1.9 cgd uio->uio_iovcnt--;
201 1.9 cgd uio->uio_iov++;
202 1.9 cgd goto again;
203 1.9 cgd }
204 1.9 cgd switch (uio->uio_segflg) {
205 1.9 cgd
206 1.9 cgd case UIO_USERSPACE:
207 1.9 cgd if (subyte(iov->iov_base, c) < 0)
208 1.9 cgd return (EFAULT);
209 1.9 cgd break;
210 1.9 cgd
211 1.9 cgd case UIO_SYSSPACE:
212 1.39 kleink *(char *)iov->iov_base = c;
213 1.9 cgd break;
214 1.9 cgd }
215 1.44 kleink iov->iov_base = (caddr_t)iov->iov_base + 1;
216 1.9 cgd iov->iov_len--;
217 1.9 cgd uio->uio_resid--;
218 1.9 cgd uio->uio_offset++;
219 1.9 cgd return (0);
220 1.9 cgd }
221 1.10 cgd
222 1.10 cgd /*
223 1.10 cgd * General routine to allocate a hash table.
224 1.42 chs * Allocate enough memory to hold at least `elements' list-head pointers.
225 1.42 chs * Return a pointer to the allocated space and set *hashmask to a pattern
226 1.42 chs * suitable for masking a value to use as an index into the returned array.
227 1.10 cgd */
228 1.10 cgd void *
229 1.72 ad hashinit(elements, htype, mtype, mflags, hashmask)
230 1.72 ad int elements;
231 1.72 ad enum hashtype htype;
232 1.72 ad int mtype, mflags;
233 1.10 cgd u_long *hashmask;
234 1.10 cgd {
235 1.10 cgd long hashsize;
236 1.72 ad LIST_HEAD(, generic) *hashtbl_list;
237 1.72 ad TAILQ_HEAD(, generic) *hashtbl_tailq;
238 1.72 ad int i, esize;
239 1.72 ad void *p;
240 1.10 cgd
241 1.10 cgd if (elements <= 0)
242 1.10 cgd panic("hashinit: bad cnt");
243 1.42 chs for (hashsize = 1; hashsize < elements; hashsize <<= 1)
244 1.10 cgd continue;
245 1.72 ad
246 1.72 ad switch (htype) {
247 1.72 ad case HASH_LIST:
248 1.72 ad esize = sizeof(*hashtbl_list);
249 1.72 ad break;
250 1.72 ad case HASH_TAILQ:
251 1.72 ad esize = sizeof(*hashtbl_tailq);
252 1.72 ad break;
253 1.72 ad #ifdef DIAGNOSTIC
254 1.72 ad default:
255 1.72 ad panic("hashinit: invalid table type");
256 1.72 ad #endif
257 1.72 ad }
258 1.72 ad
259 1.72 ad if ((p = malloc((u_long)hashsize * esize, mtype, mflags)) == NULL)
260 1.66 enami return (NULL);
261 1.72 ad
262 1.72 ad switch (htype) {
263 1.72 ad case HASH_LIST:
264 1.72 ad hashtbl_list = p;
265 1.72 ad for (i = 0; i < hashsize; i++)
266 1.72 ad LIST_INIT(&hashtbl_list[i]);
267 1.72 ad break;
268 1.72 ad case HASH_TAILQ:
269 1.72 ad hashtbl_tailq = p;
270 1.72 ad for (i = 0; i < hashsize; i++)
271 1.72 ad TAILQ_INIT(&hashtbl_tailq[i]);
272 1.72 ad break;
273 1.72 ad }
274 1.10 cgd *hashmask = hashsize - 1;
275 1.72 ad return (p);
276 1.61 jdolecek }
277 1.61 jdolecek
278 1.61 jdolecek /*
279 1.61 jdolecek * Free memory from hash table previosly allocated via hashinit().
280 1.61 jdolecek */
281 1.61 jdolecek void
282 1.72 ad hashdone(hashtbl, mtype)
283 1.61 jdolecek void *hashtbl;
284 1.72 ad int mtype;
285 1.61 jdolecek {
286 1.65 enami
287 1.72 ad free(hashtbl, mtype);
288 1.14 cgd }
289 1.14 cgd
290 1.14 cgd /*
291 1.14 cgd * "Shutdown hook" types, functions, and variables.
292 1.14 cgd */
293 1.14 cgd
294 1.14 cgd struct shutdownhook_desc {
295 1.14 cgd LIST_ENTRY(shutdownhook_desc) sfd_list;
296 1.14 cgd void (*sfd_fn) __P((void *));
297 1.14 cgd void *sfd_arg;
298 1.14 cgd };
299 1.14 cgd
300 1.14 cgd LIST_HEAD(, shutdownhook_desc) shutdownhook_list;
301 1.14 cgd
302 1.14 cgd void *
303 1.14 cgd shutdownhook_establish(fn, arg)
304 1.14 cgd void (*fn) __P((void *));
305 1.14 cgd void *arg;
306 1.14 cgd {
307 1.14 cgd struct shutdownhook_desc *ndp;
308 1.14 cgd
309 1.14 cgd ndp = (struct shutdownhook_desc *)
310 1.40 perry malloc(sizeof(*ndp), M_DEVBUF, M_NOWAIT);
311 1.14 cgd if (ndp == NULL)
312 1.59 enami return (NULL);
313 1.14 cgd
314 1.14 cgd ndp->sfd_fn = fn;
315 1.14 cgd ndp->sfd_arg = arg;
316 1.14 cgd LIST_INSERT_HEAD(&shutdownhook_list, ndp, sfd_list);
317 1.14 cgd
318 1.14 cgd return (ndp);
319 1.14 cgd }
320 1.14 cgd
321 1.14 cgd void
322 1.14 cgd shutdownhook_disestablish(vhook)
323 1.14 cgd void *vhook;
324 1.14 cgd {
325 1.14 cgd #ifdef DIAGNOSTIC
326 1.14 cgd struct shutdownhook_desc *dp;
327 1.14 cgd
328 1.14 cgd for (dp = shutdownhook_list.lh_first; dp != NULL;
329 1.14 cgd dp = dp->sfd_list.le_next)
330 1.14 cgd if (dp == vhook)
331 1.14 cgd break;
332 1.14 cgd if (dp == NULL)
333 1.14 cgd panic("shutdownhook_disestablish: hook not established");
334 1.14 cgd #endif
335 1.14 cgd
336 1.14 cgd LIST_REMOVE((struct shutdownhook_desc *)vhook, sfd_list);
337 1.17 cgd free(vhook, M_DEVBUF);
338 1.14 cgd }
339 1.14 cgd
340 1.14 cgd /*
341 1.14 cgd * Run shutdown hooks. Should be invoked immediately before the
342 1.14 cgd * system is halted or rebooted, i.e. after file systems unmounted,
343 1.14 cgd * after crash dump done, etc.
344 1.17 cgd *
345 1.17 cgd * Each shutdown hook is removed from the list before it's run, so that
346 1.17 cgd * it won't be run again.
347 1.14 cgd */
348 1.14 cgd void
349 1.14 cgd doshutdownhooks()
350 1.14 cgd {
351 1.14 cgd struct shutdownhook_desc *dp;
352 1.14 cgd
353 1.17 cgd while ((dp = shutdownhook_list.lh_first) != NULL) {
354 1.17 cgd LIST_REMOVE(dp, sfd_list);
355 1.14 cgd (*dp->sfd_fn)(dp->sfd_arg);
356 1.17 cgd #if 0
357 1.17 cgd /*
358 1.17 cgd * Don't bother freeing the hook structure,, since we may
359 1.17 cgd * be rebooting because of a memory corruption problem,
360 1.17 cgd * and this might only make things worse. It doesn't
361 1.17 cgd * matter, anyway, since the system is just about to
362 1.17 cgd * reboot.
363 1.17 cgd */
364 1.17 cgd free(dp, M_DEVBUF);
365 1.17 cgd #endif
366 1.52 augustss }
367 1.52 augustss }
368 1.52 augustss
369 1.52 augustss /*
370 1.52 augustss * "Power hook" types, functions, and variables.
371 1.71 augustss * The list of power hooks is kept ordered with the last registered hook
372 1.71 augustss * first.
373 1.71 augustss * When running the hooks on power down the hooks are called in reverse
374 1.71 augustss * registration order, when powering up in registration order.
375 1.52 augustss */
376 1.52 augustss
377 1.52 augustss struct powerhook_desc {
378 1.71 augustss CIRCLEQ_ENTRY(powerhook_desc) sfd_list;
379 1.52 augustss void (*sfd_fn) __P((int, void *));
380 1.52 augustss void *sfd_arg;
381 1.52 augustss };
382 1.52 augustss
383 1.71 augustss CIRCLEQ_HEAD(, powerhook_desc) powerhook_list =
384 1.71 augustss CIRCLEQ_HEAD_INITIALIZER(powerhook_list);
385 1.52 augustss
386 1.52 augustss void *
387 1.52 augustss powerhook_establish(fn, arg)
388 1.52 augustss void (*fn) __P((int, void *));
389 1.52 augustss void *arg;
390 1.52 augustss {
391 1.52 augustss struct powerhook_desc *ndp;
392 1.52 augustss
393 1.52 augustss ndp = (struct powerhook_desc *)
394 1.52 augustss malloc(sizeof(*ndp), M_DEVBUF, M_NOWAIT);
395 1.52 augustss if (ndp == NULL)
396 1.59 enami return (NULL);
397 1.52 augustss
398 1.52 augustss ndp->sfd_fn = fn;
399 1.52 augustss ndp->sfd_arg = arg;
400 1.71 augustss CIRCLEQ_INSERT_HEAD(&powerhook_list, ndp, sfd_list);
401 1.52 augustss
402 1.52 augustss return (ndp);
403 1.52 augustss }
404 1.52 augustss
405 1.52 augustss void
406 1.52 augustss powerhook_disestablish(vhook)
407 1.52 augustss void *vhook;
408 1.52 augustss {
409 1.52 augustss #ifdef DIAGNOSTIC
410 1.52 augustss struct powerhook_desc *dp;
411 1.52 augustss
412 1.71 augustss CIRCLEQ_FOREACH(dp, &powerhook_list, sfd_list)
413 1.52 augustss if (dp == vhook)
414 1.71 augustss goto found;
415 1.71 augustss panic("powerhook_disestablish: hook not established");
416 1.71 augustss found:
417 1.52 augustss #endif
418 1.52 augustss
419 1.71 augustss CIRCLEQ_REMOVE(&powerhook_list, (struct powerhook_desc *)vhook,
420 1.71 augustss sfd_list);
421 1.52 augustss free(vhook, M_DEVBUF);
422 1.52 augustss }
423 1.52 augustss
424 1.52 augustss /*
425 1.52 augustss * Run power hooks.
426 1.52 augustss */
427 1.52 augustss void
428 1.52 augustss dopowerhooks(why)
429 1.52 augustss int why;
430 1.52 augustss {
431 1.52 augustss struct powerhook_desc *dp;
432 1.52 augustss
433 1.73 takemura if (why == PWR_RESUME || why == PWR_SOFTRESUME) {
434 1.71 augustss CIRCLEQ_FOREACH_REVERSE(dp, &powerhook_list, sfd_list) {
435 1.71 augustss (*dp->sfd_fn)(why, dp->sfd_arg);
436 1.71 augustss }
437 1.71 augustss } else {
438 1.71 augustss CIRCLEQ_FOREACH(dp, &powerhook_list, sfd_list) {
439 1.71 augustss (*dp->sfd_fn)(why, dp->sfd_arg);
440 1.71 augustss }
441 1.18 thorpej }
442 1.18 thorpej }
443 1.18 thorpej
444 1.18 thorpej /*
445 1.18 thorpej * "Mountroot hook" types, functions, and variables.
446 1.18 thorpej */
447 1.18 thorpej
448 1.18 thorpej struct mountroothook_desc {
449 1.18 thorpej LIST_ENTRY(mountroothook_desc) mrd_list;
450 1.18 thorpej struct device *mrd_device;
451 1.18 thorpej void (*mrd_func) __P((struct device *));
452 1.18 thorpej };
453 1.18 thorpej
454 1.18 thorpej LIST_HEAD(, mountroothook_desc) mountroothook_list;
455 1.18 thorpej
456 1.18 thorpej void *
457 1.18 thorpej mountroothook_establish(func, dev)
458 1.18 thorpej void (*func) __P((struct device *));
459 1.18 thorpej struct device *dev;
460 1.18 thorpej {
461 1.18 thorpej struct mountroothook_desc *mrd;
462 1.18 thorpej
463 1.18 thorpej mrd = (struct mountroothook_desc *)
464 1.18 thorpej malloc(sizeof(*mrd), M_DEVBUF, M_NOWAIT);
465 1.18 thorpej if (mrd == NULL)
466 1.18 thorpej return (NULL);
467 1.18 thorpej
468 1.18 thorpej mrd->mrd_device = dev;
469 1.18 thorpej mrd->mrd_func = func;
470 1.18 thorpej LIST_INSERT_HEAD(&mountroothook_list, mrd, mrd_list);
471 1.18 thorpej
472 1.18 thorpej return (mrd);
473 1.18 thorpej }
474 1.18 thorpej
475 1.18 thorpej void
476 1.18 thorpej mountroothook_disestablish(vhook)
477 1.18 thorpej void *vhook;
478 1.18 thorpej {
479 1.18 thorpej #ifdef DIAGNOSTIC
480 1.18 thorpej struct mountroothook_desc *mrd;
481 1.18 thorpej
482 1.18 thorpej for (mrd = mountroothook_list.lh_first; mrd != NULL;
483 1.18 thorpej mrd = mrd->mrd_list.le_next)
484 1.18 thorpej if (mrd == vhook)
485 1.18 thorpej break;
486 1.18 thorpej if (mrd == NULL)
487 1.18 thorpej panic("mountroothook_disestablish: hook not established");
488 1.18 thorpej #endif
489 1.18 thorpej
490 1.18 thorpej LIST_REMOVE((struct mountroothook_desc *)vhook, mrd_list);
491 1.18 thorpej free(vhook, M_DEVBUF);
492 1.18 thorpej }
493 1.18 thorpej
494 1.18 thorpej void
495 1.18 thorpej mountroothook_destroy()
496 1.18 thorpej {
497 1.18 thorpej struct mountroothook_desc *mrd;
498 1.18 thorpej
499 1.18 thorpej while ((mrd = mountroothook_list.lh_first) != NULL) {
500 1.18 thorpej LIST_REMOVE(mrd, mrd_list);
501 1.18 thorpej free(mrd, M_DEVBUF);
502 1.18 thorpej }
503 1.18 thorpej }
504 1.18 thorpej
505 1.18 thorpej void
506 1.18 thorpej domountroothook()
507 1.18 thorpej {
508 1.18 thorpej struct mountroothook_desc *mrd;
509 1.18 thorpej
510 1.18 thorpej for (mrd = mountroothook_list.lh_first; mrd != NULL;
511 1.18 thorpej mrd = mrd->mrd_list.le_next) {
512 1.18 thorpej if (mrd->mrd_device == root_device) {
513 1.18 thorpej (*mrd->mrd_func)(root_device);
514 1.18 thorpej return;
515 1.18 thorpej }
516 1.53 fvdl }
517 1.53 fvdl }
518 1.53 fvdl
519 1.53 fvdl /*
520 1.53 fvdl * Exec hook code.
521 1.53 fvdl */
522 1.53 fvdl
523 1.53 fvdl struct exechook_desc {
524 1.53 fvdl LIST_ENTRY(exechook_desc) ehk_list;
525 1.53 fvdl void (*ehk_fn) __P((struct proc *, void *));
526 1.53 fvdl void *ehk_arg;
527 1.53 fvdl };
528 1.53 fvdl
529 1.53 fvdl LIST_HEAD(, exechook_desc) exechook_list;
530 1.53 fvdl
531 1.53 fvdl void *
532 1.53 fvdl exechook_establish(fn, arg)
533 1.53 fvdl void (*fn) __P((struct proc *, void *));
534 1.53 fvdl void *arg;
535 1.53 fvdl {
536 1.53 fvdl struct exechook_desc *edp;
537 1.53 fvdl
538 1.53 fvdl edp = (struct exechook_desc *)
539 1.53 fvdl malloc(sizeof(*edp), M_DEVBUF, M_NOWAIT);
540 1.53 fvdl if (edp == NULL)
541 1.59 enami return (NULL);
542 1.53 fvdl
543 1.53 fvdl edp->ehk_fn = fn;
544 1.53 fvdl edp->ehk_arg = arg;
545 1.53 fvdl LIST_INSERT_HEAD(&exechook_list, edp, ehk_list);
546 1.53 fvdl
547 1.53 fvdl return (edp);
548 1.53 fvdl }
549 1.53 fvdl
550 1.53 fvdl void
551 1.53 fvdl exechook_disestablish(vhook)
552 1.53 fvdl void *vhook;
553 1.53 fvdl {
554 1.53 fvdl #ifdef DIAGNOSTIC
555 1.53 fvdl struct exechook_desc *edp;
556 1.53 fvdl
557 1.53 fvdl for (edp = exechook_list.lh_first; edp != NULL;
558 1.53 fvdl edp = edp->ehk_list.le_next)
559 1.53 fvdl if (edp == vhook)
560 1.53 fvdl break;
561 1.53 fvdl if (edp == NULL)
562 1.53 fvdl panic("exechook_disestablish: hook not established");
563 1.53 fvdl #endif
564 1.53 fvdl
565 1.53 fvdl LIST_REMOVE((struct exechook_desc *)vhook, ehk_list);
566 1.53 fvdl free(vhook, M_DEVBUF);
567 1.53 fvdl }
568 1.53 fvdl
569 1.53 fvdl /*
570 1.53 fvdl * Run exec hooks.
571 1.53 fvdl */
572 1.53 fvdl void
573 1.53 fvdl doexechooks(p)
574 1.53 fvdl struct proc *p;
575 1.53 fvdl {
576 1.53 fvdl struct exechook_desc *edp;
577 1.53 fvdl
578 1.53 fvdl for (edp = LIST_FIRST(&exechook_list);
579 1.53 fvdl edp != NULL;
580 1.53 fvdl edp = LIST_NEXT(edp, ehk_list)) {
581 1.53 fvdl (*edp->ehk_fn)(p, edp->ehk_arg);
582 1.18 thorpej }
583 1.18 thorpej }
584 1.18 thorpej
585 1.18 thorpej /*
586 1.18 thorpej * Determine the root device and, if instructed to, the root file system.
587 1.18 thorpej */
588 1.18 thorpej
589 1.18 thorpej #include "md.h"
590 1.18 thorpej #if NMD == 0
591 1.18 thorpej #undef MEMORY_DISK_HOOKS
592 1.18 thorpej #endif
593 1.18 thorpej
594 1.18 thorpej #ifdef MEMORY_DISK_HOOKS
595 1.28 leo static struct device fakemdrootdev[NMD];
596 1.18 thorpej #endif
597 1.18 thorpej
598 1.57 oster #include "raid.h"
599 1.57 oster #if NRAID == 1
600 1.57 oster #define BOOT_FROM_RAID_HOOKS 1
601 1.57 oster #endif
602 1.57 oster
603 1.57 oster #ifdef BOOT_FROM_RAID_HOOKS
604 1.57 oster extern int numraid;
605 1.57 oster extern struct device *raidrootdev;
606 1.57 oster #endif
607 1.57 oster
608 1.18 thorpej void
609 1.51 thorpej setroot(bootdv, bootpartition)
610 1.18 thorpej struct device *bootdv;
611 1.18 thorpej int bootpartition;
612 1.18 thorpej {
613 1.18 thorpej struct device *dv;
614 1.55 enami int len;
615 1.37 hannken #ifdef MEMORY_DISK_HOOKS
616 1.37 hannken int i;
617 1.37 hannken #endif
618 1.25 mrg dev_t nrootdev;
619 1.26 thorpej dev_t ndumpdev = NODEV;
620 1.18 thorpej char buf[128];
621 1.18 thorpej const char *rootdevname;
622 1.26 thorpej const char *dumpdevname;
623 1.26 thorpej struct device *rootdv = NULL; /* XXX gcc -Wuninitialized */
624 1.26 thorpej struct device *dumpdv = NULL;
625 1.18 thorpej struct ifnet *ifp;
626 1.18 thorpej const char *deffsname;
627 1.18 thorpej struct vfsops *vops;
628 1.18 thorpej extern int (*mountroot) __P((void));
629 1.18 thorpej
630 1.28 leo #ifdef MEMORY_DISK_HOOKS
631 1.28 leo for (i = 0; i < NMD; i++) {
632 1.28 leo fakemdrootdev[i].dv_class = DV_DISK;
633 1.28 leo fakemdrootdev[i].dv_cfdata = NULL;
634 1.28 leo fakemdrootdev[i].dv_unit = i;
635 1.28 leo fakemdrootdev[i].dv_parent = NULL;
636 1.28 leo sprintf(fakemdrootdev[i].dv_xname, "md%d", i);
637 1.28 leo }
638 1.28 leo #endif /* MEMORY_DISK_HOOKS */
639 1.28 leo
640 1.22 leo #ifdef MEMORY_DISK_IS_ROOT
641 1.28 leo bootdv = &fakemdrootdev[0];
642 1.18 thorpej bootpartition = 0;
643 1.18 thorpej #endif
644 1.18 thorpej
645 1.18 thorpej /*
646 1.18 thorpej * If NFS is specified as the file system, and we found
647 1.18 thorpej * a DV_DISK boot device (or no boot device at all), then
648 1.18 thorpej * find a reasonable network interface for "rootspec".
649 1.18 thorpej */
650 1.18 thorpej vops = vfs_getopsbyname("nfs");
651 1.18 thorpej if (vops != NULL && vops->vfs_mountroot == mountroot &&
652 1.18 thorpej rootspec == NULL &&
653 1.18 thorpej (bootdv == NULL || bootdv->dv_class != DV_IFNET)) {
654 1.18 thorpej for (ifp = ifnet.tqh_first; ifp != NULL;
655 1.18 thorpej ifp = ifp->if_list.tqe_next)
656 1.18 thorpej if ((ifp->if_flags &
657 1.18 thorpej (IFF_LOOPBACK|IFF_POINTOPOINT)) == 0)
658 1.18 thorpej break;
659 1.18 thorpej if (ifp == NULL) {
660 1.18 thorpej /*
661 1.18 thorpej * Can't find a suitable interface; ask the
662 1.18 thorpej * user.
663 1.18 thorpej */
664 1.18 thorpej boothowto |= RB_ASKNAME;
665 1.18 thorpej } else {
666 1.18 thorpej /*
667 1.18 thorpej * Have a suitable interface; behave as if
668 1.18 thorpej * the user specified this interface.
669 1.18 thorpej */
670 1.18 thorpej rootspec = (const char *)ifp->if_xname;
671 1.18 thorpej }
672 1.18 thorpej }
673 1.24 thorpej
674 1.24 thorpej /*
675 1.24 thorpej * If wildcarded root and we the boot device wasn't determined,
676 1.24 thorpej * ask the user.
677 1.24 thorpej */
678 1.24 thorpej if (rootspec == NULL && bootdv == NULL)
679 1.24 thorpej boothowto |= RB_ASKNAME;
680 1.18 thorpej
681 1.18 thorpej top:
682 1.18 thorpej if (boothowto & RB_ASKNAME) {
683 1.26 thorpej struct device *defdumpdv;
684 1.26 thorpej
685 1.18 thorpej for (;;) {
686 1.18 thorpej printf("root device");
687 1.18 thorpej if (bootdv != NULL) {
688 1.18 thorpej printf(" (default %s", bootdv->dv_xname);
689 1.18 thorpej if (bootdv->dv_class == DV_DISK)
690 1.18 thorpej printf("%c", bootpartition + 'a');
691 1.18 thorpej printf(")");
692 1.18 thorpej }
693 1.18 thorpej printf(": ");
694 1.64 itojun len = cngetsn(buf, sizeof(buf));
695 1.18 thorpej if (len == 0 && bootdv != NULL) {
696 1.18 thorpej strcpy(buf, bootdv->dv_xname);
697 1.18 thorpej len = strlen(buf);
698 1.18 thorpej }
699 1.18 thorpej if (len > 0 && buf[len - 1] == '*') {
700 1.18 thorpej buf[--len] = '\0';
701 1.51 thorpej dv = getdisk(buf, len, 1, &nrootdev, 0);
702 1.18 thorpej if (dv != NULL) {
703 1.18 thorpej rootdv = dv;
704 1.25 mrg break;
705 1.18 thorpej }
706 1.18 thorpej }
707 1.51 thorpej dv = getdisk(buf, len, bootpartition, &nrootdev, 0);
708 1.18 thorpej if (dv != NULL) {
709 1.18 thorpej rootdv = dv;
710 1.18 thorpej break;
711 1.18 thorpej }
712 1.18 thorpej }
713 1.18 thorpej
714 1.26 thorpej /*
715 1.26 thorpej * Set up the default dump device. If root is on
716 1.26 thorpej * a network device, there is no default dump
717 1.26 thorpej * device, since we don't support dumps to the
718 1.26 thorpej * network.
719 1.26 thorpej */
720 1.26 thorpej if (rootdv->dv_class == DV_IFNET)
721 1.26 thorpej defdumpdv = NULL;
722 1.26 thorpej else
723 1.26 thorpej defdumpdv = rootdv;
724 1.26 thorpej
725 1.26 thorpej for (;;) {
726 1.26 thorpej printf("dump device");
727 1.26 thorpej if (defdumpdv != NULL) {
728 1.26 thorpej /*
729 1.26 thorpej * Note, we know it's a disk if we get here.
730 1.26 thorpej */
731 1.26 thorpej printf(" (default %sb)", defdumpdv->dv_xname);
732 1.26 thorpej }
733 1.26 thorpej printf(": ");
734 1.64 itojun len = cngetsn(buf, sizeof(buf));
735 1.26 thorpej if (len == 0) {
736 1.26 thorpej if (defdumpdv != NULL) {
737 1.26 thorpej ndumpdev = MAKEDISKDEV(major(nrootdev),
738 1.26 thorpej DISKUNIT(nrootdev), 1);
739 1.26 thorpej }
740 1.54 enami dumpdv = defdumpdv;
741 1.26 thorpej break;
742 1.26 thorpej }
743 1.26 thorpej if (len == 4 && strcmp(buf, "none") == 0) {
744 1.55 enami dumpdv = NULL;
745 1.54 enami break;
746 1.26 thorpej }
747 1.51 thorpej dv = getdisk(buf, len, 1, &ndumpdev, 1);
748 1.55 enami if (dv != NULL) {
749 1.26 thorpej dumpdv = dv;
750 1.26 thorpej break;
751 1.26 thorpej }
752 1.26 thorpej }
753 1.26 thorpej
754 1.18 thorpej rootdev = nrootdev;
755 1.26 thorpej dumpdev = ndumpdev;
756 1.18 thorpej
757 1.36 thorpej for (vops = LIST_FIRST(&vfs_list); vops != NULL;
758 1.36 thorpej vops = LIST_NEXT(vops, vfs_list)) {
759 1.36 thorpej if (vops->vfs_mountroot != NULL &&
760 1.36 thorpej vops->vfs_mountroot == mountroot)
761 1.38 fvdl break;
762 1.18 thorpej }
763 1.36 thorpej
764 1.36 thorpej if (vops == NULL) {
765 1.18 thorpej mountroot = NULL;
766 1.18 thorpej deffsname = "generic";
767 1.18 thorpej } else
768 1.36 thorpej deffsname = vops->vfs_name;
769 1.36 thorpej
770 1.18 thorpej for (;;) {
771 1.18 thorpej printf("file system (default %s): ", deffsname);
772 1.64 itojun len = cngetsn(buf, sizeof(buf));
773 1.18 thorpej if (len == 0)
774 1.18 thorpej break;
775 1.18 thorpej if (len == 4 && strcmp(buf, "halt") == 0)
776 1.23 gwr cpu_reboot(RB_HALT, NULL);
777 1.76 thorpej else if (len == 6 && strcmp(buf, "reboot") == 0)
778 1.76 thorpej cpu_reboot(0, NULL);
779 1.78 thorpej #if defined(DDB)
780 1.78 thorpej else if (len == 3 && strcmp(buf, "ddb") == 0) {
781 1.78 thorpej console_debugger();
782 1.78 thorpej }
783 1.78 thorpej #endif
784 1.18 thorpej else if (len == 7 && strcmp(buf, "generic") == 0) {
785 1.18 thorpej mountroot = NULL;
786 1.18 thorpej break;
787 1.18 thorpej }
788 1.18 thorpej vops = vfs_getopsbyname(buf);
789 1.18 thorpej if (vops == NULL || vops->vfs_mountroot == NULL) {
790 1.18 thorpej printf("use one of: generic");
791 1.36 thorpej for (vops = LIST_FIRST(&vfs_list);
792 1.36 thorpej vops != NULL;
793 1.36 thorpej vops = LIST_NEXT(vops, vfs_list)) {
794 1.36 thorpej if (vops->vfs_mountroot != NULL)
795 1.36 thorpej printf(" %s", vops->vfs_name);
796 1.36 thorpej }
797 1.78 thorpej #if defined(DDB)
798 1.78 thorpej printf(" ddb");
799 1.78 thorpej #endif
800 1.76 thorpej printf(" halt reboot\n");
801 1.18 thorpej } else {
802 1.18 thorpej mountroot = vops->vfs_mountroot;
803 1.18 thorpej break;
804 1.18 thorpej }
805 1.18 thorpej }
806 1.18 thorpej
807 1.18 thorpej } else if (rootspec == NULL) {
808 1.18 thorpej int majdev;
809 1.18 thorpej
810 1.18 thorpej /*
811 1.18 thorpej * Wildcarded root; use the boot device.
812 1.18 thorpej */
813 1.26 thorpej rootdv = bootdv;
814 1.26 thorpej
815 1.51 thorpej majdev = findblkmajor(bootdv->dv_xname);
816 1.18 thorpej if (majdev >= 0) {
817 1.18 thorpej /*
818 1.25 mrg * Root is on a disk. `bootpartition' is root.
819 1.18 thorpej */
820 1.18 thorpej rootdev = MAKEDISKDEV(majdev, bootdv->dv_unit,
821 1.18 thorpej bootpartition);
822 1.18 thorpej }
823 1.18 thorpej } else {
824 1.18 thorpej
825 1.18 thorpej /*
826 1.25 mrg * `root on <dev> ...'
827 1.18 thorpej */
828 1.18 thorpej
829 1.18 thorpej /*
830 1.18 thorpej * If it's a network interface, we can bail out
831 1.18 thorpej * early.
832 1.18 thorpej */
833 1.56 enami dv = finddevice(rootspec);
834 1.18 thorpej if (dv != NULL && dv->dv_class == DV_IFNET) {
835 1.26 thorpej rootdv = dv;
836 1.26 thorpej goto haveroot;
837 1.18 thorpej }
838 1.18 thorpej
839 1.51 thorpej rootdevname = findblkname(major(rootdev));
840 1.18 thorpej if (rootdevname == NULL) {
841 1.18 thorpej printf("unknown device major 0x%x\n", rootdev);
842 1.18 thorpej boothowto |= RB_ASKNAME;
843 1.18 thorpej goto top;
844 1.18 thorpej }
845 1.41 perry memset(buf, 0, sizeof(buf));
846 1.18 thorpej sprintf(buf, "%s%d", rootdevname, DISKUNIT(rootdev));
847 1.18 thorpej
848 1.56 enami rootdv = finddevice(buf);
849 1.26 thorpej if (rootdv == NULL) {
850 1.18 thorpej printf("device %s (0x%x) not configured\n",
851 1.18 thorpej buf, rootdev);
852 1.18 thorpej boothowto |= RB_ASKNAME;
853 1.18 thorpej goto top;
854 1.18 thorpej }
855 1.26 thorpej }
856 1.18 thorpej
857 1.26 thorpej haveroot:
858 1.18 thorpej
859 1.18 thorpej root_device = rootdv;
860 1.18 thorpej
861 1.18 thorpej switch (rootdv->dv_class) {
862 1.18 thorpej case DV_IFNET:
863 1.55 enami printf("root on %s", rootdv->dv_xname);
864 1.26 thorpej break;
865 1.18 thorpej
866 1.18 thorpej case DV_DISK:
867 1.18 thorpej printf("root on %s%c", rootdv->dv_xname,
868 1.18 thorpej DISKPART(rootdev) + 'a');
869 1.18 thorpej break;
870 1.18 thorpej
871 1.18 thorpej default:
872 1.18 thorpej printf("can't determine root device\n");
873 1.18 thorpej boothowto |= RB_ASKNAME;
874 1.18 thorpej goto top;
875 1.18 thorpej }
876 1.26 thorpej
877 1.26 thorpej /*
878 1.26 thorpej * Now configure the dump device.
879 1.55 enami *
880 1.26 thorpej * If we haven't figured out the dump device, do so, with
881 1.26 thorpej * the following rules:
882 1.26 thorpej *
883 1.26 thorpej * (a) We already know dumpdv in the RB_ASKNAME case.
884 1.26 thorpej *
885 1.26 thorpej * (b) If dumpspec is set, try to use it. If the device
886 1.26 thorpej * is not available, punt.
887 1.26 thorpej *
888 1.26 thorpej * (c) If dumpspec is not set, the dump device is
889 1.26 thorpej * wildcarded or unspecified. If the root device
890 1.26 thorpej * is DV_IFNET, punt. Otherwise, use partition b
891 1.26 thorpej * of the root device.
892 1.26 thorpej */
893 1.26 thorpej
894 1.55 enami if (boothowto & RB_ASKNAME) { /* (a) */
895 1.55 enami if (dumpdv == NULL)
896 1.55 enami goto nodumpdev;
897 1.55 enami } else if (dumpspec != NULL) { /* (b) */
898 1.55 enami if (strcmp(dumpspec, "none") == 0 || dumpdev == NODEV) {
899 1.26 thorpej /*
900 1.55 enami * Operator doesn't want a dump device.
901 1.55 enami * Or looks like they tried to pick a network
902 1.26 thorpej * device. Oops.
903 1.26 thorpej */
904 1.26 thorpej goto nodumpdev;
905 1.26 thorpej }
906 1.26 thorpej
907 1.51 thorpej dumpdevname = findblkname(major(dumpdev));
908 1.26 thorpej if (dumpdevname == NULL)
909 1.26 thorpej goto nodumpdev;
910 1.41 perry memset(buf, 0, sizeof(buf));
911 1.26 thorpej sprintf(buf, "%s%d", dumpdevname, DISKUNIT(dumpdev));
912 1.26 thorpej
913 1.56 enami dumpdv = finddevice(buf);
914 1.56 enami if (dumpdv == NULL) {
915 1.26 thorpej /*
916 1.26 thorpej * Device not configured.
917 1.26 thorpej */
918 1.26 thorpej goto nodumpdev;
919 1.26 thorpej }
920 1.55 enami } else { /* (c) */
921 1.55 enami if (rootdv->dv_class == DV_IFNET)
922 1.55 enami goto nodumpdev;
923 1.55 enami else {
924 1.55 enami dumpdv = rootdv;
925 1.55 enami dumpdev = MAKEDISKDEV(major(rootdev),
926 1.55 enami dumpdv->dv_unit, 1);
927 1.55 enami }
928 1.26 thorpej }
929 1.26 thorpej
930 1.26 thorpej printf(" dumps on %s%c\n", dumpdv->dv_xname, DISKPART(dumpdev) + 'a');
931 1.26 thorpej return;
932 1.26 thorpej
933 1.26 thorpej nodumpdev:
934 1.26 thorpej dumpdev = NODEV;
935 1.55 enami printf("\n");
936 1.18 thorpej }
937 1.18 thorpej
938 1.18 thorpej static int
939 1.51 thorpej findblkmajor(name)
940 1.18 thorpej const char *name;
941 1.18 thorpej {
942 1.18 thorpej int i;
943 1.18 thorpej
944 1.51 thorpej for (i = 0; dev_name2blk[i].d_name != NULL; i++)
945 1.51 thorpej if (strncmp(name, dev_name2blk[i].d_name,
946 1.51 thorpej strlen(dev_name2blk[i].d_name)) == 0)
947 1.51 thorpej return (dev_name2blk[i].d_maj);
948 1.18 thorpej return (-1);
949 1.18 thorpej }
950 1.18 thorpej
951 1.18 thorpej const char *
952 1.51 thorpej findblkname(maj)
953 1.18 thorpej int maj;
954 1.18 thorpej {
955 1.18 thorpej int i;
956 1.18 thorpej
957 1.51 thorpej for (i = 0; dev_name2blk[i].d_name != NULL; i++)
958 1.51 thorpej if (dev_name2blk[i].d_maj == maj)
959 1.51 thorpej return (dev_name2blk[i].d_name);
960 1.18 thorpej return (NULL);
961 1.18 thorpej }
962 1.18 thorpej
963 1.18 thorpej static struct device *
964 1.56 enami finddevice(name)
965 1.56 enami const char *name;
966 1.56 enami {
967 1.56 enami struct device *dv;
968 1.57 oster #ifdef BOOT_FROM_RAID_HOOKS
969 1.57 oster int j;
970 1.57 oster
971 1.57 oster for (j = 0; j < numraid; j++) {
972 1.57 oster if (strcmp(name, raidrootdev[j].dv_xname) == 0) {
973 1.57 oster dv = &raidrootdev[j];
974 1.59 enami return (dv);
975 1.57 oster }
976 1.57 oster }
977 1.75 cgd #endif
978 1.56 enami
979 1.56 enami for (dv = TAILQ_FIRST(&alldevs); dv != NULL;
980 1.56 enami dv = TAILQ_NEXT(dv, dv_list))
981 1.56 enami if (strcmp(dv->dv_xname, name) == 0)
982 1.56 enami break;
983 1.56 enami return (dv);
984 1.56 enami }
985 1.56 enami
986 1.56 enami static struct device *
987 1.51 thorpej getdisk(str, len, defpart, devp, isdump)
988 1.18 thorpej char *str;
989 1.18 thorpej int len, defpart;
990 1.18 thorpej dev_t *devp;
991 1.26 thorpej int isdump;
992 1.18 thorpej {
993 1.28 leo struct device *dv;
994 1.29 drochner #ifdef MEMORY_DISK_HOOKS
995 1.28 leo int i;
996 1.29 drochner #endif
997 1.57 oster #ifdef BOOT_FROM_RAID_HOOKS
998 1.57 oster int j;
999 1.57 oster #endif
1000 1.18 thorpej
1001 1.51 thorpej if ((dv = parsedisk(str, len, defpart, devp)) == NULL) {
1002 1.18 thorpej printf("use one of:");
1003 1.18 thorpej #ifdef MEMORY_DISK_HOOKS
1004 1.26 thorpej if (isdump == 0)
1005 1.28 leo for (i = 0; i < NMD; i++)
1006 1.28 leo printf(" %s[a-%c]", fakemdrootdev[i].dv_xname,
1007 1.28 leo 'a' + MAXPARTITIONS - 1);
1008 1.18 thorpej #endif
1009 1.57 oster #ifdef BOOT_FROM_RAID_HOOKS
1010 1.57 oster if (isdump == 0)
1011 1.57 oster for (j = 0; j < numraid; j++)
1012 1.59 enami printf(" %s[a-%c]", raidrootdev[j].dv_xname,
1013 1.59 enami 'a' + MAXPARTITIONS - 1);
1014 1.57 oster #endif
1015 1.18 thorpej for (dv = alldevs.tqh_first; dv != NULL;
1016 1.18 thorpej dv = dv->dv_list.tqe_next) {
1017 1.18 thorpej if (dv->dv_class == DV_DISK)
1018 1.18 thorpej printf(" %s[a-%c]", dv->dv_xname,
1019 1.19 cgd 'a' + MAXPARTITIONS - 1);
1020 1.26 thorpej if (isdump == 0 && dv->dv_class == DV_IFNET)
1021 1.18 thorpej printf(" %s", dv->dv_xname);
1022 1.18 thorpej }
1023 1.26 thorpej if (isdump)
1024 1.26 thorpej printf(" none");
1025 1.78 thorpej #if defined(DDB)
1026 1.78 thorpej printf(" ddb");
1027 1.78 thorpej #endif
1028 1.76 thorpej printf(" halt reboot\n");
1029 1.18 thorpej }
1030 1.18 thorpej return (dv);
1031 1.18 thorpej }
1032 1.18 thorpej
1033 1.18 thorpej static struct device *
1034 1.51 thorpej parsedisk(str, len, defpart, devp)
1035 1.18 thorpej char *str;
1036 1.18 thorpej int len, defpart;
1037 1.18 thorpej dev_t *devp;
1038 1.18 thorpej {
1039 1.18 thorpej struct device *dv;
1040 1.18 thorpej char *cp, c;
1041 1.29 drochner int majdev, part;
1042 1.29 drochner #ifdef MEMORY_DISK_HOOKS
1043 1.29 drochner int i;
1044 1.29 drochner #endif
1045 1.18 thorpej if (len == 0)
1046 1.18 thorpej return (NULL);
1047 1.18 thorpej
1048 1.18 thorpej if (len == 4 && strcmp(str, "halt") == 0)
1049 1.23 gwr cpu_reboot(RB_HALT, NULL);
1050 1.76 thorpej else if (len == 6 && strcmp(str, "reboot") == 0)
1051 1.76 thorpej cpu_reboot(0, NULL);
1052 1.78 thorpej #if defined(DDB)
1053 1.78 thorpej else if (len == 3 && strcmp(str, "ddb") == 0)
1054 1.78 thorpej console_debugger();
1055 1.78 thorpej #endif
1056 1.18 thorpej
1057 1.18 thorpej cp = str + len - 1;
1058 1.18 thorpej c = *cp;
1059 1.18 thorpej if (c >= 'a' && c <= ('a' + MAXPARTITIONS - 1)) {
1060 1.18 thorpej part = c - 'a';
1061 1.18 thorpej *cp = '\0';
1062 1.18 thorpej } else
1063 1.18 thorpej part = defpart;
1064 1.18 thorpej
1065 1.18 thorpej #ifdef MEMORY_DISK_HOOKS
1066 1.28 leo for (i = 0; i < NMD; i++)
1067 1.28 leo if (strcmp(str, fakemdrootdev[i].dv_xname) == 0) {
1068 1.28 leo dv = &fakemdrootdev[i];
1069 1.28 leo goto gotdisk;
1070 1.28 leo }
1071 1.18 thorpej #endif
1072 1.18 thorpej
1073 1.56 enami dv = finddevice(str);
1074 1.56 enami if (dv != NULL) {
1075 1.56 enami if (dv->dv_class == DV_DISK) {
1076 1.18 thorpej #ifdef MEMORY_DISK_HOOKS
1077 1.18 thorpej gotdisk:
1078 1.18 thorpej #endif
1079 1.51 thorpej majdev = findblkmajor(dv->dv_xname);
1080 1.18 thorpej if (majdev < 0)
1081 1.18 thorpej panic("parsedisk");
1082 1.18 thorpej *devp = MAKEDISKDEV(majdev, dv->dv_unit, part);
1083 1.18 thorpej }
1084 1.18 thorpej
1085 1.56 enami if (dv->dv_class == DV_IFNET)
1086 1.18 thorpej *devp = NODEV;
1087 1.18 thorpej }
1088 1.18 thorpej
1089 1.18 thorpej *cp = c;
1090 1.18 thorpej return (dv);
1091 1.48 lukem }
1092 1.48 lukem
1093 1.48 lukem /*
1094 1.49 lukem * snprintf() `bytes' into `buf', reformatting it so that the number,
1095 1.49 lukem * plus a possible `x' + suffix extension) fits into len bytes (including
1096 1.49 lukem * the terminating NUL).
1097 1.60 enami * Returns the number of bytes stored in buf, or -1 if there was a problem.
1098 1.49 lukem * E.g, given a len of 9 and a suffix of `B':
1099 1.48 lukem * bytes result
1100 1.48 lukem * ----- ------
1101 1.49 lukem * 99999 `99999 B'
1102 1.48 lukem * 100000 `97 KB'
1103 1.48 lukem * 66715648 `65152 KB'
1104 1.48 lukem * 252215296 `240 MB'
1105 1.48 lukem */
1106 1.48 lukem int
1107 1.58 sommerfe humanize_number(buf, len, bytes, suffix, divisor)
1108 1.48 lukem char *buf;
1109 1.48 lukem size_t len;
1110 1.48 lukem u_int64_t bytes;
1111 1.49 lukem const char *suffix;
1112 1.58 sommerfe int divisor;
1113 1.48 lukem {
1114 1.49 lukem /* prefixes are: (none), Kilo, Mega, Giga, Tera, Peta, Exa */
1115 1.48 lukem static const char prefixes[] = " KMGTPE";
1116 1.48 lukem
1117 1.48 lukem int i, r;
1118 1.48 lukem u_int64_t max;
1119 1.49 lukem size_t suffixlen;
1120 1.48 lukem
1121 1.49 lukem if (buf == NULL || suffix == NULL)
1122 1.49 lukem return (-1);
1123 1.48 lukem if (len > 0)
1124 1.48 lukem buf[0] = '\0';
1125 1.49 lukem suffixlen = strlen(suffix);
1126 1.49 lukem /* check if enough room for `x y' + suffix + `\0' */
1127 1.49 lukem if (len < 4 + suffixlen)
1128 1.49 lukem return (-1);
1129 1.48 lukem
1130 1.49 lukem max = 1;
1131 1.49 lukem for (i = 0; i < len - suffixlen - 3; i++)
1132 1.48 lukem max *= 10;
1133 1.48 lukem for (i = 0; bytes >= max && i < sizeof(prefixes); i++)
1134 1.58 sommerfe bytes /= divisor;
1135 1.48 lukem
1136 1.50 lukem r = snprintf(buf, len, "%qu%s%c%s", (unsigned long long)bytes,
1137 1.49 lukem i == 0 ? "" : " ", prefixes[i], suffix);
1138 1.48 lukem
1139 1.48 lukem return (r);
1140 1.49 lukem }
1141 1.49 lukem
1142 1.49 lukem int
1143 1.49 lukem format_bytes(buf, len, bytes)
1144 1.49 lukem char *buf;
1145 1.49 lukem size_t len;
1146 1.49 lukem u_int64_t bytes;
1147 1.49 lukem {
1148 1.49 lukem int rv;
1149 1.49 lukem size_t nlen;
1150 1.49 lukem
1151 1.58 sommerfe rv = humanize_number(buf, len, bytes, "B", 1024);
1152 1.49 lukem if (rv != -1) {
1153 1.49 lukem /* nuke the trailing ` B' if it exists */
1154 1.49 lukem nlen = strlen(buf) - 2;
1155 1.49 lukem if (strcmp(&buf[nlen], " B") == 0)
1156 1.49 lukem buf[nlen] = '\0';
1157 1.49 lukem }
1158 1.49 lukem return (rv);
1159 1.10 cgd }
1160