Re: [U-Boot] [PATCH v2 41/47] efi: Add a command to display the memory map

On Fri, Jul 31, 2015 at 11:31 PM, Simon Glass sjg@chromium.org wrote:

The EFI memory map is passed from the stub to U-Boot in a table. Add a command to display it in a vaguely readable fashion.

Signed-off-by: Simon Glass sjg@chromium.org

Changes in v2:

• Drop a left-over debug printf()
• Fix alignment of region index field in the output
• Fix efi_mem_desc_VERSION typo
• Output the region index in decimal

common/Makefile | 1 + common/cmd_efi.c | 257 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 258 insertions(+) create mode 100644 common/cmd_efi.c

diff --git a/common/Makefile b/common/Makefile index d6c1d48..6dc4c89 100644 --- a/common/Makefile +++ b/common/Makefile @@ -90,6 +90,7 @@ obj-$(CONFIG_CMD_DTT) += cmd_dtt.o obj-$(CONFIG_CMD_ECHO) += cmd_echo.o obj-$(CONFIG_ENV_IS_IN_EEPROM) += cmd_eeprom.o obj-$(CONFIG_CMD_EEPROM) += cmd_eeprom.o +obj-$(CONFIG_EFI_STUB) += cmd_efi.o obj-$(CONFIG_CMD_ELF) += cmd_elf.o obj-$(CONFIG_SYS_HUSH_PARSER) += cmd_exit.o obj-$(CONFIG_CMD_EXT4) += cmd_ext4.o diff --git a/common/cmd_efi.c b/common/cmd_efi.c new file mode 100644 index 0000000..c76296e --- /dev/null +++ b/common/cmd_efi.c @@ -0,0 +1,257 @@ +/*

• • (C) Copyright 2015 Google, Inc
• • Written by Simon Glass sjg@chromium.org
• • SPDX-License-Identifier: GPL-2.0+
• */

+#include <common.h> +#include <command.h> +#include <efi.h> +#include <errno.h> +#include <malloc.h>

+static const char *const type_name[] = {

•   "reserved",
  

•   "loader_code",
  

•   "loader_data",
  

•   "bs_code",
  

•   "bs_data",
  

•   "rt_code",
  

•   "rt_data",
  

•   "conv",
  

•   "unusable",
  

•   "acpi_reclaim",
  

•   "acpi_nvs",
  

•   "io",
  

•   "io_port",
  

•   "pal_code",
  

+};

+static struct attr_info {

•   int shift;
  

•   const char *name;
  

+} mem_attr[] = {

•   { EFI_MEMORY_UC_SHIFT, "uncached" },
  

•   { EFI_MEMORY_WC_SHIFT, "write-coalescing" },
  

•   { EFI_MEMORY_WT_SHIFT, "write-through" },
  

•   { EFI_MEMORY_WB_SHIFT, "write-back" },
  

•   { EFI_MEMORY_UCE_SHIFT, "uncached & exported" },
  

•   { EFI_MEMORY_WP_SHIFT, "write-protect" },
  

•   { EFI_MEMORY_RP_SHIFT, "read-protect" },
  

•   { EFI_MEMORY_XP_SHIFT, "execute-protect" },
  

•   { EFI_MEMORY_RUNTIME_SHIFT, "needs runtime mapping" }
  

+};

+/* Maximum different attribute values we can track */ +#define ATTR_SEEN_MAX 30

+static inline bool is_boot_services(int type) +{

•   return type == EFI_LOADER_CODE || type == EFI_LOADER_DATA ||
  

•           type == EFI_BOOT_SERVICES_CODE ||
  

•           type == EFI_BOOT_SERVICES_DATA;
  

+}

+static int h_cmp_entry(const void *v1, const void *v2) +{

•   const struct efi_mem_desc *desc1 = v1;
  

•   const struct efi_mem_desc *desc2 = v2;
  

•   int64_t diff = desc1->physical_start - desc2->physical_start;
  

•   /*
  

•    * Manually calculate the difference to avoid sign loss in the 64-bit
  

•    * to 32-bit conversion
  

•    */
  

•   return diff < 0 ? -1 : diff > 0 ? 1 : 0;
  

+}

+void *efi_build_mem_table(struct efi_entry_memmap *map, int size, bool skip_bs) +{

•   struct efi_mem_desc *desc, *end, *base, *dest, *prev;
  

•   int count;
  

•   u64 addr;
  

•   base = malloc(size + sizeof(*desc));
  

•   if (!base) {
  

•           debug("%s: Cannot allocate %#x bytes\n", __func__, size);
  

•           return NULL;
  

•   }
  

•   end = (struct efi_mem_desc *)((ulong)map + size);
  

•   count = ((ulong)end - (ulong)map->desc) / map->desc_size;
  

•   memcpy(base, map->desc, (ulong)end - (ulong)map->desc);
  

•   qsort(base, count, map->desc_size, h_cmp_entry);
  

•   prev = NULL;
  

•   addr = 0;
  

•   dest = base;
  

•   end = base + count;
  

•   for (desc = base; desc < end; desc = efi_get_next_mem_desc(map, desc)) {
  

•           bool merge = true;
  

•           int type = desc->type;
  

•           if (skip_bs && is_boot_services(desc->type))
  

•                   type = EFI_CONVENTIONAL_MEMORY;
  

•           memcpy(dest, desc, map->desc_size);
  

•           dest->type = type;
  

•           if (!skip_bs || !prev)
  

•                   merge = false;
  

•           else if (desc->physical_start != addr)
  

•                   merge = false;
  

•           else if (type != EFI_CONVENTIONAL_MEMORY)
  

•                   merge = false;
  

•           else if (prev->type != EFI_CONVENTIONAL_MEMORY)
  

•                   merge = false;
  

•           if (merge) {
  

•                   prev->num_pages += desc->num_pages;
  

•           } else {
  

•                   prev = dest;
  

•                   dest = efi_get_next_mem_desc(map, dest);
  

•           }
  

•           addr = desc->physical_start + (desc->num_pages <<
  

•                           EFI_PAGE_SHIFT);
  

•   }
  

•   /* Mark the end */
  

•   dest->type = EFI_TABLE_END;
  

•   return base;
  

+}

+static void efi_print_mem_table(struct efi_entry_memmap *map,

•                           struct efi_mem_desc *desc, bool skip_bs)
  

+{

•   u64 attr_seen[ATTR_SEEN_MAX];
  

•   int attr_seen_count;
  

•   int upto, i;
  

•   u64 addr;
  

•   printf(" #  %-14s  %10s  %10s  %10s  %s\n", "Type", "Physical",
  

•          "Virtual", "Size", "Attributes");
  

•   /* Keep track of all the different attributes we have seen */
  

•   attr_seen_count = 0;
  

•   addr = 0;
  

•   for (upto = 0; desc->type != EFI_TABLE_END;
  

•        upto++, desc = efi_get_next_mem_desc(map, desc)) {
  

•           const char *name;
  

•           u64 size;
  

•           if (skip_bs && is_boot_services(desc->type))
  

•                   continue;
  

•           if (desc->physical_start != addr) {
  

•                   printf("    %-14s  %010llx  %10s  %010llx\n", "<gap>",
  

•                          addr, "", desc->physical_start - addr);
  

•           }
  

•           size = desc->num_pages << EFI_PAGE_SHIFT;
  

•           name = desc->type < ARRAY_SIZE(type_name) ?
  

•                           type_name[desc->type] : "<invalid>";
  

•           printf("%2d  %x:%-12s  %010llx  %010llx  %010llx  ", upto,
  

•                  desc->type, name, desc->physical_start,
  

•                  desc->virtual_start, size);
  

•           if (desc->attribute & EFI_MEMORY_RUNTIME)
  

•                   putc('r');
  

•           printf("%llx", desc->attribute & ~EFI_MEMORY_RUNTIME);
  

•           putc('\n');
  

•           for (i = 0; i < attr_seen_count; i++) {
  

•                   if (attr_seen[i] == desc->attribute)
  

•                           break;
  

•           }
  

•           if (i == attr_seen_count && i < ATTR_SEEN_MAX)
  

•                   attr_seen[attr_seen_count++] = desc->attribute;
  

•           addr = desc->physical_start + size;
  

•   }
  

•   printf("\nAttributes key:\n");
  

•   for (i = 0; i < attr_seen_count; i++) {
  

•           u64 attr = attr_seen[i];
  

•           bool first;
  

•           int j;
  

•           printf("%c%llx: ", attr & EFI_MEMORY_RUNTIME ? 'r' : ' ',
  

•                  attr & ~EFI_MEMORY_RUNTIME);
  

•           for (j = 0, first = true; j < ARRAY_SIZE(mem_attr); j++) {
  

•                   if (attr & (1ULL << mem_attr[j].shift)) {
  

•                           if (first)
  

•                                   first = false;
  

•                           else
  

•                                   printf(", ");
  

•                           printf("%s", mem_attr[j].name);
  

•                   }
  

•           }
  

•           putc('\n');
  

•   }
  

•   if (skip_bs)
  

•           printf("*Some areas are merged (use 'all' to see)\n");
  

+}

+static int do_efi_mem(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) +{

•   struct efi_mem_desc *desc;
  

•   struct efi_entry_memmap *map;
  

•   int size, ret;
  

•   bool skip_bs;
  

•   skip_bs = !argc || *argv[0] != 'a';
  

•   ret = efi_info_get(EFIET_MEMORY_MAP, (void **)&map, &size);
  

•   switch (ret) {
  

•   case -ENOENT:
  

•           printf("No EFI table available\n");
  

•           goto done;
  

•   case -EPROTONOSUPPORT:
  

•           printf("Incorrect EFI table version\n");
  

•           goto done;
  

•   }
  

•   printf("EFI table at %lx, memory map %p, size %x, version %x, descr. size %#x\n",
  

•          gd->arch.table, map, size, map->version, map->desc_size);
  

•   if (map->version != EFI_MEM_DESC_VERSION) {
  

•           printf("Incorrect memory map version\n");
  

•           ret = -EPROTONOSUPPORT;
  

•           goto done;
  

•   }
  

•   desc = efi_build_mem_table(map, size, skip_bs);
  

•   if (!desc) {
  

•           ret = -ENOMEM;
  

•           goto done;
  

•   }
  

•   efi_print_mem_table(map, desc, skip_bs);
  

•   free(desc);
  

+done:

•   if (ret)
  

•           printf("Error: %d\n", ret);
  

•   return ret ? CMD_RET_FAILURE : 0;
  

+}

+static cmd_tbl_t efi_commands[] = {

•   U_BOOT_CMD_MKENT(mem, 1, 1, do_efi_mem, "", ""),
  

+};

+static int do_efi(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) +{

•   cmd_tbl_t *efi_cmd;
  

•   int ret;
  

•   if (argc < 2)
  

•           return CMD_RET_USAGE;
  

•   efi_cmd = find_cmd_tbl(argv[1], efi_commands, ARRAY_SIZE(efi_commands));
  

•   argc -= 2;
  

•   argv += 2;
  

•   if (!efi_cmd || argc > efi_cmd->maxargs)
  

•           return CMD_RET_USAGE;
  

•   ret = efi_cmd->cmd(efi_cmd, flag, argc, argv);
  

•   return cmd_process_error(efi_cmd, ret);
  

+}

+U_BOOT_CMD(

•   efi,     3,      1,      do_efi,
  

•   "EFI access",
  

•   "mem [all]        Dump memory information [include boot services]"
  

+);

Reviewed-by: Bin Meng bmeng.cn@gmail.com

Tested on QEMU Tested-by: Bin Meng bmeng.cn@gmail.com