]>
git.ipfire.org Git - people/ms/u-boot.git/blob - common/cmd_gpt.c
2 * cmd_gpt.c -- GPT (GUID Partition Table) handling command
4 * Copyright (C) 2012 Samsung Electronics
5 * author: Lukasz Majewski <l.majewski@samsung.com>
6 * author: Piotr Wilczek <p.wilczek@samsung.com>
8 * SPDX-License-Identifier: GPL-2.0+
16 #include <linux/ctype.h>
19 #ifndef CONFIG_PARTITION_UUIDS
20 #error CONFIG_PARTITION_UUIDS must be enabled for CONFIG_CMD_GPT to be enabled
24 * extract_env(): Expand env name from string format '&{env_name}'
25 * and return pointer to the env (if the env is set)
27 * @param str - pointer to string
28 * @param env - pointer to pointer to extracted env
30 * @return - zero on successful expand and env is set
32 static int extract_env(const char *str
, char **env
)
36 #ifdef CONFIG_RANDOM_UUID
37 char uuid_str
[UUID_STR_LEN
+ 1];
40 if (!str
|| strlen(str
) < 4)
43 if (!((strncmp(str
, "${", 2) == 0) && (str
[strlen(str
) - 1] == '}')))
50 memset(s
+ strlen(s
) - 1, '\0', 1);
51 memmove(s
, s
+ 2, strlen(s
) - 1);
55 #ifdef CONFIG_RANDOM_UUID
56 debug("%s unset. ", str
);
57 gen_rand_uuid_str(uuid_str
, UUID_STR_FORMAT_STD
);
62 debug("Set to random.\n");
65 debug("Can't get random UUID.\n");
68 debug("%s unset.\n", str
);
71 debug("%s get from environment.\n", str
);
82 * extract_val(): Extract value from a key=value pair list (comma separated).
83 * Only value for the given key is returend.
84 * Function allocates memory for the value, remember to free!
86 * @param str - pointer to string with key=values pairs
87 * @param key - pointer to the key to search for
89 * @return - pointer to allocated string with the value
91 static char *extract_val(const char *str
, const char *key
)
97 strcopy
= strdup(str
);
109 if (strcmp(k
, key
) == 0) {
121 * set_gpt_info(): Fill partition information from string
122 * function allocates memory, remember to free!
124 * @param dev_desc - pointer block device descriptor
125 * @param str_part - pointer to string with partition information
126 * @param str_disk_guid - pointer to pointer to allocated string with disk guid
127 * @param partitions - pointer to pointer to allocated partitions array
128 * @param parts_count - number of partitions
130 * @return - zero on success, otherwise error
133 static int set_gpt_info(block_dev_desc_t
*dev_desc
,
134 const char *str_part
,
135 char **str_disk_guid
,
136 disk_partition_t
**partitions
,
143 disk_partition_t
*parts
;
145 uint64_t size_ll
, start_ll
;
147 debug("%s: lba num: 0x%x %d\n", __func__
,
148 (unsigned int)dev_desc
->lba
, (unsigned int)dev_desc
->lba
);
150 if (str_part
== NULL
)
153 str
= strdup(str_part
);
155 /* extract disk guid */
157 tok
= strsep(&s
, ";");
158 val
= extract_val(tok
, "uuid_disk");
163 if (extract_env(val
, &p
))
165 *str_disk_guid
= strdup(p
);
175 /* calculate expected number of partitions */
183 /* allocate memory for partitions */
184 parts
= calloc(sizeof(disk_partition_t
), p_count
);
186 /* retrieve partitions data from string */
187 for (i
= 0; i
< p_count
; i
++) {
188 tok
= strsep(&s
, ";");
194 val
= extract_val(tok
, "uuid");
195 if (!val
) { /* 'uuid' is mandatory */
199 if (extract_env(val
, &p
))
201 if (strlen(p
) >= sizeof(parts
[i
].uuid
)) {
202 printf("Wrong uuid format for partition %d\n", i
);
206 strcpy((char *)parts
[i
].uuid
, p
);
210 val
= extract_val(tok
, "name");
211 if (!val
) { /* name is mandatory */
215 if (extract_env(val
, &p
))
217 if (strlen(p
) >= sizeof(parts
[i
].name
)) {
221 strcpy((char *)parts
[i
].name
, p
);
225 val
= extract_val(tok
, "size");
226 if (!val
) { /* 'size' is mandatory */
230 if (extract_env(val
, &p
))
232 size_ll
= ustrtoull(p
, &p
, 0);
233 parts
[i
].size
= lldiv(size_ll
, dev_desc
->blksz
);
237 val
= extract_val(tok
, "start");
238 if (val
) { /* start address is optional */
239 if (extract_env(val
, &p
))
241 start_ll
= ustrtoull(p
, &p
, 0);
242 parts
[i
].start
= lldiv(start_ll
, dev_desc
->blksz
);
247 *parts_count
= p_count
;
254 free(*str_disk_guid
);
260 static int gpt_default(block_dev_desc_t
*blk_dev_desc
, const char *str_part
)
265 disk_partition_t
*partitions
= NULL
;
270 /* fill partitions */
271 ret
= set_gpt_info(blk_dev_desc
, str_part
,
272 &str_disk_guid
, &partitions
, &part_count
);
275 printf("No partition list provided\n");
277 printf("Missing disk guid\n");
278 if ((ret
== -3) || (ret
== -4))
279 printf("Partition list incomplete\n");
283 /* save partitions layout to disk */
284 gpt_restore(blk_dev_desc
, str_disk_guid
, partitions
, part_count
);
292 * do_gpt(): Perform GPT operations
294 * @param cmdtp - command name
299 * @return zero on success; otherwise error
301 static int do_gpt(cmd_tbl_t
*cmdtp
, int flag
, int argc
, char * const argv
[])
303 int ret
= CMD_RET_SUCCESS
;
306 block_dev_desc_t
*blk_dev_desc
;
309 return CMD_RET_USAGE
;
311 /* command: 'write' */
312 if ((strcmp(argv
[1], "write") == 0) && (argc
== 5)) {
313 dev
= (int)simple_strtoul(argv
[3], &ep
, 10);
314 if (!ep
|| ep
[0] != '\0') {
315 printf("'%s' is not a number\n", argv
[3]);
316 return CMD_RET_USAGE
;
318 blk_dev_desc
= get_dev(argv
[2], dev
);
320 printf("%s: %s dev %d NOT available\n",
321 __func__
, argv
[2], dev
);
322 return CMD_RET_FAILURE
;
325 puts("Writing GPT: ");
327 ret
= gpt_default(blk_dev_desc
, argv
[4]);
330 return CMD_RET_SUCCESS
;
333 return CMD_RET_FAILURE
;
336 return CMD_RET_USAGE
;
341 U_BOOT_CMD(gpt
, CONFIG_SYS_MAXARGS
, 1, do_gpt
,
342 "GUID Partition Table",
343 "<command> <interface> <dev> <partitions_list>\n"
344 " - GUID partition table restoration\n"
345 " Restore GPT information on a device connected\n"