Merge tag 'u-boot-rockchip-20231007' of https://source.denx.de/u-boot/custodians...
[pandora-u-boot.git] / common / spl / spl_fit.c
index e567fe2..ce6b8aa 100644 (file)
@@ -44,7 +44,7 @@ static int find_node_from_desc(const void *fit, int node, const char *str)
        for (child = fdt_first_subnode(fit, node); child >= 0;
             child = fdt_next_subnode(fit, child)) {
                int len;
-               const char *desc = fdt_getprop(fit, child, "description", &len);
+               const char *desc = fdt_getprop(fit, child, FIT_DESC_PROP, &len);
 
                if (!desc)
                        continue;
@@ -208,7 +208,7 @@ static int get_aligned_image_size(struct spl_load_info *info, int data_size,
 }
 
 /**
- * spl_load_fit_image(): load the image described in a certain FIT node
+ * load_simple_fit(): load the image described in a certain FIT node
  * @info:      points to information about the device to load data from
  * @sector:    the start sector of the FIT image on the device
  * @ctx:       points to the FIT context structure
@@ -221,9 +221,9 @@ static int get_aligned_image_size(struct spl_load_info *info, int data_size,
  *
  * Return:     0 on success or a negative error number.
  */
-static int spl_load_fit_image(struct spl_load_info *info, ulong sector,
-                             const struct spl_fit_info *ctx, int node,
-                             struct spl_image_info *image_info)
+static int load_simple_fit(struct spl_load_info *info, ulong sector,
+                          const struct spl_fit_info *ctx, int node,
+                          struct spl_image_info *image_info)
 {
        int offset;
        size_t length;
@@ -400,8 +400,7 @@ static int spl_fit_append_fdt(struct spl_image_info *spl_image,
                else
                        return node;
        } else {
-               ret = spl_load_fit_image(info, sector, ctx, node,
-                                        &image_info);
+               ret = load_simple_fit(info, sector, ctx, node, &image_info);
                if (ret < 0)
                        return ret;
        }
@@ -440,8 +439,8 @@ static int spl_fit_append_fdt(struct spl_image_info *spl_image,
                                              __func__);
                        }
                        image_info.load_addr = (ulong)tmpbuffer;
-                       ret = spl_load_fit_image(info, sector, ctx,
-                                                node, &image_info);
+                       ret = load_simple_fit(info, sector, ctx, node,
+                                             &image_info);
                        if (ret < 0)
                                break;
 
@@ -491,10 +490,11 @@ static int spl_fit_record_loadable(const struct spl_fit_info *ctx, int index,
        node = spl_fit_get_image_node(ctx, "loadables", index);
 
        ret = fdt_record_loadable(blob, index, name, image->load_addr,
-                                 image->size, image->entry_point,
-                                 fdt_getprop(ctx->fit, node, "type", NULL),
-                                 fdt_getprop(ctx->fit, node, "os", NULL),
-                                 fdt_getprop(ctx->fit, node, "arch", NULL));
+                       image->size, image->entry_point,
+                       fdt_getprop(ctx->fit, node, FIT_TYPE_PROP, NULL),
+                       fdt_getprop(ctx->fit, node, FIT_OS_PROP, NULL),
+                       fdt_getprop(ctx->fit, node, FIT_ARCH_PROP, NULL));
+
        return ret;
 }
 
@@ -545,7 +545,7 @@ static void *spl_get_fit_load_buffer(size_t size)
        buf = malloc_cache_aligned(size);
        if (!buf) {
                pr_err("Could not get FIT buffer of %lu bytes\n", (ulong)size);
-               pr_err("\tcheck CONFIG_SYS_SPL_MALLOC_SIZE\n");
+               pr_err("\tcheck CONFIG_SPL_SYS_MALLOC_SIZE\n");
                buf = spl_get_load_buffer(0, size);
        }
        return buf;
@@ -631,7 +631,7 @@ static int spl_fit_load_fpga(struct spl_fit_info *ctx,
        warn_deprecated("'fpga' property in config node. Use 'loadables'");
 
        /* Load the image and set up the fpga_image structure */
-       ret = spl_load_fit_image(info, sector, ctx, node, &fpga_image);
+       ret = load_simple_fit(info, sector, ctx, node, &fpga_image);
        if (ret) {
                printf("%s: Cannot load the FPGA: %i\n", __func__, ret);
                return ret;
@@ -756,7 +756,7 @@ int spl_load_simple_fit(struct spl_image_info *spl_image,
        }
 
        /* Load the image and set up the spl_image structure */
-       ret = spl_load_fit_image(info, sector, &ctx, node, spl_image);
+       ret = load_simple_fit(info, sector, &ctx, node, spl_image);
        if (ret)
                return ret;
 
@@ -797,7 +797,7 @@ int spl_load_simple_fit(struct spl_image_info *spl_image,
                        continue;
 
                image_info.load_addr = 0;
-               ret = spl_load_fit_image(info, sector, &ctx, node, &image_info);
+               ret = load_simple_fit(info, sector, &ctx, node, &image_info);
                if (ret < 0) {
                        printf("%s: can't load image loadables index %d (ret = %d)\n",
                               __func__, index, ret);
@@ -842,3 +842,96 @@ int spl_load_simple_fit(struct spl_image_info *spl_image,
 
        return 0;
 }
+
+/* Parse and load full fitImage in SPL */
+int spl_load_fit_image(struct spl_image_info *spl_image,
+                      const struct legacy_img_hdr *header)
+{
+       struct bootm_headers images;
+       const char *fit_uname_config = NULL;
+       uintptr_t fdt_hack;
+       const char *uname;
+       ulong fw_data = 0, dt_data = 0, img_data = 0;
+       ulong fw_len = 0, dt_len = 0, img_len = 0;
+       int idx, conf_noffset;
+       int ret;
+
+#ifdef CONFIG_SPL_FIT_SIGNATURE
+       images.verify = 1;
+#endif
+       ret = fit_image_load(&images, (ulong)header,
+                            NULL, &fit_uname_config,
+                            IH_ARCH_DEFAULT, IH_TYPE_STANDALONE, -1,
+                            FIT_LOAD_OPTIONAL, &fw_data, &fw_len);
+       if (ret >= 0) {
+               printf("DEPRECATED: 'standalone = ' property.");
+               printf("Please use either 'firmware =' or 'kernel ='\n");
+       } else {
+               ret = fit_image_load(&images, (ulong)header, NULL,
+                                    &fit_uname_config, IH_ARCH_DEFAULT,
+                                    IH_TYPE_FIRMWARE, -1, FIT_LOAD_OPTIONAL,
+                                    &fw_data, &fw_len);
+       }
+
+       if (ret < 0) {
+               ret = fit_image_load(&images, (ulong)header, NULL,
+                                    &fit_uname_config, IH_ARCH_DEFAULT,
+                                    IH_TYPE_KERNEL, -1, FIT_LOAD_OPTIONAL,
+                                    &fw_data, &fw_len);
+       }
+
+       if (ret < 0)
+               return ret;
+
+       spl_image->size = fw_len;
+       spl_image->entry_point = fw_data;
+       spl_image->load_addr = fw_data;
+       if (fit_image_get_os(header, ret, &spl_image->os))
+               spl_image->os = IH_OS_INVALID;
+       spl_image->name = genimg_get_os_name(spl_image->os);
+
+       debug(SPL_TPL_PROMPT "payload image: %32s load addr: 0x%lx size: %d\n",
+             spl_image->name, spl_image->load_addr, spl_image->size);
+
+#ifdef CONFIG_SPL_FIT_SIGNATURE
+       images.verify = 1;
+#endif
+       ret = fit_image_load(&images, (ulong)header, NULL, &fit_uname_config,
+                            IH_ARCH_DEFAULT, IH_TYPE_FLATDT, -1,
+                            FIT_LOAD_OPTIONAL, &dt_data, &dt_len);
+       if (ret >= 0) {
+               spl_image->fdt_addr = (void *)dt_data;
+
+               if (spl_image->os == IH_OS_U_BOOT) {
+                       /* HACK: U-Boot expects FDT at a specific address */
+                       fdt_hack = spl_image->load_addr + spl_image->size;
+                       fdt_hack = (fdt_hack + 3) & ~3;
+                       debug("Relocating FDT to %p\n", spl_image->fdt_addr);
+                       memcpy((void *)fdt_hack, spl_image->fdt_addr, dt_len);
+               }
+       }
+
+       conf_noffset = fit_conf_get_node((const void *)header,
+                                        fit_uname_config);
+       if (conf_noffset < 0)
+               return 0;
+
+       for (idx = 0;
+            uname = fdt_stringlist_get((const void *)header, conf_noffset,
+                                       FIT_LOADABLE_PROP, idx,
+                               NULL), uname;
+            idx++) {
+#ifdef CONFIG_SPL_FIT_SIGNATURE
+               images.verify = 1;
+#endif
+               ret = fit_image_load(&images, (ulong)header,
+                                    &uname, &fit_uname_config,
+                                    IH_ARCH_DEFAULT, IH_TYPE_LOADABLE, -1,
+                                    FIT_LOAD_OPTIONAL_NON_ZERO,
+                                    &img_data, &img_len);
+               if (ret < 0)
+                       return ret;
+       }
+
+       return 0;
+}