189a4d6b1SPrafulla Wadaskar /* 289a4d6b1SPrafulla Wadaskar * (C) Copyright 2008 Semihalf 389a4d6b1SPrafulla Wadaskar * 489a4d6b1SPrafulla Wadaskar * (C) Copyright 2000-2004 589a4d6b1SPrafulla Wadaskar * DENX Software Engineering 689a4d6b1SPrafulla Wadaskar * Wolfgang Denk, wd@denx.de 789a4d6b1SPrafulla Wadaskar * 889a4d6b1SPrafulla Wadaskar * Updated-by: Prafulla Wadaskar <prafulla@marvell.com> 989a4d6b1SPrafulla Wadaskar * FIT image specific code abstracted from mkimage.c 1089a4d6b1SPrafulla Wadaskar * some functions added to address abstraction 1189a4d6b1SPrafulla Wadaskar * 1289a4d6b1SPrafulla Wadaskar * All rights reserved. 1389a4d6b1SPrafulla Wadaskar * 141a459660SWolfgang Denk * SPDX-License-Identifier: GPL-2.0+ 1589a4d6b1SPrafulla Wadaskar */ 1689a4d6b1SPrafulla Wadaskar 17f86ed6a8SGuilherme Maciel Ferreira #include "imagetool.h" 186bf4ca07SHeiko Schocher #include "fit_common.h" 1989a4d6b1SPrafulla Wadaskar #include "mkimage.h" 2089a4d6b1SPrafulla Wadaskar #include <image.h> 218e35bb07SSimon Glass #include <stdarg.h> 228e35bb07SSimon Glass #include <version.h> 2389a4d6b1SPrafulla Wadaskar #include <u-boot/crc.h> 2489a4d6b1SPrafulla Wadaskar 2589a4d6b1SPrafulla Wadaskar static image_header_t header; 2689a4d6b1SPrafulla Wadaskar 27a9468115SSimon Glass static int fit_add_file_data(struct image_tool_params *params, size_t size_inc, 28a9468115SSimon Glass const char *tmpfile) 29a9468115SSimon Glass { 30a9468115SSimon Glass int tfd, destfd = 0; 31a9468115SSimon Glass void *dest_blob = NULL; 32a9468115SSimon Glass off_t destfd_size = 0; 33a9468115SSimon Glass struct stat sbuf; 34a9468115SSimon Glass void *ptr; 35a9468115SSimon Glass int ret = 0; 36a9468115SSimon Glass 37a9468115SSimon Glass tfd = mmap_fdt(params->cmdname, tmpfile, size_inc, &ptr, &sbuf, true); 38a9468115SSimon Glass if (tfd < 0) 39a9468115SSimon Glass return -EIO; 40a9468115SSimon Glass 41a9468115SSimon Glass if (params->keydest) { 42a9468115SSimon Glass struct stat dest_sbuf; 43a9468115SSimon Glass 44a9468115SSimon Glass destfd = mmap_fdt(params->cmdname, params->keydest, size_inc, 45a9468115SSimon Glass &dest_blob, &dest_sbuf, false); 46a9468115SSimon Glass if (destfd < 0) { 47a9468115SSimon Glass ret = -EIO; 48a9468115SSimon Glass goto err_keydest; 49a9468115SSimon Glass } 50a9468115SSimon Glass destfd_size = dest_sbuf.st_size; 51a9468115SSimon Glass } 52a9468115SSimon Glass 53a9468115SSimon Glass /* for first image creation, add a timestamp at offset 0 i.e., root */ 545847084fSVagrant Cascadian if (params->datafile) { 555847084fSVagrant Cascadian time_t time = imagetool_get_source_date(params, 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, 63f1ca1fdeSGeorge McCollister params->engine_id); 64a9468115SSimon Glass } 65a9468115SSimon Glass 66a9468115SSimon Glass if (dest_blob) { 67a9468115SSimon Glass munmap(dest_blob, destfd_size); 68a9468115SSimon Glass close(destfd); 69a9468115SSimon Glass } 70a9468115SSimon Glass 71a9468115SSimon Glass err_keydest: 72a9468115SSimon Glass munmap(ptr, sbuf.st_size); 73a9468115SSimon Glass close(tfd); 74a9468115SSimon Glass 75a9468115SSimon Glass return ret; 76a9468115SSimon Glass } 77a9468115SSimon Glass 7889a4d6b1SPrafulla Wadaskar /** 798e35bb07SSimon Glass * fit_calc_size() - Calculate the approximate size of the FIT we will generate 808e35bb07SSimon Glass */ 818e35bb07SSimon Glass static int fit_calc_size(struct image_tool_params *params) 828e35bb07SSimon Glass { 83fb4cce0fSSimon Glass struct content_info *cont; 848e35bb07SSimon Glass int size, total_size; 858e35bb07SSimon Glass 868e35bb07SSimon Glass size = imagetool_get_filesize(params, params->datafile); 878e35bb07SSimon Glass if (size < 0) 888e35bb07SSimon Glass return -1; 898e35bb07SSimon Glass total_size = size; 900f7c6cdcSTomeu Vizoso 910f7c6cdcSTomeu Vizoso if (params->fit_ramdisk) { 920f7c6cdcSTomeu Vizoso size = imagetool_get_filesize(params, params->fit_ramdisk); 930f7c6cdcSTomeu Vizoso if (size < 0) 940f7c6cdcSTomeu Vizoso return -1; 950f7c6cdcSTomeu Vizoso total_size += size; 960f7c6cdcSTomeu Vizoso } 970f7c6cdcSTomeu Vizoso 98fb4cce0fSSimon Glass for (cont = params->content_head; cont; cont = cont->next) { 99fb4cce0fSSimon Glass size = imagetool_get_filesize(params, cont->fname); 100fb4cce0fSSimon Glass if (size < 0) 101fb4cce0fSSimon Glass return -1; 1028e35bb07SSimon Glass 103fb4cce0fSSimon Glass /* Add space for properties */ 104fb4cce0fSSimon Glass total_size += size + 300; 105fb4cce0fSSimon Glass } 1068e35bb07SSimon Glass 1078e35bb07SSimon Glass /* Add plenty of space for headers, properties, nodes, etc. */ 1088e35bb07SSimon Glass total_size += 4096; 1098e35bb07SSimon Glass 1108e35bb07SSimon Glass return total_size; 1118e35bb07SSimon Glass } 1128e35bb07SSimon Glass 1138e35bb07SSimon Glass static int fdt_property_file(struct image_tool_params *params, 1148e35bb07SSimon Glass void *fdt, const char *name, const char *fname) 1158e35bb07SSimon Glass { 1168e35bb07SSimon Glass struct stat sbuf; 1178e35bb07SSimon Glass void *ptr; 1188e35bb07SSimon Glass int ret; 1198e35bb07SSimon Glass int fd; 1208e35bb07SSimon Glass 1218e35bb07SSimon Glass fd = open(fname, O_RDWR | O_BINARY); 1228e35bb07SSimon Glass if (fd < 0) { 1238e35bb07SSimon Glass fprintf(stderr, "%s: Can't open %s: %s\n", 1248e35bb07SSimon Glass params->cmdname, fname, strerror(errno)); 1258e35bb07SSimon Glass return -1; 1268e35bb07SSimon Glass } 1278e35bb07SSimon Glass 1288e35bb07SSimon Glass if (fstat(fd, &sbuf) < 0) { 1298e35bb07SSimon Glass fprintf(stderr, "%s: Can't stat %s: %s\n", 1308e35bb07SSimon Glass params->cmdname, fname, strerror(errno)); 1318e35bb07SSimon Glass goto err; 1328e35bb07SSimon Glass } 1338e35bb07SSimon Glass 1348e35bb07SSimon Glass ret = fdt_property_placeholder(fdt, "data", sbuf.st_size, &ptr); 1358e35bb07SSimon Glass if (ret) 1363bd3a54aSSimon Glass goto err; 1378e35bb07SSimon Glass ret = read(fd, ptr, sbuf.st_size); 1388e35bb07SSimon Glass if (ret != sbuf.st_size) { 1398e35bb07SSimon Glass fprintf(stderr, "%s: Can't read %s: %s\n", 1408e35bb07SSimon Glass params->cmdname, fname, strerror(errno)); 1418e35bb07SSimon Glass goto err; 1428e35bb07SSimon Glass } 1433bd3a54aSSimon Glass close(fd); 1448e35bb07SSimon Glass 1458e35bb07SSimon Glass return 0; 1468e35bb07SSimon Glass err: 1478e35bb07SSimon Glass close(fd); 1488e35bb07SSimon Glass return -1; 1498e35bb07SSimon Glass } 1508e35bb07SSimon Glass 1518e35bb07SSimon Glass static int fdt_property_strf(void *fdt, const char *name, const char *fmt, ...) 1528e35bb07SSimon Glass { 1538e35bb07SSimon Glass char str[100]; 1548e35bb07SSimon Glass va_list ptr; 1558e35bb07SSimon Glass 1568e35bb07SSimon Glass va_start(ptr, fmt); 1578e35bb07SSimon Glass vsnprintf(str, sizeof(str), fmt, ptr); 1588e35bb07SSimon Glass va_end(ptr); 1598e35bb07SSimon Glass return fdt_property_string(fdt, name, str); 1608e35bb07SSimon Glass } 1618e35bb07SSimon Glass 162fb4cce0fSSimon Glass static void get_basename(char *str, int size, const char *fname) 163fb4cce0fSSimon Glass { 164fb4cce0fSSimon Glass const char *p, *start, *end; 165fb4cce0fSSimon Glass int len; 166fb4cce0fSSimon Glass 167fb4cce0fSSimon Glass /* 168fb4cce0fSSimon Glass * Use the base name as the 'name' field. So for example: 169fb4cce0fSSimon Glass * 170fb4cce0fSSimon Glass * "arch/arm/dts/sun7i-a20-bananapro.dtb" 171fb4cce0fSSimon Glass * becomes "sun7i-a20-bananapro" 172fb4cce0fSSimon Glass */ 173fb4cce0fSSimon Glass p = strrchr(fname, '/'); 174fb4cce0fSSimon Glass start = p ? p + 1 : fname; 175fb4cce0fSSimon Glass p = strrchr(fname, '.'); 176fb4cce0fSSimon Glass end = p ? p : fname + strlen(fname); 177fb4cce0fSSimon Glass len = end - start; 178fb4cce0fSSimon Glass if (len >= size) 179fb4cce0fSSimon Glass len = size - 1; 180fb4cce0fSSimon Glass memcpy(str, start, len); 181fb4cce0fSSimon Glass str[len] = '\0'; 182fb4cce0fSSimon Glass } 183fb4cce0fSSimon Glass 1848e35bb07SSimon Glass /** 1858e35bb07SSimon Glass * fit_write_images() - Write out a list of images to the FIT 1868e35bb07SSimon Glass * 187fb4cce0fSSimon Glass * We always include the main image (params->datafile). If there are device 188fb4cce0fSSimon Glass * tree files, we include an fdt@ node for each of those too. 1898e35bb07SSimon Glass */ 1908e35bb07SSimon Glass static int fit_write_images(struct image_tool_params *params, char *fdt) 1918e35bb07SSimon Glass { 192fb4cce0fSSimon Glass struct content_info *cont; 1938e35bb07SSimon Glass const char *typename; 1948e35bb07SSimon Glass char str[100]; 195fb4cce0fSSimon Glass int upto; 1968e35bb07SSimon Glass int ret; 1978e35bb07SSimon Glass 1988e35bb07SSimon Glass fdt_begin_node(fdt, "images"); 1998e35bb07SSimon Glass 2008e35bb07SSimon Glass /* First the main image */ 2018e35bb07SSimon Glass typename = genimg_get_type_short_name(params->fit_image_type); 2028e35bb07SSimon Glass snprintf(str, sizeof(str), "%s@1", typename); 2038e35bb07SSimon Glass fdt_begin_node(fdt, str); 2048e35bb07SSimon Glass fdt_property_string(fdt, "description", params->imagename); 2058e35bb07SSimon Glass fdt_property_string(fdt, "type", typename); 2063a45f38dSSimon Glass fdt_property_string(fdt, "arch", 2073a45f38dSSimon Glass genimg_get_arch_short_name(params->arch)); 2088e35bb07SSimon Glass fdt_property_string(fdt, "os", genimg_get_os_short_name(params->os)); 2098e35bb07SSimon Glass fdt_property_string(fdt, "compression", 2108e35bb07SSimon Glass genimg_get_comp_short_name(params->comp)); 2118e35bb07SSimon Glass fdt_property_u32(fdt, "load", params->addr); 2128e35bb07SSimon Glass fdt_property_u32(fdt, "entry", params->ep); 2138e35bb07SSimon Glass 2148e35bb07SSimon Glass /* 2158e35bb07SSimon Glass * Put data last since it is large. SPL may only load the first part 2168e35bb07SSimon Glass * of the DT, so this way it can access all the above fields. 2178e35bb07SSimon Glass */ 2188e35bb07SSimon Glass ret = fdt_property_file(params, fdt, "data", params->datafile); 2198e35bb07SSimon Glass if (ret) 2208e35bb07SSimon Glass return ret; 2218e35bb07SSimon Glass fdt_end_node(fdt); 2228e35bb07SSimon Glass 223fb4cce0fSSimon Glass /* Now the device tree files if available */ 224fb4cce0fSSimon Glass upto = 0; 225fb4cce0fSSimon Glass for (cont = params->content_head; cont; cont = cont->next) { 226fb4cce0fSSimon Glass if (cont->type != IH_TYPE_FLATDT) 227fb4cce0fSSimon Glass continue; 228fb4cce0fSSimon Glass snprintf(str, sizeof(str), "%s@%d", FIT_FDT_PROP, ++upto); 229fb4cce0fSSimon Glass fdt_begin_node(fdt, str); 230fb4cce0fSSimon Glass 231fb4cce0fSSimon Glass get_basename(str, sizeof(str), cont->fname); 232fb4cce0fSSimon Glass fdt_property_string(fdt, "description", str); 233fb4cce0fSSimon Glass ret = fdt_property_file(params, fdt, "data", cont->fname); 234fb4cce0fSSimon Glass if (ret) 235fb4cce0fSSimon Glass return ret; 236fb4cce0fSSimon Glass fdt_property_string(fdt, "type", typename); 237fb4cce0fSSimon Glass fdt_property_string(fdt, "arch", 238fb4cce0fSSimon Glass genimg_get_arch_short_name(params->arch)); 239fb4cce0fSSimon Glass fdt_property_string(fdt, "compression", 240fb4cce0fSSimon Glass genimg_get_comp_short_name(IH_COMP_NONE)); 241fb4cce0fSSimon Glass fdt_end_node(fdt); 242fb4cce0fSSimon Glass } 243fb4cce0fSSimon Glass 2440f7c6cdcSTomeu Vizoso /* And a ramdisk file if available */ 2450f7c6cdcSTomeu Vizoso if (params->fit_ramdisk) { 2460f7c6cdcSTomeu Vizoso fdt_begin_node(fdt, FIT_RAMDISK_PROP "@1"); 2470f7c6cdcSTomeu Vizoso 2480f7c6cdcSTomeu Vizoso fdt_property_string(fdt, "type", FIT_RAMDISK_PROP); 2490f7c6cdcSTomeu Vizoso fdt_property_string(fdt, "os", genimg_get_os_short_name(params->os)); 2500f7c6cdcSTomeu Vizoso 2510f7c6cdcSTomeu Vizoso ret = fdt_property_file(params, fdt, "data", params->fit_ramdisk); 2520f7c6cdcSTomeu Vizoso if (ret) 2530f7c6cdcSTomeu Vizoso return ret; 2540f7c6cdcSTomeu Vizoso 2550f7c6cdcSTomeu Vizoso fdt_end_node(fdt); 2560f7c6cdcSTomeu Vizoso } 2570f7c6cdcSTomeu Vizoso 2588e35bb07SSimon Glass fdt_end_node(fdt); 2598e35bb07SSimon Glass 2608e35bb07SSimon Glass return 0; 2618e35bb07SSimon Glass } 2628e35bb07SSimon Glass 2638e35bb07SSimon Glass /** 2648e35bb07SSimon Glass * fit_write_configs() - Write out a list of configurations to the FIT 2658e35bb07SSimon Glass * 266fb4cce0fSSimon Glass * If there are device tree files, we include a configuration for each, which 267fb4cce0fSSimon Glass * selects the main image (params->datafile) and its corresponding device 268fb4cce0fSSimon Glass * tree file. 269fb4cce0fSSimon Glass * 270fb4cce0fSSimon Glass * Otherwise we just create a configuration with the main image in it. 2718e35bb07SSimon Glass */ 2728e35bb07SSimon Glass static void fit_write_configs(struct image_tool_params *params, char *fdt) 2738e35bb07SSimon Glass { 274fb4cce0fSSimon Glass struct content_info *cont; 2758e35bb07SSimon Glass const char *typename; 2768e35bb07SSimon Glass char str[100]; 277fb4cce0fSSimon Glass int upto; 2788e35bb07SSimon Glass 2798e35bb07SSimon Glass fdt_begin_node(fdt, "configurations"); 2808e35bb07SSimon Glass fdt_property_string(fdt, "default", "conf@1"); 2818e35bb07SSimon Glass 282fb4cce0fSSimon Glass upto = 0; 283fb4cce0fSSimon Glass for (cont = params->content_head; cont; cont = cont->next) { 284fb4cce0fSSimon Glass if (cont->type != IH_TYPE_FLATDT) 285fb4cce0fSSimon Glass continue; 286fb4cce0fSSimon Glass typename = genimg_get_type_short_name(cont->type); 287fb4cce0fSSimon Glass snprintf(str, sizeof(str), "conf@%d", ++upto); 288fb4cce0fSSimon Glass fdt_begin_node(fdt, str); 289fb4cce0fSSimon Glass 290fb4cce0fSSimon Glass get_basename(str, sizeof(str), cont->fname); 291fb4cce0fSSimon Glass fdt_property_string(fdt, "description", str); 292fb4cce0fSSimon Glass 293fb4cce0fSSimon Glass typename = genimg_get_type_short_name(params->fit_image_type); 294fb4cce0fSSimon Glass snprintf(str, sizeof(str), "%s@1", typename); 295fb4cce0fSSimon Glass fdt_property_string(fdt, typename, str); 296fb4cce0fSSimon Glass 2970f7c6cdcSTomeu Vizoso if (params->fit_ramdisk) 2980f7c6cdcSTomeu Vizoso fdt_property_string(fdt, FIT_RAMDISK_PROP, 2990f7c6cdcSTomeu Vizoso FIT_RAMDISK_PROP "@1"); 3000f7c6cdcSTomeu Vizoso 301fb4cce0fSSimon Glass snprintf(str, sizeof(str), FIT_FDT_PROP "@%d", upto); 302fb4cce0fSSimon Glass fdt_property_string(fdt, FIT_FDT_PROP, str); 303fb4cce0fSSimon Glass fdt_end_node(fdt); 304fb4cce0fSSimon Glass } 3050f7c6cdcSTomeu Vizoso 306fb4cce0fSSimon Glass if (!upto) { 3078e35bb07SSimon Glass fdt_begin_node(fdt, "conf@1"); 3088e35bb07SSimon Glass typename = genimg_get_type_short_name(params->fit_image_type); 3098e35bb07SSimon Glass snprintf(str, sizeof(str), "%s@1", typename); 3108e35bb07SSimon Glass fdt_property_string(fdt, typename, str); 3110f7c6cdcSTomeu Vizoso 3120f7c6cdcSTomeu Vizoso if (params->fit_ramdisk) 3130f7c6cdcSTomeu Vizoso fdt_property_string(fdt, FIT_RAMDISK_PROP, 3140f7c6cdcSTomeu Vizoso FIT_RAMDISK_PROP "@1"); 3150f7c6cdcSTomeu Vizoso 3168e35bb07SSimon Glass fdt_end_node(fdt); 317fb4cce0fSSimon Glass } 3188e35bb07SSimon Glass 3198e35bb07SSimon Glass fdt_end_node(fdt); 3208e35bb07SSimon Glass } 3218e35bb07SSimon Glass 3228e35bb07SSimon Glass static int fit_build_fdt(struct image_tool_params *params, char *fdt, int size) 3238e35bb07SSimon Glass { 3248e35bb07SSimon Glass int ret; 3258e35bb07SSimon Glass 3268e35bb07SSimon Glass ret = fdt_create(fdt, size); 3278e35bb07SSimon Glass if (ret) 3288e35bb07SSimon Glass return ret; 3298e35bb07SSimon Glass fdt_finish_reservemap(fdt); 3308e35bb07SSimon Glass fdt_begin_node(fdt, ""); 3318e35bb07SSimon Glass fdt_property_strf(fdt, "description", 3328e35bb07SSimon Glass "%s image with one or more FDT blobs", 3338e35bb07SSimon Glass genimg_get_type_name(params->fit_image_type)); 3348e35bb07SSimon Glass fdt_property_strf(fdt, "creator", "U-Boot mkimage %s", PLAIN_VERSION); 3358e35bb07SSimon Glass fdt_property_u32(fdt, "#address-cells", 1); 3368e35bb07SSimon Glass ret = fit_write_images(params, fdt); 3378e35bb07SSimon Glass if (ret) 3388e35bb07SSimon Glass return ret; 3398e35bb07SSimon Glass fit_write_configs(params, fdt); 3408e35bb07SSimon Glass fdt_end_node(fdt); 3418e35bb07SSimon Glass ret = fdt_finish(fdt); 3428e35bb07SSimon Glass if (ret) 3438e35bb07SSimon Glass return ret; 3448e35bb07SSimon Glass 3458e35bb07SSimon Glass return fdt_totalsize(fdt); 3468e35bb07SSimon Glass } 3478e35bb07SSimon Glass 3488e35bb07SSimon Glass static int fit_build(struct image_tool_params *params, const char *fname) 3498e35bb07SSimon Glass { 3508e35bb07SSimon Glass char *buf; 3518e35bb07SSimon Glass int size; 3528e35bb07SSimon Glass int ret; 3538e35bb07SSimon Glass int fd; 3548e35bb07SSimon Glass 3558e35bb07SSimon Glass size = fit_calc_size(params); 3568e35bb07SSimon Glass if (size < 0) 3578e35bb07SSimon Glass return -1; 3588e35bb07SSimon Glass buf = malloc(size); 3598e35bb07SSimon Glass if (!buf) { 3608e35bb07SSimon Glass fprintf(stderr, "%s: Out of memory (%d bytes)\n", 3618e35bb07SSimon Glass params->cmdname, size); 3628e35bb07SSimon Glass return -1; 3638e35bb07SSimon Glass } 3648e35bb07SSimon Glass ret = fit_build_fdt(params, buf, size); 3658e35bb07SSimon Glass if (ret < 0) { 3668e35bb07SSimon Glass fprintf(stderr, "%s: Failed to build FIT image\n", 3678e35bb07SSimon Glass params->cmdname); 3687b0bbd88SSimon Glass goto err_buf; 3698e35bb07SSimon Glass } 3708e35bb07SSimon Glass size = ret; 3718e35bb07SSimon Glass fd = open(fname, O_RDWR | O_CREAT | O_TRUNC | O_BINARY, 0666); 3728e35bb07SSimon Glass if (fd < 0) { 3738e35bb07SSimon Glass fprintf(stderr, "%s: Can't open %s: %s\n", 3748e35bb07SSimon Glass params->cmdname, fname, strerror(errno)); 3753c2dff54STom Rini goto err_buf; 3768e35bb07SSimon Glass } 3778e35bb07SSimon Glass ret = write(fd, buf, size); 3788e35bb07SSimon Glass if (ret != size) { 3798e35bb07SSimon Glass fprintf(stderr, "%s: Can't write %s: %s\n", 3808e35bb07SSimon Glass params->cmdname, fname, strerror(errno)); 3818e35bb07SSimon Glass goto err; 3828e35bb07SSimon Glass } 3838e35bb07SSimon Glass close(fd); 3847b0bbd88SSimon Glass free(buf); 3858e35bb07SSimon Glass 3868e35bb07SSimon Glass return 0; 3878e35bb07SSimon Glass err: 3887b0bbd88SSimon Glass close(fd); 3897b0bbd88SSimon Glass err_buf: 3908e35bb07SSimon Glass free(buf); 3918e35bb07SSimon Glass return -1; 3928e35bb07SSimon Glass } 3938e35bb07SSimon Glass 3948e35bb07SSimon Glass /** 395722ebc8fSSimon Glass * fit_extract_data() - Move all data outside the FIT 396722ebc8fSSimon Glass * 397722ebc8fSSimon Glass * This takes a normal FIT file and removes all the 'data' properties from it. 398722ebc8fSSimon Glass * The data is placed in an area after the FIT so that it can be accessed 399722ebc8fSSimon Glass * using an offset into that area. The 'data' properties turn into 400722ebc8fSSimon Glass * 'data-offset' properties. 401722ebc8fSSimon Glass * 402722ebc8fSSimon Glass * This function cannot cope with FITs with 'data-offset' properties. All 403722ebc8fSSimon Glass * data must be in 'data' properties on entry. 404722ebc8fSSimon Glass */ 405722ebc8fSSimon Glass static int fit_extract_data(struct image_tool_params *params, const char *fname) 406722ebc8fSSimon Glass { 407722ebc8fSSimon Glass void *buf; 408722ebc8fSSimon Glass int buf_ptr; 409722ebc8fSSimon Glass int fit_size, new_size; 410722ebc8fSSimon Glass int fd; 411722ebc8fSSimon Glass struct stat sbuf; 412722ebc8fSSimon Glass void *fdt; 413722ebc8fSSimon Glass int ret; 414722ebc8fSSimon Glass int images; 415722ebc8fSSimon Glass int node; 416722ebc8fSSimon Glass 417722ebc8fSSimon Glass fd = mmap_fdt(params->cmdname, fname, 0, &fdt, &sbuf, false); 418722ebc8fSSimon Glass if (fd < 0) 419722ebc8fSSimon Glass return -EIO; 420722ebc8fSSimon Glass fit_size = fdt_totalsize(fdt); 421722ebc8fSSimon Glass 422722ebc8fSSimon Glass /* Allocate space to hold the image data we will extract */ 423722ebc8fSSimon Glass buf = malloc(fit_size); 424722ebc8fSSimon Glass if (!buf) { 425722ebc8fSSimon Glass ret = -ENOMEM; 426b97d71e2SSimon Glass goto err_munmap; 427722ebc8fSSimon Glass } 428722ebc8fSSimon Glass buf_ptr = 0; 429722ebc8fSSimon Glass 430722ebc8fSSimon Glass images = fdt_path_offset(fdt, FIT_IMAGES_PATH); 431722ebc8fSSimon Glass if (images < 0) { 432722ebc8fSSimon Glass debug("%s: Cannot find /images node: %d\n", __func__, images); 433722ebc8fSSimon Glass ret = -EINVAL; 434b97d71e2SSimon Glass goto err_munmap; 435722ebc8fSSimon Glass } 436722ebc8fSSimon Glass 437722ebc8fSSimon Glass for (node = fdt_first_subnode(fdt, images); 438722ebc8fSSimon Glass node >= 0; 439722ebc8fSSimon Glass node = fdt_next_subnode(fdt, node)) { 440722ebc8fSSimon Glass const char *data; 441722ebc8fSSimon Glass int len; 442722ebc8fSSimon Glass 443722ebc8fSSimon Glass data = fdt_getprop(fdt, node, "data", &len); 444722ebc8fSSimon Glass if (!data) 445722ebc8fSSimon Glass continue; 446722ebc8fSSimon Glass memcpy(buf + buf_ptr, data, len); 447722ebc8fSSimon Glass debug("Extracting data size %x\n", len); 448722ebc8fSSimon Glass 449722ebc8fSSimon Glass ret = fdt_delprop(fdt, node, "data"); 450722ebc8fSSimon Glass if (ret) { 451722ebc8fSSimon Glass ret = -EPERM; 452b97d71e2SSimon Glass goto err_munmap; 453722ebc8fSSimon Glass } 454f8f9107dSTeddy Reed if (params->external_offset > 0) { 455f8f9107dSTeddy Reed /* An external offset positions the data absolutely. */ 456f8f9107dSTeddy Reed fdt_setprop_u32(fdt, node, "data-position", 457f8f9107dSTeddy Reed params->external_offset + buf_ptr); 458f8f9107dSTeddy Reed } else { 459722ebc8fSSimon Glass fdt_setprop_u32(fdt, node, "data-offset", buf_ptr); 460f8f9107dSTeddy Reed } 461722ebc8fSSimon Glass fdt_setprop_u32(fdt, node, "data-size", len); 462722ebc8fSSimon Glass 463722ebc8fSSimon Glass buf_ptr += (len + 3) & ~3; 464722ebc8fSSimon Glass } 465722ebc8fSSimon Glass 466722ebc8fSSimon Glass /* Pack the FDT and place the data after it */ 467722ebc8fSSimon Glass fdt_pack(fdt); 468722ebc8fSSimon Glass 469722ebc8fSSimon Glass debug("Size reduced from %x to %x\n", fit_size, fdt_totalsize(fdt)); 470722ebc8fSSimon Glass debug("External data size %x\n", buf_ptr); 471722ebc8fSSimon Glass new_size = fdt_totalsize(fdt); 472722ebc8fSSimon Glass new_size = (new_size + 3) & ~3; 473722ebc8fSSimon Glass munmap(fdt, sbuf.st_size); 474722ebc8fSSimon Glass 475722ebc8fSSimon Glass if (ftruncate(fd, new_size)) { 476722ebc8fSSimon Glass debug("%s: Failed to truncate file: %s\n", __func__, 477722ebc8fSSimon Glass strerror(errno)); 478722ebc8fSSimon Glass ret = -EIO; 479722ebc8fSSimon Glass goto err; 480722ebc8fSSimon Glass } 481f8f9107dSTeddy Reed 482f8f9107dSTeddy Reed /* Check if an offset for the external data was set. */ 483f8f9107dSTeddy Reed if (params->external_offset > 0) { 484f8f9107dSTeddy Reed if (params->external_offset < new_size) { 485f8f9107dSTeddy Reed debug("External offset %x overlaps FIT length %x", 486f8f9107dSTeddy Reed params->external_offset, new_size); 487f8f9107dSTeddy Reed ret = -EINVAL; 488f8f9107dSTeddy Reed goto err; 489f8f9107dSTeddy Reed } 490f8f9107dSTeddy Reed new_size = params->external_offset; 491f8f9107dSTeddy Reed } 492722ebc8fSSimon Glass if (lseek(fd, new_size, SEEK_SET) < 0) { 493722ebc8fSSimon Glass debug("%s: Failed to seek to end of file: %s\n", __func__, 494722ebc8fSSimon Glass strerror(errno)); 495722ebc8fSSimon Glass ret = -EIO; 496722ebc8fSSimon Glass goto err; 497722ebc8fSSimon Glass } 498722ebc8fSSimon Glass if (write(fd, buf, buf_ptr) != buf_ptr) { 499722ebc8fSSimon Glass debug("%s: Failed to write external data to file %s\n", 500722ebc8fSSimon Glass __func__, strerror(errno)); 501722ebc8fSSimon Glass ret = -EIO; 502722ebc8fSSimon Glass goto err; 503722ebc8fSSimon Glass } 5043c2dff54STom Rini free(buf); 505b97d71e2SSimon Glass close(fd); 506b97d71e2SSimon Glass return 0; 507722ebc8fSSimon Glass 508b97d71e2SSimon Glass err_munmap: 509b97d71e2SSimon Glass munmap(fdt, sbuf.st_size); 510722ebc8fSSimon Glass err: 51121c2975aSSimon Glass if (buf) 51221c2975aSSimon Glass free(buf); 513722ebc8fSSimon Glass close(fd); 514722ebc8fSSimon Glass return ret; 515722ebc8fSSimon Glass } 516722ebc8fSSimon Glass 517529fd188SSimon Glass static int fit_import_data(struct image_tool_params *params, const char *fname) 518529fd188SSimon Glass { 519529fd188SSimon Glass void *fdt, *old_fdt; 520529fd188SSimon Glass int fit_size, new_size, size, data_base; 521529fd188SSimon Glass int fd; 522529fd188SSimon Glass struct stat sbuf; 523529fd188SSimon Glass int ret; 524529fd188SSimon Glass int images; 525529fd188SSimon Glass int node; 526529fd188SSimon Glass 527529fd188SSimon Glass fd = mmap_fdt(params->cmdname, fname, 0, &old_fdt, &sbuf, false); 528529fd188SSimon Glass if (fd < 0) 529529fd188SSimon Glass return -EIO; 530529fd188SSimon Glass fit_size = fdt_totalsize(old_fdt); 531529fd188SSimon Glass data_base = (fit_size + 3) & ~3; 532529fd188SSimon Glass 533529fd188SSimon Glass /* Allocate space to hold the new FIT */ 534529fd188SSimon Glass size = sbuf.st_size + 16384; 535529fd188SSimon Glass fdt = malloc(size); 536529fd188SSimon Glass if (!fdt) { 537529fd188SSimon Glass fprintf(stderr, "%s: Failed to allocate memory (%d bytes)\n", 538529fd188SSimon Glass __func__, size); 539529fd188SSimon Glass ret = -ENOMEM; 540bf52fcdeSTom Rini goto err_has_fd; 541529fd188SSimon Glass } 542529fd188SSimon Glass ret = fdt_open_into(old_fdt, fdt, size); 543529fd188SSimon Glass if (ret) { 544529fd188SSimon Glass debug("%s: Failed to expand FIT: %s\n", __func__, 545529fd188SSimon Glass fdt_strerror(errno)); 546529fd188SSimon Glass ret = -EINVAL; 547bf52fcdeSTom Rini goto err_has_fd; 548529fd188SSimon Glass } 549529fd188SSimon Glass 550529fd188SSimon Glass images = fdt_path_offset(fdt, FIT_IMAGES_PATH); 551529fd188SSimon Glass if (images < 0) { 552529fd188SSimon Glass debug("%s: Cannot find /images node: %d\n", __func__, images); 553529fd188SSimon Glass ret = -EINVAL; 554bf52fcdeSTom Rini goto err_has_fd; 555529fd188SSimon Glass } 556529fd188SSimon Glass 557529fd188SSimon Glass for (node = fdt_first_subnode(fdt, images); 558529fd188SSimon Glass node >= 0; 559529fd188SSimon Glass node = fdt_next_subnode(fdt, node)) { 560529fd188SSimon Glass int buf_ptr; 561529fd188SSimon Glass int len; 562529fd188SSimon Glass 563529fd188SSimon Glass buf_ptr = fdtdec_get_int(fdt, node, "data-offset", -1); 564529fd188SSimon Glass len = fdtdec_get_int(fdt, node, "data-size", -1); 565529fd188SSimon Glass if (buf_ptr == -1 || len == -1) 566529fd188SSimon Glass continue; 567529fd188SSimon Glass debug("Importing data size %x\n", len); 568529fd188SSimon Glass 569529fd188SSimon Glass ret = fdt_setprop(fdt, node, "data", fdt + data_base + buf_ptr, 570529fd188SSimon Glass len); 571529fd188SSimon Glass if (ret) { 572529fd188SSimon Glass debug("%s: Failed to write property: %s\n", __func__, 573529fd188SSimon Glass fdt_strerror(ret)); 574529fd188SSimon Glass ret = -EINVAL; 575bf52fcdeSTom Rini goto err_has_fd; 576529fd188SSimon Glass } 577529fd188SSimon Glass } 578529fd188SSimon Glass 579bf52fcdeSTom Rini /* Close the old fd so we can re-use it. */ 580529fd188SSimon Glass close(fd); 581529fd188SSimon Glass 582529fd188SSimon Glass /* Pack the FDT and place the data after it */ 583529fd188SSimon Glass fdt_pack(fdt); 584529fd188SSimon Glass 585529fd188SSimon Glass new_size = fdt_totalsize(fdt); 586529fd188SSimon Glass debug("Size expanded from %x to %x\n", fit_size, new_size); 587529fd188SSimon Glass 588529fd188SSimon Glass fd = open(fname, O_RDWR | O_CREAT | O_TRUNC | O_BINARY, 0666); 589529fd188SSimon Glass if (fd < 0) { 590529fd188SSimon Glass fprintf(stderr, "%s: Can't open %s: %s\n", 591529fd188SSimon Glass params->cmdname, fname, strerror(errno)); 592bf52fcdeSTom Rini ret = -EIO; 593bf52fcdeSTom Rini goto err_no_fd; 594529fd188SSimon Glass } 595529fd188SSimon Glass if (write(fd, fdt, new_size) != new_size) { 596529fd188SSimon Glass debug("%s: Failed to write external data to file %s\n", 597529fd188SSimon Glass __func__, strerror(errno)); 598529fd188SSimon Glass ret = -EIO; 599bf52fcdeSTom Rini goto err_has_fd; 600529fd188SSimon Glass } 601529fd188SSimon Glass 602529fd188SSimon Glass ret = 0; 603529fd188SSimon Glass 604bf52fcdeSTom Rini err_has_fd: 605bf52fcdeSTom Rini close(fd); 606bf52fcdeSTom Rini err_no_fd: 6073c2dff54STom Rini munmap(old_fdt, sbuf.st_size); 6086e0ffce6SSimon Glass free(fdt); 609529fd188SSimon Glass return ret; 610529fd188SSimon Glass } 611529fd188SSimon Glass 612722ebc8fSSimon Glass /** 61389a4d6b1SPrafulla Wadaskar * fit_handle_file - main FIT file processing function 61489a4d6b1SPrafulla Wadaskar * 61589a4d6b1SPrafulla Wadaskar * fit_handle_file() runs dtc to convert .its to .itb, includes 61689a4d6b1SPrafulla Wadaskar * binary data, updates timestamp property and calculates hashes. 61789a4d6b1SPrafulla Wadaskar * 61889a4d6b1SPrafulla Wadaskar * datafile - .its file 61989a4d6b1SPrafulla Wadaskar * imagefile - .itb file 62089a4d6b1SPrafulla Wadaskar * 62189a4d6b1SPrafulla Wadaskar * returns: 62289a4d6b1SPrafulla Wadaskar * only on success, otherwise calls exit (EXIT_FAILURE); 62389a4d6b1SPrafulla Wadaskar */ 624f86ed6a8SGuilherme Maciel Ferreira static int fit_handle_file(struct image_tool_params *params) 62589a4d6b1SPrafulla Wadaskar { 62689a4d6b1SPrafulla Wadaskar char tmpfile[MKIMAGE_MAX_TMPFILE_LEN]; 62789a4d6b1SPrafulla Wadaskar char cmd[MKIMAGE_MAX_DTC_CMDLINE_LEN]; 628a9468115SSimon Glass size_t size_inc; 629a9468115SSimon Glass int ret; 63089a4d6b1SPrafulla Wadaskar 63189a4d6b1SPrafulla Wadaskar /* Flattened Image Tree (FIT) format handling */ 63289a4d6b1SPrafulla Wadaskar debug ("FIT format handling\n"); 63389a4d6b1SPrafulla Wadaskar 63489a4d6b1SPrafulla Wadaskar /* call dtc to include binary properties into the tmp file */ 63589a4d6b1SPrafulla Wadaskar if (strlen (params->imagefile) + 63689a4d6b1SPrafulla Wadaskar strlen (MKIMAGE_TMPFILE_SUFFIX) + 1 > sizeof (tmpfile)) { 63789a4d6b1SPrafulla Wadaskar fprintf (stderr, "%s: Image file name (%s) too long, " 63889a4d6b1SPrafulla Wadaskar "can't create tmpfile", 63989a4d6b1SPrafulla Wadaskar params->imagefile, params->cmdname); 64089a4d6b1SPrafulla Wadaskar return (EXIT_FAILURE); 64189a4d6b1SPrafulla Wadaskar } 64289a4d6b1SPrafulla Wadaskar sprintf (tmpfile, "%s%s", params->imagefile, MKIMAGE_TMPFILE_SUFFIX); 64389a4d6b1SPrafulla Wadaskar 64495d77b44SSimon Glass /* We either compile the source file, or use the existing FIT image */ 6458e35bb07SSimon Glass if (params->auto_its) { 6468e35bb07SSimon Glass if (fit_build(params, tmpfile)) { 6478e35bb07SSimon Glass fprintf(stderr, "%s: failed to build FIT\n", 6488e35bb07SSimon Glass params->cmdname); 6498e35bb07SSimon Glass return EXIT_FAILURE; 6508e35bb07SSimon Glass } 6518e35bb07SSimon Glass *cmd = '\0'; 6528e35bb07SSimon Glass } else if (params->datafile) { 653ee7c0fc7SLars Rasmusson /* dtc -I dts -O dtb -p 500 datafile > tmpfile */ 654*a6e98104SMirza, Taimoor snprintf(cmd, sizeof(cmd), "%s %s \"%s\" > \"%s\"", 65589a4d6b1SPrafulla Wadaskar MKIMAGE_DTC, params->dtc, params->datafile, tmpfile); 65689a4d6b1SPrafulla Wadaskar debug("Trying to execute \"%s\"\n", cmd); 65795d77b44SSimon Glass } else { 658*a6e98104SMirza, Taimoor snprintf(cmd, sizeof(cmd), "cp \"%s\" \"%s\"", 65995d77b44SSimon Glass params->imagefile, tmpfile); 66095d77b44SSimon Glass } 6618e35bb07SSimon Glass if (*cmd && system(cmd) == -1) { 66289a4d6b1SPrafulla Wadaskar fprintf (stderr, "%s: system(%s) failed: %s\n", 66389a4d6b1SPrafulla Wadaskar params->cmdname, cmd, strerror(errno)); 664aa6d6db4SSimon Glass goto err_system; 66589a4d6b1SPrafulla Wadaskar } 66689a4d6b1SPrafulla Wadaskar 667529fd188SSimon Glass /* Move the data so it is internal to the FIT, if needed */ 668529fd188SSimon Glass ret = fit_import_data(params, tmpfile); 669529fd188SSimon Glass if (ret) 670529fd188SSimon Glass goto err_system; 671529fd188SSimon Glass 672a9468115SSimon Glass /* 673a9468115SSimon Glass * Set hashes for images in the blob. Unfortunately we may need more 674a9468115SSimon Glass * space in either FDT, so keep trying until we succeed. 675a9468115SSimon Glass * 676a9468115SSimon Glass * Note: this is pretty inefficient for signing, since we must 677a9468115SSimon Glass * calculate the signature every time. It would be better to calculate 678a9468115SSimon Glass * all the data and then store it in a separate step. However, this 679a9468115SSimon Glass * would be considerably more complex to implement. Generally a few 680a9468115SSimon Glass * steps of this loop is enough to sign with several keys. 681a9468115SSimon Glass */ 682a9468115SSimon Glass for (size_inc = 0; size_inc < 64 * 1024; size_inc += 1024) { 683a9468115SSimon Glass ret = fit_add_file_data(params, size_inc, tmpfile); 684a9468115SSimon Glass if (!ret || ret != -ENOSPC) 685a9468115SSimon Glass break; 686e29495d3SSimon Glass } 687e29495d3SSimon Glass 688a9468115SSimon Glass if (ret) { 689655cc696SSimon Glass fprintf(stderr, "%s Can't add hashes to FIT blob: %d\n", 690655cc696SSimon Glass params->cmdname, ret); 691a9468115SSimon Glass goto err_system; 692e29495d3SSimon Glass } 69389a4d6b1SPrafulla Wadaskar 694722ebc8fSSimon Glass /* Move the data so it is external to the FIT, if requested */ 695722ebc8fSSimon Glass if (params->external_data) { 696722ebc8fSSimon Glass ret = fit_extract_data(params, tmpfile); 697722ebc8fSSimon Glass if (ret) 698722ebc8fSSimon Glass goto err_system; 699722ebc8fSSimon Glass } 700722ebc8fSSimon Glass 70189a4d6b1SPrafulla Wadaskar if (rename (tmpfile, params->imagefile) == -1) { 70289a4d6b1SPrafulla Wadaskar fprintf (stderr, "%s: Can't rename %s to %s: %s\n", 70389a4d6b1SPrafulla Wadaskar params->cmdname, tmpfile, params->imagefile, 70489a4d6b1SPrafulla Wadaskar strerror (errno)); 70589a4d6b1SPrafulla Wadaskar unlink (tmpfile); 70689a4d6b1SPrafulla Wadaskar unlink (params->imagefile); 707a9468115SSimon Glass return EXIT_FAILURE; 70889a4d6b1SPrafulla Wadaskar } 709a9468115SSimon Glass return EXIT_SUCCESS; 710aa6d6db4SSimon Glass 711aa6d6db4SSimon Glass err_system: 712aa6d6db4SSimon Glass unlink(tmpfile); 713aa6d6db4SSimon Glass return -1; 71489a4d6b1SPrafulla Wadaskar } 71589a4d6b1SPrafulla Wadaskar 71639931f96SGuilherme Maciel Ferreira /** 71739931f96SGuilherme Maciel Ferreira * fit_image_extract - extract a FIT component image 71839931f96SGuilherme Maciel Ferreira * @fit: pointer to the FIT format image header 71939931f96SGuilherme Maciel Ferreira * @image_noffset: offset of the component image node 72039931f96SGuilherme Maciel Ferreira * @file_name: name of the file to store the FIT sub-image 72139931f96SGuilherme Maciel Ferreira * 72239931f96SGuilherme Maciel Ferreira * returns: 72339931f96SGuilherme Maciel Ferreira * zero in case of success or a negative value if fail. 72439931f96SGuilherme Maciel Ferreira */ 72539931f96SGuilherme Maciel Ferreira static int fit_image_extract( 72639931f96SGuilherme Maciel Ferreira const void *fit, 72739931f96SGuilherme Maciel Ferreira int image_noffset, 72839931f96SGuilherme Maciel Ferreira const char *file_name) 72939931f96SGuilherme Maciel Ferreira { 73039931f96SGuilherme Maciel Ferreira const void *file_data; 73139931f96SGuilherme Maciel Ferreira size_t file_size = 0; 73239931f96SGuilherme Maciel Ferreira 73339931f96SGuilherme Maciel Ferreira /* get the "data" property of component at offset "image_noffset" */ 73439931f96SGuilherme Maciel Ferreira fit_image_get_data(fit, image_noffset, &file_data, &file_size); 73539931f96SGuilherme Maciel Ferreira 73639931f96SGuilherme Maciel Ferreira /* save the "file_data" into the file specified by "file_name" */ 73739931f96SGuilherme Maciel Ferreira return imagetool_save_subimage(file_name, (ulong) file_data, file_size); 73839931f96SGuilherme Maciel Ferreira } 73939931f96SGuilherme Maciel Ferreira 74039931f96SGuilherme Maciel Ferreira /** 74139931f96SGuilherme Maciel Ferreira * fit_extract_contents - retrieve a sub-image component from the FIT image 74239931f96SGuilherme Maciel Ferreira * @ptr: pointer to the FIT format image header 74339931f96SGuilherme Maciel Ferreira * @params: command line parameters 74439931f96SGuilherme Maciel Ferreira * 74539931f96SGuilherme Maciel Ferreira * returns: 74639931f96SGuilherme Maciel Ferreira * zero in case of success or a negative value if fail. 74739931f96SGuilherme Maciel Ferreira */ 74839931f96SGuilherme Maciel Ferreira static int fit_extract_contents(void *ptr, struct image_tool_params *params) 74939931f96SGuilherme Maciel Ferreira { 75039931f96SGuilherme Maciel Ferreira int images_noffset; 75139931f96SGuilherme Maciel Ferreira int noffset; 75239931f96SGuilherme Maciel Ferreira int ndepth; 75339931f96SGuilherme Maciel Ferreira const void *fit = ptr; 75439931f96SGuilherme Maciel Ferreira int count = 0; 75539931f96SGuilherme Maciel Ferreira const char *p; 75639931f96SGuilherme Maciel Ferreira 75739931f96SGuilherme Maciel Ferreira /* Indent string is defined in header image.h */ 75839931f96SGuilherme Maciel Ferreira p = IMAGE_INDENT_STRING; 75939931f96SGuilherme Maciel Ferreira 76039931f96SGuilherme Maciel Ferreira if (!fit_check_format(fit)) { 76139931f96SGuilherme Maciel Ferreira printf("Bad FIT image format\n"); 76239931f96SGuilherme Maciel Ferreira return -1; 76339931f96SGuilherme Maciel Ferreira } 76439931f96SGuilherme Maciel Ferreira 76539931f96SGuilherme Maciel Ferreira /* Find images parent node offset */ 76639931f96SGuilherme Maciel Ferreira images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH); 76739931f96SGuilherme Maciel Ferreira if (images_noffset < 0) { 76839931f96SGuilherme Maciel Ferreira printf("Can't find images parent node '%s' (%s)\n", 76939931f96SGuilherme Maciel Ferreira FIT_IMAGES_PATH, fdt_strerror(images_noffset)); 77039931f96SGuilherme Maciel Ferreira return -1; 77139931f96SGuilherme Maciel Ferreira } 77239931f96SGuilherme Maciel Ferreira 77339931f96SGuilherme Maciel Ferreira /* Avoid any overrun */ 77439931f96SGuilherme Maciel Ferreira count = fit_get_subimage_count(fit, images_noffset); 77539931f96SGuilherme Maciel Ferreira if ((params->pflag < 0) || (count <= params->pflag)) { 77639931f96SGuilherme Maciel Ferreira printf("No such component at '%d'\n", params->pflag); 77739931f96SGuilherme Maciel Ferreira return -1; 77839931f96SGuilherme Maciel Ferreira } 77939931f96SGuilherme Maciel Ferreira 78039931f96SGuilherme Maciel Ferreira /* Process its subnodes, extract the desired component from image */ 78139931f96SGuilherme Maciel Ferreira for (ndepth = 0, count = 0, 78239931f96SGuilherme Maciel Ferreira noffset = fdt_next_node(fit, images_noffset, &ndepth); 78339931f96SGuilherme Maciel Ferreira (noffset >= 0) && (ndepth > 0); 78439931f96SGuilherme Maciel Ferreira noffset = fdt_next_node(fit, noffset, &ndepth)) { 78539931f96SGuilherme Maciel Ferreira if (ndepth == 1) { 78639931f96SGuilherme Maciel Ferreira /* 78739931f96SGuilherme Maciel Ferreira * Direct child node of the images parent node, 78839931f96SGuilherme Maciel Ferreira * i.e. component image node. 78939931f96SGuilherme Maciel Ferreira */ 79039931f96SGuilherme Maciel Ferreira if (params->pflag == count) { 79139931f96SGuilherme Maciel Ferreira printf("Extracted:\n%s Image %u (%s)\n", p, 79239931f96SGuilherme Maciel Ferreira count, fit_get_name(fit, noffset, NULL)); 79339931f96SGuilherme Maciel Ferreira 79439931f96SGuilherme Maciel Ferreira fit_image_print(fit, noffset, p); 79539931f96SGuilherme Maciel Ferreira 79639931f96SGuilherme Maciel Ferreira return fit_image_extract(fit, noffset, 79739931f96SGuilherme Maciel Ferreira params->outfile); 79839931f96SGuilherme Maciel Ferreira } 79939931f96SGuilherme Maciel Ferreira 80039931f96SGuilherme Maciel Ferreira count++; 80139931f96SGuilherme Maciel Ferreira } 80239931f96SGuilherme Maciel Ferreira } 80339931f96SGuilherme Maciel Ferreira 80439931f96SGuilherme Maciel Ferreira return 0; 80539931f96SGuilherme Maciel Ferreira } 80639931f96SGuilherme Maciel Ferreira 807f86ed6a8SGuilherme Maciel Ferreira static int fit_check_params(struct image_tool_params *params) 80889a4d6b1SPrafulla Wadaskar { 8098e35bb07SSimon Glass if (params->auto_its) 8108e35bb07SSimon Glass return 0; 81189a4d6b1SPrafulla Wadaskar return ((params->dflag && (params->fflag || params->lflag)) || 81289a4d6b1SPrafulla Wadaskar (params->fflag && (params->dflag || params->lflag)) || 81389a4d6b1SPrafulla Wadaskar (params->lflag && (params->dflag || params->fflag))); 81489a4d6b1SPrafulla Wadaskar } 81589a4d6b1SPrafulla Wadaskar 816a93648d1SGuilherme Maciel Ferreira U_BOOT_IMAGE_TYPE( 817a93648d1SGuilherme Maciel Ferreira fitimage, 818a93648d1SGuilherme Maciel Ferreira "FIT Image support", 819a93648d1SGuilherme Maciel Ferreira sizeof(image_header_t), 820a93648d1SGuilherme Maciel Ferreira (void *)&header, 821a93648d1SGuilherme Maciel Ferreira fit_check_params, 822a93648d1SGuilherme Maciel Ferreira fit_verify_header, 823a93648d1SGuilherme Maciel Ferreira fit_print_contents, 824a93648d1SGuilherme Maciel Ferreira NULL, 82539931f96SGuilherme Maciel Ferreira fit_extract_contents, 826a93648d1SGuilherme Maciel Ferreira fit_check_image_types, 827a93648d1SGuilherme Maciel Ferreira fit_handle_file, 828a93648d1SGuilherme Maciel Ferreira NULL /* FIT images use DTB header */ 829a93648d1SGuilherme Maciel Ferreira ); 830