1 // SPDX-License-Identifier: GPL-2.0+
5 * Copyright (c) 2018 Heinrich Schuchardt <xypron.glpk@gmx.de>
7 * This unit test checks the following runtime services:
8 * AllocatePages, FreePages, GetMemoryMap
10 * The memory type used for the device tree is checked.
13 #include <efi_selftest.h>
15 #define EFI_ST_NUM_PAGES 8
17 static const efi_guid_t fdt_guid = EFI_FDT_GUID;
18 static struct efi_boot_services *boottime;
22 * setup() - setup unit test
24 * @handle: handle of the loaded image
25 * @systable: system table
26 * Return: EFI_ST_SUCCESS for success
28 static int setup(const efi_handle_t handle,
29 const struct efi_system_table *systable)
33 boottime = systable->boottime;
35 for (i = 0; i < systable->nr_tables; ++i) {
36 if (!efi_st_memcmp(&systable->tables[i].guid, &fdt_guid,
37 sizeof(efi_guid_t))) {
39 efi_st_error("Duplicate device tree\n");
40 return EFI_ST_FAILURE;
42 fdt_addr = (uintptr_t)systable->tables[i].table;
45 return EFI_ST_SUCCESS;
49 * find_in_memory_map() - check matching memory map entry exists
51 * @memory_map: memory map
52 * @desc_size: number of memory map entries
53 * @addr: physical address to find in the map
54 * @type: expected memory type
55 * Return: EFI_ST_SUCCESS for success
57 static int find_in_memory_map(efi_uintn_t map_size,
58 struct efi_mem_desc *memory_map,
59 efi_uintn_t desc_size,
60 u64 addr, int memory_type)
65 for (i = 0; map_size; ++i, map_size -= desc_size) {
66 struct efi_mem_desc *entry = &memory_map[i];
68 if (addr >= entry->physical_start &&
69 addr < entry->physical_start +
70 (entry->num_pages << EFI_PAGE_SHIFT)) {
72 efi_st_error("Duplicate memory map entry\n");
73 return EFI_ST_FAILURE;
76 if (memory_type != entry->type) {
78 ("Wrong memory type %d, expected %d\n",
79 entry->type, memory_type);
80 return EFI_ST_FAILURE;
85 efi_st_error("Missing memory map entry\n");
86 return EFI_ST_FAILURE;
88 return EFI_ST_SUCCESS;
92 * execute() - execute unit test
94 * Return: EFI_ST_SUCCESS for success
96 static int execute(void)
100 efi_uintn_t map_size = 0;
102 efi_uintn_t desc_size;
104 struct efi_mem_desc *memory_map;
107 /* Allocate two page ranges with different memory type */
108 ret = boottime->allocate_pages(EFI_ALLOCATE_ANY_PAGES,
109 EFI_RUNTIME_SERVICES_CODE,
110 EFI_ST_NUM_PAGES, &p1);
111 if (ret != EFI_SUCCESS) {
112 efi_st_error("AllocatePages did not return EFI_SUCCESS\n");
113 return EFI_ST_FAILURE;
115 ret = boottime->allocate_pages(EFI_ALLOCATE_ANY_PAGES,
116 EFI_RUNTIME_SERVICES_DATA,
117 EFI_ST_NUM_PAGES, &p2);
118 if (ret != EFI_SUCCESS) {
119 efi_st_error("AllocatePages did not return EFI_SUCCESS\n");
120 return EFI_ST_FAILURE;
123 /* Load memory map */
124 ret = boottime->get_memory_map(&map_size, NULL, &map_key, &desc_size,
126 if (ret != EFI_BUFFER_TOO_SMALL) {
128 ("GetMemoryMap did not return EFI_BUFFER_TOO_SMALL\n");
129 return EFI_ST_FAILURE;
131 /* Allocate extra space for newly allocated memory */
132 map_size += sizeof(struct efi_mem_desc);
133 ret = boottime->allocate_pool(EFI_BOOT_SERVICES_DATA, map_size,
134 (void **)&memory_map);
135 if (ret != EFI_SUCCESS) {
136 efi_st_error("AllocatePool did not return EFI_SUCCESS\n");
137 return EFI_ST_FAILURE;
139 ret = boottime->get_memory_map(&map_size, memory_map, &map_key,
140 &desc_size, &desc_version);
141 if (ret != EFI_SUCCESS) {
142 efi_st_error("GetMemoryMap did not return EFI_SUCCESS\n");
143 return EFI_ST_FAILURE;
146 /* Check memory map entries */
147 if (find_in_memory_map(map_size, memory_map, desc_size, p1,
148 EFI_RUNTIME_SERVICES_CODE) != EFI_ST_SUCCESS)
149 return EFI_ST_FAILURE;
150 if (find_in_memory_map(map_size, memory_map, desc_size, p2,
151 EFI_RUNTIME_SERVICES_DATA) != EFI_ST_SUCCESS)
152 return EFI_ST_FAILURE;
155 ret = boottime->free_pages(p1, EFI_ST_NUM_PAGES);
156 if (ret != EFI_SUCCESS) {
157 efi_st_error("FreePages did not return EFI_SUCCESS\n");
158 return EFI_ST_FAILURE;
160 ret = boottime->free_pages(p2, EFI_ST_NUM_PAGES);
161 if (ret != EFI_SUCCESS) {
162 efi_st_error("FreePages did not return EFI_SUCCESS\n");
163 return EFI_ST_FAILURE;
165 ret = boottime->free_pool(memory_map);
166 if (ret != EFI_SUCCESS) {
167 efi_st_error("FreePool did not return EFI_SUCCESS\n");
168 return EFI_ST_FAILURE;
171 /* Check memory reservation for the device tree */
173 find_in_memory_map(map_size, memory_map, desc_size, fdt_addr,
174 EFI_RUNTIME_SERVICES_DATA) != EFI_ST_SUCCESS) {
176 ("Device tree not marked as runtime services data\n");
177 return EFI_ST_FAILURE;
179 return EFI_ST_SUCCESS;
182 EFI_UNIT_TEST(memory) = {
184 .phase = EFI_EXECUTE_BEFORE_BOOTTIME_EXIT,