blob: f1f3e5bceff74259a190391efaab502cd4486ea6 [file] [log] [blame]
Tom Rini83d290c2018-05-06 17:58:06 -04001// SPDX-License-Identifier: GPL-2.0+
richardretanubun07f3d782008-09-26 11:13:22 -04002/*
3 * Copyright (C) 2008 RuggedCom, Inc.
4 * Richard Retanubun <RichardRetanubun@RuggedCom.com>
richardretanubun07f3d782008-09-26 11:13:22 -04005 */
6
7/*
Steve Raee04350d2014-05-26 11:52:23 -07008 * NOTE:
9 * when CONFIG_SYS_64BIT_LBA is not defined, lbaint_t is 32 bits; this
Heinrich Schuchardt643b5e72020-09-17 17:57:21 +020010 * limits the maximum size of addressable storage to < 2 tebibytes
richardretanubun07f3d782008-09-26 11:13:22 -040011 */
Simon Glasse6f6f9e2020-05-10 11:39:58 -060012#include <common.h>
13#include <blk.h>
Simon Glassf7ae49f2020-05-10 11:40:05 -060014#include <log.h>
Simon Glasse6f6f9e2020-05-10 11:39:58 -060015#include <part.h>
Simon Glassba06b3c2020-05-10 11:39:52 -060016#include <uuid.h>
Simon Glass90526e92020-05-10 11:39:56 -060017#include <asm/cache.h>
Simon Glass401d1c42020-10-30 21:38:53 -060018#include <asm/global_data.h>
Marc Dietrich8faefad2013-03-29 07:57:10 +000019#include <asm/unaligned.h>
richardretanubun07f3d782008-09-26 11:13:22 -040020#include <command.h>
Philipp Tomsich02e43532017-03-01 21:10:39 +010021#include <fdtdec.h>
richardretanubun07f3d782008-09-26 11:13:22 -040022#include <ide.h>
23#include <malloc.h>
Simon Glasscf92e052015-09-02 17:24:58 -060024#include <memalign.h>
Chang Hyun Parkfae2bf22012-12-11 11:09:45 +010025#include <part_efi.h>
Simon Glass7de8bd02021-08-07 07:24:01 -060026#include <dm/ofnode.h>
Philipp Tomsich02e43532017-03-01 21:10:39 +010027#include <linux/compiler.h>
Lei Wen6eecc032011-09-07 18:11:19 +000028#include <linux/ctype.h>
Simon Glass3db71102019-11-14 12:57:16 -070029#include <u-boot/crc.h>
richardretanubun07f3d782008-09-26 11:13:22 -040030
Adam Ford1811a922018-02-06 12:43:56 -060031#ifdef CONFIG_HAVE_BLOCK_DEVICE
Simon Glassa594b412021-07-02 12:36:14 -060032
33/* GUID for basic data partitons */
34#if CONFIG_IS_ENABLED(EFI_PARTITION)
35static const efi_guid_t partition_basic_data_guid = PARTITION_BASIC_DATA_GUID;
36#endif
37
richardretanubun07f3d782008-09-26 11:13:22 -040038/**
39 * efi_crc32() - EFI version of crc32 function
40 * @buf: buffer to calculate crc32 of
41 * @len - length of buf
42 *
43 * Description: Returns EFI-style CRC32 value for @buf
44 */
Chang Hyun Parkfae2bf22012-12-11 11:09:45 +010045static inline u32 efi_crc32(const void *buf, u32 len)
richardretanubun07f3d782008-09-26 11:13:22 -040046{
47 return crc32(0, buf, len);
48}
49
50/*
51 * Private function prototypes
52 */
53
54static int pmbr_part_valid(struct partition *part);
55static int is_pmbr_valid(legacy_mbr * mbr);
Simon Glass4101f682016-02-29 15:25:34 -070056static int is_gpt_valid(struct blk_desc *dev_desc, u64 lba,
Steve Raee04350d2014-05-26 11:52:23 -070057 gpt_header *pgpt_head, gpt_entry **pgpt_pte);
Simon Glass4101f682016-02-29 15:25:34 -070058static gpt_entry *alloc_read_gpt_entries(struct blk_desc *dev_desc,
59 gpt_header *pgpt_head);
richardretanubun07f3d782008-09-26 11:13:22 -040060static int is_pte_valid(gpt_entry * pte);
Urja Rannikko20c568c2019-04-11 20:27:50 +000061static int find_valid_gpt(struct blk_desc *dev_desc, gpt_header *gpt_head,
62 gpt_entry **pgpt_pte);
richardretanubun07f3d782008-09-26 11:13:22 -040063
Lei Wen6eecc032011-09-07 18:11:19 +000064static char *print_efiname(gpt_entry *pte)
65{
66 static char name[PARTNAME_SZ + 1];
67 int i;
68 for (i = 0; i < PARTNAME_SZ; i++) {
69 u8 c;
70 c = pte->partition_name[i] & 0xff;
71 c = (c && !isprint(c)) ? '.' : c;
72 name[i] = c;
73 }
74 name[PARTNAME_SZ] = 0;
75 return name;
76}
77
Heinrich Schuchardt7de24b22019-01-12 11:25:25 +010078static const efi_guid_t system_guid = PARTITION_SYSTEM_GUID;
Stephen Warrenb4414f42012-10-08 08:14:37 +000079
Heinrich Schuchardt25801ac2020-03-19 13:49:34 +010080static int get_bootable(gpt_entry *p)
Stephen Warrenb4414f42012-10-08 08:14:37 +000081{
Heinrich Schuchardt25801ac2020-03-19 13:49:34 +010082 int ret = 0;
83
84 if (!memcmp(&p->partition_type_guid, &system_guid, sizeof(efi_guid_t)))
85 ret |= PART_EFI_SYSTEM_PARTITION;
86 if (p->attributes.fields.legacy_bios_bootable)
87 ret |= PART_BOOTABLE;
88 return ret;
Stephen Warrenb4414f42012-10-08 08:14:37 +000089}
90
Steve Raee1f6b0a2014-12-18 12:13:42 +010091static int validate_gpt_header(gpt_header *gpt_h, lbaint_t lba,
92 lbaint_t lastlba)
93{
94 uint32_t crc32_backup = 0;
95 uint32_t calc_crc32;
96
97 /* Check the GPT header signature */
Simon Glass67b90522018-10-01 12:22:35 -060098 if (le64_to_cpu(gpt_h->signature) != GPT_HEADER_SIGNATURE_UBOOT) {
Steve Raee1f6b0a2014-12-18 12:13:42 +010099 printf("%s signature is wrong: 0x%llX != 0x%llX\n",
100 "GUID Partition Table Header",
101 le64_to_cpu(gpt_h->signature),
Simon Glass67b90522018-10-01 12:22:35 -0600102 GPT_HEADER_SIGNATURE_UBOOT);
Steve Raee1f6b0a2014-12-18 12:13:42 +0100103 return -1;
104 }
105
106 /* Check the GUID Partition Table CRC */
107 memcpy(&crc32_backup, &gpt_h->header_crc32, sizeof(crc32_backup));
108 memset(&gpt_h->header_crc32, 0, sizeof(gpt_h->header_crc32));
109
110 calc_crc32 = efi_crc32((const unsigned char *)gpt_h,
111 le32_to_cpu(gpt_h->header_size));
112
113 memcpy(&gpt_h->header_crc32, &crc32_backup, sizeof(crc32_backup));
114
115 if (calc_crc32 != le32_to_cpu(crc32_backup)) {
116 printf("%s CRC is wrong: 0x%x != 0x%x\n",
117 "GUID Partition Table Header",
118 le32_to_cpu(crc32_backup), calc_crc32);
119 return -1;
120 }
121
122 /*
123 * Check that the my_lba entry points to the LBA that contains the GPT
124 */
125 if (le64_to_cpu(gpt_h->my_lba) != lba) {
126 printf("GPT: my_lba incorrect: %llX != " LBAF "\n",
127 le64_to_cpu(gpt_h->my_lba),
128 lba);
129 return -1;
130 }
131
132 /*
133 * Check that the first_usable_lba and that the last_usable_lba are
134 * within the disk.
135 */
136 if (le64_to_cpu(gpt_h->first_usable_lba) > lastlba) {
137 printf("GPT: first_usable_lba incorrect: %llX > " LBAF "\n",
138 le64_to_cpu(gpt_h->first_usable_lba), lastlba);
139 return -1;
140 }
141 if (le64_to_cpu(gpt_h->last_usable_lba) > lastlba) {
142 printf("GPT: last_usable_lba incorrect: %llX > " LBAF "\n",
143 le64_to_cpu(gpt_h->last_usable_lba), lastlba);
144 return -1;
145 }
146
147 debug("GPT: first_usable_lba: %llX last_usable_lba: %llX last lba: "
148 LBAF "\n", le64_to_cpu(gpt_h->first_usable_lba),
149 le64_to_cpu(gpt_h->last_usable_lba), lastlba);
150
151 return 0;
152}
153
154static int validate_gpt_entries(gpt_header *gpt_h, gpt_entry *gpt_e)
155{
156 uint32_t calc_crc32;
157
158 /* Check the GUID Partition Table Entry Array CRC */
159 calc_crc32 = efi_crc32((const unsigned char *)gpt_e,
160 le32_to_cpu(gpt_h->num_partition_entries) *
161 le32_to_cpu(gpt_h->sizeof_partition_entry));
162
163 if (calc_crc32 != le32_to_cpu(gpt_h->partition_entry_array_crc32)) {
164 printf("%s: 0x%x != 0x%x\n",
165 "GUID Partition Table Entry Array CRC is wrong",
166 le32_to_cpu(gpt_h->partition_entry_array_crc32),
167 calc_crc32);
168 return -1;
169 }
170
171 return 0;
172}
173
174static void prepare_backup_gpt_header(gpt_header *gpt_h)
175{
176 uint32_t calc_crc32;
177 uint64_t val;
178
179 /* recalculate the values for the Backup GPT Header */
180 val = le64_to_cpu(gpt_h->my_lba);
181 gpt_h->my_lba = gpt_h->alternate_lba;
182 gpt_h->alternate_lba = cpu_to_le64(val);
Steve Rae0ff7e582014-12-12 15:51:54 -0800183 gpt_h->partition_entry_lba =
184 cpu_to_le64(le64_to_cpu(gpt_h->last_usable_lba) + 1);
Steve Raee1f6b0a2014-12-18 12:13:42 +0100185 gpt_h->header_crc32 = 0;
186
187 calc_crc32 = efi_crc32((const unsigned char *)gpt_h,
188 le32_to_cpu(gpt_h->header_size));
189 gpt_h->header_crc32 = cpu_to_le32(calc_crc32);
190}
191
Patrick Delaunaybd42a942017-01-27 11:00:41 +0100192#if CONFIG_IS_ENABLED(EFI_PARTITION)
Stephen Warrenf07cd2c2012-10-08 08:14:34 +0000193/*
194 * Public Functions (include/part.h)
195 */
196
Alison Chaiken73d6d182017-06-25 16:43:23 -0700197/*
198 * UUID is displayed as 32 hexadecimal digits, in 5 groups,
199 * separated by hyphens, in the form 8-4-4-4-12 for a total of 36 characters
200 */
201int get_disk_guid(struct blk_desc * dev_desc, char *guid)
202{
203 ALLOC_CACHE_ALIGN_BUFFER_PAD(gpt_header, gpt_head, 1, dev_desc->blksz);
204 gpt_entry *gpt_pte = NULL;
205 unsigned char *guid_bin;
206
207 /* This function validates AND fills in the GPT header and PTE */
Urja Rannikko20c568c2019-04-11 20:27:50 +0000208 if (find_valid_gpt(dev_desc, gpt_head, &gpt_pte) != 1)
209 return -EINVAL;
Alison Chaiken73d6d182017-06-25 16:43:23 -0700210
211 guid_bin = gpt_head->disk_guid.b;
212 uuid_bin_to_str(guid_bin, guid, UUID_STR_FORMAT_GUID);
213
Eugeniu Rosca1cfe9692019-04-30 04:53:44 +0200214 /* Remember to free pte */
215 free(gpt_pte);
Alison Chaiken73d6d182017-06-25 16:43:23 -0700216 return 0;
217}
218
Simon Glass084bf4c2016-02-29 15:26:04 -0700219void part_print_efi(struct blk_desc *dev_desc)
Stephen Warrenf07cd2c2012-10-08 08:14:34 +0000220{
Egbert Eichae1768a2013-04-09 06:03:36 +0000221 ALLOC_CACHE_ALIGN_BUFFER_PAD(gpt_header, gpt_head, 1, dev_desc->blksz);
Stephen Warrenf07cd2c2012-10-08 08:14:34 +0000222 gpt_entry *gpt_pte = NULL;
223 int i = 0;
Heinrich Schuchardta4492ee2022-01-16 12:23:19 +0100224 unsigned char *uuid;
Stephen Warrenf07cd2c2012-10-08 08:14:34 +0000225
Stephen Warrenf07cd2c2012-10-08 08:14:34 +0000226 /* This function validates AND fills in the GPT header and PTE */
Urja Rannikko20c568c2019-04-11 20:27:50 +0000227 if (find_valid_gpt(dev_desc, gpt_head, &gpt_pte) != 1)
228 return;
Stephen Warrenf07cd2c2012-10-08 08:14:34 +0000229
230 debug("%s: gpt-entry at %p\n", __func__, gpt_pte);
231
232 printf("Part\tStart LBA\tEnd LBA\t\tName\n");
Stephen Warren13bf2f52012-10-08 08:14:36 +0000233 printf("\tAttributes\n");
Przemyslaw Marczakd718ded2014-04-02 10:20:03 +0200234 printf("\tType GUID\n");
235 printf("\tPartition GUID\n");
Stephen Warrenf07cd2c2012-10-08 08:14:34 +0000236
Chang Hyun Parkfae2bf22012-12-11 11:09:45 +0100237 for (i = 0; i < le32_to_cpu(gpt_head->num_partition_entries); i++) {
Heinrich Schuchardt504dbd22022-01-11 15:43:05 +0100238 /* Skip invalid PTE */
Stephen Warrenf07cd2c2012-10-08 08:14:34 +0000239 if (!is_pte_valid(&gpt_pte[i]))
Heinrich Schuchardt504dbd22022-01-11 15:43:05 +0100240 continue;
Stephen Warrenf07cd2c2012-10-08 08:14:34 +0000241
242 printf("%3d\t0x%08llx\t0x%08llx\t\"%s\"\n", (i + 1),
Chang Hyun Parkfae2bf22012-12-11 11:09:45 +0100243 le64_to_cpu(gpt_pte[i].starting_lba),
244 le64_to_cpu(gpt_pte[i].ending_lba),
Stephen Warrenf07cd2c2012-10-08 08:14:34 +0000245 print_efiname(&gpt_pte[i]));
Stephen Warren13bf2f52012-10-08 08:14:36 +0000246 printf("\tattrs:\t0x%016llx\n", gpt_pte[i].attributes.raw);
Heinrich Schuchardta4492ee2022-01-16 12:23:19 +0100247 uuid = (unsigned char *)gpt_pte[i].partition_type_guid.b;
248 if (CONFIG_IS_ENABLED(PARTITION_TYPE_GUID))
249 printf("\ttype:\t%pUl\n\t\t(%pUs)\n", uuid, uuid);
250 else
251 printf("\ttype:\t%pUl\n", uuid);
252 uuid = (unsigned char *)gpt_pte[i].unique_partition_guid.b;
253 printf("\tguid:\t%pUl\n", uuid);
Stephen Warrenf07cd2c2012-10-08 08:14:34 +0000254 }
255
256 /* Remember to free pte */
257 free(gpt_pte);
258 return;
259}
Stephen Warren894bfbb2012-09-21 09:50:59 +0000260
Simon Glass3e8bd462016-02-29 15:25:48 -0700261int part_get_info_efi(struct blk_desc *dev_desc, int part,
Simon Glass05289792020-05-10 11:39:57 -0600262 struct disk_partition *info)
richardretanubun07f3d782008-09-26 11:13:22 -0400263{
Egbert Eichae1768a2013-04-09 06:03:36 +0000264 ALLOC_CACHE_ALIGN_BUFFER_PAD(gpt_header, gpt_head, 1, dev_desc->blksz);
Doug Andersondeb5ca82011-10-19 09:47:31 +0000265 gpt_entry *gpt_pte = NULL;
richardretanubun07f3d782008-09-26 11:13:22 -0400266
267 /* "part" argument must be at least 1 */
Simon Glass4708a072016-03-16 07:45:36 -0600268 if (part < 1) {
Doug Andersondf70b1c2011-10-19 08:04:46 +0000269 printf("%s: Invalid Argument(s)\n", __func__);
richardretanubun07f3d782008-09-26 11:13:22 -0400270 return -1;
271 }
272
273 /* This function validates AND fills in the GPT header and PTE */
Urja Rannikko20c568c2019-04-11 20:27:50 +0000274 if (find_valid_gpt(dev_desc, gpt_head, &gpt_pte) != 1)
275 return -1;
richardretanubun07f3d782008-09-26 11:13:22 -0400276
Chang Hyun Parkfae2bf22012-12-11 11:09:45 +0100277 if (part > le32_to_cpu(gpt_head->num_partition_entries) ||
Stephen Warrenc04d68c2012-09-21 09:50:58 +0000278 !is_pte_valid(&gpt_pte[part - 1])) {
Mark Langsdorf6d2ee5a2013-09-10 15:14:56 -0500279 debug("%s: *** ERROR: Invalid partition number %d ***\n",
Stephen Warrenc04d68c2012-09-21 09:50:58 +0000280 __func__, part);
Mark Langsdorf6d2ee5a2013-09-10 15:14:56 -0500281 free(gpt_pte);
Stephen Warrenc04d68c2012-09-21 09:50:58 +0000282 return -1;
283 }
284
Steve Raee04350d2014-05-26 11:52:23 -0700285 /* The 'lbaint_t' casting may limit the maximum disk size to 2 TB */
286 info->start = (lbaint_t)le64_to_cpu(gpt_pte[part - 1].starting_lba);
Richard Retanubun50970832009-01-26 08:45:14 -0500287 /* The ending LBA is inclusive, to calculate size, add 1 to it */
Steve Raee04350d2014-05-26 11:52:23 -0700288 info->size = (lbaint_t)le64_to_cpu(gpt_pte[part - 1].ending_lba) + 1
Richard Retanubun50970832009-01-26 08:45:14 -0500289 - info->start;
Egbert Eichae1768a2013-04-09 06:03:36 +0000290 info->blksz = dev_desc->blksz;
richardretanubun07f3d782008-09-26 11:13:22 -0400291
Heinrich Schuchardt5375ee52019-07-05 21:27:13 +0200292 snprintf((char *)info->name, sizeof(info->name), "%s",
293 print_efiname(&gpt_pte[part - 1]));
Ben Whitten192bc692015-12-30 13:05:58 +0000294 strcpy((char *)info->type, "U-Boot");
Heinrich Schuchardt25801ac2020-03-19 13:49:34 +0100295 info->bootable = get_bootable(&gpt_pte[part - 1]);
Patrick Delaunayb331cd62017-01-27 11:00:42 +0100296#if CONFIG_IS_ENABLED(PARTITION_UUIDS)
Przemyslaw Marczakd718ded2014-04-02 10:20:03 +0200297 uuid_bin_to_str(gpt_pte[part - 1].unique_partition_guid.b, info->uuid,
298 UUID_STR_FORMAT_GUID);
Stephen Warren894bfbb2012-09-21 09:50:59 +0000299#endif
Patrick Delaunay7561b252015-10-27 11:00:27 +0100300#ifdef CONFIG_PARTITION_TYPE_GUID
301 uuid_bin_to_str(gpt_pte[part - 1].partition_type_guid.b,
302 info->type_guid, UUID_STR_FORMAT_GUID);
303#endif
richardretanubun07f3d782008-09-26 11:13:22 -0400304
Steve Rae60bf9412014-05-26 11:52:24 -0700305 debug("%s: start 0x" LBAF ", size 0x" LBAF ", name %s\n", __func__,
Frederic Leroy04735e92013-06-26 18:11:25 +0200306 info->start, info->size, info->name);
richardretanubun07f3d782008-09-26 11:13:22 -0400307
308 /* Remember to free pte */
Doug Andersondeb5ca82011-10-19 09:47:31 +0000309 free(gpt_pte);
richardretanubun07f3d782008-09-26 11:13:22 -0400310 return 0;
311}
312
Simon Glass084bf4c2016-02-29 15:26:04 -0700313static int part_test_efi(struct blk_desc *dev_desc)
richardretanubun07f3d782008-09-26 11:13:22 -0400314{
Egbert Eichae1768a2013-04-09 06:03:36 +0000315 ALLOC_CACHE_ALIGN_BUFFER_PAD(legacy_mbr, legacymbr, 1, dev_desc->blksz);
richardretanubun07f3d782008-09-26 11:13:22 -0400316
317 /* Read legacy MBR from block 0 and validate it */
Simon Glass2a981dc2016-02-29 15:25:52 -0700318 if ((blk_dread(dev_desc, 0, 1, (ulong *)legacymbr) != 1)
Anton staaff75dd582011-10-12 13:56:04 +0000319 || (is_pmbr_valid(legacymbr) != 1)) {
richardretanubun07f3d782008-09-26 11:13:22 -0400320 return -1;
321 }
322 return 0;
323}
324
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100325/**
326 * set_protective_mbr(): Set the EFI protective MBR
327 * @param dev_desc - block device descriptor
328 *
Heinrich Schuchardt185f8122022-01-19 18:05:50 +0100329 * Return: - zero on success, otherwise error
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100330 */
Simon Glass4101f682016-02-29 15:25:34 -0700331static int set_protective_mbr(struct blk_desc *dev_desc)
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100332{
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100333 /* Setup the Protective MBR */
Patrick Delaunay3cc56612017-11-17 10:08:18 +0100334 ALLOC_CACHE_ALIGN_BUFFER_PAD(legacy_mbr, p_mbr, 1, dev_desc->blksz);
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100335 if (p_mbr == NULL) {
336 printf("%s: calloc failed!\n", __func__);
337 return -1;
338 }
Vincent Tinellie163a932017-01-30 15:46:07 +0300339
340 /* Read MBR to backup boot code if it exists */
341 if (blk_dread(dev_desc, 0, 1, p_mbr) != 1) {
Masahiro Yamada9b643e32017-09-16 14:10:41 +0900342 pr_err("** Can't read from device %d **\n", dev_desc->devnum);
Vincent Tinellie163a932017-01-30 15:46:07 +0300343 return -1;
344 }
345
Sam Protsenko955575c2018-05-22 02:04:21 +0300346 /* Clear all data in MBR except of backed up boot code */
347 memset((char *)p_mbr + MSDOS_MBR_BOOT_CODE_SIZE, 0, sizeof(*p_mbr) -
348 MSDOS_MBR_BOOT_CODE_SIZE);
349
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100350 /* Append signature */
351 p_mbr->signature = MSDOS_MBR_SIGNATURE;
352 p_mbr->partition_record[0].sys_ind = EFI_PMBR_OSTYPE_EFI_GPT;
353 p_mbr->partition_record[0].start_sect = 1;
Maxime Ripardb349abb2015-01-08 12:26:44 +0100354 p_mbr->partition_record[0].nr_sects = (u32) dev_desc->lba - 1;
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100355
356 /* Write MBR sector to the MMC device */
Simon Glass2a981dc2016-02-29 15:25:52 -0700357 if (blk_dwrite(dev_desc, 0, 1, p_mbr) != 1) {
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100358 printf("** Can't write to device %d **\n",
Simon Glassbcce53d2016-02-29 15:25:51 -0700359 dev_desc->devnum);
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100360 return -1;
361 }
362
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100363 return 0;
364}
365
Simon Glass4101f682016-02-29 15:25:34 -0700366int write_gpt_table(struct blk_desc *dev_desc,
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100367 gpt_header *gpt_h, gpt_entry *gpt_e)
368{
Egbert Eichae1768a2013-04-09 06:03:36 +0000369 const int pte_blk_cnt = BLOCK_CNT((gpt_h->num_partition_entries
370 * sizeof(gpt_entry)), dev_desc);
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100371 u32 calc_crc32;
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100372
373 debug("max lba: %x\n", (u32) dev_desc->lba);
374 /* Setup the Protective MBR */
375 if (set_protective_mbr(dev_desc) < 0)
376 goto err;
377
378 /* Generate CRC for the Primary GPT Header */
379 calc_crc32 = efi_crc32((const unsigned char *)gpt_e,
380 le32_to_cpu(gpt_h->num_partition_entries) *
381 le32_to_cpu(gpt_h->sizeof_partition_entry));
382 gpt_h->partition_entry_array_crc32 = cpu_to_le32(calc_crc32);
383
384 calc_crc32 = efi_crc32((const unsigned char *)gpt_h,
385 le32_to_cpu(gpt_h->header_size));
386 gpt_h->header_crc32 = cpu_to_le32(calc_crc32);
387
388 /* Write the First GPT to the block right after the Legacy MBR */
Simon Glass2a981dc2016-02-29 15:25:52 -0700389 if (blk_dwrite(dev_desc, 1, 1, gpt_h) != 1)
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100390 goto err;
391
Philipp Tomsich02e43532017-03-01 21:10:39 +0100392 if (blk_dwrite(dev_desc, le64_to_cpu(gpt_h->partition_entry_lba),
393 pte_blk_cnt, gpt_e) != pte_blk_cnt)
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100394 goto err;
395
Steve Raee1f6b0a2014-12-18 12:13:42 +0100396 prepare_backup_gpt_header(gpt_h);
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100397
Simon Glass2a981dc2016-02-29 15:25:52 -0700398 if (blk_dwrite(dev_desc, (lbaint_t)le64_to_cpu(gpt_h->last_usable_lba)
399 + 1, pte_blk_cnt, gpt_e) != pte_blk_cnt)
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100400 goto err;
401
Simon Glass2a981dc2016-02-29 15:25:52 -0700402 if (blk_dwrite(dev_desc, (lbaint_t)le64_to_cpu(gpt_h->my_lba), 1,
403 gpt_h) != 1)
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100404 goto err;
405
406 debug("GPT successfully written to block device!\n");
407 return 0;
408
409 err:
Simon Glassbcce53d2016-02-29 15:25:51 -0700410 printf("** Can't write to device %d **\n", dev_desc->devnum);
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100411 return -1;
412}
413
Maxime Ripard47d7ee42017-08-23 16:01:32 +0200414int gpt_fill_pte(struct blk_desc *dev_desc,
415 gpt_header *gpt_h, gpt_entry *gpt_e,
Simon Glass05289792020-05-10 11:39:57 -0600416 struct disk_partition *partitions, int parts)
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100417{
Steve Raee04350d2014-05-26 11:52:23 -0700418 lbaint_t offset = (lbaint_t)le64_to_cpu(gpt_h->first_usable_lba);
Steve Raee04350d2014-05-26 11:52:23 -0700419 lbaint_t last_usable_lba = (lbaint_t)
420 le64_to_cpu(gpt_h->last_usable_lba);
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100421 int i, k;
Marek Vasut67cd4a62013-05-19 12:53:34 +0000422 size_t efiname_len, dosname_len;
Patrick Delaunayb331cd62017-01-27 11:00:42 +0100423#if CONFIG_IS_ENABLED(PARTITION_UUIDS)
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100424 char *str_uuid;
Przemyslaw Marczaka96a0e62014-04-02 10:20:02 +0200425 unsigned char *bin_uuid;
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100426#endif
Patrick Delaunay7561b252015-10-27 11:00:27 +0100427#ifdef CONFIG_PARTITION_TYPE_GUID
428 char *str_type_guid;
429 unsigned char *bin_type_guid;
430#endif
Maxime Ripard79c59122017-08-23 16:01:33 +0200431 size_t hdr_start = gpt_h->my_lba;
432 size_t hdr_end = hdr_start + 1;
433
434 size_t pte_start = gpt_h->partition_entry_lba;
435 size_t pte_end = pte_start +
436 gpt_h->num_partition_entries * gpt_h->sizeof_partition_entry /
437 dev_desc->blksz;
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100438
439 for (i = 0; i < parts; i++) {
440 /* partition starting lba */
Maxime Ripard5276e8b2017-08-23 16:01:31 +0200441 lbaint_t start = partitions[i].start;
442 lbaint_t size = partitions[i].size;
443
Maxime Ripard79c59122017-08-23 16:01:33 +0200444 if (start) {
445 offset = start + size;
446 } else {
447 start = offset;
448 offset += size;
449 }
450
451 /*
452 * If our partition overlaps with either the GPT
453 * header, or the partition entry, reject it.
454 */
Patrick Delaunayae0e0222017-10-18 15:11:05 +0200455 if (((start < hdr_end && hdr_start < (start + size)) ||
456 (start < pte_end && pte_start < (start + size)))) {
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100457 printf("Partition overlap\n");
458 return -1;
459 }
Maxime Ripard5276e8b2017-08-23 16:01:31 +0200460
Maxime Ripard79c59122017-08-23 16:01:33 +0200461 gpt_e[i].starting_lba = cpu_to_le64(start);
462
Patrick Delaunaya5653862016-05-02 14:43:33 +0200463 if (offset > (last_usable_lba + 1)) {
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100464 printf("Partitions layout exceds disk size\n");
465 return -1;
466 }
467 /* partition ending lba */
Maxime Ripard5276e8b2017-08-23 16:01:31 +0200468 if ((i == parts - 1) && (size == 0))
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100469 /* extend the last partition to maximuim */
470 gpt_e[i].ending_lba = gpt_h->last_usable_lba;
471 else
472 gpt_e[i].ending_lba = cpu_to_le64(offset - 1);
473
Patrick Delaunay7561b252015-10-27 11:00:27 +0100474#ifdef CONFIG_PARTITION_TYPE_GUID
475 str_type_guid = partitions[i].type_guid;
476 bin_type_guid = gpt_e[i].partition_type_guid.b;
477 if (strlen(str_type_guid)) {
478 if (uuid_str_to_bin(str_type_guid, bin_type_guid,
479 UUID_STR_FORMAT_GUID)) {
480 printf("Partition no. %d: invalid type guid: %s\n",
481 i, str_type_guid);
482 return -1;
483 }
484 } else {
485 /* default partition type GUID */
486 memcpy(bin_type_guid,
Heinrich Schuchardt44ab2d32018-06-09 17:50:18 +0200487 &partition_basic_data_guid, 16);
Patrick Delaunay7561b252015-10-27 11:00:27 +0100488 }
489#else
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100490 /* partition type GUID */
491 memcpy(gpt_e[i].partition_type_guid.b,
Heinrich Schuchardt44ab2d32018-06-09 17:50:18 +0200492 &partition_basic_data_guid, 16);
Patrick Delaunay7561b252015-10-27 11:00:27 +0100493#endif
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100494
Patrick Delaunayb331cd62017-01-27 11:00:42 +0100495#if CONFIG_IS_ENABLED(PARTITION_UUIDS)
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100496 str_uuid = partitions[i].uuid;
Przemyslaw Marczaka96a0e62014-04-02 10:20:02 +0200497 bin_uuid = gpt_e[i].unique_partition_guid.b;
498
Vincent Tinelli9da52f82017-02-27 16:11:15 +0200499 if (uuid_str_to_bin(str_uuid, bin_uuid, UUID_STR_FORMAT_GUID)) {
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100500 printf("Partition no. %d: invalid guid: %s\n",
501 i, str_uuid);
502 return -1;
503 }
504#endif
505
506 /* partition attributes */
507 memset(&gpt_e[i].attributes, 0,
508 sizeof(gpt_entry_attributes));
509
Heinrich Schuchardt25801ac2020-03-19 13:49:34 +0100510 if (partitions[i].bootable & PART_BOOTABLE)
Patrick Delaunaycfdaf4c2015-11-17 11:36:52 +0100511 gpt_e[i].attributes.fields.legacy_bios_bootable = 1;
512
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100513 /* partition name */
Marek Vasut67cd4a62013-05-19 12:53:34 +0000514 efiname_len = sizeof(gpt_e[i].partition_name)
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100515 / sizeof(efi_char16_t);
Marek Vasut67cd4a62013-05-19 12:53:34 +0000516 dosname_len = sizeof(partitions[i].name);
517
518 memset(gpt_e[i].partition_name, 0,
519 sizeof(gpt_e[i].partition_name));
520
521 for (k = 0; k < min(dosname_len, efiname_len); k++)
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100522 gpt_e[i].partition_name[k] =
523 (efi_char16_t)(partitions[i].name[k]);
524
Steve Raee04350d2014-05-26 11:52:23 -0700525 debug("%s: name: %s offset[%d]: 0x" LBAF
526 " size[%d]: 0x" LBAF "\n",
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100527 __func__, partitions[i].name, i,
Maxime Ripard5276e8b2017-08-23 16:01:31 +0200528 offset, i, size);
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100529 }
530
531 return 0;
532}
533
Philipp Tomsich02e43532017-03-01 21:10:39 +0100534static uint32_t partition_entries_offset(struct blk_desc *dev_desc)
535{
536 uint32_t offset_blks = 2;
Maxime Ripard89d33a22017-08-23 16:01:30 +0200537 uint32_t __maybe_unused offset_bytes;
Philipp Tomsich02e43532017-03-01 21:10:39 +0100538 int __maybe_unused config_offset;
539
540#if defined(CONFIG_EFI_PARTITION_ENTRIES_OFF)
541 /*
542 * Some architectures require their SPL loader at a fixed
543 * address within the first 16KB of the disk. To avoid an
544 * overlap with the partition entries of the EFI partition
545 * table, the first safe offset (in bytes, from the start of
546 * the disk) for the entries can be set in
547 * CONFIG_EFI_PARTITION_ENTRIES_OFF.
548 */
Maxime Ripard89d33a22017-08-23 16:01:30 +0200549 offset_bytes =
Philipp Tomsich02e43532017-03-01 21:10:39 +0100550 PAD_TO_BLOCKSIZE(CONFIG_EFI_PARTITION_ENTRIES_OFF, dev_desc);
Maxime Ripard89d33a22017-08-23 16:01:30 +0200551 offset_blks = offset_bytes / dev_desc->blksz;
Philipp Tomsich02e43532017-03-01 21:10:39 +0100552#endif
553
554#if defined(CONFIG_OF_CONTROL)
555 /*
556 * Allow the offset of the first partition entires (in bytes
557 * from the start of the device) to be specified as a property
558 * of the device tree '/config' node.
559 */
Simon Glass7de8bd02021-08-07 07:24:01 -0600560 config_offset = ofnode_conf_read_int(
561 "u-boot,efi-partition-entries-offset", -EINVAL);
Maxime Ripard89d33a22017-08-23 16:01:30 +0200562 if (config_offset != -EINVAL) {
563 offset_bytes = PAD_TO_BLOCKSIZE(config_offset, dev_desc);
564 offset_blks = offset_bytes / dev_desc->blksz;
565 }
Philipp Tomsich02e43532017-03-01 21:10:39 +0100566#endif
567
568 debug("efi: partition entries offset (in blocks): %d\n", offset_blks);
569
570 /*
571 * The earliest LBA this can be at is LBA#2 (i.e. right behind
572 * the (protective) MBR and the GPT header.
573 */
574 if (offset_blks < 2)
575 offset_blks = 2;
576
577 return offset_blks;
578}
579
Simon Glass4101f682016-02-29 15:25:34 -0700580int gpt_fill_header(struct blk_desc *dev_desc, gpt_header *gpt_h,
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100581 char *str_guid, int parts_count)
582{
Simon Glass67b90522018-10-01 12:22:35 -0600583 gpt_h->signature = cpu_to_le64(GPT_HEADER_SIGNATURE_UBOOT);
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100584 gpt_h->revision = cpu_to_le32(GPT_HEADER_REVISION_V1);
585 gpt_h->header_size = cpu_to_le32(sizeof(gpt_header));
586 gpt_h->my_lba = cpu_to_le64(1);
587 gpt_h->alternate_lba = cpu_to_le64(dev_desc->lba - 1);
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100588 gpt_h->last_usable_lba = cpu_to_le64(dev_desc->lba - 34);
Philipp Tomsich02e43532017-03-01 21:10:39 +0100589 gpt_h->partition_entry_lba =
590 cpu_to_le64(partition_entries_offset(dev_desc));
591 gpt_h->first_usable_lba =
592 cpu_to_le64(le64_to_cpu(gpt_h->partition_entry_lba) + 32);
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100593 gpt_h->num_partition_entries = cpu_to_le32(GPT_ENTRY_NUMBERS);
594 gpt_h->sizeof_partition_entry = cpu_to_le32(sizeof(gpt_entry));
595 gpt_h->header_crc32 = 0;
596 gpt_h->partition_entry_array_crc32 = 0;
597
Przemyslaw Marczakd718ded2014-04-02 10:20:03 +0200598 if (uuid_str_to_bin(str_guid, gpt_h->disk_guid.b, UUID_STR_FORMAT_GUID))
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100599 return -1;
600
601 return 0;
602}
603
Simon Glass4101f682016-02-29 15:25:34 -0700604int gpt_restore(struct blk_desc *dev_desc, char *str_disk_guid,
Simon Glass05289792020-05-10 11:39:57 -0600605 struct disk_partition *partitions, int parts_count)
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100606{
Lukasz Majewskibb021012017-10-27 12:28:10 +0200607 gpt_header *gpt_h;
Egbert Eichae1768a2013-04-09 06:03:36 +0000608 gpt_entry *gpt_e;
Lukasz Majewskibb021012017-10-27 12:28:10 +0200609 int ret, size;
Egbert Eichae1768a2013-04-09 06:03:36 +0000610
Lukasz Majewskibb021012017-10-27 12:28:10 +0200611 size = PAD_TO_BLOCKSIZE(sizeof(gpt_header), dev_desc);
612 gpt_h = malloc_cache_aligned(size);
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100613 if (gpt_h == NULL) {
614 printf("%s: calloc failed!\n", __func__);
615 return -1;
616 }
Lukasz Majewskibb021012017-10-27 12:28:10 +0200617 memset(gpt_h, 0, size);
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100618
Lukasz Majewskibb021012017-10-27 12:28:10 +0200619 size = PAD_TO_BLOCKSIZE(GPT_ENTRY_NUMBERS * sizeof(gpt_entry),
620 dev_desc);
621 gpt_e = malloc_cache_aligned(size);
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100622 if (gpt_e == NULL) {
623 printf("%s: calloc failed!\n", __func__);
624 free(gpt_h);
625 return -1;
626 }
Lukasz Majewskibb021012017-10-27 12:28:10 +0200627 memset(gpt_e, 0, size);
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100628
629 /* Generate Primary GPT header (LBA1) */
630 ret = gpt_fill_header(dev_desc, gpt_h, str_disk_guid, parts_count);
631 if (ret)
632 goto err;
633
634 /* Generate partition entries */
Maxime Ripard47d7ee42017-08-23 16:01:32 +0200635 ret = gpt_fill_pte(dev_desc, gpt_h, gpt_e, partitions, parts_count);
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100636 if (ret)
637 goto err;
638
639 /* Write GPT partition table */
640 ret = write_gpt_table(dev_desc, gpt_h, gpt_e);
641
642err:
643 free(gpt_e);
644 free(gpt_h);
645 return ret;
646}
Steve Rae0ff7e582014-12-12 15:51:54 -0800647
Heinrich Schuchardt06e921b2019-07-14 18:12:32 +0200648/**
649 * gpt_convert_efi_name_to_char() - convert u16 string to char string
650 *
651 * TODO: this conversion only supports ANSI characters
652 *
653 * @s: target buffer
654 * @es: u16 string to be converted
655 * @n: size of target buffer
656 */
657static void gpt_convert_efi_name_to_char(char *s, void *es, int n)
Lukasz Majewskicef68bf2015-11-20 08:06:16 +0100658{
Heinrich Schuchardt06e921b2019-07-14 18:12:32 +0200659 char *ess = es;
Lukasz Majewskicef68bf2015-11-20 08:06:16 +0100660 int i, j;
661
662 memset(s, '\0', n);
663
664 for (i = 0, j = 0; j < n; i += 2, j++) {
665 s[j] = ess[i];
666 if (!ess[i])
667 return;
668 }
669}
670
Simon Glass4101f682016-02-29 15:25:34 -0700671int gpt_verify_headers(struct blk_desc *dev_desc, gpt_header *gpt_head,
Lukasz Majewskicef68bf2015-11-20 08:06:16 +0100672 gpt_entry **gpt_pte)
673{
674 /*
675 * This function validates AND
676 * fills in the GPT header and PTE
677 */
678 if (is_gpt_valid(dev_desc,
679 GPT_PRIMARY_PARTITION_TABLE_LBA,
680 gpt_head, gpt_pte) != 1) {
681 printf("%s: *** ERROR: Invalid GPT ***\n",
682 __func__);
683 return -1;
684 }
Eugeniu Rosca716f9192019-04-30 04:53:45 +0200685
686 /* Free pte before allocating again */
687 free(*gpt_pte);
688
Stefan Herbrechtsmeierd4693382021-03-08 16:07:11 +0000689 /*
690 * Check that the alternate_lba entry points to the last LBA
691 */
692 if (le64_to_cpu(gpt_head->alternate_lba) != (dev_desc->lba - 1)) {
693 printf("%s: *** ERROR: Misplaced Backup GPT ***\n",
694 __func__);
695 return -1;
696 }
697
Lukasz Majewskicef68bf2015-11-20 08:06:16 +0100698 if (is_gpt_valid(dev_desc, (dev_desc->lba - 1),
699 gpt_head, gpt_pte) != 1) {
700 printf("%s: *** ERROR: Invalid Backup GPT ***\n",
701 __func__);
702 return -1;
703 }
704
705 return 0;
706}
707
Simon Glass4101f682016-02-29 15:25:34 -0700708int gpt_verify_partitions(struct blk_desc *dev_desc,
Simon Glass05289792020-05-10 11:39:57 -0600709 struct disk_partition *partitions, int parts,
Lukasz Majewskicef68bf2015-11-20 08:06:16 +0100710 gpt_header *gpt_head, gpt_entry **gpt_pte)
711{
712 char efi_str[PARTNAME_SZ + 1];
713 u64 gpt_part_size;
714 gpt_entry *gpt_e;
715 int ret, i;
716
717 ret = gpt_verify_headers(dev_desc, gpt_head, gpt_pte);
718 if (ret)
719 return ret;
720
721 gpt_e = *gpt_pte;
722
723 for (i = 0; i < parts; i++) {
724 if (i == gpt_head->num_partition_entries) {
Masahiro Yamada9b643e32017-09-16 14:10:41 +0900725 pr_err("More partitions than allowed!\n");
Lukasz Majewskicef68bf2015-11-20 08:06:16 +0100726 return -1;
727 }
728
729 /* Check if GPT and ENV partition names match */
730 gpt_convert_efi_name_to_char(efi_str, gpt_e[i].partition_name,
731 PARTNAME_SZ + 1);
732
733 debug("%s: part: %2d name - GPT: %16s, ENV: %16s ",
734 __func__, i, efi_str, partitions[i].name);
735
736 if (strncmp(efi_str, (char *)partitions[i].name,
737 sizeof(partitions->name))) {
Masahiro Yamada9b643e32017-09-16 14:10:41 +0900738 pr_err("Partition name: %s does not match %s!\n",
Lukasz Majewskicef68bf2015-11-20 08:06:16 +0100739 efi_str, (char *)partitions[i].name);
740 return -1;
741 }
742
743 /* Check if GPT and ENV sizes match */
744 gpt_part_size = le64_to_cpu(gpt_e[i].ending_lba) -
745 le64_to_cpu(gpt_e[i].starting_lba) + 1;
746 debug("size(LBA) - GPT: %8llu, ENV: %8llu ",
Simon Glassf8d61652016-02-29 15:25:36 -0700747 (unsigned long long)gpt_part_size,
748 (unsigned long long)partitions[i].size);
Lukasz Majewskicef68bf2015-11-20 08:06:16 +0100749
750 if (le64_to_cpu(gpt_part_size) != partitions[i].size) {
Kever Yangc2fdd342016-07-29 11:12:18 +0800751 /* We do not check the extend partition size */
752 if ((i == parts - 1) && (partitions[i].size == 0))
753 continue;
754
Masahiro Yamada9b643e32017-09-16 14:10:41 +0900755 pr_err("Partition %s size: %llu does not match %llu!\n",
Simon Glassf8d61652016-02-29 15:25:36 -0700756 efi_str, (unsigned long long)gpt_part_size,
757 (unsigned long long)partitions[i].size);
Lukasz Majewskicef68bf2015-11-20 08:06:16 +0100758 return -1;
759 }
760
761 /*
762 * Start address is optional - check only if provided
763 * in '$partition' variable
764 */
765 if (!partitions[i].start) {
766 debug("\n");
767 continue;
768 }
769
770 /* Check if GPT and ENV start LBAs match */
771 debug("start LBA - GPT: %8llu, ENV: %8llu\n",
772 le64_to_cpu(gpt_e[i].starting_lba),
Simon Glassf8d61652016-02-29 15:25:36 -0700773 (unsigned long long)partitions[i].start);
Lukasz Majewskicef68bf2015-11-20 08:06:16 +0100774
775 if (le64_to_cpu(gpt_e[i].starting_lba) != partitions[i].start) {
Masahiro Yamada9b643e32017-09-16 14:10:41 +0900776 pr_err("Partition %s start: %llu does not match %llu!\n",
Lukasz Majewskicef68bf2015-11-20 08:06:16 +0100777 efi_str, le64_to_cpu(gpt_e[i].starting_lba),
Simon Glassf8d61652016-02-29 15:25:36 -0700778 (unsigned long long)partitions[i].start);
Lukasz Majewskicef68bf2015-11-20 08:06:16 +0100779 return -1;
780 }
781 }
782
783 return 0;
784}
785
Simon Glass4101f682016-02-29 15:25:34 -0700786int is_valid_gpt_buf(struct blk_desc *dev_desc, void *buf)
Steve Rae0ff7e582014-12-12 15:51:54 -0800787{
788 gpt_header *gpt_h;
789 gpt_entry *gpt_e;
790
791 /* determine start of GPT Header in the buffer */
792 gpt_h = buf + (GPT_PRIMARY_PARTITION_TABLE_LBA *
793 dev_desc->blksz);
794 if (validate_gpt_header(gpt_h, GPT_PRIMARY_PARTITION_TABLE_LBA,
795 dev_desc->lba))
796 return -1;
797
798 /* determine start of GPT Entries in the buffer */
799 gpt_e = buf + (le64_to_cpu(gpt_h->partition_entry_lba) *
800 dev_desc->blksz);
801 if (validate_gpt_entries(gpt_h, gpt_e))
802 return -1;
803
804 return 0;
805}
806
Simon Glass4101f682016-02-29 15:25:34 -0700807int write_mbr_and_gpt_partitions(struct blk_desc *dev_desc, void *buf)
Steve Rae0ff7e582014-12-12 15:51:54 -0800808{
809 gpt_header *gpt_h;
810 gpt_entry *gpt_e;
811 int gpt_e_blk_cnt;
812 lbaint_t lba;
813 int cnt;
814
815 if (is_valid_gpt_buf(dev_desc, buf))
816 return -1;
817
818 /* determine start of GPT Header in the buffer */
819 gpt_h = buf + (GPT_PRIMARY_PARTITION_TABLE_LBA *
820 dev_desc->blksz);
821
822 /* determine start of GPT Entries in the buffer */
823 gpt_e = buf + (le64_to_cpu(gpt_h->partition_entry_lba) *
824 dev_desc->blksz);
825 gpt_e_blk_cnt = BLOCK_CNT((le32_to_cpu(gpt_h->num_partition_entries) *
826 le32_to_cpu(gpt_h->sizeof_partition_entry)),
827 dev_desc);
828
829 /* write MBR */
830 lba = 0; /* MBR is always at 0 */
831 cnt = 1; /* MBR (1 block) */
Simon Glass2a981dc2016-02-29 15:25:52 -0700832 if (blk_dwrite(dev_desc, lba, cnt, buf) != cnt) {
Steve Rae0ff7e582014-12-12 15:51:54 -0800833 printf("%s: failed writing '%s' (%d blks at 0x" LBAF ")\n",
834 __func__, "MBR", cnt, lba);
835 return 1;
836 }
837
838 /* write Primary GPT */
839 lba = GPT_PRIMARY_PARTITION_TABLE_LBA;
840 cnt = 1; /* GPT Header (1 block) */
Simon Glass2a981dc2016-02-29 15:25:52 -0700841 if (blk_dwrite(dev_desc, lba, cnt, gpt_h) != cnt) {
Steve Rae0ff7e582014-12-12 15:51:54 -0800842 printf("%s: failed writing '%s' (%d blks at 0x" LBAF ")\n",
843 __func__, "Primary GPT Header", cnt, lba);
844 return 1;
845 }
846
847 lba = le64_to_cpu(gpt_h->partition_entry_lba);
848 cnt = gpt_e_blk_cnt;
Simon Glass2a981dc2016-02-29 15:25:52 -0700849 if (blk_dwrite(dev_desc, lba, cnt, gpt_e) != cnt) {
Steve Rae0ff7e582014-12-12 15:51:54 -0800850 printf("%s: failed writing '%s' (%d blks at 0x" LBAF ")\n",
851 __func__, "Primary GPT Entries", cnt, lba);
852 return 1;
853 }
854
855 prepare_backup_gpt_header(gpt_h);
856
857 /* write Backup GPT */
858 lba = le64_to_cpu(gpt_h->partition_entry_lba);
859 cnt = gpt_e_blk_cnt;
Simon Glass2a981dc2016-02-29 15:25:52 -0700860 if (blk_dwrite(dev_desc, lba, cnt, gpt_e) != cnt) {
Steve Rae0ff7e582014-12-12 15:51:54 -0800861 printf("%s: failed writing '%s' (%d blks at 0x" LBAF ")\n",
862 __func__, "Backup GPT Entries", cnt, lba);
863 return 1;
864 }
865
866 lba = le64_to_cpu(gpt_h->my_lba);
867 cnt = 1; /* GPT Header (1 block) */
Simon Glass2a981dc2016-02-29 15:25:52 -0700868 if (blk_dwrite(dev_desc, lba, cnt, gpt_h) != cnt) {
Steve Rae0ff7e582014-12-12 15:51:54 -0800869 printf("%s: failed writing '%s' (%d blks at 0x" LBAF ")\n",
870 __func__, "Backup GPT Header", cnt, lba);
871 return 1;
872 }
873
Gary Bisson4e9bd062021-01-26 14:56:23 +0100874 /* Update the partition table entries*/
875 part_init(dev_desc);
876
Steve Rae0ff7e582014-12-12 15:51:54 -0800877 return 0;
878}
Lukasz Majewski40684dd2012-12-11 11:09:46 +0100879#endif
880
richardretanubun07f3d782008-09-26 11:13:22 -0400881/*
882 * Private functions
883 */
884/*
885 * pmbr_part_valid(): Check for EFI partition signature
886 *
887 * Returns: 1 if EFI GPT partition type is found.
888 */
889static int pmbr_part_valid(struct partition *part)
890{
891 if (part->sys_ind == EFI_PMBR_OSTYPE_EFI_GPT &&
Marc Dietrich8faefad2013-03-29 07:57:10 +0000892 get_unaligned_le32(&part->start_sect) == 1UL) {
richardretanubun07f3d782008-09-26 11:13:22 -0400893 return 1;
894 }
895
896 return 0;
897}
898
899/*
900 * is_pmbr_valid(): test Protective MBR for validity
901 *
902 * Returns: 1 if PMBR is valid, 0 otherwise.
903 * Validity depends on two things:
904 * 1) MSDOS signature is in the last two bytes of the MBR
905 * 2) One partition of type 0xEE is found, checked by pmbr_part_valid()
906 */
907static int is_pmbr_valid(legacy_mbr * mbr)
908{
909 int i = 0;
910
Chang Hyun Parkfae2bf22012-12-11 11:09:45 +0100911 if (!mbr || le16_to_cpu(mbr->signature) != MSDOS_MBR_SIGNATURE)
richardretanubun07f3d782008-09-26 11:13:22 -0400912 return 0;
richardretanubun07f3d782008-09-26 11:13:22 -0400913
914 for (i = 0; i < 4; i++) {
915 if (pmbr_part_valid(&mbr->partition_record[i])) {
916 return 1;
917 }
918 }
919 return 0;
920}
921
922/**
923 * is_gpt_valid() - tests one GPT header and PTEs for validity
924 *
925 * lba is the logical block address of the GPT header to test
926 * gpt is a GPT header ptr, filled on return.
927 * ptes is a PTEs ptr, filled on return.
928 *
Urja Rannikko0557d462019-04-11 20:27:51 +0000929 * Description: returns 1 if valid, 0 on error, 2 if ignored header
richardretanubun07f3d782008-09-26 11:13:22 -0400930 * If valid, returns pointers to PTEs.
931 */
Simon Glass4101f682016-02-29 15:25:34 -0700932static int is_gpt_valid(struct blk_desc *dev_desc, u64 lba,
Steve Raee04350d2014-05-26 11:52:23 -0700933 gpt_header *pgpt_head, gpt_entry **pgpt_pte)
richardretanubun07f3d782008-09-26 11:13:22 -0400934{
Tom Rinib351ccf2017-10-03 09:38:44 -0400935 /* Confirm valid arguments prior to allocation. */
richardretanubun07f3d782008-09-26 11:13:22 -0400936 if (!dev_desc || !pgpt_head) {
Doug Andersondf70b1c2011-10-19 08:04:46 +0000937 printf("%s: Invalid Argument(s)\n", __func__);
richardretanubun07f3d782008-09-26 11:13:22 -0400938 return 0;
939 }
940
Patrick Delaunay3cc56612017-11-17 10:08:18 +0100941 ALLOC_CACHE_ALIGN_BUFFER_PAD(legacy_mbr, mbr, 1, dev_desc->blksz);
Tom Rinib351ccf2017-10-03 09:38:44 -0400942
Peter Jonesff98cb92017-09-13 18:05:25 -0400943 /* Read MBR Header from device */
944 if (blk_dread(dev_desc, 0, 1, (ulong *)mbr) != 1) {
945 printf("*** ERROR: Can't read MBR header ***\n");
946 return 0;
947 }
948
richardretanubun07f3d782008-09-26 11:13:22 -0400949 /* Read GPT Header from device */
Simon Glass2a981dc2016-02-29 15:25:52 -0700950 if (blk_dread(dev_desc, (lbaint_t)lba, 1, pgpt_head) != 1) {
richardretanubun07f3d782008-09-26 11:13:22 -0400951 printf("*** ERROR: Can't read GPT header ***\n");
952 return 0;
953 }
954
Urja Rannikko0557d462019-04-11 20:27:51 +0000955 /* Invalid but nothing to yell about. */
956 if (le64_to_cpu(pgpt_head->signature) == GPT_HEADER_CHROMEOS_IGNORE) {
957 debug("ChromeOS 'IGNOREME' GPT header found and ignored\n");
958 return 2;
959 }
960
Steve Raee1f6b0a2014-12-18 12:13:42 +0100961 if (validate_gpt_header(pgpt_head, (lbaint_t)lba, dev_desc->lba))
richardretanubun07f3d782008-09-26 11:13:22 -0400962 return 0;
richardretanubun07f3d782008-09-26 11:13:22 -0400963
Peter Jonesff98cb92017-09-13 18:05:25 -0400964 if (dev_desc->sig_type == SIG_TYPE_NONE) {
965 efi_guid_t empty = {};
966 if (memcmp(&pgpt_head->disk_guid, &empty, sizeof(empty))) {
967 dev_desc->sig_type = SIG_TYPE_GUID;
968 memcpy(&dev_desc->guid_sig, &pgpt_head->disk_guid,
969 sizeof(empty));
970 } else if (mbr->unique_mbr_signature != 0) {
971 dev_desc->sig_type = SIG_TYPE_MBR;
972 dev_desc->mbr_sig = mbr->unique_mbr_signature;
973 }
974 }
975
richardretanubun07f3d782008-09-26 11:13:22 -0400976 /* Read and allocate Partition Table Entries */
977 *pgpt_pte = alloc_read_gpt_entries(dev_desc, pgpt_head);
978 if (*pgpt_pte == NULL) {
979 printf("GPT: Failed to allocate memory for PTE\n");
980 return 0;
981 }
982
Steve Raee1f6b0a2014-12-18 12:13:42 +0100983 if (validate_gpt_entries(pgpt_head, *pgpt_pte)) {
Doug Andersondeb5ca82011-10-19 09:47:31 +0000984 free(*pgpt_pte);
richardretanubun07f3d782008-09-26 11:13:22 -0400985 return 0;
986 }
987
988 /* We're done, all's well */
989 return 1;
990}
991
992/**
Urja Rannikko20c568c2019-04-11 20:27:50 +0000993 * find_valid_gpt() - finds a valid GPT header and PTEs
994 *
995 * gpt is a GPT header ptr, filled on return.
996 * ptes is a PTEs ptr, filled on return.
997 *
998 * Description: returns 1 if found a valid gpt, 0 on error.
999 * If valid, returns pointers to PTEs.
1000 */
1001static int find_valid_gpt(struct blk_desc *dev_desc, gpt_header *gpt_head,
1002 gpt_entry **pgpt_pte)
1003{
Urja Rannikko0557d462019-04-11 20:27:51 +00001004 int r;
1005
1006 r = is_gpt_valid(dev_desc, GPT_PRIMARY_PARTITION_TABLE_LBA, gpt_head,
1007 pgpt_pte);
1008
1009 if (r != 1) {
1010 if (r != 2)
1011 printf("%s: *** ERROR: Invalid GPT ***\n", __func__);
1012
1013 if (is_gpt_valid(dev_desc, (dev_desc->lba - 1), gpt_head,
1014 pgpt_pte) != 1) {
Urja Rannikko20c568c2019-04-11 20:27:50 +00001015 printf("%s: *** ERROR: Invalid Backup GPT ***\n",
1016 __func__);
1017 return 0;
1018 }
Urja Rannikko0557d462019-04-11 20:27:51 +00001019 if (r != 2)
1020 printf("%s: *** Using Backup GPT ***\n",
1021 __func__);
Urja Rannikko20c568c2019-04-11 20:27:50 +00001022 }
1023 return 1;
1024}
1025
1026/**
richardretanubun07f3d782008-09-26 11:13:22 -04001027 * alloc_read_gpt_entries(): reads partition entries from disk
1028 * @dev_desc
1029 * @gpt - GPT header
1030 *
1031 * Description: Returns ptes on success, NULL on error.
1032 * Allocates space for PTEs based on information found in @gpt.
1033 * Notes: remember to free pte when you're done!
1034 */
Simon Glass4101f682016-02-29 15:25:34 -07001035static gpt_entry *alloc_read_gpt_entries(struct blk_desc *dev_desc,
1036 gpt_header *pgpt_head)
richardretanubun07f3d782008-09-26 11:13:22 -04001037{
Egbert Eichae1768a2013-04-09 06:03:36 +00001038 size_t count = 0, blk_cnt;
Stephen Warren7c4213f2015-12-07 11:38:48 -07001039 lbaint_t blk;
richardretanubun07f3d782008-09-26 11:13:22 -04001040 gpt_entry *pte = NULL;
1041
1042 if (!dev_desc || !pgpt_head) {
Doug Andersondf70b1c2011-10-19 08:04:46 +00001043 printf("%s: Invalid Argument(s)\n", __func__);
richardretanubun07f3d782008-09-26 11:13:22 -04001044 return NULL;
1045 }
1046
Chang Hyun Parkfae2bf22012-12-11 11:09:45 +01001047 count = le32_to_cpu(pgpt_head->num_partition_entries) *
1048 le32_to_cpu(pgpt_head->sizeof_partition_entry);
richardretanubun07f3d782008-09-26 11:13:22 -04001049
Simon Glass5afb8d12016-07-22 09:22:47 -06001050 debug("%s: count = %u * %u = %lu\n", __func__,
Chang Hyun Parkfae2bf22012-12-11 11:09:45 +01001051 (u32) le32_to_cpu(pgpt_head->num_partition_entries),
Simon Glass5afb8d12016-07-22 09:22:47 -06001052 (u32) le32_to_cpu(pgpt_head->sizeof_partition_entry),
1053 (ulong)count);
richardretanubun07f3d782008-09-26 11:13:22 -04001054
1055 /* Allocate memory for PTE, remember to FREE */
1056 if (count != 0) {
Egbert Eichae1768a2013-04-09 06:03:36 +00001057 pte = memalign(ARCH_DMA_MINALIGN,
1058 PAD_TO_BLOCKSIZE(count, dev_desc));
richardretanubun07f3d782008-09-26 11:13:22 -04001059 }
1060
1061 if (count == 0 || pte == NULL) {
Simon Glass5afb8d12016-07-22 09:22:47 -06001062 printf("%s: ERROR: Can't allocate %#lX bytes for GPT Entries\n",
1063 __func__, (ulong)count);
richardretanubun07f3d782008-09-26 11:13:22 -04001064 return NULL;
1065 }
1066
1067 /* Read GPT Entries from device */
Stephen Warren7c4213f2015-12-07 11:38:48 -07001068 blk = le64_to_cpu(pgpt_head->partition_entry_lba);
Egbert Eichae1768a2013-04-09 06:03:36 +00001069 blk_cnt = BLOCK_CNT(count, dev_desc);
Simon Glass2a981dc2016-02-29 15:25:52 -07001070 if (blk_dread(dev_desc, blk, (lbaint_t)blk_cnt, pte) != blk_cnt) {
richardretanubun07f3d782008-09-26 11:13:22 -04001071 printf("*** ERROR: Can't read GPT Entries ***\n");
1072 free(pte);
1073 return NULL;
1074 }
1075 return pte;
1076}
1077
1078/**
1079 * is_pte_valid(): validates a single Partition Table Entry
1080 * @gpt_entry - Pointer to a single Partition Table Entry
1081 *
1082 * Description: returns 1 if valid, 0 on error.
1083 */
1084static int is_pte_valid(gpt_entry * pte)
1085{
1086 efi_guid_t unused_guid;
1087
1088 if (!pte) {
Doug Andersondf70b1c2011-10-19 08:04:46 +00001089 printf("%s: Invalid Argument(s)\n", __func__);
richardretanubun07f3d782008-09-26 11:13:22 -04001090 return 0;
1091 }
1092
1093 /* Only one validation for now:
1094 * The GUID Partition Type != Unused Entry (ALL-ZERO)
1095 */
1096 memset(unused_guid.b, 0, sizeof(unused_guid.b));
1097
1098 if (memcmp(pte->partition_type_guid.b, unused_guid.b,
1099 sizeof(unused_guid.b)) == 0) {
1100
Doug Andersondf70b1c2011-10-19 08:04:46 +00001101 debug("%s: Found an unused PTE GUID at 0x%08X\n", __func__,
Taylor Hutt9936be32012-10-12 14:26:09 +00001102 (unsigned int)(uintptr_t)pte);
richardretanubun07f3d782008-09-26 11:13:22 -04001103
1104 return 0;
1105 } else {
1106 return 1;
1107 }
1108}
Simon Glass96e5b032016-02-29 15:25:47 -07001109
1110/*
1111 * Add an 'a_' prefix so it comes before 'dos' in the linker list. We need to
1112 * check EFI first, since a DOS partition is often used as a 'protective MBR'
1113 * with EFI.
1114 */
1115U_BOOT_PART_TYPE(a_efi) = {
1116 .name = "EFI",
1117 .part_type = PART_TYPE_EFI,
Petr Kulhavy87b85302016-09-09 10:27:15 +02001118 .max_entries = GPT_ENTRY_NUMBERS,
Simon Glass3e8bd462016-02-29 15:25:48 -07001119 .get_info = part_get_info_ptr(part_get_info_efi),
Simon Glass084bf4c2016-02-29 15:26:04 -07001120 .print = part_print_ptr(part_print_efi),
1121 .test = part_test_efi,
Simon Glass96e5b032016-02-29 15:25:47 -07001122};
Simon Glassa594b412021-07-02 12:36:14 -06001123#endif /* CONFIG_HAVE_BLOCK_DEVICE */