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