[J-u-boot.git] / common / image-fdt.c

// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright (c) 2013, Google Inc.
 *
 * (C) Copyright 2008 Semihalf
 *
 * (C) Copyright 2000-2006
 * Wolfgang Denk, DENX Software Engineering, [email protected].
 */

#include <common.h>
#include <fdt_support.h>
#include <fdtdec.h>
#include <env.h>
#include <errno.h>
#include <image.h>
#include <lmb.h>
#include <log.h>
#include <malloc.h>
#include <linux/libfdt.h>
#include <mapmem.h>
#include <asm/io.h>
#include <tee/optee.h>

#ifndef CONFIG_SYS_FDT_PAD
#define CONFIG_SYS_FDT_PAD 0x3000
#endif

/* adding a ramdisk needs 0x44 bytes in version 2008.10 */
#define FDT_RAMDISK_OVERHEAD	0x80

DECLARE_GLOBAL_DATA_PTR;

static void fdt_error(const char *msg)
{
	puts("ERROR: ");
	puts(msg);
	puts(" - must RESET the board to recover.\n");
}

#if CONFIG_IS_ENABLED(LEGACY_IMAGE_FORMAT)
static const image_header_t *image_get_fdt(ulong fdt_addr)
{
	const image_header_t *fdt_hdr = map_sysmem(fdt_addr, 0);

	image_print_contents(fdt_hdr);

	puts("   Verifying Checksum ... ");
	if (!image_check_hcrc(fdt_hdr)) {
		fdt_error("fdt header checksum invalid");
		return NULL;
	}

	if (!image_check_dcrc(fdt_hdr)) {
		fdt_error("fdt checksum invalid");
		return NULL;
	}
	puts("OK\n");

	if (!image_check_type(fdt_hdr, IH_TYPE_FLATDT)) {
		fdt_error("uImage is not a fdt");
		return NULL;
	}
	if (image_get_comp(fdt_hdr) != IH_COMP_NONE) {
		fdt_error("uImage is compressed");
		return NULL;
	}
	if (fdt_check_header((void *)image_get_data(fdt_hdr)) != 0) {
		fdt_error("uImage data is not a fdt");
		return NULL;
	}
	return fdt_hdr;
}
#endif

static void boot_fdt_reserve_region(struct lmb *lmb, uint64_t addr,
				    uint64_t size)
{
	long ret;

	ret = lmb_reserve(lmb, addr, size);
	if (ret >= 0) {
		debug("   reserving fdt memory region: addr=%llx size=%llx\n",
		      (unsigned long long)addr, (unsigned long long)size);
	} else {
		puts("ERROR: reserving fdt memory region failed ");
		printf("(addr=%llx size=%llx)\n",
		       (unsigned long long)addr, (unsigned long long)size);
	}
}

/**
 * boot_fdt_add_mem_rsv_regions - Mark the memreserve and reserved-memory
 * sections as unusable
 * @lmb: pointer to lmb handle, will be used for memory mgmt
 * @fdt_blob: pointer to fdt blob base address
 *
 * Adds the and reserved-memorymemreserve regions in the dtb to the lmb block.
 * Adding the memreserve regions prevents u-boot from using them to store the
 * initrd or the fdt blob.
 */
void boot_fdt_add_mem_rsv_regions(struct lmb *lmb, void *fdt_blob)
{
	uint64_t addr, size;
	int i, total, ret;
	int nodeoffset, subnode;
	struct fdt_resource res;

	if (fdt_check_header(fdt_blob) != 0)
		return;

	/* process memreserve sections */
	total = fdt_num_mem_rsv(fdt_blob);
	for (i = 0; i < total; i++) {
		if (fdt_get_mem_rsv(fdt_blob, i, &addr, &size) != 0)
			continue;
		boot_fdt_reserve_region(lmb, addr, size);
	}

	/* process reserved-memory */
	nodeoffset = fdt_subnode_offset(fdt_blob, 0, "reserved-memory");
	if (nodeoffset >= 0) {
		subnode = fdt_first_subnode(fdt_blob, nodeoffset);
		while (subnode >= 0) {
			/* check if this subnode has a reg property */
			ret = fdt_get_resource(fdt_blob, subnode, "reg", 0,
					       &res);
			if (!ret && fdtdec_get_is_enabled(fdt_blob, subnode)) {
				addr = res.start;
				size = res.end - res.start + 1;
				boot_fdt_reserve_region(lmb, addr, size);
			}

			subnode = fdt_next_subnode(fdt_blob, subnode);
		}
	}
}

/**
 * boot_relocate_fdt - relocate flat device tree
 * @lmb: pointer to lmb handle, will be used for memory mgmt
 * @of_flat_tree: pointer to a char* variable, will hold fdt start address
 * @of_size: pointer to a ulong variable, will hold fdt length
 *
 * boot_relocate_fdt() allocates a region of memory within the bootmap and
 * relocates the of_flat_tree into that region, even if the fdt is already in
 * the bootmap.  It also expands the size of the fdt by CONFIG_SYS_FDT_PAD
 * bytes.
 *
 * of_flat_tree and of_size are set to final (after relocation) values
 *
 * returns:
 *      0 - success
 *      1 - failure
 */
int boot_relocate_fdt(struct lmb *lmb, char **of_flat_tree, ulong *of_size)
{
	void	*fdt_blob = *of_flat_tree;
	void	*of_start = NULL;
	char	*fdt_high;
	ulong	of_len = 0;
	int	err;
	int	disable_relocation = 0;

	/* nothing to do */
	if (*of_size == 0)
		return 0;

	if (fdt_check_header(fdt_blob) != 0) {
		fdt_error("image is not a fdt");
		goto error;
	}

	/* position on a 4K boundary before the alloc_current */
	/* Pad the FDT by a specified amount */
	of_len = *of_size + CONFIG_SYS_FDT_PAD;

	/* If fdt_high is set use it to select the relocation address */
	fdt_high = env_get("fdt_high");
	if (fdt_high) {
		void *desired_addr = (void *)simple_strtoul(fdt_high, NULL, 16);

		if (((ulong) desired_addr) == ~0UL) {
			/* All ones means use fdt in place */
			of_start = fdt_blob;
			lmb_reserve(lmb, (ulong)of_start, of_len);
			disable_relocation = 1;
		} else if (desired_addr) {
			of_start =
			    (void *)(ulong) lmb_alloc_base(lmb, of_len, 0x1000,
							   (ulong)desired_addr);
			if (of_start == NULL) {
				puts("Failed using fdt_high value for Device Tree");
				goto error;
			}
		} else {
			of_start =
			    (void *)(ulong) lmb_alloc(lmb, of_len, 0x1000);
		}
	} else {
		of_start =
		    (void *)(ulong) lmb_alloc_base(lmb, of_len, 0x1000,
						   env_get_bootm_mapsize()
						   + env_get_bootm_low());
	}

	if (of_start == NULL) {
		puts("device tree - allocation error\n");
		goto error;
	}

	if (disable_relocation) {
		/*
		 * We assume there is space after the existing fdt to use
		 * for padding
		 */
		fdt_set_totalsize(of_start, of_len);
		printf("   Using Device Tree in place at %p, end %p\n",
		       of_start, of_start + of_len - 1);
	} else {
		debug("## device tree at %p ... %p (len=%ld [0x%lX])\n",
		      fdt_blob, fdt_blob + *of_size - 1, of_len, of_len);

		printf("   Loading Device Tree to %p, end %p ... ",
		       of_start, of_start + of_len - 1);

		err = fdt_open_into(fdt_blob, of_start, of_len);
		if (err != 0) {
			fdt_error("fdt move failed");
			goto error;
		}
		puts("OK\n");
	}

	*of_flat_tree = of_start;
	*of_size = of_len;

	if (CONFIG_IS_ENABLED(CMD_FDT))
		set_working_fdt_addr(map_to_sysmem(*of_flat_tree));
	return 0;

error:
	return 1;
}

/**
 * boot_get_fdt - main fdt handling routine
 * @argc: command argument count
 * @argv: command argument list
 * @arch: architecture (IH_ARCH_...)
 * @images: pointer to the bootm images structure
 * @of_flat_tree: pointer to a char* variable, will hold fdt start address
 * @of_size: pointer to a ulong variable, will hold fdt length
 *
 * boot_get_fdt() is responsible for finding a valid flat device tree image.
 * Curently supported are the following ramdisk sources:
 *      - multicomponent kernel/ramdisk image,
 *      - commandline provided address of decicated ramdisk image.
 *
 * returns:
 *     0, if fdt image was found and valid, or skipped
 *     of_flat_tree and of_size are set to fdt start address and length if
 *     fdt image is found and valid
 *
 *     1, if fdt image is found but corrupted
 *     of_flat_tree and of_size are set to 0 if no fdt exists
 */
int boot_get_fdt(int flag, int argc, char *const argv[], uint8_t arch,
		 bootm_headers_t *images, char **of_flat_tree, ulong *of_size)
{
#if CONFIG_IS_ENABLED(LEGACY_IMAGE_FORMAT)
	const image_header_t *fdt_hdr;
	ulong		load, load_end;
	ulong		image_start, image_data, image_end;
#endif
	ulong		img_addr;
	ulong		fdt_addr;
	char		*fdt_blob = NULL;
	void		*buf;
#if CONFIG_IS_ENABLED(FIT)
	const char	*fit_uname_config = images->fit_uname_cfg;
	const char	*fit_uname_fdt = NULL;
	ulong		default_addr;
	int		fdt_noffset;
#endif
	const char *select = NULL;

	*of_flat_tree = NULL;
	*of_size = 0;

	img_addr = (argc == 0) ? image_load_addr :
			simple_strtoul(argv[0], NULL, 16);
	buf = map_sysmem(img_addr, 0);

	if (argc > 2)
		select = argv[2];
	if (select || genimg_has_config(images)) {
#if CONFIG_IS_ENABLED(FIT)
		if (select) {
			/*
			 * If the FDT blob comes from the FIT image and the
			 * FIT image address is omitted in the command line
			 * argument, try to use ramdisk or os FIT image
			 * address or default load address.
			 */
			if (images->fit_uname_rd)
				default_addr = (ulong)images->fit_hdr_rd;
			else if (images->fit_uname_os)
				default_addr = (ulong)images->fit_hdr_os;
			else
				default_addr = image_load_addr;

			if (fit_parse_conf(select, default_addr,
					   &fdt_addr, &fit_uname_config)) {
				debug("*  fdt: config '%s' from image at 0x%08lx\n",
				      fit_uname_config, fdt_addr);
			} else if (fit_parse_subimage(select, default_addr,
				   &fdt_addr, &fit_uname_fdt)) {
				debug("*  fdt: subimage '%s' from image at 0x%08lx\n",
				      fit_uname_fdt, fdt_addr);
			} else
#endif
			{
				fdt_addr = simple_strtoul(select, NULL, 16);
				debug("*  fdt: cmdline image address = 0x%08lx\n",
				      fdt_addr);
			}
#if CONFIG_IS_ENABLED(FIT)
		} else {
			/* use FIT configuration provided in first bootm
			 * command argument
			 */
			fdt_addr = map_to_sysmem(images->fit_hdr_os);
			fdt_noffset = fit_get_node_from_config(images,
							       FIT_FDT_PROP,
							       fdt_addr);
			if (fdt_noffset == -ENOENT)
				return 0;
			else if (fdt_noffset < 0)
				return 1;
		}
#endif
		debug("## Checking for 'FDT'/'FDT Image' at %08lx\n",
		      fdt_addr);

		/*
		 * Check if there is an FDT image at the
		 * address provided in the second bootm argument
		 * check image type, for FIT images get a FIT node.
		 */
		buf = map_sysmem(fdt_addr, 0);
		switch (genimg_get_format(buf)) {
#if CONFIG_IS_ENABLED(LEGACY_IMAGE_FORMAT)
		case IMAGE_FORMAT_LEGACY:
			/* verify fdt_addr points to a valid image header */
			printf("## Flattened Device Tree from Legacy Image at %08lx\n",
			       fdt_addr);
			fdt_hdr = image_get_fdt(fdt_addr);
			if (!fdt_hdr)
				goto no_fdt;

			/*
			 * move image data to the load address,
			 * make sure we don't overwrite initial image
			 */
			image_start = (ulong)fdt_hdr;
			image_data = (ulong)image_get_data(fdt_hdr);
			image_end = image_get_image_end(fdt_hdr);

			load = image_get_load(fdt_hdr);
			load_end = load + image_get_data_size(fdt_hdr);

			if (load == image_start ||
			    load == image_data) {
				fdt_addr = load;
				break;
			}

			if ((load < image_end) && (load_end > image_start)) {
				fdt_error("fdt overwritten");
				goto error;
			}

			debug("   Loading FDT from 0x%08lx to 0x%08lx\n",
			      image_data, load);

			memmove((void *)load,
				(void *)image_data,
				image_get_data_size(fdt_hdr));

			fdt_addr = load;
			break;
#endif
		case IMAGE_FORMAT_FIT:
			/*
			 * This case will catch both: new uImage format
			 * (libfdt based) and raw FDT blob (also libfdt
			 * based).
			 */
#if CONFIG_IS_ENABLED(FIT)
			/* check FDT blob vs FIT blob */
			if (fit_check_format(buf)) {
				ulong load, len;

				fdt_noffset = boot_get_fdt_fit(images,
					fdt_addr, &fit_uname_fdt,
					&fit_uname_config,
					arch, &load, &len);

				images->fit_hdr_fdt = map_sysmem(fdt_addr, 0);
				images->fit_uname_fdt = fit_uname_fdt;
				images->fit_noffset_fdt = fdt_noffset;
				fdt_addr = load;

				break;
			} else
#endif
			{
				/*
				 * FDT blob
				 */
				debug("*  fdt: raw FDT blob\n");
				printf("## Flattened Device Tree blob at %08lx\n",
				       (long)fdt_addr);
			}
			break;
		default:
			puts("ERROR: Did not find a cmdline Flattened Device Tree\n");
			goto error;
		}

		printf("   Booting using the fdt blob at %#08lx\n", fdt_addr);
		fdt_blob = map_sysmem(fdt_addr, 0);
	} else if (images->legacy_hdr_valid &&
			image_check_type(&images->legacy_hdr_os_copy,
					 IH_TYPE_MULTI)) {
		ulong fdt_data, fdt_len;

		/*
		 * Now check if we have a legacy multi-component image,
		 * get second entry data start address and len.
		 */
		printf("## Flattened Device Tree from multi component Image at %08lX\n",
		       (ulong)images->legacy_hdr_os);

		image_multi_getimg(images->legacy_hdr_os, 2, &fdt_data,
				   &fdt_len);
		if (fdt_len) {
			fdt_blob = (char *)fdt_data;
			printf("   Booting using the fdt at 0x%p\n", fdt_blob);

			if (fdt_check_header(fdt_blob) != 0) {
				fdt_error("image is not a fdt");
				goto error;
			}

			if (fdt_totalsize(fdt_blob) != fdt_len) {
				fdt_error("fdt size != image size");
				goto error;
			}
		} else {
			debug("## No Flattened Device Tree\n");
			goto no_fdt;
		}
#ifdef CONFIG_ANDROID_BOOT_IMAGE
	} else if (genimg_get_format(buf) == IMAGE_FORMAT_ANDROID) {
		struct andr_img_hdr *hdr = buf;
		ulong fdt_data, fdt_len;

		if (!android_image_get_second(hdr, &fdt_data, &fdt_len) &&
		    !fdt_check_header((char *)fdt_data)) {
			fdt_blob = (char *)fdt_data;
			if (fdt_totalsize(fdt_blob) != fdt_len)
				goto error;

			debug("## Using FDT in Android image second area\n");
		} else {
			fdt_addr = env_get_hex("fdtaddr", 0);
			if (!fdt_addr)
				goto no_fdt;

			fdt_blob = map_sysmem(fdt_addr, 0);
			if (fdt_check_header(fdt_blob))
				goto no_fdt;

			debug("## Using FDT at ${fdtaddr}=Ox%lx\n", fdt_addr);
		}
#endif
	} else {
		debug("## No Flattened Device Tree\n");
		goto no_fdt;
	}

	*of_flat_tree = fdt_blob;
	*of_size = fdt_totalsize(fdt_blob);
	debug("   of_flat_tree at 0x%08lx size 0x%08lx\n",
	      (ulong)*of_flat_tree, *of_size);

	return 0;

no_fdt:
	debug("Continuing to boot without FDT\n");
	return 0;
error:
	return 1;
}

/*
 * Verify the device tree.
 *
 * This function is called after all device tree fix-ups have been enacted,
 * so that the final device tree can be verified.  The definition of "verified"
 * is up to the specific implementation.  However, it generally means that the
 * addresses of some of the devices in the device tree are compared with the
 * actual addresses at which U-Boot has placed them.
 *
 * Returns 1 on success, 0 on failure.  If 0 is returned, U-Boot will halt the
 * boot process.
 */
__weak int ft_verify_fdt(void *fdt)
{
	return 1;
}

__weak int arch_fixup_fdt(void *blob)
{
	return 0;
}

int image_setup_libfdt(bootm_headers_t *images, void *blob,
		       int of_size, struct lmb *lmb)
{
	ulong *initrd_start = &images->initrd_start;
	ulong *initrd_end = &images->initrd_end;
	int ret = -EPERM;
	int fdt_ret;

	if (fdt_root(blob) < 0) {
		printf("ERROR: root node setup failed\n");
		goto err;
	}
	if (fdt_chosen(blob) < 0) {
		printf("ERROR: /chosen node create failed\n");
		goto err;
	}
	if (arch_fixup_fdt(blob) < 0) {
		printf("ERROR: arch-specific fdt fixup failed\n");
		goto err;
	}
	/* Update ethernet nodes */
	fdt_fixup_ethernet(blob);
	if (IMAGE_OF_BOARD_SETUP) {
		fdt_ret = ft_board_setup(blob, gd->bd);
		if (fdt_ret) {
			printf("ERROR: board-specific fdt fixup failed: %s\n",
			       fdt_strerror(fdt_ret));
			goto err;
		}
	}
	if (IMAGE_OF_SYSTEM_SETUP) {
		fdt_ret = ft_system_setup(blob, gd->bd);
		if (fdt_ret) {
			printf("ERROR: system-specific fdt fixup failed: %s\n",
			       fdt_strerror(fdt_ret));
			goto err;
		}
	}

	fdt_ret = optee_copy_fdt_nodes(gd->fdt_blob, blob);
	if (fdt_ret) {
		printf("ERROR: transfer of optee nodes to new fdt failed: %s\n",
		       fdt_strerror(fdt_ret));
		goto err;
	}

	/* Delete the old LMB reservation */
	if (lmb)
		lmb_free(lmb, (phys_addr_t)(u32)(uintptr_t)blob,
			 (phys_size_t)fdt_totalsize(blob));

	ret = fdt_shrink_to_minimum(blob, 0);
	if (ret < 0)
		goto err;
	of_size = ret;

	if (*initrd_start && *initrd_end) {
		of_size += FDT_RAMDISK_OVERHEAD;
		fdt_set_totalsize(blob, of_size);
	}
	/* Create a new LMB reservation */
	if (lmb)
		lmb_reserve(lmb, (ulong)blob, of_size);

	fdt_initrd(blob, *initrd_start, *initrd_end);
	if (!ft_verify_fdt(blob))
		goto err;

#if defined(CONFIG_SOC_KEYSTONE)
	if (IMAGE_OF_BOARD_SETUP)
		ft_board_setup_ex(blob, gd->bd);
#endif

	return 0;
err:
	printf(" - must RESET the board to recover.\n\n");

	return ret;
}
Commit	Line	Data
83d290c5	1	// SPDX-License-Identifier: GPL-2.0+
44d3a306 SG	2	/*
	3	* Copyright (c) 2013, Google Inc.
	4	*
	5	* (C) Copyright 2008 Semihalf
	6	*
	7	* (C) Copyright 2000-2006
	8	* Wolfgang Denk, DENX Software Engineering, [email protected].
44d3a306 SG	9	*/
	10
	11	#include <common.h>
	12	#include <fdt_support.h>
e2237a2c	13	#include <fdtdec.h>
cdbff9fc	14	#include <env.h>
44d3a306 SG	15	#include <errno.h>
44d3a306 SG	16	#include <image.h>
4d72caa5	17	#include <lmb.h>
f7ae49fc	18	#include <log.h>
336d4615	19	#include <malloc.h>
b08c8c48	20	#include <linux/libfdt.h>
0eb25b61	21	#include <mapmem.h>
44d3a306	22	#include <asm/io.h>
6ccb05ea	23	#include <tee/optee.h>
44d3a306 SG	24
	25	#ifndef CONFIG_SYS_FDT_PAD
	26	#define CONFIG_SYS_FDT_PAD 0x3000
	27	#endif
	28
c960a68e MY	29	/* adding a ramdisk needs 0x44 bytes in version 2008.10 */
	30	#define FDT_RAMDISK_OVERHEAD 0x80
	31
44d3a306 SG	32	DECLARE_GLOBAL_DATA_PTR;
	33
	34	static void fdt_error(const char *msg)
	35	{
	36	puts("ERROR: ");
	37	puts(msg);
	38	puts(" - must RESET the board to recover.\n");
	39	}
	40
c76c93a3	41	#if CONFIG_IS_ENABLED(LEGACY_IMAGE_FORMAT)
44d3a306 SG	42	static const image_header_t *image_get_fdt(ulong fdt_addr)
	43	{
	44	const image_header_t *fdt_hdr = map_sysmem(fdt_addr, 0);
	45
	46	image_print_contents(fdt_hdr);
	47
	48	puts(" Verifying Checksum ... ");
	49	if (!image_check_hcrc(fdt_hdr)) {
	50	fdt_error("fdt header checksum invalid");
	51	return NULL;
	52	}
	53
	54	if (!image_check_dcrc(fdt_hdr)) {
	55	fdt_error("fdt checksum invalid");
	56	return NULL;
	57	}
	58	puts("OK\n");
	59
	60	if (!image_check_type(fdt_hdr, IH_TYPE_FLATDT)) {
	61	fdt_error("uImage is not a fdt");
	62	return NULL;
	63	}
	64	if (image_get_comp(fdt_hdr) != IH_COMP_NONE) {
	65	fdt_error("uImage is compressed");
	66	return NULL;
	67	}
2f0877c7	68	if (fdt_check_header((void *)image_get_data(fdt_hdr)) != 0) {
44d3a306 SG	69	fdt_error("uImage data is not a fdt");
	70	return NULL;
	71	}
	72	return fdt_hdr;
	73	}
21d29f7f	74	#endif
44d3a306	75
e2237a2c SG	76	static void boot_fdt_reserve_region(struct lmb *lmb, uint64_t addr,
	77	uint64_t size)
	78	{
e1d7ed34	79	long ret;
e2237a2c SG	80
e2237a2c SG	81	ret = lmb_reserve(lmb, addr, size);
e1d7ed34	82	if (ret >= 0) {
e2237a2c SG	83	debug(" reserving fdt memory region: addr=%llx size=%llx\n",
	84	(unsigned long long)addr, (unsigned long long)size);
	85	} else {
	86	puts("ERROR: reserving fdt memory region failed ");
	87	printf("(addr=%llx size=%llx)\n",
	88	(unsigned long long)addr, (unsigned long long)size);
	89	}
	90	}
	91
44d3a306	92	/**
e2237a2c SG	93	* boot_fdt_add_mem_rsv_regions - Mark the memreserve and reserved-memory
e2237a2c SG	94	* sections as unusable
44d3a306 SG	95	* @lmb: pointer to lmb handle, will be used for memory mgmt
	96	* @fdt_blob: pointer to fdt blob base address
	97	*
e2237a2c SG	98	* Adds the and reserved-memorymemreserve regions in the dtb to the lmb block.
	99	* Adding the memreserve regions prevents u-boot from using them to store the
	100	* initrd or the fdt blob.
44d3a306 SG	101	*/
	102	void boot_fdt_add_mem_rsv_regions(struct lmb lmb, void fdt_blob)
	103	{
	104	uint64_t addr, size;
e2237a2c SG	105	int i, total, ret;
	106	int nodeoffset, subnode;
	107	struct fdt_resource res;
44d3a306 SG	108
	109	if (fdt_check_header(fdt_blob) != 0)
	110	return;
	111
e2237a2c	112	/* process memreserve sections */
44d3a306 SG	113	total = fdt_num_mem_rsv(fdt_blob);
	114	for (i = 0; i < total; i++) {
	115	if (fdt_get_mem_rsv(fdt_blob, i, &addr, &size) != 0)
	116	continue;
e2237a2c SG	117	boot_fdt_reserve_region(lmb, addr, size);
	118	}
	119
	120	/* process reserved-memory */
	121	nodeoffset = fdt_subnode_offset(fdt_blob, 0, "reserved-memory");
	122	if (nodeoffset >= 0) {
	123	subnode = fdt_first_subnode(fdt_blob, nodeoffset);
	124	while (subnode >= 0) {
	125	/* check if this subnode has a reg property */
	126	ret = fdt_get_resource(fdt_blob, subnode, "reg", 0,
	127	&res);
28b417ce	128	if (!ret && fdtdec_get_is_enabled(fdt_blob, subnode)) {
e2237a2c SG	129	addr = res.start;
	130	size = res.end - res.start + 1;
	131	boot_fdt_reserve_region(lmb, addr, size);
	132	}
	133
	134	subnode = fdt_next_subnode(fdt_blob, subnode);
	135	}
44d3a306 SG	136	}
	137	}
	138
	139	/**
	140	* boot_relocate_fdt - relocate flat device tree
	141	* @lmb: pointer to lmb handle, will be used for memory mgmt
	142	* @of_flat_tree: pointer to a char* variable, will hold fdt start address
	143	* @of_size: pointer to a ulong variable, will hold fdt length
	144	*
	145	* boot_relocate_fdt() allocates a region of memory within the bootmap and
	146	* relocates the of_flat_tree into that region, even if the fdt is already in
	147	* the bootmap. It also expands the size of the fdt by CONFIG_SYS_FDT_PAD
	148	* bytes.
	149	*
	150	* of_flat_tree and of_size are set to final (after relocation) values
	151	*
	152	* returns:
	153	* 0 - success
	154	* 1 - failure
	155	*/
	156	int boot_relocate_fdt(struct lmb lmb, char of_flat_tree, ulong of_size)
	157	{
	158	void fdt_blob = of_flat_tree;
	159	void *of_start = NULL;
	160	char *fdt_high;
	161	ulong of_len = 0;
	162	int err;
	163	int disable_relocation = 0;
	164
	165	/* nothing to do */
	166	if (*of_size == 0)
	167	return 0;
	168
	169	if (fdt_check_header(fdt_blob) != 0) {
	170	fdt_error("image is not a fdt");
	171	goto error;
	172	}
	173
	174	/* position on a 4K boundary before the alloc_current */
	175	/* Pad the FDT by a specified amount */
	176	of_len = *of_size + CONFIG_SYS_FDT_PAD;
	177
	178	/* If fdt_high is set use it to select the relocation address */
00caae6d	179	fdt_high = env_get("fdt_high");
44d3a306 SG	180	if (fdt_high) {
	181	void desired_addr = (void )simple_strtoul(fdt_high, NULL, 16);
	182
	183	if (((ulong) desired_addr) == ~0UL) {
	184	/* All ones means use fdt in place */
	185	of_start = fdt_blob;
	186	lmb_reserve(lmb, (ulong)of_start, of_len);
	187	disable_relocation = 1;
	188	} else if (desired_addr) {
	189	of_start =
	190	(void *)(ulong) lmb_alloc_base(lmb, of_len, 0x1000,
	191	(ulong)desired_addr);
	192	if (of_start == NULL) {
	193	puts("Failed using fdt_high value for Device Tree");
	194	goto error;
	195	}
	196	} else {
	197	of_start =
	198	(void *)(ulong) lmb_alloc(lmb, of_len, 0x1000);
	199	}
	200	} else {
	201	of_start =
	202	(void *)(ulong) lmb_alloc_base(lmb, of_len, 0x1000,
723806cc SG	203	env_get_bootm_mapsize()
723806cc SG	204	+ env_get_bootm_low());
44d3a306 SG	205	}
	206
	207	if (of_start == NULL) {
	208	puts("device tree - allocation error\n");
	209	goto error;
	210	}
	211
	212	if (disable_relocation) {
	213	/*
	214	* We assume there is space after the existing fdt to use
	215	* for padding
	216	*/
	217	fdt_set_totalsize(of_start, of_len);
	218	printf(" Using Device Tree in place at %p, end %p\n",
	219	of_start, of_start + of_len - 1);
	220	} else {
	221	debug("## device tree at %p ... %p (len=%ld [0x%lX])\n",
	222	fdt_blob, fdt_blob + *of_size - 1, of_len, of_len);
	223
	224	printf(" Loading Device Tree to %p, end %p ... ",
	225	of_start, of_start + of_len - 1);
	226
	227	err = fdt_open_into(fdt_blob, of_start, of_len);
	228	if (err != 0) {
	229	fdt_error("fdt move failed");
	230	goto error;
	231	}
	232	puts("OK\n");
	233	}
	234
	235	*of_flat_tree = of_start;
	236	*of_size = of_len;
	237
596be5f3 SG	238	if (CONFIG_IS_ENABLED(CMD_FDT))
596be5f3 SG	239	set_working_fdt_addr(map_to_sysmem(*of_flat_tree));
44d3a306 SG	240	return 0;
	241
	242	error:
	243	return 1;
	244	}
	245
44d3a306 SG	246	/**
	247	* boot_get_fdt - main fdt handling routine
	248	* @argc: command argument count
	249	* @argv: command argument list
53f375fa	250	* @arch: architecture (IH_ARCH_...)
44d3a306 SG	251	* @images: pointer to the bootm images structure
	252	* @of_flat_tree: pointer to a char* variable, will hold fdt start address
	253	* @of_size: pointer to a ulong variable, will hold fdt length
	254	*
	255	* boot_get_fdt() is responsible for finding a valid flat device tree image.
	256	* Curently supported are the following ramdisk sources:
	257	* - multicomponent kernel/ramdisk image,
	258	* - commandline provided address of decicated ramdisk image.
	259	*
	260	* returns:
	261	* 0, if fdt image was found and valid, or skipped
	262	* of_flat_tree and of_size are set to fdt start address and length if
	263	* fdt image is found and valid
	264	*
	265	* 1, if fdt image is found but corrupted
	266	* of_flat_tree and of_size are set to 0 if no fdt exists
	267	*/
09140113 SG	268	int boot_get_fdt(int flag, int argc, char *const argv[], uint8_t arch,
09140113 SG	269	bootm_headers_t images, char of_flat_tree, ulong of_size)
44d3a306	270	{
c76c93a3	271	#if CONFIG_IS_ENABLED(LEGACY_IMAGE_FORMAT)
44d3a306	272	const image_header_t *fdt_hdr;
21d29f7f HS	273	ulong load, load_end;
	274	ulong image_start, image_data, image_end;
	275	#endif
6a7b406a	276	ulong img_addr;
44d3a306 SG	277	ulong fdt_addr;
44d3a306 SG	278	char *fdt_blob = NULL;
44d3a306	279	void *buf;
73223f0e	280	#if CONFIG_IS_ENABLED(FIT)
f320a4d8	281	const char *fit_uname_config = images->fit_uname_cfg;
44d3a306 SG	282	const char *fit_uname_fdt = NULL;
44d3a306 SG	283	ulong default_addr;
44d3a306	284	int fdt_noffset;
44d3a306	285	#endif
983c72f4	286	const char *select = NULL;
44d3a306 SG	287
	288	*of_flat_tree = NULL;
	289	*of_size = 0;
	290
bb872dd9 SG	291	img_addr = (argc == 0) ? image_load_addr :
bb872dd9 SG	292	simple_strtoul(argv[0], NULL, 16);
6a7b406a SG	293	buf = map_sysmem(img_addr, 0);
6a7b406a SG	294
983c72f4 SG	295	if (argc > 2)
	296	select = argv[2];
	297	if (select \|\| genimg_has_config(images)) {
73223f0e	298	#if CONFIG_IS_ENABLED(FIT)
983c72f4	299	if (select) {
44d3a306 SG	300	/*
	301	* If the FDT blob comes from the FIT image and the
	302	* FIT image address is omitted in the command line
	303	* argument, try to use ramdisk or os FIT image
	304	* address or default load address.
	305	*/
	306	if (images->fit_uname_rd)
	307	default_addr = (ulong)images->fit_hdr_rd;
	308	else if (images->fit_uname_os)
	309	default_addr = (ulong)images->fit_hdr_os;
	310	else
bb872dd9	311	default_addr = image_load_addr;
44d3a306	312
983c72f4	313	if (fit_parse_conf(select, default_addr,
44d3a306 SG	314	&fdt_addr, &fit_uname_config)) {
	315	debug("* fdt: config '%s' from image at 0x%08lx\n",
	316	fit_uname_config, fdt_addr);
983c72f4	317	} else if (fit_parse_subimage(select, default_addr,
44d3a306 SG	318	&fdt_addr, &fit_uname_fdt)) {
	319	debug("* fdt: subimage '%s' from image at 0x%08lx\n",
	320	fit_uname_fdt, fdt_addr);
	321	} else
	322	#endif
	323	{
983c72f4	324	fdt_addr = simple_strtoul(select, NULL, 16);
44d3a306 SG	325	debug("* fdt: cmdline image address = 0x%08lx\n",
	326	fdt_addr);
	327	}
73223f0e	328	#if CONFIG_IS_ENABLED(FIT)
44d3a306 SG	329	} else {
	330	/* use FIT configuration provided in first bootm
	331	* command argument
	332	*/
	333	fdt_addr = map_to_sysmem(images->fit_hdr_os);
53f375fa SG	334	fdt_noffset = fit_get_node_from_config(images,
	335	FIT_FDT_PROP,
	336	fdt_addr);
bd86ef11	337	if (fdt_noffset == -ENOENT)
44d3a306	338	return 0;
53f375fa SG	339	else if (fdt_noffset < 0)
53f375fa SG	340	return 1;
44d3a306 SG	341	}
44d3a306 SG	342	#endif
44d3a306 SG	343	debug("## Checking for 'FDT'/'FDT Image' at %08lx\n",
	344	fdt_addr);
	345
44d3a306 SG	346	/*
	347	* Check if there is an FDT image at the
	348	* address provided in the second bootm argument
	349	* check image type, for FIT images get a FIT node.
	350	*/
	351	buf = map_sysmem(fdt_addr, 0);
	352	switch (genimg_get_format(buf)) {
c76c93a3	353	#if CONFIG_IS_ENABLED(LEGACY_IMAGE_FORMAT)
44d3a306 SG	354	case IMAGE_FORMAT_LEGACY:
	355	/* verify fdt_addr points to a valid image header */
	356	printf("## Flattened Device Tree from Legacy Image at %08lx\n",
	357	fdt_addr);
	358	fdt_hdr = image_get_fdt(fdt_addr);
	359	if (!fdt_hdr)
48aead71	360	goto no_fdt;
44d3a306 SG	361
	362	/*
	363	* move image data to the load address,
	364	* make sure we don't overwrite initial image
	365	*/
	366	image_start = (ulong)fdt_hdr;
	367	image_data = (ulong)image_get_data(fdt_hdr);
	368	image_end = image_get_image_end(fdt_hdr);
	369
53f375fa SG	370	load = image_get_load(fdt_hdr);
53f375fa SG	371	load_end = load + image_get_data_size(fdt_hdr);
44d3a306	372
53f375fa SG	373	if (load == image_start \|\|
53f375fa SG	374	load == image_data) {
2ea47be0	375	fdt_addr = load;
44d3a306 SG	376	break;
	377	}
	378
53f375fa	379	if ((load < image_end) && (load_end > image_start)) {
44d3a306 SG	380	fdt_error("fdt overwritten");
	381	goto error;
	382	}
	383
	384	debug(" Loading FDT from 0x%08lx to 0x%08lx\n",
53f375fa	385	image_data, load);
44d3a306	386
53f375fa	387	memmove((void *)load,
44d3a306 SG	388	(void *)image_data,
	389	image_get_data_size(fdt_hdr));
	390
53f375fa	391	fdt_addr = load;
44d3a306	392	break;
21d29f7f	393	#endif
44d3a306 SG	394	case IMAGE_FORMAT_FIT:
	395	/*
	396	* This case will catch both: new uImage format
	397	* (libfdt based) and raw FDT blob (also libfdt
	398	* based).
	399	*/
73223f0e	400	#if CONFIG_IS_ENABLED(FIT)
44d3a306 SG	401	/* check FDT blob vs FIT blob */
44d3a306 SG	402	if (fit_check_format(buf)) {
53f375fa	403	ulong load, len;
44d3a306	404
169043d8	405	fdt_noffset = boot_get_fdt_fit(images,
53f375fa	406	fdt_addr, &fit_uname_fdt,
f320a4d8	407	&fit_uname_config,
169043d8	408	arch, &load, &len);
44d3a306	409
53f375fa	410	images->fit_hdr_fdt = map_sysmem(fdt_addr, 0);
44d3a306 SG	411	images->fit_uname_fdt = fit_uname_fdt;
44d3a306 SG	412	images->fit_noffset_fdt = fdt_noffset;
53f375fa	413	fdt_addr = load;
169043d8	414
44d3a306 SG	415	break;
	416	} else
	417	#endif
	418	{
	419	/*
	420	* FDT blob
	421	*/
44d3a306 SG	422	debug("* fdt: raw FDT blob\n");
	423	printf("## Flattened Device Tree blob at %08lx\n",
	424	(long)fdt_addr);
	425	}
	426	break;
	427	default:
	428	puts("ERROR: Did not find a cmdline Flattened Device Tree\n");
19c6808d	429	goto error;
44d3a306 SG	430	}
44d3a306 SG	431
53f375fa SG	432	printf(" Booting using the fdt blob at %#08lx\n", fdt_addr);
53f375fa SG	433	fdt_blob = map_sysmem(fdt_addr, 0);
44d3a306 SG	434	} else if (images->legacy_hdr_valid &&
	435	image_check_type(&images->legacy_hdr_os_copy,
	436	IH_TYPE_MULTI)) {
	437	ulong fdt_data, fdt_len;
	438
	439	/*
	440	* Now check if we have a legacy multi-component image,
	441	* get second entry data start address and len.
	442	*/
	443	printf("## Flattened Device Tree from multi component Image at %08lX\n",
	444	(ulong)images->legacy_hdr_os);
	445
	446	image_multi_getimg(images->legacy_hdr_os, 2, &fdt_data,
	447	&fdt_len);
	448	if (fdt_len) {
	449	fdt_blob = (char *)fdt_data;
	450	printf(" Booting using the fdt at 0x%p\n", fdt_blob);
	451
	452	if (fdt_check_header(fdt_blob) != 0) {
	453	fdt_error("image is not a fdt");
	454	goto error;
	455	}
	456
	457	if (fdt_totalsize(fdt_blob) != fdt_len) {
	458	fdt_error("fdt size != image size");
	459	goto error;
	460	}
	461	} else {
	462	debug("## No Flattened Device Tree\n");
48aead71	463	goto no_fdt;
44d3a306	464	}
6a7b406a SG	465	#ifdef CONFIG_ANDROID_BOOT_IMAGE
	466	} else if (genimg_get_format(buf) == IMAGE_FORMAT_ANDROID) {
	467	struct andr_img_hdr *hdr = buf;
	468	ulong fdt_data, fdt_len;
	469
18b8f2c4 ER	470	if (!android_image_get_second(hdr, &fdt_data, &fdt_len) &&
	471	!fdt_check_header((char *)fdt_data)) {
	472	fdt_blob = (char *)fdt_data;
	473	if (fdt_totalsize(fdt_blob) != fdt_len)
	474	goto error;
6a7b406a	475
18b8f2c4 ER	476	debug("## Using FDT in Android image second area\n");
18b8f2c4 ER	477	} else {
62392675 ER	478	fdt_addr = env_get_hex("fdtaddr", 0);
	479	if (!fdt_addr)
	480	goto no_fdt;
6a7b406a	481
62392675 ER	482	fdt_blob = map_sysmem(fdt_addr, 0);
	483	if (fdt_check_header(fdt_blob))
	484	goto no_fdt;
6a7b406a	485
62392675	486	debug("## Using FDT at ${fdtaddr}=Ox%lx\n", fdt_addr);
18b8f2c4	487	}
6a7b406a	488	#endif
44d3a306 SG	489	} else {
44d3a306 SG	490	debug("## No Flattened Device Tree\n");
48aead71	491	goto no_fdt;
44d3a306 SG	492	}
	493
	494	*of_flat_tree = fdt_blob;
	495	*of_size = fdt_totalsize(fdt_blob);
	496	debug(" of_flat_tree at 0x%08lx size 0x%08lx\n",
	497	(ulong)of_flat_tree, of_size);
	498
	499	return 0;
	500
48aead71	501	no_fdt:
80281829 ER	502	debug("Continuing to boot without FDT\n");
80281829 ER	503	return 0;
44d3a306	504	error:
44d3a306 SG	505	return 1;
44d3a306 SG	506	}
13d06981 SG	507
	508	/*
	509	* Verify the device tree.
	510	*
	511	* This function is called after all device tree fix-ups have been enacted,
	512	* so that the final device tree can be verified. The definition of "verified"
	513	* is up to the specific implementation. However, it generally means that the
	514	* addresses of some of the devices in the device tree are compared with the
	515	* actual addresses at which U-Boot has placed them.
	516	*
a187559e	517	* Returns 1 on success, 0 on failure. If 0 is returned, U-Boot will halt the
13d06981 SG	518	* boot process.
	519	*/
	520	__weak int ft_verify_fdt(void *fdt)
	521	{
	522	return 1;
	523	}
	524
4280342a AB	525	__weak int arch_fixup_fdt(void *blob)
	526	{
	527	return 0;
	528	}
	529
13d06981 SG	530	int image_setup_libfdt(bootm_headers_t images, void blob,
	531	int of_size, struct lmb *lmb)
	532	{
	533	ulong *initrd_start = &images->initrd_start;
	534	ulong *initrd_end = &images->initrd_end;
6f4dbc21 SG	535	int ret = -EPERM;
6f4dbc21 SG	536	int fdt_ret;
13d06981	537
10be5b5d PK	538	if (fdt_root(blob) < 0) {
	539	printf("ERROR: root node setup failed\n");
	540	goto err;
	541	}
bc6ed0f9	542	if (fdt_chosen(blob) < 0) {
6f4dbc21 SG	543	printf("ERROR: /chosen node create failed\n");
6f4dbc21 SG	544	goto err;
13d06981	545	}
e29607ed	546	if (arch_fixup_fdt(blob) < 0) {
6f4dbc21 SG	547	printf("ERROR: arch-specific fdt fixup failed\n");
	548	goto err;
	549	}
26d61195 TR	550	/* Update ethernet nodes */
26d61195 TR	551	fdt_fixup_ethernet(blob);
6f4dbc21 SG	552	if (IMAGE_OF_BOARD_SETUP) {
	553	fdt_ret = ft_board_setup(blob, gd->bd);
	554	if (fdt_ret) {
	555	printf("ERROR: board-specific fdt fixup failed: %s\n",
	556	fdt_strerror(fdt_ret));
	557	goto err;
	558	}
e29607ed	559	}
c654b517	560	if (IMAGE_OF_SYSTEM_SETUP) {
d89212b7 MK	561	fdt_ret = ft_system_setup(blob, gd->bd);
d89212b7 MK	562	if (fdt_ret) {
c654b517 SG	563	printf("ERROR: system-specific fdt fixup failed: %s\n",
	564	fdt_strerror(fdt_ret));
	565	goto err;
	566	}
	567	}
13d06981	568
6ccb05ea HS	569	fdt_ret = optee_copy_fdt_nodes(gd->fdt_blob, blob);
	570	if (fdt_ret) {
	571	printf("ERROR: transfer of optee nodes to new fdt failed: %s\n",
	572	fdt_strerror(fdt_ret));
	573	goto err;
	574	}
	575
13d06981	576	/* Delete the old LMB reservation */
dea2174d AG	577	if (lmb)
	578	lmb_free(lmb, (phys_addr_t)(u32)(uintptr_t)blob,
	579	(phys_size_t)fdt_totalsize(blob));
13d06981	580
ef476836	581	ret = fdt_shrink_to_minimum(blob, 0);
13d06981	582	if (ret < 0)
6f4dbc21	583	goto err;
13d06981 SG	584	of_size = ret;
	585
	586	if (initrd_start && initrd_end) {
	587	of_size += FDT_RAMDISK_OVERHEAD;
	588	fdt_set_totalsize(blob, of_size);
	589	}
	590	/* Create a new LMB reservation */
dea2174d AG	591	if (lmb)
dea2174d AG	592	lmb_reserve(lmb, (ulong)blob, of_size);
13d06981	593
dbe963ae	594	fdt_initrd(blob, initrd_start, initrd_end);
13d06981	595	if (!ft_verify_fdt(blob))
6f4dbc21	596	goto err;
13d06981	597
7c387646	598	#if defined(CONFIG_SOC_KEYSTONE)
00c200f1 VA	599	if (IMAGE_OF_BOARD_SETUP)
	600	ft_board_setup_ex(blob, gd->bd);
	601	#endif
	602
13d06981	603	return 0;
6f4dbc21 SG	604	err:
	605	printf(" - must RESET the board to recover.\n\n");
	606
	607	return ret;
13d06981	608	}