Home | History | Annotate | Line # | Download | only in efiboot
      1  1.20       rin /* $NetBSD: efiblock.c,v 1.20 2023/06/14 00:52:25 rin Exp $ */
      2   1.1  jmcneill 
      3   1.1  jmcneill /*-
      4   1.1  jmcneill  * Copyright (c) 2016 Kimihiro Nonaka <nonaka (at) netbsd.org>
      5   1.1  jmcneill  * Copyright (c) 2018 Jared McNeill <jmcneill (at) invisible.ca>
      6   1.1  jmcneill  * All rights reserved.
      7   1.1  jmcneill  *
      8   1.1  jmcneill  * Redistribution and use in source and binary forms, with or without
      9   1.1  jmcneill  * modification, are permitted provided that the following conditions
     10   1.1  jmcneill  * are met:
     11   1.1  jmcneill  * 1. Redistributions of source code must retain the above copyright
     12   1.1  jmcneill  *    notice, this list of conditions and the following disclaimer.
     13   1.1  jmcneill  * 2. Redistributions in binary form must reproduce the above copyright
     14   1.1  jmcneill  *    notice, this list of conditions and the following disclaimer in the
     15   1.1  jmcneill  *    documentation and/or other materials provided with the distribution.
     16   1.1  jmcneill  *
     17   1.1  jmcneill  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     18   1.1  jmcneill  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     19   1.1  jmcneill  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     20   1.1  jmcneill  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     21   1.1  jmcneill  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     22   1.1  jmcneill  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     23   1.1  jmcneill  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     24   1.1  jmcneill  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     25   1.1  jmcneill  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     26   1.1  jmcneill  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     27   1.1  jmcneill  * SUCH DAMAGE.
     28   1.1  jmcneill  */
     29   1.1  jmcneill 
     30   1.1  jmcneill #define FSTYPENAMES
     31   1.1  jmcneill 
     32   1.1  jmcneill #include <sys/param.h>
     33   1.2  jmcneill #include <sys/md5.h>
     34   1.4  jmcneill #include <sys/uuid.h>
     35   1.1  jmcneill 
     36   1.8  jmcneill #include <fs/cd9660/iso.h>
     37   1.8  jmcneill 
     38   1.1  jmcneill #include "efiboot.h"
     39   1.1  jmcneill #include "efiblock.h"
     40   1.1  jmcneill 
     41  1.14  jmcneill #define	EFI_BLOCK_READAHEAD	(64 * 1024)
     42  1.12  jmcneill #define	EFI_BLOCK_TIMEOUT	120
     43  1.12  jmcneill #define	EFI_BLOCK_TIMEOUT_CODE	0x810c0000
     44  1.12  jmcneill 
     45  1.11       mrg /*
     46  1.11       mrg  * The raidframe support is basic.  Ideally, it should be expanded to
     47  1.11       mrg  * consider raid volumes a first-class citizen like the x86 efiboot does,
     48  1.11       mrg  * but for now, we simply assume each RAID is potentially bootable.
     49  1.11       mrg  */
     50  1.11       mrg #define	RF_PROTECTED_SECTORS	64	/* XXX refer to <.../rf_optnames.h> */
     51  1.11       mrg 
     52   1.1  jmcneill static EFI_HANDLE *efi_block;
     53   1.1  jmcneill static UINTN efi_nblock;
     54   1.2  jmcneill static struct efi_block_part *efi_block_booted = NULL;
     55   1.1  jmcneill 
     56  1.14  jmcneill static bool efi_ra_enable = false;
     57  1.14  jmcneill static UINT8 *efi_ra_buffer = NULL;
     58  1.14  jmcneill static UINT32 efi_ra_media_id;
     59  1.14  jmcneill static UINT64 efi_ra_start = 0;
     60  1.14  jmcneill static UINT64 efi_ra_length = 0;
     61  1.14  jmcneill 
     62   1.1  jmcneill static TAILQ_HEAD(, efi_block_dev) efi_block_devs = TAILQ_HEAD_INITIALIZER(efi_block_devs);
     63   1.1  jmcneill 
     64   1.1  jmcneill static int
     65   1.1  jmcneill efi_block_parse(const char *fname, struct efi_block_part **pbpart, char **pfile)
     66   1.1  jmcneill {
     67   1.1  jmcneill 	struct efi_block_dev *bdev;
     68   1.1  jmcneill 	struct efi_block_part *bpart;
     69   1.1  jmcneill 	char pathbuf[PATH_MAX], *default_device, *ep = NULL;
     70   1.1  jmcneill 	const char *full_path;
     71   1.1  jmcneill 	intmax_t dev;
     72   1.1  jmcneill 	int part;
     73   1.1  jmcneill 
     74   1.1  jmcneill 	default_device = get_default_device();
     75   1.1  jmcneill 	if (strchr(fname, ':') == NULL) {
     76   1.1  jmcneill 		if (strlen(default_device) > 0) {
     77   1.1  jmcneill 			snprintf(pathbuf, sizeof(pathbuf), "%s:%s", default_device, fname);
     78   1.1  jmcneill 			full_path = pathbuf;
     79   1.1  jmcneill 			*pfile = __UNCONST(fname);
     80   1.1  jmcneill 		} else {
     81   1.1  jmcneill 			return EINVAL;
     82   1.1  jmcneill 		}
     83   1.1  jmcneill 	} else {
     84   1.1  jmcneill 		full_path = fname;
     85   1.1  jmcneill 		*pfile = strchr(fname, ':') + 1;
     86   1.1  jmcneill 	}
     87   1.1  jmcneill 
     88  1.18  jmcneill 	if (*pfile[0] == '\0') {
     89  1.18  jmcneill 		*pfile = __UNCONST("/");
     90  1.18  jmcneill 	}
     91  1.18  jmcneill 
     92   1.1  jmcneill 	if (strncasecmp(full_path, "hd", 2) != 0)
     93   1.1  jmcneill 		return EINVAL;
     94   1.1  jmcneill 	dev = strtoimax(full_path + 2, &ep, 10);
     95   1.1  jmcneill 	if (dev < 0 || dev >= efi_nblock)
     96   1.1  jmcneill 		return ENXIO;
     97   1.1  jmcneill 	if (ep[0] < 'a' || ep[0] >= 'a' + MAXPARTITIONS || ep[1] != ':')
     98   1.1  jmcneill 		return EINVAL;
     99   1.1  jmcneill 	part = ep[0] - 'a';
    100   1.1  jmcneill 	TAILQ_FOREACH(bdev, &efi_block_devs, entries) {
    101   1.1  jmcneill 		if (bdev->index == dev) {
    102   1.1  jmcneill 			TAILQ_FOREACH(bpart, &bdev->partitions, entries) {
    103   1.1  jmcneill 				if (bpart->index == part) {
    104   1.1  jmcneill 					*pbpart = bpart;
    105   1.1  jmcneill 					return 0;
    106   1.1  jmcneill 				}
    107   1.1  jmcneill 			}
    108   1.1  jmcneill 		}
    109   1.1  jmcneill 	}
    110   1.1  jmcneill 
    111   1.1  jmcneill 	return ENOENT;
    112   1.1  jmcneill }
    113   1.1  jmcneill 
    114   1.2  jmcneill static void
    115   1.2  jmcneill efi_block_generate_hash_mbr(struct efi_block_part *bpart, struct mbr_sector *mbr)
    116   1.2  jmcneill {
    117   1.2  jmcneill 	MD5_CTX md5ctx;
    118   1.2  jmcneill 
    119   1.2  jmcneill 	MD5Init(&md5ctx);
    120   1.2  jmcneill 	MD5Update(&md5ctx, (void *)mbr, sizeof(*mbr));
    121   1.2  jmcneill 	MD5Final(bpart->hash, &md5ctx);
    122   1.2  jmcneill }
    123   1.2  jmcneill 
    124  1.13  jmcneill static EFI_STATUS
    125  1.15  jmcneill efi_block_do_read_blockio(struct efi_block_dev *bdev, UINT64 off, void *buf,
    126  1.15  jmcneill     UINTN bufsize)
    127  1.15  jmcneill {
    128  1.15  jmcneill 	UINT8 *blkbuf, *blkbuf_start;
    129  1.15  jmcneill 	EFI_STATUS status;
    130  1.15  jmcneill 	EFI_LBA lba_start, lba_end;
    131  1.15  jmcneill 	UINT64 blkbuf_offset;
    132  1.19   mlelstv 	UINT64 blkbuf_size, alloc_size;
    133  1.15  jmcneill 
    134  1.15  jmcneill 	lba_start = off / bdev->bio->Media->BlockSize;
    135  1.19   mlelstv 	lba_end = (off + bufsize - 1) / bdev->bio->Media->BlockSize;
    136  1.15  jmcneill 	blkbuf_offset = off % bdev->bio->Media->BlockSize;
    137  1.19   mlelstv 	blkbuf_size = (lba_end - lba_start + 1) * bdev->bio->Media->BlockSize;
    138  1.19   mlelstv 
    139  1.19   mlelstv 	alloc_size = blkbuf_size;
    140  1.15  jmcneill 	if (bdev->bio->Media->IoAlign > 1) {
    141  1.19   mlelstv 		alloc_size = (blkbuf_size + bdev->bio->Media->IoAlign - 1) /
    142  1.16  jmcneill 		    bdev->bio->Media->IoAlign *
    143  1.16  jmcneill 		    bdev->bio->Media->IoAlign;
    144  1.15  jmcneill 	}
    145  1.15  jmcneill 
    146  1.19   mlelstv 	blkbuf = AllocatePool(alloc_size);
    147  1.15  jmcneill 	if (blkbuf == NULL) {
    148  1.15  jmcneill 		return EFI_OUT_OF_RESOURCES;
    149  1.15  jmcneill 	}
    150  1.15  jmcneill 
    151  1.15  jmcneill 	if (bdev->bio->Media->IoAlign > 1) {
    152  1.15  jmcneill 		blkbuf_start = (void *)roundup2((intptr_t)blkbuf,
    153  1.15  jmcneill 		    bdev->bio->Media->IoAlign);
    154  1.15  jmcneill 	} else {
    155  1.15  jmcneill 		blkbuf_start = blkbuf;
    156  1.15  jmcneill 	}
    157  1.15  jmcneill 
    158  1.15  jmcneill 	status = uefi_call_wrapper(bdev->bio->ReadBlocks, 5, bdev->bio,
    159  1.15  jmcneill 	    bdev->media_id, lba_start, blkbuf_size, blkbuf_start);
    160  1.15  jmcneill 	if (EFI_ERROR(status)) {
    161  1.15  jmcneill 		goto done;
    162  1.15  jmcneill 	}
    163  1.15  jmcneill 
    164  1.15  jmcneill 	memcpy(buf, blkbuf_start + blkbuf_offset, bufsize);
    165  1.15  jmcneill 
    166  1.15  jmcneill done:
    167  1.15  jmcneill 	FreePool(blkbuf);
    168  1.15  jmcneill 	return status;
    169  1.15  jmcneill }
    170  1.15  jmcneill 
    171  1.15  jmcneill static EFI_STATUS
    172  1.15  jmcneill efi_block_do_read_diskio(struct efi_block_dev *bdev, UINT64 off, void *buf,
    173  1.15  jmcneill     UINTN bufsize)
    174  1.15  jmcneill {
    175  1.15  jmcneill 	return uefi_call_wrapper(bdev->dio->ReadDisk, 5, bdev->dio,
    176  1.15  jmcneill 	    bdev->media_id, off, bufsize, buf);
    177  1.15  jmcneill }
    178  1.15  jmcneill 
    179  1.15  jmcneill static EFI_STATUS
    180  1.15  jmcneill efi_block_do_read(struct efi_block_dev *bdev, UINT64 off, void *buf,
    181  1.15  jmcneill     UINTN bufsize)
    182  1.15  jmcneill {
    183  1.15  jmcneill 	/*
    184  1.15  jmcneill 	 * Perform read access using EFI_DISK_IO_PROTOCOL if available,
    185  1.15  jmcneill 	 * otherwise use EFI_BLOCK_IO_PROTOCOL.
    186  1.15  jmcneill 	 */
    187  1.15  jmcneill 	if (bdev->dio != NULL) {
    188  1.15  jmcneill 		return efi_block_do_read_diskio(bdev, off, buf, bufsize);
    189  1.15  jmcneill 	} else {
    190  1.15  jmcneill 		return efi_block_do_read_blockio(bdev, off, buf, bufsize);
    191  1.15  jmcneill 	}
    192  1.15  jmcneill }
    193  1.15  jmcneill 
    194  1.15  jmcneill static EFI_STATUS
    195  1.15  jmcneill efi_block_readahead(struct efi_block_dev *bdev, UINT64 off, void *buf,
    196  1.14  jmcneill     UINTN bufsize)
    197  1.14  jmcneill {
    198  1.14  jmcneill 	EFI_STATUS status;
    199  1.14  jmcneill 	UINT64 mediasize, len;
    200  1.14  jmcneill 
    201  1.14  jmcneill 	if (efi_ra_buffer == NULL) {
    202  1.14  jmcneill 		efi_ra_buffer = AllocatePool(EFI_BLOCK_READAHEAD);
    203  1.14  jmcneill 		if (efi_ra_buffer == NULL) {
    204  1.14  jmcneill 			return EFI_OUT_OF_RESOURCES;
    205  1.14  jmcneill 		}
    206  1.14  jmcneill 	}
    207  1.14  jmcneill 
    208  1.14  jmcneill 	if (bdev->media_id != efi_ra_media_id ||
    209  1.14  jmcneill 	    off < efi_ra_start ||
    210  1.14  jmcneill 	    off + bufsize > efi_ra_start + efi_ra_length) {
    211  1.14  jmcneill 		mediasize = bdev->bio->Media->BlockSize *
    212  1.14  jmcneill 		    (bdev->bio->Media->LastBlock + 1);
    213  1.14  jmcneill 		len = EFI_BLOCK_READAHEAD;
    214  1.14  jmcneill 		if (len > mediasize - off) {
    215  1.14  jmcneill 			len = mediasize - off;
    216  1.14  jmcneill 		}
    217  1.15  jmcneill 		status = efi_block_do_read(bdev, off, efi_ra_buffer, len);
    218  1.14  jmcneill 		if (EFI_ERROR(status)) {
    219  1.14  jmcneill 			efi_ra_start = efi_ra_length = 0;
    220  1.14  jmcneill 			return status;
    221  1.14  jmcneill 		}
    222  1.14  jmcneill 		efi_ra_start = off;
    223  1.14  jmcneill 		efi_ra_length = len;
    224  1.14  jmcneill 		efi_ra_media_id = bdev->media_id;
    225  1.14  jmcneill 	}
    226  1.14  jmcneill 
    227  1.14  jmcneill 	memcpy(buf, &efi_ra_buffer[off - efi_ra_start], bufsize);
    228  1.14  jmcneill 	return EFI_SUCCESS;
    229  1.14  jmcneill }
    230  1.14  jmcneill 
    231  1.14  jmcneill static EFI_STATUS
    232  1.15  jmcneill efi_block_read(struct efi_block_dev *bdev, UINT64 off, void *buf,
    233  1.13  jmcneill     UINTN bufsize)
    234   1.6  jmcneill {
    235  1.14  jmcneill 	if (efi_ra_enable) {
    236  1.15  jmcneill 		return efi_block_readahead(bdev, off, buf, bufsize);
    237  1.14  jmcneill 	}
    238  1.14  jmcneill 
    239  1.15  jmcneill 	return efi_block_do_read(bdev, off, buf, bufsize);
    240   1.6  jmcneill }
    241   1.6  jmcneill 
    242   1.1  jmcneill static int
    243   1.8  jmcneill efi_block_find_partitions_cd9660(struct efi_block_dev *bdev)
    244   1.8  jmcneill {
    245   1.8  jmcneill 	struct efi_block_part *bpart;
    246  1.13  jmcneill 	struct iso_primary_descriptor vd;
    247   1.8  jmcneill 	EFI_STATUS status;
    248   1.8  jmcneill 	EFI_LBA lba;
    249   1.8  jmcneill 
    250   1.8  jmcneill 	for (lba = 16;; lba++) {
    251  1.15  jmcneill 		status = efi_block_read(bdev,
    252  1.13  jmcneill 		    lba * ISO_DEFAULT_BLOCK_SIZE, &vd, sizeof(vd));
    253  1.10  jmcneill 		if (EFI_ERROR(status)) {
    254   1.8  jmcneill 			goto io_error;
    255  1.10  jmcneill 		}
    256   1.8  jmcneill 
    257  1.13  jmcneill 		if (memcmp(vd.id, ISO_STANDARD_ID, sizeof vd.id) != 0) {
    258   1.8  jmcneill 			goto io_error;
    259  1.10  jmcneill 		}
    260  1.13  jmcneill 		if (isonum_711(vd.type) == ISO_VD_END) {
    261   1.8  jmcneill 			goto io_error;
    262  1.10  jmcneill 		}
    263  1.13  jmcneill 		if (isonum_711(vd.type) == ISO_VD_PRIMARY) {
    264   1.8  jmcneill 			break;
    265  1.10  jmcneill 		}
    266   1.8  jmcneill 	}
    267   1.8  jmcneill 
    268  1.13  jmcneill 	if (isonum_723(vd.logical_block_size) != ISO_DEFAULT_BLOCK_SIZE) {
    269   1.8  jmcneill 		goto io_error;
    270  1.10  jmcneill 	}
    271   1.8  jmcneill 
    272   1.8  jmcneill 	bpart = alloc(sizeof(*bpart));
    273   1.8  jmcneill 	bpart->index = 0;
    274   1.8  jmcneill 	bpart->bdev = bdev;
    275   1.8  jmcneill 	bpart->type = EFI_BLOCK_PART_CD9660;
    276   1.8  jmcneill 	TAILQ_INSERT_TAIL(&bdev->partitions, bpart, entries);
    277   1.8  jmcneill 
    278   1.8  jmcneill 	return 0;
    279   1.8  jmcneill 
    280   1.8  jmcneill io_error:
    281   1.8  jmcneill 	return EIO;
    282   1.8  jmcneill }
    283   1.8  jmcneill 
    284   1.8  jmcneill static int
    285  1.13  jmcneill efi_block_find_partitions_disklabel(struct efi_block_dev *bdev,
    286  1.13  jmcneill     struct mbr_sector *mbr, uint32_t start, uint32_t size)
    287   1.1  jmcneill {
    288   1.1  jmcneill 	struct efi_block_part *bpart;
    289  1.19   mlelstv 	char buf[DEV_BSIZE]; /* XXX, arbitrary size >= struct disklabel */
    290   1.1  jmcneill 	struct disklabel d;
    291   1.1  jmcneill 	struct partition *p;
    292   1.1  jmcneill 	EFI_STATUS status;
    293   1.1  jmcneill 	int n;
    294   1.1  jmcneill 
    295  1.15  jmcneill 	status = efi_block_read(bdev,
    296  1.19   mlelstv 	    ((EFI_LBA)start + LABELSECTOR) * bdev->bio->Media->BlockSize, buf, sizeof(buf));
    297  1.19   mlelstv 	if (EFI_ERROR(status) || getdisklabel(buf, &d) != NULL)
    298   1.1  jmcneill 		return EIO;
    299   1.1  jmcneill 
    300   1.1  jmcneill 	if (le32toh(d.d_magic) != DISKMAGIC || le32toh(d.d_magic2) != DISKMAGIC)
    301   1.1  jmcneill 		return EINVAL;
    302   1.1  jmcneill 	if (le16toh(d.d_npartitions) > MAXPARTITIONS)
    303   1.1  jmcneill 		return EINVAL;
    304   1.1  jmcneill 
    305   1.1  jmcneill 	for (n = 0; n < le16toh(d.d_npartitions); n++) {
    306   1.1  jmcneill 		p = &d.d_partitions[n];
    307   1.1  jmcneill 		switch (p->p_fstype) {
    308   1.1  jmcneill 		case FS_BSDFFS:
    309   1.1  jmcneill 		case FS_MSDOS:
    310   1.1  jmcneill 		case FS_BSDLFS:
    311   1.1  jmcneill 			break;
    312  1.11       mrg 		case FS_RAID:
    313  1.11       mrg 			p->p_size -= RF_PROTECTED_SECTORS;
    314  1.11       mrg 			p->p_offset += RF_PROTECTED_SECTORS;
    315  1.11       mrg 			break;
    316   1.1  jmcneill 		default:
    317   1.1  jmcneill 			continue;
    318   1.1  jmcneill 		}
    319   1.1  jmcneill 
    320   1.1  jmcneill 		bpart = alloc(sizeof(*bpart));
    321   1.1  jmcneill 		bpart->index = n;
    322   1.1  jmcneill 		bpart->bdev = bdev;
    323   1.1  jmcneill 		bpart->type = EFI_BLOCK_PART_DISKLABEL;
    324  1.11       mrg 		bpart->disklabel.secsize = d.d_secsize;
    325   1.1  jmcneill 		bpart->disklabel.part = *p;
    326   1.2  jmcneill 		efi_block_generate_hash_mbr(bpart, mbr);
    327   1.1  jmcneill 		TAILQ_INSERT_TAIL(&bdev->partitions, bpart, entries);
    328   1.1  jmcneill 	}
    329   1.1  jmcneill 
    330   1.1  jmcneill 	return 0;
    331   1.1  jmcneill }
    332   1.1  jmcneill 
    333   1.1  jmcneill static int
    334   1.1  jmcneill efi_block_find_partitions_mbr(struct efi_block_dev *bdev)
    335   1.1  jmcneill {
    336   1.1  jmcneill 	struct mbr_sector mbr;
    337   1.1  jmcneill 	struct mbr_partition *mbr_part;
    338   1.1  jmcneill 	EFI_STATUS status;
    339   1.1  jmcneill 	int n;
    340   1.1  jmcneill 
    341  1.15  jmcneill 	status = efi_block_read(bdev, 0, &mbr, sizeof(mbr));
    342  1.13  jmcneill 	if (EFI_ERROR(status))
    343   1.1  jmcneill 		return EIO;
    344   1.1  jmcneill 
    345   1.1  jmcneill 	if (le32toh(mbr.mbr_magic) != MBR_MAGIC)
    346   1.1  jmcneill 		return ENOENT;
    347   1.1  jmcneill 
    348   1.1  jmcneill 	for (n = 0; n < MBR_PART_COUNT; n++) {
    349   1.1  jmcneill 		mbr_part = &mbr.mbr_parts[n];
    350   1.1  jmcneill 		if (le32toh(mbr_part->mbrp_size) == 0)
    351   1.1  jmcneill 			continue;
    352   1.1  jmcneill 		if (mbr_part->mbrp_type == MBR_PTYPE_NETBSD) {
    353  1.13  jmcneill 			efi_block_find_partitions_disklabel(bdev, &mbr,
    354  1.13  jmcneill 			    le32toh(mbr_part->mbrp_start),
    355  1.13  jmcneill 			    le32toh(mbr_part->mbrp_size));
    356   1.1  jmcneill 			break;
    357   1.1  jmcneill 		}
    358   1.1  jmcneill 	}
    359   1.1  jmcneill 
    360   1.1  jmcneill 	return 0;
    361   1.1  jmcneill }
    362   1.1  jmcneill 
    363   1.4  jmcneill static const struct {
    364   1.4  jmcneill 	struct uuid guid;
    365   1.4  jmcneill 	uint8_t fstype;
    366   1.4  jmcneill } gpt_guid_to_str[] = {
    367   1.4  jmcneill 	{ GPT_ENT_TYPE_NETBSD_FFS,		FS_BSDFFS },
    368   1.4  jmcneill 	{ GPT_ENT_TYPE_NETBSD_LFS,		FS_BSDLFS },
    369   1.4  jmcneill 	{ GPT_ENT_TYPE_NETBSD_RAIDFRAME,	FS_RAID },
    370   1.4  jmcneill 	{ GPT_ENT_TYPE_NETBSD_CCD,		FS_CCD },
    371   1.4  jmcneill 	{ GPT_ENT_TYPE_NETBSD_CGD,		FS_CGD },
    372   1.4  jmcneill 	{ GPT_ENT_TYPE_MS_BASIC_DATA,		FS_MSDOS },	/* or NTFS? ambiguous */
    373   1.9       tnn 	{ GPT_ENT_TYPE_EFI,			FS_MSDOS },
    374   1.4  jmcneill };
    375   1.4  jmcneill 
    376   1.4  jmcneill static int
    377  1.13  jmcneill efi_block_find_partitions_gpt_entry(struct efi_block_dev *bdev,
    378  1.13  jmcneill     struct gpt_hdr *hdr, struct gpt_ent *ent, UINT32 index)
    379   1.4  jmcneill {
    380   1.4  jmcneill 	struct efi_block_part *bpart;
    381   1.4  jmcneill 	uint8_t fstype = FS_UNUSED;
    382   1.4  jmcneill 	struct uuid uuid;
    383   1.4  jmcneill 	int n;
    384   1.4  jmcneill 
    385   1.4  jmcneill 	memcpy(&uuid, ent->ent_type, sizeof(uuid));
    386   1.4  jmcneill 	for (n = 0; n < __arraycount(gpt_guid_to_str); n++)
    387  1.13  jmcneill 		if (memcmp(ent->ent_type, &gpt_guid_to_str[n].guid,
    388  1.13  jmcneill 		    sizeof(ent->ent_type)) == 0) {
    389   1.4  jmcneill 			fstype = gpt_guid_to_str[n].fstype;
    390   1.4  jmcneill 			break;
    391   1.4  jmcneill 		}
    392   1.4  jmcneill 	if (fstype == FS_UNUSED)
    393   1.4  jmcneill 		return 0;
    394   1.4  jmcneill 
    395   1.4  jmcneill 	bpart = alloc(sizeof(*bpart));
    396   1.4  jmcneill 	bpart->index = index;
    397   1.4  jmcneill 	bpart->bdev = bdev;
    398   1.4  jmcneill 	bpart->type = EFI_BLOCK_PART_GPT;
    399   1.4  jmcneill 	bpart->gpt.fstype = fstype;
    400   1.4  jmcneill 	bpart->gpt.ent = *ent;
    401  1.11       mrg 	if (fstype == FS_RAID) {
    402  1.11       mrg 		bpart->gpt.ent.ent_lba_start += RF_PROTECTED_SECTORS;
    403  1.11       mrg 		bpart->gpt.ent.ent_lba_end -= RF_PROTECTED_SECTORS;
    404  1.11       mrg 	}
    405   1.4  jmcneill 	memcpy(bpart->hash, ent->ent_guid, sizeof(bpart->hash));
    406   1.4  jmcneill 	TAILQ_INSERT_TAIL(&bdev->partitions, bpart, entries);
    407   1.4  jmcneill 
    408   1.4  jmcneill 	return 0;
    409   1.4  jmcneill }
    410   1.4  jmcneill 
    411   1.4  jmcneill static int
    412   1.4  jmcneill efi_block_find_partitions_gpt(struct efi_block_dev *bdev)
    413   1.4  jmcneill {
    414   1.4  jmcneill 	struct gpt_hdr hdr;
    415   1.4  jmcneill 	struct gpt_ent ent;
    416   1.4  jmcneill 	EFI_STATUS status;
    417  1.13  jmcneill 	UINT32 entry;
    418  1.13  jmcneill 	void *buf;
    419  1.13  jmcneill 	UINTN sz;
    420   1.4  jmcneill 
    421  1.19   mlelstv 	status = efi_block_read(bdev, (EFI_LBA)GPT_HDR_BLKNO * bdev->bio->Media->BlockSize, &hdr,
    422  1.13  jmcneill 	    sizeof(hdr));
    423   1.4  jmcneill 	if (EFI_ERROR(status)) {
    424   1.4  jmcneill 		return EIO;
    425   1.4  jmcneill 	}
    426   1.4  jmcneill 
    427   1.4  jmcneill 	if (memcmp(hdr.hdr_sig, GPT_HDR_SIG, sizeof(hdr.hdr_sig)) != 0)
    428   1.4  jmcneill 		return ENOENT;
    429   1.4  jmcneill 	if (le32toh(hdr.hdr_entsz) < sizeof(ent))
    430   1.4  jmcneill 		return EINVAL;
    431   1.4  jmcneill 
    432  1.13  jmcneill 	sz = le32toh(hdr.hdr_entsz) * le32toh(hdr.hdr_entries);
    433  1.13  jmcneill 	buf = AllocatePool(sz);
    434  1.13  jmcneill 	if (buf == NULL)
    435   1.4  jmcneill 		return ENOMEM;
    436   1.4  jmcneill 
    437  1.15  jmcneill 	status = efi_block_read(bdev,
    438  1.19   mlelstv 	    le64toh(hdr.hdr_lba_table) * bdev->bio->Media->BlockSize, buf, sz);
    439   1.4  jmcneill 	if (EFI_ERROR(status)) {
    440   1.4  jmcneill 		FreePool(buf);
    441   1.4  jmcneill 		return EIO;
    442   1.4  jmcneill 	}
    443   1.4  jmcneill 
    444   1.4  jmcneill 	for (entry = 0; entry < le32toh(hdr.hdr_entries); entry++) {
    445  1.20       rin 		memcpy(&ent, (UINT8 *)buf + (entry * le32toh(hdr.hdr_entsz)),
    446   1.6  jmcneill 			sizeof(ent));
    447   1.4  jmcneill 		efi_block_find_partitions_gpt_entry(bdev, &hdr, &ent, entry);
    448   1.4  jmcneill 	}
    449   1.4  jmcneill 
    450   1.4  jmcneill 	FreePool(buf);
    451   1.4  jmcneill 
    452   1.4  jmcneill 	return 0;
    453   1.4  jmcneill }
    454   1.4  jmcneill 
    455   1.1  jmcneill static int
    456   1.1  jmcneill efi_block_find_partitions(struct efi_block_dev *bdev)
    457   1.1  jmcneill {
    458   1.4  jmcneill 	int error;
    459   1.4  jmcneill 
    460   1.4  jmcneill 	error = efi_block_find_partitions_gpt(bdev);
    461   1.4  jmcneill 	if (error)
    462   1.4  jmcneill 		error = efi_block_find_partitions_mbr(bdev);
    463   1.8  jmcneill 	if (error)
    464   1.8  jmcneill 		error = efi_block_find_partitions_cd9660(bdev);
    465   1.4  jmcneill 
    466   1.4  jmcneill 	return error;
    467   1.1  jmcneill }
    468   1.1  jmcneill 
    469   1.1  jmcneill void
    470   1.1  jmcneill efi_block_probe(void)
    471   1.1  jmcneill {
    472   1.1  jmcneill 	struct efi_block_dev *bdev;
    473   1.4  jmcneill 	struct efi_block_part *bpart;
    474   1.1  jmcneill 	EFI_BLOCK_IO *bio;
    475  1.13  jmcneill 	EFI_DISK_IO *dio;
    476   1.1  jmcneill 	EFI_STATUS status;
    477   1.1  jmcneill 	uint16_t devindex = 0;
    478   1.1  jmcneill 	int depth = -1;
    479   1.1  jmcneill 	int n;
    480   1.1  jmcneill 
    481   1.1  jmcneill 	status = LibLocateHandle(ByProtocol, &BlockIoProtocol, NULL, &efi_nblock, &efi_block);
    482   1.1  jmcneill 	if (EFI_ERROR(status))
    483   1.1  jmcneill 		return;
    484   1.1  jmcneill 
    485   1.1  jmcneill 	if (efi_bootdp) {
    486   1.1  jmcneill 		depth = efi_device_path_depth(efi_bootdp, MEDIA_DEVICE_PATH);
    487   1.1  jmcneill 		if (depth == 0)
    488   1.1  jmcneill 			depth = 1;
    489   1.5  jmcneill 		else if (depth == -1)
    490   1.5  jmcneill 			depth = 2;
    491   1.1  jmcneill 	}
    492   1.1  jmcneill 
    493   1.1  jmcneill 	for (n = 0; n < efi_nblock; n++) {
    494  1.15  jmcneill 		/* EFI_BLOCK_IO_PROTOCOL is required */
    495  1.13  jmcneill 		status = uefi_call_wrapper(BS->HandleProtocol, 3, efi_block[n],
    496  1.13  jmcneill 		    &BlockIoProtocol, (void **)&bio);
    497   1.1  jmcneill 		if (EFI_ERROR(status) || !bio->Media->MediaPresent)
    498   1.1  jmcneill 			continue;
    499   1.1  jmcneill 
    500  1.15  jmcneill 		/* Ignore logical partitions (we do our own partition discovery) */
    501   1.1  jmcneill 		if (bio->Media->LogicalPartition)
    502   1.1  jmcneill 			continue;
    503   1.1  jmcneill 
    504  1.15  jmcneill 		/* EFI_DISK_IO_PROTOCOL is optional */
    505  1.13  jmcneill 		status = uefi_call_wrapper(BS->HandleProtocol, 3, efi_block[n],
    506  1.13  jmcneill 		    &DiskIoProtocol, (void **)&dio);
    507  1.15  jmcneill 		if (EFI_ERROR(status)) {
    508  1.15  jmcneill 			dio = NULL;
    509  1.15  jmcneill 		}
    510  1.13  jmcneill 
    511   1.1  jmcneill 		bdev = alloc(sizeof(*bdev));
    512   1.1  jmcneill 		bdev->index = devindex++;
    513   1.1  jmcneill 		bdev->bio = bio;
    514  1.13  jmcneill 		bdev->dio = dio;
    515   1.1  jmcneill 		bdev->media_id = bio->Media->MediaId;
    516   1.1  jmcneill 		bdev->path = DevicePathFromHandle(efi_block[n]);
    517   1.1  jmcneill 		TAILQ_INIT(&bdev->partitions);
    518   1.1  jmcneill 		TAILQ_INSERT_TAIL(&efi_block_devs, bdev, entries);
    519   1.1  jmcneill 
    520   1.4  jmcneill 		efi_block_find_partitions(bdev);
    521   1.4  jmcneill 
    522   1.1  jmcneill 		if (depth > 0 && efi_device_path_ncmp(efi_bootdp, DevicePathFromHandle(efi_block[n]), depth) == 0) {
    523   1.4  jmcneill 			TAILQ_FOREACH(bpart, &bdev->partitions, entries) {
    524   1.4  jmcneill 				uint8_t fstype = FS_UNUSED;
    525   1.4  jmcneill 				switch (bpart->type) {
    526   1.4  jmcneill 				case EFI_BLOCK_PART_DISKLABEL:
    527   1.4  jmcneill 					fstype = bpart->disklabel.part.p_fstype;
    528   1.4  jmcneill 					break;
    529   1.4  jmcneill 				case EFI_BLOCK_PART_GPT:
    530   1.4  jmcneill 					fstype = bpart->gpt.fstype;
    531   1.4  jmcneill 					break;
    532   1.8  jmcneill 				case EFI_BLOCK_PART_CD9660:
    533   1.8  jmcneill 					fstype = FS_ISO9660;
    534   1.8  jmcneill 					break;
    535   1.4  jmcneill 				}
    536  1.11       mrg 				if (fstype == FS_BSDFFS || fstype == FS_ISO9660 || fstype == FS_RAID) {
    537   1.4  jmcneill 					char devname[9];
    538   1.4  jmcneill 					snprintf(devname, sizeof(devname), "hd%u%c", bdev->index, bpart->index + 'a');
    539   1.4  jmcneill 					set_default_device(devname);
    540   1.8  jmcneill 					set_default_fstype(fstype);
    541   1.4  jmcneill 					break;
    542   1.4  jmcneill 				}
    543   1.4  jmcneill 			}
    544   1.1  jmcneill 		}
    545   1.4  jmcneill 	}
    546   1.4  jmcneill }
    547   1.1  jmcneill 
    548   1.4  jmcneill static void
    549   1.4  jmcneill print_guid(const uint8_t *guid)
    550   1.4  jmcneill {
    551   1.4  jmcneill 	const int index[] = { 3, 2, 1, 0, 5, 4, 7, 6, 8, 9, 10, 11, 12, 13, 14, 15 };
    552   1.4  jmcneill 	int i;
    553   1.4  jmcneill 
    554   1.4  jmcneill 	for (i = 0; i < 16; i++) {
    555   1.4  jmcneill 		printf("%02x", guid[index[i]]);
    556   1.4  jmcneill 		if (i == 3 || i == 5 || i == 7 || i == 9)
    557   1.4  jmcneill 			printf("-");
    558   1.1  jmcneill 	}
    559   1.1  jmcneill }
    560   1.1  jmcneill 
    561   1.1  jmcneill void
    562   1.1  jmcneill efi_block_show(void)
    563   1.1  jmcneill {
    564   1.1  jmcneill 	struct efi_block_dev *bdev;
    565   1.1  jmcneill 	struct efi_block_part *bpart;
    566   1.1  jmcneill 	uint64_t size;
    567   1.1  jmcneill 	CHAR16 *path;
    568   1.1  jmcneill 
    569   1.1  jmcneill 	TAILQ_FOREACH(bdev, &efi_block_devs, entries) {
    570   1.1  jmcneill 		printf("hd%u (", bdev->index);
    571   1.1  jmcneill 
    572   1.1  jmcneill 		/* Size in MB */
    573   1.1  jmcneill 		size = ((bdev->bio->Media->LastBlock + 1) * bdev->bio->Media->BlockSize) / (1024 * 1024);
    574   1.1  jmcneill 		if (size >= 10000)
    575   1.1  jmcneill 			printf("%"PRIu64" GB", size / 1024);
    576   1.1  jmcneill 		else
    577   1.1  jmcneill 			printf("%"PRIu64" MB", size);
    578   1.1  jmcneill 		printf("): ");
    579   1.1  jmcneill 
    580   1.1  jmcneill 		path = DevicePathToStr(bdev->path);
    581   1.1  jmcneill 		Print(L"%s", path);
    582   1.1  jmcneill 		FreePool(path);
    583   1.1  jmcneill 
    584   1.1  jmcneill 		printf("\n");
    585   1.1  jmcneill 
    586   1.1  jmcneill 		TAILQ_FOREACH(bpart, &bdev->partitions, entries) {
    587   1.1  jmcneill 			switch (bpart->type) {
    588   1.1  jmcneill 			case EFI_BLOCK_PART_DISKLABEL:
    589   1.1  jmcneill 				printf("  hd%u%c (", bdev->index, bpart->index + 'a');
    590   1.1  jmcneill 
    591   1.1  jmcneill 				/* Size in MB */
    592   1.1  jmcneill 				size = ((uint64_t)bpart->disklabel.secsize * bpart->disklabel.part.p_size) / (1024 * 1024);
    593   1.1  jmcneill 				if (size >= 10000)
    594   1.1  jmcneill 					printf("%"PRIu64" GB", size / 1024);
    595   1.1  jmcneill 				else
    596   1.1  jmcneill 					printf("%"PRIu64" MB", size);
    597   1.1  jmcneill 				printf("): ");
    598   1.1  jmcneill 
    599   1.1  jmcneill 				printf("%s\n", fstypenames[bpart->disklabel.part.p_fstype]);
    600   1.1  jmcneill 				break;
    601   1.4  jmcneill 			case EFI_BLOCK_PART_GPT:
    602   1.4  jmcneill 				printf("  hd%u%c ", bdev->index, bpart->index + 'a');
    603   1.4  jmcneill 
    604   1.4  jmcneill 				if (bpart->gpt.ent.ent_name[0] == 0x0000) {
    605   1.4  jmcneill 					printf("\"");
    606   1.4  jmcneill 					print_guid(bpart->gpt.ent.ent_guid);
    607   1.4  jmcneill 					printf("\"");
    608   1.4  jmcneill 				} else {
    609   1.4  jmcneill 					Print(L"\"%s\"", bpart->gpt.ent.ent_name);
    610   1.4  jmcneill 				}
    611  1.13  jmcneill 
    612   1.4  jmcneill 				/* Size in MB */
    613   1.4  jmcneill 				size = (le64toh(bpart->gpt.ent.ent_lba_end) - le64toh(bpart->gpt.ent.ent_lba_start)) * bdev->bio->Media->BlockSize;
    614   1.4  jmcneill 				size /= (1024 * 1024);
    615   1.4  jmcneill 				if (size >= 10000)
    616   1.4  jmcneill 					printf(" (%"PRIu64" GB): ", size / 1024);
    617   1.4  jmcneill 				else
    618   1.4  jmcneill 					printf(" (%"PRIu64" MB): ", size);
    619   1.4  jmcneill 
    620   1.4  jmcneill 				printf("%s\n", fstypenames[bpart->gpt.fstype]);
    621   1.4  jmcneill 				break;
    622   1.8  jmcneill 			case EFI_BLOCK_PART_CD9660:
    623   1.8  jmcneill 				printf("  hd%u%c %s\n", bdev->index, bpart->index + 'a', fstypenames[FS_ISO9660]);
    624   1.8  jmcneill 				break;
    625   1.1  jmcneill 			default:
    626   1.1  jmcneill 				break;
    627   1.1  jmcneill 			}
    628   1.1  jmcneill 		}
    629   1.1  jmcneill 	}
    630   1.1  jmcneill }
    631   1.1  jmcneill 
    632   1.2  jmcneill struct efi_block_part *
    633   1.2  jmcneill efi_block_boot_part(void)
    634   1.2  jmcneill {
    635   1.2  jmcneill 	return efi_block_booted;
    636   1.2  jmcneill }
    637   1.2  jmcneill 
    638   1.1  jmcneill int
    639   1.1  jmcneill efi_block_open(struct open_file *f, ...)
    640   1.1  jmcneill {
    641   1.1  jmcneill 	struct efi_block_part *bpart;
    642   1.1  jmcneill 	const char *fname;
    643   1.1  jmcneill 	char **file;
    644   1.1  jmcneill 	char *path;
    645   1.1  jmcneill 	va_list ap;
    646   1.1  jmcneill 	int rv, n;
    647  1.13  jmcneill 
    648   1.1  jmcneill 	va_start(ap, f);
    649   1.1  jmcneill 	fname = va_arg(ap, const char *);
    650   1.1  jmcneill 	file = va_arg(ap, char **);
    651   1.1  jmcneill 	va_end(ap);
    652   1.1  jmcneill 
    653   1.1  jmcneill 	rv = efi_block_parse(fname, &bpart, &path);
    654   1.1  jmcneill 	if (rv != 0)
    655   1.1  jmcneill 		return rv;
    656   1.1  jmcneill 
    657   1.1  jmcneill 	for (n = 0; n < ndevs; n++)
    658   1.1  jmcneill 		if (strcmp(DEV_NAME(&devsw[n]), "efiblock") == 0) {
    659   1.1  jmcneill 			f->f_dev = &devsw[n];
    660   1.1  jmcneill 			break;
    661   1.1  jmcneill 		}
    662   1.1  jmcneill 	if (n == ndevs)
    663   1.1  jmcneill 		return ENXIO;
    664   1.1  jmcneill 
    665   1.1  jmcneill 	f->f_devdata = bpart;
    666   1.1  jmcneill 
    667   1.1  jmcneill 	*file = path;
    668   1.1  jmcneill 
    669   1.2  jmcneill 	efi_block_booted = bpart;
    670   1.2  jmcneill 
    671   1.1  jmcneill 	return 0;
    672   1.1  jmcneill }
    673   1.1  jmcneill 
    674   1.1  jmcneill int
    675   1.1  jmcneill efi_block_close(struct open_file *f)
    676   1.1  jmcneill {
    677   1.1  jmcneill 	return 0;
    678   1.1  jmcneill }
    679   1.1  jmcneill 
    680   1.1  jmcneill int
    681   1.1  jmcneill efi_block_strategy(void *devdata, int rw, daddr_t dblk, size_t size, void *buf, size_t *rsize)
    682   1.1  jmcneill {
    683   1.1  jmcneill 	struct efi_block_part *bpart = devdata;
    684  1.19   mlelstv 	struct efi_block_dev *bdev = bpart->bdev;
    685   1.1  jmcneill 	EFI_STATUS status;
    686  1.13  jmcneill 	UINT64 off;
    687   1.1  jmcneill 
    688   1.1  jmcneill 	if (rw != F_READ)
    689   1.1  jmcneill 		return EROFS;
    690   1.1  jmcneill 
    691  1.12  jmcneill 	efi_set_watchdog(EFI_BLOCK_TIMEOUT, EFI_BLOCK_TIMEOUT_CODE);
    692  1.12  jmcneill 
    693   1.1  jmcneill 	switch (bpart->type) {
    694   1.1  jmcneill 	case EFI_BLOCK_PART_DISKLABEL:
    695  1.19   mlelstv 		off = ((EFI_LBA)dblk + bpart->disklabel.part.p_offset) * bdev->bio->Media->BlockSize;
    696   1.1  jmcneill 		break;
    697   1.4  jmcneill 	case EFI_BLOCK_PART_GPT:
    698  1.19   mlelstv 		off = ((EFI_LBA)dblk + le64toh(bpart->gpt.ent.ent_lba_start)) * bdev->bio->Media->BlockSize;
    699   1.4  jmcneill 		break;
    700   1.8  jmcneill 	case EFI_BLOCK_PART_CD9660:
    701  1.19   mlelstv 		off = (EFI_LBA)dblk * ISO_DEFAULT_BLOCK_SIZE;
    702   1.8  jmcneill 		break;
    703   1.1  jmcneill 	default:
    704   1.1  jmcneill 		return EINVAL;
    705   1.1  jmcneill 	}
    706   1.1  jmcneill 
    707  1.15  jmcneill 	status = efi_block_read(bpart->bdev, off, buf, size);
    708  1.13  jmcneill 	if (EFI_ERROR(status))
    709   1.1  jmcneill 		return EIO;
    710   1.1  jmcneill 
    711   1.1  jmcneill 	*rsize = size;
    712   1.1  jmcneill 
    713   1.1  jmcneill 	return 0;
    714   1.1  jmcneill }
    715  1.14  jmcneill 
    716  1.14  jmcneill void
    717  1.14  jmcneill efi_block_set_readahead(bool onoff)
    718  1.14  jmcneill {
    719  1.14  jmcneill 	efi_ra_enable = onoff;
    720  1.14  jmcneill }
    721  1.19   mlelstv 
    722  1.19   mlelstv int
    723  1.19   mlelstv efi_block_ioctl(struct open_file *f, u_long cmd, void *data)
    724  1.19   mlelstv {
    725  1.19   mlelstv 	struct efi_block_part *bpart = f->f_devdata;
    726  1.19   mlelstv 	struct efi_block_dev *bdev = bpart->bdev;
    727  1.19   mlelstv 	int error = 0;
    728  1.19   mlelstv 
    729  1.19   mlelstv 	switch (cmd) {
    730  1.19   mlelstv 	case SAIOSECSIZE:
    731  1.19   mlelstv 		*(u_int *)data = bdev->bio->Media->BlockSize;
    732  1.19   mlelstv 		break;
    733  1.19   mlelstv 	default:
    734  1.19   mlelstv 		error = ENOTTY;
    735  1.19   mlelstv 		break;
    736  1.19   mlelstv 	}
    737  1.19   mlelstv 
    738  1.19   mlelstv 	return error;
    739  1.19   mlelstv }
    740