1 /*- 2 * Copyright (c) 1998 Michael Smith <msmith@freebsd.org> 3 * Copyright (c) 2012 Andrey V. Elsukov <ae@FreeBSD.org> 4 * All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25 * SUCH DAMAGE. 26 */ 27 28 #include <sys/cdefs.h> 29 __FBSDID("$FreeBSD$"); 30 31 /* 32 * BIOS disk device handling. 33 * 34 * Ideas and algorithms from: 35 * 36 * - NetBSD libi386/biosdisk.c 37 * - FreeBSD biosboot/disk.c 38 * 39 */ 40 41 #include <sys/disk.h> 42 #include <sys/limits.h> 43 #include <stand.h> 44 #include <machine/bootinfo.h> 45 #include <stdarg.h> 46 47 #include <bootstrap.h> 48 #include <btxv86.h> 49 #include <edd.h> 50 #include "disk.h" 51 #include "libi386.h" 52 53 #ifdef LOADER_GELI_SUPPORT 54 #include "cons.h" 55 #include "drv.h" 56 #include "gpt.h" 57 #include "part.h" 58 #include <uuid.h> 59 struct pentry { 60 struct ptable_entry part; 61 uint64_t flags; 62 union { 63 uint8_t bsd; 64 uint8_t mbr; 65 uuid_t gpt; 66 uint16_t vtoc8; 67 } type; 68 STAILQ_ENTRY(pentry) entry; 69 }; 70 struct ptable { 71 enum ptable_type type; 72 uint16_t sectorsize; 73 uint64_t sectors; 74 75 STAILQ_HEAD(, pentry) entries; 76 }; 77 78 #include "geliboot.c" 79 #endif /* LOADER_GELI_SUPPORT */ 80 81 CTASSERT(sizeof(struct i386_devdesc) >= sizeof(struct disk_devdesc)); 82 83 #define BIOS_NUMDRIVES 0x475 84 #define BIOSDISK_SECSIZE 512 85 #define BUFSIZE (1 * BIOSDISK_SECSIZE) 86 87 #define DT_ATAPI 0x10 /* disk type for ATAPI floppies */ 88 #define WDMAJOR 0 /* major numbers for devices we frontend for */ 89 #define WFDMAJOR 1 90 #define FDMAJOR 2 91 #define DAMAJOR 4 92 93 #ifdef DISK_DEBUG 94 # define DEBUG(fmt, args...) printf("%s: " fmt "\n" , __func__ , ## args) 95 #else 96 # define DEBUG(fmt, args...) 97 #endif 98 99 /* 100 * List of BIOS devices, translation from disk unit number to 101 * BIOS unit number. 102 */ 103 static struct bdinfo 104 { 105 int bd_unit; /* BIOS unit number */ 106 int bd_cyl; /* BIOS geometry */ 107 int bd_hds; 108 int bd_sec; 109 int bd_flags; 110 #define BD_MODEINT13 0x0000 111 #define BD_MODEEDD1 0x0001 112 #define BD_MODEEDD3 0x0002 113 #define BD_MODEMASK 0x0003 114 #define BD_FLOPPY 0x0004 115 int bd_type; /* BIOS 'drive type' (floppy only) */ 116 uint16_t bd_sectorsize; /* Sector size */ 117 uint64_t bd_sectors; /* Disk size */ 118 int bd_open; /* reference counter */ 119 void *bd_bcache; /* buffer cache data */ 120 } bdinfo [MAXBDDEV]; 121 static int nbdinfo = 0; 122 123 #define BD(dev) (bdinfo[(dev)->d_unit]) 124 125 static int bd_read(struct disk_devdesc *dev, daddr_t dblk, int blks, 126 caddr_t dest); 127 static int bd_write(struct disk_devdesc *dev, daddr_t dblk, int blks, 128 caddr_t dest); 129 static int bd_int13probe(struct bdinfo *bd); 130 131 static int bd_init(void); 132 static int bd_strategy(void *devdata, int flag, daddr_t dblk, size_t size, 133 char *buf, size_t *rsize); 134 static int bd_realstrategy(void *devdata, int flag, daddr_t dblk, size_t size, 135 char *buf, size_t *rsize); 136 static int bd_open(struct open_file *f, ...); 137 static int bd_close(struct open_file *f); 138 static int bd_ioctl(struct open_file *f, u_long cmd, void *data); 139 static int bd_print(int verbose); 140 141 #ifdef LOADER_GELI_SUPPORT 142 enum isgeli { 143 ISGELI_UNKNOWN, 144 ISGELI_NO, 145 ISGELI_YES 146 }; 147 static enum isgeli geli_status[MAXBDDEV][MAXTBLENTS]; 148 149 int bios_read(void *, void *, off_t off, void *buf, size_t bytes); 150 #endif /* LOADER_GELI_SUPPORT */ 151 152 struct devsw biosdisk = { 153 "disk", 154 DEVT_DISK, 155 bd_init, 156 bd_strategy, 157 bd_open, 158 bd_close, 159 bd_ioctl, 160 bd_print, 161 NULL 162 }; 163 164 /* 165 * Translate between BIOS device numbers and our private unit numbers. 166 */ 167 int 168 bd_bios2unit(int biosdev) 169 { 170 int i; 171 172 DEBUG("looking for bios device 0x%x", biosdev); 173 for (i = 0; i < nbdinfo; i++) { 174 DEBUG("bd unit %d is BIOS device 0x%x", i, bdinfo[i].bd_unit); 175 if (bdinfo[i].bd_unit == biosdev) 176 return (i); 177 } 178 return (-1); 179 } 180 181 int 182 bd_unit2bios(int unit) 183 { 184 185 if ((unit >= 0) && (unit < nbdinfo)) 186 return (bdinfo[unit].bd_unit); 187 return (-1); 188 } 189 190 /* 191 * Quiz the BIOS for disk devices, save a little info about them. 192 */ 193 static int 194 bd_init(void) 195 { 196 int base, unit, nfd = 0; 197 198 #ifdef LOADER_GELI_SUPPORT 199 geli_init(); 200 #endif 201 /* sequence 0, 0x80 */ 202 for (base = 0; base <= 0x80; base += 0x80) { 203 for (unit = base; (nbdinfo < MAXBDDEV); unit++) { 204 #ifndef VIRTUALBOX 205 /* 206 * Check the BIOS equipment list for number 207 * of fixed disks. 208 */ 209 if(base == 0x80 && 210 (nfd >= *(unsigned char *)PTOV(BIOS_NUMDRIVES))) 211 break; 212 #endif 213 bdinfo[nbdinfo].bd_open = 0; 214 bdinfo[nbdinfo].bd_bcache = NULL; 215 bdinfo[nbdinfo].bd_unit = unit; 216 bdinfo[nbdinfo].bd_flags = unit < 0x80 ? BD_FLOPPY: 0; 217 if (!bd_int13probe(&bdinfo[nbdinfo])) 218 break; 219 220 /* XXX we need "disk aliases" to make this simpler */ 221 printf("BIOS drive %c: is disk%d\n", (unit < 0x80) ? 222 ('A' + unit): ('C' + unit - 0x80), nbdinfo); 223 nbdinfo++; 224 if (base == 0x80) 225 nfd++; 226 } 227 } 228 bcache_add_dev(nbdinfo); 229 return(0); 230 } 231 232 /* 233 * Try to detect a device supported by the legacy int13 BIOS 234 */ 235 static int 236 bd_int13probe(struct bdinfo *bd) 237 { 238 struct edd_params params; 239 int ret = 1; /* assume success */ 240 241 v86.ctl = V86_FLAGS; 242 v86.addr = 0x13; 243 v86.eax = 0x800; 244 v86.edx = bd->bd_unit; 245 v86int(); 246 247 /* Don't error out if we get bad sector number, try EDD as well */ 248 if (V86_CY(v86.efl) || /* carry set */ 249 (v86.edx & 0xff) <= (unsigned)(bd->bd_unit & 0x7f)) /* unit # bad */ 250 return (0); /* skip device */ 251 252 if ((v86.ecx & 0x3f) == 0) /* absurd sector number */ 253 ret = 0; /* set error */ 254 255 /* Convert max cyl # -> # of cylinders */ 256 bd->bd_cyl = ((v86.ecx & 0xc0) << 2) + ((v86.ecx & 0xff00) >> 8) + 1; 257 /* Convert max head # -> # of heads */ 258 bd->bd_hds = ((v86.edx & 0xff00) >> 8) + 1; 259 bd->bd_sec = v86.ecx & 0x3f; 260 bd->bd_type = v86.ebx & 0xff; 261 bd->bd_flags |= BD_MODEINT13; 262 263 /* Calculate sectors count from the geometry */ 264 bd->bd_sectors = bd->bd_cyl * bd->bd_hds * bd->bd_sec; 265 bd->bd_sectorsize = BIOSDISK_SECSIZE; 266 DEBUG("unit 0x%x geometry %d/%d/%d", bd->bd_unit, bd->bd_cyl, 267 bd->bd_hds, bd->bd_sec); 268 269 /* Determine if we can use EDD with this device. */ 270 v86.ctl = V86_FLAGS; 271 v86.addr = 0x13; 272 v86.eax = 0x4100; 273 v86.edx = bd->bd_unit; 274 v86.ebx = 0x55aa; 275 v86int(); 276 if (V86_CY(v86.efl) || /* carry set */ 277 (v86.ebx & 0xffff) != 0xaa55 || /* signature */ 278 (v86.ecx & EDD_INTERFACE_FIXED_DISK) == 0) 279 return (ret); /* return code from int13 AH=08 */ 280 281 /* EDD supported */ 282 bd->bd_flags |= BD_MODEEDD1; 283 if ((v86.eax & 0xff00) >= 0x3000) 284 bd->bd_flags |= BD_MODEEDD3; 285 /* Get disk params */ 286 params.len = sizeof(struct edd_params); 287 v86.ctl = V86_FLAGS; 288 v86.addr = 0x13; 289 v86.eax = 0x4800; 290 v86.edx = bd->bd_unit; 291 v86.ds = VTOPSEG(¶ms); 292 v86.esi = VTOPOFF(¶ms); 293 v86int(); 294 if (!V86_CY(v86.efl)) { 295 uint64_t total; 296 297 /* 298 * Sector size must be a multiple of 512 bytes. 299 * An alternate test would be to check power of 2, 300 * powerof2(params.sector_size). 301 */ 302 if (params.sector_size % BIOSDISK_SECSIZE) 303 bd->bd_sectorsize = BIOSDISK_SECSIZE; 304 else 305 bd->bd_sectorsize = params.sector_size; 306 307 total = bd->bd_sectorsize * params.sectors; 308 if (params.sectors != 0) { 309 /* Only update if we did not overflow. */ 310 if (total > params.sectors) 311 bd->bd_sectors = params.sectors; 312 } 313 314 total = (uint64_t)params.cylinders * 315 params.heads * params.sectors_per_track; 316 if (bd->bd_sectors < total) 317 bd->bd_sectors = total; 318 319 ret = 1; 320 } 321 DEBUG("unit 0x%x flags %x, sectors %llu, sectorsize %u", 322 bd->bd_unit, bd->bd_flags, bd->bd_sectors, bd->bd_sectorsize); 323 return (ret); 324 } 325 326 /* 327 * Print information about disks 328 */ 329 static int 330 bd_print(int verbose) 331 { 332 static char line[80]; 333 struct disk_devdesc dev; 334 int i, ret = 0; 335 336 if (nbdinfo == 0) 337 return (0); 338 339 printf("%s devices:", biosdisk.dv_name); 340 if ((ret = pager_output("\n")) != 0) 341 return (ret); 342 343 for (i = 0; i < nbdinfo; i++) { 344 snprintf(line, sizeof(line), 345 " disk%d: BIOS drive %c (%ju X %u):\n", i, 346 (bdinfo[i].bd_unit < 0x80) ? ('A' + bdinfo[i].bd_unit): 347 ('C' + bdinfo[i].bd_unit - 0x80), 348 (uintmax_t)bdinfo[i].bd_sectors, 349 bdinfo[i].bd_sectorsize); 350 if ((ret = pager_output(line)) != 0) 351 break; 352 dev.d_dev = &biosdisk; 353 dev.d_unit = i; 354 dev.d_slice = -1; 355 dev.d_partition = -1; 356 if (disk_open(&dev, 357 bdinfo[i].bd_sectorsize * bdinfo[i].bd_sectors, 358 bdinfo[i].bd_sectorsize) == 0) { 359 snprintf(line, sizeof(line), " disk%d", i); 360 ret = disk_print(&dev, line, verbose); 361 disk_close(&dev); 362 if (ret != 0) 363 return (ret); 364 } 365 } 366 return (ret); 367 } 368 369 /* 370 * Attempt to open the disk described by (dev) for use by (f). 371 * 372 * Note that the philosophy here is "give them exactly what 373 * they ask for". This is necessary because being too "smart" 374 * about what the user might want leads to complications. 375 * (eg. given no slice or partition value, with a disk that is 376 * sliced - are they after the first BSD slice, or the DOS 377 * slice before it?) 378 */ 379 static int 380 bd_open(struct open_file *f, ...) 381 { 382 struct disk_devdesc *dev, rdev; 383 struct disk_devdesc disk; 384 int err, g_err; 385 va_list ap; 386 uint64_t size; 387 388 va_start(ap, f); 389 dev = va_arg(ap, struct disk_devdesc *); 390 va_end(ap); 391 392 if (dev->d_unit < 0 || dev->d_unit >= nbdinfo) 393 return (EIO); 394 BD(dev).bd_open++; 395 if (BD(dev).bd_bcache == NULL) 396 BD(dev).bd_bcache = bcache_allocate(); 397 398 /* 399 * Read disk size from partition. 400 * This is needed to work around buggy BIOS systems returning 401 * wrong (truncated) disk media size. 402 * During bd_probe() we tested if the mulitplication of bd_sectors 403 * would overflow so it should be safe to perform here. 404 */ 405 disk.d_dev = dev->d_dev; 406 disk.d_type = dev->d_type; 407 disk.d_unit = dev->d_unit; 408 disk.d_opendata = NULL; 409 disk.d_slice = -1; 410 disk.d_partition = -1; 411 disk.d_offset = 0; 412 if (disk_open(&disk, BD(dev).bd_sectors * BD(dev).bd_sectorsize, 413 BD(dev).bd_sectorsize) == 0) { 414 415 if (disk_ioctl(&disk, DIOCGMEDIASIZE, &size) == 0) { 416 size /= BD(dev).bd_sectorsize; 417 if (size > BD(dev).bd_sectors) 418 BD(dev).bd_sectors = size; 419 } 420 disk_close(&disk); 421 } 422 423 err = disk_open(dev, BD(dev).bd_sectors * BD(dev).bd_sectorsize, 424 BD(dev).bd_sectorsize); 425 426 #ifdef LOADER_GELI_SUPPORT 427 static char gelipw[GELI_PW_MAXLEN]; 428 char *passphrase; 429 430 if (err) 431 return (err); 432 433 /* if we already know there is no GELI, skip the rest */ 434 if (geli_status[dev->d_unit][dev->d_slice] != ISGELI_UNKNOWN) 435 return (err); 436 437 struct dsk dskp; 438 struct ptable *table = NULL; 439 struct ptable_entry part; 440 struct pentry *entry; 441 int geli_part = 0; 442 443 dskp.drive = bd_unit2bios(dev->d_unit); 444 dskp.type = dev->d_type; 445 dskp.unit = dev->d_unit; 446 dskp.slice = dev->d_slice; 447 dskp.part = dev->d_partition; 448 dskp.start = dev->d_offset; 449 450 memcpy(&rdev, dev, sizeof(rdev)); 451 /* to read the GPT table, we need to read the first sector */ 452 rdev.d_offset = 0; 453 /* We need the LBA of the end of the partition */ 454 table = ptable_open(&rdev, BD(dev).bd_sectors, 455 BD(dev).bd_sectorsize, ptblread); 456 if (table == NULL) { 457 DEBUG("Can't read partition table"); 458 /* soft failure, return the exit status of disk_open */ 459 return (err); 460 } 461 462 if (table->type == PTABLE_GPT) 463 dskp.part = 255; 464 465 STAILQ_FOREACH(entry, &table->entries, entry) { 466 dskp.slice = entry->part.index; 467 dskp.start = entry->part.start; 468 if (is_geli(&dskp) == 0) { 469 geli_status[dev->d_unit][dskp.slice] = ISGELI_YES; 470 return (0); 471 } 472 if (geli_taste(bios_read, &dskp, 473 entry->part.end - entry->part.start) == 0) { 474 if (geli_havekey(&dskp) == 0) { 475 geli_status[dev->d_unit][dskp.slice] = ISGELI_YES; 476 geli_part++; 477 continue; 478 } 479 if ((passphrase = getenv("kern.geom.eli.passphrase")) 480 != NULL) { 481 /* Use the cached passphrase */ 482 bcopy(passphrase, &gelipw, GELI_PW_MAXLEN); 483 } 484 if (geli_passphrase(gelipw, dskp.unit, 'p', 485 (dskp.slice > 0 ? dskp.slice : dskp.part), 486 &dskp) == 0) { 487 setenv("kern.geom.eli.passphrase", gelipw, 1); 488 bzero(gelipw, sizeof(gelipw)); 489 geli_status[dev->d_unit][dskp.slice] = ISGELI_YES; 490 geli_part++; 491 continue; 492 } 493 } else 494 geli_status[dev->d_unit][dskp.slice] = ISGELI_NO; 495 } 496 497 /* none of the partitions on this disk have GELI */ 498 if (geli_part == 0) { 499 /* found no GELI */ 500 geli_status[dev->d_unit][dev->d_slice] = ISGELI_NO; 501 } 502 #endif /* LOADER_GELI_SUPPORT */ 503 504 return (err); 505 } 506 507 static int 508 bd_close(struct open_file *f) 509 { 510 struct disk_devdesc *dev; 511 512 dev = (struct disk_devdesc *)f->f_devdata; 513 BD(dev).bd_open--; 514 if (BD(dev).bd_open == 0) { 515 bcache_free(BD(dev).bd_bcache); 516 BD(dev).bd_bcache = NULL; 517 } 518 return (disk_close(dev)); 519 } 520 521 static int 522 bd_ioctl(struct open_file *f, u_long cmd, void *data) 523 { 524 struct disk_devdesc *dev; 525 int rc; 526 527 dev = (struct disk_devdesc *)f->f_devdata; 528 529 rc = disk_ioctl(dev, cmd, data); 530 if (rc != ENOTTY) 531 return (rc); 532 533 switch (cmd) { 534 case DIOCGSECTORSIZE: 535 *(u_int *)data = BD(dev).bd_sectorsize; 536 break; 537 case DIOCGMEDIASIZE: 538 *(uint64_t *)data = BD(dev).bd_sectors * BD(dev).bd_sectorsize; 539 break; 540 default: 541 return (ENOTTY); 542 } 543 return (0); 544 } 545 546 static int 547 bd_strategy(void *devdata, int rw, daddr_t dblk, size_t size, 548 char *buf, size_t *rsize) 549 { 550 struct bcache_devdata bcd; 551 struct disk_devdesc *dev; 552 553 dev = (struct disk_devdesc *)devdata; 554 bcd.dv_strategy = bd_realstrategy; 555 bcd.dv_devdata = devdata; 556 bcd.dv_cache = BD(dev).bd_bcache; 557 return (bcache_strategy(&bcd, rw, dblk + dev->d_offset, 558 size, buf, rsize)); 559 } 560 561 static int 562 bd_realstrategy(void *devdata, int rw, daddr_t dblk, size_t size, 563 char *buf, size_t *rsize) 564 { 565 struct disk_devdesc *dev = (struct disk_devdesc *)devdata; 566 uint64_t disk_blocks; 567 int blks, rc; 568 #ifdef BD_SUPPORT_FRAGS /* XXX: sector size */ 569 char fragbuf[BIOSDISK_SECSIZE]; 570 size_t fragsize; 571 572 fragsize = size % BIOSDISK_SECSIZE; 573 #else 574 if (size % BD(dev).bd_sectorsize) 575 panic("bd_strategy: %d bytes I/O not multiple of block size", size); 576 #endif 577 578 DEBUG("open_disk %p", dev); 579 580 /* 581 * Check the value of the size argument. We do have quite small 582 * heap (64MB), but we do not know good upper limit, so we check against 583 * INT_MAX here. This will also protect us against possible overflows 584 * while translating block count to bytes. 585 */ 586 if (size > INT_MAX) { 587 DEBUG("too large read: %zu bytes", size); 588 return (EIO); 589 } 590 591 blks = size / BD(dev).bd_sectorsize; 592 if (dblk > dblk + blks) 593 return (EIO); 594 595 if (rsize) 596 *rsize = 0; 597 598 /* Get disk blocks, this value is either for whole disk or for partition */ 599 if (disk_ioctl(dev, DIOCGMEDIASIZE, &disk_blocks)) { 600 /* DIOCGMEDIASIZE does return bytes. */ 601 disk_blocks /= BD(dev).bd_sectorsize; 602 } else { 603 /* We should not get here. Just try to survive. */ 604 disk_blocks = BD(dev).bd_sectors - dev->d_offset; 605 } 606 607 /* Validate source block address. */ 608 if (dblk < dev->d_offset || dblk >= dev->d_offset + disk_blocks) 609 return (EIO); 610 611 /* 612 * Truncate if we are crossing disk or partition end. 613 */ 614 if (dblk + blks >= dev->d_offset + disk_blocks) { 615 blks = dev->d_offset + disk_blocks - dblk; 616 size = blks * BD(dev).bd_sectorsize; 617 DEBUG("short read %d", blks); 618 } 619 620 switch (rw & F_MASK) { 621 case F_READ: 622 DEBUG("read %d from %lld to %p", blks, dblk, buf); 623 624 if (blks && (rc = bd_read(dev, dblk, blks, buf))) { 625 /* Filter out floppy controller errors */ 626 if (BD(dev).bd_flags != BD_FLOPPY || rc != 0x20) { 627 printf("read %d from %lld to %p, error: 0x%x", blks, dblk, 628 buf, rc); 629 } 630 return (EIO); 631 } 632 #ifdef BD_SUPPORT_FRAGS /* XXX: sector size */ 633 DEBUG("bd_strategy: frag read %d from %d+%d to %p", 634 fragsize, dblk, blks, buf + (blks * BIOSDISK_SECSIZE)); 635 if (fragsize && bd_read(od, dblk + blks, 1, fragsize)) { 636 DEBUG("frag read error"); 637 return(EIO); 638 } 639 bcopy(fragbuf, buf + (blks * BIOSDISK_SECSIZE), fragsize); 640 #endif 641 break; 642 case F_WRITE : 643 DEBUG("write %d from %d to %p", blks, dblk, buf); 644 645 if (blks && bd_write(dev, dblk, blks, buf)) { 646 DEBUG("write error"); 647 return (EIO); 648 } 649 #ifdef BD_SUPPORT_FRAGS 650 if(fragsize) { 651 DEBUG("Attempted to write a frag"); 652 return (EIO); 653 } 654 #endif 655 break; 656 default: 657 /* DO NOTHING */ 658 return (EROFS); 659 } 660 661 if (rsize) 662 *rsize = size; 663 return (0); 664 } 665 666 static int 667 bd_edd_io(struct disk_devdesc *dev, daddr_t dblk, int blks, caddr_t dest, 668 int write) 669 { 670 static struct edd_packet packet; 671 672 packet.len = sizeof(struct edd_packet); 673 packet.count = blks; 674 packet.off = VTOPOFF(dest); 675 packet.seg = VTOPSEG(dest); 676 packet.lba = dblk; 677 v86.ctl = V86_FLAGS; 678 v86.addr = 0x13; 679 if (write) 680 /* Should we Write with verify ?? 0x4302 ? */ 681 v86.eax = 0x4300; 682 else 683 v86.eax = 0x4200; 684 v86.edx = BD(dev).bd_unit; 685 v86.ds = VTOPSEG(&packet); 686 v86.esi = VTOPOFF(&packet); 687 v86int(); 688 if (V86_CY(v86.efl)) 689 return (v86.eax >> 8); 690 return (0); 691 } 692 693 static int 694 bd_chs_io(struct disk_devdesc *dev, daddr_t dblk, int blks, caddr_t dest, 695 int write) 696 { 697 u_int x, bpc, cyl, hd, sec; 698 699 bpc = BD(dev).bd_sec * BD(dev).bd_hds; /* blocks per cylinder */ 700 x = dblk; 701 cyl = x / bpc; /* block # / blocks per cylinder */ 702 x %= bpc; /* block offset into cylinder */ 703 hd = x / BD(dev).bd_sec; /* offset / blocks per track */ 704 sec = x % BD(dev).bd_sec; /* offset into track */ 705 706 /* correct sector number for 1-based BIOS numbering */ 707 sec++; 708 709 if (cyl > 1023) 710 /* CHS doesn't support cylinders > 1023. */ 711 return (1); 712 713 v86.ctl = V86_FLAGS; 714 v86.addr = 0x13; 715 if (write) 716 v86.eax = 0x300 | blks; 717 else 718 v86.eax = 0x200 | blks; 719 v86.ecx = ((cyl & 0xff) << 8) | ((cyl & 0x300) >> 2) | sec; 720 v86.edx = (hd << 8) | BD(dev).bd_unit; 721 v86.es = VTOPSEG(dest); 722 v86.ebx = VTOPOFF(dest); 723 v86int(); 724 if (V86_CY(v86.efl)) 725 return (v86.eax >> 8); 726 return (0); 727 } 728 729 static int 730 bd_io(struct disk_devdesc *dev, daddr_t dblk, int blks, caddr_t dest, int write) 731 { 732 u_int x, sec, result, resid, retry, maxfer; 733 caddr_t p, xp, bbuf; 734 735 /* Just in case some idiot actually tries to read/write -1 blocks... */ 736 if (blks < 0) 737 return (-1); 738 739 resid = blks; 740 p = dest; 741 742 /* Decide whether we have to bounce */ 743 if (VTOP(dest) >> 20 != 0 || (BD(dev).bd_unit < 0x80 && 744 (VTOP(dest) >> 16) != (VTOP(dest + 745 blks * BD(dev).bd_sectorsize) >> 16))) { 746 747 /* 748 * There is a 64k physical boundary somewhere in the 749 * destination buffer, or the destination buffer is above 750 * first 1MB of physical memory so we have to arrange a 751 * suitable bounce buffer. Allocate a buffer twice as large 752 * as we need to. Use the bottom half unless there is a break 753 * there, in which case we use the top half. 754 */ 755 x = V86_IO_BUFFER_SIZE / BD(dev).bd_sectorsize; 756 x = min(x, (unsigned)blks); 757 bbuf = PTOV(V86_IO_BUFFER); 758 maxfer = x; /* limit transfers to bounce region size */ 759 } else { 760 bbuf = NULL; 761 maxfer = 0; 762 } 763 764 while (resid > 0) { 765 /* 766 * Play it safe and don't cross track boundaries. 767 * (XXX this is probably unnecessary) 768 */ 769 sec = dblk % BD(dev).bd_sec; /* offset into track */ 770 x = min(BD(dev).bd_sec - sec, resid); 771 if (maxfer > 0) 772 x = min(x, maxfer); /* fit bounce buffer */ 773 774 /* where do we transfer to? */ 775 xp = bbuf == NULL ? p : bbuf; 776 777 /* 778 * Put your Data In, Put your Data out, 779 * Put your Data In, and shake it all about 780 */ 781 if (write && bbuf != NULL) 782 bcopy(p, bbuf, x * BD(dev).bd_sectorsize); 783 784 /* 785 * Loop retrying the operation a couple of times. The BIOS 786 * may also retry. 787 */ 788 for (retry = 0; retry < 3; retry++) { 789 /* if retrying, reset the drive */ 790 if (retry > 0) { 791 v86.ctl = V86_FLAGS; 792 v86.addr = 0x13; 793 v86.eax = 0; 794 v86.edx = BD(dev).bd_unit; 795 v86int(); 796 } 797 798 if (BD(dev).bd_flags & BD_MODEEDD1) 799 result = bd_edd_io(dev, dblk, x, xp, write); 800 else 801 result = bd_chs_io(dev, dblk, x, xp, write); 802 if (result == 0) 803 break; 804 } 805 806 if (write) 807 DEBUG("Write %d sector(s) from %p (0x%x) to %lld %s", x, 808 p, VTOP(p), dblk, result ? "failed" : "ok"); 809 else 810 DEBUG("Read %d sector(s) from %lld to %p (0x%x) %s", x, 811 dblk, p, VTOP(p), result ? "failed" : "ok"); 812 if (result) { 813 return (result); 814 } 815 if (!write && bbuf != NULL) 816 bcopy(bbuf, p, x * BD(dev).bd_sectorsize); 817 p += (x * BD(dev).bd_sectorsize); 818 dblk += x; 819 resid -= x; 820 } 821 822 /* hexdump(dest, (blks * BD(dev).bd_sectorsize)); */ 823 return(0); 824 } 825 826 static int 827 bd_read(struct disk_devdesc *dev, daddr_t dblk, int blks, caddr_t dest) 828 { 829 #ifdef LOADER_GELI_SUPPORT 830 struct dsk dskp; 831 off_t p_off, diff; 832 daddr_t alignlba; 833 int err, n, alignblks; 834 char *tmpbuf; 835 836 /* if we already know there is no GELI, skip the rest */ 837 if (geli_status[dev->d_unit][dev->d_slice] != ISGELI_YES) 838 return (bd_io(dev, dblk, blks, dest, 0)); 839 840 if (geli_status[dev->d_unit][dev->d_slice] == ISGELI_YES) { 841 /* 842 * Align reads to DEV_GELIBOOT_BSIZE bytes because partial 843 * sectors cannot be decrypted. Round the requested LBA down to 844 * nearest multiple of DEV_GELIBOOT_BSIZE bytes. 845 */ 846 alignlba = rounddown2(dblk * BD(dev).bd_sectorsize, 847 DEV_GELIBOOT_BSIZE) / BD(dev).bd_sectorsize; 848 /* 849 * Round number of blocks to read up to nearest multiple of 850 * DEV_GELIBOOT_BSIZE 851 */ 852 diff = (dblk - alignlba) * BD(dev).bd_sectorsize; 853 alignblks = roundup2(blks * BD(dev).bd_sectorsize + diff, 854 DEV_GELIBOOT_BSIZE) / BD(dev).bd_sectorsize; 855 856 /* 857 * If the read is rounded up to a larger size, use a temporary 858 * buffer here because the buffer provided by the caller may be 859 * too small. 860 */ 861 if (diff == 0) { 862 tmpbuf = dest; 863 } else { 864 tmpbuf = malloc(alignblks * BD(dev).bd_sectorsize); 865 if (tmpbuf == NULL) { 866 return (-1); 867 } 868 } 869 870 err = bd_io(dev, alignlba, alignblks, tmpbuf, 0); 871 if (err) 872 return (err); 873 874 dskp.drive = bd_unit2bios(dev->d_unit); 875 dskp.type = dev->d_type; 876 dskp.unit = dev->d_unit; 877 dskp.slice = dev->d_slice; 878 dskp.part = dev->d_partition; 879 dskp.start = dev->d_offset; 880 881 /* GELI needs the offset relative to the partition start */ 882 p_off = alignlba - dskp.start; 883 884 err = geli_read(&dskp, p_off * BD(dev).bd_sectorsize, (u_char *)tmpbuf, 885 alignblks * BD(dev).bd_sectorsize); 886 if (err) 887 return (err); 888 889 if (tmpbuf != dest) { 890 bcopy(tmpbuf + diff, dest, blks * BD(dev).bd_sectorsize); 891 free(tmpbuf); 892 } 893 return (0); 894 } 895 #endif /* LOADER_GELI_SUPPORT */ 896 897 return (bd_io(dev, dblk, blks, dest, 0)); 898 } 899 900 static int 901 bd_write(struct disk_devdesc *dev, daddr_t dblk, int blks, caddr_t dest) 902 { 903 904 return (bd_io(dev, dblk, blks, dest, 1)); 905 } 906 907 /* 908 * Return the BIOS geometry of a given "fixed drive" in a format 909 * suitable for the legacy bootinfo structure. Since the kernel is 910 * expecting raw int 0x13/0x8 values for N_BIOS_GEOM drives, we 911 * prefer to get the information directly, rather than rely on being 912 * able to put it together from information already maintained for 913 * different purposes and for a probably different number of drives. 914 * 915 * For valid drives, the geometry is expected in the format (31..0) 916 * "000000cc cccccccc hhhhhhhh 00ssssss"; and invalid drives are 917 * indicated by returning the geometry of a "1.2M" PC-format floppy 918 * disk. And, incidentally, what is returned is not the geometry as 919 * such but the highest valid cylinder, head, and sector numbers. 920 */ 921 u_int32_t 922 bd_getbigeom(int bunit) 923 { 924 925 v86.ctl = V86_FLAGS; 926 v86.addr = 0x13; 927 v86.eax = 0x800; 928 v86.edx = 0x80 + bunit; 929 v86int(); 930 if (V86_CY(v86.efl)) 931 return 0x4f010f; 932 return ((v86.ecx & 0xc0) << 18) | ((v86.ecx & 0xff00) << 8) | 933 (v86.edx & 0xff00) | (v86.ecx & 0x3f); 934 } 935 936 /* 937 * Return a suitable dev_t value for (dev). 938 * 939 * In the case where it looks like (dev) is a SCSI disk, we allow the number of 940 * IDE disks to be specified in $num_ide_disks. There should be a Better Way. 941 */ 942 int 943 bd_getdev(struct i386_devdesc *d) 944 { 945 struct disk_devdesc *dev; 946 int biosdev; 947 int major; 948 int rootdev; 949 char *nip, *cp; 950 int i, unit; 951 952 dev = (struct disk_devdesc *)d; 953 biosdev = bd_unit2bios(dev->d_unit); 954 DEBUG("unit %d BIOS device %d", dev->d_unit, biosdev); 955 if (biosdev == -1) /* not a BIOS device */ 956 return(-1); 957 if (disk_open(dev, BD(dev).bd_sectors * BD(dev).bd_sectorsize, 958 BD(dev).bd_sectorsize) != 0) /* oops, not a viable device */ 959 return (-1); 960 else 961 disk_close(dev); 962 963 if (biosdev < 0x80) { 964 /* floppy (or emulated floppy) or ATAPI device */ 965 if (bdinfo[dev->d_unit].bd_type == DT_ATAPI) { 966 /* is an ATAPI disk */ 967 major = WFDMAJOR; 968 } else { 969 /* is a floppy disk */ 970 major = FDMAJOR; 971 } 972 } else { 973 /* assume an IDE disk */ 974 major = WDMAJOR; 975 } 976 /* default root disk unit number */ 977 unit = biosdev & 0x7f; 978 979 /* XXX a better kludge to set the root disk unit number */ 980 if ((nip = getenv("root_disk_unit")) != NULL) { 981 i = strtol(nip, &cp, 0); 982 /* check for parse error */ 983 if ((cp != nip) && (*cp == 0)) 984 unit = i; 985 } 986 987 rootdev = MAKEBOOTDEV(major, dev->d_slice + 1, unit, dev->d_partition); 988 DEBUG("dev is 0x%x\n", rootdev); 989 return(rootdev); 990 } 991 992 #ifdef LOADER_GELI_SUPPORT 993 int 994 bios_read(void *vdev __unused, void *xpriv, off_t off, void *buf, size_t bytes) 995 { 996 struct disk_devdesc dev; 997 struct dsk *priv = xpriv; 998 999 dev.d_dev = &biosdisk; 1000 dev.d_type = priv->type; 1001 dev.d_unit = priv->unit; 1002 dev.d_slice = priv->slice; 1003 dev.d_partition = priv->part; 1004 dev.d_offset = priv->start; 1005 1006 off = off / BD(&dev).bd_sectorsize; 1007 /* GELI gives us the offset relative to the partition start */ 1008 off += dev.d_offset; 1009 bytes = bytes / BD(&dev).bd_sectorsize; 1010 1011 return (bd_io(&dev, off, bytes, buf, 0)); 1012 } 1013 #endif /* LOADER_GELI_SUPPORT */ 1014