Hi Alison,

> From: Alison Chaiken <ali...@she-devel.com>
> 
> Make the partition table available for modification by reading it from
> the user-specified device into a linked list.   Provide an accessor
> function for command-line testing.

Acked-by: Lukasz Majewski <lu...@denx.de>

> 
> Signed-off-by: Alison Chaiken <ali...@peloton-tech.com>
> ---
>  cmd/gpt.c      | 112
> +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
> include/part.h |   8 +++++ 2 files changed, 120 insertions(+)
> 
> diff --git a/cmd/gpt.c b/cmd/gpt.c
> index 3b7d929..c61d2b1 100644
> --- a/cmd/gpt.c
> +++ b/cmd/gpt.c
> @@ -19,6 +19,9 @@
>  #include <linux/ctype.h>
>  #include <div64.h>
>  #include <memalign.h>
> +#include <linux/compat.h>
> +
> +static LIST_HEAD(disk_partitions);
>  
>  /**
>   * extract_env(): Expand env name from string format '&{env_name}'
> @@ -151,6 +154,111 @@ static bool found_key(const char *str, const
> char *key) return result;
>  }
>  
> +static void del_gpt_info(void)
> +{
> +     struct list_head *pos = &disk_partitions;
> +     struct disk_part *curr;
> +     while (!list_empty(pos)) {
> +             curr = list_entry(pos->next, struct disk_part, list);
> +             list_del(pos->next);
> +             free(curr);
> +     }
> +}
> +
> +static struct disk_part *allocate_disk_part(disk_partition_t *info,
> int partnum) +{
> +     struct disk_part *newpart;
> +     newpart = (struct disk_part *)malloc(sizeof(*newpart));
> +     memset(newpart, '\0', sizeof(newpart));
> +     if (!newpart)
> +             return ERR_PTR(-ENOMEM);
> +
> +     newpart->gpt_part_info.start = info->start;
> +     newpart->gpt_part_info.size = info->size;
> +     newpart->gpt_part_info.blksz = info->blksz;
> +     strncpy((char *)newpart->gpt_part_info.name, (const char
> *)info->name, PART_NAME_LEN);
> +     memset(newpart->gpt_part_info.name + 31, '\0', 1);
> +     strncpy((char *)newpart->gpt_part_info.type, (const char
> *)info->type, PART_TYPE_LEN);
> +     memset(newpart->gpt_part_info.type + 31, '\0', 1);
> +     newpart->gpt_part_info.bootable = info->bootable;
> +#ifdef CONFIG_PARTITION_UUIDS
> +     strncpy(newpart->gpt_part_info.uuid, (const char
> *)info->uuid,
> +             UUID_STR_LEN);
> +#endif
> +     newpart->partnum = partnum;
> +
> +     return newpart;
> +}
> +
> +static void print_gpt_info(void)
> +{
> +     struct list_head *pos;
> +     struct disk_part *curr;
> +
> +     list_for_each(pos, &disk_partitions) {
> +             curr = list_entry(pos, struct disk_part, list);
> +             printf("Partition %d:\n", curr->partnum);
> +             printf("1st block %x, size %x\n",
> (unsigned)curr->gpt_part_info.start,
> +                    (unsigned)curr->gpt_part_info.size);
> +             printf("Block size %lu, name %s\n",
> curr->gpt_part_info.blksz,
> +                    curr->gpt_part_info.name);
> +             printf("Type %s, bootable %d\n",
> curr->gpt_part_info.type,
> +                    curr->gpt_part_info.bootable);
> +#ifdef CONFIG_PARTITION_UUIDS
> +             printf("UUID %s\n", curr->gpt_part_info.uuid);
> +#endif
> +             printf("\n");
> +     }
> +}
> +
> +/*
> + * read partition info into disk_partitions list where
> + * it can be printed or modified
> + */
> +static int get_gpt_info(struct blk_desc *dev_desc)
> +{
> +     /* start partition numbering at 1, as u-boot does */
> +     int valid_parts = 1, p, ret = 0;
> +     disk_partition_t info;
> +     struct disk_part *new_disk_part;
> +
> +     if (disk_partitions.next == NULL)
> +             INIT_LIST_HEAD(&disk_partitions);
> +
> +     for (p = 1; p <= MAX_SEARCH_PARTITIONS; p++) {
> +             ret = part_get_info(dev_desc, p, &info);
> +             if (ret)
> +                     continue;
> +
> +             new_disk_part = allocate_disk_part(&info,
> valid_parts);
> +             if (IS_ERR(new_disk_part) && (valid_parts >= 2))
> +                     return -1;
> +
> +             list_add_tail(&new_disk_part->list,
> &disk_partitions);
> +             valid_parts++;
> +     }
> +     if (!valid_parts) {
> +             printf("** No valid partitions found **\n");
> +             del_gpt_info();
> +             return -ENODEV;
> +     }
> +     return --valid_parts;
> +}
> +
> +/* a wrapper to test get_gpt_info */
> +static int do_get_gpt_info(struct blk_desc *dev_desc)
> +{
> +     int ret;
> +
> +     ret = get_gpt_info(dev_desc);
> +     if (ret > 0) {
> +             print_gpt_info();
> +             del_gpt_info();
> +     }
> +     return ret;
> +}
> +
> +
>  /**
>   * set_gpt_info(): Fill partition information from string
>   *           function allocates memory, remember to free!
> @@ -457,6 +565,8 @@ static int do_gpt(cmd_tbl_t *cmdtp, int flag, int
> argc, char * const argv[]) if (argc == 5)
>                      strcpy(varname, argv[4]);
>               return do_disk_guid(blk_dev_desc, varname);
> +     } else if (strcmp(argv[1], "read") == 0) {
> +             return do_get_gpt_info(blk_dev_desc);
>       } else {
>               return CMD_RET_USAGE;
>       }
> @@ -479,6 +589,8 @@ U_BOOT_CMD(gpt, CONFIG_SYS_MAXARGS, 1, do_gpt,
>       " Example usage:\n"
>       " gpt write mmc 0 $partitions\n"
>       " gpt verify mmc 0 $partitions\n"
> +     " read <interface> <dev>\n"
> +     "    - read GPT into a data structure for manipulation\n"
>       " guid <interface> <dev>\n"
>       "    - print disk GUID\n"
>       " guid <interface> <dev> <varname>\n"
> diff --git a/include/part.h b/include/part.h
> index 16c4a46..7d7052a 100644
> --- a/include/part.h
> +++ b/include/part.h
> @@ -10,6 +10,7 @@
>  #include <blk.h>
>  #include <ide.h>
>  #include <uuid.h>
> +#include <linux/list.h>
>  
>  struct block_drvr {
>       char *name;
> @@ -49,6 +50,7 @@ struct block_drvr {
>  
>  #define PART_NAME_LEN 32
>  #define PART_TYPE_LEN 32
> +#define MAX_SEARCH_PARTITIONS 16
>  
>  typedef struct disk_partition {
>       lbaint_t        start;  /* # of first block in
> partition     */ @@ -68,6 +70,12 @@ typedef struct disk_partition {
>  #endif
>  } disk_partition_t;
>  
> +struct disk_part {
> +     int partnum;
> +     disk_partition_t gpt_part_info;
> +     struct list_head list;
> +};
> +
>  /* Misc _get_dev functions */
>  #ifdef CONFIG_PARTITIONS
>  /**




Best regards,

Lukasz Majewski

--

DENX Software Engineering GmbH,      Managing Director: Wolfgang Denk
HRB 165235 Munich, Office: Kirchenstr.5, D-82194 Groebenzell, Germany
Phone: (+49)-8142-66989-10 Fax: (+49)-8142-66989-80 Email: w...@denx.de
_______________________________________________
U-Boot mailing list
U-Boot@lists.denx.de
https://lists.denx.de/listinfo/u-boot

Reply via email to