efi.c 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257
  1. /*
  2. * (C) Copyright 2015 Google, Inc
  3. * Written by Simon Glass <sjg@chromium.org>
  4. *
  5. * SPDX-License-Identifier: GPL-2.0+
  6. */
  7. #include <common.h>
  8. #include <command.h>
  9. #include <efi.h>
  10. #include <errno.h>
  11. #include <malloc.h>
  12. static const char *const type_name[] = {
  13. "reserved",
  14. "loader_code",
  15. "loader_data",
  16. "bs_code",
  17. "bs_data",
  18. "rt_code",
  19. "rt_data",
  20. "conv",
  21. "unusable",
  22. "acpi_reclaim",
  23. "acpi_nvs",
  24. "io",
  25. "io_port",
  26. "pal_code",
  27. };
  28. static struct attr_info {
  29. int shift;
  30. const char *name;
  31. } mem_attr[] = {
  32. { EFI_MEMORY_UC_SHIFT, "uncached" },
  33. { EFI_MEMORY_WC_SHIFT, "write-coalescing" },
  34. { EFI_MEMORY_WT_SHIFT, "write-through" },
  35. { EFI_MEMORY_WB_SHIFT, "write-back" },
  36. { EFI_MEMORY_UCE_SHIFT, "uncached & exported" },
  37. { EFI_MEMORY_WP_SHIFT, "write-protect" },
  38. { EFI_MEMORY_RP_SHIFT, "read-protect" },
  39. { EFI_MEMORY_XP_SHIFT, "execute-protect" },
  40. { EFI_MEMORY_RUNTIME_SHIFT, "needs runtime mapping" }
  41. };
  42. /* Maximum different attribute values we can track */
  43. #define ATTR_SEEN_MAX 30
  44. static inline bool is_boot_services(int type)
  45. {
  46. return type == EFI_LOADER_CODE || type == EFI_LOADER_DATA ||
  47. type == EFI_BOOT_SERVICES_CODE ||
  48. type == EFI_BOOT_SERVICES_DATA;
  49. }
  50. static int h_cmp_entry(const void *v1, const void *v2)
  51. {
  52. const struct efi_mem_desc *desc1 = v1;
  53. const struct efi_mem_desc *desc2 = v2;
  54. int64_t diff = desc1->physical_start - desc2->physical_start;
  55. /*
  56. * Manually calculate the difference to avoid sign loss in the 64-bit
  57. * to 32-bit conversion
  58. */
  59. return diff < 0 ? -1 : diff > 0 ? 1 : 0;
  60. }
  61. void *efi_build_mem_table(struct efi_entry_memmap *map, int size, bool skip_bs)
  62. {
  63. struct efi_mem_desc *desc, *end, *base, *dest, *prev;
  64. int count;
  65. u64 addr;
  66. base = malloc(size + sizeof(*desc));
  67. if (!base) {
  68. debug("%s: Cannot allocate %#x bytes\n", __func__, size);
  69. return NULL;
  70. }
  71. end = (struct efi_mem_desc *)((ulong)map + size);
  72. count = ((ulong)end - (ulong)map->desc) / map->desc_size;
  73. memcpy(base, map->desc, (ulong)end - (ulong)map->desc);
  74. qsort(base, count, map->desc_size, h_cmp_entry);
  75. prev = NULL;
  76. addr = 0;
  77. dest = base;
  78. end = base + count;
  79. for (desc = base; desc < end; desc = efi_get_next_mem_desc(map, desc)) {
  80. bool merge = true;
  81. int type = desc->type;
  82. if (skip_bs && is_boot_services(desc->type))
  83. type = EFI_CONVENTIONAL_MEMORY;
  84. memcpy(dest, desc, map->desc_size);
  85. dest->type = type;
  86. if (!skip_bs || !prev)
  87. merge = false;
  88. else if (desc->physical_start != addr)
  89. merge = false;
  90. else if (type != EFI_CONVENTIONAL_MEMORY)
  91. merge = false;
  92. else if (prev->type != EFI_CONVENTIONAL_MEMORY)
  93. merge = false;
  94. if (merge) {
  95. prev->num_pages += desc->num_pages;
  96. } else {
  97. prev = dest;
  98. dest = efi_get_next_mem_desc(map, dest);
  99. }
  100. addr = desc->physical_start + (desc->num_pages <<
  101. EFI_PAGE_SHIFT);
  102. }
  103. /* Mark the end */
  104. dest->type = EFI_TABLE_END;
  105. return base;
  106. }
  107. static void efi_print_mem_table(struct efi_entry_memmap *map,
  108. struct efi_mem_desc *desc, bool skip_bs)
  109. {
  110. u64 attr_seen[ATTR_SEEN_MAX];
  111. int attr_seen_count;
  112. int upto, i;
  113. u64 addr;
  114. printf(" # %-14s %10s %10s %10s %s\n", "Type", "Physical",
  115. "Virtual", "Size", "Attributes");
  116. /* Keep track of all the different attributes we have seen */
  117. attr_seen_count = 0;
  118. addr = 0;
  119. for (upto = 0; desc->type != EFI_TABLE_END;
  120. upto++, desc = efi_get_next_mem_desc(map, desc)) {
  121. const char *name;
  122. u64 size;
  123. if (skip_bs && is_boot_services(desc->type))
  124. continue;
  125. if (desc->physical_start != addr) {
  126. printf(" %-14s %010llx %10s %010llx\n", "<gap>",
  127. addr, "", desc->physical_start - addr);
  128. }
  129. size = desc->num_pages << EFI_PAGE_SHIFT;
  130. name = desc->type < ARRAY_SIZE(type_name) ?
  131. type_name[desc->type] : "<invalid>";
  132. printf("%2d %x:%-12s %010llx %010llx %010llx ", upto,
  133. desc->type, name, desc->physical_start,
  134. desc->virtual_start, size);
  135. if (desc->attribute & EFI_MEMORY_RUNTIME)
  136. putc('r');
  137. printf("%llx", desc->attribute & ~EFI_MEMORY_RUNTIME);
  138. putc('\n');
  139. for (i = 0; i < attr_seen_count; i++) {
  140. if (attr_seen[i] == desc->attribute)
  141. break;
  142. }
  143. if (i == attr_seen_count && i < ATTR_SEEN_MAX)
  144. attr_seen[attr_seen_count++] = desc->attribute;
  145. addr = desc->physical_start + size;
  146. }
  147. printf("\nAttributes key:\n");
  148. for (i = 0; i < attr_seen_count; i++) {
  149. u64 attr = attr_seen[i];
  150. bool first;
  151. int j;
  152. printf("%c%llx: ", attr & EFI_MEMORY_RUNTIME ? 'r' : ' ',
  153. attr & ~EFI_MEMORY_RUNTIME);
  154. for (j = 0, first = true; j < ARRAY_SIZE(mem_attr); j++) {
  155. if (attr & (1ULL << mem_attr[j].shift)) {
  156. if (first)
  157. first = false;
  158. else
  159. printf(", ");
  160. printf("%s", mem_attr[j].name);
  161. }
  162. }
  163. putc('\n');
  164. }
  165. if (skip_bs)
  166. printf("*Some areas are merged (use 'all' to see)\n");
  167. }
  168. static int do_efi_mem(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  169. {
  170. struct efi_mem_desc *desc;
  171. struct efi_entry_memmap *map;
  172. int size, ret;
  173. bool skip_bs;
  174. skip_bs = !argc || *argv[0] != 'a';
  175. ret = efi_info_get(EFIET_MEMORY_MAP, (void **)&map, &size);
  176. switch (ret) {
  177. case -ENOENT:
  178. printf("No EFI table available\n");
  179. goto done;
  180. case -EPROTONOSUPPORT:
  181. printf("Incorrect EFI table version\n");
  182. goto done;
  183. }
  184. printf("EFI table at %lx, memory map %p, size %x, version %x, descr. size %#x\n",
  185. gd->arch.table, map, size, map->version, map->desc_size);
  186. if (map->version != EFI_MEM_DESC_VERSION) {
  187. printf("Incorrect memory map version\n");
  188. ret = -EPROTONOSUPPORT;
  189. goto done;
  190. }
  191. desc = efi_build_mem_table(map, size, skip_bs);
  192. if (!desc) {
  193. ret = -ENOMEM;
  194. goto done;
  195. }
  196. efi_print_mem_table(map, desc, skip_bs);
  197. free(desc);
  198. done:
  199. if (ret)
  200. printf("Error: %d\n", ret);
  201. return ret ? CMD_RET_FAILURE : 0;
  202. }
  203. static cmd_tbl_t efi_commands[] = {
  204. U_BOOT_CMD_MKENT(mem, 1, 1, do_efi_mem, "", ""),
  205. };
  206. static int do_efi(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  207. {
  208. cmd_tbl_t *efi_cmd;
  209. int ret;
  210. if (argc < 2)
  211. return CMD_RET_USAGE;
  212. efi_cmd = find_cmd_tbl(argv[1], efi_commands, ARRAY_SIZE(efi_commands));
  213. argc -= 2;
  214. argv += 2;
  215. if (!efi_cmd || argc > efi_cmd->maxargs)
  216. return CMD_RET_USAGE;
  217. ret = efi_cmd->cmd(efi_cmd, flag, argc, argv);
  218. return cmd_process_error(efi_cmd, ret);
  219. }
  220. U_BOOT_CMD(
  221. efi, 3, 1, do_efi,
  222. "EFI access",
  223. "mem [all] Dump memory information [include boot services]"
  224. );