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