sys/boot/common/part.c

   1 /*-
   2  * Copyright (c) 2012 Andrey V. Elsukov <ae@FreeBSD.org>
   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 <crc32.h>
  41 #include <part.h>
  42 #include <uuid.h>
  43
  44 #ifdef PART_DEBUG
  45 #define DEBUG(fmt, args...) printf("%s: " fmt "\n" , __func__ , ## args)
  46 #else
  47 #define DEBUG(fmt, args...)
  48 #endif
  49
  50 #ifdef LOADER_GPT_SUPPORT
  51 #define MAXTBLSZ        64
  52 static const uuid_t gpt_uuid_unused = GPT_ENT_TYPE_UNUSED;
  53 static const uuid_t gpt_uuid_ms_basic_data = GPT_ENT_TYPE_MS_BASIC_DATA;
  54 static const uuid_t gpt_uuid_freebsd_ufs = GPT_ENT_TYPE_FREEBSD_UFS;
  55 static const uuid_t gpt_uuid_efi = GPT_ENT_TYPE_EFI;
  56 static const uuid_t gpt_uuid_freebsd_boot = GPT_ENT_TYPE_FREEBSD_BOOT;
  57 static const uuid_t gpt_uuid_freebsd_nandfs = GPT_ENT_TYPE_FREEBSD_NANDFS;
  58 static const uuid_t gpt_uuid_freebsd_swap = GPT_ENT_TYPE_FREEBSD_SWAP;
  59 static const uuid_t gpt_uuid_freebsd_zfs = GPT_ENT_TYPE_FREEBSD_ZFS;
  60 static const uuid_t gpt_uuid_freebsd_vinum = GPT_ENT_TYPE_FREEBSD_VINUM;
  61 #endif
  62
  63 struct pentry {
  64         struct ptable_entry     part;
  65         uint64_t                flags;
  66         union {
  67                 uint8_t bsd;
  68                 uint8_t mbr;
  69                 uuid_t  gpt;
  70                 uint16_t vtoc8;
  71         } type;
  72         STAILQ_ENTRY(pentry)    entry;
  73 };
  74
  75 struct ptable {
  76         enum ptable_type        type;
  77         uint16_t                sectorsize;
  78         uint64_t                sectors;
  79
  80         STAILQ_HEAD(, pentry)   entries;
  81 };
  82
  83 static struct parttypes {
  84         enum partition_type     type;
  85         const char              *desc;
  86 } ptypes[] = {
  87         { PART_UNKNOWN,         "Unknown" },
  88         { PART_EFI,             "EFI" },
  89         { PART_FREEBSD,         "FreeBSD" },
  90         { PART_FREEBSD_BOOT,    "FreeBSD boot" },
  91         { PART_FREEBSD_NANDFS,  "FreeBSD nandfs" },
  92         { PART_FREEBSD_UFS,     "FreeBSD UFS" },
  93         { PART_FREEBSD_ZFS,     "FreeBSD ZFS" },
  94         { PART_FREEBSD_SWAP,    "FreeBSD swap" },
  95         { PART_FREEBSD_VINUM,   "FreeBSD vinum" },
  96         { PART_LINUX,           "Linux" },
  97         { PART_LINUX_SWAP,      "Linux swap" },
  98         { PART_DOS,             "DOS/Windows" },
  99 };
 100
 101 const char *
 102 parttype2str(enum partition_type type)
 103 {
 104         int i;
 105
 106         for (i = 0; i < sizeof(ptypes) / sizeof(ptypes[0]); i++)
 107                 if (ptypes[i].type == type)
 108                         return (ptypes[i].desc);
 109         return (ptypes[0].desc);
 110 }
 111
 112 #ifdef LOADER_GPT_SUPPORT
 113 static void
 114 uuid_letoh(uuid_t *uuid)
 115 {
 116
 117         uuid->time_low = le32toh(uuid->time_low);
 118         uuid->time_mid = le16toh(uuid->time_mid);
 119         uuid->time_hi_and_version = le16toh(uuid->time_hi_and_version);
 120 }
 121
 122 static enum partition_type
 123 gpt_parttype(uuid_t type)
 124 {
 125
 126         if (uuid_equal(&type, &gpt_uuid_efi, NULL))
 127                 return (PART_EFI);
 128         else if (uuid_equal(&type, &gpt_uuid_ms_basic_data, NULL))
 129                 return (PART_DOS);
 130         else if (uuid_equal(&type, &gpt_uuid_freebsd_boot, NULL))
 131                 return (PART_FREEBSD_BOOT);
 132         else if (uuid_equal(&type, &gpt_uuid_freebsd_ufs, NULL))
 133                 return (PART_FREEBSD_UFS);
 134         else if (uuid_equal(&type, &gpt_uuid_freebsd_zfs, NULL))
 135                 return (PART_FREEBSD_ZFS);
 136         else if (uuid_equal(&type, &gpt_uuid_freebsd_swap, NULL))
 137                 return (PART_FREEBSD_SWAP);
 138         else if (uuid_equal(&type, &gpt_uuid_freebsd_vinum, NULL))
 139                 return (PART_FREEBSD_VINUM);
 140         else if (uuid_equal(&type, &gpt_uuid_freebsd_nandfs, NULL))
 141                 return (PART_FREEBSD_NANDFS);
 142         return (PART_UNKNOWN);
 143 }
 144
 145 static struct gpt_hdr*
 146 gpt_checkhdr(struct gpt_hdr *hdr, uint64_t lba_self, uint64_t lba_last,
 147     uint16_t sectorsize)
 148 {
 149         uint32_t sz, crc;
 150
 151         if (memcmp(hdr->hdr_sig, GPT_HDR_SIG, sizeof(hdr->hdr_sig)) != 0) {
 152                 DEBUG("no GPT signature");
 153                 return (NULL);
 154         }
 155         sz = le32toh(hdr->hdr_size);
 156         if (sz < 92 || sz > sectorsize) {
 157                 DEBUG("invalid GPT header size: %d", sz);
 158                 return (NULL);
 159         }
 160         crc = le32toh(hdr->hdr_crc_self);
 161         hdr->hdr_crc_self = 0;
 162         if (crc32(hdr, sz) != crc) {
 163                 DEBUG("GPT header's CRC doesn't match");
 164                 return (NULL);
 165         }
 166         hdr->hdr_crc_self = crc;
 167         hdr->hdr_revision = le32toh(hdr->hdr_revision);
 168         if (hdr->hdr_revision < GPT_HDR_REVISION) {
 169                 DEBUG("unsupported GPT revision %d", hdr->hdr_revision);
 170                 return (NULL);
 171         }
 172         hdr->hdr_lba_self = le64toh(hdr->hdr_lba_self);
 173         if (hdr->hdr_lba_self != lba_self) {
 174                 DEBUG("self LBA doesn't match");
 175                 return (NULL);
 176         }
 177         hdr->hdr_lba_alt = le64toh(hdr->hdr_lba_alt);
 178         if (hdr->hdr_lba_alt == hdr->hdr_lba_self) {
 179                 DEBUG("invalid alternate LBA");
 180                 return (NULL);
 181         }
 182         hdr->hdr_entries = le32toh(hdr->hdr_entries);
 183         hdr->hdr_entsz = le32toh(hdr->hdr_entsz);
 184         if (hdr->hdr_entries == 0 ||
 185             hdr->hdr_entsz < sizeof(struct gpt_ent) ||
 186             sectorsize % hdr->hdr_entsz != 0) {
 187                 DEBUG("invalid entry size or number of entries");
 188                 return (NULL);
 189         }
 190         hdr->hdr_lba_start = le64toh(hdr->hdr_lba_start);
 191         hdr->hdr_lba_end = le64toh(hdr->hdr_lba_end);
 192         hdr->hdr_lba_table = le64toh(hdr->hdr_lba_table);
 193         hdr->hdr_crc_table = le32toh(hdr->hdr_crc_table);
 194         uuid_letoh(&hdr->hdr_uuid);
 195         return (hdr);
 196 }
 197
 198 static int
 199 gpt_checktbl(const struct gpt_hdr *hdr, u_char *tbl, size_t size,
 200     uint64_t lba_last)
 201 {
 202         struct gpt_ent *ent;
 203         int i, cnt;
 204
 205         cnt = size / hdr->hdr_entsz;
 206         if (hdr->hdr_entries <= cnt) {
 207                 cnt = hdr->hdr_entries;
 208                 /* Check CRC only when buffer size is enough for table. */
 209                 if (hdr->hdr_crc_table !=
 210                     crc32(tbl, hdr->hdr_entries * hdr->hdr_entsz)) {
 211                         DEBUG("GPT table's CRC doesn't match");
 212                         return (-1);
 213                 }
 214         }
 215         for (i = 0; i < cnt; i++) {
 216                 ent = (struct gpt_ent *)(tbl + i * hdr->hdr_entsz);
 217                 uuid_letoh(&ent->ent_type);
 218                 if (uuid_equal(&ent->ent_type, &gpt_uuid_unused, NULL))
 219                         continue;
 220                 ent->ent_lba_start = le64toh(ent->ent_lba_start);
 221                 ent->ent_lba_end = le64toh(ent->ent_lba_end);
 222         }
 223         return (0);
 224 }
 225
 226 static struct ptable*
 227 ptable_gptread(struct ptable *table, void *dev, diskread_t dread)
 228 {
 229         struct pentry *entry;
 230         struct gpt_hdr *phdr, hdr;
 231         struct gpt_ent *ent;
 232         u_char *buf, *tbl;
 233         uint64_t offset;
 234         int pri, sec, i;
 235         size_t size;
 236
 237         buf = malloc(table->sectorsize);
 238         if (buf == NULL)
 239                 return (NULL);
 240         tbl = malloc(table->sectorsize * MAXTBLSZ);
 241         if (tbl == NULL) {
 242                 free(buf);
 243                 return (NULL);
 244         }
 245         /* Read the primary GPT header. */
 246         if (dread(dev, buf, 1, 1) != 0) {
 247                 ptable_close(table);
 248                 table = NULL;
 249                 goto out;
 250         }
 251         pri = sec = 0;
 252         /* Check the primary GPT header. */
 253         phdr = gpt_checkhdr((struct gpt_hdr *)buf, 1, table->sectors - 1,
 254             table->sectorsize);
 255         if (phdr != NULL) {
 256                 /* Read the primary GPT table. */
 257                 size = MIN(MAXTBLSZ, (phdr->hdr_entries * phdr->hdr_entsz +
 258                     table->sectorsize - 1) / table->sectorsize);
 259                 if (dread(dev, tbl, size, phdr->hdr_lba_table) == 0 &&
 260                     gpt_checktbl(phdr, tbl, size * table->sectorsize,
 261                     table->sectors - 1) == 0) {
 262                         memcpy(&hdr, phdr, sizeof(hdr));
 263                         pri = 1;
 264                 }
 265         }
 266         offset = pri ? hdr.hdr_lba_alt: table->sectors - 1;
 267         /* Read the backup GPT header. */
 268         if (dread(dev, buf, 1, offset) != 0)
 269                 phdr = NULL;
 270         else
 271                 phdr = gpt_checkhdr((struct gpt_hdr *)buf, offset,
 272                     table->sectors - 1, table->sectorsize);
 273         if (phdr != NULL) {
 274                 /*
 275                  * Compare primary and backup headers.
 276                  * If they are equal, then we do not need to read backup
 277                  * table. If they are different, then prefer backup header
 278                  * and try to read backup table.
 279                  */
 280                 if (pri == 0 ||
 281                     uuid_equal(&hdr.hdr_uuid, &phdr->hdr_uuid, NULL) == 0 ||
 282                     hdr.hdr_revision != phdr->hdr_revision ||
 283                     hdr.hdr_size != phdr->hdr_size ||
 284                     hdr.hdr_lba_start != phdr->hdr_lba_start ||
 285                     hdr.hdr_lba_end != phdr->hdr_lba_end ||
 286                     hdr.hdr_entries != phdr->hdr_entries ||
 287                     hdr.hdr_entsz != phdr->hdr_entsz ||
 288                     hdr.hdr_crc_table != phdr->hdr_crc_table) {
 289                         /* Read the backup GPT table. */
 290                         size = MIN(MAXTBLSZ, (phdr->hdr_entries *
 291                             phdr->hdr_entsz + table->sectorsize - 1) /
 292                             table->sectorsize);
 293                         if (dread(dev, tbl, size, phdr->hdr_lba_table) == 0 &&
 294                             gpt_checktbl(phdr, tbl, size * table->sectorsize,
 295                             table->sectors - 1) == 0) {
 296                                 memcpy(&hdr, phdr, sizeof(hdr));
 297                                 sec = 1;
 298                         }
 299                 }
 300         }
 301         if (pri == 0 && sec == 0) {
 302                 /* Both primary and backup tables are invalid. */
 303                 table->type = PTABLE_NONE;
 304                 goto out;
 305         }
 306         size = MIN(hdr.hdr_entries * hdr.hdr_entsz,
 307             MAXTBLSZ * table->sectorsize);
 308         for (i = 0; i < size / hdr.hdr_entsz; i++) {
 309                 ent = (struct gpt_ent *)(tbl + i * hdr.hdr_entsz);
 310                 if (uuid_equal(&ent->ent_type, &gpt_uuid_unused, NULL))
 311                         continue;
 312                 entry = malloc(sizeof(*entry));
 313                 if (entry == NULL)
 314                         break;
 315                 entry->part.start = ent->ent_lba_start;
 316                 entry->part.end = ent->ent_lba_end;
 317                 entry->part.index = i + 1;
 318                 entry->part.type = gpt_parttype(ent->ent_type);
 319                 entry->flags = le64toh(ent->ent_attr);
 320                 memcpy(&entry->type.gpt, &ent->ent_type, sizeof(uuid_t));
 321                 STAILQ_INSERT_TAIL(&table->entries, entry, entry);
 322                 DEBUG("new GPT partition added");
 323         }
 324 out:
 325         free(buf);
 326         free(tbl);
 327         return (table);
 328 }
 329 #endif /* LOADER_GPT_SUPPORT */
 330
 331 #ifdef LOADER_MBR_SUPPORT
 332 /* We do not need to support too many EBR partitions in the loader */
 333 #define MAXEBRENTRIES           8
 334 static enum partition_type
 335 mbr_parttype(uint8_t type)
 336 {
 337
 338         switch (type) {
 339         case DOSPTYP_386BSD:
 340                 return (PART_FREEBSD);
 341         case DOSPTYP_LINSWP:
 342                 return (PART_LINUX_SWAP);
 343         case DOSPTYP_LINUX:
 344                 return (PART_LINUX);
 345         case 0x01:
 346         case 0x04:
 347         case 0x06:
 348         case 0x07:
 349         case 0x0b:
 350         case 0x0c:
 351         case 0x0e:
 352                 return (PART_DOS);
 353         }
 354         return (PART_UNKNOWN);
 355 }
 356
 357 struct ptable*
 358 ptable_ebrread(struct ptable *table, void *dev, diskread_t dread)
 359 {
 360         struct dos_partition *dp;
 361         struct pentry *e1, *entry;
 362         uint32_t start, end, offset;
 363         u_char *buf;
 364         int i, index;
 365
 366         STAILQ_FOREACH(e1, &table->entries, entry) {
 367                 if (e1->type.mbr == DOSPTYP_EXT ||
 368                     e1->type.mbr == DOSPTYP_EXTLBA)
 369                         break;
 370         }
 371         if (e1 == NULL)
 372                 return (table);
 373         index = 5;
 374         offset = e1->part.start;
 375         buf = malloc(table->sectorsize);
 376         if (buf == NULL)
 377                 return (table);
 378         for (i = 0; i < MAXEBRENTRIES; i++) {
 379 #if 0   /* Some BIOSes return an incorrect number of sectors */
 380                 if (offset >= table->sectors)
 381                         break;
 382 #endif
 383                 if (dread(dev, buf, 1, offset) != 0)
 384                         break;
 385                 dp = (struct dos_partition *)(buf + DOSPARTOFF);
 386                 if (dp[0].dp_typ == 0)
 387                         break;
 388                 start = le32toh(dp[0].dp_start);
 389                 if (dp[0].dp_typ == DOSPTYP_EXT &&
 390                     dp[1].dp_typ == 0) {
 391                         offset = e1->part.start + start;
 392                         continue;
 393                 }
 394                 end = le32toh(dp[0].dp_size);
 395                 entry = malloc(sizeof(*entry));
 396                 if (entry == NULL)
 397                         break;
 398                 entry->part.start = offset + start;
 399                 entry->part.end = entry->part.start + end - 1;
 400                 entry->part.index = index++;
 401                 entry->part.type = mbr_parttype(dp[0].dp_typ);
 402                 entry->flags = dp[0].dp_flag;
 403                 entry->type.mbr = dp[0].dp_typ;
 404                 STAILQ_INSERT_TAIL(&table->entries, entry, entry);
 405                 DEBUG("new EBR partition added");
 406                 if (dp[1].dp_typ == 0)
 407                         break;
 408                 offset = e1->part.start + le32toh(dp[1].dp_start);
 409         }
 410         free(buf);
 411         return (table);
 412 }
 413 #endif /* LOADER_MBR_SUPPORT */
 414
 415 static enum partition_type
 416 bsd_parttype(uint8_t type)
 417 {
 418
 419         switch (type) {
 420         case FS_NANDFS:
 421                 return (PART_FREEBSD_NANDFS);
 422         case FS_SWAP:
 423                 return (PART_FREEBSD_SWAP);
 424         case FS_BSDFFS:
 425                 return (PART_FREEBSD_UFS);
 426         case FS_VINUM:
 427                 return (PART_FREEBSD_VINUM);
 428         case FS_ZFS:
 429                 return (PART_FREEBSD_ZFS);
 430         }
 431         return (PART_UNKNOWN);
 432 }
 433
 434 struct ptable*
 435 ptable_bsdread(struct ptable *table, void *dev, diskread_t dread)
 436 {
 437         struct disklabel *dl;
 438         struct partition *part;
 439         struct pentry *entry;
 440         u_char *buf;
 441         uint32_t raw_offset;
 442         int i;
 443
 444         if (table->sectorsize < sizeof(struct disklabel)) {
 445                 DEBUG("Too small sectorsize");
 446                 return (table);
 447         }
 448         buf = malloc(table->sectorsize);
 449         if (buf == NULL)
 450                 return (table);
 451         if (dread(dev, buf, 1, 1) != 0) {
 452                 DEBUG("read failed");
 453                 ptable_close(table);
 454                 table = NULL;
 455                 goto out;
 456         }
 457         dl = (struct disklabel *)buf;
 458         if (le32toh(dl->d_magic) != DISKMAGIC &&
 459             le32toh(dl->d_magic2) != DISKMAGIC)
 460                 goto out;
 461         if (le32toh(dl->d_secsize) != table->sectorsize) {
 462                 DEBUG("unsupported sector size");
 463                 goto out;
 464         }
 465         dl->d_npartitions = le16toh(dl->d_npartitions);
 466         if (dl->d_npartitions > 20 || dl->d_npartitions < 8) {
 467                 DEBUG("invalid number of partitions");
 468                 goto out;
 469         }
 470         part = &dl->d_partitions[0];
 471         raw_offset = le32toh(part[RAW_PART].p_offset);
 472         for (i = 0; i < dl->d_npartitions; i++, part++) {
 473                 if (i == RAW_PART)
 474                         continue;
 475                 if (part->p_size == 0)
 476                         continue;
 477                 entry = malloc(sizeof(*entry));
 478                 if (entry == NULL)
 479                         break;
 480                 entry->part.start = le32toh(part->p_offset) - raw_offset;
 481                 entry->part.end = entry->part.start +
 482                     le32toh(part->p_size) + 1;
 483                 entry->part.type = bsd_parttype(part->p_fstype);
 484                 entry->part.index = i; /* starts from zero */
 485                 entry->type.bsd = part->p_fstype;
 486                 STAILQ_INSERT_TAIL(&table->entries, entry, entry);
 487                 DEBUG("new BSD partition added");
 488         }
 489         table->type = PTABLE_BSD;
 490 out:
 491         free(buf);
 492         return (table);
 493 }
 494
 495 #ifdef LOADER_VTOC8_SUPPORT
 496 static enum partition_type
 497 vtoc8_parttype(uint16_t type)
 498 {
 499
 500         switch (type) {
 501         case VTOC_TAG_FREEBSD_NANDFS:
 502                 return (PART_FREEBSD_NANDFS);
 503         case VTOC_TAG_FREEBSD_SWAP:
 504                 return (PART_FREEBSD_SWAP);
 505         case VTOC_TAG_FREEBSD_UFS:
 506                 return (PART_FREEBSD_UFS);
 507         case VTOC_TAG_FREEBSD_VINUM:
 508                 return (PART_FREEBSD_VINUM);
 509         case VTOC_TAG_FREEBSD_ZFS:
 510                 return (PART_FREEBSD_ZFS);
 511         };
 512         return (PART_UNKNOWN);
 513 }
 514
 515 static struct ptable*
 516 ptable_vtoc8read(struct ptable *table, void *dev, diskread_t dread)
 517 {
 518         struct pentry *entry;
 519         struct vtoc8 *dl;
 520         u_char *buf;
 521         uint16_t sum, heads, sectors;
 522         int i;
 523
 524         if (table->sectorsize != sizeof(struct vtoc8))
 525                 return (table);
 526         buf = malloc(table->sectorsize);
 527         if (buf == NULL)
 528                 return (table);
 529         if (dread(dev, buf, 1, 0) != 0) {
 530                 DEBUG("read failed");
 531                 ptable_close(table);
 532                 table = NULL;
 533                 goto out;
 534         }
 535         dl = (struct vtoc8 *)buf;
 536         /* Check the sum */
 537         for (i = sum = 0; i < sizeof(struct vtoc8); i += sizeof(sum))
 538                 sum ^= be16dec(buf + i);
 539         if (sum != 0) {
 540                 DEBUG("incorrect checksum");
 541                 goto out;
 542         }
 543         if (be16toh(dl->nparts) != VTOC8_NPARTS) {
 544                 DEBUG("invalid number of entries");
 545                 goto out;
 546         }
 547         sectors = be16toh(dl->nsecs);
 548         heads = be16toh(dl->nheads);
 549         if (sectors * heads == 0) {
 550                 DEBUG("invalid geometry");
 551                 goto out;
 552         }
 553         for (i = 0; i < VTOC8_NPARTS; i++) {
 554                 dl->part[i].tag = be16toh(dl->part[i].tag);
 555                 if (i == VTOC_RAW_PART ||
 556                     dl->part[i].tag == VTOC_TAG_UNASSIGNED)
 557                         continue;
 558                 entry = malloc(sizeof(*entry));
 559                 if (entry == NULL)
 560                         break;
 561                 entry->part.start = be32toh(dl->map[i].cyl) * heads * sectors;
 562                 entry->part.end = be32toh(dl->map[i].nblks) +
 563                     entry->part.start - 1;
 564                 entry->part.type = vtoc8_parttype(dl->part[i].tag);
 565                 entry->part.index = i; /* starts from zero */
 566                 entry->type.vtoc8 = dl->part[i].tag;
 567                 STAILQ_INSERT_TAIL(&table->entries, entry, entry);
 568                 DEBUG("new VTOC8 partition added");
 569         }
 570         table->type = PTABLE_VTOC8;
 571 out:
 572         free(buf);
 573         return (table);
 574
 575 }
 576 #endif /* LOADER_VTOC8_SUPPORT */
 577
 578 struct ptable*
 579 ptable_open(void *dev, off_t sectors, uint16_t sectorsize,
 580     diskread_t *dread)
 581 {
 582         struct dos_partition *dp;
 583         struct ptable *table;
 584         u_char *buf;
 585         int i, count;
 586 #ifdef LOADER_MBR_SUPPORT
 587         struct pentry *entry;
 588         uint32_t start, end;
 589         int has_ext;
 590 #endif
 591         table = NULL;
 592         buf = malloc(sectorsize);
 593         if (buf == NULL)
 594                 return (NULL);
 595         /* First, read the MBR. */
 596         if (dread(dev, buf, 1, DOSBBSECTOR) != 0) {
 597                 DEBUG("read failed");
 598                 goto out;
 599         }
 600
 601         table = malloc(sizeof(*table));
 602         if (table == NULL)
 603                 goto out;
 604         table->sectors = sectors;
 605         table->sectorsize = sectorsize;
 606         table->type = PTABLE_NONE;
 607         STAILQ_INIT(&table->entries);
 608
 609 #ifdef LOADER_VTOC8_SUPPORT
 610         if (be16dec(buf + offsetof(struct vtoc8, magic)) == VTOC_MAGIC) {
 611                 if (ptable_vtoc8read(table, dev, dread) == NULL) {
 612                         /* Read error. */
 613                         table = NULL;
 614                         goto out;
 615                 } else if (table->type == PTABLE_VTOC8)
 616                         goto out;
 617         }
 618 #endif
 619         /* Check the BSD label. */
 620         if (ptable_bsdread(table, dev, dread) == NULL) { /* Read error. */
 621                 table = NULL;
 622                 goto out;
 623         } else if (table->type == PTABLE_BSD)
 624                 goto out;
 625
 626 #if defined(LOADER_GPT_SUPPORT) || defined(LOADER_MBR_SUPPORT)
 627         /* Check the MBR magic. */
 628         if (buf[DOSMAGICOFFSET] != 0x55 ||
 629             buf[DOSMAGICOFFSET + 1] != 0xaa) {
 630                 DEBUG("magic sequence not found");
 631                 goto out;
 632         }
 633         /* Check that we have PMBR. Also do some validation. */
 634         dp = (struct dos_partition *)(buf + DOSPARTOFF);
 635         for (i = 0, count = 0; i < NDOSPART; i++) {
 636                 if (dp[i].dp_flag != 0 && dp[i].dp_flag != 0x80) {
 637                         DEBUG("invalid partition flag %x", dp[i].dp_flag);
 638                         goto out;
 639                 }
 640 #ifdef LOADER_GPT_SUPPORT
 641                 if (dp[i].dp_typ == DOSPTYP_PMBR) {
 642                         table->type = PTABLE_GPT;
 643                         DEBUG("PMBR detected");
 644                 }
 645 #endif
 646                 if (dp[i].dp_typ != 0)
 647                         count++;
 648         }
 649         /* Do we have some invalid values? */
 650         if (table->type == PTABLE_GPT && count > 1) {
 651                 if (dp[1].dp_typ != DOSPTYP_HFS) {
 652                         table->type = PTABLE_NONE;
 653                         DEBUG("Incorrect PMBR, ignore it");
 654                 } else
 655                         DEBUG("Bootcamp detected");
 656         }
 657 #ifdef LOADER_GPT_SUPPORT
 658         if (table->type == PTABLE_GPT) {
 659                 table = ptable_gptread(table, dev, dread);
 660                 goto out;
 661         }
 662 #endif
 663 #ifdef LOADER_MBR_SUPPORT
 664         /* Read MBR. */
 665         table->type = PTABLE_MBR;
 666         for (i = has_ext = 0; i < NDOSPART; i++) {
 667                 if (dp[i].dp_typ == 0)
 668                         continue;
 669                 start = le32dec(&(dp[i].dp_start));
 670                 end = le32dec(&(dp[i].dp_size));
 671                 if (start == 0 || end == 0)
 672                         continue;
 673 #if 0   /* Some BIOSes return an incorrect number of sectors */
 674                 if (start + end - 1 >= sectors)
 675                         continue;       /* XXX: ignore */
 676 #endif
 677                 if (dp[i].dp_typ == DOSPTYP_EXT ||
 678                     dp[i].dp_typ == DOSPTYP_EXTLBA)
 679                         has_ext = 1;
 680                 entry = malloc(sizeof(*entry));
 681                 if (entry == NULL)
 682                         break;
 683                 entry->part.start = start;
 684                 entry->part.end = start + end - 1;
 685                 entry->part.index = i + 1;
 686                 entry->part.type = mbr_parttype(dp[i].dp_typ);
 687                 entry->flags = dp[i].dp_flag;
 688                 entry->type.mbr = dp[i].dp_typ;
 689                 STAILQ_INSERT_TAIL(&table->entries, entry, entry);
 690                 DEBUG("new MBR partition added");
 691         }
 692         if (has_ext) {
 693                 table = ptable_ebrread(table, dev, dread);
 694                 /* FALLTHROUGH */
 695         }
 696 #endif /* LOADER_MBR_SUPPORT */
 697 #endif /* LOADER_MBR_SUPPORT || LOADER_GPT_SUPPORT */
 698 out:
 699         free(buf);
 700         return (table);
 701 }
 702
 703 void
 704 ptable_close(struct ptable *table)
 705 {
 706         struct pentry *entry;
 707
 708         while (!STAILQ_EMPTY(&table->entries)) {
 709                 entry = STAILQ_FIRST(&table->entries);
 710                 STAILQ_REMOVE_HEAD(&table->entries, entry);
 711                 free(entry);
 712         }
 713         free(table);
 714 }
 715
 716 enum ptable_type
 717 ptable_gettype(const struct ptable *table)
 718 {
 719
 720         return (table->type);
 721 }
 722
 723 int
 724 ptable_getpart(const struct ptable *table, struct ptable_entry *part, int index)
 725 {
 726         struct pentry *entry;
 727
 728         if (part == NULL || table == NULL)
 729                 return (EINVAL);
 730
 731         STAILQ_FOREACH(entry, &table->entries, entry) {
 732                 if (entry->part.index != index)
 733                         continue;
 734                 memcpy(part, &entry->part, sizeof(*part));
 735                 return (0);
 736         }
 737         return (ENOENT);
 738 }
 739
 740 /*
 741  * Search for a slice with the following preferences:
 742  *
 743  * 1: Active FreeBSD slice
 744  * 2: Non-active FreeBSD slice
 745  * 3: Active Linux slice
 746  * 4: non-active Linux slice
 747  * 5: Active FAT/FAT32 slice
 748  * 6: non-active FAT/FAT32 slice
 749  */
 750 #define PREF_RAWDISK    0
 751 #define PREF_FBSD_ACT   1
 752 #define PREF_FBSD       2
 753 #define PREF_LINUX_ACT  3
 754 #define PREF_LINUX      4
 755 #define PREF_DOS_ACT    5
 756 #define PREF_DOS        6
 757 #define PREF_NONE       7
 758 int
 759 ptable_getbestpart(const struct ptable *table, struct ptable_entry *part)
 760 {
 761         struct pentry *entry, *best;
 762         int pref, preflevel;
 763
 764         if (part == NULL || table == NULL)
 765                 return (EINVAL);
 766
 767         best = NULL;
 768         preflevel = pref = PREF_NONE;
 769         STAILQ_FOREACH(entry, &table->entries, entry) {
 770 #ifdef LOADER_MBR_SUPPORT
 771                 if (table->type == PTABLE_MBR) {
 772                         switch (entry->type.mbr) {
 773                         case DOSPTYP_386BSD:
 774                                 pref = entry->flags & 0x80 ? PREF_FBSD_ACT:
 775                                     PREF_FBSD;
 776                                 break;
 777                         case DOSPTYP_LINUX:
 778                                 pref = entry->flags & 0x80 ? PREF_LINUX_ACT:
 779                                     PREF_LINUX;
 780                                 break;
 781                         case 0x01:              /* DOS/Windows */
 782                         case 0x04:
 783                         case 0x06:
 784                         case 0x0c:
 785                         case 0x0e:
 786                         case DOSPTYP_FAT32:
 787                                 pref = entry->flags & 0x80 ? PREF_DOS_ACT:
 788                                     PREF_DOS;
 789                                 break;
 790                         default:
 791                                 pref = PREF_NONE;
 792                         }
 793                 }
 794 #endif /* LOADER_MBR_SUPPORT */
 795 #ifdef LOADER_GPT_SUPPORT
 796                 if (table->type == PTABLE_GPT) {
 797                         if (entry->part.type == PART_DOS)
 798                                 pref = PREF_DOS;
 799                         else if (entry->part.type == PART_FREEBSD_UFS ||
 800                             entry->part.type == PART_FREEBSD_ZFS)
 801                                 pref = PREF_FBSD;
 802                         else
 803                                 pref = PREF_NONE;
 804                 }
 805 #endif /* LOADER_GPT_SUPPORT */
 806                 if (pref < preflevel) {
 807                         preflevel = pref;
 808                         best = entry;
 809                 }
 810         }
 811         if (best != NULL) {
 812                 memcpy(part, &best->part, sizeof(*part));
 813                 return (0);
 814         }
 815         return (ENOENT);
 816 }
 817
 818 void
 819 ptable_iterate(const struct ptable *table, void *arg, ptable_iterate_t *iter)
 820 {
 821         struct pentry *entry;
 822         char name[32];
 823
 824         name[0] = '\0';
 825         STAILQ_FOREACH(entry, &table->entries, entry) {
 826 #ifdef LOADER_MBR_SUPPORT
 827                 if (table->type == PTABLE_MBR)
 828                         sprintf(name, "s%d", entry->part.index);
 829                 else
 830 #endif
 831 #ifdef LOADER_GPT_SUPPORT
 832                 if (table->type == PTABLE_GPT)
 833                         sprintf(name, "p%d", entry->part.index);
 834                 else
 835 #endif
 836 #ifdef LOADER_VTOC8_SUPPORT
 837                 if (table->type == PTABLE_VTOC8)
 838                         sprintf(name, "%c", (u_char) 'a' +
 839                             entry->part.index);
 840                 else
 841 #endif
 842                 if (table->type == PTABLE_BSD)
 843                         sprintf(name, "%c", (u_char) 'a' +
 844                             entry->part.index);
 845                 iter(arg, name, &entry->part);
 846         }
 847 }
 848