1 /*-
2 * Copyright (c) 2012 Andrey V. Elsukov <[email protected]>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
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.
13 *
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
24 * SUCH DAMAGE.
25 */
26
27 #include <sys/cdefs.h>
28 __FBSDID("$FreeBSD$");
29
30 #include <stand.h>
31 #include <sys/param.h>
32 #include <sys/diskmbr.h>
33 #include <sys/disklabel.h>
34 #include <sys/endian.h>
35 #include <sys/gpt.h>
36 #include <sys/stddef.h>
37 #include <sys/queue.h>
38 #include <sys/vtoc.h>
39
40 #include <fs/cd9660/iso.h>
41
42 #include <zlib.h>
43 #include <part.h>
44 #include <uuid.h>
45
46 #ifdef PART_DEBUG
47 #define DPRINTF(fmt, args...) printf("%s: " fmt "\n", __func__, ## args)
48 #else
49 #define DPRINTF(fmt, args...) ((void)0)
50 #endif
51
52 #ifdef LOADER_GPT_SUPPORT
53 #define MAXTBLSZ 64
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_swap = GPT_ENT_TYPE_FREEBSD_SWAP;
61 static const uuid_t gpt_uuid_freebsd_zfs = GPT_ENT_TYPE_FREEBSD_ZFS;
62 static const uuid_t gpt_uuid_freebsd_vinum = GPT_ENT_TYPE_FREEBSD_VINUM;
63 static const uuid_t gpt_uuid_apple_apfs = GPT_ENT_TYPE_APPLE_APFS;
64 #endif
65
66 struct pentry {
67 struct ptable_entry part;
68 uint64_t flags;
69 union {
70 uint8_t bsd;
71 uint8_t mbr;
72 uuid_t gpt;
73 uint16_t vtoc8;
74 } type;
75 STAILQ_ENTRY(pentry) entry;
76 };
77
78 struct ptable {
79 enum ptable_type type;
80 uint16_t sectorsize;
81 uint64_t sectors;
82
83 STAILQ_HEAD(, pentry) entries;
84 };
85
86 static struct parttypes {
87 enum partition_type type;
88 const char *desc;
89 } ptypes[] = {
90 { PART_UNKNOWN, "Unknown" },
91 { PART_EFI, "EFI" },
92 { PART_FREEBSD, "FreeBSD" },
93 { PART_FREEBSD_BOOT, "FreeBSD boot" },
94 { PART_FREEBSD_UFS, "FreeBSD UFS" },
95 { PART_FREEBSD_ZFS, "FreeBSD ZFS" },
96 { PART_FREEBSD_SWAP, "FreeBSD swap" },
97 { PART_FREEBSD_VINUM, "FreeBSD vinum" },
98 { PART_LINUX, "Linux" },
99 { PART_LINUX_SWAP, "Linux swap" },
100 { PART_DOS, "DOS/Windows" },
101 { PART_ISO9660, "ISO9660" },
102 { PART_APFS, "APFS" },
103 };
104
105 const char *
parttype2str(enum partition_type type)106 parttype2str(enum partition_type type)
107 {
108 size_t i;
109
110 for (i = 0; i < nitems(ptypes); i++)
111 if (ptypes[i].type == type)
112 return (ptypes[i].desc);
113 return (ptypes[0].desc);
114 }
115
116 #ifdef LOADER_GPT_SUPPORT
117 static void
uuid_letoh(uuid_t * uuid)118 uuid_letoh(uuid_t *uuid)
119 {
120
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);
124 }
125
126 static enum partition_type
gpt_parttype(uuid_t type)127 gpt_parttype(uuid_t type)
128 {
129
130 if (uuid_equal(&type, &gpt_uuid_efi, NULL))
131 return (PART_EFI);
132 else if (uuid_equal(&type, &gpt_uuid_ms_basic_data, NULL))
133 return (PART_DOS);
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, NULL))
145 return (PART_FREEBSD);
146 else if (uuid_equal(&type, &gpt_uuid_apple_apfs, NULL))
147 return (PART_APFS);
148 return (PART_UNKNOWN);
149 }
150
151 static struct gpt_hdr *
gpt_checkhdr(struct gpt_hdr * hdr,uint64_t lba_self,uint64_t lba_last,uint16_t sectorsize)152 gpt_checkhdr(struct gpt_hdr *hdr, uint64_t lba_self, uint64_t lba_last,
153 uint16_t sectorsize)
154 {
155 uint32_t sz, crc;
156
157 if (memcmp(hdr->hdr_sig, GPT_HDR_SIG, sizeof(hdr->hdr_sig)) != 0) {
158 DPRINTF("no GPT signature");
159 return (NULL);
160 }
161 sz = le32toh(hdr->hdr_size);
162 if (sz < 92 || sz > sectorsize) {
163 DPRINTF("invalid GPT header size: %d", sz);
164 return (NULL);
165 }
166 crc = le32toh(hdr->hdr_crc_self);
167 hdr->hdr_crc_self = crc32(0, Z_NULL, 0);
168 if (crc32(hdr->hdr_crc_self, (const Bytef *)hdr, sz) != crc) {
169 DPRINTF("GPT header's CRC doesn't match");
170 return (NULL);
171 }
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);
176 return (NULL);
177 }
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");
181 return (NULL);
182 }
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");
186 return (NULL);
187 }
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");
194 return (NULL);
195 }
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);
201 return (hdr);
202 }
203
204 static int
gpt_checktbl(const struct gpt_hdr * hdr,uint8_t * tbl,size_t size,uint64_t lba_last)205 gpt_checktbl(const struct gpt_hdr *hdr, uint8_t *tbl, size_t size,
206 uint64_t lba_last)
207 {
208 struct gpt_ent *ent;
209 uint32_t i, cnt;
210
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(0, tbl, hdr->hdr_entries * hdr->hdr_entsz)) {
217 DPRINTF("GPT table's CRC doesn't match");
218 return (-1);
219 }
220 }
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))
225 continue;
226 ent->ent_lba_start = le64toh(ent->ent_lba_start);
227 ent->ent_lba_end = le64toh(ent->ent_lba_end);
228 }
229 return (0);
230 }
231
232 static struct ptable *
ptable_gptread(struct ptable * table,void * dev,diskread_t dread)233 ptable_gptread(struct ptable *table, void *dev, diskread_t dread)
234 {
235 struct pentry *entry;
236 struct gpt_hdr *phdr, hdr;
237 struct gpt_ent *ent;
238 uint8_t *buf, *tbl;
239 uint64_t offset;
240 int pri, sec;
241 size_t size, i;
242
243 buf = malloc(table->sectorsize);
244 if (buf == NULL)
245 return (NULL);
246 tbl = malloc(table->sectorsize * MAXTBLSZ);
247 if (tbl == NULL) {
248 free(buf);
249 return (NULL);
250 }
251 /* Read the primary GPT header. */
252 if (dread(dev, buf, 1, 1) != 0) {
253 ptable_close(table);
254 table = NULL;
255 goto out;
256 }
257 pri = sec = 0;
258 /* Check the primary GPT header. */
259 phdr = gpt_checkhdr((struct gpt_hdr *)buf, 1, table->sectors - 1,
260 table->sectorsize);
261 if (phdr != NULL) {
262 /* Read the primary GPT table. */
263 size = MIN(MAXTBLSZ,
264 howmany(phdr->hdr_entries * phdr->hdr_entsz,
265 table->sectorsize));
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));
270 pri = 1;
271 }
272 }
273 offset = pri ? hdr.hdr_lba_alt: table->sectors - 1;
274 /* Read the backup GPT header. */
275 if (dread(dev, buf, 1, offset) != 0)
276 phdr = NULL;
277 else
278 phdr = gpt_checkhdr((struct gpt_hdr *)buf, offset,
279 table->sectors - 1, table->sectorsize);
280 if (phdr != NULL) {
281 /*
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.
286 */
287 if (pri == 0 ||
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. */
297 size = MIN(MAXTBLSZ,
298 howmany(phdr->hdr_entries * phdr->hdr_entsz,
299 table->sectorsize));
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));
304 sec = 1;
305 }
306 }
307 }
308 if (pri == 0 && sec == 0) {
309 /* Both primary and backup tables are invalid. */
310 table->type = PTABLE_NONE;
311 goto out;
312 }
313 DPRINTF("GPT detected");
314 size = MIN(hdr.hdr_entries * hdr.hdr_entsz,
315 MAXTBLSZ * table->sectorsize);
316
317 /*
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.
322 *
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.
325 */
326 table->sectors = hdr.hdr_lba_alt + 1;
327
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))
331 continue;
332
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)
337 continue;
338
339 entry = malloc(sizeof(*entry));
340 if (entry == NULL)
341 break;
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");
350 }
351 out:
352 free(buf);
353 free(tbl);
354 return (table);
355 }
356 #endif /* LOADER_GPT_SUPPORT */
357
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
mbr_parttype(uint8_t type)362 mbr_parttype(uint8_t type)
363 {
364
365 switch (type) {
366 case DOSPTYP_386BSD:
367 return (PART_FREEBSD);
368 case DOSPTYP_LINSWP:
369 return (PART_LINUX_SWAP);
370 case DOSPTYP_LINUX:
371 return (PART_LINUX);
372 case 0x01:
373 case 0x04:
374 case 0x06:
375 case 0x07:
376 case 0x0b:
377 case 0x0c:
378 case 0x0e:
379 return (PART_DOS);
380 }
381 return (PART_UNKNOWN);
382 }
383
384 static struct ptable *
ptable_ebrread(struct ptable * table,void * dev,diskread_t dread)385 ptable_ebrread(struct ptable *table, void *dev, diskread_t dread)
386 {
387 struct dos_partition *dp;
388 struct pentry *e1, *entry;
389 uint32_t start, end, offset;
390 u_char *buf;
391 int i, index;
392
393 STAILQ_FOREACH(e1, &table->entries, entry) {
394 if (e1->type.mbr == DOSPTYP_EXT ||
395 e1->type.mbr == DOSPTYP_EXTLBA)
396 break;
397 }
398 if (e1 == NULL)
399 return (table);
400 index = 5;
401 offset = e1->part.start;
402 buf = malloc(table->sectorsize);
403 if (buf == NULL)
404 return (table);
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)
409 break;
410 #endif
411 if (dread(dev, buf, 1, offset) != 0)
412 break;
413 dp = (struct dos_partition *)(buf + DOSPARTOFF);
414 if (dp[0].dp_typ == 0)
415 break;
416 start = le32toh(dp[0].dp_start);
417 if (dp[0].dp_typ == DOSPTYP_EXT &&
418 dp[1].dp_typ == 0) {
419 offset = e1->part.start + start;
420 continue;
421 }
422 end = le32toh(dp[0].dp_size);
423 entry = malloc(sizeof(*entry));
424 if (entry == NULL)
425 break;
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)
435 break;
436 offset = e1->part.start + le32toh(dp[1].dp_start);
437 }
438 free(buf);
439 return (table);
440 }
441 #endif /* LOADER_MBR_SUPPORT */
442
443 static enum partition_type
bsd_parttype(uint8_t type)444 bsd_parttype(uint8_t type)
445 {
446
447 switch (type) {
448 case FS_SWAP:
449 return (PART_FREEBSD_SWAP);
450 case FS_BSDFFS:
451 return (PART_FREEBSD_UFS);
452 case FS_VINUM:
453 return (PART_FREEBSD_VINUM);
454 case FS_ZFS:
455 return (PART_FREEBSD_ZFS);
456 }
457 return (PART_UNKNOWN);
458 }
459
460 static struct ptable *
ptable_bsdread(struct ptable * table,void * dev,diskread_t dread)461 ptable_bsdread(struct ptable *table, void *dev, diskread_t dread)
462 {
463 struct disklabel *dl;
464 struct partition *part;
465 struct pentry *entry;
466 uint8_t *buf;
467 uint32_t raw_offset;
468 int i;
469
470 if (table->sectorsize < sizeof(struct disklabel)) {
471 DPRINTF("Too small sectorsize");
472 return (table);
473 }
474 buf = malloc(table->sectorsize);
475 if (buf == NULL)
476 return (table);
477 if (dread(dev, buf, 1, 1) != 0) {
478 DPRINTF("read failed");
479 ptable_close(table);
480 table = NULL;
481 goto out;
482 }
483 dl = (struct disklabel *)buf;
484 if (le32toh(dl->d_magic) != DISKMAGIC &&
485 le32toh(dl->d_magic2) != DISKMAGIC)
486 goto out;
487 if (le32toh(dl->d_secsize) != table->sectorsize) {
488 DPRINTF("unsupported sector size");
489 goto out;
490 }
491 dl->d_npartitions = le16toh(dl->d_npartitions);
492 if (dl->d_npartitions > 20 || dl->d_npartitions < 8) {
493 DPRINTF("invalid number of partitions");
494 goto out;
495 }
496 DPRINTF("BSD detected");
497 part = &dl->d_partitions[0];
498 raw_offset = le32toh(part[RAW_PART].p_offset);
499 for (i = 0; i < dl->d_npartitions; i++, part++) {
500 if (i == RAW_PART)
501 continue;
502 if (part->p_size == 0)
503 continue;
504 entry = malloc(sizeof(*entry));
505 if (entry == NULL)
506 break;
507 entry->part.start = le32toh(part->p_offset) - raw_offset;
508 entry->part.end = entry->part.start +
509 le32toh(part->p_size) - 1;
510 entry->part.type = bsd_parttype(part->p_fstype);
511 entry->part.index = i; /* starts from zero */
512 entry->type.bsd = part->p_fstype;
513 STAILQ_INSERT_TAIL(&table->entries, entry, entry);
514 DPRINTF("new BSD partition added");
515 }
516 table->type = PTABLE_BSD;
517 out:
518 free(buf);
519 return (table);
520 }
521
522 #ifdef LOADER_VTOC8_SUPPORT
523 static enum partition_type
vtoc8_parttype(uint16_t type)524 vtoc8_parttype(uint16_t type)
525 {
526
527 switch (type) {
528 case VTOC_TAG_FREEBSD_SWAP:
529 return (PART_FREEBSD_SWAP);
530 case VTOC_TAG_FREEBSD_UFS:
531 return (PART_FREEBSD_UFS);
532 case VTOC_TAG_FREEBSD_VINUM:
533 return (PART_FREEBSD_VINUM);
534 case VTOC_TAG_FREEBSD_ZFS:
535 return (PART_FREEBSD_ZFS);
536 }
537 return (PART_UNKNOWN);
538 }
539
540 static struct ptable *
ptable_vtoc8read(struct ptable * table,void * dev,diskread_t dread)541 ptable_vtoc8read(struct ptable *table, void *dev, diskread_t dread)
542 {
543 struct pentry *entry;
544 struct vtoc8 *dl;
545 uint8_t *buf;
546 uint16_t sum, heads, sectors;
547 int i;
548
549 if (table->sectorsize != sizeof(struct vtoc8))
550 return (table);
551 buf = malloc(table->sectorsize);
552 if (buf == NULL)
553 return (table);
554 if (dread(dev, buf, 1, 0) != 0) {
555 DPRINTF("read failed");
556 ptable_close(table);
557 table = NULL;
558 goto out;
559 }
560 dl = (struct vtoc8 *)buf;
561 /* Check the sum */
562 for (i = sum = 0; i < sizeof(struct vtoc8); i += sizeof(sum))
563 sum ^= be16dec(buf + i);
564 if (sum != 0) {
565 DPRINTF("incorrect checksum");
566 goto out;
567 }
568 if (be16toh(dl->nparts) != VTOC8_NPARTS) {
569 DPRINTF("invalid number of entries");
570 goto out;
571 }
572 sectors = be16toh(dl->nsecs);
573 heads = be16toh(dl->nheads);
574 if (sectors * heads == 0) {
575 DPRINTF("invalid geometry");
576 goto out;
577 }
578 DPRINTF("VTOC8 detected");
579 for (i = 0; i < VTOC8_NPARTS; i++) {
580 dl->part[i].tag = be16toh(dl->part[i].tag);
581 if (i == VTOC_RAW_PART ||
582 dl->part[i].tag == VTOC_TAG_UNASSIGNED)
583 continue;
584 entry = malloc(sizeof(*entry));
585 if (entry == NULL)
586 break;
587 entry->part.start = be32toh(dl->map[i].cyl) * heads * sectors;
588 entry->part.end = be32toh(dl->map[i].nblks) +
589 entry->part.start - 1;
590 entry->part.type = vtoc8_parttype(dl->part[i].tag);
591 entry->part.index = i; /* starts from zero */
592 entry->type.vtoc8 = dl->part[i].tag;
593 STAILQ_INSERT_TAIL(&table->entries, entry, entry);
594 DPRINTF("new VTOC8 partition added");
595 }
596 table->type = PTABLE_VTOC8;
597 out:
598 free(buf);
599 return (table);
600
601 }
602 #endif /* LOADER_VTOC8_SUPPORT */
603
604 #define cdb2devb(bno) ((bno) * ISO_DEFAULT_BLOCK_SIZE / table->sectorsize)
605
606 static struct ptable *
ptable_iso9660read(struct ptable * table,void * dev,diskread_t dread)607 ptable_iso9660read(struct ptable *table, void *dev, diskread_t dread)
608 {
609 uint8_t *buf;
610 struct iso_primary_descriptor *vd;
611 struct pentry *entry;
612
613 buf = malloc(table->sectorsize);
614 if (buf == NULL)
615 return (table);
616
617 if (dread(dev, buf, 1, cdb2devb(16)) != 0) {
618 DPRINTF("read failed");
619 ptable_close(table);
620 table = NULL;
621 goto out;
622 }
623 vd = (struct iso_primary_descriptor *)buf;
624 if (bcmp(vd->id, ISO_STANDARD_ID, sizeof vd->id) != 0)
625 goto out;
626
627 entry = malloc(sizeof(*entry));
628 if (entry == NULL)
629 goto out;
630 entry->part.start = 0;
631 entry->part.end = table->sectors;
632 entry->part.type = PART_ISO9660;
633 entry->part.index = 0;
634 STAILQ_INSERT_TAIL(&table->entries, entry, entry);
635
636 table->type = PTABLE_ISO9660;
637
638 out:
639 free(buf);
640 return (table);
641 }
642
643 struct ptable *
ptable_open(void * dev,uint64_t sectors,uint16_t sectorsize,diskread_t * dread)644 ptable_open(void *dev, uint64_t sectors, uint16_t sectorsize,
645 diskread_t *dread)
646 {
647 struct dos_partition *dp;
648 struct ptable *table;
649 uint8_t *buf;
650 #ifdef LOADER_MBR_SUPPORT
651 struct pentry *entry;
652 uint32_t start, end;
653 int has_ext;
654 #endif
655 table = NULL;
656 dp = NULL;
657 buf = malloc(sectorsize);
658 if (buf == NULL)
659 return (NULL);
660 /* First, read the MBR. */
661 if (dread(dev, buf, 1, DOSBBSECTOR) != 0) {
662 DPRINTF("read failed");
663 goto out;
664 }
665
666 table = malloc(sizeof(*table));
667 if (table == NULL)
668 goto out;
669 table->sectors = sectors;
670 table->sectorsize = sectorsize;
671 table->type = PTABLE_NONE;
672 STAILQ_INIT(&table->entries);
673
674 if (ptable_iso9660read(table, dev, dread) == NULL) {
675 /* Read error. */
676 table = NULL;
677 goto out;
678 } else if (table->type == PTABLE_ISO9660)
679 goto out;
680
681 #ifdef LOADER_VTOC8_SUPPORT
682 if (be16dec(buf + offsetof(struct vtoc8, magic)) == VTOC_MAGIC) {
683 if (ptable_vtoc8read(table, dev, dread) == NULL) {
684 /* Read error. */
685 table = NULL;
686 goto out;
687 } else if (table->type == PTABLE_VTOC8)
688 goto out;
689 }
690 #endif
691 /* Check the BSD label. */
692 if (ptable_bsdread(table, dev, dread) == NULL) { /* Read error. */
693 table = NULL;
694 goto out;
695 } else if (table->type == PTABLE_BSD)
696 goto out;
697
698 #if defined(LOADER_GPT_SUPPORT) || defined(LOADER_MBR_SUPPORT)
699 /* Check the MBR magic. */
700 if (buf[DOSMAGICOFFSET] != 0x55 ||
701 buf[DOSMAGICOFFSET + 1] != 0xaa) {
702 DPRINTF("magic sequence not found");
703 #if defined(LOADER_GPT_SUPPORT)
704 /* There is no PMBR, check that we have backup GPT */
705 table->type = PTABLE_GPT;
706 table = ptable_gptread(table, dev, dread);
707 #endif
708 goto out;
709 }
710 /* Check that we have PMBR. Also do some validation. */
711 dp = malloc(NDOSPART * sizeof(struct dos_partition));
712 if (dp == NULL)
713 goto out;
714 bcopy(buf + DOSPARTOFF, dp, NDOSPART * sizeof(struct dos_partition));
715
716 /*
717 * In mac we can have PMBR partition in hybrid MBR;
718 * that is, MBR partition which has DOSPTYP_PMBR entry defined as
719 * start sector 1. After DOSPTYP_PMBR, there may be other partitions.
720 * UEFI compliant PMBR has no other partitions.
721 */
722 for (int i = 0; i < NDOSPART; i++) {
723 if (dp[i].dp_flag != 0 && dp[i].dp_flag != 0x80) {
724 DPRINTF("invalid partition flag %x", dp[i].dp_flag);
725 goto out;
726 }
727 #ifdef LOADER_GPT_SUPPORT
728 if (dp[i].dp_typ == DOSPTYP_PMBR && dp[i].dp_start == 1) {
729 table->type = PTABLE_GPT;
730 DPRINTF("PMBR detected");
731 }
732 #endif
733 }
734 #ifdef LOADER_GPT_SUPPORT
735 if (table->type == PTABLE_GPT) {
736 table = ptable_gptread(table, dev, dread);
737 goto out;
738 }
739 #endif
740 #ifdef LOADER_MBR_SUPPORT
741 /* Read MBR. */
742 DPRINTF("MBR detected");
743 table->type = PTABLE_MBR;
744 for (int i = has_ext = 0; i < NDOSPART; i++) {
745 if (dp[i].dp_typ == 0)
746 continue;
747 start = le32dec(&(dp[i].dp_start));
748 end = le32dec(&(dp[i].dp_size));
749 if (start == 0 || end == 0)
750 continue;
751 #if 0 /* Some BIOSes return an incorrect number of sectors */
752 if (start + end - 1 >= sectors)
753 continue; /* XXX: ignore */
754 #endif
755 if (dp[i].dp_typ == DOSPTYP_EXT ||
756 dp[i].dp_typ == DOSPTYP_EXTLBA)
757 has_ext = 1;
758 entry = malloc(sizeof(*entry));
759 if (entry == NULL)
760 break;
761 entry->part.start = start;
762 entry->part.end = start + end - 1;
763 entry->part.index = i + 1;
764 entry->part.type = mbr_parttype(dp[i].dp_typ);
765 entry->flags = dp[i].dp_flag;
766 entry->type.mbr = dp[i].dp_typ;
767 STAILQ_INSERT_TAIL(&table->entries, entry, entry);
768 DPRINTF("new MBR partition added");
769 }
770 if (has_ext) {
771 table = ptable_ebrread(table, dev, dread);
772 /* FALLTHROUGH */
773 }
774 #endif /* LOADER_MBR_SUPPORT */
775 #endif /* LOADER_MBR_SUPPORT || LOADER_GPT_SUPPORT */
776 out:
777 free(dp);
778 free(buf);
779 return (table);
780 }
781
782 void
ptable_close(struct ptable * table)783 ptable_close(struct ptable *table)
784 {
785 struct pentry *entry;
786
787 if (table == NULL)
788 return;
789
790 while (!STAILQ_EMPTY(&table->entries)) {
791 entry = STAILQ_FIRST(&table->entries);
792 STAILQ_REMOVE_HEAD(&table->entries, entry);
793 free(entry);
794 }
795 free(table);
796 }
797
798 enum ptable_type
ptable_gettype(const struct ptable * table)799 ptable_gettype(const struct ptable *table)
800 {
801
802 return (table->type);
803 }
804
805 int
ptable_getsize(const struct ptable * table,uint64_t * sizep)806 ptable_getsize(const struct ptable *table, uint64_t *sizep)
807 {
808 uint64_t tmp = table->sectors * table->sectorsize;
809
810 if (tmp < table->sectors)
811 return (EOVERFLOW);
812
813 if (sizep != NULL)
814 *sizep = tmp;
815 return (0);
816 }
817
818 int
ptable_getpart(const struct ptable * table,struct ptable_entry * part,int index)819 ptable_getpart(const struct ptable *table, struct ptable_entry *part, int index)
820 {
821 struct pentry *entry;
822
823 if (part == NULL || table == NULL)
824 return (EINVAL);
825
826 STAILQ_FOREACH(entry, &table->entries, entry) {
827 if (entry->part.index != index)
828 continue;
829 memcpy(part, &entry->part, sizeof(*part));
830 return (0);
831 }
832 return (ENOENT);
833 }
834
835 /*
836 * Search for a slice with the following preferences:
837 *
838 * 1: Active FreeBSD slice
839 * 2: Non-active FreeBSD slice
840 * 3: Active Linux slice
841 * 4: non-active Linux slice
842 * 5: Active FAT/FAT32 slice
843 * 6: non-active FAT/FAT32 slice
844 */
845 #define PREF_RAWDISK 0
846 #define PREF_FBSD_ACT 1
847 #define PREF_FBSD 2
848 #define PREF_LINUX_ACT 3
849 #define PREF_LINUX 4
850 #define PREF_DOS_ACT 5
851 #define PREF_DOS 6
852 #define PREF_NONE 7
853 int
ptable_getbestpart(const struct ptable * table,struct ptable_entry * part)854 ptable_getbestpart(const struct ptable *table, struct ptable_entry *part)
855 {
856 struct pentry *entry, *best;
857 int pref, preflevel;
858
859 if (part == NULL || table == NULL)
860 return (EINVAL);
861
862 best = NULL;
863 preflevel = pref = PREF_NONE;
864 STAILQ_FOREACH(entry, &table->entries, entry) {
865 #ifdef LOADER_MBR_SUPPORT
866 if (table->type == PTABLE_MBR) {
867 switch (entry->type.mbr) {
868 case DOSPTYP_386BSD:
869 pref = entry->flags & 0x80 ? PREF_FBSD_ACT:
870 PREF_FBSD;
871 break;
872 case DOSPTYP_LINUX:
873 pref = entry->flags & 0x80 ? PREF_LINUX_ACT:
874 PREF_LINUX;
875 break;
876 case 0x01: /* DOS/Windows */
877 case 0x04:
878 case 0x06:
879 case 0x0c:
880 case 0x0e:
881 case DOSPTYP_FAT32:
882 pref = entry->flags & 0x80 ? PREF_DOS_ACT:
883 PREF_DOS;
884 break;
885 default:
886 pref = PREF_NONE;
887 }
888 }
889 #endif /* LOADER_MBR_SUPPORT */
890 #ifdef LOADER_GPT_SUPPORT
891 if (table->type == PTABLE_GPT) {
892 if (entry->part.type == PART_DOS)
893 pref = PREF_DOS;
894 else if (entry->part.type == PART_FREEBSD_UFS ||
895 entry->part.type == PART_FREEBSD_ZFS)
896 pref = PREF_FBSD;
897 else
898 pref = PREF_NONE;
899 }
900 #endif /* LOADER_GPT_SUPPORT */
901 if (pref < preflevel) {
902 preflevel = pref;
903 best = entry;
904 }
905 }
906 if (best != NULL) {
907 memcpy(part, &best->part, sizeof(*part));
908 return (0);
909 }
910 return (ENOENT);
911 }
912
913 int
ptable_iterate(const struct ptable * table,void * arg,ptable_iterate_t * iter)914 ptable_iterate(const struct ptable *table, void *arg, ptable_iterate_t *iter)
915 {
916 struct pentry *entry;
917 char name[32];
918 int ret = 0;
919
920 name[0] = '\0';
921 STAILQ_FOREACH(entry, &table->entries, entry) {
922 #ifdef LOADER_MBR_SUPPORT
923 if (table->type == PTABLE_MBR)
924 sprintf(name, "s%d", entry->part.index);
925 else
926 #endif
927 #ifdef LOADER_GPT_SUPPORT
928 if (table->type == PTABLE_GPT)
929 sprintf(name, "p%d", entry->part.index);
930 else
931 #endif
932 #ifdef LOADER_VTOC8_SUPPORT
933 if (table->type == PTABLE_VTOC8)
934 sprintf(name, "%c", (uint8_t) 'a' +
935 entry->part.index);
936 else
937 #endif
938 if (table->type == PTABLE_BSD)
939 sprintf(name, "%c", (uint8_t) 'a' +
940 entry->part.index);
941 if ((ret = iter(arg, name, &entry->part)) != 0)
942 return (ret);
943 }
944 return (ret);
945 }
946