123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192 |
- // SPDX-License-Identifier: GPL-2.0+
- /*
- * efi_selftest_memory
- *
- * Copyright (c) 2018 Heinrich Schuchardt <xypron.glpk@gmx.de>
- *
- * This unit test checks the following boottime services:
- * AllocatePages, FreePages, GetMemoryMap
- *
- * The memory type used for the device tree is checked.
- */
- #include <efi_selftest.h>
- #define EFI_ST_NUM_PAGES 8
- static const efi_guid_t fdt_guid = EFI_FDT_GUID;
- static struct efi_boot_services *boottime;
- static u64 fdt_addr;
- /**
- * setup() - setup unit test
- *
- * @handle: handle of the loaded image
- * @systable: system table
- * Return: EFI_ST_SUCCESS for success
- */
- static int setup(const efi_handle_t handle,
- const struct efi_system_table *systable)
- {
- size_t i;
- boottime = systable->boottime;
- for (i = 0; i < systable->nr_tables; ++i) {
- if (!memcmp(&systable->tables[i].guid, &fdt_guid,
- sizeof(efi_guid_t))) {
- if (fdt_addr) {
- efi_st_error("Duplicate device tree\n");
- return EFI_ST_FAILURE;
- }
- fdt_addr = (uintptr_t)systable->tables[i].table;
- }
- }
- return EFI_ST_SUCCESS;
- }
- /**
- * find_in_memory_map() - check matching memory map entry exists
- *
- * @memory_map: memory map
- * @desc_size: number of memory map entries
- * @addr: physical address to find in the map
- * @type: expected memory type
- * Return: EFI_ST_SUCCESS for success
- */
- static int find_in_memory_map(efi_uintn_t map_size,
- struct efi_mem_desc *memory_map,
- efi_uintn_t desc_size,
- u64 addr, int memory_type)
- {
- efi_uintn_t i;
- bool found = false;
- for (i = 0; map_size; ++i, map_size -= desc_size) {
- struct efi_mem_desc *entry = &memory_map[i];
- if (entry->physical_start != entry->virtual_start) {
- efi_st_error("Physical and virtual addresses do not match\n");
- return EFI_ST_FAILURE;
- }
- if (addr >= entry->physical_start &&
- addr < entry->physical_start +
- (entry->num_pages << EFI_PAGE_SHIFT)) {
- if (found) {
- efi_st_error("Duplicate memory map entry\n");
- return EFI_ST_FAILURE;
- }
- found = true;
- if (memory_type != entry->type) {
- efi_st_error
- ("Wrong memory type %d, expected %d\n",
- entry->type, memory_type);
- return EFI_ST_FAILURE;
- }
- }
- }
- if (!found) {
- efi_st_error("Missing memory map entry\n");
- return EFI_ST_FAILURE;
- }
- return EFI_ST_SUCCESS;
- }
- /*
- * execute() - execute unit test
- *
- * Return: EFI_ST_SUCCESS for success
- */
- static int execute(void)
- {
- u64 p1;
- u64 p2;
- efi_uintn_t map_size = 0;
- efi_uintn_t map_key;
- efi_uintn_t desc_size;
- u32 desc_version;
- struct efi_mem_desc *memory_map;
- efi_status_t ret;
- /* Allocate two page ranges with different memory type */
- ret = boottime->allocate_pages(EFI_ALLOCATE_ANY_PAGES,
- EFI_RUNTIME_SERVICES_CODE,
- EFI_ST_NUM_PAGES, &p1);
- if (ret != EFI_SUCCESS) {
- efi_st_error("AllocatePages did not return EFI_SUCCESS\n");
- return EFI_ST_FAILURE;
- }
- ret = boottime->allocate_pages(EFI_ALLOCATE_ANY_PAGES,
- EFI_RUNTIME_SERVICES_DATA,
- EFI_ST_NUM_PAGES, &p2);
- if (ret != EFI_SUCCESS) {
- efi_st_error("AllocatePages did not return EFI_SUCCESS\n");
- return EFI_ST_FAILURE;
- }
- /* Load memory map */
- ret = boottime->get_memory_map(&map_size, NULL, &map_key, &desc_size,
- &desc_version);
- if (ret != EFI_BUFFER_TOO_SMALL) {
- efi_st_error
- ("GetMemoryMap did not return EFI_BUFFER_TOO_SMALL\n");
- return EFI_ST_FAILURE;
- }
- /* Allocate extra space for newly allocated memory */
- map_size += sizeof(struct efi_mem_desc);
- ret = boottime->allocate_pool(EFI_BOOT_SERVICES_DATA, map_size,
- (void **)&memory_map);
- if (ret != EFI_SUCCESS) {
- efi_st_error("AllocatePool did not return EFI_SUCCESS\n");
- return EFI_ST_FAILURE;
- }
- ret = boottime->get_memory_map(&map_size, memory_map, &map_key,
- &desc_size, &desc_version);
- if (ret != EFI_SUCCESS) {
- efi_st_error("GetMemoryMap did not return EFI_SUCCESS\n");
- return EFI_ST_FAILURE;
- }
- /* Check memory map entries */
- if (find_in_memory_map(map_size, memory_map, desc_size, p1,
- EFI_RUNTIME_SERVICES_CODE) != EFI_ST_SUCCESS)
- return EFI_ST_FAILURE;
- if (find_in_memory_map(map_size, memory_map, desc_size, p2,
- EFI_RUNTIME_SERVICES_DATA) != EFI_ST_SUCCESS)
- return EFI_ST_FAILURE;
- /* Free memory */
- ret = boottime->free_pages(p1, EFI_ST_NUM_PAGES);
- if (ret != EFI_SUCCESS) {
- efi_st_error("FreePages did not return EFI_SUCCESS\n");
- return EFI_ST_FAILURE;
- }
- ret = boottime->free_pages(p2, EFI_ST_NUM_PAGES);
- if (ret != EFI_SUCCESS) {
- efi_st_error("FreePages did not return EFI_SUCCESS\n");
- return EFI_ST_FAILURE;
- }
- ret = boottime->free_pool(memory_map);
- if (ret != EFI_SUCCESS) {
- efi_st_error("FreePool did not return EFI_SUCCESS\n");
- return EFI_ST_FAILURE;
- }
- /* Check memory reservation for the device tree */
- if (fdt_addr &&
- find_in_memory_map(map_size, memory_map, desc_size, fdt_addr,
- EFI_ACPI_RECLAIM_MEMORY) != EFI_ST_SUCCESS) {
- efi_st_error
- ("Device tree not marked as ACPI reclaim memory\n");
- return EFI_ST_FAILURE;
- }
- return EFI_ST_SUCCESS;
- }
- EFI_UNIT_TEST(memory) = {
- .name = "memory",
- .phase = EFI_EXECUTE_BEFORE_BOOTTIME_EXIT,
- .setup = setup,
- .execute = execute,
- };
|