efi_selftest_rng.c 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117
  1. // SPDX-License-Identifier: GPL-2.0+
  2. /*
  3. * efi_selftest_rng
  4. *
  5. * Copyright (c) 2019 Heinrich Schuchardt <xypron.glpk@gmx.de>
  6. *
  7. * Test the random number generator service.
  8. */
  9. #include <efi_selftest.h>
  10. #include <efi_rng.h>
  11. #define RNG_LEN 9
  12. static struct efi_boot_services *boottime;
  13. static efi_guid_t efi_rng_guid = EFI_RNG_PROTOCOL_GUID;
  14. /*
  15. * Setup unit test.
  16. *
  17. * @handle: handle of the loaded image
  18. * @systable: system table
  19. * @return: EFI_ST_SUCCESS for success
  20. */
  21. static int setup(const efi_handle_t handle,
  22. const struct efi_system_table *systable)
  23. {
  24. boottime = systable->boottime;
  25. return EFI_ST_SUCCESS;
  26. }
  27. /*
  28. * Execute unit test.
  29. *
  30. * Retrieve available RNG algorithms.
  31. * Retrieve two random values and compare them.
  32. *
  33. * @return: EFI_ST_SUCCESS for success
  34. */
  35. static int execute(void)
  36. {
  37. efi_status_t ret;
  38. efi_uintn_t size;
  39. struct efi_rng_protocol *rng;
  40. efi_guid_t *algo_list;
  41. u8 rnd1[RNG_LEN] __aligned(4), rnd2[RNG_LEN] __aligned(4);
  42. int r;
  43. /* Get random number generator protocol */
  44. ret = boottime->locate_protocol(&efi_rng_guid, NULL, (void **)&rng);
  45. if (ret != EFI_SUCCESS) {
  46. efi_st_error(
  47. "Random number generator protocol not available\n");
  48. return EFI_ST_FAILURE;
  49. }
  50. ret = rng->get_info(rng, &size, NULL);
  51. if (ret != EFI_BUFFER_TOO_SMALL) {
  52. efi_st_error("Could not retrieve alorithm list size\n");
  53. return EFI_ST_FAILURE;
  54. }
  55. if (size < sizeof(efi_guid_t)) {
  56. efi_st_error("Empty alorithm list\n");
  57. return EFI_ST_FAILURE;
  58. }
  59. ret = boottime->allocate_pool(EFI_LOADER_DATA, size,
  60. (void **)&algo_list);
  61. if (ret != EFI_SUCCESS) {
  62. efi_st_error("Could not allocate pool memory\n");
  63. return EFI_ST_FAILURE;
  64. }
  65. ret = rng->get_info(rng, &size, algo_list);
  66. if (ret != EFI_SUCCESS) {
  67. efi_st_error("Could not get info\n");
  68. return EFI_ST_FAILURE;
  69. }
  70. if (size < sizeof(efi_guid_t)) {
  71. efi_st_error("Empty alorithm list\n");
  72. return EFI_ST_FAILURE;
  73. }
  74. memset(rnd1, 0, RNG_LEN);
  75. memset(rnd2, 0, RNG_LEN);
  76. ret = rng->get_rng(rng, NULL, RNG_LEN - 1, &rnd1[1]);
  77. if (ret != EFI_SUCCESS) {
  78. efi_st_error("Could not get random value\n");
  79. return EFI_ST_FAILURE;
  80. }
  81. ret = rng->get_rng(rng, algo_list, RNG_LEN - 1, &rnd2[1]);
  82. if (ret != EFI_SUCCESS) {
  83. efi_st_error("Could not get random value\n");
  84. return EFI_ST_FAILURE;
  85. }
  86. r = memcmp(rnd1, rnd2, RNG_LEN);
  87. if (!r) {
  88. efi_st_error("Two equal consecutive random numbers\n");
  89. return EFI_ST_FAILURE;
  90. }
  91. ret = boottime->free_pool(algo_list);
  92. if (ret != EFI_SUCCESS) {
  93. efi_st_error("Could not free pool memory\n");
  94. return EFI_ST_FAILURE;
  95. }
  96. return EFI_ST_SUCCESS;
  97. }
  98. EFI_UNIT_TEST(rng) = {
  99. .name = "random number generator",
  100. .phase = EFI_EXECUTE_BEFORE_BOOTTIME_EXIT,
  101. .setup = setup,
  102. .execute = execute,
  103. };