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