xref: /openbmc/u-boot/tools/fit_image.c (revision 66c433ed)
183d290c5STom Rini // SPDX-License-Identifier: GPL-2.0+
289a4d6b1SPrafulla Wadaskar /*
389a4d6b1SPrafulla Wadaskar  * (C) Copyright 2008 Semihalf
489a4d6b1SPrafulla Wadaskar  *
589a4d6b1SPrafulla Wadaskar  * (C) Copyright 2000-2004
689a4d6b1SPrafulla Wadaskar  * DENX Software Engineering
789a4d6b1SPrafulla Wadaskar  * Wolfgang Denk, wd@denx.de
889a4d6b1SPrafulla Wadaskar  *
989a4d6b1SPrafulla Wadaskar  * Updated-by: Prafulla Wadaskar <prafulla@marvell.com>
1089a4d6b1SPrafulla Wadaskar  *		FIT image specific code abstracted from mkimage.c
1189a4d6b1SPrafulla Wadaskar  *		some functions added to address abstraction
1289a4d6b1SPrafulla Wadaskar  *
1389a4d6b1SPrafulla Wadaskar  * All rights reserved.
1489a4d6b1SPrafulla Wadaskar  */
1589a4d6b1SPrafulla Wadaskar 
16f86ed6a8SGuilherme Maciel Ferreira #include "imagetool.h"
176bf4ca07SHeiko Schocher #include "fit_common.h"
1889a4d6b1SPrafulla Wadaskar #include "mkimage.h"
1989a4d6b1SPrafulla Wadaskar #include <image.h>
208e35bb07SSimon Glass #include <stdarg.h>
218e35bb07SSimon Glass #include <version.h>
2289a4d6b1SPrafulla Wadaskar #include <u-boot/crc.h>
2389a4d6b1SPrafulla Wadaskar 
2489a4d6b1SPrafulla Wadaskar static image_header_t header;
2589a4d6b1SPrafulla Wadaskar 
fit_add_file_data(struct image_tool_params * params,size_t size_inc,const char * tmpfile)26a9468115SSimon Glass static int fit_add_file_data(struct image_tool_params *params, size_t size_inc,
27a9468115SSimon Glass 			     const char *tmpfile)
28a9468115SSimon Glass {
29a9468115SSimon Glass 	int tfd, destfd = 0;
30a9468115SSimon Glass 	void *dest_blob = NULL;
31a9468115SSimon Glass 	off_t destfd_size = 0;
32a9468115SSimon Glass 	struct stat sbuf;
33a9468115SSimon Glass 	void *ptr;
34a9468115SSimon Glass 	int ret = 0;
35a9468115SSimon Glass 
36a9468115SSimon Glass 	tfd = mmap_fdt(params->cmdname, tmpfile, size_inc, &ptr, &sbuf, true);
37a9468115SSimon Glass 	if (tfd < 0)
38a9468115SSimon Glass 		return -EIO;
39a9468115SSimon Glass 
40a9468115SSimon Glass 	if (params->keydest) {
41a9468115SSimon Glass 		struct stat dest_sbuf;
42a9468115SSimon Glass 
43a9468115SSimon Glass 		destfd = mmap_fdt(params->cmdname, params->keydest, size_inc,
44a9468115SSimon Glass 				  &dest_blob, &dest_sbuf, false);
45a9468115SSimon Glass 		if (destfd < 0) {
46a9468115SSimon Glass 			ret = -EIO;
47a9468115SSimon Glass 			goto err_keydest;
48a9468115SSimon Glass 		}
49a9468115SSimon Glass 		destfd_size = dest_sbuf.st_size;
50a9468115SSimon Glass 	}
51a9468115SSimon Glass 
52a9468115SSimon Glass 	/* for first image creation, add a timestamp at offset 0 i.e., root  */
535847084fSVagrant Cascadian 	if (params->datafile) {
5487925df2SAlex Kiernan 		time_t time = imagetool_get_source_date(params->cmdname,
5587925df2SAlex Kiernan 							sbuf.st_mtime);
565847084fSVagrant Cascadian 		ret = fit_set_timestamp(ptr, 0, time);
575847084fSVagrant Cascadian 	}
58a9468115SSimon Glass 
59a9468115SSimon Glass 	if (!ret) {
60a9468115SSimon Glass 		ret = fit_add_verification_data(params->keydir, dest_blob, ptr,
61a9468115SSimon Glass 						params->comment,
62f1ca1fdeSGeorge McCollister 						params->require_keys,
63795f452eSAlex Kiernan 						params->engine_id,
64795f452eSAlex Kiernan 						params->cmdname);
65a9468115SSimon Glass 	}
66a9468115SSimon Glass 
67a9468115SSimon Glass 	if (dest_blob) {
68a9468115SSimon Glass 		munmap(dest_blob, destfd_size);
69a9468115SSimon Glass 		close(destfd);
70a9468115SSimon Glass 	}
71a9468115SSimon Glass 
72a9468115SSimon Glass err_keydest:
73a9468115SSimon Glass 	munmap(ptr, sbuf.st_size);
74a9468115SSimon Glass 	close(tfd);
75a9468115SSimon Glass 
76a9468115SSimon Glass 	return ret;
77a9468115SSimon Glass }
78a9468115SSimon Glass 
7989a4d6b1SPrafulla Wadaskar /**
808e35bb07SSimon Glass  * fit_calc_size() - Calculate the approximate size of the FIT we will generate
818e35bb07SSimon Glass  */
fit_calc_size(struct image_tool_params * params)828e35bb07SSimon Glass static int fit_calc_size(struct image_tool_params *params)
838e35bb07SSimon Glass {
84fb4cce0fSSimon Glass 	struct content_info *cont;
858e35bb07SSimon Glass 	int size, total_size;
868e35bb07SSimon Glass 
878e35bb07SSimon Glass 	size = imagetool_get_filesize(params, params->datafile);
888e35bb07SSimon Glass 	if (size < 0)
898e35bb07SSimon Glass 		return -1;
908e35bb07SSimon Glass 	total_size = size;
910f7c6cdcSTomeu Vizoso 
920f7c6cdcSTomeu Vizoso 	if (params->fit_ramdisk) {
930f7c6cdcSTomeu Vizoso 		size = imagetool_get_filesize(params, params->fit_ramdisk);
940f7c6cdcSTomeu Vizoso 		if (size < 0)
950f7c6cdcSTomeu Vizoso 			return -1;
960f7c6cdcSTomeu Vizoso 		total_size += size;
970f7c6cdcSTomeu Vizoso 	}
980f7c6cdcSTomeu Vizoso 
99fb4cce0fSSimon Glass 	for (cont = params->content_head; cont; cont = cont->next) {
100fb4cce0fSSimon Glass 		size = imagetool_get_filesize(params, cont->fname);
101fb4cce0fSSimon Glass 		if (size < 0)
102fb4cce0fSSimon Glass 			return -1;
1038e35bb07SSimon Glass 
104fb4cce0fSSimon Glass 		/* Add space for properties */
105fb4cce0fSSimon Glass 		total_size += size + 300;
106fb4cce0fSSimon Glass 	}
1078e35bb07SSimon Glass 
1088e35bb07SSimon Glass 	/* Add plenty of space for headers, properties, nodes, etc. */
1098e35bb07SSimon Glass 	total_size += 4096;
1108e35bb07SSimon Glass 
1118e35bb07SSimon Glass 	return total_size;
1128e35bb07SSimon Glass }
1138e35bb07SSimon Glass 
fdt_property_file(struct image_tool_params * params,void * fdt,const char * name,const char * fname)1148e35bb07SSimon Glass static int fdt_property_file(struct image_tool_params *params,
1158e35bb07SSimon Glass 			     void *fdt, const char *name, const char *fname)
1168e35bb07SSimon Glass {
1178e35bb07SSimon Glass 	struct stat sbuf;
1188e35bb07SSimon Glass 	void *ptr;
1198e35bb07SSimon Glass 	int ret;
1208e35bb07SSimon Glass 	int fd;
1218e35bb07SSimon Glass 
1228e35bb07SSimon Glass 	fd = open(fname, O_RDWR | O_BINARY);
1238e35bb07SSimon Glass 	if (fd < 0) {
1248e35bb07SSimon Glass 		fprintf(stderr, "%s: Can't open %s: %s\n",
1258e35bb07SSimon Glass 			params->cmdname, fname, strerror(errno));
1268e35bb07SSimon Glass 		return -1;
1278e35bb07SSimon Glass 	}
1288e35bb07SSimon Glass 
1298e35bb07SSimon Glass 	if (fstat(fd, &sbuf) < 0) {
1308e35bb07SSimon Glass 		fprintf(stderr, "%s: Can't stat %s: %s\n",
1318e35bb07SSimon Glass 			params->cmdname, fname, strerror(errno));
1328e35bb07SSimon Glass 		goto err;
1338e35bb07SSimon Glass 	}
1348e35bb07SSimon Glass 
1358e35bb07SSimon Glass 	ret = fdt_property_placeholder(fdt, "data", sbuf.st_size, &ptr);
1368e35bb07SSimon Glass 	if (ret)
1373bd3a54aSSimon Glass 		goto err;
1388e35bb07SSimon Glass 	ret = read(fd, ptr, sbuf.st_size);
1398e35bb07SSimon Glass 	if (ret != sbuf.st_size) {
1408e35bb07SSimon Glass 		fprintf(stderr, "%s: Can't read %s: %s\n",
1418e35bb07SSimon Glass 			params->cmdname, fname, strerror(errno));
1428e35bb07SSimon Glass 		goto err;
1438e35bb07SSimon Glass 	}
1443bd3a54aSSimon Glass 	close(fd);
1458e35bb07SSimon Glass 
1468e35bb07SSimon Glass 	return 0;
1478e35bb07SSimon Glass err:
1488e35bb07SSimon Glass 	close(fd);
1498e35bb07SSimon Glass 	return -1;
1508e35bb07SSimon Glass }
1518e35bb07SSimon Glass 
fdt_property_strf(void * fdt,const char * name,const char * fmt,...)1528e35bb07SSimon Glass static int fdt_property_strf(void *fdt, const char *name, const char *fmt, ...)
1538e35bb07SSimon Glass {
1548e35bb07SSimon Glass 	char str[100];
1558e35bb07SSimon Glass 	va_list ptr;
1568e35bb07SSimon Glass 
1578e35bb07SSimon Glass 	va_start(ptr, fmt);
1588e35bb07SSimon Glass 	vsnprintf(str, sizeof(str), fmt, ptr);
1598e35bb07SSimon Glass 	va_end(ptr);
1608e35bb07SSimon Glass 	return fdt_property_string(fdt, name, str);
1618e35bb07SSimon Glass }
1628e35bb07SSimon Glass 
get_basename(char * str,int size,const char * fname)163fb4cce0fSSimon Glass static void get_basename(char *str, int size, const char *fname)
164fb4cce0fSSimon Glass {
165fb4cce0fSSimon Glass 	const char *p, *start, *end;
166fb4cce0fSSimon Glass 	int len;
167fb4cce0fSSimon Glass 
168fb4cce0fSSimon Glass 	/*
169fb4cce0fSSimon Glass 	 * Use the base name as the 'name' field. So for example:
170fb4cce0fSSimon Glass 	 *
171fb4cce0fSSimon Glass 	 * "arch/arm/dts/sun7i-a20-bananapro.dtb"
172fb4cce0fSSimon Glass 	 * becomes "sun7i-a20-bananapro"
173fb4cce0fSSimon Glass 	 */
174fb4cce0fSSimon Glass 	p = strrchr(fname, '/');
175fb4cce0fSSimon Glass 	start = p ? p + 1 : fname;
176fb4cce0fSSimon Glass 	p = strrchr(fname, '.');
177fb4cce0fSSimon Glass 	end = p ? p : fname + strlen(fname);
178fb4cce0fSSimon Glass 	len = end - start;
179fb4cce0fSSimon Glass 	if (len >= size)
180fb4cce0fSSimon Glass 		len = size - 1;
181fb4cce0fSSimon Glass 	memcpy(str, start, len);
182fb4cce0fSSimon Glass 	str[len] = '\0';
183fb4cce0fSSimon Glass }
184fb4cce0fSSimon Glass 
1858e35bb07SSimon Glass /**
1868e35bb07SSimon Glass  * fit_write_images() - Write out a list of images to the FIT
1878e35bb07SSimon Glass  *
188fb4cce0fSSimon Glass  * We always include the main image (params->datafile). If there are device
1892eda8e9aSAndre Przywara  * tree files, we include an fdt- node for each of those too.
1908e35bb07SSimon Glass  */
fit_write_images(struct image_tool_params * params,char * fdt)1918e35bb07SSimon Glass static int fit_write_images(struct image_tool_params *params, char *fdt)
1928e35bb07SSimon Glass {
193fb4cce0fSSimon Glass 	struct content_info *cont;
1948e35bb07SSimon Glass 	const char *typename;
1958e35bb07SSimon Glass 	char str[100];
196fb4cce0fSSimon Glass 	int upto;
1978e35bb07SSimon Glass 	int ret;
1988e35bb07SSimon Glass 
1998e35bb07SSimon Glass 	fdt_begin_node(fdt, "images");
2008e35bb07SSimon Glass 
2018e35bb07SSimon Glass 	/* First the main image */
2028e35bb07SSimon Glass 	typename = genimg_get_type_short_name(params->fit_image_type);
2032eda8e9aSAndre Przywara 	snprintf(str, sizeof(str), "%s-1", typename);
2048e35bb07SSimon Glass 	fdt_begin_node(fdt, str);
2054a8b6e01SMichal Simek 	fdt_property_string(fdt, FIT_DESC_PROP, params->imagename);
2064a8b6e01SMichal Simek 	fdt_property_string(fdt, FIT_TYPE_PROP, typename);
2074a8b6e01SMichal Simek 	fdt_property_string(fdt, FIT_ARCH_PROP,
2083a45f38dSSimon Glass 			    genimg_get_arch_short_name(params->arch));
2094a8b6e01SMichal Simek 	fdt_property_string(fdt, FIT_OS_PROP,
2104a8b6e01SMichal Simek 			    genimg_get_os_short_name(params->os));
2114a8b6e01SMichal Simek 	fdt_property_string(fdt, FIT_COMP_PROP,
2128e35bb07SSimon Glass 			    genimg_get_comp_short_name(params->comp));
2134a8b6e01SMichal Simek 	fdt_property_u32(fdt, FIT_LOAD_PROP, params->addr);
2144a8b6e01SMichal Simek 	fdt_property_u32(fdt, FIT_ENTRY_PROP, params->ep);
2158e35bb07SSimon Glass 
2168e35bb07SSimon Glass 	/*
2178e35bb07SSimon Glass 	 * Put data last since it is large. SPL may only load the first part
2188e35bb07SSimon Glass 	 * of the DT, so this way it can access all the above fields.
2198e35bb07SSimon Glass 	 */
2204a8b6e01SMichal Simek 	ret = fdt_property_file(params, fdt, FIT_DATA_PROP, params->datafile);
2218e35bb07SSimon Glass 	if (ret)
2228e35bb07SSimon Glass 		return ret;
2238e35bb07SSimon Glass 	fdt_end_node(fdt);
2248e35bb07SSimon Glass 
225fb4cce0fSSimon Glass 	/* Now the device tree files if available */
226fb4cce0fSSimon Glass 	upto = 0;
227fb4cce0fSSimon Glass 	for (cont = params->content_head; cont; cont = cont->next) {
228fb4cce0fSSimon Glass 		if (cont->type != IH_TYPE_FLATDT)
229fb4cce0fSSimon Glass 			continue;
2302eda8e9aSAndre Przywara 		snprintf(str, sizeof(str), "%s-%d", FIT_FDT_PROP, ++upto);
231fb4cce0fSSimon Glass 		fdt_begin_node(fdt, str);
232fb4cce0fSSimon Glass 
233fb4cce0fSSimon Glass 		get_basename(str, sizeof(str), cont->fname);
2344a8b6e01SMichal Simek 		fdt_property_string(fdt, FIT_DESC_PROP, str);
2354a8b6e01SMichal Simek 		ret = fdt_property_file(params, fdt, FIT_DATA_PROP,
2364a8b6e01SMichal Simek 					cont->fname);
237fb4cce0fSSimon Glass 		if (ret)
238fb4cce0fSSimon Glass 			return ret;
2394a8b6e01SMichal Simek 		fdt_property_string(fdt, FIT_TYPE_PROP, typename);
2404a8b6e01SMichal Simek 		fdt_property_string(fdt, FIT_ARCH_PROP,
241fb4cce0fSSimon Glass 				    genimg_get_arch_short_name(params->arch));
2424a8b6e01SMichal Simek 		fdt_property_string(fdt, FIT_COMP_PROP,
243fb4cce0fSSimon Glass 				    genimg_get_comp_short_name(IH_COMP_NONE));
244fb4cce0fSSimon Glass 		fdt_end_node(fdt);
245fb4cce0fSSimon Glass 	}
246fb4cce0fSSimon Glass 
2470f7c6cdcSTomeu Vizoso 	/* And a ramdisk file if available */
2480f7c6cdcSTomeu Vizoso 	if (params->fit_ramdisk) {
2492eda8e9aSAndre Przywara 		fdt_begin_node(fdt, FIT_RAMDISK_PROP "-1");
2500f7c6cdcSTomeu Vizoso 
2514a8b6e01SMichal Simek 		fdt_property_string(fdt, FIT_TYPE_PROP, FIT_RAMDISK_PROP);
2524a8b6e01SMichal Simek 		fdt_property_string(fdt, FIT_OS_PROP,
2534a8b6e01SMichal Simek 				    genimg_get_os_short_name(params->os));
2540f7c6cdcSTomeu Vizoso 
2554a8b6e01SMichal Simek 		ret = fdt_property_file(params, fdt, FIT_DATA_PROP,
2564a8b6e01SMichal Simek 					params->fit_ramdisk);
2570f7c6cdcSTomeu Vizoso 		if (ret)
2580f7c6cdcSTomeu Vizoso 			return ret;
2590f7c6cdcSTomeu Vizoso 
2600f7c6cdcSTomeu Vizoso 		fdt_end_node(fdt);
2610f7c6cdcSTomeu Vizoso 	}
2620f7c6cdcSTomeu Vizoso 
2638e35bb07SSimon Glass 	fdt_end_node(fdt);
2648e35bb07SSimon Glass 
2658e35bb07SSimon Glass 	return 0;
2668e35bb07SSimon Glass }
2678e35bb07SSimon Glass 
2688e35bb07SSimon Glass /**
2698e35bb07SSimon Glass  * fit_write_configs() - Write out a list of configurations to the FIT
2708e35bb07SSimon Glass  *
271fb4cce0fSSimon Glass  * If there are device tree files, we include a configuration for each, which
272fb4cce0fSSimon Glass  * selects the main image (params->datafile) and its corresponding device
273fb4cce0fSSimon Glass  * tree file.
274fb4cce0fSSimon Glass  *
275fb4cce0fSSimon Glass  * Otherwise we just create a configuration with the main image in it.
2768e35bb07SSimon Glass  */
fit_write_configs(struct image_tool_params * params,char * fdt)2778e35bb07SSimon Glass static void fit_write_configs(struct image_tool_params *params, char *fdt)
2788e35bb07SSimon Glass {
279fb4cce0fSSimon Glass 	struct content_info *cont;
2808e35bb07SSimon Glass 	const char *typename;
2818e35bb07SSimon Glass 	char str[100];
282fb4cce0fSSimon Glass 	int upto;
2838e35bb07SSimon Glass 
2848e35bb07SSimon Glass 	fdt_begin_node(fdt, "configurations");
2854a8b6e01SMichal Simek 	fdt_property_string(fdt, FIT_DEFAULT_PROP, "conf-1");
2868e35bb07SSimon Glass 
287fb4cce0fSSimon Glass 	upto = 0;
288fb4cce0fSSimon Glass 	for (cont = params->content_head; cont; cont = cont->next) {
289fb4cce0fSSimon Glass 		if (cont->type != IH_TYPE_FLATDT)
290fb4cce0fSSimon Glass 			continue;
291fb4cce0fSSimon Glass 		typename = genimg_get_type_short_name(cont->type);
2922eda8e9aSAndre Przywara 		snprintf(str, sizeof(str), "conf-%d", ++upto);
293fb4cce0fSSimon Glass 		fdt_begin_node(fdt, str);
294fb4cce0fSSimon Glass 
295fb4cce0fSSimon Glass 		get_basename(str, sizeof(str), cont->fname);
2964a8b6e01SMichal Simek 		fdt_property_string(fdt, FIT_DESC_PROP, str);
297fb4cce0fSSimon Glass 
298fb4cce0fSSimon Glass 		typename = genimg_get_type_short_name(params->fit_image_type);
2992eda8e9aSAndre Przywara 		snprintf(str, sizeof(str), "%s-1", typename);
300fb4cce0fSSimon Glass 		fdt_property_string(fdt, typename, str);
301*cabde449SAbel Vesa 		fdt_property_string(fdt, FIT_LOADABLE_PROP, str);
302fb4cce0fSSimon Glass 
3030f7c6cdcSTomeu Vizoso 		if (params->fit_ramdisk)
3040f7c6cdcSTomeu Vizoso 			fdt_property_string(fdt, FIT_RAMDISK_PROP,
3052eda8e9aSAndre Przywara 					    FIT_RAMDISK_PROP "-1");
3060f7c6cdcSTomeu Vizoso 
3072eda8e9aSAndre Przywara 		snprintf(str, sizeof(str), FIT_FDT_PROP "-%d", upto);
308fb4cce0fSSimon Glass 		fdt_property_string(fdt, FIT_FDT_PROP, str);
309fb4cce0fSSimon Glass 		fdt_end_node(fdt);
310fb4cce0fSSimon Glass 	}
3110f7c6cdcSTomeu Vizoso 
312fb4cce0fSSimon Glass 	if (!upto) {
3132eda8e9aSAndre Przywara 		fdt_begin_node(fdt, "conf-1");
3148e35bb07SSimon Glass 		typename = genimg_get_type_short_name(params->fit_image_type);
3152eda8e9aSAndre Przywara 		snprintf(str, sizeof(str), "%s-1", typename);
3168e35bb07SSimon Glass 		fdt_property_string(fdt, typename, str);
3170f7c6cdcSTomeu Vizoso 
3180f7c6cdcSTomeu Vizoso 		if (params->fit_ramdisk)
3190f7c6cdcSTomeu Vizoso 			fdt_property_string(fdt, FIT_RAMDISK_PROP,
3202eda8e9aSAndre Przywara 					    FIT_RAMDISK_PROP "-1");
3210f7c6cdcSTomeu Vizoso 
3228e35bb07SSimon Glass 		fdt_end_node(fdt);
323fb4cce0fSSimon Glass 	}
3248e35bb07SSimon Glass 
3258e35bb07SSimon Glass 	fdt_end_node(fdt);
3268e35bb07SSimon Glass }
3278e35bb07SSimon Glass 
fit_build_fdt(struct image_tool_params * params,char * fdt,int size)3288e35bb07SSimon Glass static int fit_build_fdt(struct image_tool_params *params, char *fdt, int size)
3298e35bb07SSimon Glass {
3308e35bb07SSimon Glass 	int ret;
3318e35bb07SSimon Glass 
3328e35bb07SSimon Glass 	ret = fdt_create(fdt, size);
3338e35bb07SSimon Glass 	if (ret)
3348e35bb07SSimon Glass 		return ret;
3358e35bb07SSimon Glass 	fdt_finish_reservemap(fdt);
3368e35bb07SSimon Glass 	fdt_begin_node(fdt, "");
3374a8b6e01SMichal Simek 	fdt_property_strf(fdt, FIT_DESC_PROP,
3388e35bb07SSimon Glass 			  "%s image with one or more FDT blobs",
3398e35bb07SSimon Glass 			  genimg_get_type_name(params->fit_image_type));
3408e35bb07SSimon Glass 	fdt_property_strf(fdt, "creator", "U-Boot mkimage %s", PLAIN_VERSION);
3418e35bb07SSimon Glass 	fdt_property_u32(fdt, "#address-cells", 1);
3428e35bb07SSimon Glass 	ret = fit_write_images(params, fdt);
3438e35bb07SSimon Glass 	if (ret)
3448e35bb07SSimon Glass 		return ret;
3458e35bb07SSimon Glass 	fit_write_configs(params, fdt);
3468e35bb07SSimon Glass 	fdt_end_node(fdt);
3478e35bb07SSimon Glass 	ret = fdt_finish(fdt);
3488e35bb07SSimon Glass 	if (ret)
3498e35bb07SSimon Glass 		return ret;
3508e35bb07SSimon Glass 
3518e35bb07SSimon Glass 	return fdt_totalsize(fdt);
3528e35bb07SSimon Glass }
3538e35bb07SSimon Glass 
fit_build(struct image_tool_params * params,const char * fname)3548e35bb07SSimon Glass static int fit_build(struct image_tool_params *params, const char *fname)
3558e35bb07SSimon Glass {
3568e35bb07SSimon Glass 	char *buf;
3578e35bb07SSimon Glass 	int size;
3588e35bb07SSimon Glass 	int ret;
3598e35bb07SSimon Glass 	int fd;
3608e35bb07SSimon Glass 
3618e35bb07SSimon Glass 	size = fit_calc_size(params);
3628e35bb07SSimon Glass 	if (size < 0)
3638e35bb07SSimon Glass 		return -1;
3648e35bb07SSimon Glass 	buf = malloc(size);
3658e35bb07SSimon Glass 	if (!buf) {
3668e35bb07SSimon Glass 		fprintf(stderr, "%s: Out of memory (%d bytes)\n",
3678e35bb07SSimon Glass 			params->cmdname, size);
3688e35bb07SSimon Glass 		return -1;
3698e35bb07SSimon Glass 	}
3708e35bb07SSimon Glass 	ret = fit_build_fdt(params, buf, size);
3718e35bb07SSimon Glass 	if (ret < 0) {
3728e35bb07SSimon Glass 		fprintf(stderr, "%s: Failed to build FIT image\n",
3738e35bb07SSimon Glass 			params->cmdname);
3747b0bbd88SSimon Glass 		goto err_buf;
3758e35bb07SSimon Glass 	}
3768e35bb07SSimon Glass 	size = ret;
3778e35bb07SSimon Glass 	fd = open(fname, O_RDWR | O_CREAT | O_TRUNC | O_BINARY, 0666);
3788e35bb07SSimon Glass 	if (fd < 0) {
3798e35bb07SSimon Glass 		fprintf(stderr, "%s: Can't open %s: %s\n",
3808e35bb07SSimon Glass 			params->cmdname, fname, strerror(errno));
3813c2dff54STom Rini 		goto err_buf;
3828e35bb07SSimon Glass 	}
3838e35bb07SSimon Glass 	ret = write(fd, buf, size);
3848e35bb07SSimon Glass 	if (ret != size) {
3858e35bb07SSimon Glass 		fprintf(stderr, "%s: Can't write %s: %s\n",
3868e35bb07SSimon Glass 			params->cmdname, fname, strerror(errno));
3878e35bb07SSimon Glass 		goto err;
3888e35bb07SSimon Glass 	}
3898e35bb07SSimon Glass 	close(fd);
3907b0bbd88SSimon Glass 	free(buf);
3918e35bb07SSimon Glass 
3928e35bb07SSimon Glass 	return 0;
3938e35bb07SSimon Glass err:
3947b0bbd88SSimon Glass 	close(fd);
3957b0bbd88SSimon Glass err_buf:
3968e35bb07SSimon Glass 	free(buf);
3978e35bb07SSimon Glass 	return -1;
3988e35bb07SSimon Glass }
3998e35bb07SSimon Glass 
4008e35bb07SSimon Glass /**
401722ebc8fSSimon Glass  * fit_extract_data() - Move all data outside the FIT
402722ebc8fSSimon Glass  *
403722ebc8fSSimon Glass  * This takes a normal FIT file and removes all the 'data' properties from it.
404722ebc8fSSimon Glass  * The data is placed in an area after the FIT so that it can be accessed
405722ebc8fSSimon Glass  * using an offset into that area. The 'data' properties turn into
406722ebc8fSSimon Glass  * 'data-offset' properties.
407722ebc8fSSimon Glass  *
408722ebc8fSSimon Glass  * This function cannot cope with FITs with 'data-offset' properties. All
409722ebc8fSSimon Glass  * data must be in 'data' properties on entry.
410722ebc8fSSimon Glass  */
fit_extract_data(struct image_tool_params * params,const char * fname)411722ebc8fSSimon Glass static int fit_extract_data(struct image_tool_params *params, const char *fname)
412722ebc8fSSimon Glass {
413722ebc8fSSimon Glass 	void *buf;
414722ebc8fSSimon Glass 	int buf_ptr;
415722ebc8fSSimon Glass 	int fit_size, new_size;
416722ebc8fSSimon Glass 	int fd;
417722ebc8fSSimon Glass 	struct stat sbuf;
418722ebc8fSSimon Glass 	void *fdt;
419722ebc8fSSimon Glass 	int ret;
420722ebc8fSSimon Glass 	int images;
421722ebc8fSSimon Glass 	int node;
422722ebc8fSSimon Glass 
423722ebc8fSSimon Glass 	fd = mmap_fdt(params->cmdname, fname, 0, &fdt, &sbuf, false);
424722ebc8fSSimon Glass 	if (fd < 0)
425722ebc8fSSimon Glass 		return -EIO;
426722ebc8fSSimon Glass 	fit_size = fdt_totalsize(fdt);
427722ebc8fSSimon Glass 
428722ebc8fSSimon Glass 	/* Allocate space to hold the image data we will extract */
429722ebc8fSSimon Glass 	buf = malloc(fit_size);
430722ebc8fSSimon Glass 	if (!buf) {
431722ebc8fSSimon Glass 		ret = -ENOMEM;
432b97d71e2SSimon Glass 		goto err_munmap;
433722ebc8fSSimon Glass 	}
434722ebc8fSSimon Glass 	buf_ptr = 0;
435722ebc8fSSimon Glass 
436722ebc8fSSimon Glass 	images = fdt_path_offset(fdt, FIT_IMAGES_PATH);
437722ebc8fSSimon Glass 	if (images < 0) {
438722ebc8fSSimon Glass 		debug("%s: Cannot find /images node: %d\n", __func__, images);
439722ebc8fSSimon Glass 		ret = -EINVAL;
440b97d71e2SSimon Glass 		goto err_munmap;
441722ebc8fSSimon Glass 	}
442722ebc8fSSimon Glass 
443722ebc8fSSimon Glass 	for (node = fdt_first_subnode(fdt, images);
444722ebc8fSSimon Glass 	     node >= 0;
445722ebc8fSSimon Glass 	     node = fdt_next_subnode(fdt, node)) {
446722ebc8fSSimon Glass 		const char *data;
447722ebc8fSSimon Glass 		int len;
448722ebc8fSSimon Glass 
4494a8b6e01SMichal Simek 		data = fdt_getprop(fdt, node, FIT_DATA_PROP, &len);
450722ebc8fSSimon Glass 		if (!data)
451722ebc8fSSimon Glass 			continue;
452722ebc8fSSimon Glass 		memcpy(buf + buf_ptr, data, len);
453722ebc8fSSimon Glass 		debug("Extracting data size %x\n", len);
454722ebc8fSSimon Glass 
4554a8b6e01SMichal Simek 		ret = fdt_delprop(fdt, node, FIT_DATA_PROP);
456722ebc8fSSimon Glass 		if (ret) {
457722ebc8fSSimon Glass 			ret = -EPERM;
458b97d71e2SSimon Glass 			goto err_munmap;
459722ebc8fSSimon Glass 		}
460f8f9107dSTeddy Reed 		if (params->external_offset > 0) {
461f8f9107dSTeddy Reed 			/* An external offset positions the data absolutely. */
4624a8b6e01SMichal Simek 			fdt_setprop_u32(fdt, node, FIT_DATA_POSITION_PROP,
463f8f9107dSTeddy Reed 					params->external_offset + buf_ptr);
464f8f9107dSTeddy Reed 		} else {
4654a8b6e01SMichal Simek 			fdt_setprop_u32(fdt, node, FIT_DATA_OFFSET_PROP,
4664a8b6e01SMichal Simek 					buf_ptr);
467f8f9107dSTeddy Reed 		}
4684a8b6e01SMichal Simek 		fdt_setprop_u32(fdt, node, FIT_DATA_SIZE_PROP, len);
469722ebc8fSSimon Glass 
470722ebc8fSSimon Glass 		buf_ptr += (len + 3) & ~3;
471722ebc8fSSimon Glass 	}
472722ebc8fSSimon Glass 
473722ebc8fSSimon Glass 	/* Pack the FDT and place the data after it */
474722ebc8fSSimon Glass 	fdt_pack(fdt);
475722ebc8fSSimon Glass 
476722ebc8fSSimon Glass 	debug("Size reduced from %x to %x\n", fit_size, fdt_totalsize(fdt));
477722ebc8fSSimon Glass 	debug("External data size %x\n", buf_ptr);
478722ebc8fSSimon Glass 	new_size = fdt_totalsize(fdt);
479722ebc8fSSimon Glass 	new_size = (new_size + 3) & ~3;
480722ebc8fSSimon Glass 	munmap(fdt, sbuf.st_size);
481722ebc8fSSimon Glass 
482722ebc8fSSimon Glass 	if (ftruncate(fd, new_size)) {
483722ebc8fSSimon Glass 		debug("%s: Failed to truncate file: %s\n", __func__,
484722ebc8fSSimon Glass 		      strerror(errno));
485722ebc8fSSimon Glass 		ret = -EIO;
486722ebc8fSSimon Glass 		goto err;
487722ebc8fSSimon Glass 	}
488f8f9107dSTeddy Reed 
489f8f9107dSTeddy Reed 	/* Check if an offset for the external data was set. */
490f8f9107dSTeddy Reed 	if (params->external_offset > 0) {
491f8f9107dSTeddy Reed 		if (params->external_offset < new_size) {
492f8f9107dSTeddy Reed 			debug("External offset %x overlaps FIT length %x",
493f8f9107dSTeddy Reed 			      params->external_offset, new_size);
494f8f9107dSTeddy Reed 			ret = -EINVAL;
495f8f9107dSTeddy Reed 			goto err;
496f8f9107dSTeddy Reed 		}
497f8f9107dSTeddy Reed 		new_size = params->external_offset;
498f8f9107dSTeddy Reed 	}
499722ebc8fSSimon Glass 	if (lseek(fd, new_size, SEEK_SET) < 0) {
500722ebc8fSSimon Glass 		debug("%s: Failed to seek to end of file: %s\n", __func__,
501722ebc8fSSimon Glass 		      strerror(errno));
502722ebc8fSSimon Glass 		ret = -EIO;
503722ebc8fSSimon Glass 		goto err;
504722ebc8fSSimon Glass 	}
505722ebc8fSSimon Glass 	if (write(fd, buf, buf_ptr) != buf_ptr) {
506722ebc8fSSimon Glass 		debug("%s: Failed to write external data to file %s\n",
507722ebc8fSSimon Glass 		      __func__, strerror(errno));
508722ebc8fSSimon Glass 		ret = -EIO;
509722ebc8fSSimon Glass 		goto err;
510722ebc8fSSimon Glass 	}
5113c2dff54STom Rini 	free(buf);
512b97d71e2SSimon Glass 	close(fd);
513b97d71e2SSimon Glass 	return 0;
514722ebc8fSSimon Glass 
515b97d71e2SSimon Glass err_munmap:
516b97d71e2SSimon Glass 	munmap(fdt, sbuf.st_size);
517722ebc8fSSimon Glass err:
51821c2975aSSimon Glass 	if (buf)
51921c2975aSSimon Glass 		free(buf);
520722ebc8fSSimon Glass 	close(fd);
521722ebc8fSSimon Glass 	return ret;
522722ebc8fSSimon Glass }
523722ebc8fSSimon Glass 
fit_import_data(struct image_tool_params * params,const char * fname)524529fd188SSimon Glass static int fit_import_data(struct image_tool_params *params, const char *fname)
525529fd188SSimon Glass {
526529fd188SSimon Glass 	void *fdt, *old_fdt;
527529fd188SSimon Glass 	int fit_size, new_size, size, data_base;
528529fd188SSimon Glass 	int fd;
529529fd188SSimon Glass 	struct stat sbuf;
530529fd188SSimon Glass 	int ret;
531529fd188SSimon Glass 	int images;
532529fd188SSimon Glass 	int node;
533529fd188SSimon Glass 
534529fd188SSimon Glass 	fd = mmap_fdt(params->cmdname, fname, 0, &old_fdt, &sbuf, false);
535529fd188SSimon Glass 	if (fd < 0)
536529fd188SSimon Glass 		return -EIO;
537529fd188SSimon Glass 	fit_size = fdt_totalsize(old_fdt);
538529fd188SSimon Glass 	data_base = (fit_size + 3) & ~3;
539529fd188SSimon Glass 
540529fd188SSimon Glass 	/* Allocate space to hold the new FIT */
541529fd188SSimon Glass 	size = sbuf.st_size + 16384;
542529fd188SSimon Glass 	fdt = malloc(size);
543529fd188SSimon Glass 	if (!fdt) {
544529fd188SSimon Glass 		fprintf(stderr, "%s: Failed to allocate memory (%d bytes)\n",
545529fd188SSimon Glass 			__func__, size);
546529fd188SSimon Glass 		ret = -ENOMEM;
547bf52fcdeSTom Rini 		goto err_has_fd;
548529fd188SSimon Glass 	}
549529fd188SSimon Glass 	ret = fdt_open_into(old_fdt, fdt, size);
550529fd188SSimon Glass 	if (ret) {
551529fd188SSimon Glass 		debug("%s: Failed to expand FIT: %s\n", __func__,
552529fd188SSimon Glass 		      fdt_strerror(errno));
553529fd188SSimon Glass 		ret = -EINVAL;
554bf52fcdeSTom Rini 		goto err_has_fd;
555529fd188SSimon Glass 	}
556529fd188SSimon Glass 
557529fd188SSimon Glass 	images = fdt_path_offset(fdt, FIT_IMAGES_PATH);
558529fd188SSimon Glass 	if (images < 0) {
559529fd188SSimon Glass 		debug("%s: Cannot find /images node: %d\n", __func__, images);
560529fd188SSimon Glass 		ret = -EINVAL;
561bf52fcdeSTom Rini 		goto err_has_fd;
562529fd188SSimon Glass 	}
563529fd188SSimon Glass 
564529fd188SSimon Glass 	for (node = fdt_first_subnode(fdt, images);
565529fd188SSimon Glass 	     node >= 0;
566529fd188SSimon Glass 	     node = fdt_next_subnode(fdt, node)) {
567529fd188SSimon Glass 		int buf_ptr;
568529fd188SSimon Glass 		int len;
569529fd188SSimon Glass 
570529fd188SSimon Glass 		buf_ptr = fdtdec_get_int(fdt, node, "data-offset", -1);
571529fd188SSimon Glass 		len = fdtdec_get_int(fdt, node, "data-size", -1);
572529fd188SSimon Glass 		if (buf_ptr == -1 || len == -1)
573529fd188SSimon Glass 			continue;
574529fd188SSimon Glass 		debug("Importing data size %x\n", len);
575529fd188SSimon Glass 
576529fd188SSimon Glass 		ret = fdt_setprop(fdt, node, "data", fdt + data_base + buf_ptr,
577529fd188SSimon Glass 				  len);
578529fd188SSimon Glass 		if (ret) {
579529fd188SSimon Glass 			debug("%s: Failed to write property: %s\n", __func__,
580529fd188SSimon Glass 			      fdt_strerror(ret));
581529fd188SSimon Glass 			ret = -EINVAL;
582bf52fcdeSTom Rini 			goto err_has_fd;
583529fd188SSimon Glass 		}
584529fd188SSimon Glass 	}
585529fd188SSimon Glass 
586bf52fcdeSTom Rini 	/* Close the old fd so we can re-use it. */
587529fd188SSimon Glass 	close(fd);
588529fd188SSimon Glass 
589529fd188SSimon Glass 	/* Pack the FDT and place the data after it */
590529fd188SSimon Glass 	fdt_pack(fdt);
591529fd188SSimon Glass 
592529fd188SSimon Glass 	new_size = fdt_totalsize(fdt);
593529fd188SSimon Glass 	debug("Size expanded from %x to %x\n", fit_size, new_size);
594529fd188SSimon Glass 
595529fd188SSimon Glass 	fd = open(fname, O_RDWR | O_CREAT | O_TRUNC | O_BINARY, 0666);
596529fd188SSimon Glass 	if (fd < 0) {
597529fd188SSimon Glass 		fprintf(stderr, "%s: Can't open %s: %s\n",
598529fd188SSimon Glass 			params->cmdname, fname, strerror(errno));
599bf52fcdeSTom Rini 		ret = -EIO;
600bf52fcdeSTom Rini 		goto err_no_fd;
601529fd188SSimon Glass 	}
602529fd188SSimon Glass 	if (write(fd, fdt, new_size) != new_size) {
603529fd188SSimon Glass 		debug("%s: Failed to write external data to file %s\n",
604529fd188SSimon Glass 		      __func__, strerror(errno));
605529fd188SSimon Glass 		ret = -EIO;
606bf52fcdeSTom Rini 		goto err_has_fd;
607529fd188SSimon Glass 	}
608529fd188SSimon Glass 
609529fd188SSimon Glass 	ret = 0;
610529fd188SSimon Glass 
611bf52fcdeSTom Rini err_has_fd:
612bf52fcdeSTom Rini 	close(fd);
613bf52fcdeSTom Rini err_no_fd:
6143c2dff54STom Rini 	munmap(old_fdt, sbuf.st_size);
6156e0ffce6SSimon Glass 	free(fdt);
616529fd188SSimon Glass 	return ret;
617529fd188SSimon Glass }
618529fd188SSimon Glass 
619722ebc8fSSimon Glass /**
62089a4d6b1SPrafulla Wadaskar  * fit_handle_file - main FIT file processing function
62189a4d6b1SPrafulla Wadaskar  *
62289a4d6b1SPrafulla Wadaskar  * fit_handle_file() runs dtc to convert .its to .itb, includes
62389a4d6b1SPrafulla Wadaskar  * binary data, updates timestamp property and calculates hashes.
62489a4d6b1SPrafulla Wadaskar  *
62589a4d6b1SPrafulla Wadaskar  * datafile  - .its file
62689a4d6b1SPrafulla Wadaskar  * imagefile - .itb file
62789a4d6b1SPrafulla Wadaskar  *
62889a4d6b1SPrafulla Wadaskar  * returns:
62989a4d6b1SPrafulla Wadaskar  *     only on success, otherwise calls exit (EXIT_FAILURE);
63089a4d6b1SPrafulla Wadaskar  */
fit_handle_file(struct image_tool_params * params)631f86ed6a8SGuilherme Maciel Ferreira static int fit_handle_file(struct image_tool_params *params)
63289a4d6b1SPrafulla Wadaskar {
63389a4d6b1SPrafulla Wadaskar 	char tmpfile[MKIMAGE_MAX_TMPFILE_LEN];
63489a4d6b1SPrafulla Wadaskar 	char cmd[MKIMAGE_MAX_DTC_CMDLINE_LEN];
635a9468115SSimon Glass 	size_t size_inc;
636a9468115SSimon Glass 	int ret;
63789a4d6b1SPrafulla Wadaskar 
63889a4d6b1SPrafulla Wadaskar 	/* Flattened Image Tree (FIT) format  handling */
63989a4d6b1SPrafulla Wadaskar 	debug ("FIT format handling\n");
64089a4d6b1SPrafulla Wadaskar 
64189a4d6b1SPrafulla Wadaskar 	/* call dtc to include binary properties into the tmp file */
64289a4d6b1SPrafulla Wadaskar 	if (strlen (params->imagefile) +
64389a4d6b1SPrafulla Wadaskar 		strlen (MKIMAGE_TMPFILE_SUFFIX) + 1 > sizeof (tmpfile)) {
64489a4d6b1SPrafulla Wadaskar 		fprintf (stderr, "%s: Image file name (%s) too long, "
64589a4d6b1SPrafulla Wadaskar 				"can't create tmpfile",
64689a4d6b1SPrafulla Wadaskar 				params->imagefile, params->cmdname);
64789a4d6b1SPrafulla Wadaskar 		return (EXIT_FAILURE);
64889a4d6b1SPrafulla Wadaskar 	}
64989a4d6b1SPrafulla Wadaskar 	sprintf (tmpfile, "%s%s", params->imagefile, MKIMAGE_TMPFILE_SUFFIX);
65089a4d6b1SPrafulla Wadaskar 
65195d77b44SSimon Glass 	/* We either compile the source file, or use the existing FIT image */
6528e35bb07SSimon Glass 	if (params->auto_its) {
6538e35bb07SSimon Glass 		if (fit_build(params, tmpfile)) {
6548e35bb07SSimon Glass 			fprintf(stderr, "%s: failed to build FIT\n",
6558e35bb07SSimon Glass 				params->cmdname);
6568e35bb07SSimon Glass 			return EXIT_FAILURE;
6578e35bb07SSimon Glass 		}
6588e35bb07SSimon Glass 		*cmd = '\0';
6598e35bb07SSimon Glass 	} else if (params->datafile) {
66063f881d4SStefan Theil 		/* dtc -I dts -O dtb -p 500 -o tmpfile datafile */
66163f881d4SStefan Theil 		snprintf(cmd, sizeof(cmd), "%s %s -o \"%s\" \"%s\"",
66263f881d4SStefan Theil 			 MKIMAGE_DTC, params->dtc, tmpfile, params->datafile);
66389a4d6b1SPrafulla Wadaskar 		debug("Trying to execute \"%s\"\n", cmd);
66495d77b44SSimon Glass 	} else {
665a6e98104SMirza, Taimoor 		snprintf(cmd, sizeof(cmd), "cp \"%s\" \"%s\"",
66695d77b44SSimon Glass 			 params->imagefile, tmpfile);
66795d77b44SSimon Glass 	}
6688e35bb07SSimon Glass 	if (*cmd && system(cmd) == -1) {
66989a4d6b1SPrafulla Wadaskar 		fprintf (stderr, "%s: system(%s) failed: %s\n",
67089a4d6b1SPrafulla Wadaskar 				params->cmdname, cmd, strerror(errno));
671aa6d6db4SSimon Glass 		goto err_system;
67289a4d6b1SPrafulla Wadaskar 	}
67389a4d6b1SPrafulla Wadaskar 
674529fd188SSimon Glass 	/* Move the data so it is internal to the FIT, if needed */
675529fd188SSimon Glass 	ret = fit_import_data(params, tmpfile);
676529fd188SSimon Glass 	if (ret)
677529fd188SSimon Glass 		goto err_system;
678529fd188SSimon Glass 
679a9468115SSimon Glass 	/*
680a9468115SSimon Glass 	 * Set hashes for images in the blob. Unfortunately we may need more
681a9468115SSimon Glass 	 * space in either FDT, so keep trying until we succeed.
682a9468115SSimon Glass 	 *
683a9468115SSimon Glass 	 * Note: this is pretty inefficient for signing, since we must
684a9468115SSimon Glass 	 * calculate the signature every time. It would be better to calculate
685a9468115SSimon Glass 	 * all the data and then store it in a separate step. However, this
686a9468115SSimon Glass 	 * would be considerably more complex to implement. Generally a few
687a9468115SSimon Glass 	 * steps of this loop is enough to sign with several keys.
688a9468115SSimon Glass 	 */
689a9468115SSimon Glass 	for (size_inc = 0; size_inc < 64 * 1024; size_inc += 1024) {
690a9468115SSimon Glass 		ret = fit_add_file_data(params, size_inc, tmpfile);
691a9468115SSimon Glass 		if (!ret || ret != -ENOSPC)
692a9468115SSimon Glass 			break;
693e29495d3SSimon Glass 	}
694e29495d3SSimon Glass 
695a9468115SSimon Glass 	if (ret) {
696655cc696SSimon Glass 		fprintf(stderr, "%s Can't add hashes to FIT blob: %d\n",
697655cc696SSimon Glass 			params->cmdname, ret);
698a9468115SSimon Glass 		goto err_system;
699e29495d3SSimon Glass 	}
70089a4d6b1SPrafulla Wadaskar 
701722ebc8fSSimon Glass 	/* Move the data so it is external to the FIT, if requested */
702722ebc8fSSimon Glass 	if (params->external_data) {
703722ebc8fSSimon Glass 		ret = fit_extract_data(params, tmpfile);
704722ebc8fSSimon Glass 		if (ret)
705722ebc8fSSimon Glass 			goto err_system;
706722ebc8fSSimon Glass 	}
707722ebc8fSSimon Glass 
70889a4d6b1SPrafulla Wadaskar 	if (rename (tmpfile, params->imagefile) == -1) {
70989a4d6b1SPrafulla Wadaskar 		fprintf (stderr, "%s: Can't rename %s to %s: %s\n",
71089a4d6b1SPrafulla Wadaskar 				params->cmdname, tmpfile, params->imagefile,
71189a4d6b1SPrafulla Wadaskar 				strerror (errno));
71289a4d6b1SPrafulla Wadaskar 		unlink (tmpfile);
71389a4d6b1SPrafulla Wadaskar 		unlink (params->imagefile);
714a9468115SSimon Glass 		return EXIT_FAILURE;
71589a4d6b1SPrafulla Wadaskar 	}
716a9468115SSimon Glass 	return EXIT_SUCCESS;
717aa6d6db4SSimon Glass 
718aa6d6db4SSimon Glass err_system:
719aa6d6db4SSimon Glass 	unlink(tmpfile);
720aa6d6db4SSimon Glass 	return -1;
72189a4d6b1SPrafulla Wadaskar }
72289a4d6b1SPrafulla Wadaskar 
72339931f96SGuilherme Maciel Ferreira /**
72439931f96SGuilherme Maciel Ferreira  * fit_image_extract - extract a FIT component image
72539931f96SGuilherme Maciel Ferreira  * @fit: pointer to the FIT format image header
72639931f96SGuilherme Maciel Ferreira  * @image_noffset: offset of the component image node
72739931f96SGuilherme Maciel Ferreira  * @file_name: name of the file to store the FIT sub-image
72839931f96SGuilherme Maciel Ferreira  *
72939931f96SGuilherme Maciel Ferreira  * returns:
73039931f96SGuilherme Maciel Ferreira  *     zero in case of success or a negative value if fail.
73139931f96SGuilherme Maciel Ferreira  */
fit_image_extract(const void * fit,int image_noffset,const char * file_name)73239931f96SGuilherme Maciel Ferreira static int fit_image_extract(
73339931f96SGuilherme Maciel Ferreira 	const void *fit,
73439931f96SGuilherme Maciel Ferreira 	int image_noffset,
73539931f96SGuilherme Maciel Ferreira 	const char *file_name)
73639931f96SGuilherme Maciel Ferreira {
73739931f96SGuilherme Maciel Ferreira 	const void *file_data;
73839931f96SGuilherme Maciel Ferreira 	size_t file_size = 0;
73939931f96SGuilherme Maciel Ferreira 
74039931f96SGuilherme Maciel Ferreira 	/* get the "data" property of component at offset "image_noffset" */
74139931f96SGuilherme Maciel Ferreira 	fit_image_get_data(fit, image_noffset, &file_data, &file_size);
74239931f96SGuilherme Maciel Ferreira 
74339931f96SGuilherme Maciel Ferreira 	/* save the "file_data" into the file specified by "file_name" */
74439931f96SGuilherme Maciel Ferreira 	return imagetool_save_subimage(file_name, (ulong) file_data, file_size);
74539931f96SGuilherme Maciel Ferreira }
74639931f96SGuilherme Maciel Ferreira 
74739931f96SGuilherme Maciel Ferreira /**
74839931f96SGuilherme Maciel Ferreira  * fit_extract_contents - retrieve a sub-image component from the FIT image
74939931f96SGuilherme Maciel Ferreira  * @ptr: pointer to the FIT format image header
75039931f96SGuilherme Maciel Ferreira  * @params: command line parameters
75139931f96SGuilherme Maciel Ferreira  *
75239931f96SGuilherme Maciel Ferreira  * returns:
75339931f96SGuilherme Maciel Ferreira  *     zero in case of success or a negative value if fail.
75439931f96SGuilherme Maciel Ferreira  */
fit_extract_contents(void * ptr,struct image_tool_params * params)75539931f96SGuilherme Maciel Ferreira static int fit_extract_contents(void *ptr, struct image_tool_params *params)
75639931f96SGuilherme Maciel Ferreira {
75739931f96SGuilherme Maciel Ferreira 	int images_noffset;
75839931f96SGuilherme Maciel Ferreira 	int noffset;
75939931f96SGuilherme Maciel Ferreira 	int ndepth;
76039931f96SGuilherme Maciel Ferreira 	const void *fit = ptr;
76139931f96SGuilherme Maciel Ferreira 	int count = 0;
76239931f96SGuilherme Maciel Ferreira 	const char *p;
76339931f96SGuilherme Maciel Ferreira 
76439931f96SGuilherme Maciel Ferreira 	/* Indent string is defined in header image.h */
76539931f96SGuilherme Maciel Ferreira 	p = IMAGE_INDENT_STRING;
76639931f96SGuilherme Maciel Ferreira 
76739931f96SGuilherme Maciel Ferreira 	if (!fit_check_format(fit)) {
76839931f96SGuilherme Maciel Ferreira 		printf("Bad FIT image format\n");
76939931f96SGuilherme Maciel Ferreira 		return -1;
77039931f96SGuilherme Maciel Ferreira 	}
77139931f96SGuilherme Maciel Ferreira 
77239931f96SGuilherme Maciel Ferreira 	/* Find images parent node offset */
77339931f96SGuilherme Maciel Ferreira 	images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH);
77439931f96SGuilherme Maciel Ferreira 	if (images_noffset < 0) {
77539931f96SGuilherme Maciel Ferreira 		printf("Can't find images parent node '%s' (%s)\n",
77639931f96SGuilherme Maciel Ferreira 		       FIT_IMAGES_PATH, fdt_strerror(images_noffset));
77739931f96SGuilherme Maciel Ferreira 		return -1;
77839931f96SGuilherme Maciel Ferreira 	}
77939931f96SGuilherme Maciel Ferreira 
78039931f96SGuilherme Maciel Ferreira 	/* Avoid any overrun */
78139931f96SGuilherme Maciel Ferreira 	count = fit_get_subimage_count(fit, images_noffset);
78239931f96SGuilherme Maciel Ferreira 	if ((params->pflag < 0) || (count <= params->pflag)) {
78339931f96SGuilherme Maciel Ferreira 		printf("No such component at '%d'\n", params->pflag);
78439931f96SGuilherme Maciel Ferreira 		return -1;
78539931f96SGuilherme Maciel Ferreira 	}
78639931f96SGuilherme Maciel Ferreira 
78739931f96SGuilherme Maciel Ferreira 	/* Process its subnodes, extract the desired component from image */
78839931f96SGuilherme Maciel Ferreira 	for (ndepth = 0, count = 0,
78939931f96SGuilherme Maciel Ferreira 		noffset = fdt_next_node(fit, images_noffset, &ndepth);
79039931f96SGuilherme Maciel Ferreira 		(noffset >= 0) && (ndepth > 0);
79139931f96SGuilherme Maciel Ferreira 		noffset = fdt_next_node(fit, noffset, &ndepth)) {
79239931f96SGuilherme Maciel Ferreira 		if (ndepth == 1) {
79339931f96SGuilherme Maciel Ferreira 			/*
79439931f96SGuilherme Maciel Ferreira 			 * Direct child node of the images parent node,
79539931f96SGuilherme Maciel Ferreira 			 * i.e. component image node.
79639931f96SGuilherme Maciel Ferreira 			 */
79739931f96SGuilherme Maciel Ferreira 			if (params->pflag == count) {
79839931f96SGuilherme Maciel Ferreira 				printf("Extracted:\n%s Image %u (%s)\n", p,
79939931f96SGuilherme Maciel Ferreira 				       count, fit_get_name(fit, noffset, NULL));
80039931f96SGuilherme Maciel Ferreira 
80139931f96SGuilherme Maciel Ferreira 				fit_image_print(fit, noffset, p);
80239931f96SGuilherme Maciel Ferreira 
80339931f96SGuilherme Maciel Ferreira 				return fit_image_extract(fit, noffset,
80439931f96SGuilherme Maciel Ferreira 						params->outfile);
80539931f96SGuilherme Maciel Ferreira 			}
80639931f96SGuilherme Maciel Ferreira 
80739931f96SGuilherme Maciel Ferreira 			count++;
80839931f96SGuilherme Maciel Ferreira 		}
80939931f96SGuilherme Maciel Ferreira 	}
81039931f96SGuilherme Maciel Ferreira 
81139931f96SGuilherme Maciel Ferreira 	return 0;
81239931f96SGuilherme Maciel Ferreira }
81339931f96SGuilherme Maciel Ferreira 
fit_check_params(struct image_tool_params * params)814f86ed6a8SGuilherme Maciel Ferreira static int fit_check_params(struct image_tool_params *params)
81589a4d6b1SPrafulla Wadaskar {
8168e35bb07SSimon Glass 	if (params->auto_its)
8178e35bb07SSimon Glass 		return 0;
81889a4d6b1SPrafulla Wadaskar 	return	((params->dflag && (params->fflag || params->lflag)) ||
81989a4d6b1SPrafulla Wadaskar 		(params->fflag && (params->dflag || params->lflag)) ||
82089a4d6b1SPrafulla Wadaskar 		(params->lflag && (params->dflag || params->fflag)));
82189a4d6b1SPrafulla Wadaskar }
82289a4d6b1SPrafulla Wadaskar 
823a93648d1SGuilherme Maciel Ferreira U_BOOT_IMAGE_TYPE(
824a93648d1SGuilherme Maciel Ferreira 	fitimage,
825a93648d1SGuilherme Maciel Ferreira 	"FIT Image support",
826a93648d1SGuilherme Maciel Ferreira 	sizeof(image_header_t),
827a93648d1SGuilherme Maciel Ferreira 	(void *)&header,
828a93648d1SGuilherme Maciel Ferreira 	fit_check_params,
829a93648d1SGuilherme Maciel Ferreira 	fit_verify_header,
830a93648d1SGuilherme Maciel Ferreira 	fit_print_contents,
831a93648d1SGuilherme Maciel Ferreira 	NULL,
83239931f96SGuilherme Maciel Ferreira 	fit_extract_contents,
833a93648d1SGuilherme Maciel Ferreira 	fit_check_image_types,
834a93648d1SGuilherme Maciel Ferreira 	fit_handle_file,
835a93648d1SGuilherme Maciel Ferreira 	NULL /* FIT images use DTB header */
836a93648d1SGuilherme Maciel Ferreira );
837