1 // SPDX-License-Identifier: GPL-2.0+
3 * (C) Copyright 2020, Heinrich Schuchardt <xypron.glpk@gmx.de>
7 #include <efi_loader.h>
12 #define UT_REG_CAPACITY 6
14 static int lib_test_efi_image_region_add(struct unit_test_state *uts)
16 struct efi_image_regions *regs;
18 regs = calloc(sizeof(*regs) +
19 sizeof(struct image_region) * UT_REG_CAPACITY, 1);
22 regs->max = UT_REG_CAPACITY;
24 ut_asserteq(0, regs->num);
25 ut_asserteq_64(EFI_INVALID_PARAMETER,
26 efi_image_region_add(regs, (void *)0x4000,
28 ut_asserteq(0, regs->num);
29 ut_asserteq_64(EFI_SUCCESS,
30 efi_image_region_add(regs, (void *)0x3100,
32 ut_asserteq(1, regs->num);
33 ut_asserteq_64(EFI_SUCCESS,
34 efi_image_region_add(regs, (void *)0x2000,
36 ut_asserteq(2, regs->num);
37 ut_asserteq_64(EFI_SUCCESS,
38 efi_image_region_add(regs, (void *)0x1000,
40 ut_asserteq(3, regs->num);
41 ut_asserteq_64(EFI_SUCCESS,
42 efi_image_region_add(regs, (void *)0x4000,
44 ut_asserteq(4, regs->num);
45 ut_asserteq_64(EFI_SUCCESS,
46 efi_image_region_add(regs, (void *)0x1f00,
48 ut_asserteq(5, regs->num);
50 ut_asserteq_ptr((void *)0x3100, regs->reg[0].data);
51 ut_asserteq(0x0f00, regs->reg[0].size);
53 ut_asserteq_ptr((void *)0x2000, regs->reg[1].data);
54 ut_asserteq(0x1100, regs->reg[1].size);
56 ut_asserteq_ptr((void *)0x1000, regs->reg[2].data);
57 ut_asserteq(0x0f00, regs->reg[2].size);
59 ut_asserteq_ptr((void *)0x4000, regs->reg[3].data);
60 ut_asserteq(0x0e00, regs->reg[3].size);
62 ut_asserteq_ptr((void *)0x1f00, regs->reg[4].data);
63 ut_asserteq(0x0101, regs->reg[4].size);
70 LIB_TEST(lib_test_efi_image_region_add, 0);
72 static int lib_test_efi_image_region_sort(struct unit_test_state *uts)
74 struct efi_image_regions *regs;
76 regs = calloc(sizeof(*regs) +
77 sizeof(struct image_region) * UT_REG_CAPACITY, 1);
80 regs->max = UT_REG_CAPACITY;
82 ut_asserteq(0, regs->num);
83 ut_asserteq_64(EFI_INVALID_PARAMETER,
84 efi_image_region_add(regs, (void *)0x4000,
86 ut_asserteq(0, regs->num);
87 ut_asserteq_64(EFI_SUCCESS,
88 efi_image_region_add(regs, (void *)0x3100,
90 ut_asserteq(1, regs->num);
91 ut_asserteq_64(EFI_SUCCESS,
92 efi_image_region_add(regs, (void *)0x2000,
94 ut_asserteq(2, regs->num);
95 ut_asserteq_64(EFI_SUCCESS,
96 efi_image_region_add(regs, (void *)0x1000,
98 ut_asserteq(3, regs->num);
99 ut_asserteq_64(EFI_SUCCESS,
100 efi_image_region_add(regs, (void *)0x4000,
102 ut_asserteq(4, regs->num);
103 ut_asserteq_64(EFI_INVALID_PARAMETER,
104 efi_image_region_add(regs, (void *)0x1f00,
106 ut_asserteq(4, regs->num);
107 ut_asserteq_64(EFI_INVALID_PARAMETER,
108 efi_image_region_add(regs, (void *)0x10ff,
110 ut_asserteq(4, regs->num);
111 ut_asserteq_64(EFI_INVALID_PARAMETER,
112 efi_image_region_add(regs, (void *)0x0000,
114 ut_asserteq(4, regs->num);
115 ut_asserteq_64(EFI_INVALID_PARAMETER,
116 efi_image_region_add(regs, (void *)0x3100,
118 ut_asserteq(4, regs->num);
119 ut_asserteq_64(EFI_INVALID_PARAMETER,
120 efi_image_region_add(regs, (void *)0x3200,
122 ut_asserteq(4, regs->num);
123 ut_asserteq_64(EFI_INVALID_PARAMETER,
124 efi_image_region_add(regs, (void *)0x3200,
126 ut_asserteq(4, regs->num);
127 ut_asserteq_64(EFI_SUCCESS,
128 efi_image_region_add(regs, (void *)0x1f00,
130 ut_asserteq(5, regs->num);
131 ut_asserteq_64(EFI_SUCCESS,
132 efi_image_region_add(regs, (void *)0x4000,
134 ut_asserteq(6, regs->num);
135 ut_asserteq_64(EFI_OUT_OF_RESOURCES,
136 efi_image_region_add(regs, (void *)0x6000,
138 ut_asserteq(6, regs->num);
140 ut_asserteq_ptr((void *)0x1000, regs->reg[0].data);
141 ut_asserteq(0x0f00, regs->reg[0].size);
143 ut_asserteq_ptr((void *)0x1f00, regs->reg[1].data);
144 ut_asserteq(0x0100, regs->reg[1].size);
146 ut_asserteq_ptr((void *)0x2000, regs->reg[2].data);
147 ut_asserteq(0x1100, regs->reg[2].size);
149 ut_asserteq_ptr((void *)0x3100, regs->reg[3].data);
150 ut_asserteq(0x0f00, regs->reg[3].size);
152 ut_asserteq_ptr((void *)0x4000, regs->reg[4].data);
153 ut_asserteq(0x0000, regs->reg[4].size);
155 ut_asserteq_ptr((void *)0x4000, regs->reg[5].data);
156 ut_asserteq(0x0e00, regs->reg[5].size);
163 LIB_TEST(lib_test_efi_image_region_sort, 0);