1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 * efi_selftest_variables
4 *
5 * Copyright (c) 2018 Heinrich Schuchardt <xypron.glpk@gmx.de>
6 *
7 * This unit test checks the following protocol services:
8 * ConnectController, DisconnectController,
9 * InstallProtocol, ReinstallProtocol, UninstallProtocol,
10 * OpenProtocol, CloseProtcol, OpenProtocolInformation
11 */
12
13 #include <efi_selftest.h>
14
15 #define EFI_ST_MAX_DATA_SIZE 16
16 #define EFI_ST_MAX_VARNAME_SIZE 40
17
18 static struct efi_boot_services *boottime;
19 static struct efi_runtime_services *runtime;
20 static efi_guid_t guid_vendor0 =
21 EFI_GUID(0x67029eb5, 0x0af2, 0xf6b1,
22 0xda, 0x53, 0xfc, 0xb5, 0x66, 0xdd, 0x1c, 0xe6);
23 static efi_guid_t guid_vendor1 =
24 EFI_GUID(0xff629290, 0x1fc1, 0xd73f,
25 0x8f, 0xb1, 0x32, 0xf9, 0x0c, 0xa0, 0x42, 0xea);
26
27 /*
28 * Setup unit test.
29 *
30 * @handle handle of the loaded image
31 * @systable system table
32 */
setup(const efi_handle_t img_handle,const struct efi_system_table * systable)33 static int setup(const efi_handle_t img_handle,
34 const struct efi_system_table *systable)
35 {
36 boottime = systable->boottime;
37 runtime = systable->runtime;
38
39 return EFI_ST_SUCCESS;
40 }
41
42 /*
43 * Execute unit test.
44 */
execute(void)45 static int execute(void)
46 {
47 efi_status_t ret;
48 efi_uintn_t len;
49 u32 attr;
50 u8 v[16] = {0x5d, 0xd1, 0x5e, 0x51, 0x5a, 0x05, 0xc7, 0x0c,
51 0x35, 0x4a, 0xae, 0x87, 0xa5, 0xdf, 0x0f, 0x65,};
52 u8 data[EFI_ST_MAX_DATA_SIZE];
53 u16 varname[EFI_ST_MAX_VARNAME_SIZE];
54 int flag;
55 efi_guid_t guid;
56 u64 max_storage, rem_storage, max_size;
57
58 ret = runtime->query_variable_info(EFI_VARIABLE_BOOTSERVICE_ACCESS,
59 &max_storage, &rem_storage,
60 &max_size);
61 if (ret != EFI_SUCCESS) {
62 efi_st_todo("QueryVariableInfo failed\n");
63 } else if (!max_storage || !rem_storage || !max_size) {
64 efi_st_error("QueryVariableInfo: wrong info\n");
65 return EFI_ST_FAILURE;
66 }
67 /* Set variable 0 */
68 ret = runtime->set_variable(L"efi_st_var0", &guid_vendor0,
69 EFI_VARIABLE_BOOTSERVICE_ACCESS,
70 3, v + 4);
71 if (ret != EFI_SUCCESS) {
72 efi_st_error("SetVariable failed\n");
73 return EFI_ST_FAILURE;
74 }
75 data[3] = 0xff;
76 len = 3;
77 ret = runtime->get_variable(L"efi_st_var0", &guid_vendor0,
78 &attr, &len, data);
79 if (ret != EFI_SUCCESS) {
80 efi_st_error("GetVariable failed\n");
81 return EFI_ST_FAILURE;
82 }
83 if (efi_st_memcmp(data, v + 4, 3)) {
84 efi_st_error("GetVariable returned wrong value\n");
85 return EFI_ST_FAILURE;
86 }
87 if (data[3] != 0xff) {
88 efi_st_error("GetVariable wrote past the end of the buffer\n");
89 return EFI_ST_FAILURE;
90 }
91 /* Set variable 1 */
92 ret = runtime->set_variable(L"efi_st_var1", &guid_vendor1,
93 EFI_VARIABLE_BOOTSERVICE_ACCESS,
94 8, v);
95 if (ret != EFI_SUCCESS) {
96 efi_st_error("SetVariable failed\n");
97 return EFI_ST_FAILURE;
98 }
99 len = EFI_ST_MAX_DATA_SIZE;
100 ret = runtime->get_variable(L"efi_st_var1", &guid_vendor1,
101 &attr, &len, data);
102 if (ret != EFI_SUCCESS) {
103 efi_st_error("GetVariable failed\n");
104 return EFI_ST_FAILURE;
105 }
106 if (len != 8) {
107 efi_st_error("GetVariable returned wrong length %u\n",
108 (unsigned int)len);
109 return EFI_ST_FAILURE;
110 }
111 if (efi_st_memcmp(data, v, 8)) {
112 efi_st_error("GetVariable returned wrong value\n");
113 return EFI_ST_FAILURE;
114 }
115 /* Append variable 1 */
116 ret = runtime->set_variable(L"efi_st_var1", &guid_vendor1,
117 EFI_VARIABLE_BOOTSERVICE_ACCESS |
118 EFI_VARIABLE_APPEND_WRITE,
119 7, v + 8);
120 if (ret != EFI_SUCCESS) {
121 efi_st_error("SetVariable failed\n");
122 return EFI_ST_FAILURE;
123 }
124 len = EFI_ST_MAX_DATA_SIZE;
125 ret = runtime->get_variable(L"efi_st_var1", &guid_vendor1,
126 &attr, &len, data);
127 if (ret != EFI_SUCCESS) {
128 efi_st_error("GetVariable failed\n");
129 return EFI_ST_FAILURE;
130 }
131 if (len != 15)
132 efi_st_todo("GetVariable returned wrong length %u\n",
133 (unsigned int)len);
134 if (efi_st_memcmp(data, v, len))
135 efi_st_todo("GetVariable returned wrong value\n");
136 /* Enumerate variables */
137 boottime->set_mem(&guid, 16, 0);
138 *varname = 0;
139 flag = 0;
140 for (;;) {
141 len = EFI_ST_MAX_VARNAME_SIZE;
142 ret = runtime->get_next_variable_name(&len, varname, &guid);
143 if (ret == EFI_NOT_FOUND)
144 break;
145 if (ret != EFI_SUCCESS) {
146 efi_st_todo("GetNextVariableName failed\n");
147 break;
148 }
149 if (!efi_st_memcmp(&guid, &guid_vendor0, sizeof(efi_guid_t)) &&
150 !efi_st_strcmp_16_8(varname, "efi_st_var0"))
151 flag |= 2;
152 if (!efi_st_memcmp(&guid, &guid_vendor1, sizeof(efi_guid_t)) &&
153 !efi_st_strcmp_16_8(varname, "efi_st_var1"))
154 flag |= 2;
155 }
156 if (flag != 3)
157 efi_st_todo(
158 "GetNextVariableName did not return all variables\n");
159 /* Delete variable 1 */
160 ret = runtime->set_variable(L"efi_st_var1", &guid_vendor1,
161 0, 0, NULL);
162 if (ret != EFI_SUCCESS) {
163 efi_st_error("SetVariable failed\n");
164 return EFI_ST_FAILURE;
165 }
166 len = EFI_ST_MAX_DATA_SIZE;
167 ret = runtime->get_variable(L"efi_st_var1", &guid_vendor1,
168 &attr, &len, data);
169 if (ret != EFI_NOT_FOUND) {
170 efi_st_error("Variable was not deleted\n");
171 return EFI_ST_FAILURE;
172 }
173 /* Delete variable 0 */
174 ret = runtime->set_variable(L"efi_st_var0", &guid_vendor0,
175 0, 0, NULL);
176 if (ret != EFI_SUCCESS) {
177 efi_st_error("SetVariable failed\n");
178 return EFI_ST_FAILURE;
179 }
180 len = EFI_ST_MAX_DATA_SIZE;
181 ret = runtime->get_variable(L"efi_st_var0", &guid_vendor0,
182 &attr, &len, data);
183 if (ret != EFI_NOT_FOUND) {
184 efi_st_error("Variable was not deleted\n");
185 return EFI_ST_FAILURE;
186 }
187
188 return EFI_ST_SUCCESS;
189 }
190
191 EFI_UNIT_TEST(variables) = {
192 .name = "variables",
193 .phase = EFI_EXECUTE_BEFORE_BOOTTIME_EXIT,
194 .setup = setup,
195 .execute = execute,
196 };
197