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 val = extract_val(str, "uuid_disk");
159 #ifdef CONFIG_RANDOM_UUID
160 *str_disk_guid = malloc(UUID_STR_LEN + 1);
161 gen_rand_uuid_str(*str_disk_guid, UUID_STR_FORMAT_STD);
170 if (extract_env(v, &p))
172 *str_disk_guid = strdup(p);
174 /* Move s to first partition */
184 /* calculate expected number of partitions */
192 /* allocate memory for partitions */
193 parts = calloc(sizeof(disk_partition_t), p_count);
195 /* retrieve partitions data from string */
196 for (i = 0; i < p_count; i++) {
197 tok = strsep(&s, ";");
203 val = extract_val(tok, "uuid");
205 /* 'uuid' is optional if random uuid's are enabled */
206 #ifdef CONFIG_RANDOM_UUID
207 gen_rand_uuid_str(parts[i].uuid, UUID_STR_FORMAT_STD);
213 if (extract_env(val, &p))
215 if (strlen(p) >= sizeof(parts[i].uuid)) {
216 printf("Wrong uuid format for partition %d\n", i);
220 strcpy((char *)parts[i].uuid, p);
224 val = extract_val(tok, "name");
225 if (!val) { /* name is mandatory */
229 if (extract_env(val, &p))
231 if (strlen(p) >= sizeof(parts[i].name)) {
235 strcpy((char *)parts[i].name, p);
239 val = extract_val(tok, "size");
240 if (!val) { /* 'size' is mandatory */
244 if (extract_env(val, &p))
246 size_ll = ustrtoull(p, &p, 0);
247 parts[i].size = lldiv(size_ll, dev_desc->blksz);
251 val = extract_val(tok, "start");
252 if (val) { /* start address is optional */
253 if (extract_env(val, &p))
255 start_ll = ustrtoull(p, &p, 0);
256 parts[i].start = lldiv(start_ll, dev_desc->blksz);
261 *parts_count = p_count;
268 free(*str_disk_guid);
272 *str_disk_guid = NULL;
278 static int gpt_default(block_dev_desc_t *blk_dev_desc, const char *str_part)
283 disk_partition_t *partitions = NULL;
288 /* fill partitions */
289 ret = set_gpt_info(blk_dev_desc, str_part,
290 &str_disk_guid, &partitions, &part_count);
293 printf("No partition list provided\n");
295 printf("Missing disk guid\n");
296 if ((ret == -3) || (ret == -4))
297 printf("Partition list incomplete\n");
301 /* save partitions layout to disk */
302 ret = gpt_restore(blk_dev_desc, str_disk_guid, partitions, part_count);
310 * do_gpt(): Perform GPT operations
312 * @param cmdtp - command name
317 * @return zero on success; otherwise error
319 static int do_gpt(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
321 int ret = CMD_RET_SUCCESS;
324 block_dev_desc_t *blk_dev_desc;
327 return CMD_RET_USAGE;
329 /* command: 'write' */
330 if ((strcmp(argv[1], "write") == 0) && (argc == 5)) {
331 dev = (int)simple_strtoul(argv[3], &ep, 10);
332 if (!ep || ep[0] != '\0') {
333 printf("'%s' is not a number\n", argv[3]);
334 return CMD_RET_USAGE;
336 blk_dev_desc = get_dev(argv[2], dev);
338 printf("%s: %s dev %d NOT available\n",
339 __func__, argv[2], dev);
340 return CMD_RET_FAILURE;
343 puts("Writing GPT: ");
345 ret = gpt_default(blk_dev_desc, argv[4]);
348 return CMD_RET_SUCCESS;
351 return CMD_RET_FAILURE;
354 return CMD_RET_USAGE;
359 U_BOOT_CMD(gpt, CONFIG_SYS_MAXARGS, 1, do_gpt,
360 "GUID Partition Table",
361 "<command> <interface> <dev> <partitions_list>\n"
362 " - GUID partition table restoration\n"
363 " Restore GPT information on a device connected\n"