git.sur5r.net Git - u-boot/blob - lib/efi_selftest/efi_selftest_variables.c

   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  */
  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  */
  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         /* Set variable 1 */
  76         ret = runtime->set_variable(L"efi_st_var1", &guid_vendor1,
  77                                     EFI_VARIABLE_BOOTSERVICE_ACCESS,
  78                                     8, v);
  79         if (ret != EFI_SUCCESS) {
  80                 efi_st_error("SetVariable failed\n");
  81                 return EFI_ST_FAILURE;
  82         }
  83         len = EFI_ST_MAX_DATA_SIZE;
  84         ret = runtime->get_variable(L"efi_st_var1", &guid_vendor1,
  85                                     &attr, &len, data);
  86         if (ret != EFI_SUCCESS) {
  87                 efi_st_error("GetVariable failed\n");
  88                 return EFI_ST_FAILURE;
  89         }
  90         if (len != 8) {
  91                 efi_st_error("GetVariable returned wrong length %u\n",
  92                              (unsigned int)len);
  93                 return EFI_ST_FAILURE;
  94         }
  95         if (efi_st_memcmp(data, v, 8)) {
  96                 efi_st_error("GetVariable returned wrong value\n");
  97                 return EFI_ST_FAILURE;
  98         }
  99         /* Append variable 1 */
 100         ret = runtime->set_variable(L"efi_st_var1", &guid_vendor1,
 101                                     EFI_VARIABLE_BOOTSERVICE_ACCESS |
 102                                     EFI_VARIABLE_APPEND_WRITE,
 103                                     7, v + 8);
 104         if (ret != EFI_SUCCESS) {
 105                 efi_st_error("SetVariable failed\n");
 106                 return EFI_ST_FAILURE;
 107         }
 108         len = EFI_ST_MAX_DATA_SIZE;
 109         ret = runtime->get_variable(L"efi_st_var1", &guid_vendor1,
 110                                     &attr, &len, data);
 111         if (ret != EFI_SUCCESS) {
 112                 efi_st_error("GetVariable failed\n");
 113                 return EFI_ST_FAILURE;
 114         }
 115         if (len != 15)
 116                 efi_st_todo("GetVariable returned wrong length %u\n",
 117                             (unsigned int)len);
 118         if (efi_st_memcmp(data, v, len))
 119                 efi_st_todo("GetVariable returned wrong value\n");
 120         /* Enumerate variables */
 121         boottime->set_mem(&guid, 16, 0);
 122         *varname = 0;
 123         flag = 0;
 124         for (;;) {
 125                 len = EFI_ST_MAX_VARNAME_SIZE;
 126                 ret = runtime->get_next_variable_name(&len, varname, &guid);
 127                 if (ret == EFI_NOT_FOUND)
 128                         break;
 129                 if (ret != EFI_SUCCESS) {
 130                         efi_st_todo("GetNextVariableName failed\n");
 131                         break;
 132                 }
 133                 if (!efi_st_memcmp(&guid, &guid_vendor0, sizeof(efi_guid_t)) &&
 134                     !efi_st_strcmp_16_8(varname, "efi_st_var0"))
 135                         flag |= 2;
 136                 if (!efi_st_memcmp(&guid, &guid_vendor1, sizeof(efi_guid_t)) &&
 137                     !efi_st_strcmp_16_8(varname, "efi_st_var1"))
 138                         flag |= 2;
 139         }
 140         if (flag != 3)
 141                 efi_st_todo(
 142                         "GetNextVariableName did not return all variables\n");
 143         /* Delete variable 1 */
 144         ret = runtime->set_variable(L"efi_st_var1", &guid_vendor1,
 145                                     0, 0, NULL);
 146         if (ret != EFI_SUCCESS) {
 147                 efi_st_error("SetVariable failed\n");
 148                 return EFI_ST_FAILURE;
 149         }
 150         len = EFI_ST_MAX_DATA_SIZE;
 151         ret = runtime->get_variable(L"efi_st_var1", &guid_vendor1,
 152                                     &attr, &len, data);
 153         if (ret != EFI_NOT_FOUND) {
 154                 efi_st_error("Variable was not deleted\n");
 155                 return EFI_ST_FAILURE;
 156         }
 157         /* Delete variable 0 */
 158         ret = runtime->set_variable(L"efi_st_var0", &guid_vendor0,
 159                                     0, 0, NULL);
 160         if (ret != EFI_SUCCESS) {
 161                 efi_st_error("SetVariable failed\n");
 162                 return EFI_ST_FAILURE;
 163         }
 164         len = EFI_ST_MAX_DATA_SIZE;
 165         ret = runtime->get_variable(L"efi_st_var0", &guid_vendor0,
 166                                     &attr, &len, data);
 167         if (ret != EFI_NOT_FOUND) {
 168                 efi_st_error("Variable was not deleted\n");
 169                 return EFI_ST_FAILURE;
 170         }
 171
 172         return EFI_ST_SUCCESS;
 173 }
 174
 175 EFI_UNIT_TEST(variables) = {
 176         .name = "variables",
 177         .phase = EFI_EXECUTE_BEFORE_BOOTTIME_EXIT,
 178         .setup = setup,
 179         .execute = execute,
 180 };