xref: /rk3399_rockchip-uboot/common/spl/spl_fit.c (revision 39b9f515bcfc818f97d14ddffdb86218fccf930c)
1f1dcee59SSimon Glass /*
2f1dcee59SSimon Glass  * Copyright (C) 2016 Google, Inc
3f1dcee59SSimon Glass  * Written by Simon Glass <sjg@chromium.org>
4f1dcee59SSimon Glass  *
5f1dcee59SSimon Glass  * SPDX-License-Identifier:	GPL-2.0+
6f1dcee59SSimon Glass  */
7f1dcee59SSimon Glass 
8f1dcee59SSimon Glass #include <common.h>
9e12dde2dSJoseph Chen #include <boot_rkimg.h>
10f1dcee59SSimon Glass #include <errno.h>
11f1dcee59SSimon Glass #include <image.h>
12f1dcee59SSimon Glass #include <spl.h>
1390d1164aSKever Yang #include <malloc.h>
14a741b19cSJason Zhu #include <mtd_blk.h>
15a741b19cSJason Zhu #include <linux/libfdt.h>
16f1dcee59SSimon Glass 
17b81c4739SYork Sun #ifndef CONFIG_SYS_BOOTM_LEN
18b81c4739SYork Sun #define CONFIG_SYS_BOOTM_LEN	(64 << 20)
19b81c4739SYork Sun #endif
20b81c4739SYork Sun 
21736806fbSAndre Przywara /**
221f1cf67bSPhilipp Tomsich  * spl_fit_get_image_name(): By using the matching configuration subnode,
23736806fbSAndre Przywara  * retrieve the name of an image, specified by a property name and an index
24736806fbSAndre Przywara  * into that.
25736806fbSAndre Przywara  * @fit:	Pointer to the FDT blob.
26736806fbSAndre Przywara  * @images:	Offset of the /images subnode.
27736806fbSAndre Przywara  * @type:	Name of the property within the configuration subnode.
28736806fbSAndre Przywara  * @index:	Index into the list of strings in this property.
291f1cf67bSPhilipp Tomsich  * @outname:	Name of the image
30736806fbSAndre Przywara  *
311f1cf67bSPhilipp Tomsich  * Return:	0 on success, or a negative error number
32736806fbSAndre Przywara  */
331f1cf67bSPhilipp Tomsich static int spl_fit_get_image_name(const void *fit, int images,
341f1cf67bSPhilipp Tomsich 				  const char *type, int index,
351f1cf67bSPhilipp Tomsich 				  char **outname)
364b9340abSAndre Przywara {
374b9340abSAndre Przywara 	const char *name, *str;
381f1cf67bSPhilipp Tomsich 	__maybe_unused int node;
391f1cf67bSPhilipp Tomsich 	int conf_node;
404b9340abSAndre Przywara 	int len, i;
41f1dcee59SSimon Glass 
423863f840SCooper Jr., Franklin 	conf_node = fit_find_config_node(fit);
434b9340abSAndre Przywara 	if (conf_node < 0) {
44f1dcee59SSimon Glass #ifdef CONFIG_SPL_LIBCOMMON_SUPPORT
45f1dcee59SSimon Glass 		printf("No matching DT out of these options:\n");
464b9340abSAndre Przywara 		for (node = fdt_first_subnode(fit, conf_node);
47f1dcee59SSimon Glass 		     node >= 0;
484b9340abSAndre Przywara 		     node = fdt_next_subnode(fit, node)) {
494b9340abSAndre Przywara 			name = fdt_getprop(fit, node, "description", &len);
50f1dcee59SSimon Glass 			printf("   %s\n", name);
51f1dcee59SSimon Glass 		}
52f1dcee59SSimon Glass #endif
534b9340abSAndre Przywara 		return conf_node;
544b9340abSAndre Przywara 	}
55f1dcee59SSimon Glass 
564b9340abSAndre Przywara 	name = fdt_getprop(fit, conf_node, type, &len);
574b9340abSAndre Przywara 	if (!name) {
584b9340abSAndre Przywara 		debug("cannot find property '%s': %d\n", type, len);
594b9340abSAndre Przywara 		return -EINVAL;
604b9340abSAndre Przywara 	}
614b9340abSAndre Przywara 
624b9340abSAndre Przywara 	str = name;
634b9340abSAndre Przywara 	for (i = 0; i < index; i++) {
644b9340abSAndre Przywara 		str = strchr(str, '\0') + 1;
654b9340abSAndre Przywara 		if (!str || (str - name >= len)) {
664b9340abSAndre Przywara 			debug("no string for index %d\n", index);
674b9340abSAndre Przywara 			return -E2BIG;
684b9340abSAndre Przywara 		}
694b9340abSAndre Przywara 	}
704b9340abSAndre Przywara 
711f1cf67bSPhilipp Tomsich 	*outname = (char *)str;
721f1cf67bSPhilipp Tomsich 	return 0;
731f1cf67bSPhilipp Tomsich }
741f1cf67bSPhilipp Tomsich 
751f1cf67bSPhilipp Tomsich /**
761f1cf67bSPhilipp Tomsich  * spl_fit_get_image_node(): By using the matching configuration subnode,
771f1cf67bSPhilipp Tomsich  * retrieve the name of an image, specified by a property name and an index
781f1cf67bSPhilipp Tomsich  * into that.
791f1cf67bSPhilipp Tomsich  * @fit:	Pointer to the FDT blob.
801f1cf67bSPhilipp Tomsich  * @images:	Offset of the /images subnode.
811f1cf67bSPhilipp Tomsich  * @type:	Name of the property within the configuration subnode.
821f1cf67bSPhilipp Tomsich  * @index:	Index into the list of strings in this property.
831f1cf67bSPhilipp Tomsich  *
841f1cf67bSPhilipp Tomsich  * Return:	the node offset of the respective image node or a negative
851f1cf67bSPhilipp Tomsich  *		error number.
861f1cf67bSPhilipp Tomsich  */
871f1cf67bSPhilipp Tomsich static int spl_fit_get_image_node(const void *fit, int images,
881f1cf67bSPhilipp Tomsich 				  const char *type, int index)
891f1cf67bSPhilipp Tomsich {
901f1cf67bSPhilipp Tomsich 	char *str;
911f1cf67bSPhilipp Tomsich 	int err;
921f1cf67bSPhilipp Tomsich 	int node;
931f1cf67bSPhilipp Tomsich 
941f1cf67bSPhilipp Tomsich 	err = spl_fit_get_image_name(fit, images, type, index, &str);
951f1cf67bSPhilipp Tomsich 	if (err)
961f1cf67bSPhilipp Tomsich 		return err;
971f1cf67bSPhilipp Tomsich 
984b9340abSAndre Przywara 	debug("%s: '%s'\n", type, str);
991f1cf67bSPhilipp Tomsich 
1004b9340abSAndre Przywara 	node = fdt_subnode_offset(fit, images, str);
1014b9340abSAndre Przywara 	if (node < 0) {
1024b9340abSAndre Przywara 		debug("cannot find image node '%s': %d\n", str, node);
1034b9340abSAndre Przywara 		return -EINVAL;
1044b9340abSAndre Przywara 	}
1054b9340abSAndre Przywara 
106736806fbSAndre Przywara 	return node;
107f1dcee59SSimon Glass }
108f1dcee59SSimon Glass 
109eafd5410SLokesh Vutla static int get_aligned_image_offset(struct spl_load_info *info, int offset)
110eafd5410SLokesh Vutla {
111eafd5410SLokesh Vutla 	/*
112eafd5410SLokesh Vutla 	 * If it is a FS read, get the first address before offset which is
113eafd5410SLokesh Vutla 	 * aligned to ARCH_DMA_MINALIGN. If it is raw read return the
114eafd5410SLokesh Vutla 	 * block number to which offset belongs.
115eafd5410SLokesh Vutla 	 */
116eafd5410SLokesh Vutla 	if (info->filename)
117eafd5410SLokesh Vutla 		return offset & ~(ARCH_DMA_MINALIGN - 1);
118eafd5410SLokesh Vutla 
119eafd5410SLokesh Vutla 	return offset / info->bl_len;
120eafd5410SLokesh Vutla }
121eafd5410SLokesh Vutla 
122eafd5410SLokesh Vutla static int get_aligned_image_overhead(struct spl_load_info *info, int offset)
123eafd5410SLokesh Vutla {
124eafd5410SLokesh Vutla 	/*
125eafd5410SLokesh Vutla 	 * If it is a FS read, get the difference between the offset and
126eafd5410SLokesh Vutla 	 * the first address before offset which is aligned to
127eafd5410SLokesh Vutla 	 * ARCH_DMA_MINALIGN. If it is raw read return the offset within the
128eafd5410SLokesh Vutla 	 * block.
129eafd5410SLokesh Vutla 	 */
130eafd5410SLokesh Vutla 	if (info->filename)
131eafd5410SLokesh Vutla 		return offset & (ARCH_DMA_MINALIGN - 1);
132eafd5410SLokesh Vutla 
133eafd5410SLokesh Vutla 	return offset % info->bl_len;
134eafd5410SLokesh Vutla }
135eafd5410SLokesh Vutla 
136eafd5410SLokesh Vutla static int get_aligned_image_size(struct spl_load_info *info, int data_size,
137eafd5410SLokesh Vutla 				  int offset)
138eafd5410SLokesh Vutla {
1393cc1f380SLokesh Vutla 	data_size = data_size + get_aligned_image_overhead(info, offset);
1403cc1f380SLokesh Vutla 
141eafd5410SLokesh Vutla 	if (info->filename)
1423cc1f380SLokesh Vutla 		return data_size;
143eafd5410SLokesh Vutla 
144eafd5410SLokesh Vutla 	return (data_size + info->bl_len - 1) / info->bl_len;
145eafd5410SLokesh Vutla }
146eafd5410SLokesh Vutla 
1478baa3818SAndre Przywara /**
1488baa3818SAndre Przywara  * spl_load_fit_image(): load the image described in a certain FIT node
1498baa3818SAndre Przywara  * @info:	points to information about the device to load data from
1508baa3818SAndre Przywara  * @sector:	the start sector of the FIT image on the device
1518baa3818SAndre Przywara  * @fit:	points to the flattened device tree blob describing the FIT
1528baa3818SAndre Przywara  *		image
1538baa3818SAndre Przywara  * @base_offset: the beginning of the data area containing the actual
1548baa3818SAndre Przywara  *		image data, relative to the beginning of the FIT
1558baa3818SAndre Przywara  * @node:	offset of the DT node describing the image to load (relative
1568baa3818SAndre Przywara  *		to @fit)
1578baa3818SAndre Przywara  * @image_info:	will be filled with information about the loaded image
1588baa3818SAndre Przywara  *		If the FIT node does not contain a "load" (address) property,
1598baa3818SAndre Przywara  *		the image gets loaded to the address pointed to by the
1608baa3818SAndre Przywara  *		load_addr member in this struct.
1618baa3818SAndre Przywara  *
1628baa3818SAndre Przywara  * Return:	0 on success or a negative error number.
1638baa3818SAndre Przywara  */
1648baa3818SAndre Przywara static int spl_load_fit_image(struct spl_load_info *info, ulong sector,
1658baa3818SAndre Przywara 			      void *fit, ulong base_offset, int node,
1668baa3818SAndre Przywara 			      struct spl_image_info *image_info)
1678baa3818SAndre Przywara {
1680ea10b9fSYork Sun 	int offset;
1698baa3818SAndre Przywara 	size_t length;
1700ea10b9fSYork Sun 	int len;
171d18ad2e6SYork Sun 	ulong size;
172d3bfb68bSJoseph Chen 	ulong comp_addr, load_addr, load_ptr;
1738baa3818SAndre Przywara 	void *src;
1748baa3818SAndre Przywara 	ulong overhead;
1758baa3818SAndre Przywara 	int nr_sectors;
1768baa3818SAndre Przywara 	int align_len = ARCH_DMA_MINALIGN - 1;
177b81c4739SYork Sun 	uint8_t image_comp = -1, type = -1;
1780ea10b9fSYork Sun 	const void *data;
179b86dc419SPeng Fan 	bool external_data = false;
180b81c4739SYork Sun 
181b81c4739SYork Sun 	if (IS_ENABLED(CONFIG_SPL_OS_BOOT) && IS_ENABLED(CONFIG_SPL_GZIP)) {
182b81c4739SYork Sun 		if (fit_image_get_comp(fit, node, &image_comp))
183b81c4739SYork Sun 			puts("Cannot get image compression format.\n");
184b81c4739SYork Sun 		else
185b81c4739SYork Sun 			debug("%s ", genimg_get_comp_name(image_comp));
186b81c4739SYork Sun 
187b81c4739SYork Sun 		if (fit_image_get_type(fit, node, &type))
188b81c4739SYork Sun 			puts("Cannot get image type.\n");
189b81c4739SYork Sun 		else
190b81c4739SYork Sun 			debug("%s ", genimg_get_type_name(type));
191d3bfb68bSJoseph Chen 	} else {
192d3bfb68bSJoseph Chen 		fit_image_get_comp(fit, node, &image_comp);
193b81c4739SYork Sun 	}
1948baa3818SAndre Przywara 
1950ea10b9fSYork Sun 	if (fit_image_get_load(fit, node, &load_addr))
1968baa3818SAndre Przywara 		load_addr = image_info->load_addr;
1970ea10b9fSYork Sun 
198d3bfb68bSJoseph Chen 	if (image_comp != IH_COMP_NONE && image_comp != IH_COMP_ZIMAGE) {
199d3bfb68bSJoseph Chen 		/* Empirically, 1MB is enough for U-Boot, tee and atf */
200d3bfb68bSJoseph Chen 		if (fit_image_get_comp_addr(fit, node, &comp_addr))
201d3bfb68bSJoseph Chen 			comp_addr = load_addr + SZ_1M;
202d3bfb68bSJoseph Chen 	} else {
203d3bfb68bSJoseph Chen 		comp_addr = load_addr;
204d3bfb68bSJoseph Chen 	}
205d3bfb68bSJoseph Chen 
206b86dc419SPeng Fan 	if (!fit_image_get_data_position(fit, node, &offset)) {
207b86dc419SPeng Fan 		external_data = true;
208b86dc419SPeng Fan 	} else if (!fit_image_get_data_offset(fit, node, &offset)) {
2090ea10b9fSYork Sun 		offset += base_offset;
210b86dc419SPeng Fan 		external_data = true;
211b86dc419SPeng Fan 	}
212b86dc419SPeng Fan 
213b86dc419SPeng Fan 	if (external_data) {
214b86dc419SPeng Fan 		/* External data */
2150ea10b9fSYork Sun 		if (fit_image_get_data_size(fit, node, &len))
2160ea10b9fSYork Sun 			return -ENOENT;
2170ea10b9fSYork Sun 
218d3bfb68bSJoseph Chen 		load_ptr = (comp_addr + align_len) & ~align_len;
21963363c40SAndy Yan #if  defined(CONFIG_ARCH_ROCKCHIP)
220a5e7a2d4SKever Yang 		if ((load_ptr < CONFIG_SYS_SDRAM_BASE) ||
221a5e7a2d4SKever Yang 		     (load_ptr >= CONFIG_SYS_SDRAM_BASE + SDRAM_MAX_SIZE))
22290d1164aSKever Yang 			load_ptr = (ulong)memalign(ARCH_DMA_MINALIGN, len);
22390d1164aSKever Yang #endif
2240ea10b9fSYork Sun 		length = len;
2258baa3818SAndre Przywara 
2268baa3818SAndre Przywara 		overhead = get_aligned_image_overhead(info, offset);
2278baa3818SAndre Przywara 		nr_sectors = get_aligned_image_size(info, length, offset);
2288baa3818SAndre Przywara 
2290ea10b9fSYork Sun 		if (info->read(info,
2300ea10b9fSYork Sun 			       sector + get_aligned_image_offset(info, offset),
2318baa3818SAndre Przywara 			       nr_sectors, (void *)load_ptr) != nr_sectors)
2328baa3818SAndre Przywara 			return -EIO;
2338baa3818SAndre Przywara 
2340ea10b9fSYork Sun 		debug("External data: dst=%lx, offset=%x, size=%lx\n",
2350ea10b9fSYork Sun 		      load_ptr, offset, (unsigned long)length);
2368baa3818SAndre Przywara 		src = (void *)load_ptr + overhead;
2370ea10b9fSYork Sun 	} else {
2380ea10b9fSYork Sun 		/* Embedded data */
2390ea10b9fSYork Sun 		if (fit_image_get_data(fit, node, &data, &length)) {
2400ea10b9fSYork Sun 			puts("Cannot get image data/size\n");
2410ea10b9fSYork Sun 			return -ENOENT;
2420ea10b9fSYork Sun 		}
2430ea10b9fSYork Sun 		debug("Embedded data: dst=%lx, size=%lx\n", load_addr,
2440ea10b9fSYork Sun 		      (unsigned long)length);
2450ea10b9fSYork Sun 		src = (void *)data;
2460ea10b9fSYork Sun 	}
2470ea10b9fSYork Sun 
248a515b6d2SJoseph Chen 	/* Check hashes and signature */
249d3bfb68bSJoseph Chen 	if (image_comp != IH_COMP_NONE && image_comp != IH_COMP_ZIMAGE)
250d3bfb68bSJoseph Chen 		printf("## Checking %s 0x%08lx (%s @0x%08lx) ... ",
251d3bfb68bSJoseph Chen 		       fit_get_name(fit, node, NULL), load_addr,
252d3bfb68bSJoseph Chen 		       (char *)fdt_getprop(fit, node, FIT_COMP_PROP, NULL),
253d3bfb68bSJoseph Chen 		       (long)src);
254d3bfb68bSJoseph Chen 	else
255d3bfb68bSJoseph Chen 		printf("## Checking %s 0x%08lx ... ",
256d3bfb68bSJoseph Chen 		       fit_get_name(fit, node, NULL), load_addr);
257d3bfb68bSJoseph Chen 
258b92c7cf4SJoseph Chen #ifdef CONFIG_FIT_SPL_PRINT
259b92c7cf4SJoseph Chen 	printf("\n");
260b92c7cf4SJoseph Chen 	fit_image_print(fit, node, "");
261b92c7cf4SJoseph Chen #endif
262f3219d95SBen Whitten 	if (!fit_image_verify_with_data(fit, node,
263f3219d95SBen Whitten 					 src, length))
264f3219d95SBen Whitten 		return -EPERM;
265f3219d95SBen Whitten 
2668baa3818SAndre Przywara #ifdef CONFIG_SPL_FIT_IMAGE_POST_PROCESS
2679c633288SJoseph Chen 	board_fit_image_post_process(fit, node, (ulong *)&load_addr,
2689c633288SJoseph Chen 				     (ulong **)&src, &length);
2698baa3818SAndre Przywara #endif
27050466f58SJoseph Chen 	puts("OK\n");
2718baa3818SAndre Przywara 
272b81c4739SYork Sun 	if (IS_ENABLED(CONFIG_SPL_OS_BOOT)	&&
273b81c4739SYork Sun 	    IS_ENABLED(CONFIG_SPL_GZIP)		&&
274b81c4739SYork Sun 	    image_comp == IH_COMP_GZIP		&&
275b81c4739SYork Sun 	    type == IH_TYPE_KERNEL) {
276d18ad2e6SYork Sun 		size = length;
277b81c4739SYork Sun 		if (gunzip((void *)load_addr, CONFIG_SYS_BOOTM_LEN,
278d18ad2e6SYork Sun 			   src, &size)) {
279b81c4739SYork Sun 			puts("Uncompressing error\n");
280b81c4739SYork Sun 			return -EIO;
281b81c4739SYork Sun 		}
282d18ad2e6SYork Sun 		length = size;
283b81c4739SYork Sun 	} else {
2848baa3818SAndre Przywara 		memcpy((void *)load_addr, src, length);
285b81c4739SYork Sun 	}
2868baa3818SAndre Przywara 
2878baa3818SAndre Przywara 	if (image_info) {
2888baa3818SAndre Przywara 		image_info->load_addr = load_addr;
2898baa3818SAndre Przywara 		image_info->size = length;
2908baa3818SAndre Przywara 		image_info->entry_point = fdt_getprop_u32(fit, node, "entry");
2918baa3818SAndre Przywara 	}
2928baa3818SAndre Przywara 
2938baa3818SAndre Przywara 	return 0;
2948baa3818SAndre Przywara }
2958baa3818SAndre Przywara 
2969719e14fSPhilipp Tomsich static int spl_fit_append_fdt(struct spl_image_info *spl_image,
2979719e14fSPhilipp Tomsich 			      struct spl_load_info *info, ulong sector,
2989719e14fSPhilipp Tomsich 			      void *fit, int images, ulong base_offset)
2999719e14fSPhilipp Tomsich {
3009719e14fSPhilipp Tomsich 	struct spl_image_info image_info;
3019719e14fSPhilipp Tomsich 	int node, ret;
3029719e14fSPhilipp Tomsich 
3039719e14fSPhilipp Tomsich 	/* Figure out which device tree the board wants to use */
3049719e14fSPhilipp Tomsich 	node = spl_fit_get_image_node(fit, images, FIT_FDT_PROP, 0);
3059719e14fSPhilipp Tomsich 	if (node < 0) {
3069719e14fSPhilipp Tomsich 		debug("%s: cannot find FDT node\n", __func__);
3079719e14fSPhilipp Tomsich 		return node;
3089719e14fSPhilipp Tomsich 	}
3099719e14fSPhilipp Tomsich 
3109719e14fSPhilipp Tomsich 	/*
3119719e14fSPhilipp Tomsich 	 * Read the device tree and place it after the image.
3129719e14fSPhilipp Tomsich 	 * Align the destination address to ARCH_DMA_MINALIGN.
3139719e14fSPhilipp Tomsich 	 */
3149719e14fSPhilipp Tomsich 	image_info.load_addr = spl_image->load_addr + spl_image->size;
3159719e14fSPhilipp Tomsich 	ret = spl_load_fit_image(info, sector, fit, base_offset, node,
3169719e14fSPhilipp Tomsich 				 &image_info);
3171f1cf67bSPhilipp Tomsich 
3181f1cf67bSPhilipp Tomsich 	if (ret < 0)
3191f1cf67bSPhilipp Tomsich 		return ret;
3201f1cf67bSPhilipp Tomsich 
3211f1cf67bSPhilipp Tomsich 	/* Make the load-address of the FDT available for the SPL framework */
3221f1cf67bSPhilipp Tomsich 	spl_image->fdt_addr = (void *)image_info.load_addr;
3238118cce3SPhilipp Tomsich #if !CONFIG_IS_ENABLED(FIT_IMAGE_TINY)
3241f1cf67bSPhilipp Tomsich 	/* Try to make space, so we can inject details on the loadables */
3251f1cf67bSPhilipp Tomsich 	ret = fdt_shrink_to_minimum(spl_image->fdt_addr, 8192);
3268118cce3SPhilipp Tomsich #endif
3271f1cf67bSPhilipp Tomsich 
3289d23fbe3SJoseph Chen 	/*
3299d23fbe3SJoseph Chen 	 * If need, load kernel FDT right after U-Boot FDT.
3309d23fbe3SJoseph Chen 	 *
3319d23fbe3SJoseph Chen 	 * kernel FDT is for U-Boot if there is not valid one
3329d23fbe3SJoseph Chen 	 * from images, ie: resource.img, boot.img or recovery.img.
3339d23fbe3SJoseph Chen 	 */
3349d23fbe3SJoseph Chen 	node = fdt_subnode_offset(fit, images, FIT_KERNEL_FDT_PROP);
3359d23fbe3SJoseph Chen 	if (node < 0)
3369d23fbe3SJoseph Chen 		return ret;
3379d23fbe3SJoseph Chen 
3389d23fbe3SJoseph Chen 	image_info.load_addr =
3399d23fbe3SJoseph Chen 		(ulong)spl_image->fdt_addr + fdt_totalsize(spl_image->fdt_addr);
3409d23fbe3SJoseph Chen 	ret = spl_load_fit_image(info, sector, fit, base_offset, node,
3419d23fbe3SJoseph Chen 				 &image_info);
3429d23fbe3SJoseph Chen 
3431f1cf67bSPhilipp Tomsich 	return ret;
3441f1cf67bSPhilipp Tomsich }
3451f1cf67bSPhilipp Tomsich 
3461f1cf67bSPhilipp Tomsich static int spl_fit_record_loadable(const void *fit, int images, int index,
3471f1cf67bSPhilipp Tomsich 				   void *blob, struct spl_image_info *image)
3481f1cf67bSPhilipp Tomsich {
3498118cce3SPhilipp Tomsich 	int ret = 0;
3508118cce3SPhilipp Tomsich #if !CONFIG_IS_ENABLED(FIT_IMAGE_TINY)
3511f1cf67bSPhilipp Tomsich 	char *name;
3528118cce3SPhilipp Tomsich 	int node;
3531f1cf67bSPhilipp Tomsich 
3541f1cf67bSPhilipp Tomsich 	ret = spl_fit_get_image_name(fit, images, "loadables",
3551f1cf67bSPhilipp Tomsich 				     index, &name);
3561f1cf67bSPhilipp Tomsich 	if (ret < 0)
3571f1cf67bSPhilipp Tomsich 		return ret;
3581f1cf67bSPhilipp Tomsich 
3591f1cf67bSPhilipp Tomsich 	node = spl_fit_get_image_node(fit, images, "loadables", index);
3601f1cf67bSPhilipp Tomsich 
3611f1cf67bSPhilipp Tomsich 	ret = fdt_record_loadable(blob, index, name, image->load_addr,
3621f1cf67bSPhilipp Tomsich 				  image->size, image->entry_point,
3631f1cf67bSPhilipp Tomsich 				  fdt_getprop(fit, node, "type", NULL),
3641f1cf67bSPhilipp Tomsich 				  fdt_getprop(fit, node, "os", NULL));
3658118cce3SPhilipp Tomsich #endif
3669719e14fSPhilipp Tomsich 	return ret;
3679719e14fSPhilipp Tomsich }
3689719e14fSPhilipp Tomsich 
3698118cce3SPhilipp Tomsich static int spl_fit_image_get_os(const void *fit, int noffset, uint8_t *os)
3708118cce3SPhilipp Tomsich {
3718118cce3SPhilipp Tomsich #if CONFIG_IS_ENABLED(FIT_IMAGE_TINY)
3728118cce3SPhilipp Tomsich 	return -ENOTSUPP;
3738118cce3SPhilipp Tomsich #else
3748118cce3SPhilipp Tomsich 	return fit_image_get_os(fit, noffset, os);
3758118cce3SPhilipp Tomsich #endif
3768118cce3SPhilipp Tomsich }
3778118cce3SPhilipp Tomsich 
3789a65720bSJason Zhu __weak int spl_fit_standalone_release(uintptr_t entry_point)
379342d050eSJason Zhu {
380342d050eSJason Zhu 	return 0;
381342d050eSJason Zhu }
382342d050eSJason Zhu 
383569a1737SJoseph Chen static void *spl_fit_load_blob(struct spl_load_info *info,
384569a1737SJoseph Chen 			       ulong sector, void *fit_header,
385569a1737SJoseph Chen 			       int *base_offset)
386f1dcee59SSimon Glass {
387569a1737SJoseph Chen 	int align_len = ARCH_DMA_MINALIGN - 1;
388569a1737SJoseph Chen 	ulong count;
3898baa3818SAndre Przywara 	ulong size;
390569a1737SJoseph Chen 	int sectors;
391569a1737SJoseph Chen 	void *fit;
392f1dcee59SSimon Glass 
393f1dcee59SSimon Glass 	/*
394ebec805aSYork Sun 	 * For FIT with external data, figure out where the external images
395ebec805aSYork Sun 	 * start. This is the base for the data-offset properties in each
396ebec805aSYork Sun 	 * image.
397f1dcee59SSimon Glass 	 */
398569a1737SJoseph Chen 	size = fdt_totalsize(fit_header);
399acd43290SJoseph Chen 	size = FIT_ALIGN(size);
400569a1737SJoseph Chen 	*base_offset = FIT_ALIGN(size);
401f1dcee59SSimon Glass 
402f1dcee59SSimon Glass 	/*
403f1dcee59SSimon Glass 	 * So far we only have one block of data from the FIT. Read the entire
404f1dcee59SSimon Glass 	 * thing, including that first block, placing it so it finishes before
405f1dcee59SSimon Glass 	 * where we will load the image.
406f1dcee59SSimon Glass 	 *
407f1dcee59SSimon Glass 	 * Note that we will load the image such that its first byte will be
408f1dcee59SSimon Glass 	 * at the load address. Since that byte may be part-way through a
409f1dcee59SSimon Glass 	 * block, we may load the image up to one block before the load
410f1dcee59SSimon Glass 	 * address. So take account of that here by subtracting an addition
411f1dcee59SSimon Glass 	 * block length from the FIT start position.
412f1dcee59SSimon Glass 	 *
413f1dcee59SSimon Glass 	 * In fact the FIT has its own load address, but we assume it cannot
414f1dcee59SSimon Glass 	 * be before CONFIG_SYS_TEXT_BASE.
415ebec805aSYork Sun 	 *
416ebec805aSYork Sun 	 * For FIT with data embedded, data is loaded as part of FIT image.
417ebec805aSYork Sun 	 * For FIT with external data, data is not loaded in this step.
418f1dcee59SSimon Glass 	 */
4198b528709SLokesh Vutla 	fit = (void *)((CONFIG_SYS_TEXT_BASE - size - info->bl_len -
4208b528709SLokesh Vutla 			align_len) & ~align_len);
421eafd5410SLokesh Vutla 	sectors = get_aligned_image_size(info, size, 0);
422f1dcee59SSimon Glass 	count = info->read(info, sector, sectors, fit);
423a741b19cSJason Zhu #ifdef CONFIG_MTD_BLK
424a741b19cSJason Zhu 	mtd_blk_map_fit(info->dev, sector, fit);
425a741b19cSJason Zhu #endif
426f1dcee59SSimon Glass 	debug("fit read sector %lx, sectors=%d, dst=%p, count=%lu\n",
427f1dcee59SSimon Glass 	      sector, sectors, fit, count);
428f1dcee59SSimon Glass 	if (count == 0)
429569a1737SJoseph Chen 		return NULL;
430569a1737SJoseph Chen 
431569a1737SJoseph Chen 	return fit;
432569a1737SJoseph Chen }
433569a1737SJoseph Chen 
434a00b516fSJoseph Chen #ifdef CONFIG_SPL_KERNEL_BOOT
435e12dde2dSJoseph Chen #ifdef CONFIG_SPL_LIBDISK_SUPPORT
436e12dde2dSJoseph Chen __weak const char *spl_kernel_partition(struct spl_image_info *spl,
437e12dde2dSJoseph Chen 					struct spl_load_info *info)
438e12dde2dSJoseph Chen {
439e12dde2dSJoseph Chen 	return PART_BOOT;
440e12dde2dSJoseph Chen }
441e12dde2dSJoseph Chen #endif
442e12dde2dSJoseph Chen 
443e12dde2dSJoseph Chen static int spl_load_kernel_fit(struct spl_image_info *spl_image,
444e12dde2dSJoseph Chen 			       struct spl_load_info *info)
445e12dde2dSJoseph Chen {
446e12dde2dSJoseph Chen 	/*
447e12dde2dSJoseph Chen 	 * Never change the image order.
448e12dde2dSJoseph Chen 	 *
449e12dde2dSJoseph Chen 	 * Considering thunder-boot feature, there maybe asynchronous
450e12dde2dSJoseph Chen 	 * loading operation of these images and ramdisk is usually to
451e12dde2dSJoseph Chen 	 * be the last one.
452e12dde2dSJoseph Chen 	 *
453e12dde2dSJoseph Chen 	 * The .its content rule of kernel fit image follows U-Boot proper.
454e12dde2dSJoseph Chen 	 */
455e12dde2dSJoseph Chen 	const char *images[] = { FIT_FDT_PROP, FIT_KERNEL_PROP, FIT_RAMDISK_PROP, };
456e12dde2dSJoseph Chen 	struct spl_image_info image_info;
457e12dde2dSJoseph Chen 	char fit_header[info->bl_len];
458e12dde2dSJoseph Chen 	int images_noffset;
459e12dde2dSJoseph Chen 	int base_offset;
460e12dde2dSJoseph Chen 	int sector;
461e12dde2dSJoseph Chen 	int node, ret, i;
462e12dde2dSJoseph Chen 	void *fit;
463e12dde2dSJoseph Chen 
464e12dde2dSJoseph Chen 	if (spl_image->next_stage != SPL_NEXT_STAGE_KERNEL)
465e12dde2dSJoseph Chen 		return 0;
466e12dde2dSJoseph Chen 
467e12dde2dSJoseph Chen #ifdef CONFIG_SPL_LIBDISK_SUPPORT
468e12dde2dSJoseph Chen 	const char *part_name = PART_BOOT;
469e12dde2dSJoseph Chen 	disk_partition_t part_info;
470e12dde2dSJoseph Chen 
471e12dde2dSJoseph Chen 	part_name = spl_kernel_partition(spl_image, info);
472e12dde2dSJoseph Chen 	if (part_get_info_by_name(info->dev, part_name, &part_info) <= 0) {
473e12dde2dSJoseph Chen 		printf("%s: no partition\n", __func__);
474e12dde2dSJoseph Chen 		return -EINVAL;
475e12dde2dSJoseph Chen 	}
476e12dde2dSJoseph Chen 	sector = part_info.start;
477e12dde2dSJoseph Chen #else
478a00b516fSJoseph Chen 	sector = CONFIG_SPL_KERNEL_BOOT_SECTOR;
479e12dde2dSJoseph Chen #endif
480e12dde2dSJoseph Chen 	if (info->read(info, sector, 1, &fit_header) != 1) {
48129725e85SJoseph Chen 		debug("%s: Failed to read header\n", __func__);
482e12dde2dSJoseph Chen 		return -EIO;
483e12dde2dSJoseph Chen 	}
484e12dde2dSJoseph Chen 
485e12dde2dSJoseph Chen 	if (image_get_magic((void *)&fit_header) != FDT_MAGIC) {
48629725e85SJoseph Chen 		printf("%s: Not fit magic\n", __func__);
487e12dde2dSJoseph Chen 		return -EINVAL;
488e12dde2dSJoseph Chen 	}
489e12dde2dSJoseph Chen 
490e12dde2dSJoseph Chen 	fit = spl_fit_load_blob(info, sector, fit_header, &base_offset);
491e12dde2dSJoseph Chen 	if (!fit) {
492e12dde2dSJoseph Chen 		debug("%s: Cannot load blob\n", __func__);
493e12dde2dSJoseph Chen 		return -ENODEV;
494e12dde2dSJoseph Chen 	}
495e12dde2dSJoseph Chen 
496e12dde2dSJoseph Chen 	/* verify the configure node by keys, if required */
497e12dde2dSJoseph Chen #ifdef CONFIG_SPL_FIT_SIGNATURE
498e12dde2dSJoseph Chen 	int conf_noffset;
499e12dde2dSJoseph Chen 
500e12dde2dSJoseph Chen 	conf_noffset = fit_conf_get_node(fit, NULL);
501e12dde2dSJoseph Chen 	if (conf_noffset <= 0) {
502e12dde2dSJoseph Chen 		printf("No default config node\n");
503e12dde2dSJoseph Chen 		return -EINVAL;
504e12dde2dSJoseph Chen 	}
505e12dde2dSJoseph Chen 
506e12dde2dSJoseph Chen 	ret = fit_config_verify(fit, conf_noffset);
507e12dde2dSJoseph Chen 	if (ret) {
508e12dde2dSJoseph Chen 		printf("fit verify configure failed, ret=%d\n", ret);
509e12dde2dSJoseph Chen 		return ret;
510e12dde2dSJoseph Chen 	}
511e12dde2dSJoseph Chen 	printf("\n");
512e12dde2dSJoseph Chen #endif
513e12dde2dSJoseph Chen 	images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH);
514e12dde2dSJoseph Chen 	if (images_noffset < 0) {
515e12dde2dSJoseph Chen 		debug("%s: Cannot find /images node: %d\n",
516e12dde2dSJoseph Chen 		      __func__, images_noffset);
517e12dde2dSJoseph Chen 		return images_noffset;
518e12dde2dSJoseph Chen 	}
519e12dde2dSJoseph Chen 
520e12dde2dSJoseph Chen 	for (i = 0; i < ARRAY_SIZE(images); i++) {
521e12dde2dSJoseph Chen 		node = spl_fit_get_image_node(fit, images_noffset,
522e12dde2dSJoseph Chen 					      images[i], 0);
523e12dde2dSJoseph Chen 		if (node < 0) {
524e12dde2dSJoseph Chen 			debug("No image: %s\n", images[i]);
525e12dde2dSJoseph Chen 			continue;
526e12dde2dSJoseph Chen 		}
527e12dde2dSJoseph Chen 
528e12dde2dSJoseph Chen 		ret = spl_load_fit_image(info, sector, fit, base_offset,
529e12dde2dSJoseph Chen 					 node, &image_info);
530e12dde2dSJoseph Chen 		if (ret)
531e12dde2dSJoseph Chen 			return ret;
532e12dde2dSJoseph Chen 
533e12dde2dSJoseph Chen 		/* initial addr or entry point */
534e12dde2dSJoseph Chen 		if (!strcmp(images[i], FIT_FDT_PROP))
535e12dde2dSJoseph Chen 			spl_image->fdt_addr = (void *)image_info.load_addr;
536e12dde2dSJoseph Chen 		else if (!strcmp(images[i], FIT_KERNEL_PROP))
537a00b516fSJoseph Chen #if CONFIG_IS_ENABLED(OPTEE)
538e12dde2dSJoseph Chen 			spl_image->entry_point_os = image_info.load_addr;
539a00b516fSJoseph Chen #endif
540a00b516fSJoseph Chen #if CONFIG_IS_ENABLED(ATF)
541a00b516fSJoseph Chen 			spl_image->entry_point_bl33 = image_info.load_addr;
542a00b516fSJoseph Chen #endif
543e12dde2dSJoseph Chen 	}
544e12dde2dSJoseph Chen 
545a00b516fSJoseph Chen 	debug("fdt_addr=0x%08lx, entry_point=0x%08lx, entry_point_os=0x%08lx\n",
546a00b516fSJoseph Chen 	      (ulong)spl_image->fdt_addr,
547a00b516fSJoseph Chen 	      spl_image->entry_point,
548a00b516fSJoseph Chen #if CONFIG_IS_ENABLED(OPTEE)
549a00b516fSJoseph Chen 	      spl_image->entry_point_os);
550a00b516fSJoseph Chen #endif
551a00b516fSJoseph Chen #if CONFIG_IS_ENABLED(ATF)
552a00b516fSJoseph Chen 	      spl_image->entry_point_bl33);
553a00b516fSJoseph Chen #endif
554e12dde2dSJoseph Chen 
555e12dde2dSJoseph Chen 	return 0;
556e12dde2dSJoseph Chen }
557e12dde2dSJoseph Chen #endif
558e12dde2dSJoseph Chen 
559569a1737SJoseph Chen static int spl_internal_load_simple_fit(struct spl_image_info *spl_image,
560569a1737SJoseph Chen 					struct spl_load_info *info,
561569a1737SJoseph Chen 					ulong sector, void *fit_header)
562569a1737SJoseph Chen {
563569a1737SJoseph Chen 	struct spl_image_info image_info;
564569a1737SJoseph Chen 	int base_offset;
565569a1737SJoseph Chen 	int images, ret;
566569a1737SJoseph Chen 	int index = 0;
567569a1737SJoseph Chen 	int node = -1;
568569a1737SJoseph Chen 	void *fit;
569569a1737SJoseph Chen 
570569a1737SJoseph Chen 	fit = spl_fit_load_blob(info, sector, fit_header, &base_offset);
571569a1737SJoseph Chen 	if (!fit) {
572569a1737SJoseph Chen 		debug("%s: Cannot load blob\n", __func__);
573569a1737SJoseph Chen 		return -1;
574569a1737SJoseph Chen 	}
575f1dcee59SSimon Glass 
576736806fbSAndre Przywara 	/* find the node holding the images information */
577f1dcee59SSimon Glass 	images = fdt_path_offset(fit, FIT_IMAGES_PATH);
578f1dcee59SSimon Glass 	if (images < 0) {
579f1dcee59SSimon Glass 		debug("%s: Cannot find /images node: %d\n", __func__, images);
580f1dcee59SSimon Glass 		return -1;
581f1dcee59SSimon Glass 	}
582736806fbSAndre Przywara 
5831f452cbfSJoseph Chen 	/* if board sigs verify required, check self */
5841f452cbfSJoseph Chen 	if (fit_board_verify_required_sigs() &&
5851f452cbfSJoseph Chen 	    !IS_ENABLED(CONFIG_SPL_FIT_SIGNATURE)) {
5861f452cbfSJoseph Chen 		printf("Verified-boot requires CONFIG_SPL_FIT_SIGNATURE enabled\n");
5871f452cbfSJoseph Chen 		hang();
5881f452cbfSJoseph Chen 	}
5891f452cbfSJoseph Chen 
590583377c4SJoseph Chen 	/* verify the configure node by keys, if required */
591583377c4SJoseph Chen #ifdef CONFIG_SPL_FIT_SIGNATURE
592583377c4SJoseph Chen 	int conf_noffset;
593583377c4SJoseph Chen 
594583377c4SJoseph Chen 	conf_noffset = fit_conf_get_node(fit, NULL);
595a7560f55SJoseph Chen 	if (conf_noffset <= 0) {
596a7560f55SJoseph Chen 		printf("No default config node\n");
597a7560f55SJoseph Chen 		return -EINVAL;
598a7560f55SJoseph Chen 	}
599a7560f55SJoseph Chen 
600583377c4SJoseph Chen 	ret = fit_config_verify(fit, conf_noffset);
601583377c4SJoseph Chen 	if (ret) {
602583377c4SJoseph Chen 		printf("fit verify configure failed, ret=%d\n", ret);
603583377c4SJoseph Chen 		return ret;
604583377c4SJoseph Chen 	}
605583377c4SJoseph Chen 	printf("\n");
606583377c4SJoseph Chen 
6077a137075SJoseph Chen #ifdef CONFIG_SPL_FIT_ROLLBACK_PROTECT
6087a137075SJoseph Chen 	uint32_t this_index, min_index;
6097a137075SJoseph Chen 
6107a137075SJoseph Chen 	ret = fit_rollback_index_verify(fit, FIT_ROLLBACK_INDEX_SPL,
6117a137075SJoseph Chen 					&this_index, &min_index);
6127a137075SJoseph Chen 	if (ret) {
6137a137075SJoseph Chen 		printf("fit failed to get rollback index, ret=%d\n", ret);
6147a137075SJoseph Chen 		return ret;
6157a137075SJoseph Chen 	} else if (this_index < min_index) {
6167a137075SJoseph Chen 		printf("fit reject rollback: %d < %d(min)\n",
6177a137075SJoseph Chen 		       this_index, min_index);
6187a137075SJoseph Chen 		return -EINVAL;
6197a137075SJoseph Chen 	}
6207a137075SJoseph Chen 
621e5ca21e8SJoseph Chen 	printf("rollback index: %d >= %d(min), OK\n", this_index, min_index);
6227a137075SJoseph Chen #endif
6237a137075SJoseph Chen #endif
624342d050eSJason Zhu 
625342d050eSJason Zhu 	/*
626342d050eSJason Zhu 	 * If required to start the other core before load "loadables"
627342d050eSJason Zhu 	 * firmwares, use the config "standalone" to load the other core's
628342d050eSJason Zhu 	 * firmware, then start it.
629342d050eSJason Zhu 	 * Normally, different cores' firmware is attach to the config
630342d050eSJason Zhu 	 * "loadables" and load them together.
631342d050eSJason Zhu 	 */
632342d050eSJason Zhu 	if (node < 0)
633342d050eSJason Zhu 		node = spl_fit_get_image_node(fit, images, FIT_STANDALONE_PROP,
634342d050eSJason Zhu 					      0);
635342d050eSJason Zhu 	if (node > 0) {
636342d050eSJason Zhu 		/* Load the image and set up the spl_image structure */
637342d050eSJason Zhu 		ret = spl_load_fit_image(info, sector, fit, base_offset, node,
6389a65720bSJason Zhu 					 &image_info);
639342d050eSJason Zhu 		if (!ret) {
6409a65720bSJason Zhu 			if (image_info.entry_point == FDT_ERROR)
6419a65720bSJason Zhu 				image_info.entry_point = image_info.load_addr;
6429a65720bSJason Zhu 
6439a65720bSJason Zhu 			ret = spl_fit_standalone_release(image_info.entry_point);
644342d050eSJason Zhu 			if (ret)
645342d050eSJason Zhu 				printf("Start standalone fail, ret = %d\n",
646342d050eSJason Zhu 				       ret);
647342d050eSJason Zhu 		}
648342d050eSJason Zhu 
649e12dde2dSJoseph Chen 		/* standalone is special one, continue to find others */
650342d050eSJason Zhu 		node = -1;
651342d050eSJason Zhu 	}
652342d050eSJason Zhu 
6539719e14fSPhilipp Tomsich 	/*
6549719e14fSPhilipp Tomsich 	 * Find the U-Boot image using the following search order:
6559719e14fSPhilipp Tomsich 	 *   - start at 'firmware' (e.g. an ARM Trusted Firmware)
6569719e14fSPhilipp Tomsich 	 *   - fall back 'kernel' (e.g. a Falcon-mode OS boot
6579719e14fSPhilipp Tomsich 	 *   - fall back to using the first 'loadables' entry
6589719e14fSPhilipp Tomsich 	 */
659ebec805aSYork Sun 	if (node < 0)
660c04fe8bfSMichal Simek 		node = spl_fit_get_image_node(fit, images, FIT_FIRMWARE_PROP,
661c04fe8bfSMichal Simek 					      0);
6629719e14fSPhilipp Tomsich #ifdef CONFIG_SPL_OS_BOOT
6639719e14fSPhilipp Tomsich 	if (node < 0)
6649719e14fSPhilipp Tomsich 		node = spl_fit_get_image_node(fit, images, FIT_KERNEL_PROP, 0);
6659719e14fSPhilipp Tomsich #endif
666f1dcee59SSimon Glass 	if (node < 0) {
667736806fbSAndre Przywara 		debug("could not find firmware image, trying loadables...\n");
668736806fbSAndre Przywara 		node = spl_fit_get_image_node(fit, images, "loadables", 0);
669411cf32dSAndre Przywara 		/*
670411cf32dSAndre Przywara 		 * If we pick the U-Boot image from "loadables", start at
671411cf32dSAndre Przywara 		 * the second image when later loading additional images.
672411cf32dSAndre Przywara 		 */
673411cf32dSAndre Przywara 		index = 1;
674736806fbSAndre Przywara 	}
675736806fbSAndre Przywara 	if (node < 0) {
676736806fbSAndre Przywara 		debug("%s: Cannot find u-boot image node: %d\n",
677736806fbSAndre Przywara 		      __func__, node);
678f1dcee59SSimon Glass 		return -1;
679f1dcee59SSimon Glass 	}
680f1dcee59SSimon Glass 
6818baa3818SAndre Przywara 	/* Load the image and set up the spl_image structure */
6828baa3818SAndre Przywara 	ret = spl_load_fit_image(info, sector, fit, base_offset, node,
6838baa3818SAndre Przywara 				 spl_image);
6848baa3818SAndre Przywara 	if (ret)
6858baa3818SAndre Przywara 		return ret;
6868baa3818SAndre Przywara 
6879719e14fSPhilipp Tomsich 	/*
6889719e14fSPhilipp Tomsich 	 * For backward compatibility, we treat the first node that is
6899719e14fSPhilipp Tomsich 	 * as a U-Boot image, if no OS-type has been declared.
6909719e14fSPhilipp Tomsich 	 */
6918118cce3SPhilipp Tomsich 	if (!spl_fit_image_get_os(fit, node, &spl_image->os))
692ebec805aSYork Sun 		debug("Image OS is %s\n", genimg_get_os_name(spl_image->os));
6939719e14fSPhilipp Tomsich #if !defined(CONFIG_SPL_OS_BOOT)
6949719e14fSPhilipp Tomsich 	else
695f4d7d859SSimon Glass 		spl_image->os = IH_OS_U_BOOT;
696ebec805aSYork Sun #endif
697f1dcee59SSimon Glass 
698f1dcee59SSimon Glass 	/*
6999719e14fSPhilipp Tomsich 	 * Booting a next-stage U-Boot may require us to append the FDT.
7009719e14fSPhilipp Tomsich 	 * We allow this to fail, as the U-Boot image might embed its FDT.
701f1dcee59SSimon Glass 	 */
7029719e14fSPhilipp Tomsich 	if (spl_image->os == IH_OS_U_BOOT)
7039719e14fSPhilipp Tomsich 		spl_fit_append_fdt(spl_image, info, sector, fit,
7049719e14fSPhilipp Tomsich 				   images, base_offset);
705411cf32dSAndre Przywara 
706411cf32dSAndre Przywara 	/* Now check if there are more images for us to load */
707411cf32dSAndre Przywara 	for (; ; index++) {
7089719e14fSPhilipp Tomsich 		uint8_t os_type = IH_OS_INVALID;
7099719e14fSPhilipp Tomsich 
710411cf32dSAndre Przywara 		node = spl_fit_get_image_node(fit, images, "loadables", index);
711411cf32dSAndre Przywara 		if (node < 0)
712411cf32dSAndre Przywara 			break;
713411cf32dSAndre Przywara 
714e12dde2dSJoseph Chen 		if (!spl_fit_image_get_os(fit, node, &os_type))
715e12dde2dSJoseph Chen 			debug("Loadable is %s\n", genimg_get_os_name(os_type));
716e12dde2dSJoseph Chen 
717e12dde2dSJoseph Chen 		/* skip U-Boot ? */
718e12dde2dSJoseph Chen 		if (spl_image->next_stage == SPL_NEXT_STAGE_KERNEL &&
719e12dde2dSJoseph Chen 		    os_type == IH_OS_U_BOOT)
720e12dde2dSJoseph Chen 		    continue;
721e12dde2dSJoseph Chen 
722411cf32dSAndre Przywara 		ret = spl_load_fit_image(info, sector, fit, base_offset, node,
723411cf32dSAndre Przywara 					 &image_info);
724411cf32dSAndre Przywara 		if (ret < 0)
725411cf32dSAndre Przywara 			continue;
726411cf32dSAndre Przywara 
7271f1cf67bSPhilipp Tomsich 		if (os_type == IH_OS_U_BOOT) {
728*39b9f515SJoseph Chen #if CONFIG_IS_ENABLED(ATF)
729*39b9f515SJoseph Chen 			spl_image->entry_point_bl33 = image_info.load_addr;
730*39b9f515SJoseph Chen #elif CONFIG_IS_ENABLED(OPTEE)
731*39b9f515SJoseph Chen 			spl_image->entry_point_os = image_info.load_addr;
732*39b9f515SJoseph Chen #endif
7331f1cf67bSPhilipp Tomsich 			spl_fit_append_fdt(&image_info, info, sector,
7349719e14fSPhilipp Tomsich 					   fit, images, base_offset);
7351f1cf67bSPhilipp Tomsich 			spl_image->fdt_addr = image_info.fdt_addr;
7361f1cf67bSPhilipp Tomsich 		}
7379719e14fSPhilipp Tomsich 
738411cf32dSAndre Przywara 		/*
739411cf32dSAndre Przywara 		 * If the "firmware" image did not provide an entry point,
740411cf32dSAndre Przywara 		 * use the first valid entry point from the loadables.
741411cf32dSAndre Przywara 		 */
742411cf32dSAndre Przywara 		if (spl_image->entry_point == FDT_ERROR &&
743411cf32dSAndre Przywara 		    image_info.entry_point != FDT_ERROR)
744411cf32dSAndre Przywara 			spl_image->entry_point = image_info.entry_point;
7451f1cf67bSPhilipp Tomsich 
7461f1cf67bSPhilipp Tomsich 		/* Record our loadables into the FDT */
7471f1cf67bSPhilipp Tomsich 		if (spl_image->fdt_addr)
7481f1cf67bSPhilipp Tomsich 			spl_fit_record_loadable(fit, images, index,
7491f1cf67bSPhilipp Tomsich 						spl_image->fdt_addr,
7501f1cf67bSPhilipp Tomsich 						&image_info);
751411cf32dSAndre Przywara 	}
752411cf32dSAndre Przywara 
753411cf32dSAndre Przywara 	/*
754411cf32dSAndre Przywara 	 * If a platform does not provide CONFIG_SYS_UBOOT_START, U-Boot's
755411cf32dSAndre Przywara 	 * Makefile will set it to 0 and it will end up as the entry point
756411cf32dSAndre Przywara 	 * here. What it actually means is: use the load address.
757411cf32dSAndre Przywara 	 */
758411cf32dSAndre Przywara 	if (spl_image->entry_point == FDT_ERROR || spl_image->entry_point == 0)
759411cf32dSAndre Przywara 		spl_image->entry_point = spl_image->load_addr;
760411cf32dSAndre Przywara 
761411cf32dSAndre Przywara 	return 0;
762f1dcee59SSimon Glass }
76322c7c1a8SJoseph Chen 
76422c7c1a8SJoseph Chen int spl_load_simple_fit(struct spl_image_info *spl_image,
76522c7c1a8SJoseph Chen 			struct spl_load_info *info, ulong sector, void *fit)
76622c7c1a8SJoseph Chen {
76722c7c1a8SJoseph Chen 	ulong sector_offs = sector;
768e12dde2dSJoseph Chen 	int ret = -EINVAL;
76922c7c1a8SJoseph Chen 	int i;
77022c7c1a8SJoseph Chen 
77122c7c1a8SJoseph Chen 	for (i = 0; i < CONFIG_SPL_FIT_IMAGE_MULTIPLE; i++) {
77222c7c1a8SJoseph Chen 		if (i > 0) {
77322c7c1a8SJoseph Chen 			sector_offs +=
77422c7c1a8SJoseph Chen 			   i * ((CONFIG_SPL_FIT_IMAGE_KB << 10) / info->bl_len);
77522c7c1a8SJoseph Chen 			printf("Trying fit image at 0x%lx sector\n", sector_offs);
77622c7c1a8SJoseph Chen 			if (info->read(info, sector_offs, 1, fit) != 1) {
77722c7c1a8SJoseph Chen 				printf("IO error\n");
77822c7c1a8SJoseph Chen 				continue;
77922c7c1a8SJoseph Chen 			}
78022c7c1a8SJoseph Chen 		}
78122c7c1a8SJoseph Chen 
78222c7c1a8SJoseph Chen 		if (image_get_magic(fit) != FDT_MAGIC) {
783e12dde2dSJoseph Chen 			printf("Not fit magic\n");
78422c7c1a8SJoseph Chen 			continue;
78522c7c1a8SJoseph Chen 		}
78622c7c1a8SJoseph Chen 
787e12dde2dSJoseph Chen 		ret = spl_internal_load_simple_fit(spl_image, info,
788e12dde2dSJoseph Chen 						   sector_offs, fit);
789e12dde2dSJoseph Chen 		if (!ret) {
790a00b516fSJoseph Chen #ifdef CONFIG_SPL_KERNEL_BOOT
791e12dde2dSJoseph Chen 			ret = spl_load_kernel_fit(spl_image, info);
792e12dde2dSJoseph Chen #endif
793e12dde2dSJoseph Chen 			return ret;
794e12dde2dSJoseph Chen 		}
79522c7c1a8SJoseph Chen 	}
79622c7c1a8SJoseph Chen 
797e12dde2dSJoseph Chen 	return ret;
79822c7c1a8SJoseph Chen }
79922c7c1a8SJoseph Chen 
800