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