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 */ 54a9468115SSimon Glass if (params->datafile) 55a9468115SSimon Glass ret = fit_set_timestamp(ptr, 0, sbuf.st_mtime); 56a9468115SSimon Glass 57a9468115SSimon Glass if (!ret) { 58a9468115SSimon Glass ret = fit_add_verification_data(params->keydir, dest_blob, ptr, 59a9468115SSimon Glass params->comment, 60a9468115SSimon Glass params->require_keys); 61a9468115SSimon Glass } 62a9468115SSimon Glass 63a9468115SSimon Glass if (dest_blob) { 64a9468115SSimon Glass munmap(dest_blob, destfd_size); 65a9468115SSimon Glass close(destfd); 66a9468115SSimon Glass } 67a9468115SSimon Glass 68a9468115SSimon Glass err_keydest: 69a9468115SSimon Glass munmap(ptr, sbuf.st_size); 70a9468115SSimon Glass close(tfd); 71a9468115SSimon Glass 72a9468115SSimon Glass return ret; 73a9468115SSimon Glass } 74a9468115SSimon Glass 7589a4d6b1SPrafulla Wadaskar /** 768e35bb07SSimon Glass * fit_calc_size() - Calculate the approximate size of the FIT we will generate 778e35bb07SSimon Glass */ 788e35bb07SSimon Glass static int fit_calc_size(struct image_tool_params *params) 798e35bb07SSimon Glass { 80fb4cce0fSSimon Glass struct content_info *cont; 818e35bb07SSimon Glass int size, total_size; 828e35bb07SSimon Glass 838e35bb07SSimon Glass size = imagetool_get_filesize(params, params->datafile); 848e35bb07SSimon Glass if (size < 0) 858e35bb07SSimon Glass return -1; 868e35bb07SSimon Glass 878e35bb07SSimon Glass total_size = size; 88fb4cce0fSSimon Glass for (cont = params->content_head; cont; cont = cont->next) { 89fb4cce0fSSimon Glass size = imagetool_get_filesize(params, cont->fname); 90fb4cce0fSSimon Glass if (size < 0) 91fb4cce0fSSimon Glass return -1; 928e35bb07SSimon Glass 93fb4cce0fSSimon Glass /* Add space for properties */ 94fb4cce0fSSimon Glass total_size += size + 300; 95fb4cce0fSSimon Glass } 968e35bb07SSimon Glass 978e35bb07SSimon Glass /* Add plenty of space for headers, properties, nodes, etc. */ 988e35bb07SSimon Glass total_size += 4096; 998e35bb07SSimon Glass 1008e35bb07SSimon Glass return total_size; 1018e35bb07SSimon Glass } 1028e35bb07SSimon Glass 1038e35bb07SSimon Glass static int fdt_property_file(struct image_tool_params *params, 1048e35bb07SSimon Glass void *fdt, const char *name, const char *fname) 1058e35bb07SSimon Glass { 1068e35bb07SSimon Glass struct stat sbuf; 1078e35bb07SSimon Glass void *ptr; 1088e35bb07SSimon Glass int ret; 1098e35bb07SSimon Glass int fd; 1108e35bb07SSimon Glass 1118e35bb07SSimon Glass fd = open(fname, O_RDWR | O_BINARY); 1128e35bb07SSimon Glass if (fd < 0) { 1138e35bb07SSimon Glass fprintf(stderr, "%s: Can't open %s: %s\n", 1148e35bb07SSimon Glass params->cmdname, fname, strerror(errno)); 1158e35bb07SSimon Glass return -1; 1168e35bb07SSimon Glass } 1178e35bb07SSimon Glass 1188e35bb07SSimon Glass if (fstat(fd, &sbuf) < 0) { 1198e35bb07SSimon Glass fprintf(stderr, "%s: Can't stat %s: %s\n", 1208e35bb07SSimon Glass params->cmdname, fname, strerror(errno)); 1218e35bb07SSimon Glass goto err; 1228e35bb07SSimon Glass } 1238e35bb07SSimon Glass 1248e35bb07SSimon Glass ret = fdt_property_placeholder(fdt, "data", sbuf.st_size, &ptr); 1258e35bb07SSimon Glass if (ret) 1268e35bb07SSimon Glass return ret; 1278e35bb07SSimon Glass ret = read(fd, ptr, sbuf.st_size); 1288e35bb07SSimon Glass if (ret != sbuf.st_size) { 1298e35bb07SSimon Glass fprintf(stderr, "%s: Can't read %s: %s\n", 1308e35bb07SSimon Glass params->cmdname, fname, strerror(errno)); 1318e35bb07SSimon Glass goto err; 1328e35bb07SSimon Glass } 1338e35bb07SSimon Glass 1348e35bb07SSimon Glass return 0; 1358e35bb07SSimon Glass err: 1368e35bb07SSimon Glass close(fd); 1378e35bb07SSimon Glass return -1; 1388e35bb07SSimon Glass } 1398e35bb07SSimon Glass 1408e35bb07SSimon Glass static int fdt_property_strf(void *fdt, const char *name, const char *fmt, ...) 1418e35bb07SSimon Glass { 1428e35bb07SSimon Glass char str[100]; 1438e35bb07SSimon Glass va_list ptr; 1448e35bb07SSimon Glass 1458e35bb07SSimon Glass va_start(ptr, fmt); 1468e35bb07SSimon Glass vsnprintf(str, sizeof(str), fmt, ptr); 1478e35bb07SSimon Glass va_end(ptr); 1488e35bb07SSimon Glass return fdt_property_string(fdt, name, str); 1498e35bb07SSimon Glass } 1508e35bb07SSimon Glass 151fb4cce0fSSimon Glass static void get_basename(char *str, int size, const char *fname) 152fb4cce0fSSimon Glass { 153fb4cce0fSSimon Glass const char *p, *start, *end; 154fb4cce0fSSimon Glass int len; 155fb4cce0fSSimon Glass 156fb4cce0fSSimon Glass /* 157fb4cce0fSSimon Glass * Use the base name as the 'name' field. So for example: 158fb4cce0fSSimon Glass * 159fb4cce0fSSimon Glass * "arch/arm/dts/sun7i-a20-bananapro.dtb" 160fb4cce0fSSimon Glass * becomes "sun7i-a20-bananapro" 161fb4cce0fSSimon Glass */ 162fb4cce0fSSimon Glass p = strrchr(fname, '/'); 163fb4cce0fSSimon Glass start = p ? p + 1 : fname; 164fb4cce0fSSimon Glass p = strrchr(fname, '.'); 165fb4cce0fSSimon Glass end = p ? p : fname + strlen(fname); 166fb4cce0fSSimon Glass len = end - start; 167fb4cce0fSSimon Glass if (len >= size) 168fb4cce0fSSimon Glass len = size - 1; 169fb4cce0fSSimon Glass memcpy(str, start, len); 170fb4cce0fSSimon Glass str[len] = '\0'; 171fb4cce0fSSimon Glass } 172fb4cce0fSSimon Glass 1738e35bb07SSimon Glass /** 1748e35bb07SSimon Glass * fit_write_images() - Write out a list of images to the FIT 1758e35bb07SSimon Glass * 176fb4cce0fSSimon Glass * We always include the main image (params->datafile). If there are device 177fb4cce0fSSimon Glass * tree files, we include an fdt@ node for each of those too. 1788e35bb07SSimon Glass */ 1798e35bb07SSimon Glass static int fit_write_images(struct image_tool_params *params, char *fdt) 1808e35bb07SSimon Glass { 181fb4cce0fSSimon Glass struct content_info *cont; 1828e35bb07SSimon Glass const char *typename; 1838e35bb07SSimon Glass char str[100]; 184fb4cce0fSSimon Glass int upto; 1858e35bb07SSimon Glass int ret; 1868e35bb07SSimon Glass 1878e35bb07SSimon Glass fdt_begin_node(fdt, "images"); 1888e35bb07SSimon Glass 1898e35bb07SSimon Glass /* First the main image */ 1908e35bb07SSimon Glass typename = genimg_get_type_short_name(params->fit_image_type); 1918e35bb07SSimon Glass snprintf(str, sizeof(str), "%s@1", typename); 1928e35bb07SSimon Glass fdt_begin_node(fdt, str); 1938e35bb07SSimon Glass fdt_property_string(fdt, "description", params->imagename); 1948e35bb07SSimon Glass fdt_property_string(fdt, "type", typename); 1958e35bb07SSimon Glass fdt_property_string(fdt, "arch", genimg_get_arch_name(params->arch)); 1968e35bb07SSimon Glass fdt_property_string(fdt, "os", genimg_get_os_short_name(params->os)); 1978e35bb07SSimon Glass fdt_property_string(fdt, "compression", 1988e35bb07SSimon Glass genimg_get_comp_short_name(params->comp)); 1998e35bb07SSimon Glass fdt_property_u32(fdt, "load", params->addr); 2008e35bb07SSimon Glass fdt_property_u32(fdt, "entry", params->ep); 2018e35bb07SSimon Glass 2028e35bb07SSimon Glass /* 2038e35bb07SSimon Glass * Put data last since it is large. SPL may only load the first part 2048e35bb07SSimon Glass * of the DT, so this way it can access all the above fields. 2058e35bb07SSimon Glass */ 2068e35bb07SSimon Glass ret = fdt_property_file(params, fdt, "data", params->datafile); 2078e35bb07SSimon Glass if (ret) 2088e35bb07SSimon Glass return ret; 2098e35bb07SSimon Glass fdt_end_node(fdt); 2108e35bb07SSimon Glass 211fb4cce0fSSimon Glass /* Now the device tree files if available */ 212fb4cce0fSSimon Glass upto = 0; 213fb4cce0fSSimon Glass for (cont = params->content_head; cont; cont = cont->next) { 214fb4cce0fSSimon Glass if (cont->type != IH_TYPE_FLATDT) 215fb4cce0fSSimon Glass continue; 216fb4cce0fSSimon Glass snprintf(str, sizeof(str), "%s@%d", FIT_FDT_PROP, ++upto); 217fb4cce0fSSimon Glass fdt_begin_node(fdt, str); 218fb4cce0fSSimon Glass 219fb4cce0fSSimon Glass get_basename(str, sizeof(str), cont->fname); 220fb4cce0fSSimon Glass fdt_property_string(fdt, "description", str); 221fb4cce0fSSimon Glass ret = fdt_property_file(params, fdt, "data", cont->fname); 222fb4cce0fSSimon Glass if (ret) 223fb4cce0fSSimon Glass return ret; 224fb4cce0fSSimon Glass fdt_property_string(fdt, "type", typename); 225fb4cce0fSSimon Glass fdt_property_string(fdt, "arch", 226fb4cce0fSSimon Glass genimg_get_arch_short_name(params->arch)); 227fb4cce0fSSimon Glass fdt_property_string(fdt, "compression", 228fb4cce0fSSimon Glass genimg_get_comp_short_name(IH_COMP_NONE)); 229fb4cce0fSSimon Glass fdt_end_node(fdt); 230fb4cce0fSSimon Glass } 231fb4cce0fSSimon Glass 2328e35bb07SSimon Glass fdt_end_node(fdt); 2338e35bb07SSimon Glass 2348e35bb07SSimon Glass return 0; 2358e35bb07SSimon Glass } 2368e35bb07SSimon Glass 2378e35bb07SSimon Glass /** 2388e35bb07SSimon Glass * fit_write_configs() - Write out a list of configurations to the FIT 2398e35bb07SSimon Glass * 240fb4cce0fSSimon Glass * If there are device tree files, we include a configuration for each, which 241fb4cce0fSSimon Glass * selects the main image (params->datafile) and its corresponding device 242fb4cce0fSSimon Glass * tree file. 243fb4cce0fSSimon Glass * 244fb4cce0fSSimon Glass * Otherwise we just create a configuration with the main image in it. 2458e35bb07SSimon Glass */ 2468e35bb07SSimon Glass static void fit_write_configs(struct image_tool_params *params, char *fdt) 2478e35bb07SSimon Glass { 248fb4cce0fSSimon Glass struct content_info *cont; 2498e35bb07SSimon Glass const char *typename; 2508e35bb07SSimon Glass char str[100]; 251fb4cce0fSSimon Glass int upto; 2528e35bb07SSimon Glass 2538e35bb07SSimon Glass fdt_begin_node(fdt, "configurations"); 2548e35bb07SSimon Glass fdt_property_string(fdt, "default", "conf@1"); 2558e35bb07SSimon Glass 256fb4cce0fSSimon Glass upto = 0; 257fb4cce0fSSimon Glass for (cont = params->content_head; cont; cont = cont->next) { 258fb4cce0fSSimon Glass if (cont->type != IH_TYPE_FLATDT) 259fb4cce0fSSimon Glass continue; 260fb4cce0fSSimon Glass typename = genimg_get_type_short_name(cont->type); 261fb4cce0fSSimon Glass snprintf(str, sizeof(str), "conf@%d", ++upto); 262fb4cce0fSSimon Glass fdt_begin_node(fdt, str); 263fb4cce0fSSimon Glass 264fb4cce0fSSimon Glass get_basename(str, sizeof(str), cont->fname); 265fb4cce0fSSimon Glass fdt_property_string(fdt, "description", str); 266fb4cce0fSSimon Glass 267fb4cce0fSSimon Glass typename = genimg_get_type_short_name(params->fit_image_type); 268fb4cce0fSSimon Glass snprintf(str, sizeof(str), "%s@1", typename); 269fb4cce0fSSimon Glass fdt_property_string(fdt, typename, str); 270fb4cce0fSSimon Glass 271fb4cce0fSSimon Glass snprintf(str, sizeof(str), FIT_FDT_PROP "@%d", upto); 272fb4cce0fSSimon Glass fdt_property_string(fdt, FIT_FDT_PROP, str); 273fb4cce0fSSimon Glass fdt_end_node(fdt); 274fb4cce0fSSimon Glass } 275fb4cce0fSSimon Glass if (!upto) { 2768e35bb07SSimon Glass fdt_begin_node(fdt, "conf@1"); 2778e35bb07SSimon Glass typename = genimg_get_type_short_name(params->fit_image_type); 2788e35bb07SSimon Glass snprintf(str, sizeof(str), "%s@1", typename); 2798e35bb07SSimon Glass fdt_property_string(fdt, typename, str); 2808e35bb07SSimon Glass fdt_end_node(fdt); 281fb4cce0fSSimon Glass } 2828e35bb07SSimon Glass 2838e35bb07SSimon Glass fdt_end_node(fdt); 2848e35bb07SSimon Glass } 2858e35bb07SSimon Glass 2868e35bb07SSimon Glass static int fit_build_fdt(struct image_tool_params *params, char *fdt, int size) 2878e35bb07SSimon Glass { 2888e35bb07SSimon Glass int ret; 2898e35bb07SSimon Glass 2908e35bb07SSimon Glass ret = fdt_create(fdt, size); 2918e35bb07SSimon Glass if (ret) 2928e35bb07SSimon Glass return ret; 2938e35bb07SSimon Glass fdt_finish_reservemap(fdt); 2948e35bb07SSimon Glass fdt_begin_node(fdt, ""); 2958e35bb07SSimon Glass fdt_property_strf(fdt, "description", 2968e35bb07SSimon Glass "%s image with one or more FDT blobs", 2978e35bb07SSimon Glass genimg_get_type_name(params->fit_image_type)); 2988e35bb07SSimon Glass fdt_property_strf(fdt, "creator", "U-Boot mkimage %s", PLAIN_VERSION); 2998e35bb07SSimon Glass fdt_property_u32(fdt, "#address-cells", 1); 3008e35bb07SSimon Glass ret = fit_write_images(params, fdt); 3018e35bb07SSimon Glass if (ret) 3028e35bb07SSimon Glass return ret; 3038e35bb07SSimon Glass fit_write_configs(params, fdt); 3048e35bb07SSimon Glass fdt_end_node(fdt); 3058e35bb07SSimon Glass ret = fdt_finish(fdt); 3068e35bb07SSimon Glass if (ret) 3078e35bb07SSimon Glass return ret; 3088e35bb07SSimon Glass 3098e35bb07SSimon Glass return fdt_totalsize(fdt); 3108e35bb07SSimon Glass } 3118e35bb07SSimon Glass 3128e35bb07SSimon Glass static int fit_build(struct image_tool_params *params, const char *fname) 3138e35bb07SSimon Glass { 3148e35bb07SSimon Glass char *buf; 3158e35bb07SSimon Glass int size; 3168e35bb07SSimon Glass int ret; 3178e35bb07SSimon Glass int fd; 3188e35bb07SSimon Glass 3198e35bb07SSimon Glass size = fit_calc_size(params); 3208e35bb07SSimon Glass if (size < 0) 3218e35bb07SSimon Glass return -1; 3228e35bb07SSimon Glass buf = malloc(size); 3238e35bb07SSimon Glass if (!buf) { 3248e35bb07SSimon Glass fprintf(stderr, "%s: Out of memory (%d bytes)\n", 3258e35bb07SSimon Glass params->cmdname, size); 3268e35bb07SSimon Glass return -1; 3278e35bb07SSimon Glass } 3288e35bb07SSimon Glass ret = fit_build_fdt(params, buf, size); 3298e35bb07SSimon Glass if (ret < 0) { 3308e35bb07SSimon Glass fprintf(stderr, "%s: Failed to build FIT image\n", 3318e35bb07SSimon Glass params->cmdname); 3328e35bb07SSimon Glass goto err; 3338e35bb07SSimon Glass } 3348e35bb07SSimon Glass size = ret; 3358e35bb07SSimon Glass fd = open(fname, O_RDWR | O_CREAT | O_TRUNC | O_BINARY, 0666); 3368e35bb07SSimon Glass if (fd < 0) { 3378e35bb07SSimon Glass fprintf(stderr, "%s: Can't open %s: %s\n", 3388e35bb07SSimon Glass params->cmdname, fname, strerror(errno)); 3398e35bb07SSimon Glass goto err; 3408e35bb07SSimon Glass } 3418e35bb07SSimon Glass ret = write(fd, buf, size); 3428e35bb07SSimon Glass if (ret != size) { 3438e35bb07SSimon Glass fprintf(stderr, "%s: Can't write %s: %s\n", 3448e35bb07SSimon Glass params->cmdname, fname, strerror(errno)); 3458e35bb07SSimon Glass close(fd); 3468e35bb07SSimon Glass goto err; 3478e35bb07SSimon Glass } 3488e35bb07SSimon Glass close(fd); 3498e35bb07SSimon Glass 3508e35bb07SSimon Glass return 0; 3518e35bb07SSimon Glass err: 3528e35bb07SSimon Glass free(buf); 3538e35bb07SSimon Glass return -1; 3548e35bb07SSimon Glass } 3558e35bb07SSimon Glass 3568e35bb07SSimon Glass /** 357722ebc8fSSimon Glass * fit_extract_data() - Move all data outside the FIT 358722ebc8fSSimon Glass * 359722ebc8fSSimon Glass * This takes a normal FIT file and removes all the 'data' properties from it. 360722ebc8fSSimon Glass * The data is placed in an area after the FIT so that it can be accessed 361722ebc8fSSimon Glass * using an offset into that area. The 'data' properties turn into 362722ebc8fSSimon Glass * 'data-offset' properties. 363722ebc8fSSimon Glass * 364722ebc8fSSimon Glass * This function cannot cope with FITs with 'data-offset' properties. All 365722ebc8fSSimon Glass * data must be in 'data' properties on entry. 366722ebc8fSSimon Glass */ 367722ebc8fSSimon Glass static int fit_extract_data(struct image_tool_params *params, const char *fname) 368722ebc8fSSimon Glass { 369722ebc8fSSimon Glass void *buf; 370722ebc8fSSimon Glass int buf_ptr; 371722ebc8fSSimon Glass int fit_size, new_size; 372722ebc8fSSimon Glass int fd; 373722ebc8fSSimon Glass struct stat sbuf; 374722ebc8fSSimon Glass void *fdt; 375722ebc8fSSimon Glass int ret; 376722ebc8fSSimon Glass int images; 377722ebc8fSSimon Glass int node; 378722ebc8fSSimon Glass 379722ebc8fSSimon Glass fd = mmap_fdt(params->cmdname, fname, 0, &fdt, &sbuf, false); 380722ebc8fSSimon Glass if (fd < 0) 381722ebc8fSSimon Glass return -EIO; 382722ebc8fSSimon Glass fit_size = fdt_totalsize(fdt); 383722ebc8fSSimon Glass 384722ebc8fSSimon Glass /* Allocate space to hold the image data we will extract */ 385722ebc8fSSimon Glass buf = malloc(fit_size); 386722ebc8fSSimon Glass if (!buf) { 387722ebc8fSSimon Glass ret = -ENOMEM; 388*b97d71e2SSimon Glass goto err_munmap; 389722ebc8fSSimon Glass } 390722ebc8fSSimon Glass buf_ptr = 0; 391722ebc8fSSimon Glass 392722ebc8fSSimon Glass images = fdt_path_offset(fdt, FIT_IMAGES_PATH); 393722ebc8fSSimon Glass if (images < 0) { 394722ebc8fSSimon Glass debug("%s: Cannot find /images node: %d\n", __func__, images); 395722ebc8fSSimon Glass ret = -EINVAL; 396*b97d71e2SSimon Glass goto err_munmap; 397722ebc8fSSimon Glass } 398722ebc8fSSimon Glass 399722ebc8fSSimon Glass for (node = fdt_first_subnode(fdt, images); 400722ebc8fSSimon Glass node >= 0; 401722ebc8fSSimon Glass node = fdt_next_subnode(fdt, node)) { 402722ebc8fSSimon Glass const char *data; 403722ebc8fSSimon Glass int len; 404722ebc8fSSimon Glass 405722ebc8fSSimon Glass data = fdt_getprop(fdt, node, "data", &len); 406722ebc8fSSimon Glass if (!data) 407722ebc8fSSimon Glass continue; 408722ebc8fSSimon Glass memcpy(buf + buf_ptr, data, len); 409722ebc8fSSimon Glass debug("Extracting data size %x\n", len); 410722ebc8fSSimon Glass 411722ebc8fSSimon Glass ret = fdt_delprop(fdt, node, "data"); 412722ebc8fSSimon Glass if (ret) { 413722ebc8fSSimon Glass ret = -EPERM; 414*b97d71e2SSimon Glass goto err_munmap; 415722ebc8fSSimon Glass } 416722ebc8fSSimon Glass fdt_setprop_u32(fdt, node, "data-offset", buf_ptr); 417722ebc8fSSimon Glass fdt_setprop_u32(fdt, node, "data-size", len); 418722ebc8fSSimon Glass 419722ebc8fSSimon Glass buf_ptr += (len + 3) & ~3; 420722ebc8fSSimon Glass } 421722ebc8fSSimon Glass 422722ebc8fSSimon Glass /* Pack the FDT and place the data after it */ 423722ebc8fSSimon Glass fdt_pack(fdt); 424722ebc8fSSimon Glass 425722ebc8fSSimon Glass debug("Size reduced from %x to %x\n", fit_size, fdt_totalsize(fdt)); 426722ebc8fSSimon Glass debug("External data size %x\n", buf_ptr); 427722ebc8fSSimon Glass new_size = fdt_totalsize(fdt); 428722ebc8fSSimon Glass new_size = (new_size + 3) & ~3; 429722ebc8fSSimon Glass munmap(fdt, sbuf.st_size); 430722ebc8fSSimon Glass 431722ebc8fSSimon Glass if (ftruncate(fd, new_size)) { 432722ebc8fSSimon Glass debug("%s: Failed to truncate file: %s\n", __func__, 433722ebc8fSSimon Glass strerror(errno)); 434722ebc8fSSimon Glass ret = -EIO; 435722ebc8fSSimon Glass goto err; 436722ebc8fSSimon Glass } 437722ebc8fSSimon Glass if (lseek(fd, new_size, SEEK_SET) < 0) { 438722ebc8fSSimon Glass debug("%s: Failed to seek to end of file: %s\n", __func__, 439722ebc8fSSimon Glass strerror(errno)); 440722ebc8fSSimon Glass ret = -EIO; 441722ebc8fSSimon Glass goto err; 442722ebc8fSSimon Glass } 443722ebc8fSSimon Glass if (write(fd, buf, buf_ptr) != buf_ptr) { 444722ebc8fSSimon Glass debug("%s: Failed to write external data to file %s\n", 445722ebc8fSSimon Glass __func__, strerror(errno)); 446722ebc8fSSimon Glass ret = -EIO; 447722ebc8fSSimon Glass goto err; 448722ebc8fSSimon Glass } 449*b97d71e2SSimon Glass close(fd); 450*b97d71e2SSimon Glass return 0; 451722ebc8fSSimon Glass 452*b97d71e2SSimon Glass err_munmap: 453*b97d71e2SSimon Glass munmap(fdt, sbuf.st_size); 454722ebc8fSSimon Glass err: 455722ebc8fSSimon Glass close(fd); 456722ebc8fSSimon Glass return ret; 457722ebc8fSSimon Glass } 458722ebc8fSSimon Glass 459529fd188SSimon Glass static int fit_import_data(struct image_tool_params *params, const char *fname) 460529fd188SSimon Glass { 461529fd188SSimon Glass void *fdt, *old_fdt; 462529fd188SSimon Glass int fit_size, new_size, size, data_base; 463529fd188SSimon Glass int fd; 464529fd188SSimon Glass struct stat sbuf; 465529fd188SSimon Glass int ret; 466529fd188SSimon Glass int images; 467529fd188SSimon Glass int node; 468529fd188SSimon Glass 469529fd188SSimon Glass fd = mmap_fdt(params->cmdname, fname, 0, &old_fdt, &sbuf, false); 470529fd188SSimon Glass if (fd < 0) 471529fd188SSimon Glass return -EIO; 472529fd188SSimon Glass fit_size = fdt_totalsize(old_fdt); 473529fd188SSimon Glass data_base = (fit_size + 3) & ~3; 474529fd188SSimon Glass 475529fd188SSimon Glass /* Allocate space to hold the new FIT */ 476529fd188SSimon Glass size = sbuf.st_size + 16384; 477529fd188SSimon Glass fdt = malloc(size); 478529fd188SSimon Glass if (!fdt) { 479529fd188SSimon Glass fprintf(stderr, "%s: Failed to allocate memory (%d bytes)\n", 480529fd188SSimon Glass __func__, size); 481529fd188SSimon Glass ret = -ENOMEM; 482529fd188SSimon Glass goto err; 483529fd188SSimon Glass } 484529fd188SSimon Glass ret = fdt_open_into(old_fdt, fdt, size); 485529fd188SSimon Glass if (ret) { 486529fd188SSimon Glass debug("%s: Failed to expand FIT: %s\n", __func__, 487529fd188SSimon Glass fdt_strerror(errno)); 488529fd188SSimon Glass ret = -EINVAL; 489529fd188SSimon Glass goto err; 490529fd188SSimon Glass } 491529fd188SSimon Glass 492529fd188SSimon Glass images = fdt_path_offset(fdt, FIT_IMAGES_PATH); 493529fd188SSimon Glass if (images < 0) { 494529fd188SSimon Glass debug("%s: Cannot find /images node: %d\n", __func__, images); 495529fd188SSimon Glass ret = -EINVAL; 496529fd188SSimon Glass goto err; 497529fd188SSimon Glass } 498529fd188SSimon Glass 499529fd188SSimon Glass for (node = fdt_first_subnode(fdt, images); 500529fd188SSimon Glass node >= 0; 501529fd188SSimon Glass node = fdt_next_subnode(fdt, node)) { 502529fd188SSimon Glass int buf_ptr; 503529fd188SSimon Glass int len; 504529fd188SSimon Glass 505529fd188SSimon Glass buf_ptr = fdtdec_get_int(fdt, node, "data-offset", -1); 506529fd188SSimon Glass len = fdtdec_get_int(fdt, node, "data-size", -1); 507529fd188SSimon Glass if (buf_ptr == -1 || len == -1) 508529fd188SSimon Glass continue; 509529fd188SSimon Glass debug("Importing data size %x\n", len); 510529fd188SSimon Glass 511529fd188SSimon Glass ret = fdt_setprop(fdt, node, "data", fdt + data_base + buf_ptr, 512529fd188SSimon Glass len); 513529fd188SSimon Glass if (ret) { 514529fd188SSimon Glass debug("%s: Failed to write property: %s\n", __func__, 515529fd188SSimon Glass fdt_strerror(ret)); 516529fd188SSimon Glass ret = -EINVAL; 517529fd188SSimon Glass goto err; 518529fd188SSimon Glass } 519529fd188SSimon Glass } 520529fd188SSimon Glass 5219cff2d1eSSimon Glass munmap(old_fdt, sbuf.st_size); 522529fd188SSimon Glass close(fd); 523529fd188SSimon Glass 524529fd188SSimon Glass /* Pack the FDT and place the data after it */ 525529fd188SSimon Glass fdt_pack(fdt); 526529fd188SSimon Glass 527529fd188SSimon Glass new_size = fdt_totalsize(fdt); 528529fd188SSimon Glass debug("Size expanded from %x to %x\n", fit_size, new_size); 529529fd188SSimon Glass 530529fd188SSimon Glass fd = open(fname, O_RDWR | O_CREAT | O_TRUNC | O_BINARY, 0666); 531529fd188SSimon Glass if (fd < 0) { 532529fd188SSimon Glass fprintf(stderr, "%s: Can't open %s: %s\n", 533529fd188SSimon Glass params->cmdname, fname, strerror(errno)); 534508b028aSSimon Glass ret = -EIO; 535529fd188SSimon Glass goto err; 536529fd188SSimon Glass } 537529fd188SSimon Glass if (write(fd, fdt, new_size) != new_size) { 538529fd188SSimon Glass debug("%s: Failed to write external data to file %s\n", 539529fd188SSimon Glass __func__, strerror(errno)); 540529fd188SSimon Glass ret = -EIO; 541529fd188SSimon Glass goto err; 542529fd188SSimon Glass } 543529fd188SSimon Glass 544529fd188SSimon Glass ret = 0; 545529fd188SSimon Glass 546529fd188SSimon Glass err: 5476e0ffce6SSimon Glass free(fdt); 548529fd188SSimon Glass close(fd); 549529fd188SSimon Glass return ret; 550529fd188SSimon Glass } 551529fd188SSimon Glass 552722ebc8fSSimon Glass /** 55389a4d6b1SPrafulla Wadaskar * fit_handle_file - main FIT file processing function 55489a4d6b1SPrafulla Wadaskar * 55589a4d6b1SPrafulla Wadaskar * fit_handle_file() runs dtc to convert .its to .itb, includes 55689a4d6b1SPrafulla Wadaskar * binary data, updates timestamp property and calculates hashes. 55789a4d6b1SPrafulla Wadaskar * 55889a4d6b1SPrafulla Wadaskar * datafile - .its file 55989a4d6b1SPrafulla Wadaskar * imagefile - .itb file 56089a4d6b1SPrafulla Wadaskar * 56189a4d6b1SPrafulla Wadaskar * returns: 56289a4d6b1SPrafulla Wadaskar * only on success, otherwise calls exit (EXIT_FAILURE); 56389a4d6b1SPrafulla Wadaskar */ 564f86ed6a8SGuilherme Maciel Ferreira static int fit_handle_file(struct image_tool_params *params) 56589a4d6b1SPrafulla Wadaskar { 56689a4d6b1SPrafulla Wadaskar char tmpfile[MKIMAGE_MAX_TMPFILE_LEN]; 56789a4d6b1SPrafulla Wadaskar char cmd[MKIMAGE_MAX_DTC_CMDLINE_LEN]; 568a9468115SSimon Glass size_t size_inc; 569a9468115SSimon Glass int ret; 57089a4d6b1SPrafulla Wadaskar 57189a4d6b1SPrafulla Wadaskar /* Flattened Image Tree (FIT) format handling */ 57289a4d6b1SPrafulla Wadaskar debug ("FIT format handling\n"); 57389a4d6b1SPrafulla Wadaskar 57489a4d6b1SPrafulla Wadaskar /* call dtc to include binary properties into the tmp file */ 57589a4d6b1SPrafulla Wadaskar if (strlen (params->imagefile) + 57689a4d6b1SPrafulla Wadaskar strlen (MKIMAGE_TMPFILE_SUFFIX) + 1 > sizeof (tmpfile)) { 57789a4d6b1SPrafulla Wadaskar fprintf (stderr, "%s: Image file name (%s) too long, " 57889a4d6b1SPrafulla Wadaskar "can't create tmpfile", 57989a4d6b1SPrafulla Wadaskar params->imagefile, params->cmdname); 58089a4d6b1SPrafulla Wadaskar return (EXIT_FAILURE); 58189a4d6b1SPrafulla Wadaskar } 58289a4d6b1SPrafulla Wadaskar sprintf (tmpfile, "%s%s", params->imagefile, MKIMAGE_TMPFILE_SUFFIX); 58389a4d6b1SPrafulla Wadaskar 58495d77b44SSimon Glass /* We either compile the source file, or use the existing FIT image */ 5858e35bb07SSimon Glass if (params->auto_its) { 5868e35bb07SSimon Glass if (fit_build(params, tmpfile)) { 5878e35bb07SSimon Glass fprintf(stderr, "%s: failed to build FIT\n", 5888e35bb07SSimon Glass params->cmdname); 5898e35bb07SSimon Glass return EXIT_FAILURE; 5908e35bb07SSimon Glass } 5918e35bb07SSimon Glass *cmd = '\0'; 5928e35bb07SSimon Glass } else if (params->datafile) { 593ee7c0fc7SLars Rasmusson /* dtc -I dts -O dtb -p 500 datafile > tmpfile */ 59495d77b44SSimon Glass snprintf(cmd, sizeof(cmd), "%s %s %s > %s", 59589a4d6b1SPrafulla Wadaskar MKIMAGE_DTC, params->dtc, params->datafile, tmpfile); 59689a4d6b1SPrafulla Wadaskar debug("Trying to execute \"%s\"\n", cmd); 59795d77b44SSimon Glass } else { 59895d77b44SSimon Glass snprintf(cmd, sizeof(cmd), "cp %s %s", 59995d77b44SSimon Glass params->imagefile, tmpfile); 60095d77b44SSimon Glass } 6018e35bb07SSimon Glass if (*cmd && system(cmd) == -1) { 60289a4d6b1SPrafulla Wadaskar fprintf (stderr, "%s: system(%s) failed: %s\n", 60389a4d6b1SPrafulla Wadaskar params->cmdname, cmd, strerror(errno)); 604aa6d6db4SSimon Glass goto err_system; 60589a4d6b1SPrafulla Wadaskar } 60689a4d6b1SPrafulla Wadaskar 607529fd188SSimon Glass /* Move the data so it is internal to the FIT, if needed */ 608529fd188SSimon Glass ret = fit_import_data(params, tmpfile); 609529fd188SSimon Glass if (ret) 610529fd188SSimon Glass goto err_system; 611529fd188SSimon Glass 612a9468115SSimon Glass /* 613a9468115SSimon Glass * Set hashes for images in the blob. Unfortunately we may need more 614a9468115SSimon Glass * space in either FDT, so keep trying until we succeed. 615a9468115SSimon Glass * 616a9468115SSimon Glass * Note: this is pretty inefficient for signing, since we must 617a9468115SSimon Glass * calculate the signature every time. It would be better to calculate 618a9468115SSimon Glass * all the data and then store it in a separate step. However, this 619a9468115SSimon Glass * would be considerably more complex to implement. Generally a few 620a9468115SSimon Glass * steps of this loop is enough to sign with several keys. 621a9468115SSimon Glass */ 622a9468115SSimon Glass for (size_inc = 0; size_inc < 64 * 1024; size_inc += 1024) { 623a9468115SSimon Glass ret = fit_add_file_data(params, size_inc, tmpfile); 624a9468115SSimon Glass if (!ret || ret != -ENOSPC) 625a9468115SSimon Glass break; 626e29495d3SSimon Glass } 627e29495d3SSimon Glass 628a9468115SSimon Glass if (ret) { 629399c744bSSimon Glass fprintf(stderr, "%s Can't add hashes to FIT blob\n", 63089a4d6b1SPrafulla Wadaskar params->cmdname); 631a9468115SSimon Glass goto err_system; 632e29495d3SSimon Glass } 63389a4d6b1SPrafulla Wadaskar 634722ebc8fSSimon Glass /* Move the data so it is external to the FIT, if requested */ 635722ebc8fSSimon Glass if (params->external_data) { 636722ebc8fSSimon Glass ret = fit_extract_data(params, tmpfile); 637722ebc8fSSimon Glass if (ret) 638722ebc8fSSimon Glass goto err_system; 639722ebc8fSSimon Glass } 640722ebc8fSSimon Glass 64189a4d6b1SPrafulla Wadaskar if (rename (tmpfile, params->imagefile) == -1) { 64289a4d6b1SPrafulla Wadaskar fprintf (stderr, "%s: Can't rename %s to %s: %s\n", 64389a4d6b1SPrafulla Wadaskar params->cmdname, tmpfile, params->imagefile, 64489a4d6b1SPrafulla Wadaskar strerror (errno)); 64589a4d6b1SPrafulla Wadaskar unlink (tmpfile); 64689a4d6b1SPrafulla Wadaskar unlink (params->imagefile); 647a9468115SSimon Glass return EXIT_FAILURE; 64889a4d6b1SPrafulla Wadaskar } 649a9468115SSimon Glass return EXIT_SUCCESS; 650aa6d6db4SSimon Glass 651aa6d6db4SSimon Glass err_system: 652aa6d6db4SSimon Glass unlink(tmpfile); 653aa6d6db4SSimon Glass return -1; 65489a4d6b1SPrafulla Wadaskar } 65589a4d6b1SPrafulla Wadaskar 65639931f96SGuilherme Maciel Ferreira /** 65739931f96SGuilherme Maciel Ferreira * fit_image_extract - extract a FIT component image 65839931f96SGuilherme Maciel Ferreira * @fit: pointer to the FIT format image header 65939931f96SGuilherme Maciel Ferreira * @image_noffset: offset of the component image node 66039931f96SGuilherme Maciel Ferreira * @file_name: name of the file to store the FIT sub-image 66139931f96SGuilherme Maciel Ferreira * 66239931f96SGuilherme Maciel Ferreira * returns: 66339931f96SGuilherme Maciel Ferreira * zero in case of success or a negative value if fail. 66439931f96SGuilherme Maciel Ferreira */ 66539931f96SGuilherme Maciel Ferreira static int fit_image_extract( 66639931f96SGuilherme Maciel Ferreira const void *fit, 66739931f96SGuilherme Maciel Ferreira int image_noffset, 66839931f96SGuilherme Maciel Ferreira const char *file_name) 66939931f96SGuilherme Maciel Ferreira { 67039931f96SGuilherme Maciel Ferreira const void *file_data; 67139931f96SGuilherme Maciel Ferreira size_t file_size = 0; 67239931f96SGuilherme Maciel Ferreira 67339931f96SGuilherme Maciel Ferreira /* get the "data" property of component at offset "image_noffset" */ 67439931f96SGuilherme Maciel Ferreira fit_image_get_data(fit, image_noffset, &file_data, &file_size); 67539931f96SGuilherme Maciel Ferreira 67639931f96SGuilherme Maciel Ferreira /* save the "file_data" into the file specified by "file_name" */ 67739931f96SGuilherme Maciel Ferreira return imagetool_save_subimage(file_name, (ulong) file_data, file_size); 67839931f96SGuilherme Maciel Ferreira } 67939931f96SGuilherme Maciel Ferreira 68039931f96SGuilherme Maciel Ferreira /** 68139931f96SGuilherme Maciel Ferreira * fit_extract_contents - retrieve a sub-image component from the FIT image 68239931f96SGuilherme Maciel Ferreira * @ptr: pointer to the FIT format image header 68339931f96SGuilherme Maciel Ferreira * @params: command line parameters 68439931f96SGuilherme Maciel Ferreira * 68539931f96SGuilherme Maciel Ferreira * returns: 68639931f96SGuilherme Maciel Ferreira * zero in case of success or a negative value if fail. 68739931f96SGuilherme Maciel Ferreira */ 68839931f96SGuilherme Maciel Ferreira static int fit_extract_contents(void *ptr, struct image_tool_params *params) 68939931f96SGuilherme Maciel Ferreira { 69039931f96SGuilherme Maciel Ferreira int images_noffset; 69139931f96SGuilherme Maciel Ferreira int noffset; 69239931f96SGuilherme Maciel Ferreira int ndepth; 69339931f96SGuilherme Maciel Ferreira const void *fit = ptr; 69439931f96SGuilherme Maciel Ferreira int count = 0; 69539931f96SGuilherme Maciel Ferreira const char *p; 69639931f96SGuilherme Maciel Ferreira 69739931f96SGuilherme Maciel Ferreira /* Indent string is defined in header image.h */ 69839931f96SGuilherme Maciel Ferreira p = IMAGE_INDENT_STRING; 69939931f96SGuilherme Maciel Ferreira 70039931f96SGuilherme Maciel Ferreira if (!fit_check_format(fit)) { 70139931f96SGuilherme Maciel Ferreira printf("Bad FIT image format\n"); 70239931f96SGuilherme Maciel Ferreira return -1; 70339931f96SGuilherme Maciel Ferreira } 70439931f96SGuilherme Maciel Ferreira 70539931f96SGuilherme Maciel Ferreira /* Find images parent node offset */ 70639931f96SGuilherme Maciel Ferreira images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH); 70739931f96SGuilherme Maciel Ferreira if (images_noffset < 0) { 70839931f96SGuilherme Maciel Ferreira printf("Can't find images parent node '%s' (%s)\n", 70939931f96SGuilherme Maciel Ferreira FIT_IMAGES_PATH, fdt_strerror(images_noffset)); 71039931f96SGuilherme Maciel Ferreira return -1; 71139931f96SGuilherme Maciel Ferreira } 71239931f96SGuilherme Maciel Ferreira 71339931f96SGuilherme Maciel Ferreira /* Avoid any overrun */ 71439931f96SGuilherme Maciel Ferreira count = fit_get_subimage_count(fit, images_noffset); 71539931f96SGuilherme Maciel Ferreira if ((params->pflag < 0) || (count <= params->pflag)) { 71639931f96SGuilherme Maciel Ferreira printf("No such component at '%d'\n", params->pflag); 71739931f96SGuilherme Maciel Ferreira return -1; 71839931f96SGuilherme Maciel Ferreira } 71939931f96SGuilherme Maciel Ferreira 72039931f96SGuilherme Maciel Ferreira /* Process its subnodes, extract the desired component from image */ 72139931f96SGuilherme Maciel Ferreira for (ndepth = 0, count = 0, 72239931f96SGuilherme Maciel Ferreira noffset = fdt_next_node(fit, images_noffset, &ndepth); 72339931f96SGuilherme Maciel Ferreira (noffset >= 0) && (ndepth > 0); 72439931f96SGuilherme Maciel Ferreira noffset = fdt_next_node(fit, noffset, &ndepth)) { 72539931f96SGuilherme Maciel Ferreira if (ndepth == 1) { 72639931f96SGuilherme Maciel Ferreira /* 72739931f96SGuilherme Maciel Ferreira * Direct child node of the images parent node, 72839931f96SGuilherme Maciel Ferreira * i.e. component image node. 72939931f96SGuilherme Maciel Ferreira */ 73039931f96SGuilherme Maciel Ferreira if (params->pflag == count) { 73139931f96SGuilherme Maciel Ferreira printf("Extracted:\n%s Image %u (%s)\n", p, 73239931f96SGuilherme Maciel Ferreira count, fit_get_name(fit, noffset, NULL)); 73339931f96SGuilherme Maciel Ferreira 73439931f96SGuilherme Maciel Ferreira fit_image_print(fit, noffset, p); 73539931f96SGuilherme Maciel Ferreira 73639931f96SGuilherme Maciel Ferreira return fit_image_extract(fit, noffset, 73739931f96SGuilherme Maciel Ferreira params->outfile); 73839931f96SGuilherme Maciel Ferreira } 73939931f96SGuilherme Maciel Ferreira 74039931f96SGuilherme Maciel Ferreira count++; 74139931f96SGuilherme Maciel Ferreira } 74239931f96SGuilherme Maciel Ferreira } 74339931f96SGuilherme Maciel Ferreira 74439931f96SGuilherme Maciel Ferreira return 0; 74539931f96SGuilherme Maciel Ferreira } 74639931f96SGuilherme Maciel Ferreira 747f86ed6a8SGuilherme Maciel Ferreira static int fit_check_params(struct image_tool_params *params) 74889a4d6b1SPrafulla Wadaskar { 7498e35bb07SSimon Glass if (params->auto_its) 7508e35bb07SSimon Glass return 0; 75189a4d6b1SPrafulla Wadaskar return ((params->dflag && (params->fflag || params->lflag)) || 75289a4d6b1SPrafulla Wadaskar (params->fflag && (params->dflag || params->lflag)) || 75389a4d6b1SPrafulla Wadaskar (params->lflag && (params->dflag || params->fflag))); 75489a4d6b1SPrafulla Wadaskar } 75589a4d6b1SPrafulla Wadaskar 756a93648d1SGuilherme Maciel Ferreira U_BOOT_IMAGE_TYPE( 757a93648d1SGuilherme Maciel Ferreira fitimage, 758a93648d1SGuilherme Maciel Ferreira "FIT Image support", 759a93648d1SGuilherme Maciel Ferreira sizeof(image_header_t), 760a93648d1SGuilherme Maciel Ferreira (void *)&header, 761a93648d1SGuilherme Maciel Ferreira fit_check_params, 762a93648d1SGuilherme Maciel Ferreira fit_verify_header, 763a93648d1SGuilherme Maciel Ferreira fit_print_contents, 764a93648d1SGuilherme Maciel Ferreira NULL, 76539931f96SGuilherme Maciel Ferreira fit_extract_contents, 766a93648d1SGuilherme Maciel Ferreira fit_check_image_types, 767a93648d1SGuilherme Maciel Ferreira fit_handle_file, 768a93648d1SGuilherme Maciel Ferreira NULL /* FIT images use DTB header */ 769a93648d1SGuilherme Maciel Ferreira ); 770