lib/libefi/rdwr_efi.c

   1 /*
   2  * CDDL HEADER START
   3  *
   4  * The contents of this file are subject to the terms of the
   5  * Common Development and Distribution License (the "License").
   6  * You may not use this file except in compliance with the License.
   7  *
   8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
   9  * or http://www.opensolaris.org/os/licensing.
  10  * See the License for the specific language governing permissions
  11  * and limitations under the License.
  12  *
  13  * When distributing Covered Code, include this CDDL HEADER in each
  14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  15  * If applicable, add the following below this CDDL HEADER, with the
  16  * fields enclosed by brackets "[]" replaced with your own identifying
  17  * information: Portions Copyright [yyyy] [name of copyright owner]
  18  *
  19  * CDDL HEADER END
  20  */
  21
  22 /*
  23  * Copyright (c) 2002, 2010, Oracle and/or its affiliates. All rights reserved.
  24  */
  25
  26 #include <stdio.h>
  27 #include <stdlib.h>
  28 #include <errno.h>
  29 #include <strings.h>
  30 #include <unistd.h>
  31 #include <uuid/uuid.h>
  32 #include <zlib.h>
  33 #include <libintl.h>
  34 #include <sys/types.h>
  35 #include <sys/dkio.h>
  36 #include <sys/vtoc.h>
  37 #include <sys/mhd.h>
  38 #include <sys/param.h>
  39 #include <sys/dktp/fdisk.h>
  40 #include <sys/efi_partition.h>
  41 #include <sys/byteorder.h>
  42 #if defined(__linux__)
  43 #include <linux/fs.h>
  44 #endif
  45
  46 static struct uuid_to_ptag {
  47         struct uuid     uuid;
  48 } conversion_array[] = {
  49         { EFI_UNUSED },
  50         { EFI_BOOT },
  51         { EFI_ROOT },
  52         { EFI_SWAP },
  53         { EFI_USR },
  54         { EFI_BACKUP },
  55         { EFI_UNUSED },         /* STAND is never used */
  56         { EFI_VAR },
  57         { EFI_HOME },
  58         { EFI_ALTSCTR },
  59         { EFI_UNUSED },         /* CACHE (cachefs) is never used */
  60         { EFI_RESERVED },
  61         { EFI_SYSTEM },
  62         { EFI_LEGACY_MBR },
  63         { EFI_SYMC_PUB },
  64         { EFI_SYMC_CDS },
  65         { EFI_MSFT_RESV },
  66         { EFI_DELL_BASIC },
  67         { EFI_DELL_RAID },
  68         { EFI_DELL_SWAP },
  69         { EFI_DELL_LVM },
  70         { EFI_DELL_RESV },
  71         { EFI_AAPL_HFS },
  72         { EFI_AAPL_UFS }
  73 };
  74
  75 /*
  76  * Default vtoc information for non-SVr4 partitions
  77  */
  78 struct dk_map2  default_vtoc_map[NDKMAP] = {
  79         {       V_ROOT,         0       },              /* a - 0 */
  80         {       V_SWAP,         V_UNMNT },              /* b - 1 */
  81         {       V_BACKUP,       V_UNMNT },              /* c - 2 */
  82         {       V_UNASSIGNED,   0       },              /* d - 3 */
  83         {       V_UNASSIGNED,   0       },              /* e - 4 */
  84         {       V_UNASSIGNED,   0       },              /* f - 5 */
  85         {       V_USR,          0       },              /* g - 6 */
  86         {       V_UNASSIGNED,   0       },              /* h - 7 */
  87
  88 #if defined(_SUNOS_VTOC_16)
  89
  90 #if defined(i386) || defined(__amd64)
  91         {       V_BOOT,         V_UNMNT },              /* i - 8 */
  92         {       V_ALTSCTR,      0       },              /* j - 9 */
  93
  94 #else
  95 #error No VTOC format defined.
  96 #endif                  /* defined(i386) */
  97
  98         {       V_UNASSIGNED,   0       },              /* k - 10 */
  99         {       V_UNASSIGNED,   0       },              /* l - 11 */
 100         {       V_UNASSIGNED,   0       },              /* m - 12 */
 101         {       V_UNASSIGNED,   0       },              /* n - 13 */
 102         {       V_UNASSIGNED,   0       },              /* o - 14 */
 103         {       V_UNASSIGNED,   0       },              /* p - 15 */
 104 #endif                  /* defined(_SUNOS_VTOC_16) */
 105 };
 106
 107 #ifdef DEBUG
 108 int efi_debug = 1;
 109 #else
 110 int efi_debug = 0;
 111 #endif
 112
 113 static int efi_read(int, struct dk_gpt *);
 114
 115 /*
 116  * Return a 32-bit CRC of the contents of the buffer.  Pre-and-post
 117  * one's conditioning will be handled by crc32() internally.
 118  */
 119 static uint32_t
 120 efi_crc32(const unsigned char *buf, unsigned int size)
 121 {
 122         uint32_t crc = crc32(0, Z_NULL, 0);
 123
 124         crc = crc32(crc, buf, size);
 125
 126         return (crc);
 127 }
 128
 129 static int
 130 read_disk_info(int fd, diskaddr_t *capacity, uint_t *lbsize)
 131 {
 132         int sector_size;
 133         unsigned long long capacity_size;
 134
 135         if (ioctl(fd, BLKSSZGET, &sector_size) < 0)
 136                 return (-1);
 137
 138         if (ioctl(fd, BLKGETSIZE64, &capacity_size) < 0)
 139                 return (-1);
 140
 141         *lbsize = (uint_t)sector_size;
 142         *capacity = (diskaddr_t)(capacity_size / sector_size);
 143
 144         return (0);
 145 }
 146
 147 static int
 148 efi_get_info(int fd, struct dk_cinfo *dki_info)
 149 {
 150 #if defined(__linux__)
 151         char *path;
 152         char *dev_path;
 153         int rval = 0;
 154
 155         memset(dki_info, 0, sizeof(*dki_info));
 156
 157         path = calloc(PATH_MAX, 1);
 158         if (path == NULL)
 159                 goto error;
 160
 161         /*
 162          * The simplest way to get the partition number under linux is
 163          * to parse it out of the /dev/<disk><parition> block device name.
 164          * The kernel creates this using the partition number when it
 165          * populates /dev/ so it may be trusted.  The tricky bit here is
 166          * that the naming convention is based on the block device type.
 167          * So we need to take this in to account when parsing out the
 168          * partition information.  Another issue is that the libefi API
 169          * API only provides the open fd and not the file path.  To handle
 170          * this realpath(3) is used to resolve the block device name from
 171          * /proc/self/fd/<fd>.  Aside from the partition number we collect
 172          * some additional device info.
 173          */
 174         (void) sprintf(path, "/proc/self/fd/%d", fd);
 175         dev_path = realpath(path, NULL);
 176         free(path);
 177
 178         if (dev_path == NULL)
 179                 goto error;
 180
 181         if ((strncmp(dev_path, "/dev/sd", 7) == 0)) {
 182                 strcpy(dki_info->dki_cname, "sd");
 183                 dki_info->dki_ctype = DKC_SCSI_CCS;
 184                 rval = sscanf(dev_path, "/dev/%[a-zA-Z]%hu",
 185                               dki_info->dki_dname,
 186                               &dki_info->dki_partition);
 187         } else if ((strncmp(dev_path, "/dev/hd", 7) == 0)) {
 188                 strcpy(dki_info->dki_cname, "hd");
 189                 dki_info->dki_ctype = DKC_DIRECT;
 190                 rval = sscanf(dev_path, "/dev/%[a-zA-Z]%hu",
 191                               dki_info->dki_dname,
 192                               &dki_info->dki_partition);
 193         } else if ((strncmp(dev_path, "/dev/md", 7) == 0)) {
 194                 strcpy(dki_info->dki_cname, "pseudo");
 195                 dki_info->dki_ctype = DKC_MD;
 196                 rval = sscanf(dev_path, "/dev/%[a-zA-Z0-9]p%hu",
 197                               dki_info->dki_dname,
 198                               &dki_info->dki_partition);
 199         } else if ((strncmp(dev_path, "/dev/dm-", 8) == 0)) {
 200                 strcpy(dki_info->dki_cname, "pseudo");
 201                 dki_info->dki_ctype = DKC_VBD;
 202                 rval = sscanf(dev_path, "/dev/%[a-zA-Z0-9-]p%hu",
 203                               dki_info->dki_dname,
 204                               &dki_info->dki_partition);
 205         } else if ((strncmp(dev_path, "/dev/ram", 8) == 0)) {
 206                 strcpy(dki_info->dki_cname, "pseudo");
 207                 dki_info->dki_ctype = DKC_PCMCIA_MEM;
 208                 rval = sscanf(dev_path, "/dev/%[a-zA-Z0-9]p%hu",
 209                               dki_info->dki_dname,
 210                               &dki_info->dki_partition);
 211         } else if ((strncmp(dev_path, "/dev/loop", 9) == 0)) {
 212                 strcpy(dki_info->dki_cname, "pseudo");
 213                 dki_info->dki_ctype = DKC_VBD;
 214                 rval = sscanf(dev_path, "/dev/%[a-zA-Z0-9]p%hu",
 215                               dki_info->dki_dname,
 216                               &dki_info->dki_partition);
 217         } else {
 218                 strcpy(dki_info->dki_dname, "unknown");
 219                 strcpy(dki_info->dki_cname, "unknown");
 220                 dki_info->dki_ctype = DKC_UNKNOWN;
 221         }
 222
 223         switch (rval) {
 224         case 0:
 225                 errno = EINVAL;
 226                 goto error;
 227         case 1:
 228                 dki_info->dki_partition = 0;
 229         }
 230
 231         free(dev_path);
 232 #else
 233         if (ioctl(fd, DKIOCINFO, (caddr_t)dki_info) == -1)
 234                 goto error;
 235 #endif
 236         return (0);
 237 error:
 238         if (efi_debug)
 239                 (void) fprintf(stderr, "DKIOCINFO errno 0x%x\n", errno);
 240
 241         switch (errno) {
 242         case EIO:
 243                 return (VT_EIO);
 244         case EINVAL:
 245                 return (VT_EINVAL);
 246         default:
 247                 return (VT_ERROR);
 248         }
 249 }
 250
 251 /*
 252  * the number of blocks the EFI label takes up (round up to nearest
 253  * block)
 254  */
 255 #define NBLOCKS(p, l)   (1 + ((((p) * (int)sizeof (efi_gpe_t))  + \
 256                                 ((l) - 1)) / (l)))
 257 /* number of partitions -- limited by what we can malloc */
 258 #define MAX_PARTS       ((4294967295UL - sizeof (struct dk_gpt)) / \
 259                             sizeof (struct dk_part))
 260
 261 int
 262 efi_alloc_and_init(int fd, uint32_t nparts, struct dk_gpt **vtoc)
 263 {
 264         diskaddr_t      capacity = 0;
 265         uint_t          lbsize = 0;
 266         uint_t          nblocks;
 267         size_t          length;
 268         struct dk_gpt   *vptr;
 269         struct uuid     uuid;
 270         struct dk_cinfo dki_info;
 271
 272         if (read_disk_info(fd, &capacity, &lbsize) != 0) {
 273                 if (efi_debug)
 274                         (void) fprintf(stderr,
 275                             "couldn't read disk information\n");
 276                 return (-1);
 277         }
 278 #if defined(__linux__)
 279         if (efi_get_info(fd, &dki_info) != 0) {
 280                 if (efi_debug)
 281                         (void) fprintf(stderr,
 282                             "couldn't read disk information\n");
 283                 return (-1);
 284         }
 285
 286         if (dki_info.dki_partition != 0)
 287                 return (-1);
 288
 289         if ((dki_info.dki_ctype == DKC_PCMCIA_MEM) ||
 290             (dki_info.dki_ctype == DKC_VBD) ||
 291             (dki_info.dki_ctype == DKC_UNKNOWN))
 292                 return (-1);
 293 #endif
 294
 295         nblocks = NBLOCKS(nparts, lbsize);
 296         if ((nblocks * lbsize) < EFI_MIN_ARRAY_SIZE + lbsize) {
 297                 /* 16K plus one block for the GPT */
 298                 nblocks = EFI_MIN_ARRAY_SIZE / lbsize + 1;
 299         }
 300
 301         if (nparts > MAX_PARTS) {
 302                 if (efi_debug) {
 303                         (void) fprintf(stderr,
 304                         "the maximum number of partitions supported is %lu\n",
 305                             MAX_PARTS);
 306                 }
 307                 return (-1);
 308         }
 309
 310         length = sizeof (struct dk_gpt) +
 311             sizeof (struct dk_part) * (nparts - 1);
 312
 313         if ((*vtoc = calloc(length, 1)) == NULL)
 314                 return (-1);
 315
 316         vptr = *vtoc;
 317
 318         vptr->efi_version = EFI_VERSION_CURRENT;
 319         vptr->efi_lbasize = lbsize;
 320         vptr->efi_nparts = nparts;
 321         /*
 322          * add one block here for the PMBR; on disks with a 512 byte
 323          * block size and 128 or fewer partitions, efi_first_u_lba
 324          * should work out to "34"
 325          */
 326         vptr->efi_first_u_lba = nblocks + 1;
 327         vptr->efi_last_lba = capacity - 1;
 328         vptr->efi_altern_lba = capacity -1;
 329         vptr->efi_last_u_lba = vptr->efi_last_lba - nblocks;
 330
 331         (void) uuid_generate((uchar_t *)&uuid);
 332         UUID_LE_CONVERT(vptr->efi_disk_uguid, uuid);
 333         return (0);
 334 }
 335
 336 /*
 337  * Read EFI - return partition number upon success.
 338  */
 339 int
 340 efi_alloc_and_read(int fd, struct dk_gpt **vtoc)
 341 {
 342         int                     rval;
 343         uint32_t                nparts;
 344         int                     length;
 345
 346         /* figure out the number of entries that would fit into 16K */
 347         nparts = EFI_MIN_ARRAY_SIZE / sizeof (efi_gpe_t);
 348         length = (int) sizeof (struct dk_gpt) +
 349             (int) sizeof (struct dk_part) * (nparts - 1);
 350         if ((*vtoc = calloc(length, 1)) == NULL)
 351                 return (VT_ERROR);
 352
 353         (*vtoc)->efi_nparts = nparts;
 354         rval = efi_read(fd, *vtoc);
 355
 356         if ((rval == VT_EINVAL) && (*vtoc)->efi_nparts > nparts) {
 357                 void *tmp;
 358                 length = (int) sizeof (struct dk_gpt) +
 359                     (int) sizeof (struct dk_part) *
 360                     ((*vtoc)->efi_nparts - 1);
 361                 nparts = (*vtoc)->efi_nparts;
 362                 if ((tmp = realloc(*vtoc, length)) == NULL) {
 363                         free (*vtoc);
 364                         *vtoc = NULL;
 365                         return (VT_ERROR);
 366                 } else {
 367                         *vtoc = tmp;
 368                         rval = efi_read(fd, *vtoc);
 369                 }
 370         }
 371
 372         if (rval < 0) {
 373                 if (efi_debug) {
 374                         (void) fprintf(stderr,
 375                             "read of EFI table failed, rval=%d\n", rval);
 376                 }
 377                 free (*vtoc);
 378                 *vtoc = NULL;
 379         }
 380
 381         return (rval);
 382 }
 383
 384 static int
 385 efi_ioctl(int fd, int cmd, dk_efi_t *dk_ioc)
 386 {
 387         void *data = dk_ioc->dki_data;
 388         int error;
 389 #if defined(__linux__)
 390         diskaddr_t capacity;
 391         uint_t lbsize;
 392
 393         /*
 394          * When the IO is not being performed in kernel as an ioctl we need
 395          * to know the sector size so we can seek to the proper byte offset.
 396          */
 397         if (read_disk_info(fd, &capacity, &lbsize) == -1) {
 398                 if (efi_debug)
 399                         fprintf(stderr,"unable to read disk info: %d",errno);
 400
 401                 errno = EIO;
 402                 return -1;
 403         }
 404
 405         switch (cmd) {
 406         case DKIOCGETEFI:
 407                 if (lbsize == 0) {
 408                         if (efi_debug)
 409                                 (void) fprintf(stderr, "DKIOCGETEFI assuming "
 410                                                "LBA %d bytes\n", DEV_BSIZE);
 411
 412                         lbsize = DEV_BSIZE;
 413                 }
 414
 415                 error = lseek(fd, dk_ioc->dki_lba * lbsize, SEEK_SET);
 416                 if (error == -1) {
 417                         if (efi_debug)
 418                                 (void) fprintf(stderr, "DKIOCGETEFI lseek "
 419                                                "error: %d\n", errno);
 420                         return error;
 421                 }
 422
 423                 error = read(fd, data, dk_ioc->dki_length);
 424                 if (error == -1) {
 425                         if (efi_debug)
 426                                 (void) fprintf(stderr, "DKIOCGETEFI read "
 427                                                "error: %d\n", errno);
 428                         return error;
 429                 }
 430
 431                 if (error != dk_ioc->dki_length) {
 432                         if (efi_debug)
 433                                 (void) fprintf(stderr, "DKIOCGETEFI short "
 434                                                "read of %d bytes\n", error);
 435                         errno = EIO;
 436                         return -1;
 437                 }
 438                 error = 0;
 439                 break;
 440
 441         case DKIOCSETEFI:
 442                 if (lbsize == 0) {
 443                         if (efi_debug)
 444                                 (void) fprintf(stderr, "DKIOCSETEFI unknown "
 445                                                "LBA size\n");
 446                         errno = EIO;
 447                         return -1;
 448                 }
 449
 450                 error = lseek(fd, dk_ioc->dki_lba * lbsize, SEEK_SET);
 451                 if (error == -1) {
 452                         if (efi_debug)
 453                                 (void) fprintf(stderr, "DKIOCSETEFI lseek "
 454                                                "error: %d\n", errno);
 455                         return error;
 456                 }
 457
 458                 error = write(fd, data, dk_ioc->dki_length);
 459                 if (error == -1) {
 460                         if (efi_debug)
 461                                 (void) fprintf(stderr, "DKIOCSETEFI write "
 462                                                "error: %d\n", errno);
 463                         return error;
 464                 }
 465
 466                 if (error != dk_ioc->dki_length) {
 467                         if (efi_debug)
 468                                 (void) fprintf(stderr, "DKIOCSETEFI short "
 469                                                "write of %d bytes\n", error);
 470                         errno = EIO;
 471                         return -1;
 472                 }
 473
 474                 /* Sync the new EFI table to disk */
 475                 error = fsync(fd);
 476                 if (error == -1)
 477                         return error;
 478
 479                 /* Ensure any local disk cache is also flushed */
 480                 if (ioctl(fd, BLKFLSBUF, 0) == -1)
 481                         return error;
 482
 483                 error = 0;
 484                 break;
 485
 486         default:
 487                 if (efi_debug)
 488                         (void) fprintf(stderr, "unsupported ioctl()\n");
 489
 490                 errno = EIO;
 491                 return -1;
 492         }
 493 #else
 494         dk_ioc->dki_data_64 = (uint64_t)(uintptr_t)data;
 495         error = ioctl(fd, cmd, (void *)dk_ioc);
 496         dk_ioc->dki_data = data;
 497 #endif
 498         return (error);
 499 }
 500
 501 #if defined(__linux__)
 502 static int
 503 efi_rescan(int fd)
 504 {
 505         int retry = 5;
 506         int error;
 507
 508         /* Notify the kernel a devices partition table has been updated */
 509         while ((error = ioctl(fd, BLKRRPART)) != 0) {
 510                 if (--retry == 0) {
 511                         (void) fprintf(stderr, "the kernel failed to rescan "
 512                                        "the partition table: %d\n", errno);
 513                         return (-1);
 514                 }
 515         }
 516
 517         return (0);
 518 }
 519 #endif
 520
 521 static int
 522 check_label(int fd, dk_efi_t *dk_ioc)
 523 {
 524         efi_gpt_t               *efi;
 525         uint_t                  crc;
 526
 527         if (efi_ioctl(fd, DKIOCGETEFI, dk_ioc) == -1) {
 528                 switch (errno) {
 529                 case EIO:
 530                         return (VT_EIO);
 531                 default:
 532                         return (VT_ERROR);
 533                 }
 534         }
 535         efi = dk_ioc->dki_data;
 536         if (efi->efi_gpt_Signature != LE_64(EFI_SIGNATURE)) {
 537                 if (efi_debug)
 538                         (void) fprintf(stderr,
 539                             "Bad EFI signature: 0x%llx != 0x%llx\n",
 540                             (long long)efi->efi_gpt_Signature,
 541                             (long long)LE_64(EFI_SIGNATURE));
 542                 return (VT_EINVAL);
 543         }
 544
 545         /*
 546          * check CRC of the header; the size of the header should
 547          * never be larger than one block
 548          */
 549         crc = efi->efi_gpt_HeaderCRC32;
 550         efi->efi_gpt_HeaderCRC32 = 0;
 551
 552         if (((len_t)LE_32(efi->efi_gpt_HeaderSize) > dk_ioc->dki_length) ||
 553             crc != LE_32(efi_crc32((unsigned char *)efi,
 554             LE_32(efi->efi_gpt_HeaderSize)))) {
 555                 if (efi_debug)
 556                         (void) fprintf(stderr,
 557                             "Bad EFI CRC: 0x%x != 0x%x\n",
 558                             crc,
 559                             LE_32(efi_crc32((unsigned char *)efi,
 560                             sizeof (struct efi_gpt))));
 561                 return (VT_EINVAL);
 562         }
 563
 564         return (0);
 565 }
 566
 567 static int
 568 efi_read(int fd, struct dk_gpt *vtoc)
 569 {
 570         int                     i, j;
 571         int                     label_len;
 572         int                     rval = 0;
 573         int                     md_flag = 0;
 574         int                     vdc_flag = 0;
 575         diskaddr_t              capacity = 0;
 576         uint_t                  lbsize = 0;
 577         struct dk_minfo         disk_info;
 578         dk_efi_t                dk_ioc;
 579         efi_gpt_t               *efi;
 580         efi_gpe_t               *efi_parts;
 581         struct dk_cinfo         dki_info;
 582         uint32_t                user_length;
 583         boolean_t               legacy_label = B_FALSE;
 584
 585         /*
 586          * get the partition number for this file descriptor.
 587          */
 588         if ((rval = efi_get_info(fd, &dki_info)) != 0)
 589                 return rval;
 590
 591         if ((strncmp(dki_info.dki_cname, "pseudo", 7) == 0) &&
 592             (strncmp(dki_info.dki_dname, "md", 3) == 0)) {
 593                 md_flag++;
 594         } else if ((strncmp(dki_info.dki_cname, "vdc", 4) == 0) &&
 595             (strncmp(dki_info.dki_dname, "vdc", 4) == 0)) {
 596                 /*
 597                  * The controller and drive name "vdc" (virtual disk client)
 598                  * indicates a LDoms virtual disk.
 599                  */
 600                 vdc_flag++;
 601         }
 602
 603         /* get the LBA size */
 604         if (read_disk_info(fd, &capacity, &lbsize) == -1) {
 605                 if (efi_debug) {
 606                         (void) fprintf(stderr,
 607                                        "unable to read disk info: %d",
 608                                        errno);
 609                 }
 610                 return (VT_EINVAL);
 611         }
 612
 613         disk_info.dki_lbsize = lbsize;
 614         disk_info.dki_capacity = capacity;
 615
 616         if (disk_info.dki_lbsize == 0) {
 617                 if (efi_debug) {
 618                         (void) fprintf(stderr,
 619                             "efi_read: assuming LBA 512 bytes\n");
 620                 }
 621                 disk_info.dki_lbsize = DEV_BSIZE;
 622         }
 623         /*
 624          * Read the EFI GPT to figure out how many partitions we need
 625          * to deal with.
 626          */
 627         dk_ioc.dki_lba = 1;
 628         if (NBLOCKS(vtoc->efi_nparts, disk_info.dki_lbsize) < 34) {
 629                 label_len = EFI_MIN_ARRAY_SIZE + disk_info.dki_lbsize;
 630         } else {
 631                 label_len = vtoc->efi_nparts * (int) sizeof (efi_gpe_t) +
 632                     disk_info.dki_lbsize;
 633                 if (label_len % disk_info.dki_lbsize) {
 634                         /* pad to physical sector size */
 635                         label_len += disk_info.dki_lbsize;
 636                         label_len &= ~(disk_info.dki_lbsize - 1);
 637                 }
 638         }
 639
 640         if (posix_memalign((void **)&dk_ioc.dki_data,
 641                            disk_info.dki_lbsize, label_len))
 642                 return (VT_ERROR);
 643
 644         memset(dk_ioc.dki_data, 0, label_len);
 645         dk_ioc.dki_length = disk_info.dki_lbsize;
 646         user_length = vtoc->efi_nparts;
 647         efi = dk_ioc.dki_data;
 648         if (md_flag) {
 649                 dk_ioc.dki_length = label_len;
 650                 if (efi_ioctl(fd, DKIOCGETEFI, &dk_ioc) == -1) {
 651                         switch (errno) {
 652                         case EIO:
 653                                 return (VT_EIO);
 654                         default:
 655                                 return (VT_ERROR);
 656                         }
 657                 }
 658         } else if ((rval = check_label(fd, &dk_ioc)) == VT_EINVAL) {
 659                 /*
 660                  * No valid label here; try the alternate. Note that here
 661                  * we just read GPT header and save it into dk_ioc.data,
 662                  * Later, we will read GUID partition entry array if we
 663                  * can get valid GPT header.
 664                  */
 665
 666                 /*
 667                  * This is a workaround for legacy systems. In the past, the
 668                  * last sector of SCSI disk was invisible on x86 platform. At
 669                  * that time, backup label was saved on the next to the last
 670                  * sector. It is possible for users to move a disk from previous
 671                  * solaris system to present system. Here, we attempt to search
 672                  * legacy backup EFI label first.
 673                  */
 674                 dk_ioc.dki_lba = disk_info.dki_capacity - 2;
 675                 dk_ioc.dki_length = disk_info.dki_lbsize;
 676                 rval = check_label(fd, &dk_ioc);
 677                 if (rval == VT_EINVAL) {
 678                         /*
 679                          * we didn't find legacy backup EFI label, try to
 680                          * search backup EFI label in the last block.
 681                          */
 682                         dk_ioc.dki_lba = disk_info.dki_capacity - 1;
 683                         dk_ioc.dki_length = disk_info.dki_lbsize;
 684                         rval = check_label(fd, &dk_ioc);
 685                         if (rval == 0) {
 686                                 legacy_label = B_TRUE;
 687                                 if (efi_debug)
 688                                         (void) fprintf(stderr,
 689                                             "efi_read: primary label corrupt; "
 690                                             "using EFI backup label located on"
 691                                             " the last block\n");
 692                         }
 693                 } else {
 694                         if ((efi_debug) && (rval == 0))
 695                                 (void) fprintf(stderr, "efi_read: primary label"
 696                                     " corrupt; using legacy EFI backup label "
 697                                     " located on the next to last block\n");
 698                 }
 699
 700                 if (rval == 0) {
 701                         dk_ioc.dki_lba = LE_64(efi->efi_gpt_PartitionEntryLBA);
 702                         vtoc->efi_flags |= EFI_GPT_PRIMARY_CORRUPT;
 703                         vtoc->efi_nparts =
 704                             LE_32(efi->efi_gpt_NumberOfPartitionEntries);
 705                         /*
 706                          * Partition tables are between backup GPT header
 707                          * table and ParitionEntryLBA (the starting LBA of
 708                          * the GUID partition entries array). Now that we
 709                          * already got valid GPT header and saved it in
 710                          * dk_ioc.dki_data, we try to get GUID partition
 711                          * entry array here.
 712                          */
 713                         /* LINTED */
 714                         dk_ioc.dki_data = (efi_gpt_t *)((char *)dk_ioc.dki_data
 715                             + disk_info.dki_lbsize);
 716                         if (legacy_label)
 717                                 dk_ioc.dki_length = disk_info.dki_capacity - 1 -
 718                                     dk_ioc.dki_lba;
 719                         else
 720                                 dk_ioc.dki_length = disk_info.dki_capacity - 2 -
 721                                     dk_ioc.dki_lba;
 722                         dk_ioc.dki_length *= disk_info.dki_lbsize;
 723                         if (dk_ioc.dki_length >
 724                             ((len_t)label_len - sizeof (*dk_ioc.dki_data))) {
 725                                 rval = VT_EINVAL;
 726                         } else {
 727                                 /*
 728                                  * read GUID partition entry array
 729                                  */
 730                                 rval = efi_ioctl(fd, DKIOCGETEFI, &dk_ioc);
 731                         }
 732                 }
 733
 734         } else if (rval == 0) {
 735
 736                 dk_ioc.dki_lba = LE_64(efi->efi_gpt_PartitionEntryLBA);
 737                 /* LINTED */
 738                 dk_ioc.dki_data = (efi_gpt_t *)((char *)dk_ioc.dki_data
 739                     + disk_info.dki_lbsize);
 740                 dk_ioc.dki_length = label_len - disk_info.dki_lbsize;
 741                 rval = efi_ioctl(fd, DKIOCGETEFI, &dk_ioc);
 742
 743         } else if (vdc_flag && rval == VT_ERROR && errno == EINVAL) {
 744                 /*
 745                  * When the device is a LDoms virtual disk, the DKIOCGETEFI
 746                  * ioctl can fail with EINVAL if the virtual disk backend
 747                  * is a ZFS volume serviced by a domain running an old version
 748                  * of Solaris. This is because the DKIOCGETEFI ioctl was
 749                  * initially incorrectly implemented for a ZFS volume and it
 750                  * expected the GPT and GPE to be retrieved with a single ioctl.
 751                  * So we try to read the GPT and the GPE using that old style
 752                  * ioctl.
 753                  */
 754                 dk_ioc.dki_lba = 1;
 755                 dk_ioc.dki_length = label_len;
 756                 rval = check_label(fd, &dk_ioc);
 757         }
 758
 759         if (rval < 0) {
 760                 free(efi);
 761                 return (rval);
 762         }
 763
 764         /* LINTED -- always longlong aligned */
 765         efi_parts = (efi_gpe_t *)(((char *)efi) + disk_info.dki_lbsize);
 766
 767         /*
 768          * Assemble this into a "dk_gpt" struct for easier
 769          * digestibility by applications.
 770          */
 771         vtoc->efi_version = LE_32(efi->efi_gpt_Revision);
 772         vtoc->efi_nparts = LE_32(efi->efi_gpt_NumberOfPartitionEntries);
 773         vtoc->efi_part_size = LE_32(efi->efi_gpt_SizeOfPartitionEntry);
 774         vtoc->efi_lbasize = disk_info.dki_lbsize;
 775         vtoc->efi_last_lba = disk_info.dki_capacity - 1;
 776         vtoc->efi_first_u_lba = LE_64(efi->efi_gpt_FirstUsableLBA);
 777         vtoc->efi_last_u_lba = LE_64(efi->efi_gpt_LastUsableLBA);
 778         vtoc->efi_altern_lba = LE_64(efi->efi_gpt_AlternateLBA);
 779         UUID_LE_CONVERT(vtoc->efi_disk_uguid, efi->efi_gpt_DiskGUID);
 780
 781         /*
 782          * If the array the user passed in is too small, set the length
 783          * to what it needs to be and return
 784          */
 785         if (user_length < vtoc->efi_nparts) {
 786                 return (VT_EINVAL);
 787         }
 788
 789         for (i = 0; i < vtoc->efi_nparts; i++) {
 790
 791                 UUID_LE_CONVERT(vtoc->efi_parts[i].p_guid,
 792                     efi_parts[i].efi_gpe_PartitionTypeGUID);
 793
 794                 for (j = 0;
 795                     j < sizeof (conversion_array)
 796                     / sizeof (struct uuid_to_ptag); j++) {
 797
 798                         if (bcmp(&vtoc->efi_parts[i].p_guid,
 799                             &conversion_array[j].uuid,
 800                             sizeof (struct uuid)) == 0) {
 801                                 vtoc->efi_parts[i].p_tag = j;
 802                                 break;
 803                         }
 804                 }
 805                 if (vtoc->efi_parts[i].p_tag == V_UNASSIGNED)
 806                         continue;
 807                 vtoc->efi_parts[i].p_flag =
 808                     LE_16(efi_parts[i].efi_gpe_Attributes.PartitionAttrs);
 809                 vtoc->efi_parts[i].p_start =
 810                     LE_64(efi_parts[i].efi_gpe_StartingLBA);
 811                 vtoc->efi_parts[i].p_size =
 812                     LE_64(efi_parts[i].efi_gpe_EndingLBA) -
 813                     vtoc->efi_parts[i].p_start + 1;
 814                 for (j = 0; j < EFI_PART_NAME_LEN; j++) {
 815                         vtoc->efi_parts[i].p_name[j] =
 816                             (uchar_t)LE_16(
 817                             efi_parts[i].efi_gpe_PartitionName[j]);
 818                 }
 819
 820                 UUID_LE_CONVERT(vtoc->efi_parts[i].p_uguid,
 821                     efi_parts[i].efi_gpe_UniquePartitionGUID);
 822         }
 823         free(efi);
 824
 825         return (dki_info.dki_partition);
 826 }
 827
 828 /* writes a "protective" MBR */
 829 static int
 830 write_pmbr(int fd, struct dk_gpt *vtoc)
 831 {
 832         dk_efi_t        dk_ioc;
 833         struct mboot    mb;
 834         uchar_t         *cp;
 835         diskaddr_t      size_in_lba;
 836         uchar_t         *buf;
 837         int             len;
 838
 839         len = (vtoc->efi_lbasize == 0) ? sizeof (mb) : vtoc->efi_lbasize;
 840         if (posix_memalign((void **)&buf, len, len))
 841                 return (VT_ERROR);
 842
 843         /*
 844          * Preserve any boot code and disk signature if the first block is
 845          * already an MBR.
 846          */
 847         memset(buf, 0, len);
 848         dk_ioc.dki_lba = 0;
 849         dk_ioc.dki_length = len;
 850         /* LINTED -- always longlong aligned */
 851         dk_ioc.dki_data = (efi_gpt_t *)buf;
 852         if (efi_ioctl(fd, DKIOCGETEFI, &dk_ioc) == -1) {
 853                 (void *) memcpy(&mb, buf, sizeof (mb));
 854                 bzero(&mb, sizeof (mb));
 855                 mb.signature = LE_16(MBB_MAGIC);
 856         } else {
 857                 (void *) memcpy(&mb, buf, sizeof (mb));
 858                 if (mb.signature != LE_16(MBB_MAGIC)) {
 859                         bzero(&mb, sizeof (mb));
 860                         mb.signature = LE_16(MBB_MAGIC);
 861                 }
 862         }
 863
 864         bzero(&mb.parts, sizeof (mb.parts));
 865         cp = (uchar_t *)&mb.parts[0];
 866         /* bootable or not */
 867         *cp++ = 0;
 868         /* beginning CHS; 0xffffff if not representable */
 869         *cp++ = 0xff;
 870         *cp++ = 0xff;
 871         *cp++ = 0xff;
 872         /* OS type */
 873         *cp++ = EFI_PMBR;
 874         /* ending CHS; 0xffffff if not representable */
 875         *cp++ = 0xff;
 876         *cp++ = 0xff;
 877         *cp++ = 0xff;
 878         /* starting LBA: 1 (little endian format) by EFI definition */
 879         *cp++ = 0x01;
 880         *cp++ = 0x00;
 881         *cp++ = 0x00;
 882         *cp++ = 0x00;
 883         /* ending LBA: last block on the disk (little endian format) */
 884         size_in_lba = vtoc->efi_last_lba;
 885         if (size_in_lba < 0xffffffff) {
 886                 *cp++ = (size_in_lba & 0x000000ff);
 887                 *cp++ = (size_in_lba & 0x0000ff00) >> 8;
 888                 *cp++ = (size_in_lba & 0x00ff0000) >> 16;
 889                 *cp++ = (size_in_lba & 0xff000000) >> 24;
 890         } else {
 891                 *cp++ = 0xff;
 892                 *cp++ = 0xff;
 893                 *cp++ = 0xff;
 894                 *cp++ = 0xff;
 895         }
 896
 897         (void *) memcpy(buf, &mb, sizeof (mb));
 898         /* LINTED -- always longlong aligned */
 899         dk_ioc.dki_data = (efi_gpt_t *)buf;
 900         dk_ioc.dki_lba = 0;
 901         dk_ioc.dki_length = len;
 902         if (efi_ioctl(fd, DKIOCSETEFI, &dk_ioc) == -1) {
 903                 free(buf);
 904                 switch (errno) {
 905                 case EIO:
 906                         return (VT_EIO);
 907                 case EINVAL:
 908                         return (VT_EINVAL);
 909                 default:
 910                         return (VT_ERROR);
 911                 }
 912         }
 913         free(buf);
 914         return (0);
 915 }
 916
 917 /* make sure the user specified something reasonable */
 918 static int
 919 check_input(struct dk_gpt *vtoc)
 920 {
 921         int                     resv_part = -1;
 922         int                     i, j;
 923         diskaddr_t              istart, jstart, isize, jsize, endsect;
 924
 925         /*
 926          * Sanity-check the input (make sure no partitions overlap)
 927          */
 928         for (i = 0; i < vtoc->efi_nparts; i++) {
 929                 /* It can't be unassigned and have an actual size */
 930                 if ((vtoc->efi_parts[i].p_tag == V_UNASSIGNED) &&
 931                     (vtoc->efi_parts[i].p_size != 0)) {
 932                         if (efi_debug) {
 933                                 (void) fprintf(stderr, "partition %d is "
 934                                     "\"unassigned\" but has a size of %llu",
 935                                     i, vtoc->efi_parts[i].p_size);
 936                         }
 937                         return (VT_EINVAL);
 938                 }
 939                 if (vtoc->efi_parts[i].p_tag == V_UNASSIGNED) {
 940                         if (uuid_is_null((uchar_t *)&vtoc->efi_parts[i].p_guid))
 941                                 continue;
 942                         /* we have encountered an unknown uuid */
 943                         vtoc->efi_parts[i].p_tag = 0xff;
 944                 }
 945                 if (vtoc->efi_parts[i].p_tag == V_RESERVED) {
 946                         if (resv_part != -1) {
 947                                 if (efi_debug) {
 948                                         (void) fprintf(stderr, "found "
 949                                             "duplicate reserved partition "
 950                                             "at %d\n", i);
 951                                 }
 952                                 return (VT_EINVAL);
 953                         }
 954                         resv_part = i;
 955                 }
 956                 if ((vtoc->efi_parts[i].p_start < vtoc->efi_first_u_lba) ||
 957                     (vtoc->efi_parts[i].p_start > vtoc->efi_last_u_lba)) {
 958                         if (efi_debug) {
 959                                 (void) fprintf(stderr,
 960                                     "Partition %d starts at %llu.  ",
 961                                     i,
 962                                     vtoc->efi_parts[i].p_start);
 963                                 (void) fprintf(stderr,
 964                                     "It must be between %llu and %llu.\n",
 965                                     vtoc->efi_first_u_lba,
 966                                     vtoc->efi_last_u_lba);
 967                         }
 968                         return (VT_EINVAL);
 969                 }
 970                 if ((vtoc->efi_parts[i].p_start +
 971                     vtoc->efi_parts[i].p_size <
 972                     vtoc->efi_first_u_lba) ||
 973                     (vtoc->efi_parts[i].p_start +
 974                     vtoc->efi_parts[i].p_size >
 975                     vtoc->efi_last_u_lba + 1)) {
 976                         if (efi_debug) {
 977                                 (void) fprintf(stderr,
 978                                     "Partition %d ends at %llu.  ",
 979                                     i,
 980                                     vtoc->efi_parts[i].p_start +
 981                                     vtoc->efi_parts[i].p_size);
 982                                 (void) fprintf(stderr,
 983                                     "It must be between %llu and %llu.\n",
 984                                     vtoc->efi_first_u_lba,
 985                                     vtoc->efi_last_u_lba);
 986                         }
 987                         return (VT_EINVAL);
 988                 }
 989
 990                 for (j = 0; j < vtoc->efi_nparts; j++) {
 991                         isize = vtoc->efi_parts[i].p_size;
 992                         jsize = vtoc->efi_parts[j].p_size;
 993                         istart = vtoc->efi_parts[i].p_start;
 994                         jstart = vtoc->efi_parts[j].p_start;
 995                         if ((i != j) && (isize != 0) && (jsize != 0)) {
 996                                 endsect = jstart + jsize -1;
 997                                 if ((jstart <= istart) &&
 998                                     (istart <= endsect)) {
 999                                         if (efi_debug) {
1000                                                 (void) fprintf(stderr,
1001                                                     "Partition %d overlaps "
1002                                                     "partition %d.", i, j);
1003                                         }
1004                                         return (VT_EINVAL);
1005                                 }
1006                         }
1007                 }
1008         }
1009         /* just a warning for now */
1010         if ((resv_part == -1) && efi_debug) {
1011                 (void) fprintf(stderr,
1012                     "no reserved partition found\n");
1013         }
1014         return (0);
1015 }
1016
1017 /*
1018  * add all the unallocated space to the current label
1019  */
1020 int
1021 efi_use_whole_disk(int fd)
1022 {
1023         struct dk_gpt           *efi_label;
1024         int                     rval;
1025         int                     i;
1026         uint_t                  phy_last_slice = 0;
1027         diskaddr_t              pl_start = 0;
1028         diskaddr_t              pl_size;
1029
1030         rval = efi_alloc_and_read(fd, &efi_label);
1031         if (rval < 0) {
1032                 return (rval);
1033         }
1034
1035         /* find the last physically non-zero partition */
1036         for (i = 0; i < efi_label->efi_nparts - 2; i ++) {
1037                 if (pl_start < efi_label->efi_parts[i].p_start) {
1038                         pl_start = efi_label->efi_parts[i].p_start;
1039                         phy_last_slice = i;
1040                 }
1041         }
1042         pl_size = efi_label->efi_parts[phy_last_slice].p_size;
1043
1044         /*
1045          * If alter_lba is 1, we are using the backup label.
1046          * Since we can locate the backup label by disk capacity,
1047          * there must be no unallocated space.
1048          */
1049         if ((efi_label->efi_altern_lba == 1) || (efi_label->efi_altern_lba
1050             >= efi_label->efi_last_lba)) {
1051                 if (efi_debug) {
1052                         (void) fprintf(stderr,
1053                             "efi_use_whole_disk: requested space not found\n");
1054                 }
1055                 efi_free(efi_label);
1056                 return (VT_ENOSPC);
1057         }
1058
1059         /*
1060          * If there is space between the last physically non-zero partition
1061          * and the reserved partition, just add the unallocated space to this
1062          * area. Otherwise, the unallocated space is added to the last
1063          * physically non-zero partition.
1064          */
1065         if (pl_start + pl_size - 1 == efi_label->efi_last_u_lba -
1066             EFI_MIN_RESV_SIZE) {
1067                 efi_label->efi_parts[phy_last_slice].p_size +=
1068                     efi_label->efi_last_lba - efi_label->efi_altern_lba;
1069         }
1070
1071         /*
1072          * Move the reserved partition. There is currently no data in
1073          * here except fabricated devids (which get generated via
1074          * efi_write()). So there is no need to copy data.
1075          */
1076         efi_label->efi_parts[efi_label->efi_nparts - 1].p_start +=
1077             efi_label->efi_last_lba - efi_label->efi_altern_lba;
1078         efi_label->efi_last_u_lba += efi_label->efi_last_lba
1079             - efi_label->efi_altern_lba;
1080
1081         rval = efi_write(fd, efi_label);
1082         if (rval < 0) {
1083                 if (efi_debug) {
1084                         (void) fprintf(stderr,
1085                             "efi_use_whole_disk:fail to write label, rval=%d\n",
1086                             rval);
1087                 }
1088                 efi_free(efi_label);
1089                 return (rval);
1090         }
1091
1092         efi_free(efi_label);
1093         return (0);
1094 }
1095
1096
1097 /*
1098  * write EFI label and backup label
1099  */
1100 int
1101 efi_write(int fd, struct dk_gpt *vtoc)
1102 {
1103         dk_efi_t                dk_ioc;
1104         efi_gpt_t               *efi;
1105         efi_gpe_t               *efi_parts;
1106         int                     i, j;
1107         struct dk_cinfo         dki_info;
1108         int                     rval;
1109         int                     md_flag = 0;
1110         int                     nblocks;
1111         diskaddr_t              lba_backup_gpt_hdr;
1112
1113         if ((rval = efi_get_info(fd, &dki_info)) != 0)
1114                 return rval;
1115
1116         /* check if we are dealing wih a metadevice */
1117         if ((strncmp(dki_info.dki_cname, "pseudo", 7) == 0) &&
1118             (strncmp(dki_info.dki_dname, "md", 3) == 0)) {
1119                 md_flag = 1;
1120         }
1121
1122         if (check_input(vtoc)) {
1123                 /*
1124                  * not valid; if it's a metadevice just pass it down
1125                  * because SVM will do its own checking
1126                  */
1127                 if (md_flag == 0) {
1128                         return (VT_EINVAL);
1129                 }
1130         }
1131
1132         dk_ioc.dki_lba = 1;
1133         if (NBLOCKS(vtoc->efi_nparts, vtoc->efi_lbasize) < 34) {
1134                 dk_ioc.dki_length = EFI_MIN_ARRAY_SIZE + vtoc->efi_lbasize;
1135         } else {
1136                 dk_ioc.dki_length = NBLOCKS(vtoc->efi_nparts,
1137                     vtoc->efi_lbasize) *
1138                     vtoc->efi_lbasize;
1139         }
1140
1141         /*
1142          * the number of blocks occupied by GUID partition entry array
1143          */
1144         nblocks = dk_ioc.dki_length / vtoc->efi_lbasize - 1;
1145
1146         /*
1147          * Backup GPT header is located on the block after GUID
1148          * partition entry array. Here, we calculate the address
1149          * for backup GPT header.
1150          */
1151         lba_backup_gpt_hdr = vtoc->efi_last_u_lba + 1 + nblocks;
1152         if (posix_memalign((void **)&dk_ioc.dki_data,
1153                            vtoc->efi_lbasize, dk_ioc.dki_length))
1154                 return (VT_ERROR);
1155
1156         memset(dk_ioc.dki_data, 0, dk_ioc.dki_length);
1157         efi = dk_ioc.dki_data;
1158
1159         /* stuff user's input into EFI struct */
1160         efi->efi_gpt_Signature = LE_64(EFI_SIGNATURE);
1161         efi->efi_gpt_Revision = LE_32(vtoc->efi_version); /* 0x02000100 */
1162         efi->efi_gpt_HeaderSize = LE_32(sizeof (struct efi_gpt));
1163         efi->efi_gpt_Reserved1 = 0;
1164         efi->efi_gpt_MyLBA = LE_64(1ULL);
1165         efi->efi_gpt_AlternateLBA = LE_64(lba_backup_gpt_hdr);
1166         efi->efi_gpt_FirstUsableLBA = LE_64(vtoc->efi_first_u_lba);
1167         efi->efi_gpt_LastUsableLBA = LE_64(vtoc->efi_last_u_lba);
1168         efi->efi_gpt_PartitionEntryLBA = LE_64(2ULL);
1169         efi->efi_gpt_NumberOfPartitionEntries = LE_32(vtoc->efi_nparts);
1170         efi->efi_gpt_SizeOfPartitionEntry = LE_32(sizeof (struct efi_gpe));
1171         UUID_LE_CONVERT(efi->efi_gpt_DiskGUID, vtoc->efi_disk_uguid);
1172
1173         /* LINTED -- always longlong aligned */
1174         efi_parts = (efi_gpe_t *)((char *)dk_ioc.dki_data + vtoc->efi_lbasize);
1175
1176         for (i = 0; i < vtoc->efi_nparts; i++) {
1177                 for (j = 0;
1178                     j < sizeof (conversion_array) /
1179                     sizeof (struct uuid_to_ptag); j++) {
1180
1181                         if (vtoc->efi_parts[i].p_tag == j) {
1182                                 UUID_LE_CONVERT(
1183                                     efi_parts[i].efi_gpe_PartitionTypeGUID,
1184                                     conversion_array[j].uuid);
1185                                 break;
1186                         }
1187                 }
1188
1189                 if (j == sizeof (conversion_array) /
1190                     sizeof (struct uuid_to_ptag)) {
1191                         /*
1192                          * If we didn't have a matching uuid match, bail here.
1193                          * Don't write a label with unknown uuid.
1194                          */
1195                         if (efi_debug) {
1196                                 (void) fprintf(stderr,
1197                                     "Unknown uuid for p_tag %d\n",
1198                                     vtoc->efi_parts[i].p_tag);
1199                         }
1200                         return (VT_EINVAL);
1201                 }
1202
1203                 /* Zero's should be written for empty partitions */
1204                 if (vtoc->efi_parts[i].p_tag == V_UNASSIGNED)
1205                         continue;
1206
1207                 efi_parts[i].efi_gpe_StartingLBA =
1208                     LE_64(vtoc->efi_parts[i].p_start);
1209                 efi_parts[i].efi_gpe_EndingLBA =
1210                     LE_64(vtoc->efi_parts[i].p_start +
1211                     vtoc->efi_parts[i].p_size - 1);
1212                 efi_parts[i].efi_gpe_Attributes.PartitionAttrs =
1213                     LE_16(vtoc->efi_parts[i].p_flag);
1214                 for (j = 0; j < EFI_PART_NAME_LEN; j++) {
1215                         efi_parts[i].efi_gpe_PartitionName[j] =
1216                             LE_16((ushort_t)vtoc->efi_parts[i].p_name[j]);
1217                 }
1218                 if ((vtoc->efi_parts[i].p_tag != V_UNASSIGNED) &&
1219                     uuid_is_null((uchar_t *)&vtoc->efi_parts[i].p_uguid)) {
1220                         (void) uuid_generate((uchar_t *)
1221                             &vtoc->efi_parts[i].p_uguid);
1222                 }
1223                 bcopy(&vtoc->efi_parts[i].p_uguid,
1224                     &efi_parts[i].efi_gpe_UniquePartitionGUID,
1225                     sizeof (uuid_t));
1226         }
1227         efi->efi_gpt_PartitionEntryArrayCRC32 =
1228             LE_32(efi_crc32((unsigned char *)efi_parts,
1229             vtoc->efi_nparts * (int)sizeof (struct efi_gpe)));
1230         efi->efi_gpt_HeaderCRC32 =
1231             LE_32(efi_crc32((unsigned char *)efi, sizeof (struct efi_gpt)));
1232
1233         if (efi_ioctl(fd, DKIOCSETEFI, &dk_ioc) == -1) {
1234                 free(dk_ioc.dki_data);
1235                 switch (errno) {
1236                 case EIO:
1237                         return (VT_EIO);
1238                 case EINVAL:
1239                         return (VT_EINVAL);
1240                 default:
1241                         return (VT_ERROR);
1242                 }
1243         }
1244         /* if it's a metadevice we're done */
1245         if (md_flag) {
1246                 free(dk_ioc.dki_data);
1247                 return (0);
1248         }
1249
1250         /* write backup partition array */
1251         dk_ioc.dki_lba = vtoc->efi_last_u_lba + 1;
1252         dk_ioc.dki_length -= vtoc->efi_lbasize;
1253         /* LINTED */
1254         dk_ioc.dki_data = (efi_gpt_t *)((char *)dk_ioc.dki_data +
1255             vtoc->efi_lbasize);
1256
1257         if (efi_ioctl(fd, DKIOCSETEFI, &dk_ioc) == -1) {
1258                 /*
1259                  * we wrote the primary label okay, so don't fail
1260                  */
1261                 if (efi_debug) {
1262                         (void) fprintf(stderr,
1263                             "write of backup partitions to block %llu "
1264                             "failed, errno %d\n",
1265                             vtoc->efi_last_u_lba + 1,
1266                             errno);
1267                 }
1268         }
1269         /*
1270          * now swap MyLBA and AlternateLBA fields and write backup
1271          * partition table header
1272          */
1273         dk_ioc.dki_lba = lba_backup_gpt_hdr;
1274         dk_ioc.dki_length = vtoc->efi_lbasize;
1275         /* LINTED */
1276         dk_ioc.dki_data = (efi_gpt_t *)((char *)dk_ioc.dki_data -
1277             vtoc->efi_lbasize);
1278         efi->efi_gpt_AlternateLBA = LE_64(1ULL);
1279         efi->efi_gpt_MyLBA = LE_64(lba_backup_gpt_hdr);
1280         efi->efi_gpt_PartitionEntryLBA = LE_64(vtoc->efi_last_u_lba + 1);
1281         efi->efi_gpt_HeaderCRC32 = 0;
1282         efi->efi_gpt_HeaderCRC32 =
1283             LE_32(efi_crc32((unsigned char *)dk_ioc.dki_data,
1284             sizeof (struct efi_gpt)));
1285
1286         if (efi_ioctl(fd, DKIOCSETEFI, &dk_ioc) == -1) {
1287                 if (efi_debug) {
1288                         (void) fprintf(stderr,
1289                             "write of backup header to block %llu failed, "
1290                             "errno %d\n",
1291                             lba_backup_gpt_hdr,
1292                             errno);
1293                 }
1294         }
1295         /* write the PMBR */
1296         (void) write_pmbr(fd, vtoc);
1297         free(dk_ioc.dki_data);
1298
1299 #if defined(__linux__)
1300         rval = efi_rescan(fd);
1301         if (rval)
1302                 return (VT_ERROR);
1303 #endif
1304
1305         return (0);
1306 }
1307
1308 void
1309 efi_free(struct dk_gpt *ptr)
1310 {
1311         free(ptr);
1312 }
1313
1314 /*
1315  * Input: File descriptor
1316  * Output: 1 if disk has an EFI label, or > 2TB with no VTOC or legacy MBR.
1317  * Otherwise 0.
1318  */
1319 int
1320 efi_type(int fd)
1321 {
1322 #if 0
1323         struct vtoc vtoc;
1324         struct extvtoc extvtoc;
1325
1326         if (ioctl(fd, DKIOCGEXTVTOC, &extvtoc) == -1) {
1327                 if (errno == ENOTSUP)
1328                         return (1);
1329                 else if (errno == ENOTTY) {
1330                         if (ioctl(fd, DKIOCGVTOC, &vtoc) == -1)
1331                                 if (errno == ENOTSUP)
1332                                         return (1);
1333                 }
1334         }
1335         return (0);
1336 #else
1337         return (ENOSYS);
1338 #endif
1339 }
1340
1341 void
1342 efi_err_check(struct dk_gpt *vtoc)
1343 {
1344         int                     resv_part = -1;
1345         int                     i, j;
1346         diskaddr_t              istart, jstart, isize, jsize, endsect;
1347         int                     overlap = 0;
1348
1349         /*
1350          * make sure no partitions overlap
1351          */
1352         for (i = 0; i < vtoc->efi_nparts; i++) {
1353                 /* It can't be unassigned and have an actual size */
1354                 if ((vtoc->efi_parts[i].p_tag == V_UNASSIGNED) &&
1355                     (vtoc->efi_parts[i].p_size != 0)) {
1356                         (void) fprintf(stderr,
1357                             "partition %d is \"unassigned\" but has a size "
1358                             "of %llu\n", i, vtoc->efi_parts[i].p_size);
1359                 }
1360                 if (vtoc->efi_parts[i].p_tag == V_UNASSIGNED) {
1361                         continue;
1362                 }
1363                 if (vtoc->efi_parts[i].p_tag == V_RESERVED) {
1364                         if (resv_part != -1) {
1365                                 (void) fprintf(stderr,
1366                                     "found duplicate reserved partition at "
1367                                     "%d\n", i);
1368                         }
1369                         resv_part = i;
1370                         if (vtoc->efi_parts[i].p_size != EFI_MIN_RESV_SIZE)
1371                                 (void) fprintf(stderr,
1372                                     "Warning: reserved partition size must "
1373                                     "be %d sectors\n", EFI_MIN_RESV_SIZE);
1374                 }
1375                 if ((vtoc->efi_parts[i].p_start < vtoc->efi_first_u_lba) ||
1376                     (vtoc->efi_parts[i].p_start > vtoc->efi_last_u_lba)) {
1377                         (void) fprintf(stderr,
1378                             "Partition %d starts at %llu\n",
1379                             i,
1380                             vtoc->efi_parts[i].p_start);
1381                         (void) fprintf(stderr,
1382                             "It must be between %llu and %llu.\n",
1383                             vtoc->efi_first_u_lba,
1384                             vtoc->efi_last_u_lba);
1385                 }
1386                 if ((vtoc->efi_parts[i].p_start +
1387                     vtoc->efi_parts[i].p_size <
1388                     vtoc->efi_first_u_lba) ||
1389                     (vtoc->efi_parts[i].p_start +
1390                     vtoc->efi_parts[i].p_size >
1391                     vtoc->efi_last_u_lba + 1)) {
1392                         (void) fprintf(stderr,
1393                             "Partition %d ends at %llu\n",
1394                             i,
1395                             vtoc->efi_parts[i].p_start +
1396                             vtoc->efi_parts[i].p_size);
1397                         (void) fprintf(stderr,
1398                             "It must be between %llu and %llu.\n",
1399                             vtoc->efi_first_u_lba,
1400                             vtoc->efi_last_u_lba);
1401                 }
1402
1403                 for (j = 0; j < vtoc->efi_nparts; j++) {
1404                         isize = vtoc->efi_parts[i].p_size;
1405                         jsize = vtoc->efi_parts[j].p_size;
1406                         istart = vtoc->efi_parts[i].p_start;
1407                         jstart = vtoc->efi_parts[j].p_start;
1408                         if ((i != j) && (isize != 0) && (jsize != 0)) {
1409                                 endsect = jstart + jsize -1;
1410                                 if ((jstart <= istart) &&
1411                                     (istart <= endsect)) {
1412                                         if (!overlap) {
1413                                         (void) fprintf(stderr,
1414                                             "label error: EFI Labels do not "
1415                                             "support overlapping partitions\n");
1416                                         }
1417                                         (void) fprintf(stderr,
1418                                             "Partition %d overlaps partition "
1419                                             "%d.\n", i, j);
1420                                         overlap = 1;
1421                                 }
1422                         }
1423                 }
1424         }
1425         /* make sure there is a reserved partition */
1426         if (resv_part == -1) {
1427                 (void) fprintf(stderr,
1428                     "no reserved partition found\n");
1429         }
1430 }
1431
1432 /*
1433  * We need to get information necessary to construct a *new* efi
1434  * label type
1435  */
1436 int
1437 efi_auto_sense(int fd, struct dk_gpt **vtoc)
1438 {
1439
1440         int     i;
1441
1442         /*
1443          * Now build the default partition table
1444          */
1445         if (efi_alloc_and_init(fd, EFI_NUMPAR, vtoc) != 0) {
1446                 if (efi_debug) {
1447                         (void) fprintf(stderr, "efi_alloc_and_init failed.\n");
1448                 }
1449                 return (-1);
1450         }
1451
1452         for (i = 0; i < MIN((*vtoc)->efi_nparts, V_NUMPAR); i++) {
1453                 (*vtoc)->efi_parts[i].p_tag = default_vtoc_map[i].p_tag;
1454                 (*vtoc)->efi_parts[i].p_flag = default_vtoc_map[i].p_flag;
1455                 (*vtoc)->efi_parts[i].p_start = 0;
1456                 (*vtoc)->efi_parts[i].p_size = 0;
1457         }
1458         /*
1459          * Make constants first
1460          * and variable partitions later
1461          */
1462
1463         /* root partition - s0 128 MB */
1464         (*vtoc)->efi_parts[0].p_start = 34;
1465         (*vtoc)->efi_parts[0].p_size = 262144;
1466
1467         /* partition - s1  128 MB */
1468         (*vtoc)->efi_parts[1].p_start = 262178;
1469         (*vtoc)->efi_parts[1].p_size = 262144;
1470
1471         /* partition -s2 is NOT the Backup disk */
1472         (*vtoc)->efi_parts[2].p_tag = V_UNASSIGNED;
1473
1474         /* partition -s6 /usr partition - HOG */
1475         (*vtoc)->efi_parts[6].p_start = 524322;
1476         (*vtoc)->efi_parts[6].p_size = (*vtoc)->efi_last_u_lba - 524322
1477             - (1024 * 16);
1478
1479         /* efi reserved partition - s9 16K */
1480         (*vtoc)->efi_parts[8].p_start = (*vtoc)->efi_last_u_lba - (1024 * 16);
1481         (*vtoc)->efi_parts[8].p_size = (1024 * 16);
1482         (*vtoc)->efi_parts[8].p_tag = V_RESERVED;
1483         return (0);
1484 }