efi.c 6.3 KB

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