kpartx/gpt.c

   1 /*
   2     gpt.[ch]
   3
   4     Copyright (C) 2000-2001 Dell Computer Corporation <Matt_Domsch@dell.com>
   5
   6     EFI GUID Partition Table handling
   7     Per Intel EFI Specification v1.02
   8     http://developer.intel.com/technology/efi/efi.htm
   9
  10     This program is free software; you can redistribute it and/or modify
  11     it under the terms of the GNU General Public License as published by
  12     the Free Software Foundation; either version 2 of the License, or
  13     (at your option) any later version.
  14
  15     This program is distributed in the hope that it will be useful,
  16     but WITHOUT ANY WARRANTY; without even the implied warranty of
  17     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18     GNU General Public License for more details.
  19
  20     You should have received a copy of the GNU General Public License
  21     along with this program; if not, write to the Free Software
  22     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  23 */
  24
  25 #define _FILE_OFFSET_BITS 64
  26
  27 #include "gpt.h"
  28 #include <stdio.h>
  29 #include <string.h>
  30 #include <stdlib.h>
  31 #include <inttypes.h>
  32 #include <sys/stat.h>
  33 #include <sys/ioctl.h>
  34 #include <fcntl.h>
  35 #include <unistd.h>
  36 #include <errno.h>
  37 #include <endian.h>
  38 #include <byteswap.h>
  39 #include <linux/fs.h>
  40 #include "crc32.h"
  41
  42 #if BYTE_ORDER == LITTLE_ENDIAN
  43 #  define __le16_to_cpu(x) (x)
  44 #  define __le32_to_cpu(x) (x)
  45 #  define __le64_to_cpu(x) (x)
  46 #  define __cpu_to_le32(x) (x)
  47 #elif BYTE_ORDER == BIG_ENDIAN
  48 #  define __le16_to_cpu(x) bswap_16(x)
  49 #  define __le32_to_cpu(x) bswap_32(x)
  50 #  define __le64_to_cpu(x) bswap_64(x)
  51 #  define __cpu_to_le32(x) bswap_32(x)
  52 #endif
  53
  54 #ifndef BLKGETLASTSECT
  55 #define BLKGETLASTSECT  _IO(0x12,108)   /* get last sector of block device */
  56 #endif
  57 #ifndef BLKGETSIZE
  58 #define BLKGETSIZE _IO(0x12,96)         /* return device size */
  59 #endif
  60 #ifndef BLKSSZGET
  61 #define BLKSSZGET  _IO(0x12,104)        /* get block device sector size */
  62 #endif
  63 #ifndef BLKGETSIZE64
  64 #define BLKGETSIZE64 _IOR(0x12,114,sizeof(uint64_t))    /* return device size in bytes (u64 *arg) */
  65 #endif
  66
  67 struct blkdev_ioctl_param {
  68         unsigned int block;
  69         size_t content_length;
  70         char * block_contents;
  71 };
  72
  73 /**
  74  * efi_crc32() - EFI version of crc32 function
  75  * @buf: buffer to calculate crc32 of
  76  * @len - length of buf
  77  *
  78  * Description: Returns EFI-style CRC32 value for @buf
  79  *
  80  * This function uses the little endian Ethernet polynomial
  81  * but seeds the function with ~0, and xor's with ~0 at the end.
  82  * Note, the EFI Specification, v1.02, has a reference to
  83  * Dr. Dobbs Journal, May 1994 (actually it's in May 1992).
  84  */
  85 static inline uint32_t
  86 efi_crc32(const void *buf, unsigned long len)
  87 {
  88         return (crc32(~0L, buf, len) ^ ~0L);
  89 }
  90
  91 /**
  92  * is_pmbr_valid(): test Protective MBR for validity
  93  * @mbr: pointer to a legacy mbr structure
  94  *
  95  * Description: Returns 1 if PMBR is valid, 0 otherwise.
  96  * Validity depends on two things:
  97  *  1) MSDOS signature is in the last two bytes of the MBR
  98  *  2) One partition of type 0xEE is found
  99  */
 100 static int
 101 is_pmbr_valid(legacy_mbr *mbr)
 102 {
 103         int i, found = 0, signature = 0;
 104         if (!mbr)
 105                 return 0;
 106         signature = (__le16_to_cpu(mbr->signature) == MSDOS_MBR_SIGNATURE);
 107         for (i = 0; signature && i < 4; i++) {
 108                 if (mbr->partition[i].sys_type ==
 109                     EFI_PMBR_OSTYPE_EFI_GPT) {
 110                         found = 1;
 111                         break;
 112                 }
 113         }
 114         return (signature && found);
 115 }
 116
 117
 118 /************************************************************
 119  * get_sector_size
 120  * Requires:
 121  *  - filedes is an open file descriptor, suitable for reading
 122  * Modifies: nothing
 123  * Returns:
 124  *  sector size, or 512.
 125  ************************************************************/
 126 static int
 127 get_sector_size(int filedes)
 128 {
 129         int rc, sector_size = 512;
 130
 131         rc = ioctl(filedes, BLKSSZGET, &sector_size);
 132         if (rc)
 133                 sector_size = 512;
 134         return sector_size;
 135 }
 136
 137 /************************************************************
 138  * _get_num_sectors
 139  * Requires:
 140  *  - filedes is an open file descriptor, suitable for reading
 141  * Modifies: nothing
 142  * Returns:
 143  *  Last LBA value on success
 144  *  0 on error
 145  *
 146  * Try getting BLKGETSIZE64 and BLKSSZGET first,
 147  * then BLKGETSIZE if necessary.
 148  *  Kernels 2.4.15-2.4.18 and 2.5.0-2.5.3 have a broken BLKGETSIZE64
 149  *  which returns the number of 512-byte sectors, not the size of
 150  *  the disk in bytes. Fixed in kernels 2.4.18-pre8 and 2.5.4-pre3.
 151  ************************************************************/
 152 static uint64_t
 153 _get_num_sectors(int filedes)
 154 {
 155         int rc;
 156         uint64_t bytes=0;
 157
 158         rc = ioctl(filedes, BLKGETSIZE64, &bytes);
 159         if (!rc)
 160                 return bytes / get_sector_size(filedes);
 161
 162         return 0;
 163 }
 164
 165 /************************************************************
 166  * last_lba(): return number of last logical block of device
 167  *
 168  * @fd
 169  *
 170  * Description: returns Last LBA value on success, 0 on error.
 171  * Notes: The value st_blocks gives the size of the file
 172  *        in 512-byte blocks, which is OK if
 173  *        EFI_BLOCK_SIZE_SHIFT == 9.
 174  ************************************************************/
 175
 176 static uint64_t
 177 last_lba(int filedes)
 178 {
 179         int rc;
 180         uint64_t sectors = 0;
 181         struct stat s;
 182         memset(&s, 0, sizeof (s));
 183         rc = fstat(filedes, &s);
 184         if (rc == -1) {
 185                 fprintf(stderr, "last_lba() could not stat: %s\n",
 186                         strerror(errno));
 187                 return 0;
 188         }
 189
 190         if (S_ISBLK(s.st_mode)) {
 191                 sectors = _get_num_sectors(filedes);
 192         } else {
 193                 fprintf(stderr,
 194                         "last_lba(): I don't know how to handle files with mode %x\n",
 195                         s.st_mode);
 196                 sectors = 1;
 197         }
 198
 199         return sectors ? sectors - 1 : 0;
 200 }
 201
 202
 203 static ssize_t
 204 read_lastoddsector(int fd, uint64_t lba, void *buffer, size_t count)
 205 {
 206         int rc;
 207         struct blkdev_ioctl_param ioctl_param;
 208
 209         if (!buffer) return 0;
 210
 211         ioctl_param.block = 0; /* read the last sector */
 212         ioctl_param.content_length = count;
 213         ioctl_param.block_contents = buffer;
 214
 215         rc = ioctl(fd, BLKGETLASTSECT, &ioctl_param);
 216         if (rc == -1) perror("read failed");
 217
 218         return !rc;
 219 }
 220
 221 static ssize_t
 222 read_lba(int fd, uint64_t lba, void *buffer, size_t bytes)
 223 {
 224         int sector_size = get_sector_size(fd);
 225         off_t offset = lba * sector_size;
 226         uint64_t lastlba;
 227         ssize_t bytesread;
 228
 229         lseek(fd, offset, SEEK_SET);
 230         bytesread = read(fd, buffer, bytes);
 231
 232         lastlba = last_lba(fd);
 233         if (!lastlba)
 234                 return bytesread;
 235
 236         /* Kludge.  This is necessary to read/write the last
 237            block of an odd-sized disk, until Linux 2.5.x kernel fixes.
 238            This is only used by gpt.c, and only to read
 239            one sector, so we don't have to be fancy.
 240         */
 241         if (!bytesread && !(lastlba & 1) && lba == lastlba) {
 242                 bytesread = read_lastoddsector(fd, lba, buffer, bytes);
 243         }
 244         return bytesread;
 245 }
 246
 247 /**
 248  * alloc_read_gpt_entries(): reads partition entries from disk
 249  * @fd  is an open file descriptor to the whole disk
 250  * @gpt is a buffer into which the GPT will be put
 251  * Description: Returns ptes on success,  NULL on error.
 252  * Allocates space for PTEs based on information found in @gpt.
 253  * Notes: remember to free pte when you're done!
 254  */
 255 static gpt_entry *
 256 alloc_read_gpt_entries(int fd, gpt_header * gpt)
 257 {
 258         gpt_entry *pte;
 259         size_t count = __le32_to_cpu(gpt->num_partition_entries) *
 260                 __le32_to_cpu(gpt->sizeof_partition_entry);
 261
 262         if (!count) return NULL;
 263
 264         pte = (gpt_entry *)malloc(count);
 265         if (!pte)
 266                 return NULL;
 267         memset(pte, 0, count);
 268
 269         if (!read_lba(fd, __le64_to_cpu(gpt->partition_entry_lba), pte,
 270                       count)) {
 271                 free(pte);
 272                 return NULL;
 273         }
 274         return pte;
 275 }
 276
 277 /**
 278  * alloc_read_gpt_header(): Allocates GPT header, reads into it from disk
 279  * @fd  is an open file descriptor to the whole disk
 280  * @lba is the Logical Block Address of the partition table
 281  *
 282  * Description: returns GPT header on success, NULL on error.   Allocates
 283  * and fills a GPT header starting at @ from @bdev.
 284  * Note: remember to free gpt when finished with it.
 285  */
 286 static gpt_header *
 287 alloc_read_gpt_header(int fd, uint64_t lba)
 288 {
 289         gpt_header *gpt;
 290         gpt = (gpt_header *)
 291             malloc(sizeof (gpt_header));
 292         if (!gpt)
 293                 return NULL;
 294         memset(gpt, 0, sizeof (*gpt));
 295         if (!read_lba(fd, lba, gpt, sizeof (gpt_header))) {
 296                 free(gpt);
 297                 return NULL;
 298         }
 299
 300         return gpt;
 301 }
 302
 303 /**
 304  * is_gpt_valid() - tests one GPT header and PTEs for validity
 305  * @fd  is an open file descriptor to the whole disk
 306  * @lba is the logical block address of the GPT header to test
 307  * @gpt is a GPT header ptr, filled on return.
 308  * @ptes is a PTEs ptr, filled on return.
 309  *
 310  * Description: returns 1 if valid,  0 on error.
 311  * If valid, returns pointers to newly allocated GPT header and PTEs.
 312  */
 313 static int
 314 is_gpt_valid(int fd, uint64_t lba,
 315              gpt_header ** gpt, gpt_entry ** ptes)
 316 {
 317         int rc = 0;             /* default to not valid */
 318         uint32_t crc, origcrc;
 319
 320         if (!gpt || !ptes)
 321                 return 0;
 322         if (!(*gpt = alloc_read_gpt_header(fd, lba)))
 323                 return 0;
 324
 325         /* Check the GUID Partition Table signature */
 326         if (__le64_to_cpu((*gpt)->signature) != GPT_HEADER_SIGNATURE) {
 327                 /*
 328                    printf("GUID Partition Table Header signature is wrong: %" PRIx64" != %" PRIx64 "\n",
 329                    __le64_to_cpu((*gpt)->signature), GUID_PT_HEADER_SIGNATURE);
 330                  */
 331                 free(*gpt);
 332                 *gpt = NULL;
 333                 return rc;
 334         }
 335
 336         /* Check the GUID Partition Table Header CRC */
 337         origcrc = __le32_to_cpu((*gpt)->header_crc32);
 338         (*gpt)->header_crc32 = 0;
 339         crc = efi_crc32(*gpt, __le32_to_cpu((*gpt)->header_size));
 340         if (crc != origcrc) {
 341                 // printf( "GPTH CRC check failed, %x != %x.\n", origcrc, crc);
 342                 (*gpt)->header_crc32 = __cpu_to_le32(origcrc);
 343                 free(*gpt);
 344                 *gpt = NULL;
 345                 return 0;
 346         }
 347         (*gpt)->header_crc32 = __cpu_to_le32(origcrc);
 348
 349         /* Check that the my_lba entry points to the LBA
 350          * that contains the GPT we read */
 351         if (__le64_to_cpu((*gpt)->my_lba) != lba) {
 352                 /*
 353                 printf( "my_lba % PRIx64 "x != lba %"PRIx64 "x.\n",
 354                                 __le64_to_cpu((*gpt)->my_lba), lba);
 355                  */
 356                 free(*gpt);
 357                 *gpt = NULL;
 358                 return 0;
 359         }
 360
 361         /* Check that sizeof_partition_entry has the correct value */
 362         if (__le32_to_cpu((*gpt)->sizeof_partition_entry) != sizeof(gpt_entry)) {
 363                 // printf("GUID partition entry size check failed.\n");
 364                 free(*gpt);
 365                 *gpt = NULL;
 366                 return 0;
 367         }
 368
 369
 370         /* Check that sizeof_partition_entry has the correct value */
 371         if (__le32_to_cpu((*gpt)->sizeof_partition_entry) != sizeof(gpt_entry)) {
 372                 // printf("GUID partition entry size check failed.\n");
 373                 free(*gpt);
 374                 *gpt = NULL;
 375                 return 0;
 376         }
 377
 378
 379         if (!(*ptes = alloc_read_gpt_entries(fd, *gpt))) {
 380                 free(*gpt);
 381                 *gpt = NULL;
 382                 return 0;
 383         }
 384
 385         /* Check the GUID Partition Entry Array CRC */
 386         crc = efi_crc32(*ptes,
 387                         __le32_to_cpu((*gpt)->num_partition_entries) *
 388                         __le32_to_cpu((*gpt)->sizeof_partition_entry));
 389         if (crc != __le32_to_cpu((*gpt)->partition_entry_array_crc32)) {
 390                 // printf("GUID Partitition Entry Array CRC check failed.\n");
 391                 free(*gpt);
 392                 *gpt = NULL;
 393                 free(*ptes);
 394                 *ptes = NULL;
 395                 return 0;
 396         }
 397
 398         /* We're done, all's well */
 399         return 1;
 400 }
 401 /**
 402  * compare_gpts() - Search disk for valid GPT headers and PTEs
 403  * @pgpt is the primary GPT header
 404  * @agpt is the alternate GPT header
 405  * @lastlba is the last LBA number
 406  * Description: Returns nothing.  Sanity checks pgpt and agpt fields
 407  * and prints warnings on discrepancies.
 408  *
 409  */
 410 static void
 411 compare_gpts(gpt_header *pgpt, gpt_header *agpt, uint64_t lastlba)
 412 {
 413         int error_found = 0;
 414         if (!pgpt || !agpt)
 415                 return;
 416         if (__le64_to_cpu(pgpt->my_lba) != __le64_to_cpu(agpt->alternate_lba)) {
 417                 error_found++;
 418                 fprintf(stderr,
 419                        "GPT:Primary header LBA != Alt. header alternate_lba\n");
 420 #ifdef DEBUG
 421                 fprintf(stderr,  "GPT:%" PRIx64 " != %" PRIx64 "\n",
 422                        __le64_to_cpu(pgpt->my_lba),
 423                        __le64_to_cpu(agpt->alternate_lba));
 424 #endif
 425         }
 426         if (__le64_to_cpu(pgpt->alternate_lba) != __le64_to_cpu(agpt->my_lba)) {
 427                 error_found++;
 428                 fprintf(stderr,
 429                        "GPT:Primary header alternate_lba != Alt. header my_lba\n");
 430 #ifdef DEBUG
 431                 fprintf(stderr,  "GPT:%" PRIx64 " != %" PRIx64 "\n",
 432                        __le64_to_cpu(pgpt->alternate_lba),
 433                        __le64_to_cpu(agpt->my_lba));
 434 #endif
 435         }
 436         if (__le64_to_cpu(pgpt->first_usable_lba) !=
 437             __le64_to_cpu(agpt->first_usable_lba)) {
 438                 error_found++;
 439                 fprintf(stderr,  "GPT:first_usable_lbas don't match.\n");
 440 #ifdef DEBUG
 441                 fprintf(stderr,  "GPT:%" PRIx64 " != %" PRIx64 "\n",
 442                        __le64_to_cpu(pgpt->first_usable_lba),
 443                        __le64_to_cpu(agpt->first_usable_lba));
 444 #endif
 445         }
 446         if (__le64_to_cpu(pgpt->last_usable_lba) !=
 447             __le64_to_cpu(agpt->last_usable_lba)) {
 448                 error_found++;
 449                 fprintf(stderr,  "GPT:last_usable_lbas don't match.\n");
 450 #ifdef DEBUG
 451                 fprintf(stderr,  "GPT:%" PRIx64 " != %" PRIx64 "\n",
 452                        __le64_to_cpu(pgpt->last_usable_lba),
 453                        __le64_to_cpu(agpt->last_usable_lba));
 454 #endif
 455         }
 456         if (efi_guidcmp(pgpt->disk_guid, agpt->disk_guid)) {
 457                 error_found++;
 458                 fprintf(stderr,  "GPT:disk_guids don't match.\n");
 459         }
 460         if (__le32_to_cpu(pgpt->num_partition_entries) !=
 461             __le32_to_cpu(agpt->num_partition_entries)) {
 462                 error_found++;
 463                 fprintf(stderr,  "GPT:num_partition_entries don't match: "
 464                        "0x%x != 0x%x\n",
 465                        __le32_to_cpu(pgpt->num_partition_entries),
 466                        __le32_to_cpu(agpt->num_partition_entries));
 467         }
 468         if (__le32_to_cpu(pgpt->sizeof_partition_entry) !=
 469             __le32_to_cpu(agpt->sizeof_partition_entry)) {
 470                 error_found++;
 471                 fprintf(stderr,
 472                        "GPT:sizeof_partition_entry values don't match: "
 473                        "0x%x != 0x%x\n",
 474                        __le32_to_cpu(pgpt->sizeof_partition_entry),
 475                        __le32_to_cpu(agpt->sizeof_partition_entry));
 476         }
 477         if (__le32_to_cpu(pgpt->partition_entry_array_crc32) !=
 478             __le32_to_cpu(agpt->partition_entry_array_crc32)) {
 479                 error_found++;
 480                 fprintf(stderr,
 481                        "GPT:partition_entry_array_crc32 values don't match: "
 482                        "0x%x != 0x%x\n",
 483                        __le32_to_cpu(pgpt->partition_entry_array_crc32),
 484                        __le32_to_cpu(agpt->partition_entry_array_crc32));
 485         }
 486         if (__le64_to_cpu(pgpt->alternate_lba) != lastlba) {
 487                 error_found++;
 488                 fprintf(stderr,
 489                        "GPT:Primary header thinks Alt. header is not at the end of the disk.\n");
 490 #ifdef DEBUG
 491                 fprintf(stderr,  "GPT:%" PRIx64 " != %" PRIx64 "\n",
 492                        __le64_to_cpu(pgpt->alternate_lba), lastlba);
 493 #endif
 494         }
 495
 496         if (__le64_to_cpu(agpt->my_lba) != lastlba) {
 497                 error_found++;
 498                 fprintf(stderr,
 499                        "GPT:Alternate GPT header not at the end of the disk.\n");
 500 #ifdef DEBUG
 501                 fprintf(stderr,  "GPT:%" PRIx64 " != %" PRIx64 "\n",
 502                        __le64_to_cpu(agpt->my_lba), lastlba);
 503 #endif
 504         }
 505
 506         if (error_found)
 507                 fprintf(stderr,
 508                        "GPT: Use GNU Parted to correct GPT errors.\n");
 509         return;
 510 }
 511
 512 /**
 513  * find_valid_gpt() - Search disk for valid GPT headers and PTEs
 514  * @fd  is an open file descriptor to the whole disk
 515  * @gpt is a GPT header ptr, filled on return.
 516  * @ptes is a PTEs ptr, filled on return.
 517  * Description: Returns 1 if valid, 0 on error.
 518  * If valid, returns pointers to newly allocated GPT header and PTEs.
 519  * Validity depends on finding either the Primary GPT header and PTEs valid,
 520  * or the Alternate GPT header and PTEs valid, and the PMBR valid.
 521  */
 522 static int
 523 find_valid_gpt(int fd, gpt_header ** gpt, gpt_entry ** ptes)
 524 {
 525         extern int force_gpt;
 526         int good_pgpt = 0, good_agpt = 0, good_pmbr = 0;
 527         gpt_header *pgpt = NULL, *agpt = NULL;
 528         gpt_entry *pptes = NULL, *aptes = NULL;
 529         legacy_mbr *legacymbr = NULL;
 530         uint64_t lastlba;
 531         if (!gpt || !ptes)
 532                 return 0;
 533
 534         if (!(lastlba = last_lba(fd)))
 535                 return 0;
 536         good_pgpt = is_gpt_valid(fd, GPT_PRIMARY_PARTITION_TABLE_LBA,
 537                                  &pgpt, &pptes);
 538         if (good_pgpt) {
 539                 good_agpt = is_gpt_valid(fd,
 540                                          __le64_to_cpu(pgpt->alternate_lba),
 541                                          &agpt, &aptes);
 542                 if (!good_agpt) {
 543                         good_agpt = is_gpt_valid(fd, lastlba,
 544                                                  &agpt, &aptes);
 545                 }
 546         }
 547         else {
 548                 good_agpt = is_gpt_valid(fd, lastlba,
 549                                          &agpt, &aptes);
 550         }
 551
 552         /* The obviously unsuccessful case */
 553         if (!good_pgpt && !good_agpt) {
 554                 goto fail;
 555         }
 556
 557         /* This will be added to the EFI Spec. per Intel after v1.02. */
 558         legacymbr = malloc(sizeof (*legacymbr));
 559         if (legacymbr) {
 560                 memset(legacymbr, 0, sizeof (*legacymbr));
 561                 read_lba(fd, 0, (uint8_t *) legacymbr,
 562                          sizeof (*legacymbr));
 563                 good_pmbr = is_pmbr_valid(legacymbr);
 564                 free(legacymbr);
 565                 legacymbr=NULL;
 566         }
 567
 568         /* Failure due to bad PMBR */
 569         if ((good_pgpt || good_agpt) && !good_pmbr && !force_gpt) {
 570                 fprintf(stderr,
 571                        "  Warning: Disk has a valid GPT signature "
 572                        "but invalid PMBR.\n"
 573                        "  Assuming this disk is *not* a GPT disk anymore.\n"
 574                        "  Use gpt kernel option to override.  "
 575                        "Use GNU Parted to correct disk.\n");
 576                 goto fail;
 577         }
 578
 579         /* Would fail due to bad PMBR, but force GPT anyhow */
 580         if ((good_pgpt || good_agpt) && !good_pmbr && force_gpt) {
 581                 fprintf(stderr,
 582                        "  Warning: Disk has a valid GPT signature but "
 583                        "invalid PMBR.\n"
 584                        "  Use GNU Parted to correct disk.\n"
 585                        "  gpt option taken, disk treated as GPT.\n");
 586         }
 587
 588         compare_gpts(pgpt, agpt, lastlba);
 589
 590         /* The good cases */
 591         if (good_pgpt && (good_pmbr || force_gpt)) {
 592                 *gpt  = pgpt;
 593                 *ptes = pptes;
 594                 if (agpt)  { free(agpt);   agpt = NULL; }
 595                 if (aptes) { free(aptes); aptes = NULL; }
 596                 if (!good_agpt) {
 597                         fprintf(stderr,
 598                                "Alternate GPT is invalid, "
 599                                "using primary GPT.\n");
 600                 }
 601                 return 1;
 602         }
 603         else if (good_agpt && (good_pmbr || force_gpt)) {
 604                 *gpt  = agpt;
 605                 *ptes = aptes;
 606                 if (pgpt)  { free(pgpt);   pgpt = NULL; }
 607                 if (pptes) { free(pptes); pptes = NULL; }
 608                 fprintf(stderr,
 609                        "Primary GPT is invalid, using alternate GPT.\n");
 610                 return 1;
 611         }
 612
 613  fail:
 614         if (pgpt)  { free(pgpt);   pgpt=NULL; }
 615         if (agpt)  { free(agpt);   agpt=NULL; }
 616         if (pptes) { free(pptes); pptes=NULL; }
 617         if (aptes) { free(aptes); aptes=NULL; }
 618         *gpt = NULL;
 619         *ptes = NULL;
 620         return 0;
 621 }
 622
 623 /**
 624  * read_gpt_pt()
 625  * @fd
 626  * @all - slice with start/size of whole disk
 627  *
 628  *  0 if this isn't our partition table
 629  *  number of partitions if successful
 630  *
 631  */
 632 int
 633 read_gpt_pt (int fd, struct slice all, struct slice *sp, int ns)
 634 {
 635         gpt_header *gpt = NULL;
 636         gpt_entry *ptes = NULL;
 637         uint32_t i;
 638         int n = 0;
 639         int last_used_index=-1;
 640         int sector_size_mul = get_sector_size(fd)/512;
 641
 642         if (!find_valid_gpt (fd, &gpt, &ptes) || !gpt || !ptes) {
 643                 if (gpt)
 644                         free (gpt);
 645                 if (ptes)
 646                         free (ptes);
 647                 return 0;
 648         }
 649
 650         for (i = 0; i < __le32_to_cpu(gpt->num_partition_entries) && i < ns; i++) {
 651                 if (!efi_guidcmp (NULL_GUID, ptes[i].partition_type_guid)) {
 652                         sp[n].start = 0;
 653                         sp[n].size = 0;
 654                         n++;
 655                 } else {
 656                         sp[n].start = sector_size_mul *
 657                                       __le64_to_cpu(ptes[i].starting_lba);
 658                         sp[n].size  = sector_size_mul *
 659                                       (__le64_to_cpu(ptes[i].ending_lba) -
 660                                        __le64_to_cpu(ptes[i].starting_lba) + 1);
 661                         last_used_index=n;
 662                         n++;
 663                 }
 664         }
 665         free (ptes);
 666         free (gpt);
 667         return last_used_index+1;
 668 }