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 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 */ 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 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 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 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 */ 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 */ 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 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 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 */ 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 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 */ 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 */ 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 */ 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 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