From: Patrick Delaunay <patrick.delau...@st.com>

- udpate prepare_backup_gpt_header():
  calculate partition_entry_lba and no more assumed last lba + 1

- split part_print_efi : create sub-functions
  part_print_gpt()
  part_get_disk_info()

Signed-off-by: Patrick Delaunay <patrick.delau...@st.com>
Signed-off-by: Patrick Delaunay <patrick.delauna...@gmail.com>
---

Changes in v2: None

 disk/part_efi.c | 115 ++++++++++++++++++++++++++++++++------------------------
 1 file changed, 65 insertions(+), 50 deletions(-)

diff --git a/disk/part_efi.c b/disk/part_efi.c
index 1924338..a23c8ea 100644
--- a/disk/part_efi.c
+++ b/disk/part_efi.c
@@ -10,6 +10,7 @@
  *   when CONFIG_SYS_64BIT_LBA is not defined, lbaint_t is 32 bits; this
  *   limits the maximum size of addressable storage to < 2 Terra Bytes
  */
+
 #include <asm/unaligned.h>
 #include <common.h>
 #include <command.h>
@@ -159,11 +160,12 @@ static void prepare_backup_gpt_header(gpt_header *gpt_h)
        uint64_t val;
 
        /* recalculate the values for the Backup GPT Header */
-       val = le64_to_cpu(gpt_h->my_lba);
-       gpt_h->my_lba = gpt_h->alternate_lba;
-       gpt_h->alternate_lba = cpu_to_le64(val);
-       gpt_h->partition_entry_lba =
-                       cpu_to_le64(le64_to_cpu(gpt_h->last_usable_lba) + 1);
+       val = le64_to_cpu(gpt_h->alternate_lba);
+       gpt_h->alternate_lba = gpt_h->my_lba;
+       gpt_h->my_lba = cpu_to_le64(val);
+       val -= (le32_to_cpu(gpt_h->num_partition_entries) *
+               le32_to_cpu(gpt_h->sizeof_partition_entry));
+       gpt_h->partition_entry_lba = cpu_to_le64(val);
        gpt_h->header_crc32 = 0;
 
        calc_crc32 = efi_crc32((const unsigned char *)gpt_h,
@@ -171,36 +173,16 @@ static void prepare_backup_gpt_header(gpt_header *gpt_h)
        gpt_h->header_crc32 = cpu_to_le32(calc_crc32);
 }
 
+
 #ifdef CONFIG_EFI_PARTITION
-/*
- * Public Functions (include/part.h)
- */
 
-void part_print_efi(struct blk_desc *dev_desc)
+static void part_print_gpt(gpt_header *gpt_head,
+                          gpt_entry *gpt_pte)
 {
-       ALLOC_CACHE_ALIGN_BUFFER_PAD(gpt_header, gpt_head, 1, dev_desc->blksz);
-       gpt_entry *gpt_pte = NULL;
        int i = 0;
        char uuid[37];
        unsigned char *uuid_bin;
 
-       /* This function validates AND fills in the GPT header and PTE */
-       if (is_gpt_valid(dev_desc, GPT_PRIMARY_PARTITION_TABLE_LBA,
-                        gpt_head, &gpt_pte) != 1) {
-               printf("%s: *** ERROR: Invalid GPT ***\n", __func__);
-               if (is_gpt_valid(dev_desc, (dev_desc->lba - 1),
-                                gpt_head, &gpt_pte) != 1) {
-                       printf("%s: *** ERROR: Invalid Backup GPT ***\n",
-                              __func__);
-                       return;
-               } else {
-                       printf("%s: ***        Using Backup GPT ***\n",
-                              __func__);
-               }
-       }
-
-       debug("%s: gpt-entry at %p\n", __func__, gpt_pte);
-
        printf("Part\tStart LBA\tEnd LBA\t\tName\n");
        printf("\tAttributes\n");
        printf("\tType GUID\n");
@@ -227,6 +209,60 @@ void part_print_efi(struct blk_desc *dev_desc)
                uuid_bin_to_str(uuid_bin, uuid, UUID_STR_FORMAT_GUID);
                printf("\tguid:\t%s\n", uuid);
        }
+}
+
+static void part_get_disk_info(int part, unsigned int blksz,
+                              gpt_entry *gpt_pte, disk_partition_t *info)
+{
+       /* The 'lbaint_t' casting may limit the maximum disk size to 2 TB */
+       info->start = (lbaint_t)le64_to_cpu(gpt_pte[part - 1].starting_lba);
+       /* The ending LBA is inclusive, to calculate size, add 1 to it */
+       info->size = (lbaint_t)le64_to_cpu(gpt_pte[part - 1].ending_lba) + 1
+                    - info->start;
+       info->blksz = blksz;
+
+       sprintf((char *)info->name, "%s", print_efiname(&gpt_pte[part - 1]));
+       strcpy((char *)info->type, "U-Boot");
+       info->bootable = is_bootable(&gpt_pte[part - 1]);
+#ifdef CONFIG_PARTITION_UUIDS
+       uuid_bin_to_str(gpt_pte[part - 1].unique_partition_guid.b, info->uuid,
+                       UUID_STR_FORMAT_GUID);
+#endif
+#ifdef CONFIG_PARTITION_TYPE_GUID
+       uuid_bin_to_str(gpt_pte[part - 1].partition_type_guid.b,
+                       info->type_guid, UUID_STR_FORMAT_GUID);
+#endif
+
+       debug("%s: start 0x" LBAF ", size 0x" LBAF ", name %s\n", __func__,
+             info->start, info->size, info->name);
+}
+
+/*
+ * Public Functions (include/part.h)
+ */
+void part_print_efi(struct blk_desc *dev_desc)
+{
+       ALLOC_CACHE_ALIGN_BUFFER_PAD(gpt_header, gpt_head, 1, dev_desc->blksz);
+       gpt_entry *gpt_pte = NULL;
+
+       /* This function validates AND fills in the GPT header and PTE */
+       if (is_gpt_valid(dev_desc, GPT_PRIMARY_PARTITION_TABLE_LBA,
+                        gpt_head, &gpt_pte) != 1) {
+               printf("%s: *** ERROR: Invalid GPT ***\n", __func__);
+               if (is_gpt_valid(dev_desc, (dev_desc->lba - 1),
+                                gpt_head, &gpt_pte) != 1) {
+                       printf("%s: *** ERROR: Invalid Backup GPT ***\n",
+                              __func__);
+                       return;
+               } else {
+                       printf("%s: ***        Using Backup GPT ***\n",
+                              __func__);
+               }
+       }
+
+       debug("%s: gpt-entry at %p\n", __func__, gpt_pte);
+
+       part_print_gpt(gpt_head, gpt_pte);
 
        /* Remember to free pte */
        free(gpt_pte);
@@ -268,28 +304,7 @@ int part_get_info_efi(struct blk_desc *dev_desc, int part,
                return -1;
        }
 
-       /* The 'lbaint_t' casting may limit the maximum disk size to 2 TB */
-       info->start = (lbaint_t)le64_to_cpu(gpt_pte[part - 1].starting_lba);
-       /* The ending LBA is inclusive, to calculate size, add 1 to it */
-       info->size = (lbaint_t)le64_to_cpu(gpt_pte[part - 1].ending_lba) + 1
-                    - info->start;
-       info->blksz = dev_desc->blksz;
-
-       sprintf((char *)info->name, "%s",
-                       print_efiname(&gpt_pte[part - 1]));
-       strcpy((char *)info->type, "U-Boot");
-       info->bootable = is_bootable(&gpt_pte[part - 1]);
-#ifdef CONFIG_PARTITION_UUIDS
-       uuid_bin_to_str(gpt_pte[part - 1].unique_partition_guid.b, info->uuid,
-                       UUID_STR_FORMAT_GUID);
-#endif
-#ifdef CONFIG_PARTITION_TYPE_GUID
-       uuid_bin_to_str(gpt_pte[part - 1].partition_type_guid.b,
-                       info->type_guid, UUID_STR_FORMAT_GUID);
-#endif
-
-       debug("%s: start 0x" LBAF ", size 0x" LBAF ", name %s\n", __func__,
-             info->start, info->size, info->name);
+       part_get_disk_info(part, dev_desc->blksz, gpt_pte, info);
 
        /* Remember to free pte */
        free(gpt_pte);
-- 
1.9.1

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

Reply via email to