efi.c 7.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281
  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 <log.h>
  11. #include <malloc.h>
  12. #include <sort.h>
  13. static const char *const type_name[] = {
  14. "reserved",
  15. "loader_code",
  16. "loader_data",
  17. "bs_code",
  18. "bs_data",
  19. "rt_code",
  20. "rt_data",
  21. "conv",
  22. "unusable",
  23. "acpi_reclaim",
  24. "acpi_nvs",
  25. "io",
  26. "io_port",
  27. "pal_code",
  28. };
  29. static struct attr_info {
  30. u64 val;
  31. const char *name;
  32. } mem_attr[] = {
  33. { EFI_MEMORY_UC, "uncached" },
  34. { EFI_MEMORY_WC, "write-coalescing" },
  35. { EFI_MEMORY_WT, "write-through" },
  36. { EFI_MEMORY_WB, "write-back" },
  37. { EFI_MEMORY_UCE, "uncached & exported" },
  38. { EFI_MEMORY_WP, "write-protect" },
  39. { EFI_MEMORY_RP, "read-protect" },
  40. { EFI_MEMORY_XP, "execute-protect" },
  41. { EFI_MEMORY_NV, "non-volatile" },
  42. { EFI_MEMORY_MORE_RELIABLE, "higher reliability" },
  43. { EFI_MEMORY_RO, "read-only" },
  44. { EFI_MEMORY_SP, "specific purpose" },
  45. { EFI_MEMORY_RUNTIME, "needs runtime mapping" }
  46. };
  47. /* Maximum different attribute values we can track */
  48. #define ATTR_SEEN_MAX 30
  49. static inline bool is_boot_services(int type)
  50. {
  51. return type == EFI_LOADER_CODE || type == EFI_LOADER_DATA ||
  52. type == EFI_BOOT_SERVICES_CODE ||
  53. type == EFI_BOOT_SERVICES_DATA;
  54. }
  55. static int h_cmp_entry(const void *v1, const void *v2)
  56. {
  57. const struct efi_mem_desc *desc1 = v1;
  58. const struct efi_mem_desc *desc2 = v2;
  59. int64_t diff = desc1->physical_start - desc2->physical_start;
  60. /*
  61. * Manually calculate the difference to avoid sign loss in the 64-bit
  62. * to 32-bit conversion
  63. */
  64. return diff < 0 ? -1 : diff > 0 ? 1 : 0;
  65. }
  66. /**
  67. * efi_build_mem_table() - make a sorted copy of the memory table
  68. *
  69. * @map: Pointer to EFI memory map table
  70. * @size: Size of table in bytes
  71. * @skip_bs: True to skip boot-time memory and merge it with conventional
  72. * memory. This will significantly reduce the number of table
  73. * entries.
  74. * Return: pointer to the new table. It should be freed with free() by the
  75. * caller.
  76. */
  77. static void *efi_build_mem_table(struct efi_entry_memmap *map, int size,
  78. bool skip_bs)
  79. {
  80. struct efi_mem_desc *desc, *end, *base, *dest, *prev;
  81. int count;
  82. u64 addr;
  83. base = malloc(size + sizeof(*desc));
  84. if (!base) {
  85. debug("%s: Cannot allocate %#x bytes\n", __func__, size);
  86. return NULL;
  87. }
  88. end = (struct efi_mem_desc *)((ulong)map + size);
  89. count = ((ulong)end - (ulong)map->desc) / map->desc_size;
  90. memcpy(base, map->desc, (ulong)end - (ulong)map->desc);
  91. qsort(base, count, map->desc_size, h_cmp_entry);
  92. prev = NULL;
  93. addr = 0;
  94. dest = base;
  95. end = (struct efi_mem_desc *)((ulong)base + count * map->desc_size);
  96. for (desc = base; desc < end; desc = efi_get_next_mem_desc(map, desc)) {
  97. bool merge = true;
  98. u32 type = desc->type;
  99. if (type >= EFI_MAX_MEMORY_TYPE) {
  100. printf("Memory map contains invalid entry type %u\n",
  101. type);
  102. continue;
  103. }
  104. if (skip_bs && is_boot_services(desc->type))
  105. type = EFI_CONVENTIONAL_MEMORY;
  106. memcpy(dest, desc, map->desc_size);
  107. dest->type = type;
  108. if (!skip_bs || !prev)
  109. merge = false;
  110. else if (desc->physical_start != addr)
  111. merge = false;
  112. else if (type != EFI_CONVENTIONAL_MEMORY)
  113. merge = false;
  114. else if (prev->type != EFI_CONVENTIONAL_MEMORY)
  115. merge = false;
  116. if (merge) {
  117. prev->num_pages += desc->num_pages;
  118. } else {
  119. prev = dest;
  120. dest = efi_get_next_mem_desc(map, dest);
  121. }
  122. addr = desc->physical_start + (desc->num_pages <<
  123. EFI_PAGE_SHIFT);
  124. }
  125. /* Mark the end */
  126. dest->type = EFI_MAX_MEMORY_TYPE;
  127. return base;
  128. }
  129. static void efi_print_mem_table(struct efi_entry_memmap *map,
  130. struct efi_mem_desc *desc, bool skip_bs)
  131. {
  132. u64 attr_seen[ATTR_SEEN_MAX];
  133. int attr_seen_count;
  134. int upto, i;
  135. u64 addr;
  136. printf(" # %-14s %10s %10s %10s %s\n", "Type", "Physical",
  137. "Virtual", "Size", "Attributes");
  138. /* Keep track of all the different attributes we have seen */
  139. attr_seen_count = 0;
  140. addr = 0;
  141. for (upto = 0; desc->type != EFI_MAX_MEMORY_TYPE;
  142. upto++, desc = efi_get_next_mem_desc(map, desc)) {
  143. const char *name;
  144. u64 size;
  145. if (skip_bs && is_boot_services(desc->type))
  146. continue;
  147. if (desc->physical_start != addr) {
  148. printf(" %-14s %010llx %10s %010llx\n", "<gap>",
  149. addr, "", desc->physical_start - addr);
  150. }
  151. size = desc->num_pages << EFI_PAGE_SHIFT;
  152. name = desc->type < ARRAY_SIZE(type_name) ?
  153. type_name[desc->type] : "<invalid>";
  154. printf("%2d %x:%-12s %010llx %010llx %010llx ", upto,
  155. desc->type, name, desc->physical_start,
  156. desc->virtual_start, size);
  157. if (desc->attribute & EFI_MEMORY_RUNTIME)
  158. putc('r');
  159. printf("%llx", desc->attribute & ~EFI_MEMORY_RUNTIME);
  160. putc('\n');
  161. for (i = 0; i < attr_seen_count; i++) {
  162. if (attr_seen[i] == desc->attribute)
  163. break;
  164. }
  165. if (i == attr_seen_count && i < ATTR_SEEN_MAX)
  166. attr_seen[attr_seen_count++] = desc->attribute;
  167. addr = desc->physical_start + size;
  168. }
  169. printf("\nAttributes key:\n");
  170. for (i = 0; i < attr_seen_count; i++) {
  171. u64 attr = attr_seen[i];
  172. bool first;
  173. int j;
  174. printf("%c%llx: ", (attr & EFI_MEMORY_RUNTIME) ? 'r' : ' ',
  175. attr & ~EFI_MEMORY_RUNTIME);
  176. for (j = 0, first = true; j < ARRAY_SIZE(mem_attr); j++) {
  177. if (attr & mem_attr[j].val) {
  178. if (first)
  179. first = false;
  180. else
  181. printf(", ");
  182. printf("%s", mem_attr[j].name);
  183. }
  184. }
  185. putc('\n');
  186. }
  187. if (skip_bs)
  188. printf("*Some areas are merged (use 'all' to see)\n");
  189. }
  190. static int do_efi_mem(struct cmd_tbl *cmdtp, int flag, int argc,
  191. char *const argv[])
  192. {
  193. struct efi_mem_desc *desc;
  194. struct efi_entry_memmap *map;
  195. int size, ret;
  196. bool skip_bs;
  197. skip_bs = !argc || *argv[0] != 'a';
  198. ret = efi_info_get(EFIET_MEMORY_MAP, (void **)&map, &size);
  199. switch (ret) {
  200. case -ENOENT:
  201. printf("No EFI table available\n");
  202. goto done;
  203. case -EPROTONOSUPPORT:
  204. printf("Incorrect EFI table version\n");
  205. goto done;
  206. }
  207. printf("EFI table at %lx, memory map %p, size %x, version %x, descr. size %#x\n",
  208. gd->arch.table, map, size, map->version, map->desc_size);
  209. if (map->version != EFI_MEM_DESC_VERSION) {
  210. printf("Incorrect memory map version\n");
  211. ret = -EPROTONOSUPPORT;
  212. goto done;
  213. }
  214. desc = efi_build_mem_table(map, size, skip_bs);
  215. if (!desc) {
  216. ret = -ENOMEM;
  217. goto done;
  218. }
  219. efi_print_mem_table(map, desc, skip_bs);
  220. free(desc);
  221. done:
  222. if (ret)
  223. printf("Error: %d\n", ret);
  224. return ret ? CMD_RET_FAILURE : 0;
  225. }
  226. static struct cmd_tbl efi_commands[] = {
  227. U_BOOT_CMD_MKENT(mem, 1, 1, do_efi_mem, "", ""),
  228. };
  229. static int do_efi(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
  230. {
  231. struct cmd_tbl *efi_cmd;
  232. int ret;
  233. if (argc < 2)
  234. return CMD_RET_USAGE;
  235. efi_cmd = find_cmd_tbl(argv[1], efi_commands, ARRAY_SIZE(efi_commands));
  236. argc -= 2;
  237. argv += 2;
  238. if (!efi_cmd || argc > efi_cmd->maxargs)
  239. return CMD_RET_USAGE;
  240. ret = efi_cmd->cmd(efi_cmd, flag, argc, argv);
  241. return cmd_process_error(efi_cmd, ret);
  242. }
  243. U_BOOT_CMD(
  244. efi, 3, 1, do_efi,
  245. "EFI access",
  246. "mem [all] Dump memory information [include boot services]"
  247. );