zfsboot.c revision 329100
1185029Spjd/*- 2185029Spjd * Copyright (c) 1998 Robert Nordier 3185029Spjd * All rights reserved. 4185029Spjd * 5185029Spjd * Redistribution and use in source and binary forms are freely 6185029Spjd * permitted provided that the above copyright notice and this 7185029Spjd * paragraph and the following disclaimer are duplicated in all 8185029Spjd * such forms. 9185029Spjd * 10185029Spjd * This software is provided "AS IS" and without any express or 11185029Spjd * implied warranties, including, without limitation, the implied 12185029Spjd * warranties of merchantability and fitness for a particular 13185029Spjd * purpose. 14185029Spjd */ 15185029Spjd 16185029Spjd#include <sys/cdefs.h> 17185029Spjd__FBSDID("$FreeBSD: stable/11/sys/boot/i386/zfsboot/zfsboot.c 329100 2018-02-10 04:56:07Z kevans $"); 18185029Spjd 19185029Spjd#include <sys/param.h> 20185029Spjd#include <sys/errno.h> 21185029Spjd#include <sys/diskmbr.h> 22185096Sdfr#ifdef GPT 23185096Sdfr#include <sys/gpt.h> 24185096Sdfr#endif 25185029Spjd#include <sys/reboot.h> 26185029Spjd#include <sys/queue.h> 27185029Spjd 28185029Spjd#include <machine/bootinfo.h> 29185029Spjd#include <machine/elf.h> 30200309Sjhb#include <machine/pc/bios.h> 31185029Spjd 32185029Spjd#include <stdarg.h> 33185029Spjd#include <stddef.h> 34185029Spjd 35185029Spjd#include <a.out.h> 36185029Spjd 37185029Spjd#include <btxv86.h> 38185029Spjd 39185029Spjd#include "lib.h" 40213136Spjd#include "rbx.h" 41213136Spjd#include "drv.h" 42329100Skevans#include "edd.h" 43213136Spjd#include "util.h" 44213136Spjd#include "cons.h" 45235154Savg#include "bootargs.h" 46294765Simp#include "paths.h" 47185029Spjd 48235329Savg#include "libzfs.h" 49235329Savg 50297629Sallanjude#define ARGS 0x900 51297629Sallanjude#define NOPT 14 52297629Sallanjude#define NDEV 3 53185029Spjd 54297629Sallanjude#define BIOS_NUMDRIVES 0x475 55297629Sallanjude#define DRV_HARD 0x80 56297629Sallanjude#define DRV_MASK 0x7f 57185029Spjd 58297629Sallanjude#define TYPE_AD 0 59297629Sallanjude#define TYPE_DA 1 60297629Sallanjude#define TYPE_MAXHARD TYPE_DA 61297629Sallanjude#define TYPE_FD 2 62185029Spjd 63297629Sallanjude#define DEV_GELIBOOT_BSIZE 4096 64297629Sallanjude 65185029Spjdextern uint32_t _end; 66185029Spjd 67185096Sdfr#ifdef GPT 68185096Sdfrstatic const uuid_t freebsd_zfs_uuid = GPT_ENT_TYPE_FREEBSD_ZFS; 69185096Sdfr#endif 70185029Spjdstatic const char optstr[NOPT] = "DhaCcdgmnpqrsv"; /* Also 'P', 'S' */ 71185029Spjdstatic const unsigned char flags[NOPT] = { 72185029Spjd RBX_DUAL, 73185029Spjd RBX_SERIAL, 74185029Spjd RBX_ASKNAME, 75185029Spjd RBX_CDROM, 76185029Spjd RBX_CONFIG, 77185029Spjd RBX_KDB, 78185029Spjd RBX_GDB, 79185029Spjd RBX_MUTE, 80185029Spjd RBX_NOINTR, 81185029Spjd RBX_PAUSE, 82185029Spjd RBX_QUIET, 83185029Spjd RBX_DFLTROOT, 84185029Spjd RBX_SINGLE, 85185029Spjd RBX_VERBOSE 86185029Spjd}; 87213136Spjduint32_t opts; 88185029Spjd 89185029Spjdstatic const unsigned char dev_maj[NDEV] = {30, 4, 2}; 90185029Spjd 91185029Spjdstatic char cmd[512]; 92234339Savgstatic char cmddup[512]; 93185029Spjdstatic char kname[1024]; 94235329Savgstatic char rootname[256]; 95185029Spjdstatic int comspeed = SIOSPD; 96185029Spjdstatic struct bootinfo bootinfo; 97185029Spjdstatic uint32_t bootdev; 98235329Savgstatic struct zfs_boot_args zfsargs; 99235329Savgstatic struct zfsmount zfsmount; 100185029Spjd 101200309Sjhbvm_offset_t high_heap_base; 102200309Sjhbuint32_t bios_basemem, bios_extmem, high_heap_size; 103200309Sjhb 104200309Sjhbstatic struct bios_smap smap; 105200309Sjhb 106200309Sjhb/* 107200309Sjhb * The minimum amount of memory to reserve in bios_extmem for the heap. 108200309Sjhb */ 109328866Skevans#define HEAP_MIN (64 * 1024 * 1024) 110200309Sjhb 111200309Sjhbstatic char *heap_next; 112200309Sjhbstatic char *heap_end; 113200309Sjhb 114185029Spjd/* Buffers that must not span a 64k boundary. */ 115297629Sallanjude#define READ_BUF_SIZE 8192 116185029Spjdstruct dmadat { 117185029Spjd char rdbuf[READ_BUF_SIZE]; /* for reading large things */ 118185029Spjd char secbuf[READ_BUF_SIZE]; /* for MBR/disklabel */ 119185029Spjd}; 120185029Spjdstatic struct dmadat *dmadat; 121185029Spjd 122185029Spjdvoid exit(int); 123308914Savgvoid reboot(void); 124185029Spjdstatic void load(void); 125329099Skevansstatic int parse_cmd(void); 126200309Sjhbstatic void bios_getmem(void); 127296963Sallanjudevoid *malloc(size_t n); 128296963Sallanjudevoid free(void *ptr); 129329100Skevansint main(void); 130185029Spjd 131296963Sallanjudevoid * 132185029Spjdmalloc(size_t n) 133185029Spjd{ 134185029Spjd char *p = heap_next; 135185029Spjd if (p + n > heap_end) { 136185029Spjd printf("malloc failure\n"); 137185029Spjd for (;;) 138185029Spjd ; 139296963Sallanjude /* NOTREACHED */ 140296963Sallanjude return (0); 141185029Spjd } 142185029Spjd heap_next += n; 143296963Sallanjude return (p); 144185029Spjd} 145185029Spjd 146296963Sallanjudevoid 147296963Sallanjudefree(void *ptr) 148296963Sallanjude{ 149296963Sallanjude 150296963Sallanjude return; 151296963Sallanjude} 152296963Sallanjude 153185029Spjdstatic char * 154185029Spjdstrdup(const char *s) 155185029Spjd{ 156185029Spjd char *p = malloc(strlen(s) + 1); 157185029Spjd strcpy(p, s); 158296963Sallanjude return (p); 159185029Spjd} 160185029Spjd 161296963Sallanjude#ifdef LOADER_GELI_SUPPORT 162296963Sallanjude#include "geliboot.c" 163296963Sallanjudestatic char gelipw[GELI_PW_MAXLEN]; 164329099Skevansstatic struct keybuf *gelibuf; 165296963Sallanjude#endif 166296963Sallanjude 167185029Spjd#include "zfsimpl.c" 168185029Spjd 169185029Spjd/* 170185029Spjd * Read from a dnode (which must be from a ZPL filesystem). 171185029Spjd */ 172185029Spjdstatic int 173185029Spjdzfs_read(spa_t *spa, const dnode_phys_t *dnode, off_t *offp, void *start, size_t size) 174185029Spjd{ 175185029Spjd const znode_phys_t *zp = (const znode_phys_t *) dnode->dn_bonus; 176185029Spjd size_t n; 177185029Spjd int rc; 178185029Spjd 179185029Spjd n = size; 180185029Spjd if (*offp + n > zp->zp_size) 181185029Spjd n = zp->zp_size - *offp; 182308914Savg 183185029Spjd rc = dnode_read(spa, dnode, *offp, start, n); 184185029Spjd if (rc) 185185029Spjd return (-1); 186185029Spjd *offp += n; 187185029Spjd 188185029Spjd return (n); 189185029Spjd} 190185029Spjd 191185029Spjd/* 192185029Spjd * Current ZFS pool 193185029Spjd */ 194235329Savgstatic spa_t *spa; 195241293Savgstatic spa_t *primary_spa; 196241293Savgstatic vdev_t *primary_vdev; 197185029Spjd 198185029Spjd/* 199185029Spjd * A wrapper for dskread that doesn't have to worry about whether the 200185029Spjd * buffer pointer crosses a 64k boundary. 201185029Spjd */ 202185029Spjdstatic int 203185029Spjdvdev_read(vdev_t *vdev, void *priv, off_t off, void *buf, size_t bytes) 204185029Spjd{ 205185029Spjd char *p; 206297629Sallanjude daddr_t lba, alignlba; 207300257Sallanjude off_t diff; 208297629Sallanjude unsigned int nb, alignnb; 209185029Spjd struct dsk *dsk = (struct dsk *) priv; 210185029Spjd 211185029Spjd if ((off & (DEV_BSIZE - 1)) || (bytes & (DEV_BSIZE - 1))) 212185029Spjd return -1; 213185029Spjd 214185029Spjd p = buf; 215185029Spjd lba = off / DEV_BSIZE; 216213136Spjd lba += dsk->start; 217297629Sallanjude /* 218300257Sallanjude * Align reads to 4k else 4k sector GELIs will not decrypt. 219300257Sallanjude * Round LBA down to nearest multiple of DEV_GELIBOOT_BSIZE bytes. 220300257Sallanjude */ 221300257Sallanjude alignlba = rounddown2(off, DEV_GELIBOOT_BSIZE) / DEV_BSIZE; 222300257Sallanjude /* 223297629Sallanjude * The read must be aligned to DEV_GELIBOOT_BSIZE bytes relative to the 224297629Sallanjude * start of the GELI partition, not the start of the actual disk. 225297629Sallanjude */ 226297629Sallanjude alignlba += dsk->start; 227297629Sallanjude diff = (lba - alignlba) * DEV_BSIZE; 228297629Sallanjude 229185029Spjd while (bytes > 0) { 230185029Spjd nb = bytes / DEV_BSIZE; 231297629Sallanjude /* 232297629Sallanjude * Ensure that the read size plus the leading offset does not 233297629Sallanjude * exceed the size of the read buffer. 234297629Sallanjude */ 235298949Speter if (nb > (READ_BUF_SIZE - diff) / DEV_BSIZE) 236298949Speter nb = (READ_BUF_SIZE - diff) / DEV_BSIZE; 237297629Sallanjude /* 238297629Sallanjude * Round the number of blocks to read up to the nearest multiple 239297629Sallanjude * of DEV_GELIBOOT_BSIZE. 240297629Sallanjude */ 241298949Speter alignnb = roundup2(nb * DEV_BSIZE + diff, DEV_GELIBOOT_BSIZE) 242298949Speter / DEV_BSIZE; 243297629Sallanjude 244297629Sallanjude if (drvread(dsk, dmadat->rdbuf, alignlba, alignnb)) 245185029Spjd return -1; 246296963Sallanjude#ifdef LOADER_GELI_SUPPORT 247296963Sallanjude /* decrypt */ 248296963Sallanjude if (is_geli(dsk) == 0) { 249297629Sallanjude if (geli_read(dsk, ((alignlba - dsk->start) * 250297629Sallanjude DEV_BSIZE), dmadat->rdbuf, alignnb * DEV_BSIZE)) 251297629Sallanjude return (-1); 252296963Sallanjude } 253296963Sallanjude#endif 254297629Sallanjude memcpy(p, dmadat->rdbuf + diff, nb * DEV_BSIZE); 255185029Spjd p += nb * DEV_BSIZE; 256185029Spjd lba += nb; 257297629Sallanjude alignlba += alignnb; 258185029Spjd bytes -= nb * DEV_BSIZE; 259297629Sallanjude /* Don't need the leading offset after the first block. */ 260297629Sallanjude diff = 0; 261185029Spjd } 262185029Spjd 263185029Spjd return 0; 264185029Spjd} 265185029Spjd 266185029Spjdstatic int 267308914Savgvdev_write(vdev_t *vdev, void *priv, off_t off, void *buf, size_t bytes) 268308914Savg{ 269308914Savg char *p; 270308914Savg daddr_t lba; 271308914Savg unsigned int nb; 272308914Savg struct dsk *dsk = (struct dsk *) priv; 273308914Savg 274308914Savg if ((off & (DEV_BSIZE - 1)) || (bytes & (DEV_BSIZE - 1))) 275308914Savg return -1; 276308914Savg 277308914Savg p = buf; 278308914Savg lba = off / DEV_BSIZE; 279308914Savg lba += dsk->start; 280308914Savg while (bytes > 0) { 281308914Savg nb = bytes / DEV_BSIZE; 282308914Savg if (nb > READ_BUF_SIZE / DEV_BSIZE) 283308914Savg nb = READ_BUF_SIZE / DEV_BSIZE; 284308914Savg memcpy(dmadat->rdbuf, p, nb * DEV_BSIZE); 285308914Savg if (drvwrite(dsk, dmadat->rdbuf, lba, nb)) 286308914Savg return -1; 287308914Savg p += nb * DEV_BSIZE; 288308914Savg lba += nb; 289308914Savg bytes -= nb * DEV_BSIZE; 290308914Savg } 291308914Savg 292308914Savg return 0; 293308914Savg} 294308914Savg 295308914Savgstatic int 296185029Spjdxfsread(const dnode_phys_t *dnode, off_t *offp, void *buf, size_t nbyte) 297185029Spjd{ 298185029Spjd if ((size_t)zfs_read(spa, dnode, offp, buf, nbyte) != nbyte) { 299235329Savg printf("Invalid format\n"); 300185029Spjd return -1; 301185029Spjd } 302185029Spjd return 0; 303185029Spjd} 304185029Spjd 305308914Savg/* 306308914Savg * Read Pad2 (formerly "Boot Block Header") area of the first 307308914Savg * vdev label of the given vdev. 308308914Savg */ 309308914Savgstatic int 310308914Savgvdev_read_pad2(vdev_t *vdev, char *buf, size_t size) 311308914Savg{ 312308914Savg blkptr_t bp; 313308914Savg char *tmp = zap_scratch; 314308914Savg off_t off = offsetof(vdev_label_t, vl_pad2); 315308914Savg 316308914Savg if (size > VDEV_PAD_SIZE) 317308914Savg size = VDEV_PAD_SIZE; 318308914Savg 319308914Savg BP_ZERO(&bp); 320308914Savg BP_SET_LSIZE(&bp, VDEV_PAD_SIZE); 321308914Savg BP_SET_PSIZE(&bp, VDEV_PAD_SIZE); 322308914Savg BP_SET_CHECKSUM(&bp, ZIO_CHECKSUM_LABEL); 323308914Savg BP_SET_COMPRESS(&bp, ZIO_COMPRESS_OFF); 324308914Savg DVA_SET_OFFSET(BP_IDENTITY(&bp), off); 325308914Savg if (vdev_read_phys(vdev, &bp, tmp, off, 0)) 326308914Savg return (EIO); 327308914Savg memcpy(buf, tmp, size); 328308914Savg return (0); 329308914Savg} 330308914Savg 331308914Savgstatic int 332308914Savgvdev_clear_pad2(vdev_t *vdev) 333308914Savg{ 334308914Savg char *zeroes = zap_scratch; 335308914Savg uint64_t *end; 336308914Savg off_t off = offsetof(vdev_label_t, vl_pad2); 337308914Savg 338308914Savg memset(zeroes, 0, VDEV_PAD_SIZE); 339308914Savg end = (uint64_t *)(zeroes + VDEV_PAD_SIZE); 340308914Savg /* ZIO_CHECKSUM_LABEL magic and pre-calcualted checksum for all zeros */ 341308914Savg end[-5] = 0x0210da7ab10c7a11; 342308914Savg end[-4] = 0x97f48f807f6e2a3f; 343308914Savg end[-3] = 0xaf909f1658aacefc; 344308914Savg end[-2] = 0xcbd1ea57ff6db48b; 345308914Savg end[-1] = 0x6ec692db0d465fab; 346308914Savg if (vdev_write(vdev, vdev->v_read_priv, off, zeroes, VDEV_PAD_SIZE)) 347308914Savg return (EIO); 348308914Savg return (0); 349308914Savg} 350308914Savg 351200309Sjhbstatic void 352200309Sjhbbios_getmem(void) 353185029Spjd{ 354200309Sjhb uint64_t size; 355185029Spjd 356200309Sjhb /* Parse system memory map */ 357200309Sjhb v86.ebx = 0; 358200309Sjhb do { 359200309Sjhb v86.ctl = V86_FLAGS; 360200309Sjhb v86.addr = 0x15; /* int 0x15 function 0xe820*/ 361200309Sjhb v86.eax = 0xe820; 362200309Sjhb v86.ecx = sizeof(struct bios_smap); 363200309Sjhb v86.edx = SMAP_SIG; 364200309Sjhb v86.es = VTOPSEG(&smap); 365200309Sjhb v86.edi = VTOPOFF(&smap); 366200309Sjhb v86int(); 367292682Sjhb if (V86_CY(v86.efl) || (v86.eax != SMAP_SIG)) 368200309Sjhb break; 369200309Sjhb /* look for a low-memory segment that's large enough */ 370200309Sjhb if ((smap.type == SMAP_TYPE_MEMORY) && (smap.base == 0) && 371200309Sjhb (smap.length >= (512 * 1024))) 372200309Sjhb bios_basemem = smap.length; 373200309Sjhb /* look for the first segment in 'extended' memory */ 374200309Sjhb if ((smap.type == SMAP_TYPE_MEMORY) && (smap.base == 0x100000)) { 375200309Sjhb bios_extmem = smap.length; 376200309Sjhb } 377200309Sjhb 378200309Sjhb /* 379200309Sjhb * Look for the largest segment in 'extended' memory beyond 380200309Sjhb * 1MB but below 4GB. 381200309Sjhb */ 382200309Sjhb if ((smap.type == SMAP_TYPE_MEMORY) && (smap.base > 0x100000) && 383200309Sjhb (smap.base < 0x100000000ull)) { 384200309Sjhb size = smap.length; 385200309Sjhb 386200309Sjhb /* 387200309Sjhb * If this segment crosses the 4GB boundary, truncate it. 388200309Sjhb */ 389200309Sjhb if (smap.base + size > 0x100000000ull) 390200309Sjhb size = 0x100000000ull - smap.base; 391200309Sjhb 392200309Sjhb if (size > high_heap_size) { 393200309Sjhb high_heap_size = size; 394200309Sjhb high_heap_base = smap.base; 395200309Sjhb } 396200309Sjhb } 397200309Sjhb } while (v86.ebx != 0); 398200309Sjhb 399200309Sjhb /* Fall back to the old compatibility function for base memory */ 400200309Sjhb if (bios_basemem == 0) { 401200309Sjhb v86.ctl = 0; 402200309Sjhb v86.addr = 0x12; /* int 0x12 */ 403200309Sjhb v86int(); 404200309Sjhb 405200309Sjhb bios_basemem = (v86.eax & 0xffff) * 1024; 406200309Sjhb } 407200309Sjhb 408200309Sjhb /* Fall back through several compatibility functions for extended memory */ 409200309Sjhb if (bios_extmem == 0) { 410200309Sjhb v86.ctl = V86_FLAGS; 411200309Sjhb v86.addr = 0x15; /* int 0x15 function 0xe801*/ 412200309Sjhb v86.eax = 0xe801; 413200309Sjhb v86int(); 414292682Sjhb if (!V86_CY(v86.efl)) { 415200309Sjhb bios_extmem = ((v86.ecx & 0xffff) + ((v86.edx & 0xffff) * 64)) * 1024; 416200309Sjhb } 417200309Sjhb } 418200309Sjhb if (bios_extmem == 0) { 419200309Sjhb v86.ctl = 0; 420200309Sjhb v86.addr = 0x15; /* int 0x15 function 0x88*/ 421200309Sjhb v86.eax = 0x8800; 422200309Sjhb v86int(); 423200309Sjhb bios_extmem = (v86.eax & 0xffff) * 1024; 424200309Sjhb } 425200309Sjhb 426200309Sjhb /* 427200309Sjhb * If we have extended memory and did not find a suitable heap 428200309Sjhb * region in the SMAP, use the last 3MB of 'extended' memory as a 429200309Sjhb * high heap candidate. 430200309Sjhb */ 431200309Sjhb if (bios_extmem >= HEAP_MIN && high_heap_size < HEAP_MIN) { 432200309Sjhb high_heap_size = HEAP_MIN; 433200309Sjhb high_heap_base = bios_extmem + 0x100000 - HEAP_MIN; 434200309Sjhb } 435296963Sallanjude} 436200309Sjhb 437185029Spjd/* 438185029Spjd * Try to detect a device supported by the legacy int13 BIOS 439185029Spjd */ 440185029Spjdstatic int 441185029Spjdint13probe(int drive) 442185029Spjd{ 443185029Spjd v86.ctl = V86_FLAGS; 444185029Spjd v86.addr = 0x13; 445185029Spjd v86.eax = 0x800; 446185029Spjd v86.edx = drive; 447185029Spjd v86int(); 448185029Spjd 449292682Sjhb if (!V86_CY(v86.efl) && /* carry clear */ 450185029Spjd ((v86.edx & 0xff) != (drive & DRV_MASK))) { /* unit # OK */ 451185029Spjd if ((v86.ecx & 0x3f) == 0) { /* absurd sector size */ 452185029Spjd return(0); /* skip device */ 453185029Spjd } 454185029Spjd return (1); 455185029Spjd } 456185029Spjd return(0); 457185029Spjd} 458185029Spjd 459192194Sdfr/* 460192194Sdfr * We call this when we find a ZFS vdev - ZFS consumes the dsk 461192194Sdfr * structure so we must make a new one. 462192194Sdfr */ 463192194Sdfrstatic struct dsk * 464192194Sdfrcopy_dsk(struct dsk *dsk) 465192194Sdfr{ 466192194Sdfr struct dsk *newdsk; 467192194Sdfr 468192194Sdfr newdsk = malloc(sizeof(struct dsk)); 469192194Sdfr *newdsk = *dsk; 470192194Sdfr return (newdsk); 471192194Sdfr} 472192194Sdfr 473329100Skevans/* 474329100Skevans * Get disk size from eax=0x800 and 0x4800. We need to probe both 475329100Skevans * because 0x4800 may not be available and we would like to get more 476329100Skevans * or less correct disk size - if it is possible at all. 477329100Skevans * Note we do not really want to touch drv.c because that code is shared 478329100Skevans * with boot2 and we can not afford to grow that code. 479329100Skevans */ 480329100Skevansstatic uint64_t 481329100Skevansdrvsize_ext(struct dsk *dskp) 482329100Skevans{ 483329100Skevans uint64_t size, tmp; 484329100Skevans int cyl, hds, sec; 485329100Skevans 486329100Skevans v86.ctl = V86_FLAGS; 487329100Skevans v86.addr = 0x13; 488329100Skevans v86.eax = 0x800; 489329100Skevans v86.edx = dskp->drive; 490329100Skevans v86int(); 491329100Skevans 492329100Skevans /* Don't error out if we get bad sector number, try EDD as well */ 493329100Skevans if (V86_CY(v86.efl) || /* carry set */ 494329100Skevans (v86.edx & 0xff) <= (unsigned)(dskp->drive & 0x7f)) /* unit # bad */ 495329100Skevans return (0); 496329100Skevans 497329100Skevans cyl = ((v86.ecx & 0xc0) << 2) + ((v86.ecx & 0xff00) >> 8) + 1; 498329100Skevans /* Convert max head # -> # of heads */ 499329100Skevans hds = ((v86.edx & 0xff00) >> 8) + 1; 500329100Skevans sec = v86.ecx & 0x3f; 501329100Skevans 502329100Skevans size = (uint64_t)cyl * hds * sec; 503329100Skevans 504329100Skevans /* Determine if we can use EDD with this device. */ 505329100Skevans v86.ctl = V86_FLAGS; 506329100Skevans v86.addr = 0x13; 507329100Skevans v86.eax = 0x4100; 508329100Skevans v86.edx = dskp->drive; 509329100Skevans v86.ebx = 0x55aa; 510329100Skevans v86int(); 511329100Skevans if (V86_CY(v86.efl) || /* carry set */ 512329100Skevans (v86.ebx & 0xffff) != 0xaa55 || /* signature */ 513329100Skevans (v86.ecx & EDD_INTERFACE_FIXED_DISK) == 0) 514329100Skevans return (size); 515329100Skevans 516329100Skevans tmp = drvsize(dskp); 517329100Skevans if (tmp > size) 518329100Skevans size = tmp; 519329100Skevans 520329100Skevans return (size); 521329100Skevans} 522329100Skevans 523329100Skevans/* 524329100Skevans * The "layered" ioctl to read disk/partition size. Unfortunately 525329100Skevans * the zfsboot case is hardest, because we do not have full software 526329100Skevans * stack available, so we need to do some manual work here. 527329100Skevans */ 528329100Skevansuint64_t 529329100Skevansldi_get_size(void *priv) 530329100Skevans{ 531329100Skevans struct dsk *dskp = priv; 532329100Skevans uint64_t size = dskp->size; 533329100Skevans 534329100Skevans if (dskp->start == 0) 535329100Skevans size = drvsize_ext(dskp); 536329100Skevans 537329100Skevans return (size * DEV_BSIZE); 538329100Skevans} 539329100Skevans 540185029Spjdstatic void 541241294Savgprobe_drive(struct dsk *dsk) 542185029Spjd{ 543185096Sdfr#ifdef GPT 544185096Sdfr struct gpt_hdr hdr; 545185096Sdfr struct gpt_ent *ent; 546185096Sdfr unsigned part, entries_per_sec; 547299997Speter daddr_t slba; 548185096Sdfr#endif 549299997Speter#if defined(GPT) || defined(LOADER_GELI_SUPPORT) 550299997Speter daddr_t elba; 551299660Sngie#endif 552299997Speter 553185029Spjd struct dos_partition *dp; 554185029Spjd char *sec; 555185029Spjd unsigned i; 556185029Spjd 557185029Spjd /* 558296963Sallanjude * If we find a vdev on the whole disk, stop here. 559185029Spjd */ 560241294Savg if (vdev_probe(vdev_read, dsk, NULL) == 0) 561185029Spjd return; 562185029Spjd 563296963Sallanjude#ifdef LOADER_GELI_SUPPORT 564296963Sallanjude /* 565296963Sallanjude * Taste the disk, if it is GELI encrypted, decrypt it and check to see if 566296963Sallanjude * it is a usable vdev then. Otherwise dig 567296963Sallanjude * out the partition table and probe each slice/partition 568296963Sallanjude * in turn for a vdev or GELI encrypted vdev. 569296963Sallanjude */ 570329100Skevans elba = drvsize_ext(dsk); 571296963Sallanjude if (elba > 0) { 572296963Sallanjude elba--; 573296963Sallanjude } 574296963Sallanjude if (geli_taste(vdev_read, dsk, elba) == 0) { 575329099Skevans if (geli_havekey(dsk) == 0 || geli_passphrase(&gelipw, dsk->unit, 576329099Skevans ':', 0, dsk) == 0) { 577296963Sallanjude if (vdev_probe(vdev_read, dsk, NULL) == 0) { 578296963Sallanjude return; 579296963Sallanjude } 580296963Sallanjude } 581296963Sallanjude } 582296963Sallanjude#endif /* LOADER_GELI_SUPPORT */ 583296963Sallanjude 584185029Spjd sec = dmadat->secbuf; 585185029Spjd dsk->start = 0; 586185096Sdfr 587185096Sdfr#ifdef GPT 588185096Sdfr /* 589185096Sdfr * First check for GPT. 590185096Sdfr */ 591185096Sdfr if (drvread(dsk, sec, 1, 1)) { 592185096Sdfr return; 593185096Sdfr } 594185096Sdfr memcpy(&hdr, sec, sizeof(hdr)); 595185096Sdfr if (memcmp(hdr.hdr_sig, GPT_HDR_SIG, sizeof(hdr.hdr_sig)) != 0 || 596185096Sdfr hdr.hdr_lba_self != 1 || hdr.hdr_revision < 0x00010000 || 597185096Sdfr hdr.hdr_entsz < sizeof(*ent) || DEV_BSIZE % hdr.hdr_entsz != 0) { 598185096Sdfr goto trymbr; 599185096Sdfr } 600185096Sdfr 601185096Sdfr /* 602298826Spfg * Probe all GPT partitions for the presence of ZFS pools. We 603185096Sdfr * return the spa_t for the first we find (if requested). This 604185096Sdfr * will have the effect of booting from the first pool on the 605185096Sdfr * disk. 606296963Sallanjude * 607296963Sallanjude * If no vdev is found, GELI decrypting the device and try again 608185096Sdfr */ 609185096Sdfr entries_per_sec = DEV_BSIZE / hdr.hdr_entsz; 610185096Sdfr slba = hdr.hdr_lba_table; 611185096Sdfr elba = slba + hdr.hdr_entries / entries_per_sec; 612185096Sdfr while (slba < elba) { 613198420Srnoland dsk->start = 0; 614185096Sdfr if (drvread(dsk, sec, slba, 1)) 615185096Sdfr return; 616185096Sdfr for (part = 0; part < entries_per_sec; part++) { 617185096Sdfr ent = (struct gpt_ent *)(sec + part * hdr.hdr_entsz); 618185096Sdfr if (memcmp(&ent->ent_type, &freebsd_zfs_uuid, 619185096Sdfr sizeof(uuid_t)) == 0) { 620185096Sdfr dsk->start = ent->ent_lba_start; 621329100Skevans dsk->size = ent->ent_lba_end - ent->ent_lba_start + 1; 622296963Sallanjude dsk->slice = part + 1; 623296963Sallanjude dsk->part = 255; 624241294Savg if (vdev_probe(vdev_read, dsk, NULL) == 0) { 625185096Sdfr /* 626185096Sdfr * This slice had a vdev. We need a new dsk 627185096Sdfr * structure now since the vdev now owns this one. 628185096Sdfr */ 629192194Sdfr dsk = copy_dsk(dsk); 630185096Sdfr } 631296963Sallanjude#ifdef LOADER_GELI_SUPPORT 632296963Sallanjude else if (geli_taste(vdev_read, dsk, ent->ent_lba_end - 633296963Sallanjude ent->ent_lba_start) == 0) { 634329099Skevans if (geli_havekey(dsk) == 0 || geli_passphrase(&gelipw, 635329099Skevans dsk->unit, 'p', dsk->slice, dsk) == 0) { 636296963Sallanjude /* 637296963Sallanjude * This slice has GELI, check it for ZFS. 638296963Sallanjude */ 639296963Sallanjude if (vdev_probe(vdev_read, dsk, NULL) == 0) { 640296963Sallanjude /* 641296963Sallanjude * This slice had a vdev. We need a new dsk 642296963Sallanjude * structure now since the vdev now owns this one. 643296963Sallanjude */ 644296963Sallanjude dsk = copy_dsk(dsk); 645296963Sallanjude } 646296963Sallanjude break; 647296963Sallanjude } 648296963Sallanjude } 649296963Sallanjude#endif /* LOADER_GELI_SUPPORT */ 650185096Sdfr } 651185096Sdfr } 652185096Sdfr slba++; 653185096Sdfr } 654185096Sdfr return; 655185096Sdfrtrymbr: 656296963Sallanjude#endif /* GPT */ 657185096Sdfr 658185029Spjd if (drvread(dsk, sec, DOSBBSECTOR, 1)) 659185029Spjd return; 660185029Spjd dp = (void *)(sec + DOSPARTOFF); 661185029Spjd 662185029Spjd for (i = 0; i < NDOSPART; i++) { 663185029Spjd if (!dp[i].dp_typ) 664185029Spjd continue; 665185029Spjd dsk->start = dp[i].dp_start; 666329100Skevans dsk->size = dp[i].dp_size; 667296963Sallanjude dsk->slice = i + 1; 668241294Savg if (vdev_probe(vdev_read, dsk, NULL) == 0) { 669192194Sdfr dsk = copy_dsk(dsk); 670185029Spjd } 671296963Sallanjude#ifdef LOADER_GELI_SUPPORT 672296963Sallanjude else if (geli_taste(vdev_read, dsk, dp[i].dp_size - 673296963Sallanjude dp[i].dp_start) == 0) { 674329099Skevans if (geli_havekey(dsk) == 0 || geli_passphrase(&gelipw, dsk->unit, 675329099Skevans 's', i, dsk) == 0) { 676296963Sallanjude /* 677296963Sallanjude * This slice has GELI, check it for ZFS. 678296963Sallanjude */ 679296963Sallanjude if (vdev_probe(vdev_read, dsk, NULL) == 0) { 680296963Sallanjude /* 681296963Sallanjude * This slice had a vdev. We need a new dsk 682296963Sallanjude * structure now since the vdev now owns this one. 683296963Sallanjude */ 684296963Sallanjude dsk = copy_dsk(dsk); 685296963Sallanjude } 686296963Sallanjude break; 687296963Sallanjude } 688296963Sallanjude } 689296963Sallanjude#endif /* LOADER_GELI_SUPPORT */ 690185029Spjd } 691185029Spjd} 692185029Spjd 693185029Spjdint 694185029Spjdmain(void) 695185029Spjd{ 696185029Spjd dnode_phys_t dn; 697185029Spjd off_t off; 698185029Spjd struct dsk *dsk; 699308914Savg int autoboot, i; 700308914Savg int nextboot; 701308914Savg int rc; 702185029Spjd 703208388Sjhb dmadat = (void *)(roundup2(__base + (int32_t)&_end, 0x10000) - __base); 704208388Sjhb 705200309Sjhb bios_getmem(); 706200309Sjhb 707200309Sjhb if (high_heap_size > 0) { 708200309Sjhb heap_end = PTOV(high_heap_base + high_heap_size); 709200309Sjhb heap_next = PTOV(high_heap_base); 710200309Sjhb } else { 711296963Sallanjude heap_next = (char *)dmadat + sizeof(*dmadat); 712296963Sallanjude heap_end = (char *)PTOV(bios_basemem); 713200309Sjhb } 714200309Sjhb 715185029Spjd dsk = malloc(sizeof(struct dsk)); 716185029Spjd dsk->drive = *(uint8_t *)PTOV(ARGS); 717185029Spjd dsk->type = dsk->drive & DRV_HARD ? TYPE_AD : TYPE_FD; 718185029Spjd dsk->unit = dsk->drive & DRV_MASK; 719185029Spjd dsk->slice = *(uint8_t *)PTOV(ARGS + 1) + 1; 720185029Spjd dsk->part = 0; 721185029Spjd dsk->start = 0; 722329100Skevans dsk->size = 0; 723185029Spjd 724185029Spjd bootinfo.bi_version = BOOTINFO_VERSION; 725185029Spjd bootinfo.bi_size = sizeof(bootinfo); 726200309Sjhb bootinfo.bi_basemem = bios_basemem / 1024; 727200309Sjhb bootinfo.bi_extmem = bios_extmem / 1024; 728185029Spjd bootinfo.bi_memsizes_valid++; 729185029Spjd bootinfo.bi_bios_dev = dsk->drive; 730185029Spjd 731185029Spjd bootdev = MAKEBOOTDEV(dev_maj[dsk->type], 732305614Spfg dsk->slice, dsk->unit, dsk->part); 733185029Spjd 734185029Spjd /* Process configuration file */ 735185029Spjd 736185029Spjd autoboot = 1; 737185029Spjd 738296963Sallanjude#ifdef LOADER_GELI_SUPPORT 739296963Sallanjude geli_init(); 740296963Sallanjude#endif 741185029Spjd zfs_init(); 742185029Spjd 743185029Spjd /* 744185029Spjd * Probe the boot drive first - we will try to boot from whatever 745185029Spjd * pool we find on that drive. 746185029Spjd */ 747241294Savg probe_drive(dsk); 748185029Spjd 749185029Spjd /* 750185029Spjd * Probe the rest of the drives that the bios knows about. This 751185029Spjd * will find any other available pools and it may fill in missing 752185029Spjd * vdevs for the boot pool. 753185029Spjd */ 754212805Spjd#ifndef VIRTUALBOX 755212805Spjd for (i = 0; i < *(unsigned char *)PTOV(BIOS_NUMDRIVES); i++) 756212805Spjd#else 757212805Spjd for (i = 0; i < MAXBDDEV; i++) 758212805Spjd#endif 759212805Spjd { 760185029Spjd if ((i | DRV_HARD) == *(uint8_t *)PTOV(ARGS)) 761185029Spjd continue; 762185029Spjd 763192194Sdfr if (!int13probe(i | DRV_HARD)) 764192194Sdfr break; 765192194Sdfr 766185029Spjd dsk = malloc(sizeof(struct dsk)); 767185029Spjd dsk->drive = i | DRV_HARD; 768185029Spjd dsk->type = dsk->drive & TYPE_AD; 769185029Spjd dsk->unit = i; 770185029Spjd dsk->slice = 0; 771185029Spjd dsk->part = 0; 772185029Spjd dsk->start = 0; 773329100Skevans dsk->size = 0; 774241294Savg probe_drive(dsk); 775185029Spjd } 776185029Spjd 777185029Spjd /* 778241294Savg * The first discovered pool, if any, is the pool. 779185029Spjd */ 780241294Savg spa = spa_get_primary(); 781185029Spjd if (!spa) { 782241294Savg printf("%s: No ZFS pools located, can't boot\n", BOOTPROG); 783241294Savg for (;;) 784241294Savg ; 785185029Spjd } 786185029Spjd 787241293Savg primary_spa = spa; 788241293Savg primary_vdev = spa_get_primary_vdev(spa); 789241293Savg 790308914Savg nextboot = 0; 791308914Savg rc = vdev_read_pad2(primary_vdev, cmd, sizeof(cmd)); 792308914Savg if (vdev_clear_pad2(primary_vdev)) 793308914Savg printf("failed to clear pad2 area of primary vdev\n"); 794308914Savg if (rc == 0) { 795308914Savg if (*cmd) { 796308914Savg /* 797308914Savg * We could find an old-style ZFS Boot Block header here. 798308914Savg * Simply ignore it. 799308914Savg */ 800308914Savg if (*(uint64_t *)cmd != 0x2f5b007b10c) { 801308914Savg /* 802308914Savg * Note that parse() is destructive to cmd[] and we also want 803308914Savg * to honor RBX_QUIET option that could be present in cmd[]. 804308914Savg */ 805308914Savg nextboot = 1; 806308914Savg memcpy(cmddup, cmd, sizeof(cmd)); 807329099Skevans if (parse_cmd()) { 808308914Savg printf("failed to parse pad2 area of primary vdev\n"); 809308914Savg reboot(); 810308914Savg } 811308914Savg if (!OPT_CHECK(RBX_QUIET)) 812308914Savg printf("zfs nextboot: %s\n", cmddup); 813308914Savg } 814308914Savg /* Do not process this command twice */ 815308914Savg *cmd = 0; 816308914Savg } 817308914Savg } else 818308914Savg printf("failed to read pad2 area of primary vdev\n"); 819308914Savg 820308914Savg /* Mount ZFS only if it's not already mounted via nextboot parsing. */ 821308914Savg if (zfsmount.spa == NULL && 822308914Savg (zfs_spa_init(spa) != 0 || zfs_mount(spa, 0, &zfsmount) != 0)) { 823235329Savg printf("%s: failed to mount default pool %s\n", 824235329Savg BOOTPROG, spa->spa_name); 825235329Savg autoboot = 0; 826235329Savg } else if (zfs_lookup(&zfsmount, PATH_CONFIG, &dn) == 0 || 827235329Savg zfs_lookup(&zfsmount, PATH_DOTCONFIG, &dn) == 0) { 828185029Spjd off = 0; 829198079Sjhb zfs_read(spa, &dn, &off, cmd, sizeof(cmd)); 830185029Spjd } 831185029Spjd 832185029Spjd if (*cmd) { 833234339Savg /* 834329099Skevans * Note that parse_cmd() is destructive to cmd[] and we also want 835234339Savg * to honor RBX_QUIET option that could be present in cmd[]. 836234339Savg */ 837234339Savg memcpy(cmddup, cmd, sizeof(cmd)); 838329099Skevans if (parse_cmd()) 839185029Spjd autoboot = 0; 840234339Savg if (!OPT_CHECK(RBX_QUIET)) 841241288Savg printf("%s: %s\n", PATH_CONFIG, cmddup); 842185029Spjd /* Do not process this command twice */ 843185029Spjd *cmd = 0; 844185029Spjd } 845185029Spjd 846308914Savg /* Do not risk waiting at the prompt forever. */ 847308914Savg if (nextboot && !autoboot) 848308914Savg reboot(); 849308914Savg 850185029Spjd /* 851294925Simp * Try to exec /boot/loader. If interrupted by a keypress, 852185029Spjd * or in case of failure, try to load a kernel directly instead. 853185029Spjd */ 854185029Spjd 855185029Spjd if (autoboot && !*kname) { 856294925Simp memcpy(kname, PATH_LOADER_ZFS, sizeof(PATH_LOADER_ZFS)); 857213136Spjd if (!keyhit(3)) { 858185029Spjd load(); 859185029Spjd memcpy(kname, PATH_KERNEL, sizeof(PATH_KERNEL)); 860185029Spjd } 861185029Spjd } 862185029Spjd 863185029Spjd /* Present the user with the boot2 prompt. */ 864185029Spjd 865185029Spjd for (;;) { 866235329Savg if (!autoboot || !OPT_CHECK(RBX_QUIET)) { 867235329Savg printf("\nFreeBSD/x86 boot\n"); 868235329Savg if (zfs_rlookup(spa, zfsmount.rootobj, rootname) != 0) 869241288Savg printf("Default: %s/<0x%llx>:%s\n" 870235329Savg "boot: ", 871235329Savg spa->spa_name, zfsmount.rootobj, kname); 872241288Savg else if (rootname[0] != '\0') 873241288Savg printf("Default: %s/%s:%s\n" 874241288Savg "boot: ", 875241288Savg spa->spa_name, rootname, kname); 876235329Savg else 877241288Savg printf("Default: %s:%s\n" 878235329Savg "boot: ", 879241288Savg spa->spa_name, kname); 880235329Savg } 881185029Spjd if (ioctrl & IO_SERIAL) 882185029Spjd sio_flush(); 883213136Spjd if (!autoboot || keyhit(5)) 884213136Spjd getstr(cmd, sizeof(cmd)); 885185029Spjd else if (!autoboot || !OPT_CHECK(RBX_QUIET)) 886185029Spjd putchar('\n'); 887185029Spjd autoboot = 0; 888329099Skevans if (parse_cmd()) 889185029Spjd putchar('\a'); 890185029Spjd else 891185029Spjd load(); 892185029Spjd } 893185029Spjd} 894185029Spjd 895185029Spjd/* XXX - Needed for btxld to link the boot2 binary; do not remove. */ 896185029Spjdvoid 897185029Spjdexit(int x) 898185029Spjd{ 899308914Savg __exit(x); 900185029Spjd} 901185029Spjd 902308914Savgvoid 903308914Savgreboot(void) 904308914Savg{ 905308914Savg __exit(0); 906308914Savg} 907308914Savg 908185029Spjdstatic void 909185029Spjdload(void) 910185029Spjd{ 911185029Spjd union { 912185029Spjd struct exec ex; 913185029Spjd Elf32_Ehdr eh; 914185029Spjd } hdr; 915185029Spjd static Elf32_Phdr ep[2]; 916185029Spjd static Elf32_Shdr es[2]; 917185029Spjd caddr_t p; 918185029Spjd dnode_phys_t dn; 919185029Spjd off_t off; 920185029Spjd uint32_t addr, x; 921185029Spjd int fmt, i, j; 922185029Spjd 923235329Savg if (zfs_lookup(&zfsmount, kname, &dn)) { 924235329Savg printf("\nCan't find %s\n", kname); 925185029Spjd return; 926185029Spjd } 927185029Spjd off = 0; 928185029Spjd if (xfsread(&dn, &off, &hdr, sizeof(hdr))) 929185029Spjd return; 930185029Spjd if (N_GETMAGIC(hdr.ex) == ZMAGIC) 931185029Spjd fmt = 0; 932185029Spjd else if (IS_ELF(hdr.eh)) 933185029Spjd fmt = 1; 934185029Spjd else { 935185029Spjd printf("Invalid %s\n", "format"); 936185029Spjd return; 937185029Spjd } 938185029Spjd if (fmt == 0) { 939185029Spjd addr = hdr.ex.a_entry & 0xffffff; 940185029Spjd p = PTOV(addr); 941185029Spjd off = PAGE_SIZE; 942185029Spjd if (xfsread(&dn, &off, p, hdr.ex.a_text)) 943185029Spjd return; 944185029Spjd p += roundup2(hdr.ex.a_text, PAGE_SIZE); 945185029Spjd if (xfsread(&dn, &off, p, hdr.ex.a_data)) 946185029Spjd return; 947185029Spjd p += hdr.ex.a_data + roundup2(hdr.ex.a_bss, PAGE_SIZE); 948185029Spjd bootinfo.bi_symtab = VTOP(p); 949185029Spjd memcpy(p, &hdr.ex.a_syms, sizeof(hdr.ex.a_syms)); 950185029Spjd p += sizeof(hdr.ex.a_syms); 951185029Spjd if (hdr.ex.a_syms) { 952185029Spjd if (xfsread(&dn, &off, p, hdr.ex.a_syms)) 953185029Spjd return; 954185029Spjd p += hdr.ex.a_syms; 955185029Spjd if (xfsread(&dn, &off, p, sizeof(int))) 956185029Spjd return; 957185029Spjd x = *(uint32_t *)p; 958185029Spjd p += sizeof(int); 959185029Spjd x -= sizeof(int); 960185029Spjd if (xfsread(&dn, &off, p, x)) 961185029Spjd return; 962185029Spjd p += x; 963185029Spjd } 964185029Spjd } else { 965185029Spjd off = hdr.eh.e_phoff; 966185029Spjd for (j = i = 0; i < hdr.eh.e_phnum && j < 2; i++) { 967185029Spjd if (xfsread(&dn, &off, ep + j, sizeof(ep[0]))) 968185029Spjd return; 969185029Spjd if (ep[j].p_type == PT_LOAD) 970185029Spjd j++; 971185029Spjd } 972185029Spjd for (i = 0; i < 2; i++) { 973185029Spjd p = PTOV(ep[i].p_paddr & 0xffffff); 974185029Spjd off = ep[i].p_offset; 975185029Spjd if (xfsread(&dn, &off, p, ep[i].p_filesz)) 976185029Spjd return; 977185029Spjd } 978185029Spjd p += roundup2(ep[1].p_memsz, PAGE_SIZE); 979185029Spjd bootinfo.bi_symtab = VTOP(p); 980185029Spjd if (hdr.eh.e_shnum == hdr.eh.e_shstrndx + 3) { 981185029Spjd off = hdr.eh.e_shoff + sizeof(es[0]) * 982185029Spjd (hdr.eh.e_shstrndx + 1); 983185029Spjd if (xfsread(&dn, &off, &es, sizeof(es))) 984185029Spjd return; 985185029Spjd for (i = 0; i < 2; i++) { 986185029Spjd memcpy(p, &es[i].sh_size, sizeof(es[i].sh_size)); 987185029Spjd p += sizeof(es[i].sh_size); 988185029Spjd off = es[i].sh_offset; 989185029Spjd if (xfsread(&dn, &off, p, es[i].sh_size)) 990185029Spjd return; 991185029Spjd p += es[i].sh_size; 992185029Spjd } 993185029Spjd } 994185029Spjd addr = hdr.eh.e_entry & 0xffffff; 995185029Spjd } 996185029Spjd bootinfo.bi_esymtab = VTOP(p); 997185029Spjd bootinfo.bi_kernelname = VTOP(kname); 998235329Savg zfsargs.size = sizeof(zfsargs); 999235329Savg zfsargs.pool = zfsmount.spa->spa_guid; 1000235329Savg zfsargs.root = zfsmount.rootobj; 1001241293Savg zfsargs.primary_pool = primary_spa->spa_guid; 1002296963Sallanjude#ifdef LOADER_GELI_SUPPORT 1003329099Skevans explicit_bzero(gelipw, sizeof(gelipw)); 1004329099Skevans gelibuf = malloc(sizeof(struct keybuf) + (GELI_MAX_KEYS * sizeof(struct keybuf_ent))); 1005329099Skevans geli_fill_keybuf(gelibuf); 1006329099Skevans zfsargs.notapw = '\0'; 1007329099Skevans zfsargs.keybuf_sentinel = KEYBUF_SENTINEL; 1008329099Skevans zfsargs.keybuf = gelibuf; 1009296963Sallanjude#else 1010296963Sallanjude zfsargs.gelipw[0] = '\0'; 1011296963Sallanjude#endif 1012241293Savg if (primary_vdev != NULL) 1013241293Savg zfsargs.primary_vdev = primary_vdev->v_guid; 1014241293Savg else 1015241293Savg printf("failed to detect primary vdev\n"); 1016185029Spjd __exec((caddr_t)addr, RB_BOOTINFO | (opts & RBX_MASK), 1017185029Spjd bootdev, 1018235329Savg KARGS_FLAGS_ZFS | KARGS_FLAGS_EXTARG, 1019185029Spjd (uint32_t) spa->spa_guid, 1020185029Spjd (uint32_t) (spa->spa_guid >> 32), 1021235329Savg VTOP(&bootinfo), 1022235329Savg zfsargs); 1023185029Spjd} 1024185029Spjd 1025185029Spjdstatic int 1026241288Savgzfs_mount_ds(char *dsname) 1027241288Savg{ 1028241288Savg uint64_t newroot; 1029241288Savg spa_t *newspa; 1030241288Savg char *q; 1031241288Savg 1032241288Savg q = strchr(dsname, '/'); 1033241288Savg if (q) 1034241288Savg *q++ = '\0'; 1035241288Savg newspa = spa_find_by_name(dsname); 1036241288Savg if (newspa == NULL) { 1037241288Savg printf("\nCan't find ZFS pool %s\n", dsname); 1038241288Savg return -1; 1039241288Savg } 1040241288Savg 1041241288Savg if (zfs_spa_init(newspa)) 1042241288Savg return -1; 1043241288Savg 1044241288Savg newroot = 0; 1045241288Savg if (q) { 1046241288Savg if (zfs_lookup_dataset(newspa, q, &newroot)) { 1047241288Savg printf("\nCan't find dataset %s in ZFS pool %s\n", 1048241288Savg q, newspa->spa_name); 1049241288Savg return -1; 1050241288Savg } 1051241288Savg } 1052241288Savg if (zfs_mount(newspa, newroot, &zfsmount)) { 1053241288Savg printf("\nCan't mount ZFS dataset\n"); 1054241288Savg return -1; 1055241288Savg } 1056241288Savg spa = newspa; 1057241288Savg return (0); 1058241288Savg} 1059241288Savg 1060241288Savgstatic int 1061329099Skevansparse_cmd(void) 1062185029Spjd{ 1063185029Spjd char *arg = cmd; 1064185029Spjd char *ep, *p, *q; 1065185029Spjd const char *cp; 1066185029Spjd int c, i, j; 1067185029Spjd 1068185029Spjd while ((c = *arg++)) { 1069185029Spjd if (c == ' ' || c == '\t' || c == '\n') 1070185029Spjd continue; 1071185029Spjd for (p = arg; *p && *p != '\n' && *p != ' ' && *p != '\t'; p++); 1072185029Spjd ep = p; 1073185029Spjd if (*p) 1074185029Spjd *p++ = 0; 1075185029Spjd if (c == '-') { 1076185029Spjd while ((c = *arg++)) { 1077185029Spjd if (c == 'P') { 1078185029Spjd if (*(uint8_t *)PTOV(0x496) & 0x10) { 1079185029Spjd cp = "yes"; 1080185029Spjd } else { 1081185029Spjd opts |= OPT_SET(RBX_DUAL) | OPT_SET(RBX_SERIAL); 1082185029Spjd cp = "no"; 1083185029Spjd } 1084185029Spjd printf("Keyboard: %s\n", cp); 1085185029Spjd continue; 1086185029Spjd } else if (c == 'S') { 1087185029Spjd j = 0; 1088185029Spjd while ((unsigned int)(i = *arg++ - '0') <= 9) 1089185029Spjd j = j * 10 + i; 1090185029Spjd if (j > 0 && i == -'0') { 1091185029Spjd comspeed = j; 1092185029Spjd break; 1093185029Spjd } 1094185029Spjd /* Fall through to error below ('S' not in optstr[]). */ 1095185029Spjd } 1096185029Spjd for (i = 0; c != optstr[i]; i++) 1097185029Spjd if (i == NOPT - 1) 1098185029Spjd return -1; 1099185029Spjd opts ^= OPT_SET(flags[i]); 1100185029Spjd } 1101185029Spjd ioctrl = OPT_CHECK(RBX_DUAL) ? (IO_SERIAL|IO_KEYBOARD) : 1102185029Spjd OPT_CHECK(RBX_SERIAL) ? IO_SERIAL : IO_KEYBOARD; 1103241301Savg if (ioctrl & IO_SERIAL) { 1104241301Savg if (sio_init(115200 / comspeed) != 0) 1105241301Savg ioctrl &= ~IO_SERIAL; 1106241301Savg } 1107185029Spjd } if (c == '?') { 1108185029Spjd dnode_phys_t dn; 1109185029Spjd 1110235329Savg if (zfs_lookup(&zfsmount, arg, &dn) == 0) { 1111185029Spjd zap_list(spa, &dn); 1112185029Spjd } 1113185029Spjd return -1; 1114185029Spjd } else { 1115185029Spjd arg--; 1116185029Spjd 1117185029Spjd /* 1118185029Spjd * Report pool status if the comment is 'status'. Lets 1119185029Spjd * hope no-one wants to load /status as a kernel. 1120185029Spjd */ 1121185029Spjd if (!strcmp(arg, "status")) { 1122185029Spjd spa_all_status(); 1123185029Spjd return -1; 1124185029Spjd } 1125185029Spjd 1126185029Spjd /* 1127241288Savg * If there is "zfs:" prefix simply ignore it. 1128241288Savg */ 1129241288Savg if (strncmp(arg, "zfs:", 4) == 0) 1130241288Savg arg += 4; 1131241288Savg 1132241288Savg /* 1133185029Spjd * If there is a colon, switch pools. 1134185029Spjd */ 1135241288Savg q = strchr(arg, ':'); 1136185029Spjd if (q) { 1137241288Savg *q++ = '\0'; 1138241288Savg if (zfs_mount_ds(arg) != 0) 1139185029Spjd return -1; 1140241288Savg arg = q; 1141185029Spjd } 1142185029Spjd if ((i = ep - arg)) { 1143185029Spjd if ((size_t)i >= sizeof(kname)) 1144185029Spjd return -1; 1145185029Spjd memcpy(kname, arg, i + 1); 1146185029Spjd } 1147185029Spjd } 1148185029Spjd arg = p; 1149185029Spjd } 1150185029Spjd return 0; 1151185029Spjd} 1152