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 pmbr_part_valid(gpt_mbr_record *part)
 153 {
 154         if (part->os_type != EFI_PMBR_OSTYPE_EFI_GPT)
 155                 goto invalid;
 156
 157         /* set to 0x00000001 (i.e., the LBA of the GPT Partition Header) */
 158         if (le32_to_cpu(part->starting_lba) != GPT_PRIMARY_PARTITION_TABLE_LBA)
 159                 goto invalid;
 160
 161         return GPT_MBR_PROTECTIVE;
 162 invalid:
 163         return 0;
 164 }
 165
 166 /**
 167  * is_pmbr_valid(): test Protective MBR for validity
 168  * @mbr: pointer to a legacy mbr structure
 169  *
 170  * Description: Checks for a valid protective or hybrid
 171  * master boot record (MBR). The validity of a pMBR depends
 172  * on all of the following properties:
 173  *  1) MSDOS signature is in the last two bytes of the MBR
 174  *  2) One partition of type 0xEE is found
 175  *
 176  * In addition, a hybrid MBR will have up to three additional
 177  * primary partitions, which point to the same space that's
 178  * marked out by up to three GPT partitions.
 179  *
 180  * Returns 0 upon invalid MBR, or GPT_MBR_PROTECTIVE or
 181  * GPT_MBR_HYBRID depending on the device layout.
 182  */
 183 static int is_pmbr_valid(legacy_mbr *mbr)
 184 {
 185         int i, ret = 0; /* invalid by default */
 186
 187         if (!mbr || le16_to_cpu(mbr->signature) != MSDOS_MBR_SIGNATURE)
 188                 goto done;
 189
 190         for (i = 0; i < 4; i++) {
 191                 ret = pmbr_part_valid(&mbr->partition_record[i]);
 192                 if (ret == GPT_MBR_PROTECTIVE) {
 193                         /*
 194                          * Ok, we at least know that there's a protective MBR,
 195                          * now check if there are other partition types for
 196                          * hybrid MBR.
 197                          */
 198                         goto check_hybrid;
 199                 }
 200         }
 201
 202         if (ret != GPT_MBR_PROTECTIVE)
 203                 goto done;
 204 check_hybrid:
 205         for (i = 0; i < 4; i++)
 206                 if ((mbr->partition_record[i].os_type !=
 207                         EFI_PMBR_OSTYPE_EFI_GPT) &&
 208                     (mbr->partition_record[i].os_type != 0x00))
 209                         ret = GPT_MBR_HYBRID;
 210 done:
 211         return ret;
 212 }
 213
 214 /**
 215  * read_lba(): Read bytes from disk, starting at given LBA
 216  * @state
 217  * @lba
 218  * @buffer
 219  * @size_t
 220  *
 221  * Description: Reads @count bytes from @state->bdev into @buffer.
 222  * Returns number of bytes read on success, 0 on error.
 223  */
 224 static size_t read_lba(struct parsed_partitions *state,
 225                        u64 lba, u8 *buffer, size_t count)
 226 {
 227         size_t totalreadcount = 0;
 228         struct block_device *bdev = state->bdev;
 229         sector_t n = lba * (bdev_logical_block_size(bdev) / 512);
 230
 231         if (!buffer || lba > last_lba(bdev))
 232                 return 0;
 233
 234         while (count) {
 235                 int copied = 512;
 236                 Sector sect;
 237                 unsigned char *data = read_part_sector(state, n++, &sect);
 238                 if (!data)
 239                         break;
 240                 if (copied > count)
 241                         copied = count;
 242                 memcpy(buffer, data, copied);
 243                 put_dev_sector(sect);
 244                 buffer += copied;
 245                 totalreadcount +=copied;
 246                 count -= copied;
 247         }
 248         return totalreadcount;
 249 }
 250
 251 /**
 252  * alloc_read_gpt_entries(): reads partition entries from disk
 253  * @state
 254  * @gpt - GPT header
 255  *
 256  * Description: Returns ptes on success,  NULL on error.
 257  * Allocates space for PTEs based on information found in @gpt.
 258  * Notes: remember to free pte when you're done!
 259  */
 260 static gpt_entry *alloc_read_gpt_entries(struct parsed_partitions *state,
 261                                          gpt_header *gpt)
 262 {
 263         size_t count;
 264         gpt_entry *pte;
 265
 266         if (!gpt)
 267                 return NULL;
 268
 269         count = le32_to_cpu(gpt->num_partition_entries) *
 270                 le32_to_cpu(gpt->sizeof_partition_entry);
 271         if (!count)
 272                 return NULL;
 273         pte = kmalloc(count, GFP_KERNEL);
 274         if (!pte)
 275                 return NULL;
 276
 277         if (read_lba(state, le64_to_cpu(gpt->partition_entry_lba),
 278                      (u8 *) pte,
 279                      count) < count) {
 280                 kfree(pte);
 281                 pte=NULL;
 282                 return NULL;
 283         }
 284         return pte;
 285 }
 286
 287 /**
 288  * alloc_read_gpt_header(): Allocates GPT header, reads into it from disk
 289  * @state
 290  * @lba is the Logical Block Address of the partition table
 291  *
 292  * Description: returns GPT header on success, NULL on error.   Allocates
 293  * and fills a GPT header starting at @ from @state->bdev.
 294  * Note: remember to free gpt when finished with it.
 295  */
 296 static gpt_header *alloc_read_gpt_header(struct parsed_partitions *state,
 297                                          u64 lba)
 298 {
 299         gpt_header *gpt;
 300         unsigned ssz = bdev_logical_block_size(state->bdev);
 301
 302         gpt = kmalloc(ssz, GFP_KERNEL);
 303         if (!gpt)
 304                 return NULL;
 305
 306         if (read_lba(state, lba, (u8 *) gpt, ssz) < ssz) {
 307                 kfree(gpt);
 308                 gpt=NULL;
 309                 return NULL;
 310         }
 311
 312         return gpt;
 313 }
 314
 315 /**
 316  * is_gpt_valid() - tests one GPT header and PTEs for validity
 317  * @state
 318  * @lba is the logical block address of the GPT header to test
 319  * @gpt is a GPT header ptr, filled on return.
 320  * @ptes is a PTEs ptr, filled on return.
 321  *
 322  * Description: returns 1 if valid,  0 on error.
 323  * If valid, returns pointers to newly allocated GPT header and PTEs.
 324  */
 325 static int is_gpt_valid(struct parsed_partitions *state, u64 lba,
 326                         gpt_header **gpt, gpt_entry **ptes)
 327 {
 328         u32 crc, origcrc;
 329         u64 lastlba;
 330
 331         if (!ptes)
 332                 return 0;
 333         if (!(*gpt = alloc_read_gpt_header(state, lba)))
 334                 return 0;
 335
 336         /* Check the GUID Partition Table signature */
 337         if (le64_to_cpu((*gpt)->signature) != GPT_HEADER_SIGNATURE) {
 338                 pr_debug("GUID Partition Table Header signature is wrong:"
 339                          "%lld != %lld\n",
 340                          (unsigned long long)le64_to_cpu((*gpt)->signature),
 341                          (unsigned long long)GPT_HEADER_SIGNATURE);
 342                 goto fail;
 343         }
 344
 345         /* Check the GUID Partition Table header size is too big */
 346         if (le32_to_cpu((*gpt)->header_size) >
 347                         bdev_logical_block_size(state->bdev)) {
 348                 pr_debug("GUID Partition Table Header size is too large: %u > %u\n",
 349                         le32_to_cpu((*gpt)->header_size),
 350                         bdev_logical_block_size(state->bdev));
 351                 goto fail;
 352         }
 353
 354         /* Check the GUID Partition Table header size is too small */
 355         if (le32_to_cpu((*gpt)->header_size) < sizeof(gpt_header)) {
 356                 pr_debug("GUID Partition Table Header size is too small: %u < %zu\n",
 357                         le32_to_cpu((*gpt)->header_size),
 358                         sizeof(gpt_header));
 359                 goto fail;
 360         }
 361
 362         /* Check the GUID Partition Table CRC */
 363         origcrc = le32_to_cpu((*gpt)->header_crc32);
 364         (*gpt)->header_crc32 = 0;
 365         crc = efi_crc32((const unsigned char *) (*gpt), le32_to_cpu((*gpt)->header_size));
 366
 367         if (crc != origcrc) {
 368                 pr_debug("GUID Partition Table Header CRC is wrong: %x != %x\n",
 369                          crc, origcrc);
 370                 goto fail;
 371         }
 372         (*gpt)->header_crc32 = cpu_to_le32(origcrc);
 373
 374         /* Check that the my_lba entry points to the LBA that contains
 375          * the GUID Partition Table */
 376         if (le64_to_cpu((*gpt)->my_lba) != lba) {
 377                 pr_debug("GPT my_lba incorrect: %lld != %lld\n",
 378                          (unsigned long long)le64_to_cpu((*gpt)->my_lba),
 379                          (unsigned long long)lba);
 380                 goto fail;
 381         }
 382
 383         /* Check the first_usable_lba and last_usable_lba are
 384          * within the disk.
 385          */
 386         lastlba = last_lba(state->bdev);
 387         if (le64_to_cpu((*gpt)->first_usable_lba) > lastlba) {
 388                 pr_debug("GPT: first_usable_lba incorrect: %lld > %lld\n",
 389                          (unsigned long long)le64_to_cpu((*gpt)->first_usable_lba),
 390                          (unsigned long long)lastlba);
 391                 goto fail;
 392         }
 393         if (le64_to_cpu((*gpt)->last_usable_lba) > lastlba) {
 394                 pr_debug("GPT: last_usable_lba incorrect: %lld > %lld\n",
 395                          (unsigned long long)le64_to_cpu((*gpt)->last_usable_lba),
 396                          (unsigned long long)lastlba);
 397                 goto fail;
 398         }
 399
 400         /* Check that sizeof_partition_entry has the correct value */
 401         if (le32_to_cpu((*gpt)->sizeof_partition_entry) != sizeof(gpt_entry)) {
 402                 pr_debug("GUID Partitition Entry Size check failed.\n");
 403                 goto fail;
 404         }
 405
 406         if (!(*ptes = alloc_read_gpt_entries(state, *gpt)))
 407                 goto fail;
 408
 409         /* Check the GUID Partition Entry Array CRC */
 410         crc = efi_crc32((const unsigned char *) (*ptes),
 411                         le32_to_cpu((*gpt)->num_partition_entries) *
 412                         le32_to_cpu((*gpt)->sizeof_partition_entry));
 413
 414         if (crc != le32_to_cpu((*gpt)->partition_entry_array_crc32)) {
 415                 pr_debug("GUID Partitition Entry Array CRC check failed.\n");
 416                 goto fail_ptes;
 417         }
 418
 419         /* We're done, all's well */
 420         return 1;
 421
 422  fail_ptes:
 423         kfree(*ptes);
 424         *ptes = NULL;
 425  fail:
 426         kfree(*gpt);
 427         *gpt = NULL;
 428         return 0;
 429 }
 430
 431 /**
 432  * is_pte_valid() - tests one PTE for validity
 433  * @pte is the pte to check
 434  * @lastlba is last lba of the disk
 435  *
 436  * Description: returns 1 if valid,  0 on error.
 437  */
 438 static inline int
 439 is_pte_valid(const gpt_entry *pte, const u64 lastlba)
 440 {
 441         if ((!efi_guidcmp(pte->partition_type_guid, NULL_GUID)) ||
 442             le64_to_cpu(pte->starting_lba) > lastlba         ||
 443             le64_to_cpu(pte->ending_lba)   > lastlba)
 444                 return 0;
 445         return 1;
 446 }
 447
 448 /**
 449  * compare_gpts() - Search disk for valid GPT headers and PTEs
 450  * @pgpt is the primary GPT header
 451  * @agpt is the alternate GPT header
 452  * @lastlba is the last LBA number
 453  * Description: Returns nothing.  Sanity checks pgpt and agpt fields
 454  * and prints warnings on discrepancies.
 455  *
 456  */
 457 static void
 458 compare_gpts(gpt_header *pgpt, gpt_header *agpt, u64 lastlba)
 459 {
 460         int error_found = 0;
 461         if (!pgpt || !agpt)
 462                 return;
 463         if (le64_to_cpu(pgpt->my_lba) != le64_to_cpu(agpt->alternate_lba)) {
 464                 printk(KERN_WARNING
 465                        "GPT:Primary header LBA != Alt. header alternate_lba\n");
 466                 printk(KERN_WARNING "GPT:%lld != %lld\n",
 467                        (unsigned long long)le64_to_cpu(pgpt->my_lba),
 468                        (unsigned long long)le64_to_cpu(agpt->alternate_lba));
 469                 error_found++;
 470         }
 471         if (le64_to_cpu(pgpt->alternate_lba) != le64_to_cpu(agpt->my_lba)) {
 472                 printk(KERN_WARNING
 473                        "GPT:Primary header alternate_lba != Alt. header my_lba\n");
 474                 printk(KERN_WARNING "GPT:%lld != %lld\n",
 475                        (unsigned long long)le64_to_cpu(pgpt->alternate_lba),
 476                        (unsigned long long)le64_to_cpu(agpt->my_lba));
 477                 error_found++;
 478         }
 479         if (le64_to_cpu(pgpt->first_usable_lba) !=
 480             le64_to_cpu(agpt->first_usable_lba)) {
 481                 printk(KERN_WARNING "GPT:first_usable_lbas don't match.\n");
 482                 printk(KERN_WARNING "GPT:%lld != %lld\n",
 483                        (unsigned long long)le64_to_cpu(pgpt->first_usable_lba),
 484                        (unsigned long long)le64_to_cpu(agpt->first_usable_lba));
 485                 error_found++;
 486         }
 487         if (le64_to_cpu(pgpt->last_usable_lba) !=
 488             le64_to_cpu(agpt->last_usable_lba)) {
 489                 printk(KERN_WARNING "GPT:last_usable_lbas don't match.\n");
 490                 printk(KERN_WARNING "GPT:%lld != %lld\n",
 491                        (unsigned long long)le64_to_cpu(pgpt->last_usable_lba),
 492                        (unsigned long long)le64_to_cpu(agpt->last_usable_lba));
 493                 error_found++;
 494         }
 495         if (efi_guidcmp(pgpt->disk_guid, agpt->disk_guid)) {
 496                 printk(KERN_WARNING "GPT:disk_guids don't match.\n");
 497                 error_found++;
 498         }
 499         if (le32_to_cpu(pgpt->num_partition_entries) !=
 500             le32_to_cpu(agpt->num_partition_entries)) {
 501                 printk(KERN_WARNING "GPT:num_partition_entries don't match: "
 502                        "0x%x != 0x%x\n",
 503                        le32_to_cpu(pgpt->num_partition_entries),
 504                        le32_to_cpu(agpt->num_partition_entries));
 505                 error_found++;
 506         }
 507         if (le32_to_cpu(pgpt->sizeof_partition_entry) !=
 508             le32_to_cpu(agpt->sizeof_partition_entry)) {
 509                 printk(KERN_WARNING
 510                        "GPT:sizeof_partition_entry values don't match: "
 511                        "0x%x != 0x%x\n",
 512                        le32_to_cpu(pgpt->sizeof_partition_entry),
 513                        le32_to_cpu(agpt->sizeof_partition_entry));
 514                 error_found++;
 515         }
 516         if (le32_to_cpu(pgpt->partition_entry_array_crc32) !=
 517             le32_to_cpu(agpt->partition_entry_array_crc32)) {
 518                 printk(KERN_WARNING
 519                        "GPT:partition_entry_array_crc32 values don't match: "
 520                        "0x%x != 0x%x\n",
 521                        le32_to_cpu(pgpt->partition_entry_array_crc32),
 522                        le32_to_cpu(agpt->partition_entry_array_crc32));
 523                 error_found++;
 524         }
 525         if (le64_to_cpu(pgpt->alternate_lba) != lastlba) {
 526                 printk(KERN_WARNING
 527                        "GPT:Primary header thinks Alt. header is not at the end of the disk.\n");
 528                 printk(KERN_WARNING "GPT:%lld != %lld\n",
 529                         (unsigned long long)le64_to_cpu(pgpt->alternate_lba),
 530                         (unsigned long long)lastlba);
 531                 error_found++;
 532         }
 533
 534         if (le64_to_cpu(agpt->my_lba) != lastlba) {
 535                 printk(KERN_WARNING
 536                        "GPT:Alternate GPT header not at the end of the disk.\n");
 537                 printk(KERN_WARNING "GPT:%lld != %lld\n",
 538                         (unsigned long long)le64_to_cpu(agpt->my_lba),
 539                         (unsigned long long)lastlba);
 540                 error_found++;
 541         }
 542
 543         if (error_found)
 544                 printk(KERN_WARNING
 545                        "GPT: Use GNU Parted to correct GPT errors.\n");
 546         return;
 547 }
 548
 549 /**
 550  * find_valid_gpt() - Search disk for valid GPT headers and PTEs
 551  * @state
 552  * @gpt is a GPT header ptr, filled on return.
 553  * @ptes is a PTEs ptr, filled on return.
 554  * Description: Returns 1 if valid, 0 on error.
 555  * If valid, returns pointers to newly allocated GPT header and PTEs.
 556  * Validity depends on PMBR being valid (or being overridden by the
 557  * 'gpt' kernel command line option) and finding either the Primary
 558  * GPT header and PTEs valid, or the Alternate GPT header and PTEs
 559  * valid.  If the Primary GPT header is not valid, the Alternate GPT header
 560  * is not checked unless the 'gpt' kernel command line option is passed.
 561  * This protects against devices which misreport their size, and forces
 562  * the user to decide to use the Alternate GPT.
 563  */
 564 static int find_valid_gpt(struct parsed_partitions *state, gpt_header **gpt,
 565                           gpt_entry **ptes)
 566 {
 567         int good_pgpt = 0, good_agpt = 0, good_pmbr = 0;
 568         gpt_header *pgpt = NULL, *agpt = NULL;
 569         gpt_entry *pptes = NULL, *aptes = NULL;
 570         legacy_mbr *legacymbr;
 571         u64 lastlba;
 572
 573         if (!ptes)
 574                 return 0;
 575
 576         lastlba = last_lba(state->bdev);
 577         if (!force_gpt) {
 578                 /* This will be added to the EFI Spec. per Intel after v1.02. */
 579                 legacymbr = kzalloc(sizeof(*legacymbr), GFP_KERNEL);
 580                 if (!legacymbr)
 581                         goto fail;
 582
 583                 read_lba(state, 0, (u8 *)legacymbr, sizeof(*legacymbr));
 584                 good_pmbr = is_pmbr_valid(legacymbr);
 585                 kfree(legacymbr);
 586
 587                 if (!good_pmbr)
 588                         goto fail;
 589
 590                 pr_debug("Device has a %s MBR\n",
 591                          good_pmbr == GPT_MBR_PROTECTIVE ?
 592                                                 "protective" : "hybrid");
 593         }
 594
 595         good_pgpt = is_gpt_valid(state, GPT_PRIMARY_PARTITION_TABLE_LBA,
 596                                  &pgpt, &pptes);
 597         if (good_pgpt)
 598                 good_agpt = is_gpt_valid(state,
 599                                          le64_to_cpu(pgpt->alternate_lba),
 600                                          &agpt, &aptes);
 601         if (!good_agpt && force_gpt)
 602                 good_agpt = is_gpt_valid(state, lastlba, &agpt, &aptes);
 603
 604         /* The obviously unsuccessful case */
 605         if (!good_pgpt && !good_agpt)
 606                 goto fail;
 607
 608         compare_gpts(pgpt, agpt, lastlba);
 609
 610         /* The good cases */
 611         if (good_pgpt) {
 612                 *gpt  = pgpt;
 613                 *ptes = pptes;
 614                 kfree(agpt);
 615                 kfree(aptes);
 616                 if (!good_agpt) {
 617                         printk(KERN_WARNING
 618                                "Alternate GPT is invalid, "
 619                                "using primary GPT.\n");
 620                 }
 621                 return 1;
 622         }
 623         else if (good_agpt) {
 624                 *gpt  = agpt;
 625                 *ptes = aptes;
 626                 kfree(pgpt);
 627                 kfree(pptes);
 628                 printk(KERN_WARNING
 629                        "Primary GPT is invalid, using alternate GPT.\n");
 630                 return 1;
 631         }
 632
 633  fail:
 634         kfree(pgpt);
 635         kfree(agpt);
 636         kfree(pptes);
 637         kfree(aptes);
 638         *gpt = NULL;
 639         *ptes = NULL;
 640         return 0;
 641 }
 642
 643 /**
 644  * efi_partition(struct parsed_partitions *state)
 645  * @state
 646  *
 647  * Description: called from check.c, if the disk contains GPT
 648  * partitions, sets up partition entries in the kernel.
 649  *
 650  * If the first block on the disk is a legacy MBR,
 651  * it will get handled by msdos_partition().
 652  * If it's a Protective MBR, we'll handle it here.
 653  *
 654  * We do not create a Linux partition for GPT, but
 655  * only for the actual data partitions.
 656  * Returns:
 657  * -1 if unable to read the partition table
 658  *  0 if this isn't our partition table
 659  *  1 if successful
 660  *
 661  */
 662 int efi_partition(struct parsed_partitions *state)
 663 {
 664         gpt_header *gpt = NULL;
 665         gpt_entry *ptes = NULL;
 666         u32 i;
 667         unsigned ssz = bdev_logical_block_size(state->bdev) / 512;
 668
 669         if (!find_valid_gpt(state, &gpt, &ptes) || !gpt || !ptes) {
 670                 kfree(gpt);
 671                 kfree(ptes);
 672                 return 0;
 673         }
 674
 675         pr_debug("GUID Partition Table is valid!  Yea!\n");
 676
 677         for (i = 0; i < le32_to_cpu(gpt->num_partition_entries) && i < state->limit-1; i++) {
 678                 struct partition_meta_info *info;
 679                 unsigned label_count = 0;
 680                 unsigned label_max;
 681                 u64 start = le64_to_cpu(ptes[i].starting_lba);
 682                 u64 size = le64_to_cpu(ptes[i].ending_lba) -
 683                            le64_to_cpu(ptes[i].starting_lba) + 1ULL;
 684
 685                 if (!is_pte_valid(&ptes[i], last_lba(state->bdev)))
 686                         continue;
 687
 688                 put_partition(state, i+1, start * ssz, size * ssz);
 689
 690                 /* If this is a RAID volume, tell md */
 691                 if (!efi_guidcmp(ptes[i].partition_type_guid,
 692                                  PARTITION_LINUX_RAID_GUID))
 693                         state->parts[i + 1].flags = ADDPART_FLAG_RAID;
 694
 695                 info = &state->parts[i + 1].info;
 696                 efi_guid_unparse(&ptes[i].unique_partition_guid, info->uuid);
 697
 698                 /* Naively convert UTF16-LE to 7 bits. */
 699                 label_max = min(sizeof(info->volname) - 1,
 700                                 sizeof(ptes[i].partition_name));
 701                 info->volname[label_max] = 0;
 702                 while (label_count < label_max) {
 703                         u8 c = ptes[i].partition_name[label_count] & 0xff;
 704                         if (c && !isprint(c))
 705                                 c = '!';
 706                         info->volname[label_count] = c;
 707                         label_count++;
 708                 }
 709                 state->parts[i + 1].has_info = true;
 710         }
 711         kfree(ptes);
 712         kfree(gpt);
 713         strlcat(state->pp_buf, "\n", PAGE_SIZE);
 714         return 1;
 715 }