spl: mmc: Report device number when we fail

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

diff --git a/common/spl/spl_nor.c b/common/spl/spl_nor.c
index d07ca84382468278e066f69598b6c490f3b68661..969e319de06a5457a70b7a1093533f68f0b96ede 100644 (file)
--- a/common/spl/spl_nor.c
+++ b/common/spl/spl_nor.c
@@ -1,16 +1,30 @@
+// SPDX-License-Identifier: GPL-2.0+
 /*
  * Copyright (C) 2012 Stefan Roese <sr@denx.de>
- *
- * SPDX-License-Identifier:    GPL-2.0+
  */
 
 #include <common.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)
+{
+       debug("%s: sector %lx, count %lx, buf %p\n",
+             __func__, sector, count, buf);
+       memcpy(buf, (void *)sector, count);
+
+       return count;
+}
+#endif
+
 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;
+
        /*
         * Loading of the payload to SDRAM is done with skipping of
         * the mkimage header in this SPL NOR driver
@@ -19,14 +33,24 @@ static int spl_nor_load_image(struct spl_image_info *spl_image,
 
 #ifdef CONFIG_SPL_OS_BOOT
        if (!spl_start_uboot()) {
-               const struct image_header *header;
-
                /*
                 * Load Linux from its location in NOR flash to its defined
                 * location in SDRAM
                 */
                header = (const struct image_header *)CONFIG_SYS_OS_BASE;
+#ifdef CONFIG_SPL_LOAD_FIT
+               if (image_get_magic(header) == FDT_MAGIC) {
+                       debug("Found FIT\n");
+                       load.bl_len = 1;
+                       load.read = spl_nor_load_read;
+
+                       ret = spl_load_simple_fit(spl_image, &load,
+                                                 CONFIG_SYS_OS_BASE,
+                                                 (void *)header);
 
+                       return ret;
+               }
+#endif
                if (image_get_os(header) == IH_OS_LINUX) {
                        /* happy - was a Linux */
 
@@ -38,14 +62,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);
-
-                       /*
-                        * Copy DT blob (fdt) to SDRAM. Passing pointer to
-                        * flash doesn't work
-                        */
-                       memcpy((void *)CONFIG_SYS_SPL_ARGS_ADDR,
-                              (void *)(CONFIG_SYS_FDT_BASE),
-                              CONFIG_SYS_FDT_SIZE);
+#ifdef CONFIG_SYS_FDT_BASE
+                       spl_image->arg = (void *)CONFIG_SYS_FDT_BASE;
+#endif
 
                        return 0;
                } else {
@@ -60,6 +79,19 @@ static int spl_nor_load_image(struct spl_image_info *spl_image,
         * Load real U-Boot from its location in NOR flash to its
         * defined location in SDRAM
         */
+#ifdef CONFIG_SPL_LOAD_FIT
+       header = (const struct image_header *)CONFIG_SYS_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;
+       }
+#endif
        ret = spl_parse_image_header(spl_image,
                        (const struct image_header *)CONFIG_SYS_UBOOT_BASE);
        if (ret)