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