| // SPDX-License-Identifier: GPL-2.0+ |
| /* |
| * efi_selftest_variables_runtime |
| * |
| * Copyright (c) 2019 Heinrich Schuchardt <xypron.glpk@gmx.de> |
| * |
| * This unit test checks the runtime services for variables after |
| * ExitBootServices(): |
| * GetVariable, GetNextVariableName, SetVariable, QueryVariableInfo. |
| */ |
| |
| #include <efi_selftest.h> |
| #include <efi_variable.h> |
| #include <u-boot/crc.h> |
| |
| #define EFI_ST_MAX_DATA_SIZE 16 |
| #define EFI_ST_MAX_VARNAME_SIZE 40 |
| |
| static struct efi_boot_services *boottime; |
| static struct efi_runtime_services *runtime; |
| static const efi_guid_t guid_vendor0 = EFI_GLOBAL_VARIABLE_GUID; |
| static const efi_guid_t __efi_runtime_data efi_rt_var_guid = |
| U_BOOT_EFI_RT_VAR_FILE_GUID; |
| |
| /* |
| * Setup unit test. |
| * |
| * @handle handle of the loaded image |
| * @systable system table |
| */ |
| static int setup(const efi_handle_t img_handle, |
| const struct efi_system_table *systable) |
| { |
| boottime = systable->boottime; |
| runtime = systable->runtime; |
| |
| return EFI_ST_SUCCESS; |
| } |
| |
| /** |
| * execute() - execute unit test |
| * |
| * As runtime support is not implmented expect EFI_UNSUPPORTED to be returned. |
| */ |
| static int execute(void) |
| { |
| efi_status_t ret; |
| efi_uintn_t len, avail, append_len = 17; |
| u32 attr; |
| u8 v[16] = {0x5d, 0xd1, 0x5e, 0x51, 0x5a, 0x05, 0xc7, 0x0c, |
| 0x35, 0x4a, 0xae, 0x87, 0xa5, 0xdf, 0x0f, 0x65,}; |
| u8 v2[CONFIG_EFI_VAR_BUF_SIZE]; |
| u8 data[EFI_ST_MAX_DATA_SIZE]; |
| u8 data2[CONFIG_EFI_VAR_BUF_SIZE]; |
| u16 varname[EFI_ST_MAX_VARNAME_SIZE]; |
| efi_guid_t guid; |
| u64 max_storage, rem_storage, max_size; |
| |
| memset(v2, 0x1, sizeof(v2)); |
| ret = runtime->query_variable_info(EFI_VARIABLE_BOOTSERVICE_ACCESS, |
| &max_storage, &rem_storage, |
| &max_size); |
| if (ret != EFI_UNSUPPORTED) { |
| efi_st_error("QueryVariableInfo failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| |
| ret = runtime->set_variable(u"efi_st_var0", &guid_vendor0, |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | |
| EFI_VARIABLE_RUNTIME_ACCESS, |
| 3, v + 4); |
| if (IS_ENABLED(CONFIG_EFI_RT_VOLATILE_STORE)) { |
| efi_uintn_t prev_len, delta; |
| struct efi_var_entry *var; |
| struct efi_var_file *hdr; |
| |
| /* At runtime only non-volatile variables may be set. */ |
| if (ret != EFI_INVALID_PARAMETER) { |
| efi_st_error("SetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| |
| /* runtime atttribute must be set */ |
| ret = runtime->set_variable(u"efi_st_var0", &guid_vendor0, |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | |
| EFI_VARIABLE_NON_VOLATILE, |
| 3, v + 4); |
| if (ret != EFI_INVALID_PARAMETER) { |
| efi_st_error("SetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| |
| len = sizeof(data); |
| ret = runtime->get_variable(u"RTStorageVolatile", |
| &efi_rt_var_guid, |
| &attr, &len, data); |
| if (ret != EFI_SUCCESS) { |
| efi_st_error("GetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| |
| if (len != sizeof(EFI_VAR_FILE_NAME) || |
| memcmp(data, EFI_VAR_FILE_NAME, sizeof(EFI_VAR_FILE_NAME))) { |
| data[len - 1] = 0; |
| efi_st_error("RTStorageVolatile = %s\n", data); |
| return EFI_ST_FAILURE; |
| } |
| |
| len = sizeof(data2); |
| ret = runtime->get_variable(u"VarToFile", &efi_rt_var_guid, |
| &attr, &len, data2); |
| if (ret != EFI_SUCCESS) { |
| efi_st_error("GetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| /* |
| * VarToFile size must change once a variable is inserted |
| * Store it now, we'll use it later |
| */ |
| prev_len = len; |
| ret = runtime->set_variable(u"efi_st_var0", &guid_vendor0, |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | |
| EFI_VARIABLE_RUNTIME_ACCESS | |
| EFI_VARIABLE_NON_VOLATILE, |
| sizeof(v2), |
| v2); |
| /* |
| * This will try to update VarToFile as well and must fail, |
| * without changing or deleting VarToFile |
| */ |
| if (ret != EFI_OUT_OF_RESOURCES) { |
| efi_st_error("SetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| len = sizeof(data2); |
| ret = runtime->get_variable(u"VarToFile", &efi_rt_var_guid, |
| &attr, &len, data2); |
| if (ret != EFI_SUCCESS || prev_len != len) { |
| efi_st_error("Get/SetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| |
| /* Add an 8byte aligned variable */ |
| ret = runtime->set_variable(u"efi_st_var0", &guid_vendor0, |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | |
| EFI_VARIABLE_RUNTIME_ACCESS | |
| EFI_VARIABLE_NON_VOLATILE, |
| sizeof(v), v); |
| if (ret != EFI_SUCCESS) { |
| efi_st_error("SetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| |
| /* Delete it by setting the attrs to 0 */ |
| ret = runtime->set_variable(u"efi_st_var0", &guid_vendor0, |
| 0, sizeof(v), v); |
| if (ret != EFI_SUCCESS) { |
| efi_st_error("SetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| |
| /* Add it back */ |
| ret = runtime->set_variable(u"efi_st_var0", &guid_vendor0, |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | |
| EFI_VARIABLE_RUNTIME_ACCESS | |
| EFI_VARIABLE_NON_VOLATILE, |
| sizeof(v), v); |
| if (ret != EFI_SUCCESS) { |
| efi_st_error("SetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| |
| /* Delete it again by setting the size to 0 */ |
| ret = runtime->set_variable(u"efi_st_var0", &guid_vendor0, |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | |
| EFI_VARIABLE_RUNTIME_ACCESS | |
| EFI_VARIABLE_NON_VOLATILE, |
| 0, NULL); |
| if (ret != EFI_SUCCESS) { |
| efi_st_error("SetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| |
| /* Delete it again and make sure it's not there */ |
| ret = runtime->set_variable(u"efi_st_var0", &guid_vendor0, |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | |
| EFI_VARIABLE_RUNTIME_ACCESS | |
| EFI_VARIABLE_NON_VOLATILE, |
| 0, NULL); |
| if (ret != EFI_NOT_FOUND) { |
| efi_st_error("SetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| |
| /* |
| * Add a non-aligned variable |
| * VarToFile updates must include efi_st_var0 |
| */ |
| ret = runtime->set_variable(u"efi_st_var0", &guid_vendor0, |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | |
| EFI_VARIABLE_RUNTIME_ACCESS | |
| EFI_VARIABLE_NON_VOLATILE, |
| 9, v + 4); |
| if (ret != EFI_SUCCESS) { |
| efi_st_error("SetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| var = efi_var_mem_find(&guid_vendor0, u"efi_st_var0", NULL); |
| if (!var) { |
| efi_st_error("GetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| delta = efi_var_entry_len(var); |
| len = sizeof(data2); |
| ret = runtime->get_variable(u"VarToFile", &efi_rt_var_guid, |
| &attr, &len, data2); |
| if (ret != EFI_SUCCESS || prev_len + delta != len) { |
| efi_st_error("Get/SetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| |
| /* |
| * Append on an existing variable must update VarToFile |
| * Our variable entries are 8-byte aligned. |
| * Adding a single byte will fit on the existing space |
| */ |
| prev_len = len; |
| avail = efi_var_entry_len(var) - |
| (sizeof(u16) * (u16_strlen(var->name) + 1) + sizeof(*var)) - |
| var->length; |
| if (avail >= append_len) |
| delta = 0; |
| else |
| delta = ALIGN(append_len - avail, 8); |
| ret = runtime->set_variable(u"efi_st_var0", &guid_vendor0, |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | |
| EFI_VARIABLE_RUNTIME_ACCESS | |
| EFI_VARIABLE_APPEND_WRITE | |
| EFI_VARIABLE_NON_VOLATILE, |
| append_len, v2); |
| if (ret != EFI_SUCCESS) { |
| efi_st_error("SetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| len = sizeof(data2); |
| ret = runtime->get_variable(u"VarToFile", &efi_rt_var_guid, |
| &attr, &len, data2); |
| if (ret != EFI_SUCCESS) { |
| efi_st_error("GetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| if (prev_len + delta != len) { |
| efi_st_error("Unexpected VarToFile size"); |
| return EFI_ST_FAILURE; |
| } |
| |
| /* Make sure that variable contains a valid file */ |
| hdr = (struct efi_var_file *)data2; |
| if (hdr->magic != EFI_VAR_FILE_MAGIC || |
| len != hdr->length || |
| hdr->crc32 != crc32(0, (u8 *)((uintptr_t)data2 + sizeof(struct efi_var_file)), |
| len - sizeof(struct efi_var_file))) { |
| efi_st_error("VarToFile invalid header\n"); |
| return EFI_ST_FAILURE; |
| } |
| |
| /* Variables that are BS, RT and volatile are RO after EBS */ |
| ret = runtime->set_variable(u"VarToFile", &efi_rt_var_guid, |
| EFI_VARIABLE_BOOTSERVICE_ACCESS | |
| EFI_VARIABLE_RUNTIME_ACCESS | |
| EFI_VARIABLE_NON_VOLATILE, |
| sizeof(v), v); |
| if (ret != EFI_WRITE_PROTECTED) { |
| efi_st_error("Get/SetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| } else { |
| if (ret != EFI_UNSUPPORTED) { |
| efi_st_error("SetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| } |
| len = EFI_ST_MAX_DATA_SIZE; |
| ret = runtime->get_variable(u"PlatformLangCodes", &guid_vendor0, |
| &attr, &len, data); |
| if (ret != EFI_SUCCESS) { |
| efi_st_error("GetVariable failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| memset(&guid, 0, 16); |
| *varname = 0; |
| len = 2 * EFI_ST_MAX_VARNAME_SIZE; |
| ret = runtime->get_next_variable_name(&len, varname, &guid); |
| if (ret != EFI_SUCCESS) { |
| efi_st_error("GetNextVariableName failed\n"); |
| return EFI_ST_FAILURE; |
| } |
| |
| return EFI_ST_SUCCESS; |
| } |
| |
| EFI_UNIT_TEST(variables_run) = { |
| .name = "variables at runtime", |
| .phase = EFI_SETUP_BEFORE_BOOTTIME_EXIT, |
| .setup = setup, |
| .execute = execute, |
| }; |