1 /* $NetBSD: md.c,v 1.2 2014/08/03 16:09:40 martin Exp $ */ 2 3 /* 4 * Copyright 1997 Piermont Information Systems Inc. 5 * All rights reserved. 6 * 7 * Based on code written by Philip A. Nelson for Piermont Information 8 * Systems Inc. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 1. Redistributions of source code must retain the above copyright 14 * notice, this list of conditions and the following disclaimer. 15 * 2. Redistributions in binary form must reproduce the above copyright 16 * notice, this list of conditions and the following disclaimer in the 17 * documentation and/or other materials provided with the distribution. 18 * 3. The name of Piermont Information Systems Inc. may not be used to endorse 19 * or promote products derived from this software without specific prior 20 * written permission. 21 * 22 * THIS SOFTWARE IS PROVIDED BY PIERMONT INFORMATION SYSTEMS INC. ``AS IS'' 23 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 25 * ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE 26 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 27 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 28 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 29 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 30 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 31 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF 32 * THE POSSIBILITY OF SUCH DAMAGE. 33 */ 34 35 /* md.c -- i386 machine specific routines - also used by amd64 */ 36 37 #include <sys/param.h> 38 #include <sys/sysctl.h> 39 #include <sys/exec.h> 40 #include <sys/utsname.h> 41 #include <sys/types.h> 42 #include <sys/stat.h> 43 #include <machine/cpu.h> 44 #include <stdio.h> 45 #include <stddef.h> 46 #include <util.h> 47 #include <dirent.h> 48 #include <termios.h> 49 50 #include "defs.h" 51 #include "md.h" 52 #include "endian.h" 53 #include "msg_defs.h" 54 #include "menu_defs.h" 55 56 #ifdef NO_LBA_READS /* for testing */ 57 #undef BIFLAG_EXTINT13 58 #define BIFLAG_EXTINT13 0 59 #endif 60 61 static struct biosdisk_info *biosdisk = NULL; 62 63 /* prototypes */ 64 65 static int get_bios_info(char *); 66 static int mbr_root_above_chs(void); 67 static void md_upgrade_mbrtype(void); 68 static int md_read_bootcode(const char *, struct mbr_sector *); 69 static unsigned int get_bootmodel(void); 70 71 void 72 md_init(void) 73 { 74 } 75 76 void 77 md_init_set_status(int flags) 78 { 79 (void)flags; 80 81 /* Default to install same type of kernel as we are running */ 82 set_kernel_set(get_bootmodel()); 83 } 84 85 int 86 md_get_info(void) 87 { 88 mbr_info_t *ext; 89 struct mbr_partition *p; 90 const char *bootcode; 91 int i; 92 int names, fl, ofl; 93 #define ACTIVE_FOUND 0x0100 94 #define NETBSD_ACTIVE 0x0200 95 #define NETBSD_NAMED 0x0400 96 #define ACTIVE_NAMED 0x0800 97 98 if (pm->no_mbr) 99 return 1; 100 101 if (read_mbr(pm->diskdev, &mbr) < 0) 102 memset(&mbr.mbr, 0, sizeof mbr.mbr - 2); 103 get_bios_info(pm->diskdev); 104 105 edit: 106 if (edit_mbr(&mbr) == 0) 107 return 0; 108 109 root_limit = 0; 110 if (biosdisk != NULL && (biosdisk->bi_flags & BIFLAG_EXTINT13) == 0) { 111 if (mbr_root_above_chs()) { 112 msg_display(MSG_partabovechs); 113 process_menu(MENU_noyes, NULL); 114 if (!yesno) 115 goto edit; 116 /* The user is shooting themselves in the foot here...*/ 117 } else 118 root_limit = bcyl * bhead * bsec; 119 } 120 121 /* 122 * Ensure the install partition (at sector pm->ptstart) and the active 123 * partition are bootable. 124 * Determine whether the bootselect code is needed. 125 * Note that MBR_BS_NEWMBR is always set, so we ignore it! 126 */ 127 fl = 0; 128 names = 0; 129 for (ext = &mbr; ext != NULL; ext = ext->extended) { 130 p = ext->mbr.mbr_parts; 131 for (i = 0; i < MBR_PART_COUNT; p++, i++) { 132 if (p->mbrp_flag == MBR_PFLAG_ACTIVE) { 133 fl |= ACTIVE_FOUND; 134 if (ext->sector + p->mbrp_start == pm->ptstart) 135 fl |= NETBSD_ACTIVE; 136 } 137 if (ext->mbrb.mbrbs_nametab[i][0] == 0) { 138 /* No bootmenu label... */ 139 if (ext->sector == 0) 140 continue; 141 if (ext->sector + p->mbrp_start == pm->ptstart) 142 /* 143 * Have installed into an extended ptn 144 * force name & bootsel... 145 */ 146 names++; 147 continue; 148 } 149 /* Partition has a bootmenu label... */ 150 if (ext->sector != 0) 151 fl |= MBR_BS_EXTLBA; 152 if (ext->sector + p->mbrp_start == pm->ptstart) 153 fl |= NETBSD_NAMED; 154 else if (p->mbrp_flag == MBR_PFLAG_ACTIVE) 155 fl |= ACTIVE_NAMED; 156 else 157 names++; 158 } 159 } 160 if (!(fl & ACTIVE_FOUND)) 161 fl |= NETBSD_ACTIVE; 162 if (fl & NETBSD_NAMED && fl & NETBSD_ACTIVE) 163 fl |= ACTIVE_NAMED; 164 165 if ((names > 0 || !(fl & NETBSD_ACTIVE)) && 166 (!(fl & NETBSD_NAMED) || !(fl & ACTIVE_NAMED))) { 167 /* 168 * There appear to be multiple bootable partitions, but they 169 * don't all have bootmenu texts. 170 */ 171 msg_display(MSG_missing_bootmenu_text); 172 process_menu(MENU_yesno, NULL); 173 if (yesno) 174 goto edit; 175 } 176 177 if ((fl & MBR_BS_EXTLBA) && 178 (biosdisk == NULL || !(biosdisk->bi_flags & BIFLAG_EXTINT13))) { 179 /* Need unsupported LBA reads to read boot sectors */ 180 msg_display(MSG_no_extended_bootmenu); 181 process_menu(MENU_noyes, NULL); 182 if (!yesno) 183 goto edit; 184 } 185 186 /* Sort out the name of the mbr code we need */ 187 if (names > 0 || fl & (NETBSD_NAMED | ACTIVE_NAMED)) { 188 /* Need bootselect code */ 189 fl |= MBR_BS_ACTIVE; 190 bootcode = fl & MBR_BS_EXTLBA ? _PATH_BOOTEXT : _PATH_BOOTSEL; 191 } else 192 bootcode = _PATH_MBR; 193 194 fl &= MBR_BS_ACTIVE | MBR_BS_EXTLBA; 195 196 /* Look at what is installed */ 197 ofl = mbr.mbrb.mbrbs_flags; 198 if (ofl == 0) { 199 /* Check there is some bootcode at all... */ 200 if (mbr.mbr.mbr_magic != htole16(MBR_MAGIC) || 201 mbr.mbr.mbr_jmpboot[0] == 0 || 202 mbr_root_above_chs()) 203 /* Existing won't do, force update */ 204 fl |= MBR_BS_NEWMBR; 205 } 206 ofl = mbr.oflags & (MBR_BS_ACTIVE | MBR_BS_EXTLBA); 207 208 if (fl & ~ofl || (fl == 0 && ofl & MBR_BS_ACTIVE)) { 209 /* Existing boot code isn't the right one... */ 210 if (fl & MBR_BS_ACTIVE) 211 msg_display(MSG_installbootsel); 212 else 213 msg_display(MSG_installmbr); 214 } else 215 /* Existing code would (probably) be ok */ 216 msg_display(MSG_updatembr); 217 218 process_menu(MENU_yesno, NULL); 219 if (!yesno) 220 /* User doesn't want to update mbr code */ 221 return 1; 222 223 if (md_read_bootcode(bootcode, &mbr.mbr) == 0) 224 /* update suceeded - to memory copy */ 225 return 1; 226 227 /* This shouldn't happen since the files are in the floppy fs... */ 228 msg_display("Can't find %s", bootcode); 229 process_menu(MENU_yesno, NULL); 230 231 return 1; 232 } 233 234 /* 235 * md back-end code for menu-driven BSD disklabel editor. 236 */ 237 int 238 md_make_bsd_partitions(void) 239 { 240 return make_bsd_partitions(); 241 } 242 243 /* 244 * any additional partition validation 245 */ 246 int 247 md_check_partitions(void) 248 { 249 int rval; 250 char *bootxx; 251 252 /* check we have boot code for the root partition type */ 253 bootxx = bootxx_name(); 254 rval = access(bootxx, R_OK); 255 free(bootxx); 256 if (rval == 0) 257 return 1; 258 process_menu(MENU_ok, deconst(MSG_No_Bootcode)); 259 return 0; 260 } 261 262 /* 263 * hook called before writing new disklabel. 264 */ 265 int 266 md_pre_disklabel(void) 267 { 268 if (pm->no_mbr) 269 return 0; 270 271 msg_display(MSG_dofdisk); 272 273 /* write edited MBR onto disk. */ 274 if (write_mbr(pm->diskdev, &mbr, 1) != 0) { 275 msg_display(MSG_wmbrfail); 276 process_menu(MENU_ok, NULL); 277 return 1; 278 } 279 return 0; 280 } 281 282 /* 283 * hook called after writing disklabel to new target disk. 284 */ 285 int 286 md_post_disklabel(void) 287 { 288 if (get_ramsize() <= 32) 289 set_swap(pm->diskdev, pm->bsdlabel); 290 291 return 0; 292 } 293 294 /* 295 * hook called after upgrade() or install() has finished setting 296 * up the target disk but immediately before the user is given the 297 * ``disks are now set up'' message. 298 */ 299 int 300 md_post_newfs(void) 301 { 302 int ret; 303 size_t len; 304 int td, sd; 305 char bootxx[8192 + 4]; 306 char *bootxx_filename; 307 /* 308 * XXX - should either find some way to pull this automatically 309 * from sys/arch/i386/stand/lib/boot_params.S, or just bite the 310 * bullet and include /sbin/installboot on the ramdisk 311 */ 312 static struct x86_boot_params boottype = 313 {sizeof boottype, 0, 5, 0, 9600, { '\0' }, "", 0}; 314 static int conmib[] = {CTL_MACHDEP, CPU_CONSDEV}; 315 struct termios t; 316 dev_t condev; 317 #define bp (*(struct x86_boot_params *)(bootxx + 512 * 2 + 8)) 318 319 /* 320 * Get console device, should either be ttyE0 or tty0n. 321 * Too hard to double check, so just 'know' the device numbers. 322 */ 323 len = sizeof condev; 324 if (sysctl(conmib, nelem(conmib), &condev, &len, NULL, 0) != -1 325 && (condev & ~3) == 0x800) { 326 /* Motherboard serial port */ 327 boottype.bp_consdev = (condev & 3) + 1; 328 /* Defaulting the baud rate to that of stdin should suffice */ 329 if (tcgetattr(0, &t) != -1) 330 boottype.bp_conspeed = t.c_ispeed; 331 } 332 333 process_menu(MENU_getboottype, &boottype); 334 msg_display(MSG_dobootblks, pm->diskdev); 335 if (bp.bp_consdev == ~0u) 336 return 0; 337 338 ret = cp_to_target("/usr/mdec/boot", "/boot"); 339 if (ret) 340 return ret; 341 342 /* Copy bootstrap in by hand - /sbin/installboot explodes ramdisks */ 343 ret = 1; 344 345 snprintf(bootxx, sizeof bootxx, "/dev/r%s%c", pm->diskdev, 'a' + pm->rootpart); 346 td = open(bootxx, O_RDWR, 0); 347 bootxx_filename = bootxx_name(); 348 if (bootxx_filename != NULL) { 349 sd = open(bootxx_filename, O_RDONLY); 350 free(bootxx_filename); 351 } else 352 sd = -1; 353 if (td == -1 || sd == -1) 354 goto bad_bootxx; 355 len = read(sd, bootxx, sizeof bootxx); 356 if (len < 2048 || len > 8192) 357 goto bad_bootxx; 358 359 if (*(uint32_t *)(bootxx + 512 * 2 + 4) != X86_BOOT_MAGIC_1) 360 goto bad_bootxx; 361 362 boottype.bp_length = bp.bp_length; 363 memcpy(&bp, &boottype, min(boottype.bp_length, sizeof boottype)); 364 365 if (pwrite(td, bootxx, 512, 0) != 512) 366 goto bad_bootxx; 367 len -= 512 * 2; 368 if (pwrite(td, bootxx + 512 * 2, len, 2 * (off_t)512) - len != 0) 369 goto bad_bootxx; 370 ret = 0; 371 372 bad_bootxx: 373 close(td); 374 close(sd); 375 376 return ret; 377 } 378 379 int 380 md_post_extract(void) 381 { 382 return 0; 383 } 384 385 void 386 md_cleanup_install(void) 387 { 388 #ifndef DEBUG 389 enable_rc_conf(); 390 add_rc_conf("wscons=YES\n"); 391 392 # if defined(__i386__) && defined(SET_KERNEL_TINY) 393 /* 394 * For GENERIC_TINY, do not enable any extra screens or wsmux. 395 * Otherwise, run getty on 4 VTs. 396 */ 397 if (get_kernel_set() == SET_KERNEL_TINY) 398 run_program(RUN_CHROOT, 399 "sed -an -e '/^screen/s/^/#/;/^mux/s/^/#/;" 400 "H;$!d;g;w /etc/wscons.conf' /etc/wscons.conf"); 401 else 402 # endif 403 run_program(RUN_CHROOT, 404 "sed -an -e '/^ttyE[1-9]/s/off/on/;" 405 "H;$!d;g;w /etc/ttys' /etc/ttys"); 406 407 #endif 408 } 409 410 int 411 md_pre_update(void) 412 { 413 if (get_ramsize() <= 8) 414 set_swap(pm->diskdev, NULL); 415 return 1; 416 } 417 418 /* Upgrade support */ 419 int 420 md_update(void) 421 { 422 md_post_newfs(); 423 md_upgrade_mbrtype(); 424 return 1; 425 } 426 427 int 428 md_check_mbr(mbr_info_t *mbri) 429 { 430 return 2; 431 } 432 433 int 434 md_mbr_use_wholedisk(mbr_info_t *mbri) 435 { 436 return mbr_use_wholedisk(mbri); 437 } 438 439 static int 440 get_bios_info(char *dev) 441 { 442 static struct disklist *disklist = NULL; 443 static int mib[2] = {CTL_MACHDEP, CPU_DISKINFO}; 444 int i; 445 size_t len; 446 struct biosdisk_info *bip; 447 struct nativedisk_info *nip = NULL, *nat; 448 int cyl, head; 449 daddr_t sec; 450 451 if (disklist == NULL) { 452 if (sysctl(mib, 2, NULL, &len, NULL, 0) < 0) 453 goto nogeom; 454 disklist = malloc(len); 455 if (disklist == NULL) { 456 fprintf(stderr, "Out of memory\n"); 457 return -1; 458 } 459 sysctl(mib, 2, disklist, &len, NULL, 0); 460 } 461 462 for (i = 0; i < disklist->dl_nnativedisks; i++) { 463 nat = &disklist->dl_nativedisks[i]; 464 if (!strcmp(dev, nat->ni_devname)) { 465 nip = nat; 466 break; 467 } 468 } 469 if (nip == NULL || nip->ni_nmatches == 0) { 470 nogeom: 471 if (nip != NULL) 472 msg_display(MSG_nobiosgeom, pm->dlcyl, pm->dlhead, pm->dlsec); 473 if (guess_biosgeom_from_mbr(&mbr, &cyl, &head, &sec) >= 0 474 && nip != NULL) 475 msg_display_add(MSG_biosguess, cyl, head, sec); 476 biosdisk = NULL; 477 } else { 478 guess_biosgeom_from_mbr(&mbr, &cyl, &head, &sec); 479 if (nip->ni_nmatches == 1) { 480 bip = &disklist->dl_biosdisks[nip->ni_biosmatches[0]]; 481 msg_display(MSG_onebiosmatch); 482 msg_table_add(MSG_onebiosmatch_header); 483 msg_table_add(MSG_onebiosmatch_row, bip->bi_dev, 484 bip->bi_cyl, bip->bi_head, bip->bi_sec, 485 (unsigned)bip->bi_lbasecs, 486 (unsigned)(bip->bi_lbasecs / (1000000000 / 512))); 487 msg_display_add(MSG_biosgeom_advise); 488 biosdisk = bip; 489 process_menu(MENU_biosonematch, &biosdisk); 490 } else { 491 msg_display(MSG_biosmultmatch); 492 msg_table_add(MSG_biosmultmatch_header); 493 for (i = 0; i < nip->ni_nmatches; i++) { 494 bip = &disklist->dl_biosdisks[ 495 nip->ni_biosmatches[i]]; 496 msg_table_add(MSG_biosmultmatch_row, i, 497 bip->bi_dev, bip->bi_cyl, bip->bi_head, 498 bip->bi_sec, (unsigned)bip->bi_lbasecs, 499 (unsigned)bip->bi_lbasecs/(1000000000/512)); 500 } 501 process_menu(MENU_biosmultmatch, &i); 502 if (i == -1) 503 biosdisk = NULL; 504 else 505 biosdisk = &disklist->dl_biosdisks[ 506 nip->ni_biosmatches[i]]; 507 } 508 } 509 if (biosdisk == NULL) { 510 if (nip != NULL) { 511 set_bios_geom(cyl, head, sec); 512 } else { 513 bcyl = cyl; 514 bhead = head; 515 bsec = sec; 516 } 517 } else { 518 bcyl = biosdisk->bi_cyl; 519 bhead = biosdisk->bi_head; 520 bsec = biosdisk->bi_sec; 521 } 522 return 0; 523 } 524 525 static int 526 mbr_root_above_chs(void) 527 { 528 return pm->ptstart + DEFROOTSIZE * (MEG / 512) >= bcyl * bhead * bsec; 529 } 530 531 static void 532 md_upgrade_mbrtype(void) 533 { 534 struct mbr_partition *mbrp; 535 int i, netbsdpart = -1, oldbsdpart = -1, oldbsdcount = 0; 536 537 if (pm->no_mbr) 538 return; 539 540 if (read_mbr(pm->diskdev, &mbr) < 0) 541 return; 542 543 mbrp = &mbr.mbr.mbr_parts[0]; 544 545 for (i = 0; i < MBR_PART_COUNT; i++) { 546 if (mbrp[i].mbrp_type == MBR_PTYPE_386BSD) { 547 oldbsdpart = i; 548 oldbsdcount++; 549 } else if (mbrp[i].mbrp_type == MBR_PTYPE_NETBSD) 550 netbsdpart = i; 551 } 552 553 if (netbsdpart == -1 && oldbsdcount == 1) { 554 mbrp[oldbsdpart].mbrp_type = MBR_PTYPE_NETBSD; 555 write_mbr(pm->diskdev, &mbr, 0); 556 } 557 } 558 559 /* 560 * Read MBR code from a file. 561 * The existing partition table and bootselect configuration is kept. 562 */ 563 static int 564 md_read_bootcode(const char *path, struct mbr_sector *mbrs) 565 { 566 int fd; 567 struct stat st; 568 size_t len; 569 struct mbr_sector new_mbr; 570 uint32_t dsn; 571 572 fd = open(path, O_RDONLY); 573 if (fd < 0) 574 return -1; 575 576 if (fstat(fd, &st) < 0 || st.st_size != sizeof *mbrs) { 577 close(fd); 578 return -1; 579 } 580 581 if (read(fd, &new_mbr, sizeof new_mbr) != sizeof new_mbr) { 582 close(fd); 583 return -1; 584 } 585 close(fd); 586 587 if (new_mbr.mbr_bootsel_magic != htole16(MBR_BS_MAGIC)) 588 return -1; 589 590 if (mbrs->mbr_bootsel_magic == htole16(MBR_BS_MAGIC)) { 591 len = offsetof(struct mbr_sector, mbr_bootsel); 592 } else 593 len = offsetof(struct mbr_sector, mbr_parts); 594 595 /* Preserve the 'drive serial number' - especially for Vista */ 596 dsn = mbrs->mbr_dsn; 597 memcpy(mbrs, &new_mbr, len); 598 mbrs->mbr_dsn = dsn; 599 600 /* Keep flags from object file - indicate the properties */ 601 mbrs->mbr_bootsel.mbrbs_flags = new_mbr.mbr_bootsel.mbrbs_flags; 602 mbrs->mbr_magic = htole16(MBR_MAGIC); 603 604 return 0; 605 } 606 607 static unsigned int 608 get_bootmodel(void) 609 { 610 #if defined(__i386__) 611 struct utsname ut; 612 #ifdef DEBUG 613 char *envstr; 614 615 envstr = getenv("BOOTMODEL"); 616 if (envstr != NULL) 617 return atoi(envstr); 618 #endif 619 620 if (uname(&ut) < 0) 621 ut.version[0] = 0; 622 623 #if defined(SET_KERNEL_TINY) 624 if (strstr(ut.version, "TINY") != NULL) 625 return SET_KERNEL_TINY; 626 #endif 627 #if defined(SET_KERNEL_PS2) 628 if (strstr(ut.version, "PS2") != NULL) 629 return SET_KERNEL_PS2; 630 #endif 631 #endif 632 return SET_KERNEL_GENERIC; 633 } 634 635 636 int 637 md_pre_mount() 638 { 639 return 0; 640 } 641