@@ -4,8 +4,15 @@
*/
#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)
{
@@ -56,15 +63,50 @@ int spl_load_legacy_img(struct spl_image_info *spl_image,
const struct image_header *header,
uintptr_t dataptr, struct spl_load_info *info)
{
+ __maybe_unused SizeT lzma_len;
+ __maybe_unused void *src;
int ret;
ret = spl_parse_image_header(spl_image, header);
if (ret)
return ret;
- dataptr = (uintptr_t)header + sizeof(struct image_header);
- info->read(info, dataptr, spl_image->size,
- (void *)(unsigned long)spl_image->load_addr);
+ switch (image_get_comp(header)) {
+ case IH_COMP_NONE:
+ info->read(info, dataptr, spl_image->size,
+ (void *)(unsigned long)spl_image->load_addr);
+ break;
+
+#if IS_ENABLED(CONFIG_SPL_LZMA)
+ 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;
+ }
+
+ info->read(info, 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;
+#endif
+
+ default:
+ debug("Compression method %s is not supported\n",
+ genimg_get_comp_short_name(image_get_comp(header)));
+ return -EINVAL;
+ }
return 0;
}