2 * Copyright (c) 2012 Andrey V. Elsukov <ae@FreeBSD.org>
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 #include <sys/cdefs.h>
28 __FBSDID("$FreeBSD$");
31 #include <sys/param.h>
32 #include <sys/diskmbr.h>
33 #include <sys/disklabel.h>
34 #include <sys/endian.h>
36 #include <sys/stddef.h>
37 #include <sys/queue.h>
40 #include <fs/cd9660/iso.h>
47 #define DPRINTF(fmt, args...) printf("%s: " fmt "\n", __func__, ## args)
49 #define DPRINTF(fmt, args...)
52 #ifdef LOADER_GPT_SUPPORT
54 static const uuid_t gpt_uuid_unused = GPT_ENT_TYPE_UNUSED;
55 static const uuid_t gpt_uuid_ms_basic_data = GPT_ENT_TYPE_MS_BASIC_DATA;
56 static const uuid_t gpt_uuid_freebsd_ufs = GPT_ENT_TYPE_FREEBSD_UFS;
57 static const uuid_t gpt_uuid_efi = GPT_ENT_TYPE_EFI;
58 static const uuid_t gpt_uuid_freebsd = GPT_ENT_TYPE_FREEBSD;
59 static const uuid_t gpt_uuid_freebsd_boot = GPT_ENT_TYPE_FREEBSD_BOOT;
60 static const uuid_t gpt_uuid_freebsd_nandfs = GPT_ENT_TYPE_FREEBSD_NANDFS;
61 static const uuid_t gpt_uuid_freebsd_swap = GPT_ENT_TYPE_FREEBSD_SWAP;
62 static const uuid_t gpt_uuid_freebsd_zfs = GPT_ENT_TYPE_FREEBSD_ZFS;
63 static const uuid_t gpt_uuid_freebsd_vinum = GPT_ENT_TYPE_FREEBSD_VINUM;
67 struct ptable_entry part;
75 STAILQ_ENTRY(pentry) entry;
79 enum ptable_type type;
83 STAILQ_HEAD(, pentry) entries;
86 static struct parttypes {
87 enum partition_type type;
90 { PART_UNKNOWN, "Unknown" },
92 { PART_FREEBSD, "FreeBSD" },
93 { PART_FREEBSD_BOOT, "FreeBSD boot" },
94 { PART_FREEBSD_NANDFS, "FreeBSD nandfs" },
95 { PART_FREEBSD_UFS, "FreeBSD UFS" },
96 { PART_FREEBSD_ZFS, "FreeBSD ZFS" },
97 { PART_FREEBSD_SWAP, "FreeBSD swap" },
98 { PART_FREEBSD_VINUM, "FreeBSD vinum" },
99 { PART_LINUX, "Linux" },
100 { PART_LINUX_SWAP, "Linux swap" },
101 { PART_DOS, "DOS/Windows" },
102 { PART_ISO9660, "ISO9660" },
106 parttype2str(enum partition_type type)
110 for (i = 0; i < nitems(ptypes); i++)
111 if (ptypes[i].type == type)
112 return (ptypes[i].desc);
113 return (ptypes[0].desc);
116 #ifdef LOADER_GPT_SUPPORT
118 uuid_letoh(uuid_t *uuid)
121 uuid->time_low = le32toh(uuid->time_low);
122 uuid->time_mid = le16toh(uuid->time_mid);
123 uuid->time_hi_and_version = le16toh(uuid->time_hi_and_version);
126 static enum partition_type
127 gpt_parttype(uuid_t type)
130 if (uuid_equal(&type, &gpt_uuid_efi, NULL))
132 else if (uuid_equal(&type, &gpt_uuid_ms_basic_data, NULL))
134 else if (uuid_equal(&type, &gpt_uuid_freebsd_boot, NULL))
135 return (PART_FREEBSD_BOOT);
136 else if (uuid_equal(&type, &gpt_uuid_freebsd_ufs, NULL))
137 return (PART_FREEBSD_UFS);
138 else if (uuid_equal(&type, &gpt_uuid_freebsd_zfs, NULL))
139 return (PART_FREEBSD_ZFS);
140 else if (uuid_equal(&type, &gpt_uuid_freebsd_swap, NULL))
141 return (PART_FREEBSD_SWAP);
142 else if (uuid_equal(&type, &gpt_uuid_freebsd_vinum, NULL))
143 return (PART_FREEBSD_VINUM);
144 else if (uuid_equal(&type, &gpt_uuid_freebsd_nandfs, NULL))
145 return (PART_FREEBSD_NANDFS);
146 else if (uuid_equal(&type, &gpt_uuid_freebsd, NULL))
147 return (PART_FREEBSD);
148 return (PART_UNKNOWN);
151 static struct gpt_hdr *
152 gpt_checkhdr(struct gpt_hdr *hdr, uint64_t lba_self, uint64_t lba_last,
157 if (memcmp(hdr->hdr_sig, GPT_HDR_SIG, sizeof(hdr->hdr_sig)) != 0) {
158 DPRINTF("no GPT signature");
161 sz = le32toh(hdr->hdr_size);
162 if (sz < 92 || sz > sectorsize) {
163 DPRINTF("invalid GPT header size: %d", sz);
166 crc = le32toh(hdr->hdr_crc_self);
167 hdr->hdr_crc_self = 0;
168 if (crc32(hdr, sz) != crc) {
169 DPRINTF("GPT header's CRC doesn't match");
172 hdr->hdr_crc_self = crc;
173 hdr->hdr_revision = le32toh(hdr->hdr_revision);
174 if (hdr->hdr_revision < GPT_HDR_REVISION) {
175 DPRINTF("unsupported GPT revision %d", hdr->hdr_revision);
178 hdr->hdr_lba_self = le64toh(hdr->hdr_lba_self);
179 if (hdr->hdr_lba_self != lba_self) {
180 DPRINTF("self LBA doesn't match");
183 hdr->hdr_lba_alt = le64toh(hdr->hdr_lba_alt);
184 if (hdr->hdr_lba_alt == hdr->hdr_lba_self) {
185 DPRINTF("invalid alternate LBA");
188 hdr->hdr_entries = le32toh(hdr->hdr_entries);
189 hdr->hdr_entsz = le32toh(hdr->hdr_entsz);
190 if (hdr->hdr_entries == 0 ||
191 hdr->hdr_entsz < sizeof(struct gpt_ent) ||
192 sectorsize % hdr->hdr_entsz != 0) {
193 DPRINTF("invalid entry size or number of entries");
196 hdr->hdr_lba_start = le64toh(hdr->hdr_lba_start);
197 hdr->hdr_lba_end = le64toh(hdr->hdr_lba_end);
198 hdr->hdr_lba_table = le64toh(hdr->hdr_lba_table);
199 hdr->hdr_crc_table = le32toh(hdr->hdr_crc_table);
200 uuid_letoh(&hdr->hdr_uuid);
205 gpt_checktbl(const struct gpt_hdr *hdr, uint8_t *tbl, size_t size,
211 cnt = size / hdr->hdr_entsz;
212 if (hdr->hdr_entries <= cnt) {
213 cnt = hdr->hdr_entries;
214 /* Check CRC only when buffer size is enough for table. */
215 if (hdr->hdr_crc_table !=
216 crc32(tbl, hdr->hdr_entries * hdr->hdr_entsz)) {
217 DPRINTF("GPT table's CRC doesn't match");
221 for (i = 0; i < cnt; i++) {
222 ent = (struct gpt_ent *)(tbl + i * hdr->hdr_entsz);
223 uuid_letoh(&ent->ent_type);
224 if (uuid_equal(&ent->ent_type, &gpt_uuid_unused, NULL))
226 ent->ent_lba_start = le64toh(ent->ent_lba_start);
227 ent->ent_lba_end = le64toh(ent->ent_lba_end);
232 static struct ptable *
233 ptable_gptread(struct ptable *table, void *dev, diskread_t dread)
235 struct pentry *entry;
236 struct gpt_hdr *phdr, hdr;
243 buf = malloc(table->sectorsize);
246 tbl = malloc(table->sectorsize * MAXTBLSZ);
251 /* Read the primary GPT header. */
252 if (dread(dev, buf, 1, 1) != 0) {
258 /* Check the primary GPT header. */
259 phdr = gpt_checkhdr((struct gpt_hdr *)buf, 1, table->sectors - 1,
262 /* Read the primary GPT table. */
264 howmany(phdr->hdr_entries * phdr->hdr_entsz,
266 if (dread(dev, tbl, size, phdr->hdr_lba_table) == 0 &&
267 gpt_checktbl(phdr, tbl, size * table->sectorsize,
268 table->sectors - 1) == 0) {
269 memcpy(&hdr, phdr, sizeof(hdr));
273 offset = pri ? hdr.hdr_lba_alt: table->sectors - 1;
274 /* Read the backup GPT header. */
275 if (dread(dev, buf, 1, offset) != 0)
278 phdr = gpt_checkhdr((struct gpt_hdr *)buf, offset,
279 table->sectors - 1, table->sectorsize);
282 * Compare primary and backup headers.
283 * If they are equal, then we do not need to read backup
284 * table. If they are different, then prefer backup header
285 * and try to read backup table.
288 uuid_equal(&hdr.hdr_uuid, &phdr->hdr_uuid, NULL) == 0 ||
289 hdr.hdr_revision != phdr->hdr_revision ||
290 hdr.hdr_size != phdr->hdr_size ||
291 hdr.hdr_lba_start != phdr->hdr_lba_start ||
292 hdr.hdr_lba_end != phdr->hdr_lba_end ||
293 hdr.hdr_entries != phdr->hdr_entries ||
294 hdr.hdr_entsz != phdr->hdr_entsz ||
295 hdr.hdr_crc_table != phdr->hdr_crc_table) {
296 /* Read the backup GPT table. */
298 howmany(phdr->hdr_entries * phdr->hdr_entsz,
300 if (dread(dev, tbl, size, phdr->hdr_lba_table) == 0 &&
301 gpt_checktbl(phdr, tbl, size * table->sectorsize,
302 table->sectors - 1) == 0) {
303 memcpy(&hdr, phdr, sizeof(hdr));
308 if (pri == 0 && sec == 0) {
309 /* Both primary and backup tables are invalid. */
310 table->type = PTABLE_NONE;
313 DPRINTF("GPT detected");
314 size = MIN(hdr.hdr_entries * hdr.hdr_entsz,
315 MAXTBLSZ * table->sectorsize);
318 * If the disk's sector count is smaller than the sector count recorded
319 * in the disk's GPT table header, set the table->sectors to the value
320 * recorded in GPT tables. This is done to work around buggy firmware
321 * that returns truncated disk sizes.
323 * Note, this is still not a foolproof way to get disk's size. For
324 * example, an image file can be truncated when copied to smaller media.
326 table->sectors = hdr.hdr_lba_alt + 1;
328 for (i = 0; i < size / hdr.hdr_entsz; i++) {
329 ent = (struct gpt_ent *)(tbl + i * hdr.hdr_entsz);
330 if (uuid_equal(&ent->ent_type, &gpt_uuid_unused, NULL))
333 /* Simple sanity checks. */
334 if (ent->ent_lba_start < hdr.hdr_lba_start ||
335 ent->ent_lba_end > hdr.hdr_lba_end ||
336 ent->ent_lba_start > ent->ent_lba_end)
339 entry = malloc(sizeof(*entry));
342 entry->part.start = ent->ent_lba_start;
343 entry->part.end = ent->ent_lba_end;
344 entry->part.index = i + 1;
345 entry->part.type = gpt_parttype(ent->ent_type);
346 entry->flags = le64toh(ent->ent_attr);
347 memcpy(&entry->type.gpt, &ent->ent_type, sizeof(uuid_t));
348 STAILQ_INSERT_TAIL(&table->entries, entry, entry);
349 DPRINTF("new GPT partition added");
356 #endif /* LOADER_GPT_SUPPORT */
358 #ifdef LOADER_MBR_SUPPORT
359 /* We do not need to support too many EBR partitions in the loader */
360 #define MAXEBRENTRIES 8
361 static enum partition_type
362 mbr_parttype(uint8_t type)
367 return (PART_FREEBSD);
369 return (PART_LINUX_SWAP);
381 return (PART_UNKNOWN);
384 static struct ptable *
385 ptable_ebrread(struct ptable *table, void *dev, diskread_t dread)
387 struct dos_partition *dp;
388 struct pentry *e1, *entry;
389 uint32_t start, end, offset;
393 STAILQ_FOREACH(e1, &table->entries, entry) {
394 if (e1->type.mbr == DOSPTYP_EXT ||
395 e1->type.mbr == DOSPTYP_EXTLBA)
401 offset = e1->part.start;
402 buf = malloc(table->sectorsize);
405 DPRINTF("EBR detected");
406 for (i = 0; i < MAXEBRENTRIES; i++) {
407 #if 0 /* Some BIOSes return an incorrect number of sectors */
408 if (offset >= table->sectors)
411 if (dread(dev, buf, 1, offset) != 0)
413 dp = (struct dos_partition *)(buf + DOSPARTOFF);
414 if (dp[0].dp_typ == 0)
416 start = le32toh(dp[0].dp_start);
417 if (dp[0].dp_typ == DOSPTYP_EXT &&
419 offset = e1->part.start + start;
422 end = le32toh(dp[0].dp_size);
423 entry = malloc(sizeof(*entry));
426 entry->part.start = offset + start;
427 entry->part.end = entry->part.start + end - 1;
428 entry->part.index = index++;
429 entry->part.type = mbr_parttype(dp[0].dp_typ);
430 entry->flags = dp[0].dp_flag;
431 entry->type.mbr = dp[0].dp_typ;
432 STAILQ_INSERT_TAIL(&table->entries, entry, entry);
433 DPRINTF("new EBR partition added");
434 if (dp[1].dp_typ == 0)
436 offset = e1->part.start + le32toh(dp[1].dp_start);
441 #endif /* LOADER_MBR_SUPPORT */
443 static enum partition_type
444 bsd_parttype(uint8_t type)
449 return (PART_FREEBSD_NANDFS);
451 return (PART_FREEBSD_SWAP);
453 return (PART_FREEBSD_UFS);
455 return (PART_FREEBSD_VINUM);
457 return (PART_FREEBSD_ZFS);
459 return (PART_UNKNOWN);
462 static struct ptable *
463 ptable_bsdread(struct ptable *table, void *dev, diskread_t dread)
465 struct disklabel *dl;
466 struct partition *part;
467 struct pentry *entry;
472 if (table->sectorsize < sizeof(struct disklabel)) {
473 DPRINTF("Too small sectorsize");
476 buf = malloc(table->sectorsize);
479 if (dread(dev, buf, 1, 1) != 0) {
480 DPRINTF("read failed");
485 dl = (struct disklabel *)buf;
486 if (le32toh(dl->d_magic) != DISKMAGIC &&
487 le32toh(dl->d_magic2) != DISKMAGIC)
489 if (le32toh(dl->d_secsize) != table->sectorsize) {
490 DPRINTF("unsupported sector size");
493 dl->d_npartitions = le16toh(dl->d_npartitions);
494 if (dl->d_npartitions > 20 || dl->d_npartitions < 8) {
495 DPRINTF("invalid number of partitions");
498 DPRINTF("BSD detected");
499 part = &dl->d_partitions[0];
500 raw_offset = le32toh(part[RAW_PART].p_offset);
501 for (i = 0; i < dl->d_npartitions; i++, part++) {
504 if (part->p_size == 0)
506 entry = malloc(sizeof(*entry));
509 entry->part.start = le32toh(part->p_offset) - raw_offset;
510 entry->part.end = entry->part.start +
511 le32toh(part->p_size) - 1;
512 entry->part.type = bsd_parttype(part->p_fstype);
513 entry->part.index = i; /* starts from zero */
514 entry->type.bsd = part->p_fstype;
515 STAILQ_INSERT_TAIL(&table->entries, entry, entry);
516 DPRINTF("new BSD partition added");
518 table->type = PTABLE_BSD;
524 #ifdef LOADER_VTOC8_SUPPORT
525 static enum partition_type
526 vtoc8_parttype(uint16_t type)
530 case VTOC_TAG_FREEBSD_NANDFS:
531 return (PART_FREEBSD_NANDFS);
532 case VTOC_TAG_FREEBSD_SWAP:
533 return (PART_FREEBSD_SWAP);
534 case VTOC_TAG_FREEBSD_UFS:
535 return (PART_FREEBSD_UFS);
536 case VTOC_TAG_FREEBSD_VINUM:
537 return (PART_FREEBSD_VINUM);
538 case VTOC_TAG_FREEBSD_ZFS:
539 return (PART_FREEBSD_ZFS);
541 return (PART_UNKNOWN);
544 static struct ptable *
545 ptable_vtoc8read(struct ptable *table, void *dev, diskread_t dread)
547 struct pentry *entry;
550 uint16_t sum, heads, sectors;
553 if (table->sectorsize != sizeof(struct vtoc8))
555 buf = malloc(table->sectorsize);
558 if (dread(dev, buf, 1, 0) != 0) {
559 DPRINTF("read failed");
564 dl = (struct vtoc8 *)buf;
566 for (i = sum = 0; i < sizeof(struct vtoc8); i += sizeof(sum))
567 sum ^= be16dec(buf + i);
569 DPRINTF("incorrect checksum");
572 if (be16toh(dl->nparts) != VTOC8_NPARTS) {
573 DPRINTF("invalid number of entries");
576 sectors = be16toh(dl->nsecs);
577 heads = be16toh(dl->nheads);
578 if (sectors * heads == 0) {
579 DPRINTF("invalid geometry");
582 DPRINTF("VTOC8 detected");
583 for (i = 0; i < VTOC8_NPARTS; i++) {
584 dl->part[i].tag = be16toh(dl->part[i].tag);
585 if (i == VTOC_RAW_PART ||
586 dl->part[i].tag == VTOC_TAG_UNASSIGNED)
588 entry = malloc(sizeof(*entry));
591 entry->part.start = be32toh(dl->map[i].cyl) * heads * sectors;
592 entry->part.end = be32toh(dl->map[i].nblks) +
593 entry->part.start - 1;
594 entry->part.type = vtoc8_parttype(dl->part[i].tag);
595 entry->part.index = i; /* starts from zero */
596 entry->type.vtoc8 = dl->part[i].tag;
597 STAILQ_INSERT_TAIL(&table->entries, entry, entry);
598 DPRINTF("new VTOC8 partition added");
600 table->type = PTABLE_VTOC8;
606 #endif /* LOADER_VTOC8_SUPPORT */
608 #define cdb2devb(bno) ((bno) * ISO_DEFAULT_BLOCK_SIZE / table->sectorsize)
610 static struct ptable *
611 ptable_iso9660read(struct ptable *table, void *dev, diskread_t dread)
614 struct iso_primary_descriptor *vd;
615 struct pentry *entry;
617 buf = malloc(table->sectorsize);
621 if (dread(dev, buf, 1, cdb2devb(16)) != 0) {
622 DPRINTF("read failed");
627 vd = (struct iso_primary_descriptor *)buf;
628 if (bcmp(vd->id, ISO_STANDARD_ID, sizeof vd->id) != 0)
631 entry = malloc(sizeof(*entry));
634 entry->part.start = 0;
635 entry->part.end = table->sectors;
636 entry->part.type = PART_ISO9660;
637 entry->part.index = 0;
638 STAILQ_INSERT_TAIL(&table->entries, entry, entry);
640 table->type = PTABLE_ISO9660;
648 ptable_open(void *dev, uint64_t sectors, uint16_t sectorsize,
651 struct dos_partition *dp;
652 struct ptable *table;
655 #ifdef LOADER_MBR_SUPPORT
656 struct pentry *entry;
661 buf = malloc(sectorsize);
664 /* First, read the MBR. */
665 if (dread(dev, buf, 1, DOSBBSECTOR) != 0) {
666 DPRINTF("read failed");
670 table = malloc(sizeof(*table));
673 table->sectors = sectors;
674 table->sectorsize = sectorsize;
675 table->type = PTABLE_NONE;
676 STAILQ_INIT(&table->entries);
678 if (ptable_iso9660read(table, dev, dread) == NULL) {
682 } else if (table->type == PTABLE_ISO9660)
685 #ifdef LOADER_VTOC8_SUPPORT
686 if (be16dec(buf + offsetof(struct vtoc8, magic)) == VTOC_MAGIC) {
687 if (ptable_vtoc8read(table, dev, dread) == NULL) {
691 } else if (table->type == PTABLE_VTOC8)
695 /* Check the BSD label. */
696 if (ptable_bsdread(table, dev, dread) == NULL) { /* Read error. */
699 } else if (table->type == PTABLE_BSD)
702 #if defined(LOADER_GPT_SUPPORT) || defined(LOADER_MBR_SUPPORT)
703 /* Check the MBR magic. */
704 if (buf[DOSMAGICOFFSET] != 0x55 ||
705 buf[DOSMAGICOFFSET + 1] != 0xaa) {
706 DPRINTF("magic sequence not found");
707 #if defined(LOADER_GPT_SUPPORT)
708 /* There is no PMBR, check that we have backup GPT */
709 table->type = PTABLE_GPT;
710 table = ptable_gptread(table, dev, dread);
714 /* Check that we have PMBR. Also do some validation. */
715 dp = (struct dos_partition *)(buf + DOSPARTOFF);
716 for (i = 0, count = 0; i < NDOSPART; i++) {
717 if (dp[i].dp_flag != 0 && dp[i].dp_flag != 0x80) {
718 DPRINTF("invalid partition flag %x", dp[i].dp_flag);
721 #ifdef LOADER_GPT_SUPPORT
722 if (dp[i].dp_typ == DOSPTYP_PMBR) {
723 table->type = PTABLE_GPT;
724 DPRINTF("PMBR detected");
727 if (dp[i].dp_typ != 0)
730 /* Do we have some invalid values? */
731 if (table->type == PTABLE_GPT && count > 1) {
732 if (dp[1].dp_typ != DOSPTYP_HFS) {
733 table->type = PTABLE_NONE;
734 DPRINTF("Incorrect PMBR, ignore it");
736 DPRINTF("Bootcamp detected");
739 #ifdef LOADER_GPT_SUPPORT
740 if (table->type == PTABLE_GPT) {
741 table = ptable_gptread(table, dev, dread);
745 #ifdef LOADER_MBR_SUPPORT
747 DPRINTF("MBR detected");
748 table->type = PTABLE_MBR;
749 for (i = has_ext = 0; i < NDOSPART; i++) {
750 if (dp[i].dp_typ == 0)
752 start = le32dec(&(dp[i].dp_start));
753 end = le32dec(&(dp[i].dp_size));
754 if (start == 0 || end == 0)
756 #if 0 /* Some BIOSes return an incorrect number of sectors */
757 if (start + end - 1 >= sectors)
758 continue; /* XXX: ignore */
760 if (dp[i].dp_typ == DOSPTYP_EXT ||
761 dp[i].dp_typ == DOSPTYP_EXTLBA)
763 entry = malloc(sizeof(*entry));
766 entry->part.start = start;
767 entry->part.end = start + end - 1;
768 entry->part.index = i + 1;
769 entry->part.type = mbr_parttype(dp[i].dp_typ);
770 entry->flags = dp[i].dp_flag;
771 entry->type.mbr = dp[i].dp_typ;
772 STAILQ_INSERT_TAIL(&table->entries, entry, entry);
773 DPRINTF("new MBR partition added");
776 table = ptable_ebrread(table, dev, dread);
779 #endif /* LOADER_MBR_SUPPORT */
780 #endif /* LOADER_MBR_SUPPORT || LOADER_GPT_SUPPORT */
787 ptable_close(struct ptable *table)
789 struct pentry *entry;
794 while (!STAILQ_EMPTY(&table->entries)) {
795 entry = STAILQ_FIRST(&table->entries);
796 STAILQ_REMOVE_HEAD(&table->entries, entry);
803 ptable_gettype(const struct ptable *table)
806 return (table->type);
810 ptable_getsize(const struct ptable *table, uint64_t *sizep)
812 uint64_t tmp = table->sectors * table->sectorsize;
814 if (tmp < table->sectors)
823 ptable_getpart(const struct ptable *table, struct ptable_entry *part, int index)
825 struct pentry *entry;
827 if (part == NULL || table == NULL)
830 STAILQ_FOREACH(entry, &table->entries, entry) {
831 if (entry->part.index != index)
833 memcpy(part, &entry->part, sizeof(*part));
840 * Search for a slice with the following preferences:
842 * 1: Active FreeBSD slice
843 * 2: Non-active FreeBSD slice
844 * 3: Active Linux slice
845 * 4: non-active Linux slice
846 * 5: Active FAT/FAT32 slice
847 * 6: non-active FAT/FAT32 slice
849 #define PREF_RAWDISK 0
850 #define PREF_FBSD_ACT 1
852 #define PREF_LINUX_ACT 3
854 #define PREF_DOS_ACT 5
858 ptable_getbestpart(const struct ptable *table, struct ptable_entry *part)
860 struct pentry *entry, *best;
863 if (part == NULL || table == NULL)
867 preflevel = pref = PREF_NONE;
868 STAILQ_FOREACH(entry, &table->entries, entry) {
869 #ifdef LOADER_MBR_SUPPORT
870 if (table->type == PTABLE_MBR) {
871 switch (entry->type.mbr) {
873 pref = entry->flags & 0x80 ? PREF_FBSD_ACT:
877 pref = entry->flags & 0x80 ? PREF_LINUX_ACT:
880 case 0x01: /* DOS/Windows */
886 pref = entry->flags & 0x80 ? PREF_DOS_ACT:
893 #endif /* LOADER_MBR_SUPPORT */
894 #ifdef LOADER_GPT_SUPPORT
895 if (table->type == PTABLE_GPT) {
896 if (entry->part.type == PART_DOS)
898 else if (entry->part.type == PART_FREEBSD_UFS ||
899 entry->part.type == PART_FREEBSD_ZFS)
904 #endif /* LOADER_GPT_SUPPORT */
905 #ifdef LOADER_PC98_SUPPORT
906 if (table->type == PTABLE_PC98) {
907 switch(entry->part.type & PC98_MID_MASK) {
908 case PC98_MID_386BSD: /* FreeBSD */
909 if ((entry->part.type & PC98_MID_BOOTABLE) &&
910 (preflevel > PREF_FBSD_ACT)) {
912 preflevel = PREF_FBSD_ACT;
913 } else if (preflevel > PREF_FBSD) {
915 preflevel = PREF_FBSD;
919 case 0x11: /* DOS/Windows */
925 if ((entry->part.type & PC98_MID_BOOTABLE) &&
926 (preflevel > PREF_DOS_ACT)) {
928 preflevel = PREF_DOS_ACT;
929 } else if (preflevel > PREF_DOS) {
931 preflevel = PREF_DOS;
936 #endif /* LOADER_PC98_SUPPORT */
937 if (pref < preflevel) {
943 memcpy(part, &best->part, sizeof(*part));
950 ptable_iterate(const struct ptable *table, void *arg, ptable_iterate_t *iter)
952 struct pentry *entry;
957 STAILQ_FOREACH(entry, &table->entries, entry) {
958 #ifdef LOADER_MBR_SUPPORT
959 if (table->type == PTABLE_MBR)
960 sprintf(name, "s%d", entry->part.index);
963 #ifdef LOADER_GPT_SUPPORT
964 if (table->type == PTABLE_GPT)
965 sprintf(name, "p%d", entry->part.index);
968 #ifdef LOADER_VTOC8_SUPPORT
969 if (table->type == PTABLE_VTOC8)
970 sprintf(name, "%c", (uint8_t) 'a' +
974 if (table->type == PTABLE_BSD)
975 sprintf(name, "%c", (uint8_t) 'a' +
977 if ((ret = iter(arg, name, &entry->part)) != 0)
982 #ifdef LOADER_PC98_SUPPORT
984 bd_open_pc98(struct open_disk *od, struct i386_devdesc *dev)
986 struct pc98_partition *dptr;
987 struct disklabel *lp;
988 int sector, slice, i;
992 * Following calculations attempt to determine the correct value
993 * for d->od_boff by looking for the slice and partition specified,
994 * or searching for reasonable defaults.
998 * Find the slice in the DOS slice table.
1001 if (od->od_flags & BD_FLOPPY) {
1005 if (bd_read(od, 0, 1, buf)) {
1006 DEBUG("error reading MBR");
1011 * Check the slice table magic.
1013 if (((u_char)buf[0x1fe] != 0x55) || ((u_char)buf[0x1ff] != 0xaa)) {
1014 /* If a slice number was explicitly supplied, this is an error */
1015 if (dev->d_kind.biosdisk.slice > 0) {
1016 DEBUG("no slice table/MBR (no magic)");
1020 goto unsliced; /* may be a floppy */
1022 if (bd_read(od, 1, 1, buf)) {
1023 DEBUG("error reading MBR");
1028 * copy the partition table, then pick up any extended partitions.
1030 bcopy(buf + PC98_PARTOFF, &od->od_slicetab,
1031 sizeof(struct pc98_partition) * PC98_NPARTS);
1032 od->od_nslices = PC98_NPARTS; /* extended slices start here */
1033 od->od_flags |= BD_PARTTABOK;
1034 dptr = &od->od_slicetab[0];
1036 /* Is this a request for the whole disk? */
1037 if (dev->d_kind.biosdisk.slice == -1) {
1043 * if a slice number was supplied but not found, this is an error.
1045 if (dev->d_kind.biosdisk.slice > 0) {
1046 slice = dev->d_kind.biosdisk.slice - 1;
1047 if (slice >= od->od_nslices) {
1048 DEBUG("slice %d not found", slice);
1053 /* Try to auto-detect the best slice; this should always give a slice number */
1054 if (dev->d_kind.biosdisk.slice == 0) {
1055 slice = bd_bestslice(od);
1059 dev->d_kind.biosdisk.slice = slice;
1062 dptr = &od->od_slicetab[0];
1064 * Accept the supplied slice number unequivocally (we may be looking
1065 * at a DOS partition).
1067 dptr += (dev->d_kind.biosdisk.slice - 1); /* we number 1-4, offsets are 0-3 */
1068 sector = dptr->dp_scyl * od->od_hds * od->od_sec +
1069 dptr->dp_shd * od->od_sec + dptr->dp_ssect;
1071 int end = dptr->dp_ecyl * od->od_hds * od->od_sec +
1072 dptr->dp_ehd * od->od_sec + dptr->dp_esect;
1073 DEBUG("slice entry %d at %d, %d sectors",
1074 dev->d_kind.biosdisk.slice - 1, sector, end-sector);
1078 * If we are looking at a BSD slice, and the partition is < 0, assume the 'a' partition
1080 if ((dptr->dp_mid == DOSMID_386BSD) && (dev->d_kind.biosdisk.partition < 0))
1081 dev->d_kind.biosdisk.partition = 0;
1085 * Now we have the slice offset, look for the partition in the disklabel if we have
1086 * a partition to start with.
1088 * XXX we might want to check the label checksum.
1090 if (dev->d_kind.biosdisk.partition < 0) {
1091 od->od_boff = sector; /* no partition, must be after the slice */
1092 DEBUG("opening raw slice");
1095 if (bd_read(od, sector + LABELSECTOR, 1, buf)) {
1096 DEBUG("error reading disklabel");
1099 DEBUG("copy %d bytes of label from %p to %p", sizeof(struct disklabel), buf + LABELOFFSET, &od->od_disklabel);
1100 bcopy(buf + LABELOFFSET, &od->od_disklabel, sizeof(struct disklabel));
1101 lp = &od->od_disklabel;
1102 od->od_flags |= BD_LABELOK;
1104 if (lp->d_magic != DISKMAGIC) {
1105 DEBUG("no disklabel");
1108 if (dev->d_kind.biosdisk.partition >= lp->d_npartitions) {
1109 DEBUG("partition '%c' exceeds partitions in table (a-'%c')",
1110 'a' + dev->d_kind.biosdisk.partition, 'a' + lp->d_npartitions);
1115 /* Complain if the partition is unused unless this is a floppy. */
1116 if ((lp->d_partitions[dev->d_kind.biosdisk.partition].p_fstype == FS_UNUSED) &&
1117 !(od->od_flags & BD_FLOPPY))
1118 DEBUG("warning, partition marked as unused");
1122 lp->d_partitions[dev->d_kind.biosdisk.partition].p_offset -
1123 lp->d_partitions[RAW_PART].p_offset +
1129 bd_closedisk(struct open_disk *od)
1131 DEBUG("open_disk %p", od);
1133 /* XXX is this required? (especially if disk already open...) */
1134 if (od->od_flags & BD_FLOPPY)
1140 #endif /* LOADER_PC98_SUPPORT */