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, 62a9468115SSimon Glass params->require_keys); 63a9468115SSimon Glass } 64a9468115SSimon Glass 65a9468115SSimon Glass if (dest_blob) { 66a9468115SSimon Glass munmap(dest_blob, destfd_size); 67a9468115SSimon Glass close(destfd); 68a9468115SSimon Glass } 69a9468115SSimon Glass 70a9468115SSimon Glass err_keydest: 71a9468115SSimon Glass munmap(ptr, sbuf.st_size); 72a9468115SSimon Glass close(tfd); 73a9468115SSimon Glass 74a9468115SSimon Glass return ret; 75a9468115SSimon Glass } 76a9468115SSimon Glass 7789a4d6b1SPrafulla Wadaskar /** 788e35bb07SSimon Glass * fit_calc_size() - Calculate the approximate size of the FIT we will generate 798e35bb07SSimon Glass */ 808e35bb07SSimon Glass static int fit_calc_size(struct image_tool_params *params) 818e35bb07SSimon Glass { 82fb4cce0fSSimon Glass struct content_info *cont; 838e35bb07SSimon Glass int size, total_size; 848e35bb07SSimon Glass 858e35bb07SSimon Glass size = imagetool_get_filesize(params, params->datafile); 868e35bb07SSimon Glass if (size < 0) 878e35bb07SSimon Glass return -1; 888e35bb07SSimon Glass total_size = size; 89*0f7c6cdcSTomeu Vizoso 90*0f7c6cdcSTomeu Vizoso if (params->fit_ramdisk) { 91*0f7c6cdcSTomeu Vizoso size = imagetool_get_filesize(params, params->fit_ramdisk); 92*0f7c6cdcSTomeu Vizoso if (size < 0) 93*0f7c6cdcSTomeu Vizoso return -1; 94*0f7c6cdcSTomeu Vizoso total_size += size; 95*0f7c6cdcSTomeu Vizoso } 96*0f7c6cdcSTomeu Vizoso 97fb4cce0fSSimon Glass for (cont = params->content_head; cont; cont = cont->next) { 98fb4cce0fSSimon Glass size = imagetool_get_filesize(params, cont->fname); 99fb4cce0fSSimon Glass if (size < 0) 100fb4cce0fSSimon Glass return -1; 1018e35bb07SSimon Glass 102fb4cce0fSSimon Glass /* Add space for properties */ 103fb4cce0fSSimon Glass total_size += size + 300; 104fb4cce0fSSimon Glass } 1058e35bb07SSimon Glass 1068e35bb07SSimon Glass /* Add plenty of space for headers, properties, nodes, etc. */ 1078e35bb07SSimon Glass total_size += 4096; 1088e35bb07SSimon Glass 1098e35bb07SSimon Glass return total_size; 1108e35bb07SSimon Glass } 1118e35bb07SSimon Glass 1128e35bb07SSimon Glass static int fdt_property_file(struct image_tool_params *params, 1138e35bb07SSimon Glass void *fdt, const char *name, const char *fname) 1148e35bb07SSimon Glass { 1158e35bb07SSimon Glass struct stat sbuf; 1168e35bb07SSimon Glass void *ptr; 1178e35bb07SSimon Glass int ret; 1188e35bb07SSimon Glass int fd; 1198e35bb07SSimon Glass 1208e35bb07SSimon Glass fd = open(fname, O_RDWR | O_BINARY); 1218e35bb07SSimon Glass if (fd < 0) { 1228e35bb07SSimon Glass fprintf(stderr, "%s: Can't open %s: %s\n", 1238e35bb07SSimon Glass params->cmdname, fname, strerror(errno)); 1248e35bb07SSimon Glass return -1; 1258e35bb07SSimon Glass } 1268e35bb07SSimon Glass 1278e35bb07SSimon Glass if (fstat(fd, &sbuf) < 0) { 1288e35bb07SSimon Glass fprintf(stderr, "%s: Can't stat %s: %s\n", 1298e35bb07SSimon Glass params->cmdname, fname, strerror(errno)); 1308e35bb07SSimon Glass goto err; 1318e35bb07SSimon Glass } 1328e35bb07SSimon Glass 1338e35bb07SSimon Glass ret = fdt_property_placeholder(fdt, "data", sbuf.st_size, &ptr); 1348e35bb07SSimon Glass if (ret) 1353bd3a54aSSimon Glass goto err; 1368e35bb07SSimon Glass ret = read(fd, ptr, sbuf.st_size); 1378e35bb07SSimon Glass if (ret != sbuf.st_size) { 1388e35bb07SSimon Glass fprintf(stderr, "%s: Can't read %s: %s\n", 1398e35bb07SSimon Glass params->cmdname, fname, strerror(errno)); 1408e35bb07SSimon Glass goto err; 1418e35bb07SSimon Glass } 1423bd3a54aSSimon Glass close(fd); 1438e35bb07SSimon Glass 1448e35bb07SSimon Glass return 0; 1458e35bb07SSimon Glass err: 1468e35bb07SSimon Glass close(fd); 1478e35bb07SSimon Glass return -1; 1488e35bb07SSimon Glass } 1498e35bb07SSimon Glass 1508e35bb07SSimon Glass static int fdt_property_strf(void *fdt, const char *name, const char *fmt, ...) 1518e35bb07SSimon Glass { 1528e35bb07SSimon Glass char str[100]; 1538e35bb07SSimon Glass va_list ptr; 1548e35bb07SSimon Glass 1558e35bb07SSimon Glass va_start(ptr, fmt); 1568e35bb07SSimon Glass vsnprintf(str, sizeof(str), fmt, ptr); 1578e35bb07SSimon Glass va_end(ptr); 1588e35bb07SSimon Glass return fdt_property_string(fdt, name, str); 1598e35bb07SSimon Glass } 1608e35bb07SSimon Glass 161fb4cce0fSSimon Glass static void get_basename(char *str, int size, const char *fname) 162fb4cce0fSSimon Glass { 163fb4cce0fSSimon Glass const char *p, *start, *end; 164fb4cce0fSSimon Glass int len; 165fb4cce0fSSimon Glass 166fb4cce0fSSimon Glass /* 167fb4cce0fSSimon Glass * Use the base name as the 'name' field. So for example: 168fb4cce0fSSimon Glass * 169fb4cce0fSSimon Glass * "arch/arm/dts/sun7i-a20-bananapro.dtb" 170fb4cce0fSSimon Glass * becomes "sun7i-a20-bananapro" 171fb4cce0fSSimon Glass */ 172fb4cce0fSSimon Glass p = strrchr(fname, '/'); 173fb4cce0fSSimon Glass start = p ? p + 1 : fname; 174fb4cce0fSSimon Glass p = strrchr(fname, '.'); 175fb4cce0fSSimon Glass end = p ? p : fname + strlen(fname); 176fb4cce0fSSimon Glass len = end - start; 177fb4cce0fSSimon Glass if (len >= size) 178fb4cce0fSSimon Glass len = size - 1; 179fb4cce0fSSimon Glass memcpy(str, start, len); 180fb4cce0fSSimon Glass str[len] = '\0'; 181fb4cce0fSSimon Glass } 182fb4cce0fSSimon Glass 1838e35bb07SSimon Glass /** 1848e35bb07SSimon Glass * fit_write_images() - Write out a list of images to the FIT 1858e35bb07SSimon Glass * 186fb4cce0fSSimon Glass * We always include the main image (params->datafile). If there are device 187fb4cce0fSSimon Glass * tree files, we include an fdt@ node for each of those too. 1888e35bb07SSimon Glass */ 1898e35bb07SSimon Glass static int fit_write_images(struct image_tool_params *params, char *fdt) 1908e35bb07SSimon Glass { 191fb4cce0fSSimon Glass struct content_info *cont; 1928e35bb07SSimon Glass const char *typename; 1938e35bb07SSimon Glass char str[100]; 194fb4cce0fSSimon Glass int upto; 1958e35bb07SSimon Glass int ret; 1968e35bb07SSimon Glass 1978e35bb07SSimon Glass fdt_begin_node(fdt, "images"); 1988e35bb07SSimon Glass 1998e35bb07SSimon Glass /* First the main image */ 2008e35bb07SSimon Glass typename = genimg_get_type_short_name(params->fit_image_type); 2018e35bb07SSimon Glass snprintf(str, sizeof(str), "%s@1", typename); 2028e35bb07SSimon Glass fdt_begin_node(fdt, str); 2038e35bb07SSimon Glass fdt_property_string(fdt, "description", params->imagename); 2048e35bb07SSimon Glass fdt_property_string(fdt, "type", typename); 2053a45f38dSSimon Glass fdt_property_string(fdt, "arch", 2063a45f38dSSimon Glass genimg_get_arch_short_name(params->arch)); 2078e35bb07SSimon Glass fdt_property_string(fdt, "os", genimg_get_os_short_name(params->os)); 2088e35bb07SSimon Glass fdt_property_string(fdt, "compression", 2098e35bb07SSimon Glass genimg_get_comp_short_name(params->comp)); 2108e35bb07SSimon Glass fdt_property_u32(fdt, "load", params->addr); 2118e35bb07SSimon Glass fdt_property_u32(fdt, "entry", params->ep); 2128e35bb07SSimon Glass 2138e35bb07SSimon Glass /* 2148e35bb07SSimon Glass * Put data last since it is large. SPL may only load the first part 2158e35bb07SSimon Glass * of the DT, so this way it can access all the above fields. 2168e35bb07SSimon Glass */ 2178e35bb07SSimon Glass ret = fdt_property_file(params, fdt, "data", params->datafile); 2188e35bb07SSimon Glass if (ret) 2198e35bb07SSimon Glass return ret; 2208e35bb07SSimon Glass fdt_end_node(fdt); 2218e35bb07SSimon Glass 222fb4cce0fSSimon Glass /* Now the device tree files if available */ 223fb4cce0fSSimon Glass upto = 0; 224fb4cce0fSSimon Glass for (cont = params->content_head; cont; cont = cont->next) { 225fb4cce0fSSimon Glass if (cont->type != IH_TYPE_FLATDT) 226fb4cce0fSSimon Glass continue; 227fb4cce0fSSimon Glass snprintf(str, sizeof(str), "%s@%d", FIT_FDT_PROP, ++upto); 228fb4cce0fSSimon Glass fdt_begin_node(fdt, str); 229fb4cce0fSSimon Glass 230fb4cce0fSSimon Glass get_basename(str, sizeof(str), cont->fname); 231fb4cce0fSSimon Glass fdt_property_string(fdt, "description", str); 232fb4cce0fSSimon Glass ret = fdt_property_file(params, fdt, "data", cont->fname); 233fb4cce0fSSimon Glass if (ret) 234fb4cce0fSSimon Glass return ret; 235fb4cce0fSSimon Glass fdt_property_string(fdt, "type", typename); 236fb4cce0fSSimon Glass fdt_property_string(fdt, "arch", 237fb4cce0fSSimon Glass genimg_get_arch_short_name(params->arch)); 238fb4cce0fSSimon Glass fdt_property_string(fdt, "compression", 239fb4cce0fSSimon Glass genimg_get_comp_short_name(IH_COMP_NONE)); 240fb4cce0fSSimon Glass fdt_end_node(fdt); 241fb4cce0fSSimon Glass } 242fb4cce0fSSimon Glass 243*0f7c6cdcSTomeu Vizoso /* And a ramdisk file if available */ 244*0f7c6cdcSTomeu Vizoso if (params->fit_ramdisk) { 245*0f7c6cdcSTomeu Vizoso fdt_begin_node(fdt, FIT_RAMDISK_PROP "@1"); 246*0f7c6cdcSTomeu Vizoso 247*0f7c6cdcSTomeu Vizoso fdt_property_string(fdt, "type", FIT_RAMDISK_PROP); 248*0f7c6cdcSTomeu Vizoso fdt_property_string(fdt, "os", genimg_get_os_short_name(params->os)); 249*0f7c6cdcSTomeu Vizoso 250*0f7c6cdcSTomeu Vizoso ret = fdt_property_file(params, fdt, "data", params->fit_ramdisk); 251*0f7c6cdcSTomeu Vizoso if (ret) 252*0f7c6cdcSTomeu Vizoso return ret; 253*0f7c6cdcSTomeu Vizoso 254*0f7c6cdcSTomeu Vizoso fdt_end_node(fdt); 255*0f7c6cdcSTomeu Vizoso } 256*0f7c6cdcSTomeu Vizoso 2578e35bb07SSimon Glass fdt_end_node(fdt); 2588e35bb07SSimon Glass 2598e35bb07SSimon Glass return 0; 2608e35bb07SSimon Glass } 2618e35bb07SSimon Glass 2628e35bb07SSimon Glass /** 2638e35bb07SSimon Glass * fit_write_configs() - Write out a list of configurations to the FIT 2648e35bb07SSimon Glass * 265fb4cce0fSSimon Glass * If there are device tree files, we include a configuration for each, which 266fb4cce0fSSimon Glass * selects the main image (params->datafile) and its corresponding device 267fb4cce0fSSimon Glass * tree file. 268fb4cce0fSSimon Glass * 269fb4cce0fSSimon Glass * Otherwise we just create a configuration with the main image in it. 2708e35bb07SSimon Glass */ 2718e35bb07SSimon Glass static void fit_write_configs(struct image_tool_params *params, char *fdt) 2728e35bb07SSimon Glass { 273fb4cce0fSSimon Glass struct content_info *cont; 2748e35bb07SSimon Glass const char *typename; 2758e35bb07SSimon Glass char str[100]; 276fb4cce0fSSimon Glass int upto; 2778e35bb07SSimon Glass 2788e35bb07SSimon Glass fdt_begin_node(fdt, "configurations"); 2798e35bb07SSimon Glass fdt_property_string(fdt, "default", "conf@1"); 2808e35bb07SSimon Glass 281fb4cce0fSSimon Glass upto = 0; 282fb4cce0fSSimon Glass for (cont = params->content_head; cont; cont = cont->next) { 283fb4cce0fSSimon Glass if (cont->type != IH_TYPE_FLATDT) 284fb4cce0fSSimon Glass continue; 285fb4cce0fSSimon Glass typename = genimg_get_type_short_name(cont->type); 286fb4cce0fSSimon Glass snprintf(str, sizeof(str), "conf@%d", ++upto); 287fb4cce0fSSimon Glass fdt_begin_node(fdt, str); 288fb4cce0fSSimon Glass 289fb4cce0fSSimon Glass get_basename(str, sizeof(str), cont->fname); 290fb4cce0fSSimon Glass fdt_property_string(fdt, "description", str); 291fb4cce0fSSimon Glass 292fb4cce0fSSimon Glass typename = genimg_get_type_short_name(params->fit_image_type); 293fb4cce0fSSimon Glass snprintf(str, sizeof(str), "%s@1", typename); 294fb4cce0fSSimon Glass fdt_property_string(fdt, typename, str); 295fb4cce0fSSimon Glass 296*0f7c6cdcSTomeu Vizoso if (params->fit_ramdisk) 297*0f7c6cdcSTomeu Vizoso fdt_property_string(fdt, FIT_RAMDISK_PROP, 298*0f7c6cdcSTomeu Vizoso FIT_RAMDISK_PROP "@1"); 299*0f7c6cdcSTomeu Vizoso 300fb4cce0fSSimon Glass snprintf(str, sizeof(str), FIT_FDT_PROP "@%d", upto); 301fb4cce0fSSimon Glass fdt_property_string(fdt, FIT_FDT_PROP, str); 302fb4cce0fSSimon Glass fdt_end_node(fdt); 303fb4cce0fSSimon Glass } 304*0f7c6cdcSTomeu Vizoso 305fb4cce0fSSimon Glass if (!upto) { 3068e35bb07SSimon Glass fdt_begin_node(fdt, "conf@1"); 3078e35bb07SSimon Glass typename = genimg_get_type_short_name(params->fit_image_type); 3088e35bb07SSimon Glass snprintf(str, sizeof(str), "%s@1", typename); 3098e35bb07SSimon Glass fdt_property_string(fdt, typename, str); 310*0f7c6cdcSTomeu Vizoso 311*0f7c6cdcSTomeu Vizoso if (params->fit_ramdisk) 312*0f7c6cdcSTomeu Vizoso fdt_property_string(fdt, FIT_RAMDISK_PROP, 313*0f7c6cdcSTomeu Vizoso FIT_RAMDISK_PROP "@1"); 314*0f7c6cdcSTomeu Vizoso 3158e35bb07SSimon Glass fdt_end_node(fdt); 316fb4cce0fSSimon Glass } 3178e35bb07SSimon Glass 3188e35bb07SSimon Glass fdt_end_node(fdt); 3198e35bb07SSimon Glass } 3208e35bb07SSimon Glass 3218e35bb07SSimon Glass static int fit_build_fdt(struct image_tool_params *params, char *fdt, int size) 3228e35bb07SSimon Glass { 3238e35bb07SSimon Glass int ret; 3248e35bb07SSimon Glass 3258e35bb07SSimon Glass ret = fdt_create(fdt, size); 3268e35bb07SSimon Glass if (ret) 3278e35bb07SSimon Glass return ret; 3288e35bb07SSimon Glass fdt_finish_reservemap(fdt); 3298e35bb07SSimon Glass fdt_begin_node(fdt, ""); 3308e35bb07SSimon Glass fdt_property_strf(fdt, "description", 3318e35bb07SSimon Glass "%s image with one or more FDT blobs", 3328e35bb07SSimon Glass genimg_get_type_name(params->fit_image_type)); 3338e35bb07SSimon Glass fdt_property_strf(fdt, "creator", "U-Boot mkimage %s", PLAIN_VERSION); 3348e35bb07SSimon Glass fdt_property_u32(fdt, "#address-cells", 1); 3358e35bb07SSimon Glass ret = fit_write_images(params, fdt); 3368e35bb07SSimon Glass if (ret) 3378e35bb07SSimon Glass return ret; 3388e35bb07SSimon Glass fit_write_configs(params, fdt); 3398e35bb07SSimon Glass fdt_end_node(fdt); 3408e35bb07SSimon Glass ret = fdt_finish(fdt); 3418e35bb07SSimon Glass if (ret) 3428e35bb07SSimon Glass return ret; 3438e35bb07SSimon Glass 3448e35bb07SSimon Glass return fdt_totalsize(fdt); 3458e35bb07SSimon Glass } 3468e35bb07SSimon Glass 3478e35bb07SSimon Glass static int fit_build(struct image_tool_params *params, const char *fname) 3488e35bb07SSimon Glass { 3498e35bb07SSimon Glass char *buf; 3508e35bb07SSimon Glass int size; 3518e35bb07SSimon Glass int ret; 3528e35bb07SSimon Glass int fd; 3538e35bb07SSimon Glass 3548e35bb07SSimon Glass size = fit_calc_size(params); 3558e35bb07SSimon Glass if (size < 0) 3568e35bb07SSimon Glass return -1; 3578e35bb07SSimon Glass buf = malloc(size); 3588e35bb07SSimon Glass if (!buf) { 3598e35bb07SSimon Glass fprintf(stderr, "%s: Out of memory (%d bytes)\n", 3608e35bb07SSimon Glass params->cmdname, size); 3618e35bb07SSimon Glass return -1; 3628e35bb07SSimon Glass } 3638e35bb07SSimon Glass ret = fit_build_fdt(params, buf, size); 3648e35bb07SSimon Glass if (ret < 0) { 3658e35bb07SSimon Glass fprintf(stderr, "%s: Failed to build FIT image\n", 3668e35bb07SSimon Glass params->cmdname); 3677b0bbd88SSimon Glass goto err_buf; 3688e35bb07SSimon Glass } 3698e35bb07SSimon Glass size = ret; 3708e35bb07SSimon Glass fd = open(fname, O_RDWR | O_CREAT | O_TRUNC | O_BINARY, 0666); 3718e35bb07SSimon Glass if (fd < 0) { 3728e35bb07SSimon Glass fprintf(stderr, "%s: Can't open %s: %s\n", 3738e35bb07SSimon Glass params->cmdname, fname, strerror(errno)); 3748e35bb07SSimon Glass goto err; 3758e35bb07SSimon Glass } 3768e35bb07SSimon Glass ret = write(fd, buf, size); 3778e35bb07SSimon Glass if (ret != size) { 3788e35bb07SSimon Glass fprintf(stderr, "%s: Can't write %s: %s\n", 3798e35bb07SSimon Glass params->cmdname, fname, strerror(errno)); 3808e35bb07SSimon Glass goto err; 3818e35bb07SSimon Glass } 3828e35bb07SSimon Glass close(fd); 3837b0bbd88SSimon Glass free(buf); 3848e35bb07SSimon Glass 3858e35bb07SSimon Glass return 0; 3868e35bb07SSimon Glass err: 3877b0bbd88SSimon Glass close(fd); 3887b0bbd88SSimon Glass err_buf: 3898e35bb07SSimon Glass free(buf); 3908e35bb07SSimon Glass return -1; 3918e35bb07SSimon Glass } 3928e35bb07SSimon Glass 3938e35bb07SSimon Glass /** 394722ebc8fSSimon Glass * fit_extract_data() - Move all data outside the FIT 395722ebc8fSSimon Glass * 396722ebc8fSSimon Glass * This takes a normal FIT file and removes all the 'data' properties from it. 397722ebc8fSSimon Glass * The data is placed in an area after the FIT so that it can be accessed 398722ebc8fSSimon Glass * using an offset into that area. The 'data' properties turn into 399722ebc8fSSimon Glass * 'data-offset' properties. 400722ebc8fSSimon Glass * 401722ebc8fSSimon Glass * This function cannot cope with FITs with 'data-offset' properties. All 402722ebc8fSSimon Glass * data must be in 'data' properties on entry. 403722ebc8fSSimon Glass */ 404722ebc8fSSimon Glass static int fit_extract_data(struct image_tool_params *params, const char *fname) 405722ebc8fSSimon Glass { 406722ebc8fSSimon Glass void *buf; 407722ebc8fSSimon Glass int buf_ptr; 408722ebc8fSSimon Glass int fit_size, new_size; 409722ebc8fSSimon Glass int fd; 410722ebc8fSSimon Glass struct stat sbuf; 411722ebc8fSSimon Glass void *fdt; 412722ebc8fSSimon Glass int ret; 413722ebc8fSSimon Glass int images; 414722ebc8fSSimon Glass int node; 415722ebc8fSSimon Glass 416722ebc8fSSimon Glass fd = mmap_fdt(params->cmdname, fname, 0, &fdt, &sbuf, false); 417722ebc8fSSimon Glass if (fd < 0) 418722ebc8fSSimon Glass return -EIO; 419722ebc8fSSimon Glass fit_size = fdt_totalsize(fdt); 420722ebc8fSSimon Glass 421722ebc8fSSimon Glass /* Allocate space to hold the image data we will extract */ 422722ebc8fSSimon Glass buf = malloc(fit_size); 423722ebc8fSSimon Glass if (!buf) { 424722ebc8fSSimon Glass ret = -ENOMEM; 425b97d71e2SSimon Glass goto err_munmap; 426722ebc8fSSimon Glass } 427722ebc8fSSimon Glass buf_ptr = 0; 428722ebc8fSSimon Glass 429722ebc8fSSimon Glass images = fdt_path_offset(fdt, FIT_IMAGES_PATH); 430722ebc8fSSimon Glass if (images < 0) { 431722ebc8fSSimon Glass debug("%s: Cannot find /images node: %d\n", __func__, images); 432722ebc8fSSimon Glass ret = -EINVAL; 433b97d71e2SSimon Glass goto err_munmap; 434722ebc8fSSimon Glass } 435722ebc8fSSimon Glass 436722ebc8fSSimon Glass for (node = fdt_first_subnode(fdt, images); 437722ebc8fSSimon Glass node >= 0; 438722ebc8fSSimon Glass node = fdt_next_subnode(fdt, node)) { 439722ebc8fSSimon Glass const char *data; 440722ebc8fSSimon Glass int len; 441722ebc8fSSimon Glass 442722ebc8fSSimon Glass data = fdt_getprop(fdt, node, "data", &len); 443722ebc8fSSimon Glass if (!data) 444722ebc8fSSimon Glass continue; 445722ebc8fSSimon Glass memcpy(buf + buf_ptr, data, len); 446722ebc8fSSimon Glass debug("Extracting data size %x\n", len); 447722ebc8fSSimon Glass 448722ebc8fSSimon Glass ret = fdt_delprop(fdt, node, "data"); 449722ebc8fSSimon Glass if (ret) { 450722ebc8fSSimon Glass ret = -EPERM; 451b97d71e2SSimon Glass goto err_munmap; 452722ebc8fSSimon Glass } 453f8f9107dSTeddy Reed if (params->external_offset > 0) { 454f8f9107dSTeddy Reed /* An external offset positions the data absolutely. */ 455f8f9107dSTeddy Reed fdt_setprop_u32(fdt, node, "data-position", 456f8f9107dSTeddy Reed params->external_offset + buf_ptr); 457f8f9107dSTeddy Reed } else { 458722ebc8fSSimon Glass fdt_setprop_u32(fdt, node, "data-offset", buf_ptr); 459f8f9107dSTeddy Reed } 460722ebc8fSSimon Glass fdt_setprop_u32(fdt, node, "data-size", len); 461722ebc8fSSimon Glass 462722ebc8fSSimon Glass buf_ptr += (len + 3) & ~3; 463722ebc8fSSimon Glass } 464722ebc8fSSimon Glass 465722ebc8fSSimon Glass /* Pack the FDT and place the data after it */ 466722ebc8fSSimon Glass fdt_pack(fdt); 467722ebc8fSSimon Glass 468722ebc8fSSimon Glass debug("Size reduced from %x to %x\n", fit_size, fdt_totalsize(fdt)); 469722ebc8fSSimon Glass debug("External data size %x\n", buf_ptr); 470722ebc8fSSimon Glass new_size = fdt_totalsize(fdt); 471722ebc8fSSimon Glass new_size = (new_size + 3) & ~3; 472722ebc8fSSimon Glass munmap(fdt, sbuf.st_size); 473722ebc8fSSimon Glass 474722ebc8fSSimon Glass if (ftruncate(fd, new_size)) { 475722ebc8fSSimon Glass debug("%s: Failed to truncate file: %s\n", __func__, 476722ebc8fSSimon Glass strerror(errno)); 477722ebc8fSSimon Glass ret = -EIO; 478722ebc8fSSimon Glass goto err; 479722ebc8fSSimon Glass } 480f8f9107dSTeddy Reed 481f8f9107dSTeddy Reed /* Check if an offset for the external data was set. */ 482f8f9107dSTeddy Reed if (params->external_offset > 0) { 483f8f9107dSTeddy Reed if (params->external_offset < new_size) { 484f8f9107dSTeddy Reed debug("External offset %x overlaps FIT length %x", 485f8f9107dSTeddy Reed params->external_offset, new_size); 486f8f9107dSTeddy Reed ret = -EINVAL; 487f8f9107dSTeddy Reed goto err; 488f8f9107dSTeddy Reed } 489f8f9107dSTeddy Reed new_size = params->external_offset; 490f8f9107dSTeddy Reed } 491722ebc8fSSimon Glass if (lseek(fd, new_size, SEEK_SET) < 0) { 492722ebc8fSSimon Glass debug("%s: Failed to seek to end of file: %s\n", __func__, 493722ebc8fSSimon Glass strerror(errno)); 494722ebc8fSSimon Glass ret = -EIO; 495722ebc8fSSimon Glass goto err; 496722ebc8fSSimon Glass } 497722ebc8fSSimon Glass if (write(fd, buf, buf_ptr) != buf_ptr) { 498722ebc8fSSimon Glass debug("%s: Failed to write external data to file %s\n", 499722ebc8fSSimon Glass __func__, strerror(errno)); 500722ebc8fSSimon Glass ret = -EIO; 501722ebc8fSSimon Glass goto err; 502722ebc8fSSimon Glass } 503b97d71e2SSimon Glass close(fd); 504b97d71e2SSimon Glass return 0; 505722ebc8fSSimon Glass 506b97d71e2SSimon Glass err_munmap: 507b97d71e2SSimon Glass munmap(fdt, sbuf.st_size); 508722ebc8fSSimon Glass err: 50921c2975aSSimon Glass if (buf) 51021c2975aSSimon Glass free(buf); 511722ebc8fSSimon Glass close(fd); 512722ebc8fSSimon Glass return ret; 513722ebc8fSSimon Glass } 514722ebc8fSSimon Glass 515529fd188SSimon Glass static int fit_import_data(struct image_tool_params *params, const char *fname) 516529fd188SSimon Glass { 517529fd188SSimon Glass void *fdt, *old_fdt; 518529fd188SSimon Glass int fit_size, new_size, size, data_base; 519529fd188SSimon Glass int fd; 520529fd188SSimon Glass struct stat sbuf; 521529fd188SSimon Glass int ret; 522529fd188SSimon Glass int images; 523529fd188SSimon Glass int node; 524529fd188SSimon Glass 525529fd188SSimon Glass fd = mmap_fdt(params->cmdname, fname, 0, &old_fdt, &sbuf, false); 526529fd188SSimon Glass if (fd < 0) 527529fd188SSimon Glass return -EIO; 528529fd188SSimon Glass fit_size = fdt_totalsize(old_fdt); 529529fd188SSimon Glass data_base = (fit_size + 3) & ~3; 530529fd188SSimon Glass 531529fd188SSimon Glass /* Allocate space to hold the new FIT */ 532529fd188SSimon Glass size = sbuf.st_size + 16384; 533529fd188SSimon Glass fdt = malloc(size); 534529fd188SSimon Glass if (!fdt) { 535529fd188SSimon Glass fprintf(stderr, "%s: Failed to allocate memory (%d bytes)\n", 536529fd188SSimon Glass __func__, size); 537529fd188SSimon Glass ret = -ENOMEM; 538529fd188SSimon Glass goto err; 539529fd188SSimon Glass } 540529fd188SSimon Glass ret = fdt_open_into(old_fdt, fdt, size); 541529fd188SSimon Glass if (ret) { 542529fd188SSimon Glass debug("%s: Failed to expand FIT: %s\n", __func__, 543529fd188SSimon Glass fdt_strerror(errno)); 544529fd188SSimon Glass ret = -EINVAL; 545529fd188SSimon Glass goto err; 546529fd188SSimon Glass } 547529fd188SSimon Glass 548529fd188SSimon Glass images = fdt_path_offset(fdt, FIT_IMAGES_PATH); 549529fd188SSimon Glass if (images < 0) { 550529fd188SSimon Glass debug("%s: Cannot find /images node: %d\n", __func__, images); 551529fd188SSimon Glass ret = -EINVAL; 552529fd188SSimon Glass goto err; 553529fd188SSimon Glass } 554529fd188SSimon Glass 555529fd188SSimon Glass for (node = fdt_first_subnode(fdt, images); 556529fd188SSimon Glass node >= 0; 557529fd188SSimon Glass node = fdt_next_subnode(fdt, node)) { 558529fd188SSimon Glass int buf_ptr; 559529fd188SSimon Glass int len; 560529fd188SSimon Glass 561529fd188SSimon Glass buf_ptr = fdtdec_get_int(fdt, node, "data-offset", -1); 562529fd188SSimon Glass len = fdtdec_get_int(fdt, node, "data-size", -1); 563529fd188SSimon Glass if (buf_ptr == -1 || len == -1) 564529fd188SSimon Glass continue; 565529fd188SSimon Glass debug("Importing data size %x\n", len); 566529fd188SSimon Glass 567529fd188SSimon Glass ret = fdt_setprop(fdt, node, "data", fdt + data_base + buf_ptr, 568529fd188SSimon Glass len); 569529fd188SSimon Glass if (ret) { 570529fd188SSimon Glass debug("%s: Failed to write property: %s\n", __func__, 571529fd188SSimon Glass fdt_strerror(ret)); 572529fd188SSimon Glass ret = -EINVAL; 573529fd188SSimon Glass goto err; 574529fd188SSimon Glass } 575529fd188SSimon Glass } 576529fd188SSimon Glass 5779cff2d1eSSimon Glass munmap(old_fdt, sbuf.st_size); 578529fd188SSimon Glass close(fd); 579529fd188SSimon Glass 580529fd188SSimon Glass /* Pack the FDT and place the data after it */ 581529fd188SSimon Glass fdt_pack(fdt); 582529fd188SSimon Glass 583529fd188SSimon Glass new_size = fdt_totalsize(fdt); 584529fd188SSimon Glass debug("Size expanded from %x to %x\n", fit_size, new_size); 585529fd188SSimon Glass 586529fd188SSimon Glass fd = open(fname, O_RDWR | O_CREAT | O_TRUNC | O_BINARY, 0666); 587529fd188SSimon Glass if (fd < 0) { 588529fd188SSimon Glass fprintf(stderr, "%s: Can't open %s: %s\n", 589529fd188SSimon Glass params->cmdname, fname, strerror(errno)); 5905b15a259SSimon Glass free(fdt); 5915b15a259SSimon Glass return -EIO; 592529fd188SSimon Glass } 593529fd188SSimon Glass if (write(fd, fdt, new_size) != new_size) { 594529fd188SSimon Glass debug("%s: Failed to write external data to file %s\n", 595529fd188SSimon Glass __func__, strerror(errno)); 596529fd188SSimon Glass ret = -EIO; 597529fd188SSimon Glass goto err; 598529fd188SSimon Glass } 599529fd188SSimon Glass 600529fd188SSimon Glass ret = 0; 601529fd188SSimon Glass 602529fd188SSimon Glass err: 6036e0ffce6SSimon Glass free(fdt); 604529fd188SSimon Glass close(fd); 605529fd188SSimon Glass return ret; 606529fd188SSimon Glass } 607529fd188SSimon Glass 608722ebc8fSSimon Glass /** 60989a4d6b1SPrafulla Wadaskar * fit_handle_file - main FIT file processing function 61089a4d6b1SPrafulla Wadaskar * 61189a4d6b1SPrafulla Wadaskar * fit_handle_file() runs dtc to convert .its to .itb, includes 61289a4d6b1SPrafulla Wadaskar * binary data, updates timestamp property and calculates hashes. 61389a4d6b1SPrafulla Wadaskar * 61489a4d6b1SPrafulla Wadaskar * datafile - .its file 61589a4d6b1SPrafulla Wadaskar * imagefile - .itb file 61689a4d6b1SPrafulla Wadaskar * 61789a4d6b1SPrafulla Wadaskar * returns: 61889a4d6b1SPrafulla Wadaskar * only on success, otherwise calls exit (EXIT_FAILURE); 61989a4d6b1SPrafulla Wadaskar */ 620f86ed6a8SGuilherme Maciel Ferreira static int fit_handle_file(struct image_tool_params *params) 62189a4d6b1SPrafulla Wadaskar { 62289a4d6b1SPrafulla Wadaskar char tmpfile[MKIMAGE_MAX_TMPFILE_LEN]; 62389a4d6b1SPrafulla Wadaskar char cmd[MKIMAGE_MAX_DTC_CMDLINE_LEN]; 624a9468115SSimon Glass size_t size_inc; 625a9468115SSimon Glass int ret; 62689a4d6b1SPrafulla Wadaskar 62789a4d6b1SPrafulla Wadaskar /* Flattened Image Tree (FIT) format handling */ 62889a4d6b1SPrafulla Wadaskar debug ("FIT format handling\n"); 62989a4d6b1SPrafulla Wadaskar 63089a4d6b1SPrafulla Wadaskar /* call dtc to include binary properties into the tmp file */ 63189a4d6b1SPrafulla Wadaskar if (strlen (params->imagefile) + 63289a4d6b1SPrafulla Wadaskar strlen (MKIMAGE_TMPFILE_SUFFIX) + 1 > sizeof (tmpfile)) { 63389a4d6b1SPrafulla Wadaskar fprintf (stderr, "%s: Image file name (%s) too long, " 63489a4d6b1SPrafulla Wadaskar "can't create tmpfile", 63589a4d6b1SPrafulla Wadaskar params->imagefile, params->cmdname); 63689a4d6b1SPrafulla Wadaskar return (EXIT_FAILURE); 63789a4d6b1SPrafulla Wadaskar } 63889a4d6b1SPrafulla Wadaskar sprintf (tmpfile, "%s%s", params->imagefile, MKIMAGE_TMPFILE_SUFFIX); 63989a4d6b1SPrafulla Wadaskar 64095d77b44SSimon Glass /* We either compile the source file, or use the existing FIT image */ 6418e35bb07SSimon Glass if (params->auto_its) { 6428e35bb07SSimon Glass if (fit_build(params, tmpfile)) { 6438e35bb07SSimon Glass fprintf(stderr, "%s: failed to build FIT\n", 6448e35bb07SSimon Glass params->cmdname); 6458e35bb07SSimon Glass return EXIT_FAILURE; 6468e35bb07SSimon Glass } 6478e35bb07SSimon Glass *cmd = '\0'; 6488e35bb07SSimon Glass } else if (params->datafile) { 649ee7c0fc7SLars Rasmusson /* dtc -I dts -O dtb -p 500 datafile > tmpfile */ 65095d77b44SSimon Glass snprintf(cmd, sizeof(cmd), "%s %s %s > %s", 65189a4d6b1SPrafulla Wadaskar MKIMAGE_DTC, params->dtc, params->datafile, tmpfile); 65289a4d6b1SPrafulla Wadaskar debug("Trying to execute \"%s\"\n", cmd); 65395d77b44SSimon Glass } else { 65495d77b44SSimon Glass snprintf(cmd, sizeof(cmd), "cp %s %s", 65595d77b44SSimon Glass params->imagefile, tmpfile); 65695d77b44SSimon Glass } 6578e35bb07SSimon Glass if (*cmd && system(cmd) == -1) { 65889a4d6b1SPrafulla Wadaskar fprintf (stderr, "%s: system(%s) failed: %s\n", 65989a4d6b1SPrafulla Wadaskar params->cmdname, cmd, strerror(errno)); 660aa6d6db4SSimon Glass goto err_system; 66189a4d6b1SPrafulla Wadaskar } 66289a4d6b1SPrafulla Wadaskar 663529fd188SSimon Glass /* Move the data so it is internal to the FIT, if needed */ 664529fd188SSimon Glass ret = fit_import_data(params, tmpfile); 665529fd188SSimon Glass if (ret) 666529fd188SSimon Glass goto err_system; 667529fd188SSimon Glass 668a9468115SSimon Glass /* 669a9468115SSimon Glass * Set hashes for images in the blob. Unfortunately we may need more 670a9468115SSimon Glass * space in either FDT, so keep trying until we succeed. 671a9468115SSimon Glass * 672a9468115SSimon Glass * Note: this is pretty inefficient for signing, since we must 673a9468115SSimon Glass * calculate the signature every time. It would be better to calculate 674a9468115SSimon Glass * all the data and then store it in a separate step. However, this 675a9468115SSimon Glass * would be considerably more complex to implement. Generally a few 676a9468115SSimon Glass * steps of this loop is enough to sign with several keys. 677a9468115SSimon Glass */ 678a9468115SSimon Glass for (size_inc = 0; size_inc < 64 * 1024; size_inc += 1024) { 679a9468115SSimon Glass ret = fit_add_file_data(params, size_inc, tmpfile); 680a9468115SSimon Glass if (!ret || ret != -ENOSPC) 681a9468115SSimon Glass break; 682e29495d3SSimon Glass } 683e29495d3SSimon Glass 684a9468115SSimon Glass if (ret) { 685655cc696SSimon Glass fprintf(stderr, "%s Can't add hashes to FIT blob: %d\n", 686655cc696SSimon Glass params->cmdname, ret); 687a9468115SSimon Glass goto err_system; 688e29495d3SSimon Glass } 68989a4d6b1SPrafulla Wadaskar 690722ebc8fSSimon Glass /* Move the data so it is external to the FIT, if requested */ 691722ebc8fSSimon Glass if (params->external_data) { 692722ebc8fSSimon Glass ret = fit_extract_data(params, tmpfile); 693722ebc8fSSimon Glass if (ret) 694722ebc8fSSimon Glass goto err_system; 695722ebc8fSSimon Glass } 696722ebc8fSSimon Glass 69789a4d6b1SPrafulla Wadaskar if (rename (tmpfile, params->imagefile) == -1) { 69889a4d6b1SPrafulla Wadaskar fprintf (stderr, "%s: Can't rename %s to %s: %s\n", 69989a4d6b1SPrafulla Wadaskar params->cmdname, tmpfile, params->imagefile, 70089a4d6b1SPrafulla Wadaskar strerror (errno)); 70189a4d6b1SPrafulla Wadaskar unlink (tmpfile); 70289a4d6b1SPrafulla Wadaskar unlink (params->imagefile); 703a9468115SSimon Glass return EXIT_FAILURE; 70489a4d6b1SPrafulla Wadaskar } 705a9468115SSimon Glass return EXIT_SUCCESS; 706aa6d6db4SSimon Glass 707aa6d6db4SSimon Glass err_system: 708aa6d6db4SSimon Glass unlink(tmpfile); 709aa6d6db4SSimon Glass return -1; 71089a4d6b1SPrafulla Wadaskar } 71189a4d6b1SPrafulla Wadaskar 71239931f96SGuilherme Maciel Ferreira /** 71339931f96SGuilherme Maciel Ferreira * fit_image_extract - extract a FIT component image 71439931f96SGuilherme Maciel Ferreira * @fit: pointer to the FIT format image header 71539931f96SGuilherme Maciel Ferreira * @image_noffset: offset of the component image node 71639931f96SGuilherme Maciel Ferreira * @file_name: name of the file to store the FIT sub-image 71739931f96SGuilherme Maciel Ferreira * 71839931f96SGuilherme Maciel Ferreira * returns: 71939931f96SGuilherme Maciel Ferreira * zero in case of success or a negative value if fail. 72039931f96SGuilherme Maciel Ferreira */ 72139931f96SGuilherme Maciel Ferreira static int fit_image_extract( 72239931f96SGuilherme Maciel Ferreira const void *fit, 72339931f96SGuilherme Maciel Ferreira int image_noffset, 72439931f96SGuilherme Maciel Ferreira const char *file_name) 72539931f96SGuilherme Maciel Ferreira { 72639931f96SGuilherme Maciel Ferreira const void *file_data; 72739931f96SGuilherme Maciel Ferreira size_t file_size = 0; 72839931f96SGuilherme Maciel Ferreira 72939931f96SGuilherme Maciel Ferreira /* get the "data" property of component at offset "image_noffset" */ 73039931f96SGuilherme Maciel Ferreira fit_image_get_data(fit, image_noffset, &file_data, &file_size); 73139931f96SGuilherme Maciel Ferreira 73239931f96SGuilherme Maciel Ferreira /* save the "file_data" into the file specified by "file_name" */ 73339931f96SGuilherme Maciel Ferreira return imagetool_save_subimage(file_name, (ulong) file_data, file_size); 73439931f96SGuilherme Maciel Ferreira } 73539931f96SGuilherme Maciel Ferreira 73639931f96SGuilherme Maciel Ferreira /** 73739931f96SGuilherme Maciel Ferreira * fit_extract_contents - retrieve a sub-image component from the FIT image 73839931f96SGuilherme Maciel Ferreira * @ptr: pointer to the FIT format image header 73939931f96SGuilherme Maciel Ferreira * @params: command line parameters 74039931f96SGuilherme Maciel Ferreira * 74139931f96SGuilherme Maciel Ferreira * returns: 74239931f96SGuilherme Maciel Ferreira * zero in case of success or a negative value if fail. 74339931f96SGuilherme Maciel Ferreira */ 74439931f96SGuilherme Maciel Ferreira static int fit_extract_contents(void *ptr, struct image_tool_params *params) 74539931f96SGuilherme Maciel Ferreira { 74639931f96SGuilherme Maciel Ferreira int images_noffset; 74739931f96SGuilherme Maciel Ferreira int noffset; 74839931f96SGuilherme Maciel Ferreira int ndepth; 74939931f96SGuilherme Maciel Ferreira const void *fit = ptr; 75039931f96SGuilherme Maciel Ferreira int count = 0; 75139931f96SGuilherme Maciel Ferreira const char *p; 75239931f96SGuilherme Maciel Ferreira 75339931f96SGuilherme Maciel Ferreira /* Indent string is defined in header image.h */ 75439931f96SGuilherme Maciel Ferreira p = IMAGE_INDENT_STRING; 75539931f96SGuilherme Maciel Ferreira 75639931f96SGuilherme Maciel Ferreira if (!fit_check_format(fit)) { 75739931f96SGuilherme Maciel Ferreira printf("Bad FIT image format\n"); 75839931f96SGuilherme Maciel Ferreira return -1; 75939931f96SGuilherme Maciel Ferreira } 76039931f96SGuilherme Maciel Ferreira 76139931f96SGuilherme Maciel Ferreira /* Find images parent node offset */ 76239931f96SGuilherme Maciel Ferreira images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH); 76339931f96SGuilherme Maciel Ferreira if (images_noffset < 0) { 76439931f96SGuilherme Maciel Ferreira printf("Can't find images parent node '%s' (%s)\n", 76539931f96SGuilherme Maciel Ferreira FIT_IMAGES_PATH, fdt_strerror(images_noffset)); 76639931f96SGuilherme Maciel Ferreira return -1; 76739931f96SGuilherme Maciel Ferreira } 76839931f96SGuilherme Maciel Ferreira 76939931f96SGuilherme Maciel Ferreira /* Avoid any overrun */ 77039931f96SGuilherme Maciel Ferreira count = fit_get_subimage_count(fit, images_noffset); 77139931f96SGuilherme Maciel Ferreira if ((params->pflag < 0) || (count <= params->pflag)) { 77239931f96SGuilherme Maciel Ferreira printf("No such component at '%d'\n", params->pflag); 77339931f96SGuilherme Maciel Ferreira return -1; 77439931f96SGuilherme Maciel Ferreira } 77539931f96SGuilherme Maciel Ferreira 77639931f96SGuilherme Maciel Ferreira /* Process its subnodes, extract the desired component from image */ 77739931f96SGuilherme Maciel Ferreira for (ndepth = 0, count = 0, 77839931f96SGuilherme Maciel Ferreira noffset = fdt_next_node(fit, images_noffset, &ndepth); 77939931f96SGuilherme Maciel Ferreira (noffset >= 0) && (ndepth > 0); 78039931f96SGuilherme Maciel Ferreira noffset = fdt_next_node(fit, noffset, &ndepth)) { 78139931f96SGuilherme Maciel Ferreira if (ndepth == 1) { 78239931f96SGuilherme Maciel Ferreira /* 78339931f96SGuilherme Maciel Ferreira * Direct child node of the images parent node, 78439931f96SGuilherme Maciel Ferreira * i.e. component image node. 78539931f96SGuilherme Maciel Ferreira */ 78639931f96SGuilherme Maciel Ferreira if (params->pflag == count) { 78739931f96SGuilherme Maciel Ferreira printf("Extracted:\n%s Image %u (%s)\n", p, 78839931f96SGuilherme Maciel Ferreira count, fit_get_name(fit, noffset, NULL)); 78939931f96SGuilherme Maciel Ferreira 79039931f96SGuilherme Maciel Ferreira fit_image_print(fit, noffset, p); 79139931f96SGuilherme Maciel Ferreira 79239931f96SGuilherme Maciel Ferreira return fit_image_extract(fit, noffset, 79339931f96SGuilherme Maciel Ferreira params->outfile); 79439931f96SGuilherme Maciel Ferreira } 79539931f96SGuilherme Maciel Ferreira 79639931f96SGuilherme Maciel Ferreira count++; 79739931f96SGuilherme Maciel Ferreira } 79839931f96SGuilherme Maciel Ferreira } 79939931f96SGuilherme Maciel Ferreira 80039931f96SGuilherme Maciel Ferreira return 0; 80139931f96SGuilherme Maciel Ferreira } 80239931f96SGuilherme Maciel Ferreira 803f86ed6a8SGuilherme Maciel Ferreira static int fit_check_params(struct image_tool_params *params) 80489a4d6b1SPrafulla Wadaskar { 8058e35bb07SSimon Glass if (params->auto_its) 8068e35bb07SSimon Glass return 0; 80789a4d6b1SPrafulla Wadaskar return ((params->dflag && (params->fflag || params->lflag)) || 80889a4d6b1SPrafulla Wadaskar (params->fflag && (params->dflag || params->lflag)) || 80989a4d6b1SPrafulla Wadaskar (params->lflag && (params->dflag || params->fflag))); 81089a4d6b1SPrafulla Wadaskar } 81189a4d6b1SPrafulla Wadaskar 812a93648d1SGuilherme Maciel Ferreira U_BOOT_IMAGE_TYPE( 813a93648d1SGuilherme Maciel Ferreira fitimage, 814a93648d1SGuilherme Maciel Ferreira "FIT Image support", 815a93648d1SGuilherme Maciel Ferreira sizeof(image_header_t), 816a93648d1SGuilherme Maciel Ferreira (void *)&header, 817a93648d1SGuilherme Maciel Ferreira fit_check_params, 818a93648d1SGuilherme Maciel Ferreira fit_verify_header, 819a93648d1SGuilherme Maciel Ferreira fit_print_contents, 820a93648d1SGuilherme Maciel Ferreira NULL, 82139931f96SGuilherme Maciel Ferreira fit_extract_contents, 822a93648d1SGuilherme Maciel Ferreira fit_check_image_types, 823a93648d1SGuilherme Maciel Ferreira fit_handle_file, 824a93648d1SGuilherme Maciel Ferreira NULL /* FIT images use DTB header */ 825a93648d1SGuilherme Maciel Ferreira ); 826