1 /* $NetBSD: boot2.c,v 1.44 2009/03/21 15:01:56 ad Exp $ */ 2 3 /*- 4 * Copyright (c) 2008 The NetBSD Foundation, Inc. 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 17 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 18 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 19 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 20 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 26 * POSSIBILITY OF SUCH DAMAGE. 27 */ 28 29 /* 30 * Copyright (c) 2003 31 * David Laight. All rights reserved 32 * Copyright (c) 1996, 1997, 1999 33 * Matthias Drochner. All rights reserved. 34 * Copyright (c) 1996, 1997 35 * Perry E. Metzger. All rights reserved. 36 * Copyright (c) 1997 37 * Jason R. Thorpe. All rights reserved 38 * 39 * Redistribution and use in source and binary forms, with or without 40 * modification, are permitted provided that the following conditions 41 * are met: 42 * 1. Redistributions of source code must retain the above copyright 43 * notice, this list of conditions and the following disclaimer. 44 * 2. Redistributions in binary form must reproduce the above copyright 45 * notice, this list of conditions and the following disclaimer in the 46 * documentation and/or other materials provided with the distribution. 47 * 3. All advertising materials mentioning features or use of this software 48 * must display the following acknowledgements: 49 * This product includes software developed for the NetBSD Project 50 * by Matthias Drochner. 51 * This product includes software developed for the NetBSD Project 52 * by Perry E. Metzger. 53 * 4. The names of the authors may not be used to endorse or promote products 54 * derived from this software without specific prior written permission. 55 * 56 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 57 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 58 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 59 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 60 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 61 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 62 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 63 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 64 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 65 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 66 */ 67 68 /* Based on stand/biosboot/main.c */ 69 70 #include <sys/types.h> 71 #include <sys/reboot.h> 72 #include <sys/bootblock.h> 73 74 #include <lib/libsa/stand.h> 75 #include <lib/libsa/ufs.h> 76 #include <lib/libkern/libkern.h> 77 78 #include <libi386.h> 79 #include <bootmod.h> 80 #include <bootmenu.h> 81 #include <vbe.h> 82 #include "devopen.h" 83 84 #ifdef SUPPORT_PS2 85 #include <biosmca.h> 86 #endif 87 88 extern struct x86_boot_params boot_params; 89 90 extern const char bootprog_name[], bootprog_rev[], bootprog_kernrev[]; 91 92 int errno; 93 94 int boot_biosdev; 95 u_int boot_biossector; 96 97 static const char * const names[][2] = { 98 { "netbsd", "netbsd.gz" }, 99 { "onetbsd", "onetbsd.gz" }, 100 { "netbsd.old", "netbsd.old.gz" }, 101 }; 102 103 #define NUMNAMES (sizeof(names)/sizeof(names[0])) 104 #define DEFFILENAME names[0][0] 105 106 #define MAXDEVNAME 16 107 108 static char *default_devname; 109 static int default_unit, default_partition; 110 static const char *default_filename; 111 112 char *sprint_bootsel(const char *); 113 void bootit(const char *, int, int); 114 void print_banner(void); 115 void boot2(int, u_int); 116 117 void command_help(char *); 118 void command_ls(char *); 119 void command_quit(char *); 120 void command_boot(char *); 121 void command_dev(char *); 122 void command_consdev(char *); 123 void command_modules(char *); 124 void command_multiboot(char *); 125 126 const struct bootblk_command commands[] = { 127 { "help", command_help }, 128 { "?", command_help }, 129 { "ls", command_ls }, 130 { "quit", command_quit }, 131 { "boot", command_boot }, 132 { "dev", command_dev }, 133 { "consdev", command_consdev }, 134 { "modules", command_modules }, 135 { "load", module_add }, 136 { "multiboot", command_multiboot }, 137 { "vesa", command_vesa }, 138 { NULL, NULL }, 139 }; 140 141 int 142 parsebootfile(const char *fname, char **fsname, char **devname, 143 int *unit, int *partition, const char **file) 144 { 145 const char *col; 146 147 *fsname = "ufs"; 148 *devname = default_devname; 149 *unit = default_unit; 150 *partition = default_partition; 151 *file = default_filename; 152 153 if (fname == NULL) 154 return 0; 155 156 if ((col = strchr(fname, ':')) != NULL) { /* device given */ 157 static char savedevname[MAXDEVNAME+1]; 158 int devlen; 159 int u = 0, p = 0; 160 int i = 0; 161 162 devlen = col - fname; 163 if (devlen > MAXDEVNAME) 164 return EINVAL; 165 166 #define isvalidname(c) ((c) >= 'a' && (c) <= 'z') 167 if (!isvalidname(fname[i])) 168 return EINVAL; 169 do { 170 savedevname[i] = fname[i]; 171 i++; 172 } while (isvalidname(fname[i])); 173 savedevname[i] = '\0'; 174 175 #define isnum(c) ((c) >= '0' && (c) <= '9') 176 if (i < devlen) { 177 if (!isnum(fname[i])) 178 return EUNIT; 179 do { 180 u *= 10; 181 u += fname[i++] - '0'; 182 } while (isnum(fname[i])); 183 } 184 185 #define isvalidpart(c) ((c) >= 'a' && (c) <= 'z') 186 if (i < devlen) { 187 if (!isvalidpart(fname[i])) 188 return EPART; 189 p = fname[i++] - 'a'; 190 } 191 192 if (i != devlen) 193 return ENXIO; 194 195 *devname = savedevname; 196 *unit = u; 197 *partition = p; 198 fname = col + 1; 199 } 200 201 if (*fname) 202 *file = fname; 203 204 return 0; 205 } 206 207 char * 208 sprint_bootsel(const char *filename) 209 { 210 char *fsname, *devname; 211 int unit, partition; 212 const char *file; 213 static char buf[80]; 214 215 if (parsebootfile(filename, &fsname, &devname, &unit, 216 &partition, &file) == 0) { 217 sprintf(buf, "%s%d%c:%s", devname, unit, 'a' + partition, file); 218 return buf; 219 } 220 return "(invalid)"; 221 } 222 223 static void 224 clearit(void) 225 { 226 227 if (bootconf.clear) 228 clear_pc_screen(); 229 } 230 231 void 232 bootit(const char *filename, int howto, int tell) 233 { 234 235 if (tell) { 236 printf("booting %s", sprint_bootsel(filename)); 237 if (howto) 238 printf(" (howto 0x%x)", howto); 239 printf("\n"); 240 } 241 242 if (exec_netbsd(filename, 0, howto, boot_biosdev < 0x80, clearit) < 0) 243 printf("boot: %s: %s\n", sprint_bootsel(filename), 244 strerror(errno)); 245 else 246 printf("boot returned\n"); 247 } 248 249 void 250 print_banner(void) 251 { 252 253 clearit(); 254 #ifndef SMALL 255 int n; 256 if (bootconf.banner[0]) { 257 for (n = 0; bootconf.banner[n] && n < MAXBANNER; n++) 258 printf("%s\n", bootconf.banner[n]); 259 } else { 260 #endif /* !SMALL */ 261 printf("\n" 262 ">> %s, Revision %s (from NetBSD %s)\n" 263 ">> Memory: %d/%d k\n", 264 bootprog_name, bootprog_rev, bootprog_kernrev, 265 getbasemem(), getextmem()); 266 267 #ifndef SMALL 268 } 269 #endif /* !SMALL */ 270 } 271 272 /* 273 * Called from the initial entry point boot_start in biosboot.S 274 * 275 * biosdev: BIOS drive number the system booted from 276 * biossector: Sector number of the NetBSD partition 277 */ 278 void 279 boot2(int biosdev, u_int biossector) 280 { 281 extern char twiddle_toggle; 282 int currname; 283 char c; 284 285 twiddle_toggle = 1; /* no twiddling until we're ready */ 286 287 initio(boot_params.bp_consdev); 288 289 #ifdef SUPPORT_PS2 290 biosmca(); 291 #endif 292 gateA20(); 293 294 if (boot_params.bp_flags & X86_BP_FLAGS_RESET_VIDEO) 295 biosvideomode(); 296 297 vbe_init(); 298 299 /* need to remember these */ 300 boot_biosdev = biosdev; 301 boot_biossector = biossector; 302 303 /* try to set default device to what BIOS tells us */ 304 bios2dev(biosdev, biossector, &default_devname, &default_unit, 305 &default_partition); 306 307 /* if the user types "boot" without filename */ 308 default_filename = DEFFILENAME; 309 310 #ifndef SMALL 311 parsebootconf(BOOTCONF); 312 313 /* 314 * If console set in boot.cfg, switch to it. 315 * This will print the banner, so we don't need to explicitly do it 316 */ 317 if (bootconf.consdev) 318 command_consdev(bootconf.consdev); 319 else 320 print_banner(); 321 322 /* Display the menu, if applicable */ 323 twiddle_toggle = 0; 324 if (bootconf.nummenu > 0) { 325 /* Does not return */ 326 doboottypemenu(); 327 } 328 329 #else 330 twiddle_toggle = 0; 331 print_banner(); 332 #endif 333 334 printf("Press return to boot now, any other key for boot menu\n"); 335 for (currname = 0; currname < NUMNAMES; currname++) { 336 printf("booting %s - starting in ", 337 sprint_bootsel(names[currname][0])); 338 339 #ifdef SMALL 340 c = awaitkey(boot_params.bp_timeout, 1); 341 #else 342 c = awaitkey((bootconf.timeout < 0) ? 0 : bootconf.timeout, 1); 343 #endif 344 if ((c != '\r') && (c != '\n') && (c != '\0') && 345 ((boot_params.bp_flags & X86_BP_FLAGS_PASSWORD) == 0 346 || check_password(boot_params.bp_password))) { 347 printf("type \"?\" or \"help\" for help.\n"); 348 bootmenu(); /* does not return */ 349 } 350 351 /* 352 * try pairs of names[] entries, foo and foo.gz 353 */ 354 /* don't print "booting..." again */ 355 bootit(names[currname][0], 0, 0); 356 /* since it failed, try compressed bootfile. */ 357 bootit(names[currname][1], 0, 1); 358 } 359 360 bootmenu(); /* does not return */ 361 } 362 363 /* ARGSUSED */ 364 void 365 command_help(char *arg) 366 { 367 368 printf("commands are:\n" 369 "boot [xdNx:][filename] [-12acdqsvxz]\n" 370 " (ex. \"hd0a:netbsd.old -s\"\n" 371 "ls [path]\n" 372 "dev xd[N[x]]:\n" 373 "consdev {pc|com[0123]|com[0123]kbd|auto}\n" 374 "vesa {enabled|disabled|list|modenum}\n" 375 "modules {enabled|disabled}\n" 376 "load {path_to_module}\n" 377 "multiboot [xdNx:][filename] [<args>]\n" 378 "help|?\n" 379 "quit\n"); 380 } 381 382 void 383 command_ls(char *arg) 384 { 385 const char *save = default_filename; 386 387 default_filename = "/"; 388 ufs_ls(arg); 389 default_filename = save; 390 } 391 392 /* ARGSUSED */ 393 void 394 command_quit(char *arg) 395 { 396 397 printf("Exiting...\n"); 398 delay(1000000); 399 reboot(); 400 /* Note: we shouldn't get to this point! */ 401 panic("Could not reboot!"); 402 exit(0); 403 } 404 405 void 406 command_boot(char *arg) 407 { 408 char *filename; 409 int howto; 410 411 if (parseboot(arg, &filename, &howto)) 412 bootit(filename, howto, (howto & AB_VERBOSE) != 0); 413 } 414 415 void 416 command_dev(char *arg) 417 { 418 static char savedevname[MAXDEVNAME + 1]; 419 char *fsname, *devname; 420 const char *file; /* dummy */ 421 422 if (*arg == '\0') { 423 printf("%s%d%c:\n", default_devname, default_unit, 424 'a' + default_partition); 425 return; 426 } 427 428 if (strchr(arg, ':') == NULL || 429 parsebootfile(arg, &fsname, &devname, &default_unit, 430 &default_partition, &file)) { 431 command_help(NULL); 432 return; 433 } 434 435 /* put to own static storage */ 436 strncpy(savedevname, devname, MAXDEVNAME + 1); 437 default_devname = savedevname; 438 } 439 440 static const struct cons_devs { 441 const char *name; 442 u_int tag; 443 } cons_devs[] = { 444 { "pc", CONSDEV_PC }, 445 { "com0", CONSDEV_COM0 }, 446 { "com1", CONSDEV_COM1 }, 447 { "com2", CONSDEV_COM2 }, 448 { "com3", CONSDEV_COM3 }, 449 { "com0kbd", CONSDEV_COM0KBD }, 450 { "com1kbd", CONSDEV_COM1KBD }, 451 { "com2kbd", CONSDEV_COM2KBD }, 452 { "com3kbd", CONSDEV_COM3KBD }, 453 { "auto", CONSDEV_AUTO }, 454 { NULL, 0 } 455 }; 456 457 void 458 command_consdev(char *arg) 459 { 460 const struct cons_devs *cdp; 461 462 for (cdp = cons_devs; cdp->name; cdp++) { 463 if (strcmp(arg, cdp->name) == 0) { 464 initio(cdp->tag); 465 print_banner(); 466 return; 467 } 468 } 469 printf("invalid console device.\n"); 470 } 471 472 void 473 command_modules(char *arg) 474 { 475 476 if (strcmp(arg, "enabled") == 0 || 477 strcmp(arg, "on") == 0) 478 boot_modules_enabled = true; 479 else if (strcmp(arg, "disabled") == 0 || 480 strcmp(arg, "off") == 0) 481 boot_modules_enabled = false; 482 else 483 printf("invalid flag, must be 'enabled' or 'disabled'.\n"); 484 } 485 486 void 487 command_multiboot(char *arg) 488 { 489 char *filename; 490 491 filename = arg; 492 if (exec_multiboot(filename, gettrailer(arg)) < 0) 493 printf("multiboot: %s: %s\n", sprint_bootsel(filename), 494 strerror(errno)); 495 else 496 printf("boot returned\n"); 497 } 498 499