spl_legacy.c 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133
  1. // SPDX-License-Identifier: GPL-2.0+
  2. /*
  3. * Copyright (C) 2020 Stefan Roese <sr@denx.de>
  4. */
  5. #include <common.h>
  6. #include <image.h>
  7. #include <log.h>
  8. #include <malloc.h>
  9. #include <spl.h>
  10. #include <lzma/LzmaTypes.h>
  11. #include <lzma/LzmaDec.h>
  12. #include <lzma/LzmaTools.h>
  13. #define LZMA_LEN (1 << 20)
  14. int spl_parse_legacy_header(struct spl_image_info *spl_image,
  15. const struct image_header *header)
  16. {
  17. u32 header_size = sizeof(struct image_header);
  18. /* check uImage header CRC */
  19. if (IS_ENABLED(CONFIG_SPL_LEGACY_IMAGE_CRC_CHECK) &&
  20. !image_check_hcrc(header)) {
  21. puts("SPL: Image header CRC check failed!\n");
  22. return -EINVAL;
  23. }
  24. if (spl_image->flags & SPL_COPY_PAYLOAD_ONLY) {
  25. /*
  26. * On some system (e.g. powerpc), the load-address and
  27. * entry-point is located at address 0. We can't load
  28. * to 0-0x40. So skip header in this case.
  29. */
  30. spl_image->load_addr = image_get_load(header);
  31. spl_image->entry_point = image_get_ep(header);
  32. spl_image->size = image_get_data_size(header);
  33. } else {
  34. spl_image->entry_point = image_get_ep(header);
  35. /* Load including the header */
  36. spl_image->load_addr = image_get_load(header) -
  37. header_size;
  38. spl_image->size = image_get_data_size(header) +
  39. header_size;
  40. }
  41. #ifdef CONFIG_SPL_LEGACY_IMAGE_CRC_CHECK
  42. /* store uImage data length and CRC to check later */
  43. spl_image->dcrc_data = image_get_load(header);
  44. spl_image->dcrc_length = image_get_data_size(header);
  45. spl_image->dcrc = image_get_dcrc(header);
  46. #endif
  47. spl_image->os = image_get_os(header);
  48. spl_image->name = image_get_name(header);
  49. debug(SPL_TPL_PROMPT
  50. "payload image: %32s load addr: 0x%lx size: %d\n",
  51. spl_image->name, spl_image->load_addr, spl_image->size);
  52. return 0;
  53. }
  54. /*
  55. * This function is added explicitly to avoid code size increase, when
  56. * no compression method is enabled. The compiler will optimize the
  57. * following switch/case statement in spl_load_legacy_img() away due to
  58. * Dead Code Elimination.
  59. */
  60. static inline int spl_image_get_comp(const struct image_header *hdr)
  61. {
  62. if (IS_ENABLED(CONFIG_SPL_LZMA))
  63. return image_get_comp(hdr);
  64. return IH_COMP_NONE;
  65. }
  66. int spl_load_legacy_img(struct spl_image_info *spl_image,
  67. struct spl_load_info *load, ulong header)
  68. {
  69. __maybe_unused SizeT lzma_len;
  70. __maybe_unused void *src;
  71. struct image_header hdr;
  72. ulong dataptr;
  73. int ret;
  74. /* Read header into local struct */
  75. load->read(load, header, sizeof(hdr), &hdr);
  76. ret = spl_parse_image_header(spl_image, &hdr);
  77. if (ret)
  78. return ret;
  79. dataptr = header + sizeof(hdr);
  80. /* Read image */
  81. switch (spl_image_get_comp(&hdr)) {
  82. case IH_COMP_NONE:
  83. load->read(load, dataptr, spl_image->size,
  84. (void *)(unsigned long)spl_image->load_addr);
  85. break;
  86. case IH_COMP_LZMA:
  87. lzma_len = LZMA_LEN;
  88. debug("LZMA: Decompressing %08lx to %08lx\n",
  89. dataptr, spl_image->load_addr);
  90. src = malloc(spl_image->size);
  91. if (!src) {
  92. printf("Unable to allocate %d bytes for LZMA\n",
  93. spl_image->size);
  94. return -ENOMEM;
  95. }
  96. load->read(load, dataptr, spl_image->size, src);
  97. ret = lzmaBuffToBuffDecompress((void *)spl_image->load_addr,
  98. &lzma_len, src, spl_image->size);
  99. if (ret) {
  100. printf("LZMA decompression error: %d\n", ret);
  101. return ret;
  102. }
  103. spl_image->size = lzma_len;
  104. break;
  105. default:
  106. debug("Compression method %s is not supported\n",
  107. genimg_get_comp_short_name(image_get_comp(&hdr)));
  108. return -EINVAL;
  109. }
  110. return 0;
  111. }