New command - "gpt" is supported. It restores the GPT partition table.
It looks into the "partitions" environment variable for partitions definition.
It can be enabled at target configuration file with CONFIG_CMD_GPT.
Simple UUID generator has been implemented. It uses the the gd->start_addr_sp
for entrophy pool. Moreover the pool address is used as crc32 seed.

Signed-off-by: Lukasz Majewski <l.majew...@samsung.com>
Signed-off-by: Piotr Wilczek <p.wilc...@samsung.com>
Signed-off-by: Kyungmin Park <kyungmin.p...@samsung.com>
---
Changes for v2:
- gpt command now accepts device medium and its number (e.g. gpt mmc 0)
- UUIDs can be passed via u-boot prompt when used with gpt command
- Format of restored GPT has been changed - now key=value pairs are used
  'name="PARTS_CSA",size=8MiB;\ '
  'name="PARTS_BOOTLOADER",size=60MiB; \'
- guid_gen now accepts "pool" pointer and guid pointer
- gd->start_addr_sp is used as a primary source of entrophy
- static buffers definitions have been removed
- remove memsize_to_blocks function with call to standard ustrtoul
- doxygen comments for functions added

Changes for v3:
- Remove unnecessary braces

Changes for v4:
- partions list can be passed as environment variable or directly as text
- each value in partions list can be passed as environment variable
---
 common/Makefile  |    1 +
 common/cmd_gpt.c |  300 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 301 insertions(+), 0 deletions(-)
 create mode 100644 common/cmd_gpt.c

diff --git a/common/Makefile b/common/Makefile
index 9e43322..741781f 100644
--- a/common/Makefile
+++ b/common/Makefile
@@ -192,6 +192,7 @@ COBJS-$(CONFIG_MODEM_SUPPORT) += modem.o
 COBJS-$(CONFIG_UPDATE_TFTP) += update.o
 COBJS-$(CONFIG_USB_KEYBOARD) += usb_kbd.o
 COBJS-$(CONFIG_CMD_DFU) += cmd_dfu.o
+COBJS-$(CONFIG_CMD_GPT) += cmd_gpt.o
 endif
 
 ifdef CONFIG_SPL_BUILD
diff --git a/common/cmd_gpt.c b/common/cmd_gpt.c
new file mode 100644
index 0000000..171eeb3
--- /dev/null
+++ b/common/cmd_gpt.c
@@ -0,0 +1,300 @@
+/*
+ * cmd_gpt.c -- GPT (GUID Partition Table) handling command
+ *
+ * Copyright (C) 2012 Samsung Electronics
+ * author: Lukasz Majewski <l.majew...@samsung.com>
+ * author: Piotr Wilczek <p.wilc...@samsung.com>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+#include <common.h>
+#include <malloc.h>
+#include <command.h>
+#include <mmc.h>
+#include <part_efi.h>
+#include <exports.h>
+
+/**
+ * extract_env(): Convert string from '&{env_name}' to 'env_name'
+ *
+ * @param p - pointer to string
+ *
+ * @return - zero on success
+ */
+static int extract_env(char *p)
+{
+       char *p1, *p2;
+
+       if (!p)
+               return -1;
+
+       p1 = strstr(p, "${");
+       p2 = strstr(p, "}");
+
+       if (p1 && p2) {
+               *p2 = '\0';
+               memmove(p, p+2, p2-p1-1);
+       } else {
+               return -2;
+       }
+
+       return 0;
+}
+
+/**
+ * extract_val(): Extract value from a key=value pair
+ *
+ * @param p - pointer to string
+ * @param tab - table to store extracted value
+ * @param i - actual tab element to work on
+ * @param key - extract only given key
+ *
+ * @return - zero on success; otherwise error
+ */
+static int extract_val(char **p, char *tab[], int i, char *key)
+{
+       char *t, *e, *tok = *p;
+       char *k;
+
+       t = strsep(&tok, ",");
+       k = t;
+       strsep(&t, "=");
+
+       if (key && strcmp(k, key))
+               return -2;
+
+       if (extract_env(t) == 0) {
+               e = getenv(t);
+               if (e == NULL)
+                       return -1;
+               t = e;
+       }
+
+       tab[i] = calloc(strlen(t) + 1, 1);
+       if (tab[i] == NULL) {
+               printf("%s: calloc failed!\n", __func__);
+               return -1;
+       }
+       strcpy(tab[i], t);
+       *p = tok;
+
+       return 0;
+}
+
+/**
+ * set_gpt_info(): Fill gpt information for GPT header and partition entries
+ *
+ * @param dev_desc - block device descriptor
+ * @param str_uuid - string uuid representation
+ *
+ * @return - zero on success; otherwise error
+ *
+ */
+static int set_gpt_info(block_dev_desc_t *dev_desc, char *str_part,
+       disk_partition_t *partitions[], const int parts_count)
+{
+       char *ps[parts_count];
+       char *name[parts_count];
+       char *uuid[parts_count];
+       unsigned int size[parts_count];
+       char *tok, *p, *s, *ss;
+       int i, ret;
+
+       debug("%s: MMC lba num: 0x%x %d\n", __func__,
+             (unsigned int)dev_desc->lba, (unsigned int)dev_desc->lba);
+
+       if (!str_part)
+               return -2;
+       ret = 0;
+
+       s = str_part;
+
+       printf("PARTITIONS: %s\n", s);
+       ss = calloc(strlen(s) + 1, 1);
+       if (ss == NULL) {
+               printf("%s: calloc failed!\n", __func__);
+               return -1;
+       }
+       memcpy(ss, s, strlen(s) + 1);
+
+       for (i = 0, p = ss; i < parts_count; i++) {
+               tok = strsep(&p, ";");
+               if (tok == NULL)
+                       break;
+
+               if (extract_val(&tok, name, i, "name")) {
+                       ret = -1;
+                       goto err;
+               }
+
+               if (extract_val(&tok, ps, i, "size")) {
+                       ret = -1;
+                       free(name[i]);
+                       goto err;
+               }
+
+               if (extract_val(&tok, uuid, i, "uuid")) {
+                       /* uuid string length equals 37 */
+                       uuid[i] = calloc(37, 1);
+                       if (uuid[i] == NULL) {
+                               printf("%s: calloc failed!\n", __func__);
+                               ret = -1;
+                               free(name[i]);
+                               free(ps[i]);
+                               goto err;
+                       }
+               }
+       }
+
+       printf("found %d partitions\n", i);
+
+       for (i = 0; i < parts_count; i++) {
+               p = ps[i];
+               size[i] = ustrtoul(p, &p, 0);
+               size[i] /= dev_desc->blksz;
+       }
+
+       for (i = 0; i < parts_count; i++) {
+               partitions[i]->size = size[i];
+               partitions[i]->blksz = dev_desc->blksz;
+               strncpy((char *)&partitions[i]->name, name[i],
+                               sizeof(partitions[i]->name));
+#ifdef CONFIG_PARTITION_UUIDS
+               strncpy((char *)&partitions[i]->uuid, uuid[i],
+                       sizeof(partitions[i]->uuid));
+#endif
+       }
+
+       i = parts_count;
+ err:
+       for (i--; i >= 0; i--) {
+               free(name[i]);
+               free(uuid[i]);
+               free(ps[i]);
+       }
+
+       free(ss);
+       return ret;
+}
+
+static int gpt_mmc_default(int dev, char *str_part)
+{
+       int i;
+       u8 part_count;
+       char *s , *p;
+
+       struct mmc *mmc = find_mmc_device(dev);
+
+       if (mmc == NULL) {
+               printf("%s: mmc dev %d NOT available\n", __func__, dev);
+               return CMD_RET_FAILURE;
+       }
+
+       puts("Using default GPT UUID\n");
+
+       /* get partitions list string */
+       if (!str_part)
+#ifdef DEFAULT_PARTITIONS_ENV
+               str_part = DEFAULT_PARTITIONS_ENV;
+#else
+               str_part = "partitions";
+#endif
+       s = getenv(str_part);
+       if (s == NULL)
+               s = str_part;
+
+       /* calculate expected number of partitions */
+       part_count = 1;
+       p = s;
+       while (*p) {
+               if (*p++ == ';')
+                       part_count++;
+       }
+
+       /* allocate memory for partitions */
+       disk_partition_t *partitions[part_count];
+       for (i = 0; i < part_count; i++) {
+               partitions[i] = calloc(sizeof(disk_partition_t), part_count);
+               if (partitions[i] == NULL)
+                       goto err;
+       }
+
+       /* fill partitions */
+       if (!set_gpt_info(&mmc->block_dev, s, partitions, part_count))
+               /* save partions layout do disk */
+               gpt_fill(&mmc->block_dev, partitions, part_count);
+       else
+               printf("Partitions list incomplete\n");
+
+err:
+       /* free memory for partitions */
+       for (i--; i >= 0; i--)
+               free(partitions[i]);
+
+       return 0;
+}
+
+/**
+ * do_gpt(): Perform GPT operations
+ *
+ * @param cmdtp - command name
+ * @param flag
+ * @param argc
+ * @param argv
+ *
+ * @return zero on success; otherwise error
+ */
+static int do_gpt(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
+{
+       int ret = CMD_RET_SUCCESS;
+       char *str_part = NULL;
+       int dev = 0;
+
+       if (argc < 3)
+               return CMD_RET_USAGE;
+
+       if (argc == 4) {
+               str_part = strdup(argv[3]);
+               if (!str_part) {
+                       printf("%s: malloc failed!\n", __func__);
+                       return CMD_RET_FAILURE;
+               }
+       }
+
+       if (strcmp(argv[1], "mmc") == 0) {
+               dev = (int)simple_strtoul(argv[2], NULL, 10);
+               if (gpt_mmc_default(dev, str_part))
+                       return CMD_RET_FAILURE;
+       }
+
+       if (argc == 4)
+               free(str_part);
+
+       return ret;
+}
+
+U_BOOT_CMD(gpt, CONFIG_SYS_MAXARGS, 1, do_gpt,
+       "GUID Partition Table",
+       "<interface> <dev> <partions list>\n"
+       " partions list is in format: name=..,size=..,uuid=..;...\n"
+       " and can be passed as env or string ex.:\n"
+       "    gpt mmc 0 partitions\n"
+       "    gpt mmc 0 \"name=..,size=..;name=..,size=..;...\"\n"
+       "    gpt mmc 0 \"name=${part1_name},size=..;name=..,size=..;...\"\n"
+       " - GUID partition table restoration\n"
+       " Restore GPT information on a device connected\n"
+       " to interface\n"
+);
-- 
1.7.5.4

_______________________________________________
U-Boot mailing list
U-Boot@lists.denx.de
http://lists.denx.de/mailman/listinfo/u-boot

Reply via email to