block/partitions/efi.c

   1 /************************************************************
   2  * EFI GUID Partition Table handling
   3  *
   4  * http://www.uefi.org/specs/
   5  * http://www.intel.com/technology/efi/
   6  *
   7  * efi.[ch] by Matt Domsch <Matt_Domsch@dell.com>
   8  *   Copyright 2000,2001,2002,2004 Dell Inc.
   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  * TODO:
  26  *
  27  * Changelog:
  28  * Mon Nov 09 2004 Matt Domsch <Matt_Domsch@dell.com>
  29  * - test for valid PMBR and valid PGPT before ever reading
  30  *   AGPT, allow override with 'gpt' kernel command line option.
  31  * - check for first/last_usable_lba outside of size of disk
  32  *
  33  * Tue  Mar 26 2002 Matt Domsch <Matt_Domsch@dell.com>
  34  * - Ported to 2.5.7-pre1 and 2.5.7-dj2
  35  * - Applied patch to avoid fault in alternate header handling
  36  * - cleaned up find_valid_gpt
  37  * - On-disk structure and copy in memory is *always* LE now -
  38  *   swab fields as needed
  39  * - remove print_gpt_header()
  40  * - only use first max_p partition entries, to keep the kernel minor number
  41  *   and partition numbers tied.
  42  *
  43  * Mon  Feb 04 2002 Matt Domsch <Matt_Domsch@dell.com>
  44  * - Removed __PRIPTR_PREFIX - not being used
  45  *
  46  * Mon  Jan 14 2002 Matt Domsch <Matt_Domsch@dell.com>
  47  * - Ported to 2.5.2-pre11 + library crc32 patch Linus applied
  48  *
  49  * Thu Dec 6 2001 Matt Domsch <Matt_Domsch@dell.com>
  50  * - Added compare_gpts().
  51  * - moved le_efi_guid_to_cpus() back into this file.  GPT is the only
  52  *   thing that keeps EFI GUIDs on disk.
  53  * - Changed gpt structure names and members to be simpler and more Linux-like.
  54  *
  55  * Wed Oct 17 2001 Matt Domsch <Matt_Domsch@dell.com>
  56  * - Removed CONFIG_DEVFS_VOLUMES_UUID code entirely per Martin Wilck
  57  *
  58  * Wed Oct 10 2001 Matt Domsch <Matt_Domsch@dell.com>
  59  * - Changed function comments to DocBook style per Andreas Dilger suggestion.
  60  *
  61  * Mon Oct 08 2001 Matt Domsch <Matt_Domsch@dell.com>
  62  * - Change read_lba() to use the page cache per Al Viro's work.
  63  * - print u64s properly on all architectures
  64  * - fixed debug_printk(), now Dprintk()
  65  *
  66  * Mon Oct 01 2001 Matt Domsch <Matt_Domsch@dell.com>
  67  * - Style cleanups
  68  * - made most functions static
  69  * - Endianness addition
  70  * - remove test for second alternate header, as it's not per spec,
  71  *   and is unnecessary.  There's now a method to read/write the last
  72  *   sector of an odd-sized disk from user space.  No tools have ever
  73  *   been released which used this code, so it's effectively dead.
  74  * - Per Asit Mallick of Intel, added a test for a valid PMBR.
  75  * - Added kernel command line option 'gpt' to override valid PMBR test.
  76  *
  77  * Wed Jun  6 2001 Martin Wilck <Martin.Wilck@Fujitsu-Siemens.com>
  78  * - added devfs volume UUID support (/dev/volumes/uuids) for
  79  *   mounting file systems by the partition GUID.
  80  *
  81  * Tue Dec  5 2000 Matt Domsch <Matt_Domsch@dell.com>
  82  * - Moved crc32() to linux/lib, added efi_crc32().
  83  *
  84  * Thu Nov 30 2000 Matt Domsch <Matt_Domsch@dell.com>
  85  * - Replaced Intel's CRC32 function with an equivalent
  86  *   non-license-restricted version.
  87  *
  88  * Wed Oct 25 2000 Matt Domsch <Matt_Domsch@dell.com>
  89  * - Fixed the last_lba() call to return the proper last block
  90  *
  91  * Thu Oct 12 2000 Matt Domsch <Matt_Domsch@dell.com>
  92  * - Thanks to Andries Brouwer for his debugging assistance.
  93  * - Code works, detects all the partitions.
  94  *
  95  ************************************************************/
  96 #include <linux/crc32.h>
  97 #include <linux/ctype.h>
  98 #include <linux/math64.h>
  99 #include <linux/slab.h>
 100 #include "check.h"
 101 #include "efi.h"
 102
 103 /* This allows a kernel command line option 'gpt' to override
 104  * the test for invalid PMBR.  Not __initdata because reloading
 105  * the partition tables happens after init too.
 106  */
 107 static int force_gpt;
 108 static int __init
 109 force_gpt_fn(char *str)
 110 {
 111         force_gpt = 1;
 112         return 1;
 113 }
 114 __setup("gpt", force_gpt_fn);
 115
 116
 117 /**
 118  * efi_crc32() - EFI version of crc32 function
 119  * @buf: buffer to calculate crc32 of
 120  * @len - length of buf
 121  *
 122  * Description: Returns EFI-style CRC32 value for @buf
 123  *
 124  * This function uses the little endian Ethernet polynomial
 125  * but seeds the function with ~0, and xor's with ~0 at the end.
 126  * Note, the EFI Specification, v1.02, has a reference to
 127  * Dr. Dobbs Journal, May 1994 (actually it's in May 1992).
 128  */
 129 static inline u32
 130 efi_crc32(const void *buf, unsigned long len)
 131 {
 132         return (crc32(~0L, buf, len) ^ ~0L);
 133 }
 134
 135 /**
 136  * last_lba(): return number of last logical block of device
 137  * @bdev: block device
 138  *
 139  * Description: Returns last LBA value on success, 0 on error.
 140  * This is stored (by sd and ide-geometry) in
 141  *  the part[0] entry for this disk, and is the number of
 142  *  physical sectors available on the disk.
 143  */
 144 static u64 last_lba(struct block_device *bdev)
 145 {
 146         if (!bdev || !bdev->bd_inode)
 147                 return 0;
 148         return div_u64(bdev->bd_inode->i_size,
 149                        bdev_logical_block_size(bdev)) - 1ULL;
 150 }
 151
 152 static inline int
 153 pmbr_part_valid(struct partition *part)
 154 {
 155         if (part->sys_ind == EFI_PMBR_OSTYPE_EFI_GPT &&
 156             le32_to_cpu(part->start_sect) == 1UL)
 157                 return 1;
 158         return 0;
 159 }
 160
 161 /**
 162  * is_pmbr_valid(): test Protective MBR for validity
 163  * @mbr: pointer to a legacy mbr structure
 164  *
 165  * Description: Returns 1 if PMBR is valid, 0 otherwise.
 166  * Validity depends on two things:
 167  *  1) MSDOS signature is in the last two bytes of the MBR
 168  *  2) One partition of type 0xEE is found
 169  */
 170 static int
 171 is_pmbr_valid(legacy_mbr *mbr)
 172 {
 173         int i;
 174         if (!mbr || le16_to_cpu(mbr->signature) != MSDOS_MBR_SIGNATURE)
 175                 return 0;
 176         for (i = 0; i < 4; i++)
 177                 if (pmbr_part_valid(&mbr->partition_record[i]))
 178                         return 1;
 179         return 0;
 180 }
 181
 182 /**
 183  * read_lba(): Read bytes from disk, starting at given LBA
 184  * @state
 185  * @lba
 186  * @buffer
 187  * @size_t
 188  *
 189  * Description: Reads @count bytes from @state->bdev into @buffer.
 190  * Returns number of bytes read on success, 0 on error.
 191  */
 192 static size_t read_lba(struct parsed_partitions *state,
 193                        u64 lba, u8 *buffer, size_t count)
 194 {
 195         size_t totalreadcount = 0;
 196         struct block_device *bdev = state->bdev;
 197         sector_t n = lba * (bdev_logical_block_size(bdev) / 512);
 198
 199         if (!buffer || lba > last_lba(bdev))
 200                 return 0;
 201
 202         while (count) {
 203                 int copied = 512;
 204                 Sector sect;
 205                 unsigned char *data = read_part_sector(state, n++, &sect);
 206                 if (!data)
 207                         break;
 208                 if (copied > count)
 209                         copied = count;
 210                 memcpy(buffer, data, copied);
 211                 put_dev_sector(sect);
 212                 buffer += copied;
 213                 totalreadcount +=copied;
 214                 count -= copied;
 215         }
 216         return totalreadcount;
 217 }
 218
 219 /**
 220  * alloc_read_gpt_entries(): reads partition entries from disk
 221  * @state
 222  * @gpt - GPT header
 223  *
 224  * Description: Returns ptes on success,  NULL on error.
 225  * Allocates space for PTEs based on information found in @gpt.
 226  * Notes: remember to free pte when you're done!
 227  */
 228 static gpt_entry *alloc_read_gpt_entries(struct parsed_partitions *state,
 229                                          gpt_header *gpt)
 230 {
 231         size_t count;
 232         gpt_entry *pte;
 233
 234         if (!gpt)
 235                 return NULL;
 236
 237         count = le32_to_cpu(gpt->num_partition_entries) *
 238                 le32_to_cpu(gpt->sizeof_partition_entry);
 239         if (!count)
 240                 return NULL;
 241         pte = kzalloc(count, GFP_KERNEL);
 242         if (!pte)
 243                 return NULL;
 244
 245         if (read_lba(state, le64_to_cpu(gpt->partition_entry_lba),
 246                      (u8 *) pte,
 247                      count) < count) {
 248                 kfree(pte);
 249                 pte=NULL;
 250                 return NULL;
 251         }
 252         return pte;
 253 }
 254
 255 /**
 256  * alloc_read_gpt_header(): Allocates GPT header, reads into it from disk
 257  * @state
 258  * @lba is the Logical Block Address of the partition table
 259  *
 260  * Description: returns GPT header on success, NULL on error.   Allocates
 261  * and fills a GPT header starting at @ from @state->bdev.
 262  * Note: remember to free gpt when finished with it.
 263  */
 264 static gpt_header *alloc_read_gpt_header(struct parsed_partitions *state,
 265                                          u64 lba)
 266 {
 267         gpt_header *gpt;
 268         unsigned ssz = bdev_logical_block_size(state->bdev);
 269
 270         gpt = kzalloc(ssz, GFP_KERNEL);
 271         if (!gpt)
 272                 return NULL;
 273
 274         if (read_lba(state, lba, (u8 *) gpt, ssz) < ssz) {
 275                 kfree(gpt);
 276                 gpt=NULL;
 277                 return NULL;
 278         }
 279
 280         return gpt;
 281 }
 282
 283 /**
 284  * is_gpt_valid() - tests one GPT header and PTEs for validity
 285  * @state
 286  * @lba is the logical block address of the GPT header to test
 287  * @gpt is a GPT header ptr, filled on return.
 288  * @ptes is a PTEs ptr, filled on return.
 289  *
 290  * Description: returns 1 if valid,  0 on error.
 291  * If valid, returns pointers to newly allocated GPT header and PTEs.
 292  */
 293 static int is_gpt_valid(struct parsed_partitions *state, u64 lba,
 294                         gpt_header **gpt, gpt_entry **ptes)
 295 {
 296         u32 crc, origcrc;
 297         u64 lastlba;
 298
 299         if (!ptes)
 300                 return 0;
 301         if (!(*gpt = alloc_read_gpt_header(state, lba)))
 302                 return 0;
 303
 304         /* Check the GUID Partition Table signature */
 305         if (le64_to_cpu((*gpt)->signature) != GPT_HEADER_SIGNATURE) {
 306                 pr_debug("GUID Partition Table Header signature is wrong:"
 307                          "%lld != %lld\n",
 308                          (unsigned long long)le64_to_cpu((*gpt)->signature),
 309                          (unsigned long long)GPT_HEADER_SIGNATURE);
 310                 goto fail;
 311         }
 312
 313         /* Check the GUID Partition Table header size */
 314         if (le32_to_cpu((*gpt)->header_size) >
 315                         bdev_logical_block_size(state->bdev)) {
 316                 pr_debug("GUID Partition Table Header size is wrong: %u > %u\n",
 317                         le32_to_cpu((*gpt)->header_size),
 318                         bdev_logical_block_size(state->bdev));
 319                 goto fail;
 320         }
 321
 322         /* Check the GUID Partition Table header size */
 323         if (le32_to_cpu((*gpt)->header_size) >
 324                         bdev_logical_block_size(state->bdev)) {
 325                 pr_debug("GUID Partition Table Header size is wrong: %u > %u\n",
 326                         le32_to_cpu((*gpt)->header_size),
 327                         bdev_logical_block_size(state->bdev));
 328                 goto fail;
 329         }
 330
 331         /* Check the GUID Partition Table CRC */
 332         origcrc = le32_to_cpu((*gpt)->header_crc32);
 333         (*gpt)->header_crc32 = 0;
 334         crc = efi_crc32((const unsigned char *) (*gpt), le32_to_cpu((*gpt)->header_size));
 335
 336         if (crc != origcrc) {
 337                 pr_debug("GUID Partition Table Header CRC is wrong: %x != %x\n",
 338                          crc, origcrc);
 339                 goto fail;
 340         }
 341         (*gpt)->header_crc32 = cpu_to_le32(origcrc);
 342
 343         /* Check that the my_lba entry points to the LBA that contains
 344          * the GUID Partition Table */
 345         if (le64_to_cpu((*gpt)->my_lba) != lba) {
 346                 pr_debug("GPT my_lba incorrect: %lld != %lld\n",
 347                          (unsigned long long)le64_to_cpu((*gpt)->my_lba),
 348                          (unsigned long long)lba);
 349                 goto fail;
 350         }
 351
 352         /* Check the first_usable_lba and last_usable_lba are
 353          * within the disk.
 354          */
 355         lastlba = last_lba(state->bdev);
 356         if (le64_to_cpu((*gpt)->first_usable_lba) > lastlba) {
 357                 pr_debug("GPT: first_usable_lba incorrect: %lld > %lld\n",
 358                          (unsigned long long)le64_to_cpu((*gpt)->first_usable_lba),
 359                          (unsigned long long)lastlba);
 360                 goto fail;
 361         }
 362         if (le64_to_cpu((*gpt)->last_usable_lba) > lastlba) {
 363                 pr_debug("GPT: last_usable_lba incorrect: %lld > %lld\n",
 364                          (unsigned long long)le64_to_cpu((*gpt)->last_usable_lba),
 365                          (unsigned long long)lastlba);
 366                 goto fail;
 367         }
 368
 369         /* Check that sizeof_partition_entry has the correct value */
 370         if (le32_to_cpu((*gpt)->sizeof_partition_entry) != sizeof(gpt_entry)) {
 371                 pr_debug("GUID Partitition Entry Size check failed.\n");
 372                 goto fail;
 373         }
 374
 375         if (!(*ptes = alloc_read_gpt_entries(state, *gpt)))
 376                 goto fail;
 377
 378         /* Check the GUID Partition Entry Array CRC */
 379         crc = efi_crc32((const unsigned char *) (*ptes),
 380                         le32_to_cpu((*gpt)->num_partition_entries) *
 381                         le32_to_cpu((*gpt)->sizeof_partition_entry));
 382
 383         if (crc != le32_to_cpu((*gpt)->partition_entry_array_crc32)) {
 384                 pr_debug("GUID Partitition Entry Array CRC check failed.\n");
 385                 goto fail_ptes;
 386         }
 387
 388         /* We're done, all's well */
 389         return 1;
 390
 391  fail_ptes:
 392         kfree(*ptes);
 393         *ptes = NULL;
 394  fail:
 395         kfree(*gpt);
 396         *gpt = NULL;
 397         return 0;
 398 }
 399
 400 /**
 401  * is_pte_valid() - tests one PTE for validity
 402  * @pte is the pte to check
 403  * @lastlba is last lba of the disk
 404  *
 405  * Description: returns 1 if valid,  0 on error.
 406  */
 407 static inline int
 408 is_pte_valid(const gpt_entry *pte, const u64 lastlba)
 409 {
 410         if ((!efi_guidcmp(pte->partition_type_guid, NULL_GUID)) ||
 411             le64_to_cpu(pte->starting_lba) > lastlba         ||
 412             le64_to_cpu(pte->ending_lba)   > lastlba)
 413                 return 0;
 414         return 1;
 415 }
 416
 417 /**
 418  * compare_gpts() - Search disk for valid GPT headers and PTEs
 419  * @pgpt is the primary GPT header
 420  * @agpt is the alternate GPT header
 421  * @lastlba is the last LBA number
 422  * Description: Returns nothing.  Sanity checks pgpt and agpt fields
 423  * and prints warnings on discrepancies.
 424  *
 425  */
 426 static void
 427 compare_gpts(gpt_header *pgpt, gpt_header *agpt, u64 lastlba)
 428 {
 429         int error_found = 0;
 430         if (!pgpt || !agpt)
 431                 return;
 432         if (le64_to_cpu(pgpt->my_lba) != le64_to_cpu(agpt->alternate_lba)) {
 433                 printk(KERN_WARNING
 434                        "GPT:Primary header LBA != Alt. header alternate_lba\n");
 435                 printk(KERN_WARNING "GPT:%lld != %lld\n",
 436                        (unsigned long long)le64_to_cpu(pgpt->my_lba),
 437                        (unsigned long long)le64_to_cpu(agpt->alternate_lba));
 438                 error_found++;
 439         }
 440         if (le64_to_cpu(pgpt->alternate_lba) != le64_to_cpu(agpt->my_lba)) {
 441                 printk(KERN_WARNING
 442                        "GPT:Primary header alternate_lba != Alt. header my_lba\n");
 443                 printk(KERN_WARNING "GPT:%lld != %lld\n",
 444                        (unsigned long long)le64_to_cpu(pgpt->alternate_lba),
 445                        (unsigned long long)le64_to_cpu(agpt->my_lba));
 446                 error_found++;
 447         }
 448         if (le64_to_cpu(pgpt->first_usable_lba) !=
 449             le64_to_cpu(agpt->first_usable_lba)) {
 450                 printk(KERN_WARNING "GPT:first_usable_lbas don't match.\n");
 451                 printk(KERN_WARNING "GPT:%lld != %lld\n",
 452                        (unsigned long long)le64_to_cpu(pgpt->first_usable_lba),
 453                        (unsigned long long)le64_to_cpu(agpt->first_usable_lba));
 454                 error_found++;
 455         }
 456         if (le64_to_cpu(pgpt->last_usable_lba) !=
 457             le64_to_cpu(agpt->last_usable_lba)) {
 458                 printk(KERN_WARNING "GPT:last_usable_lbas don't match.\n");
 459                 printk(KERN_WARNING "GPT:%lld != %lld\n",
 460                        (unsigned long long)le64_to_cpu(pgpt->last_usable_lba),
 461                        (unsigned long long)le64_to_cpu(agpt->last_usable_lba));
 462                 error_found++;
 463         }
 464         if (efi_guidcmp(pgpt->disk_guid, agpt->disk_guid)) {
 465                 printk(KERN_WARNING "GPT:disk_guids don't match.\n");
 466                 error_found++;
 467         }
 468         if (le32_to_cpu(pgpt->num_partition_entries) !=
 469             le32_to_cpu(agpt->num_partition_entries)) {
 470                 printk(KERN_WARNING "GPT:num_partition_entries don't match: "
 471                        "0x%x != 0x%x\n",
 472                        le32_to_cpu(pgpt->num_partition_entries),
 473                        le32_to_cpu(agpt->num_partition_entries));
 474                 error_found++;
 475         }
 476         if (le32_to_cpu(pgpt->sizeof_partition_entry) !=
 477             le32_to_cpu(agpt->sizeof_partition_entry)) {
 478                 printk(KERN_WARNING
 479                        "GPT:sizeof_partition_entry values don't match: "
 480                        "0x%x != 0x%x\n",
 481                        le32_to_cpu(pgpt->sizeof_partition_entry),
 482                        le32_to_cpu(agpt->sizeof_partition_entry));
 483                 error_found++;
 484         }
 485         if (le32_to_cpu(pgpt->partition_entry_array_crc32) !=
 486             le32_to_cpu(agpt->partition_entry_array_crc32)) {
 487                 printk(KERN_WARNING
 488                        "GPT:partition_entry_array_crc32 values don't match: "
 489                        "0x%x != 0x%x\n",
 490                        le32_to_cpu(pgpt->partition_entry_array_crc32),
 491                        le32_to_cpu(agpt->partition_entry_array_crc32));
 492                 error_found++;
 493         }
 494         if (le64_to_cpu(pgpt->alternate_lba) != lastlba) {
 495                 printk(KERN_WARNING
 496                        "GPT:Primary header thinks Alt. header is not at the end of the disk.\n");
 497                 printk(KERN_WARNING "GPT:%lld != %lld\n",
 498                         (unsigned long long)le64_to_cpu(pgpt->alternate_lba),
 499                         (unsigned long long)lastlba);
 500                 error_found++;
 501         }
 502
 503         if (le64_to_cpu(agpt->my_lba) != lastlba) {
 504                 printk(KERN_WARNING
 505                        "GPT:Alternate GPT header not at the end of the disk.\n");
 506                 printk(KERN_WARNING "GPT:%lld != %lld\n",
 507                         (unsigned long long)le64_to_cpu(agpt->my_lba),
 508                         (unsigned long long)lastlba);
 509                 error_found++;
 510         }
 511
 512         if (error_found)
 513                 printk(KERN_WARNING
 514                        "GPT: Use GNU Parted to correct GPT errors.\n");
 515         return;
 516 }
 517
 518 /**
 519  * find_valid_gpt() - Search disk for valid GPT headers and PTEs
 520  * @state
 521  * @gpt is a GPT header ptr, filled on return.
 522  * @ptes is a PTEs ptr, filled on return.
 523  * Description: Returns 1 if valid, 0 on error.
 524  * If valid, returns pointers to newly allocated GPT header and PTEs.
 525  * Validity depends on PMBR being valid (or being overridden by the
 526  * 'gpt' kernel command line option) and finding either the Primary
 527  * GPT header and PTEs valid, or the Alternate GPT header and PTEs
 528  * valid.  If the Primary GPT header is not valid, the Alternate GPT header
 529  * is not checked unless the 'gpt' kernel command line option is passed.
 530  * This protects against devices which misreport their size, and forces
 531  * the user to decide to use the Alternate GPT.
 532  */
 533 static int find_valid_gpt(struct parsed_partitions *state, gpt_header **gpt,
 534                           gpt_entry **ptes)
 535 {
 536         int good_pgpt = 0, good_agpt = 0, good_pmbr = 0;
 537         gpt_header *pgpt = NULL, *agpt = NULL;
 538         gpt_entry *pptes = NULL, *aptes = NULL;
 539         legacy_mbr *legacymbr;
 540         u64 lastlba;
 541
 542         if (!ptes)
 543                 return 0;
 544
 545         lastlba = last_lba(state->bdev);
 546         if (!force_gpt) {
 547                 /* This will be added to the EFI Spec. per Intel after v1.02. */
 548                 legacymbr = kzalloc(sizeof (*legacymbr), GFP_KERNEL);
 549                 if (legacymbr) {
 550                         read_lba(state, 0, (u8 *) legacymbr,
 551                                  sizeof (*legacymbr));
 552                         good_pmbr = is_pmbr_valid(legacymbr);
 553                         kfree(legacymbr);
 554                 }
 555                 if (!good_pmbr)
 556                         goto fail;
 557         }
 558
 559         good_pgpt = is_gpt_valid(state, GPT_PRIMARY_PARTITION_TABLE_LBA,
 560                                  &pgpt, &pptes);
 561         if (good_pgpt)
 562                 good_agpt = is_gpt_valid(state,
 563                                          le64_to_cpu(pgpt->alternate_lba),
 564                                          &agpt, &aptes);
 565         if (!good_agpt && force_gpt)
 566                 good_agpt = is_gpt_valid(state, lastlba, &agpt, &aptes);
 567
 568         /* The obviously unsuccessful case */
 569         if (!good_pgpt && !good_agpt)
 570                 goto fail;
 571
 572         compare_gpts(pgpt, agpt, lastlba);
 573
 574         /* The good cases */
 575         if (good_pgpt) {
 576                 *gpt  = pgpt;
 577                 *ptes = pptes;
 578                 kfree(agpt);
 579                 kfree(aptes);
 580                 if (!good_agpt) {
 581                         printk(KERN_WARNING
 582                                "Alternate GPT is invalid, "
 583                                "using primary GPT.\n");
 584                 }
 585                 return 1;
 586         }
 587         else if (good_agpt) {
 588                 *gpt  = agpt;
 589                 *ptes = aptes;
 590                 kfree(pgpt);
 591                 kfree(pptes);
 592                 printk(KERN_WARNING
 593                        "Primary GPT is invalid, using alternate GPT.\n");
 594                 return 1;
 595         }
 596
 597  fail:
 598         kfree(pgpt);
 599         kfree(agpt);
 600         kfree(pptes);
 601         kfree(aptes);
 602         *gpt = NULL;
 603         *ptes = NULL;
 604         return 0;
 605 }
 606
 607 /**
 608  * efi_partition(struct parsed_partitions *state)
 609  * @state
 610  *
 611  * Description: called from check.c, if the disk contains GPT
 612  * partitions, sets up partition entries in the kernel.
 613  *
 614  * If the first block on the disk is a legacy MBR,
 615  * it will get handled by msdos_partition().
 616  * If it's a Protective MBR, we'll handle it here.
 617  *
 618  * We do not create a Linux partition for GPT, but
 619  * only for the actual data partitions.
 620  * Returns:
 621  * -1 if unable to read the partition table
 622  *  0 if this isn't our partition table
 623  *  1 if successful
 624  *
 625  */
 626 int efi_partition(struct parsed_partitions *state)
 627 {
 628         gpt_header *gpt = NULL;
 629         gpt_entry *ptes = NULL;
 630         u32 i;
 631         unsigned ssz = bdev_logical_block_size(state->bdev) / 512;
 632         u8 unparsed_guid[37];
 633
 634         if (!find_valid_gpt(state, &gpt, &ptes) || !gpt || !ptes) {
 635                 kfree(gpt);
 636                 kfree(ptes);
 637                 return 0;
 638         }
 639
 640         pr_debug("GUID Partition Table is valid!  Yea!\n");
 641
 642         for (i = 0; i < le32_to_cpu(gpt->num_partition_entries) && i < state->limit-1; i++) {
 643                 struct partition_meta_info *info;
 644                 unsigned label_count = 0;
 645                 unsigned label_max;
 646                 u64 start = le64_to_cpu(ptes[i].starting_lba);
 647                 u64 size = le64_to_cpu(ptes[i].ending_lba) -
 648                            le64_to_cpu(ptes[i].starting_lba) + 1ULL;
 649
 650                 if (!is_pte_valid(&ptes[i], last_lba(state->bdev)))
 651                         continue;
 652
 653                 put_partition(state, i+1, start * ssz, size * ssz);
 654
 655                 /* If this is a RAID volume, tell md */
 656                 if (!efi_guidcmp(ptes[i].partition_type_guid,
 657                                  PARTITION_LINUX_RAID_GUID))
 658                         state->parts[i + 1].flags = ADDPART_FLAG_RAID;
 659
 660                 info = &state->parts[i + 1].info;
 661                 /* Instead of doing a manual swap to big endian, reuse the
 662                  * common ASCII hex format as the interim.
 663                  */
 664                 efi_guid_unparse(&ptes[i].unique_partition_guid, unparsed_guid);
 665                 part_pack_uuid(unparsed_guid, info->uuid);
 666
 667                 /* Naively convert UTF16-LE to 7 bits. */
 668                 label_max = min(sizeof(info->volname) - 1,
 669                                 sizeof(ptes[i].partition_name));
 670                 info->volname[label_max] = 0;
 671                 while (label_count < label_max) {
 672                         u8 c = ptes[i].partition_name[label_count] & 0xff;
 673                         if (c && !isprint(c))
 674                                 c = '!';
 675                         info->volname[label_count] = c;
 676                         label_count++;
 677                 }
 678                 state->parts[i + 1].has_info = true;
 679         }
 680         kfree(ptes);
 681         kfree(gpt);
 682         strlcat(state->pp_buf, "\n", PAGE_SIZE);
 683         return 1;
 684 }