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