123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131 |
- // SPDX-License-Identifier: GPL-2.0+
- /*
- * Copyright (C) 2020 Stefan Roese <sr@denx.de>
- */
- #include <common.h>
- #include <malloc.h>
- #include <spl.h>
- #include <lzma/LzmaTypes.h>
- #include <lzma/LzmaDec.h>
- #include <lzma/LzmaTools.h>
- #define LZMA_LEN (1 << 20)
- int spl_parse_legacy_header(struct spl_image_info *spl_image,
- const struct image_header *header)
- {
- u32 header_size = sizeof(struct image_header);
- /* check uImage header CRC */
- if (IS_ENABLED(CONFIG_SPL_LEGACY_IMAGE_CRC_CHECK) &&
- !image_check_hcrc(header)) {
- puts("SPL: Image header CRC check failed!\n");
- return -EINVAL;
- }
- if (spl_image->flags & SPL_COPY_PAYLOAD_ONLY) {
- /*
- * On some system (e.g. powerpc), the load-address and
- * entry-point is located at address 0. We can't load
- * to 0-0x40. So skip header in this case.
- */
- spl_image->load_addr = image_get_load(header);
- spl_image->entry_point = image_get_ep(header);
- spl_image->size = image_get_data_size(header);
- } else {
- spl_image->entry_point = image_get_ep(header);
- /* Load including the header */
- spl_image->load_addr = image_get_load(header) -
- header_size;
- spl_image->size = image_get_data_size(header) +
- header_size;
- }
- #ifdef CONFIG_SPL_LEGACY_IMAGE_CRC_CHECK
- /* store uImage data length and CRC to check later */
- spl_image->dcrc_data = image_get_load(header);
- spl_image->dcrc_length = image_get_data_size(header);
- spl_image->dcrc = image_get_dcrc(header);
- #endif
- spl_image->os = image_get_os(header);
- spl_image->name = image_get_name(header);
- debug(SPL_TPL_PROMPT
- "payload image: %32s load addr: 0x%lx size: %d\n",
- spl_image->name, spl_image->load_addr, spl_image->size);
- return 0;
- }
- /*
- * This function is added explicitly to avoid code size increase, when
- * no compression method is enabled. The compiler will optimize the
- * following switch/case statement in spl_load_legacy_img() away due to
- * Dead Code Elimination.
- */
- static inline int spl_image_get_comp(const struct image_header *hdr)
- {
- if (IS_ENABLED(CONFIG_SPL_LZMA))
- return image_get_comp(hdr);
- return IH_COMP_NONE;
- }
- int spl_load_legacy_img(struct spl_image_info *spl_image,
- struct spl_load_info *load, ulong header)
- {
- __maybe_unused SizeT lzma_len;
- __maybe_unused void *src;
- struct image_header hdr;
- ulong dataptr;
- int ret;
- /* Read header into local struct */
- load->read(load, header, sizeof(hdr), &hdr);
- ret = spl_parse_image_header(spl_image, &hdr);
- if (ret)
- return ret;
- dataptr = header + sizeof(hdr);
- /* Read image */
- switch (spl_image_get_comp(&hdr)) {
- case IH_COMP_NONE:
- load->read(load, dataptr, spl_image->size,
- (void *)(unsigned long)spl_image->load_addr);
- break;
- case IH_COMP_LZMA:
- lzma_len = LZMA_LEN;
- debug("LZMA: Decompressing %08lx to %08lx\n",
- dataptr, spl_image->load_addr);
- src = malloc(spl_image->size);
- if (!src) {
- printf("Unable to allocate %d bytes for LZMA\n",
- spl_image->size);
- return -ENOMEM;
- }
- load->read(load, dataptr, spl_image->size, src);
- ret = lzmaBuffToBuffDecompress((void *)spl_image->load_addr,
- &lzma_len, src, spl_image->size);
- if (ret) {
- printf("LZMA decompression error: %d\n", ret);
- return ret;
- }
- spl_image->size = lzma_len;
- break;
- default:
- debug("Compression method %s is not supported\n",
- genimg_get_comp_short_name(image_get_comp(&hdr)));
- return -EINVAL;
- }
- return 0;
- }
|