spl_load.c 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495
  1. // SPDX-License-Identifier: GPL-2.0+
  2. /*
  3. * Copyright 2021 Google LLC
  4. * Written by Simon Glass <sjg@chromium.org>
  5. */
  6. #include <common.h>
  7. #include <image.h>
  8. #include <mapmem.h>
  9. #include <os.h>
  10. #include <spl.h>
  11. #include <test/ut.h>
  12. /* Declare a new SPL test */
  13. #define SPL_TEST(_name, _flags) UNIT_TEST(_name, _flags, spl_test)
  14. /* Context used for this test */
  15. struct text_ctx {
  16. int fd;
  17. };
  18. static ulong read_fit_image(struct spl_load_info *load, ulong sector,
  19. ulong count, void *buf)
  20. {
  21. struct text_ctx *text_ctx = load->priv;
  22. off_t offset, ret;
  23. ssize_t res;
  24. offset = sector * load->bl_len;
  25. ret = os_lseek(text_ctx->fd, offset, OS_SEEK_SET);
  26. if (ret != offset) {
  27. printf("Failed to seek to %zx, got %zx (errno=%d)\n", offset,
  28. ret, errno);
  29. return 0;
  30. }
  31. res = os_read(text_ctx->fd, buf, count * load->bl_len);
  32. if (res == -1) {
  33. printf("Failed to read %lx bytes, got %ld (errno=%d)\n",
  34. count * load->bl_len, res, errno);
  35. return 0;
  36. }
  37. return count;
  38. }
  39. int board_fit_config_name_match(const char *name)
  40. {
  41. return 0;
  42. }
  43. struct image_header *spl_get_load_buffer(ssize_t offset, size_t size)
  44. {
  45. return map_sysmem(0x100000, 0);
  46. }
  47. static int spl_test_load(struct unit_test_state *uts)
  48. {
  49. const char *cur_prefix, *next_prefix;
  50. struct spl_image_info image;
  51. struct image_header *header;
  52. struct text_ctx text_ctx;
  53. struct spl_load_info load;
  54. char fname[256];
  55. int ret;
  56. int fd;
  57. memset(&load, '\0', sizeof(load));
  58. load.bl_len = 512;
  59. load.read = read_fit_image;
  60. cur_prefix = spl_phase_prefix(spl_phase());
  61. next_prefix = spl_phase_prefix(spl_next_phase());
  62. ret = os_find_u_boot(fname, sizeof(fname), true, cur_prefix,
  63. next_prefix);
  64. if (ret) {
  65. printf("(%s not found, error %d)\n", fname, ret);
  66. return ret;
  67. }
  68. load.filename = fname;
  69. header = spl_get_load_buffer(-sizeof(*header), sizeof(*header));
  70. fd = os_open(fname, OS_O_RDONLY);
  71. ut_assert(fd >= 0);
  72. ut_asserteq(512, os_read(fd, header, 512));
  73. text_ctx.fd = fd;
  74. load.priv = &text_ctx;
  75. ut_assertok(spl_load_simple_fit(&image, &load, 0, header));
  76. return 0;
  77. }
  78. SPL_TEST(spl_test_load, 0);