spl: nand: support loading legacy image with payload compressed

Add support to load legacy image with payload compressed. This redirects
the boot flow for all legacy images. If the payload is not compressed, the
actual behavior will remain unchanged.

Reviewed-by: Daniel Schwierzeck <daniel.schwierzeck@gmail.com>
Signed-off-by: Weijie Gao <weijie.gao@mediatek.com>

diff --git a/common/spl/spl_nand.c b/common/spl/spl_nand.c
index 82a10ffa63aa7cb5ca64c249c27926f466851ccf..7b7579a2df16d77beedf1bc10edcaee2e366e8c1 100644 (file)
--- a/common/spl/spl_nand.c
+++ b/common/spl/spl_nand.c
@@ -56,6 +56,21 @@ static ulong spl_nand_fit_read(struct spl_load_info *load, ulong offs,
        return size / load->bl_len;
 }
 
+static ulong spl_nand_legacy_read(struct spl_load_info *load, ulong offs,
+                                 ulong size, void *dst)
+{
+       int err;
+
+       debug("%s: offs %lx, size %lx, dst %p\n",
+             __func__, offs, size, dst);
+
+       err = nand_spl_load_image(offs, size, dst);
+       if (err)
+               return 0;
+
+       return size;
+}
+
 struct mtd_info * __weak nand_get_mtd(void)
 {
        return NULL;
@@ -93,6 +108,18 @@ static int spl_nand_load_element(struct spl_image_info *spl_image,
                load.bl_len = bl_len;
                load.read = spl_nand_fit_read;
                return spl_load_imx_container(spl_image, &load, offset / bl_len);
+       } else if (IS_ENABLED(CONFIG_SPL_LEGACY_IMAGE_FORMAT) &&
+                  image_get_magic(header) == IH_MAGIC) {
+               struct spl_load_info load;
+
+               debug("Found legacy image\n");
+               load.dev = NULL;
+               load.priv = NULL;
+               load.filename = NULL;
+               load.bl_len = 1;
+               load.read = spl_nand_legacy_read;
+
+               return spl_load_legacy_img(spl_image, bootdev, &load, offset);
        } else {
                err = spl_parse_image_header(spl_image, bootdev, header);
                if (err)