Open Menu / test / lib / efi_memory.c
Loading...
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Test memory functions
 *
 * Copyright (c) 2025 Heinrich Schuchardt <xypron.glpk@gmx.de>
 */

#include <efi_loader.h>
#include <test/lib.h>
#include <test/test.h>
#include <test/ut.h>

static int lib_test_efi_alloc_aligned_pages(struct unit_test_state *uts)
{
	efi_status_t ret;

	void *addr;
	unsigned long align = 0x400000;

	addr = efi_alloc_aligned_pages(4096, EFI_PERSISTENT_MEMORY_TYPE,
				       EFI_PAGE_SIZE);
	ut_asserteq_ptr(NULL, addr);

	addr = efi_alloc_aligned_pages(4096, 0x6FFFFFFF, EFI_PAGE_SIZE);
	ut_asserteq_ptr(NULL, addr);

	align = 0x200;
	addr = efi_alloc_aligned_pages(4096, EFI_ACPI_RECLAIM_MEMORY, align);
	ut_assertnonnull(addr);
	ut_asserteq_64(0, (uintptr_t)addr & (align - 1));

	ret = efi_free_pages((uintptr_t) addr, 1);
	ut_asserteq_64(ret, EFI_SUCCESS);

	align = 0x400000;
	addr = efi_alloc_aligned_pages(4096, EFI_ACPI_RECLAIM_MEMORY, align);
	ut_assertnonnull(addr);
	ut_asserteq_64(0, (uintptr_t)addr & (align - 1));

	ret = efi_free_pages((uintptr_t) addr, 1);
	ut_asserteq_64(ret, EFI_SUCCESS);

	return 0;
}
LIB_TEST(lib_test_efi_alloc_aligned_pages, 0);

static int lib_test_efi_allocate_pages(struct unit_test_state *uts)
{
	efi_status_t ret;
	u64 memory;

	ret = efi_allocate_pages(EFI_ALLOCATE_ANY_PAGES,
				 EFI_ACPI_RECLAIM_MEMORY,
				 1, &memory);
	ut_asserteq_64(ret, EFI_SUCCESS);
	ut_asserteq_64(0, memory & EFI_PAGE_MASK);

	ret = efi_free_pages(memory, 1);
	ut_asserteq_64(ret, EFI_SUCCESS);

	return 0;
}
LIB_TEST(lib_test_efi_allocate_pages, 0);