Merge tag 'u-boot-rockchip-20200531' of https://gitlab.denx.de/u-boot/custodians...

[oweals/u-boot.git] / common / spl / spl_nor.c

diff --git a/common/spl/spl_nor.c b/common/spl/spl_nor.c
index b919fffb30db9ef5d33b5e560fe7bf962d2a764d..5270401db0347270a23b7b33a27898b1d0d04ae9 100644 (file)
--- a/common/spl/spl_nor.c
+++ b/common/spl/spl_nor.c
@@ -4,9 +4,10 @@
  */
 
 #include <common.h>
+#include <image.h>
+#include <log.h>
 #include <spl.h>
 
-#ifdef CONFIG_SPL_LOAD_FIT
 static ulong spl_nor_load_read(struct spl_load_info *load, ulong sector,
                               ulong count, void *buf)
 {
@@ -16,12 +17,15 @@ static ulong spl_nor_load_read(struct spl_load_info *load, ulong sector,
 
        return count;
 }
-#endif
+
+unsigned long __weak spl_nor_get_uboot_base(void)
+{
+       return CONFIG_SYS_UBOOT_BASE;
+}
 
 static int spl_nor_load_image(struct spl_image_info *spl_image,
                              struct spl_boot_device *bootdev)
 {
-       int ret;
        __maybe_unused const struct image_header *header;
        __maybe_unused struct spl_load_info load;
 
@@ -40,6 +44,8 @@ static int spl_nor_load_image(struct spl_image_info *spl_image,
                header = (const struct image_header *)CONFIG_SYS_OS_BASE;
 #ifdef CONFIG_SPL_LOAD_FIT
                if (image_get_magic(header) == FDT_MAGIC) {
+                       int ret;
+
                        debug("Found FIT\n");
                        load.bl_len = 1;
                        load.read = spl_nor_load_read;
@@ -48,11 +54,17 @@ static int spl_nor_load_image(struct spl_image_info *spl_image,
                                                  CONFIG_SYS_OS_BASE,
                                                  (void *)header);
 
+#if defined CONFIG_SYS_SPL_ARGS_ADDR && defined CONFIG_CMD_SPL_NOR_OFS
+                       memcpy((void *)CONFIG_SYS_SPL_ARGS_ADDR,
+                              (void *)CONFIG_CMD_SPL_NOR_OFS,
+                              CONFIG_CMD_SPL_WRITE_SIZE);
+#endif
                        return ret;
                }
 #endif
                if (image_get_os(header) == IH_OS_LINUX) {
                        /* happy - was a Linux */
+                       int ret;
 
                        ret = spl_parse_image_header(spl_image, header);
                        if (ret)
@@ -62,8 +74,9 @@ static int spl_nor_load_image(struct spl_image_info *spl_image,
                               (void *)(CONFIG_SYS_OS_BASE +
                                        sizeof(struct image_header)),
                               spl_image->size);
-
+#ifdef CONFIG_SYS_FDT_BASE
                        spl_image->arg = (void *)CONFIG_SYS_FDT_BASE;
+#endif
 
                        return 0;
                } else {
@@ -79,26 +92,30 @@ static int spl_nor_load_image(struct spl_image_info *spl_image,
         * defined location in SDRAM
         */
 #ifdef CONFIG_SPL_LOAD_FIT
-       header = (const struct image_header *)CONFIG_SYS_UBOOT_BASE;
+       header = (const struct image_header *)spl_nor_get_uboot_base();
        if (image_get_magic(header) == FDT_MAGIC) {
                debug("Found FIT format U-Boot\n");
                load.bl_len = 1;
                load.read = spl_nor_load_read;
-               ret = spl_load_simple_fit(spl_image, &load,
-                                         CONFIG_SYS_UBOOT_BASE,
-                                         (void *)header);
-
-               return ret;
+               return spl_load_simple_fit(spl_image, &load,
+                                          spl_nor_get_uboot_base(),
+                                          (void *)header);
        }
 #endif
-       ret = spl_parse_image_header(spl_image,
-                       (const struct image_header *)CONFIG_SYS_UBOOT_BASE);
-       if (ret)
-               return ret;
-
-       memcpy((void *)(unsigned long)spl_image->load_addr,
-              (void *)(CONFIG_SYS_UBOOT_BASE + sizeof(struct image_header)),
-              spl_image->size);
+       if (IS_ENABLED(CONFIG_SPL_LOAD_IMX_CONTAINER)) {
+               load.bl_len = 1;
+               load.read = spl_nor_load_read;
+               return spl_load_imx_container(spl_image, &load,
+                                             spl_nor_get_uboot_base());
+       }
+
+       /* Legacy image handling */
+       if (IS_ENABLED(CONFIG_SPL_LEGACY_IMAGE_SUPPORT)) {
+               load.bl_len = 1;
+               load.read = spl_nor_load_read;
+               return spl_load_legacy_img(spl_image, &load,
+                                          spl_nor_get_uboot_base());
+       }
 
        return 0;
 }