efi_selftest_variables.c 5.7 KB

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