blob: 5c6a60e85136074d98b8ad5046bb7715a2b7b1ab [file] [log] [blame]
Tom Rini10e47792018-05-06 17:58:06 -04001// SPDX-License-Identifier: GPL-2.0+
wdenk5b1d7132002-11-03 00:07:02 +00002/*
Bartlomiej Siekad7816fb2008-03-11 12:34:47 +01003 * (C) Copyright 2008 Semihalf
4 *
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +05305 * (C) Copyright 2000-2009
wdenk5b1d7132002-11-03 00:07:02 +00006 * DENX Software Engineering
7 * Wolfgang Denk, wd@denx.de
wdenk5b1d7132002-11-03 00:07:02 +00008 */
9
Kever Yang86f26012020-03-30 11:56:19 +080010#include "imagetool.h"
Marian Balakowicz41d71ed2008-01-08 18:14:09 +010011#include "mkimage.h"
Sven Ebenfeld59697a22016-11-06 16:37:56 +010012#include "imximage.h"
Simon Glasse4607262021-11-12 12:28:13 -070013#include <fit_common.h>
wdenk5b1d7132002-11-03 00:07:02 +000014#include <image.h>
Wolfgang Denke97ab722011-02-12 10:37:11 +010015#include <version.h>
Yann Dirson774aed52021-05-18 10:59:04 +020016#ifdef __linux__
17#include <sys/ioctl.h>
18#endif
wdenk5b1d7132002-11-03 00:07:02 +000019
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +053020static void copy_file(int, const char *, int);
wdenk5b1d7132002-11-03 00:07:02 +000021
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +053022/* parameters initialized by core will be used by the image type code */
Simon Glass52322d62016-02-22 22:55:38 -070023static struct image_tool_params params = {
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +053024 .os = IH_OS_LINUX,
25 .arch = IH_ARCH_PPC,
26 .type = IH_TYPE_KERNEL,
27 .comp = IH_COMP_GZIP,
28 .dtc = MKIMAGE_DEFAULT_DTC_OPTIONS,
Wolfgang Denk23416b82010-03-27 23:37:46 +010029 .imagename = "",
Shaohui Xieea65fd82012-08-10 02:49:35 +000030 .imagename2 = "",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +053031};
wdenk5b1d7132002-11-03 00:07:02 +000032
Simon Glass1d138b52016-06-30 10:52:17 -060033static enum ih_category cur_category;
34
35static int h_compare_category_name(const void *vtype1, const void *vtype2)
36{
37 const int *type1 = vtype1;
38 const int *type2 = vtype2;
39 const char *name1 = genimg_get_cat_short_name(cur_category, *type1);
40 const char *name2 = genimg_get_cat_short_name(cur_category, *type2);
41
42 return strcmp(name1, name2);
43}
44
Simon Glassbe421592016-06-30 10:52:18 -060045static int show_valid_options(enum ih_category category)
Simon Glass1d138b52016-06-30 10:52:17 -060046{
47 int *order;
48 int count;
49 int item;
50 int i;
51
52 count = genimg_get_cat_count(category);
53 order = calloc(count, sizeof(*order));
54 if (!order)
55 return -ENOMEM;
56
57 /* Sort the names in order of short name for easier reading */
Naoki Hayama089fc922020-10-07 11:21:55 +090058 for (i = 0, item = 0; i < count; i++, item++) {
59 while (!genimg_cat_has_id(category, item) && i < count) {
60 item++;
61 count--;
62 }
63 order[i] = item;
64 }
Simon Glass1d138b52016-06-30 10:52:17 -060065 cur_category = category;
66 qsort(order, count, sizeof(int), h_compare_category_name);
67
68 fprintf(stderr, "\nInvalid %s, supported are:\n",
69 genimg_get_cat_desc(category));
70 for (i = 0; i < count; i++) {
71 item = order[i];
72 fprintf(stderr, "\t%-15s %s\n",
73 genimg_get_cat_short_name(category, item),
74 genimg_get_cat_name(category, item));
75 }
76 fprintf(stderr, "\n");
Simon Glassd435d9f2016-10-27 17:54:03 -060077 free(order);
Simon Glass1d138b52016-06-30 10:52:17 -060078
79 return 0;
80}
81
Simon Glass01168b32016-02-22 22:55:37 -070082static void usage(const char *msg)
Simon Glass99a7c262016-02-22 22:55:36 -070083{
Simon Glass01168b32016-02-22 22:55:37 -070084 fprintf(stderr, "Error: %s\n", msg);
Pali Rohárcdf0d1c2022-02-13 01:09:46 +010085 fprintf(stderr, "Usage: %s [-T type] -l image\n"
86 " -l ==> list image header information\n"
Sean Anderson138d2712022-04-08 16:08:39 -040087 " -T ==> parse image file as 'type'\n"
88 " -q ==> quiet\n",
Simon Glass99a7c262016-02-22 22:55:36 -070089 params.cmdname);
90 fprintf(stderr,
91 " %s [-x] -A arch -O os -T type -C comp -a addr -e ep -n name -d data_file[:data_file...] image\n"
92 " -A ==> set architecture to 'arch'\n"
93 " -O ==> set operating system to 'os'\n"
94 " -T ==> set image type to 'type'\n"
95 " -C ==> set compression type 'comp'\n"
96 " -a ==> set load address to 'addr' (hex)\n"
97 " -e ==> set entry point to 'ep' (hex)\n"
98 " -n ==> set image name to 'name'\n"
Sean Anderson138d2712022-04-08 16:08:39 -040099 " -R ==> set second image name to 'name'\n"
Simon Glass99a7c262016-02-22 22:55:36 -0700100 " -d ==> use image data from 'datafile'\n"
Sean Anderson138d2712022-04-08 16:08:39 -0400101 " -x ==> set XIP (execute in place)\n"
102 " -s ==> create an image with no data\n"
103 " -v ==> verbose\n",
Simon Glass99a7c262016-02-22 22:55:36 -0700104 params.cmdname);
105 fprintf(stderr,
Joel Stanley00dffdd2020-12-08 14:42:15 +1030106 " %s [-D dtc_options] [-f fit-image.its|-f auto|-F] [-b <dtb> [-b <dtb>]] [-E] [-B size] [-i <ramdisk.cpio.gz>] fit-image\n"
Vagrant Cascadian3e9e6d62016-10-23 20:45:17 -0700107 " <dtb> file is used with -f auto, it may occur multiple times.\n",
Simon Glass99a7c262016-02-22 22:55:36 -0700108 params.cmdname);
109 fprintf(stderr,
110 " -D => set all options for device tree compiler\n"
Tomeu Vizoso8d83ed22016-11-04 14:22:15 +0100111 " -f => input filename for FIT source\n"
Joel Stanley00dffdd2020-12-08 14:42:15 +1030112 " -i => input filename for ramdisk file\n"
113 " -E => place data outside of the FIT structure\n"
Sean Anderson138d2712022-04-08 16:08:39 -0400114 " -B => align size in hex for FIT structure and header\n"
115 " -b => append the device tree binary to the FIT\n"
116 " -t => update the timestamp in the FIT\n");
Simon Glass99a7c262016-02-22 22:55:36 -0700117#ifdef CONFIG_FIT_SIGNATURE
118 fprintf(stderr,
Joel Stanley00dffdd2020-12-08 14:42:15 +1030119 "Signing / verified boot options: [-k keydir] [-K dtb] [ -c <comment>] [-p addr] [-r] [-N engine]\n"
Simon Glass99a7c262016-02-22 22:55:36 -0700120 " -k => set directory containing private keys\n"
121 " -K => write public keys to this .dtb file\n"
Alexandru Gagniuc8fcea122021-02-19 12:45:17 -0600122 " -G => use this signing key (in lieu of -k)\n"
Simon Glass99a7c262016-02-22 22:55:36 -0700123 " -c => add comment in signature node\n"
124 " -F => re-sign existing FIT image\n"
Teddy Reeda8457622016-06-09 19:38:02 -0700125 " -p => place external data at a static position\n"
George McCollister23d14892017-01-06 13:14:17 -0600126 " -r => mark keys used as 'required' in dtb\n"
Sean Anderson138d2712022-04-08 16:08:39 -0400127 " -N => openssl engine to use for signing\n"
128 " -o => algorithm to use for signing\n");
Simon Glass99a7c262016-02-22 22:55:36 -0700129#else
130 fprintf(stderr,
131 "Signing / verified boot not supported (CONFIG_FIT_SIGNATURE undefined)\n");
132#endif
133 fprintf(stderr, " %s -V ==> print version information and exit\n",
134 params.cmdname);
Baruch Siachf77ab202017-06-29 20:37:08 +0300135 fprintf(stderr, "Use '-T list' to see a list of available image types\n");
Simon Glass99a7c262016-02-22 22:55:36 -0700136
137 exit(EXIT_FAILURE);
138}
139
Simon Glassbd8bc5d2016-02-22 22:55:52 -0700140static int add_content(int type, const char *fname)
141{
142 struct content_info *cont;
143
144 cont = calloc(1, sizeof(*cont));
145 if (!cont)
146 return -1;
147 cont->type = type;
148 cont->fname = fname;
149 if (params.content_tail)
150 params.content_tail->next = cont;
151 else
152 params.content_head = cont;
153 params.content_tail = cont;
154
155 return 0;
156}
157
Simon Glass7106bd42016-02-22 22:55:33 -0700158static void process_args(int argc, char **argv)
wdenk5b1d7132002-11-03 00:07:02 +0000159{
Peter Tyser632d9ca2010-04-04 22:36:03 -0500160 char *ptr;
Simon Glass04b46072016-02-22 22:55:48 -0700161 int type = IH_TYPE_INVALID;
162 char *datafile = NULL;
Simon Glasscf76cd22016-02-22 22:55:34 -0700163 int opt;
wdenk5b1d7132002-11-03 00:07:02 +0000164
Simon Glasscf76cd22016-02-22 22:55:34 -0700165 while ((opt = getopt(argc, argv,
Jan Kiszka4043f322022-01-14 10:21:19 +0100166 "a:A:b:B:c:C:d:D:e:Ef:FG:k:i:K:ln:N:p:o:O:rR:qstT:vVx")) != -1) {
Simon Glasscf76cd22016-02-22 22:55:34 -0700167 switch (opt) {
Simon Glassf6fe5f62016-02-22 22:55:35 -0700168 case 'a':
169 params.addr = strtoull(optarg, &ptr, 16);
170 if (*ptr) {
171 fprintf(stderr, "%s: invalid load address %s\n",
172 params.cmdname, optarg);
173 exit(EXIT_FAILURE);
174 }
Simon Glasscf76cd22016-02-22 22:55:34 -0700175 break;
176 case 'A':
177 params.arch = genimg_get_arch_id(optarg);
Simon Glasseaa25402016-06-30 10:52:19 -0600178 if (params.arch < 0) {
179 show_valid_options(IH_ARCH);
Simon Glass01168b32016-02-22 22:55:37 -0700180 usage("Invalid architecture");
Simon Glasseaa25402016-06-30 10:52:19 -0600181 }
Icenowy Zheng4369c972021-10-14 20:53:04 -0500182 params.Aflag = 1;
Simon Glasscf76cd22016-02-22 22:55:34 -0700183 break;
Simon Glassbd8bc5d2016-02-22 22:55:52 -0700184 case 'b':
Andreas Bießmannddd34772016-05-03 15:17:03 +0200185 if (add_content(IH_TYPE_FLATDT, optarg)) {
Andreas Bießmannf4001582016-05-01 03:01:27 +0200186 fprintf(stderr,
187 "%s: Out of memory adding content '%s'",
188 params.cmdname, optarg);
189 exit(EXIT_FAILURE);
190 }
Simon Glassbd8bc5d2016-02-22 22:55:52 -0700191 break;
Kever Yang8e238b52020-03-30 11:56:24 +0800192 case 'B':
193 params.bl_len = strtoull(optarg, &ptr, 16);
194 if (*ptr) {
195 fprintf(stderr, "%s: invalid block length %s\n",
196 params.cmdname, optarg);
197 exit(EXIT_FAILURE);
198 }
199
200 break;
Simon Glasscf76cd22016-02-22 22:55:34 -0700201 case 'c':
202 params.comment = optarg;
203 break;
204 case 'C':
205 params.comp = genimg_get_comp_id(optarg);
Simon Glasseaa25402016-06-30 10:52:19 -0600206 if (params.comp < 0) {
207 show_valid_options(IH_COMP);
Simon Glass01168b32016-02-22 22:55:37 -0700208 usage("Invalid compression type");
Simon Glasseaa25402016-06-30 10:52:19 -0600209 }
Simon Glasscf76cd22016-02-22 22:55:34 -0700210 break;
Simon Glasscf76cd22016-02-22 22:55:34 -0700211 case 'd':
212 params.datafile = optarg;
213 params.dflag = 1;
214 break;
Simon Glassf6fe5f62016-02-22 22:55:35 -0700215 case 'D':
216 params.dtc = optarg;
217 break;
Simon Glasscf76cd22016-02-22 22:55:34 -0700218 case 'e':
219 params.ep = strtoull(optarg, &ptr, 16);
220 if (*ptr) {
221 fprintf(stderr, "%s: invalid entry point %s\n",
222 params.cmdname, optarg);
223 exit(EXIT_FAILURE);
wdenk5b1d7132002-11-03 00:07:02 +0000224 }
Simon Glasscf76cd22016-02-22 22:55:34 -0700225 params.eflag = 1;
226 break;
Simon Glassafd728c2016-02-22 22:55:53 -0700227 case 'E':
228 params.external_data = true;
229 break;
Simon Glasscf76cd22016-02-22 22:55:34 -0700230 case 'f':
Simon Glass88e31cb2016-02-22 22:55:51 -0700231 datafile = optarg;
232 params.auto_its = !strcmp(datafile, "auto");
Heinrich Schuchardtddc38e62020-05-09 17:10:41 +0200233 /* fallthrough */
Simon Glasscf76cd22016-02-22 22:55:34 -0700234 case 'F':
235 /*
236 * The flattened image tree (FIT) format
237 * requires a flattened device tree image type
238 */
239 params.type = IH_TYPE_FLATDT;
240 params.fflag = 1;
241 break;
Alexandru Gagniuc8fcea122021-02-19 12:45:17 -0600242 case 'G':
243 params.keyfile = optarg;
244 break;
Tomeu Vizoso8d83ed22016-11-04 14:22:15 +0100245 case 'i':
246 params.fit_ramdisk = optarg;
247 break;
Simon Glasscf76cd22016-02-22 22:55:34 -0700248 case 'k':
249 params.keydir = optarg;
250 break;
251 case 'K':
252 params.keydest = optarg;
253 break;
Simon Glassf6fe5f62016-02-22 22:55:35 -0700254 case 'l':
255 params.lflag = 1;
256 break;
Simon Glasscf76cd22016-02-22 22:55:34 -0700257 case 'n':
258 params.imagename = optarg;
259 break;
George McCollister23d14892017-01-06 13:14:17 -0600260 case 'N':
261 params.engine_id = optarg;
262 break;
Jan Kiszka4043f322022-01-14 10:21:19 +0100263 case 'o':
264 params.algo_name = optarg;
265 break;
Simon Glassf6fe5f62016-02-22 22:55:35 -0700266 case 'O':
267 params.os = genimg_get_os_id(optarg);
Simon Glasseaa25402016-06-30 10:52:19 -0600268 if (params.os < 0) {
269 show_valid_options(IH_OS);
Simon Glass01168b32016-02-22 22:55:37 -0700270 usage("Invalid operating system");
Simon Glasseaa25402016-06-30 10:52:19 -0600271 }
Simon Glassf6fe5f62016-02-22 22:55:35 -0700272 break;
Teddy Reeda8457622016-06-09 19:38:02 -0700273 case 'p':
274 params.external_offset = strtoull(optarg, &ptr, 16);
275 if (*ptr) {
276 fprintf(stderr, "%s: invalid offset size %s\n",
277 params.cmdname, optarg);
278 exit(EXIT_FAILURE);
279 }
Teddy Reed9e3cd3c2016-07-11 22:54:26 -0700280 break;
Simon Glassda1805f2016-05-01 13:55:38 -0600281 case 'q':
282 params.quiet = 1;
283 break;
Simon Glasscf76cd22016-02-22 22:55:34 -0700284 case 'r':
285 params.require_keys = 1;
286 break;
287 case 'R':
288 /*
289 * This entry is for the second configuration
290 * file, if only one is not enough.
291 */
292 params.imagename2 = optarg;
293 break;
294 case 's':
295 params.skipcpy = 1;
296 break;
Simon Glass472ee0c2020-07-09 18:39:43 -0600297 case 't':
298 params.reset_timestamp = 1;
299 break;
Simon Glassf6fe5f62016-02-22 22:55:35 -0700300 case 'T':
Baruch Siachf77ab202017-06-29 20:37:08 +0300301 if (strcmp(optarg, "list") == 0) {
302 show_valid_options(IH_TYPE);
303 exit(EXIT_SUCCESS);
304 }
Simon Glass04b46072016-02-22 22:55:48 -0700305 type = genimg_get_type_id(optarg);
306 if (type < 0) {
Simon Glassbe421592016-06-30 10:52:18 -0600307 show_valid_options(IH_TYPE);
Simon Glass01168b32016-02-22 22:55:37 -0700308 usage("Invalid image type");
Simon Glassf6fe5f62016-02-22 22:55:35 -0700309 }
310 break;
Simon Glasscf76cd22016-02-22 22:55:34 -0700311 case 'v':
312 params.vflag++;
313 break;
314 case 'V':
315 printf("mkimage version %s\n", PLAIN_VERSION);
316 exit(EXIT_SUCCESS);
317 case 'x':
318 params.xflag++;
319 break;
320 default:
Simon Glass01168b32016-02-22 22:55:37 -0700321 usage("Invalid option");
wdenk5b1d7132002-11-03 00:07:02 +0000322 }
wdenk5b1d7132002-11-03 00:07:02 +0000323 }
324
Andreas Bießmannddd34772016-05-03 15:17:03 +0200325 /* The last parameter is expected to be the imagefile */
326 if (optind < argc)
Andreas Bießmannf4001582016-05-01 03:01:27 +0200327 params.imagefile = argv[optind];
328
Simon Glass04b46072016-02-22 22:55:48 -0700329 /*
330 * For auto-generated FIT images we need to know the image type to put
331 * in the FIT, which is separate from the file's image type (which
332 * will always be IH_TYPE_FLATDT in this case).
333 */
334 if (params.type == IH_TYPE_FLATDT) {
Simon Glassa89857d2016-06-30 10:52:07 -0600335 params.fit_image_type = type ? type : IH_TYPE_KERNEL;
Simon Glass156c3492016-06-30 10:52:08 -0600336 /* For auto_its, datafile is always 'auto' */
Simon Glass88e31cb2016-02-22 22:55:51 -0700337 if (!params.auto_its)
338 params.datafile = datafile;
Simon Glass79b5ee52016-06-30 10:52:09 -0600339 else if (!params.datafile)
340 usage("Missing data file for auto-FIT (use -d)");
Pali Rohárcdf0d1c2022-02-13 01:09:46 +0100341 } else if (params.lflag || type != IH_TYPE_INVALID) {
Alex Kiernan9d74c702018-04-22 05:11:17 +0000342 if (type == IH_TYPE_SCRIPT && !params.datafile)
343 usage("Missing data file for script (use -d)");
Simon Glass04b46072016-02-22 22:55:48 -0700344 params.type = type;
345 }
346
347 if (!params.imagefile)
Simon Glass01168b32016-02-22 22:55:37 -0700348 usage("Missing output filename");
Simon Glass7106bd42016-02-22 22:55:33 -0700349}
350
Pali Rohára55eaa52022-01-14 18:34:43 +0100351static void verify_image(const struct image_type_params *tparams)
352{
353 struct stat sbuf;
354 void *ptr;
355 int ifd;
356
357 ifd = open(params.imagefile, O_RDONLY | O_BINARY);
358 if (ifd < 0) {
359 fprintf(stderr, "%s: Can't open %s: %s\n",
360 params.cmdname, params.imagefile,
361 strerror(errno));
362 exit(EXIT_FAILURE);
363 }
364
365 if (fstat(ifd, &sbuf) < 0) {
366 fprintf(stderr, "%s: Can't stat %s: %s\n",
367 params.cmdname, params.imagefile, strerror(errno));
368 exit(EXIT_FAILURE);
369 }
370 params.file_size = sbuf.st_size;
371
372 ptr = mmap(0, params.file_size, PROT_READ, MAP_SHARED, ifd, 0);
373 if (ptr == MAP_FAILED) {
374 fprintf(stderr, "%s: Can't map %s: %s\n",
375 params.cmdname, params.imagefile, strerror(errno));
376 exit(EXIT_FAILURE);
377 }
378
379 if (tparams->verify_header((unsigned char *)ptr, params.file_size, &params) != 0) {
380 fprintf(stderr, "%s: Failed to verify header of %s\n",
381 params.cmdname, params.imagefile);
382 exit(EXIT_FAILURE);
383 }
384
385 (void)munmap(ptr, params.file_size);
386 (void)close(ifd);
387}
388
Simon Glass7106bd42016-02-22 22:55:33 -0700389int main(int argc, char **argv)
390{
391 int ifd = -1;
392 struct stat sbuf;
393 char *ptr;
394 int retval = 0;
395 struct image_type_params *tparams = NULL;
396 int pad_len = 0;
397 int dfd;
Mark Tomlinsonfc497862018-08-29 10:51:14 +1200398 size_t map_len;
Simon Glass7106bd42016-02-22 22:55:33 -0700399
400 params.cmdname = *argv;
401 params.addr = 0;
402 params.ep = 0;
403
404 process_args(argc, argv);
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530405
406 /* set tparams as per input type_id */
Guilherme Maciel Ferreira28be1cf2015-01-15 02:48:07 -0200407 tparams = imagetool_get_type(params.type);
Pali Rohárcdf0d1c2022-02-13 01:09:46 +0100408 if (tparams == NULL && !params.lflag) {
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530409 fprintf (stderr, "%s: unsupported type %s\n",
410 params.cmdname, genimg_get_type_name(params.type));
411 exit (EXIT_FAILURE);
412 }
413
414 /*
415 * check the passed arguments parameters meets the requirements
416 * as per image type to be generated/listed
417 */
Pali Rohárcdf0d1c2022-02-13 01:09:46 +0100418 if (tparams && tparams->check_params)
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530419 if (tparams->check_params (&params))
Simon Glass01168b32016-02-22 22:55:37 -0700420 usage("Bad parameters for image type");
wdenk5b1d7132002-11-03 00:07:02 +0000421
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530422 if (!params.eflag) {
423 params.ep = params.addr;
wdenk5b1d7132002-11-03 00:07:02 +0000424 /* If XIP, entry point must be after the U-Boot header */
Pali Rohárcdf0d1c2022-02-13 01:09:46 +0100425 if (params.xflag && tparams)
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530426 params.ep += tparams->header_size;
wdenk5b1d7132002-11-03 00:07:02 +0000427 }
428
Peter Tyserb2f8b942009-11-24 16:42:10 -0600429 if (params.fflag){
Heinrich Schuchardt8b933b42022-03-01 08:53:56 +0100430 if (!tparams) {
431 fprintf(stderr, "%s: Missing FIT support\n",
432 params.cmdname);
433 exit (EXIT_FAILURE);
434 }
Peter Tyserb2f8b942009-11-24 16:42:10 -0600435 if (tparams->fflag_handle)
436 /*
437 * in some cases, some additional processing needs
438 * to be done if fflag is defined
439 *
440 * For ex. fit_handle_file for Fit file support
441 */
442 retval = tparams->fflag_handle(&params);
wdenk5b1d7132002-11-03 00:07:02 +0000443
Peter Tyserb2f8b942009-11-24 16:42:10 -0600444 if (retval != EXIT_SUCCESS)
Heinrich Schuchardt8b933b42022-03-01 08:53:56 +0100445 usage("Bad parameters for FIT image type");
wdenk5b1d7132002-11-03 00:07:02 +0000446 }
447
Peter Tyserb2f8b942009-11-24 16:42:10 -0600448 if (params.lflag || params.fflag) {
449 ifd = open (params.imagefile, O_RDONLY|O_BINARY);
450 } else {
451 ifd = open (params.imagefile,
452 O_RDWR|O_CREAT|O_TRUNC|O_BINARY, 0666);
453 }
454
455 if (ifd < 0) {
456 fprintf (stderr, "%s: Can't open %s: %s\n",
457 params.cmdname, params.imagefile,
458 strerror(errno));
459 exit (EXIT_FAILURE);
460 }
461
462 if (params.lflag || params.fflag) {
Yann Dirson774aed52021-05-18 10:59:04 +0200463 uint64_t size;
wdenk5b1d7132002-11-03 00:07:02 +0000464 /*
465 * list header information of existing image
466 */
467 if (fstat(ifd, &sbuf) < 0) {
468 fprintf (stderr, "%s: Can't stat %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530469 params.cmdname, params.imagefile,
470 strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000471 exit (EXIT_FAILURE);
472 }
473
Yann Dirson774aed52021-05-18 10:59:04 +0200474 if ((sbuf.st_mode & S_IFMT) == S_IFBLK) {
475#ifdef __linux__
476#if defined(__linux__) && defined(_IOR) && !defined(BLKGETSIZE64)
477#define BLKGETSIZE64 _IOR(0x12,114,size_t) /* return device size in bytes (u64 *arg) */
478#endif
479 if (ioctl(ifd, BLKGETSIZE64, &size) < 0) {
480 fprintf (stderr,
481 "%s: failed to get size of block device \"%s\"\n",
482 params.cmdname, params.imagefile);
483 exit (EXIT_FAILURE);
484 }
485#else
wdenk5b1d7132002-11-03 00:07:02 +0000486 fprintf (stderr,
Yann Dirson774aed52021-05-18 10:59:04 +0200487 "%s: \"%s\" is block device, don't know how to get its size\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530488 params.cmdname, params.imagefile);
wdenk5b1d7132002-11-03 00:07:02 +0000489 exit (EXIT_FAILURE);
Yann Dirson774aed52021-05-18 10:59:04 +0200490#endif
Pali Rohárcdf0d1c2022-02-13 01:09:46 +0100491 } else if (tparams && sbuf.st_size < (off_t)tparams->header_size) {
Yann Dirson774aed52021-05-18 10:59:04 +0200492 fprintf (stderr,
Heinrich Schuchardt5a3b8dc2022-01-15 20:12:56 +0100493 "%s: Bad size: \"%s\" is not valid image: size %llu < %u\n",
Yann Dirson774aed52021-05-18 10:59:04 +0200494 params.cmdname, params.imagefile,
Heinrich Schuchardt5a3b8dc2022-01-15 20:12:56 +0100495 (unsigned long long) sbuf.st_size,
496 tparams->header_size);
Yann Dirson774aed52021-05-18 10:59:04 +0200497 exit (EXIT_FAILURE);
498 } else {
499 size = sbuf.st_size;
wdenk5b1d7132002-11-03 00:07:02 +0000500 }
501
Yann Dirson774aed52021-05-18 10:59:04 +0200502 ptr = mmap(0, size, PROT_READ, MAP_SHARED, ifd, 0);
Mike Frysingercd7e1a32008-05-01 04:13:05 -0400503 if (ptr == MAP_FAILED) {
wdenk5b1d7132002-11-03 00:07:02 +0000504 fprintf (stderr, "%s: Can't read %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530505 params.cmdname, params.imagefile,
506 strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000507 exit (EXIT_FAILURE);
508 }
509
Pali Rohárcdf0d1c2022-02-13 01:09:46 +0100510 /*
511 * Verifies the header format based on the expected header for image
512 * type in tparams. If tparams is NULL simply check all image types
513 * to find one that matches our header.
514 */
515 retval = imagetool_verify_print_header(ptr, &sbuf, tparams, &params);
wdenk5b1d7132002-11-03 00:07:02 +0000516
wdenk5b1d7132002-11-03 00:07:02 +0000517 (void) munmap((void *)ptr, sbuf.st_size);
518 (void) close (ifd);
Simon Glasse4607262021-11-12 12:28:13 -0700519 if (!retval)
520 summary_show(&params.summary, params.imagefile,
521 params.keydest);
wdenk5b1d7132002-11-03 00:07:02 +0000522
Prafulla Wadaskar139cb082009-08-10 18:49:37 +0530523 exit (retval);
wdenk5b1d7132002-11-03 00:07:02 +0000524 }
525
Marek Vasut4cc86272015-12-07 18:01:54 +0100526 if ((params.type != IH_TYPE_MULTI) && (params.type != IH_TYPE_SCRIPT)) {
Philippe De Swert3edd0412015-12-04 00:11:23 +0200527 dfd = open(params.datafile, O_RDONLY | O_BINARY);
528 if (dfd < 0) {
529 fprintf(stderr, "%s: Can't open %s: %s\n",
530 params.cmdname, params.datafile,
531 strerror(errno));
532 exit(EXIT_FAILURE);
533 }
Simon Glass139651f2015-06-23 15:39:12 -0600534
Philippe De Swert3edd0412015-12-04 00:11:23 +0200535 if (fstat(dfd, &sbuf) < 0) {
536 fprintf(stderr, "%s: Can't stat %s: %s\n",
537 params.cmdname, params.datafile,
538 strerror(errno));
539 exit(EXIT_FAILURE);
540 }
Simon Glass139651f2015-06-23 15:39:12 -0600541
Philippe De Swert3edd0412015-12-04 00:11:23 +0200542 params.file_size = sbuf.st_size + tparams->header_size;
543 close(dfd);
544 }
Simon Glass139651f2015-06-23 15:39:12 -0600545
wdenk5b1d7132002-11-03 00:07:02 +0000546 /*
Stefano Babic4a963302011-09-15 23:50:16 +0000547 * In case there an header with a variable
548 * length will be added, the corresponding
549 * function is called. This is responsible to
550 * allocate memory for the header itself.
wdenk5b1d7132002-11-03 00:07:02 +0000551 */
Stefano Babic4a963302011-09-15 23:50:16 +0000552 if (tparams->vrec_header)
Stefano Babic6531bd92013-08-19 19:03:19 +0200553 pad_len = tparams->vrec_header(&params, tparams);
Stefano Babic4a963302011-09-15 23:50:16 +0000554 else
555 memset(tparams->hdr, 0, tparams->header_size);
wdenk5b1d7132002-11-03 00:07:02 +0000556
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530557 if (write(ifd, tparams->hdr, tparams->header_size)
558 != tparams->header_size) {
wdenk5b1d7132002-11-03 00:07:02 +0000559 fprintf (stderr, "%s: Write error on %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530560 params.cmdname, params.imagefile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000561 exit (EXIT_FAILURE);
562 }
563
Christian Rieschc14f2812011-12-09 09:47:38 +0000564 if (!params.skipcpy) {
565 if (params.type == IH_TYPE_MULTI ||
566 params.type == IH_TYPE_SCRIPT) {
567 char *file = params.datafile;
568 uint32_t size;
wdenk5b1d7132002-11-03 00:07:02 +0000569
Christian Rieschc14f2812011-12-09 09:47:38 +0000570 for (;;) {
571 char *sep = NULL;
wdenk5b1d7132002-11-03 00:07:02 +0000572
Christian Rieschc14f2812011-12-09 09:47:38 +0000573 if (file) {
574 if ((sep = strchr(file, ':')) != NULL) {
575 *sep = '\0';
576 }
577
578 if (stat (file, &sbuf) < 0) {
579 fprintf (stderr, "%s: Can't stat %s: %s\n",
580 params.cmdname, file, strerror(errno));
581 exit (EXIT_FAILURE);
582 }
583 size = cpu_to_uimage (sbuf.st_size);
584 } else {
585 size = 0;
wdenk5b1d7132002-11-03 00:07:02 +0000586 }
587
Christian Rieschc14f2812011-12-09 09:47:38 +0000588 if (write(ifd, (char *)&size, sizeof(size)) != sizeof(size)) {
589 fprintf (stderr, "%s: Write error on %s: %s\n",
590 params.cmdname, params.imagefile,
591 strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000592 exit (EXIT_FAILURE);
593 }
wdenk5b1d7132002-11-03 00:07:02 +0000594
Christian Rieschc14f2812011-12-09 09:47:38 +0000595 if (!file) {
596 break;
597 }
wdenk5b1d7132002-11-03 00:07:02 +0000598
Christian Rieschc14f2812011-12-09 09:47:38 +0000599 if (sep) {
600 *sep = ':';
601 file = sep + 1;
602 } else {
603 file = NULL;
604 }
wdenk5b1d7132002-11-03 00:07:02 +0000605 }
wdenk5b1d7132002-11-03 00:07:02 +0000606
Christian Rieschc14f2812011-12-09 09:47:38 +0000607 file = params.datafile;
wdenk5b1d7132002-11-03 00:07:02 +0000608
Christian Rieschc14f2812011-12-09 09:47:38 +0000609 for (;;) {
610 char *sep = strchr(file, ':');
611 if (sep) {
612 *sep = '\0';
613 copy_file (ifd, file, 1);
614 *sep++ = ':';
615 file = sep;
616 } else {
617 copy_file (ifd, file, 0);
618 break;
619 }
wdenk5b1d7132002-11-03 00:07:02 +0000620 }
Shaohui Xieea65fd82012-08-10 02:49:35 +0000621 } else if (params.type == IH_TYPE_PBLIMAGE) {
622 /* PBL has special Image format, implements its' own */
623 pbl_load_uboot(ifd, &params);
Alexander Graf5329d672018-04-13 14:18:52 +0200624 } else if (params.type == IH_TYPE_ZYNQMPBIF) {
625 /* Image file is meta, walk through actual targets */
626 int ret;
627
628 ret = zynqmpbif_copy_image(ifd, &params);
629 if (ret)
630 return ret;
Peng Fan60a56072018-10-16 04:50:30 +0000631 } else if (params.type == IH_TYPE_IMX8IMAGE) {
632 /* i.MX8/8X has special Image format */
633 int ret;
634
635 ret = imx8image_copy_image(ifd, &params);
636 if (ret)
637 return ret;
Peng Fan31c51da2018-11-20 10:19:36 +0000638 } else if (params.type == IH_TYPE_IMX8MIMAGE) {
639 /* i.MX8M has special Image format */
640 int ret;
641
642 ret = imx8mimage_copy_image(ifd, &params);
643 if (ret)
644 return ret;
Jeffy Chenec0d2ea2019-12-27 11:24:41 +0800645 } else if ((params.type == IH_TYPE_RKSD) ||
646 (params.type == IH_TYPE_RKSPI)) {
647 /* Rockchip has special Image format */
648 int ret;
649
650 ret = rockchip_copy_image(ifd, &params);
651 if (ret)
652 return ret;
Christian Rieschc14f2812011-12-09 09:47:38 +0000653 } else {
Stefano Babic6531bd92013-08-19 19:03:19 +0200654 copy_file(ifd, params.datafile, pad_len);
wdenk5b1d7132002-11-03 00:07:02 +0000655 }
Sven Ebenfeld59697a22016-11-06 16:37:56 +0100656 if (params.type == IH_TYPE_FIRMWARE_IVT) {
657 /* Add alignment and IVT */
Kever Yang86f26012020-03-30 11:56:19 +0800658 uint32_t aligned_filesize = ALIGN(params.file_size,
659 0x1000);
Sven Ebenfeld59697a22016-11-06 16:37:56 +0100660 flash_header_v2_t ivt_header = { { 0xd1, 0x2000, 0x40 },
661 params.addr, 0, 0, 0, params.addr
662 + aligned_filesize
663 - tparams->header_size,
664 params.addr + aligned_filesize
665 - tparams->header_size
666 + 0x20, 0 };
667 int i = params.file_size;
668 for (; i < aligned_filesize; i++) {
Sven Ebenfeld3bb71e02017-01-17 19:36:51 +0100669 if (write(ifd, (char *) &i, 1) != 1) {
Sven Ebenfeld59697a22016-11-06 16:37:56 +0100670 fprintf(stderr,
671 "%s: Write error on %s: %s\n",
672 params.cmdname,
673 params.imagefile,
674 strerror(errno));
675 exit(EXIT_FAILURE);
676 }
677 }
678 if (write(ifd, &ivt_header, sizeof(flash_header_v2_t))
679 != sizeof(flash_header_v2_t)) {
680 fprintf(stderr, "%s: Write error on %s: %s\n",
681 params.cmdname,
682 params.imagefile,
683 strerror(errno));
684 exit(EXIT_FAILURE);
685 }
686 }
wdenk5b1d7132002-11-03 00:07:02 +0000687 }
688
689 /* We're a bit of paranoid */
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530690#if defined(_POSIX_SYNCHRONIZED_IO) && \
691 !defined(__sun__) && \
692 !defined(__FreeBSD__) && \
Luka Perkov98c689b2014-08-09 18:17:47 +0200693 !defined(__OpenBSD__) && \
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530694 !defined(__APPLE__)
wdenk5b1d7132002-11-03 00:07:02 +0000695 (void) fdatasync (ifd);
696#else
697 (void) fsync (ifd);
698#endif
699
700 if (fstat(ifd, &sbuf) < 0) {
701 fprintf (stderr, "%s: Can't stat %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530702 params.cmdname, params.imagefile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000703 exit (EXIT_FAILURE);
704 }
Simon Glass139651f2015-06-23 15:39:12 -0600705 params.file_size = sbuf.st_size;
wdenk5b1d7132002-11-03 00:07:02 +0000706
Mark Tomlinsonfc497862018-08-29 10:51:14 +1200707 map_len = sbuf.st_size;
708 ptr = mmap(0, map_len, PROT_READ | PROT_WRITE, MAP_SHARED, ifd, 0);
Mike Frysingercd7e1a32008-05-01 04:13:05 -0400709 if (ptr == MAP_FAILED) {
wdenk5b1d7132002-11-03 00:07:02 +0000710 fprintf (stderr, "%s: Can't map %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530711 params.cmdname, params.imagefile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000712 exit (EXIT_FAILURE);
713 }
714
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530715 /* Setup the image header as per input image type*/
716 if (tparams->set_header)
717 tparams->set_header (ptr, &sbuf, ifd, &params);
718 else {
719 fprintf (stderr, "%s: Can't set header for %s: %s\n",
720 params.cmdname, tparams->name, strerror(errno));
721 exit (EXIT_FAILURE);
722 }
wdenk5b1d7132002-11-03 00:07:02 +0000723
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530724 /* Print the image information by processing image header */
725 if (tparams->print_header)
726 tparams->print_header (ptr);
727 else {
Guillaume GARDETdfe25fa2018-03-30 10:28:19 +0200728 fprintf (stderr, "%s: Can't print header for %s\n",
729 params.cmdname, tparams->name);
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530730 }
wdenk5b1d7132002-11-03 00:07:02 +0000731
Mark Tomlinsonfc497862018-08-29 10:51:14 +1200732 (void)munmap((void *)ptr, map_len);
wdenk5b1d7132002-11-03 00:07:02 +0000733
734 /* We're a bit of paranoid */
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530735#if defined(_POSIX_SYNCHRONIZED_IO) && \
736 !defined(__sun__) && \
737 !defined(__FreeBSD__) && \
Luka Perkov98c689b2014-08-09 18:17:47 +0200738 !defined(__OpenBSD__) && \
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530739 !defined(__APPLE__)
wdenk5b1d7132002-11-03 00:07:02 +0000740 (void) fdatasync (ifd);
741#else
742 (void) fsync (ifd);
743#endif
744
745 if (close(ifd)) {
746 fprintf (stderr, "%s: Write error on %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530747 params.cmdname, params.imagefile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000748 exit (EXIT_FAILURE);
749 }
750
Pali Rohára55eaa52022-01-14 18:34:43 +0100751 if (tparams->verify_header)
752 verify_image(tparams);
753
wdenk5b1d7132002-11-03 00:07:02 +0000754 exit (EXIT_SUCCESS);
755}
756
757static void
758copy_file (int ifd, const char *datafile, int pad)
759{
760 int dfd;
761 struct stat sbuf;
762 unsigned char *ptr;
763 int tail;
764 int zero = 0;
Stefano Babic6531bd92013-08-19 19:03:19 +0200765 uint8_t zeros[4096];
wdenk5b1d7132002-11-03 00:07:02 +0000766 int offset = 0;
Mylène Josserand9e0e9a52020-07-08 11:52:50 +0200767 int size, ret;
Guilherme Maciel Ferreira28be1cf2015-01-15 02:48:07 -0200768 struct image_type_params *tparams = imagetool_get_type(params.type);
wdenk5b1d7132002-11-03 00:07:02 +0000769
Stefano Babic6531bd92013-08-19 19:03:19 +0200770 memset(zeros, 0, sizeof(zeros));
771
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530772 if (params.vflag) {
wdenk5b1d7132002-11-03 00:07:02 +0000773 fprintf (stderr, "Adding Image %s\n", datafile);
774 }
775
wdenk1ee550e2003-10-08 22:14:02 +0000776 if ((dfd = open(datafile, O_RDONLY|O_BINARY)) < 0) {
wdenk5b1d7132002-11-03 00:07:02 +0000777 fprintf (stderr, "%s: Can't open %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530778 params.cmdname, datafile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000779 exit (EXIT_FAILURE);
780 }
781
782 if (fstat(dfd, &sbuf) < 0) {
783 fprintf (stderr, "%s: Can't stat %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530784 params.cmdname, datafile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000785 exit (EXIT_FAILURE);
786 }
787
Thomas Hebb13895a32021-08-01 15:23:13 -0700788 if (sbuf.st_size == 0) {
789 fprintf (stderr, "%s: Input file %s is empty, bailing out\n",
790 params.cmdname, datafile);
791 exit (EXIT_FAILURE);
792 }
793
Mike Frysingercd7e1a32008-05-01 04:13:05 -0400794 ptr = mmap(0, sbuf.st_size, PROT_READ, MAP_SHARED, dfd, 0);
795 if (ptr == MAP_FAILED) {
wdenk5b1d7132002-11-03 00:07:02 +0000796 fprintf (stderr, "%s: Can't read %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530797 params.cmdname, datafile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000798 exit (EXIT_FAILURE);
799 }
800
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530801 if (params.xflag) {
wdenk5b1d7132002-11-03 00:07:02 +0000802 unsigned char *p = NULL;
803 /*
804 * XIP: do not append the image_header_t at the
805 * beginning of the file, but consume the space
806 * reserved for it.
807 */
808
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530809 if ((unsigned)sbuf.st_size < tparams->header_size) {
wdenk5b1d7132002-11-03 00:07:02 +0000810 fprintf (stderr,
811 "%s: Bad size: \"%s\" is too small for XIP\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530812 params.cmdname, datafile);
wdenk5b1d7132002-11-03 00:07:02 +0000813 exit (EXIT_FAILURE);
814 }
815
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530816 for (p = ptr; p < ptr + tparams->header_size; p++) {
wdenk5b1d7132002-11-03 00:07:02 +0000817 if ( *p != 0xff ) {
818 fprintf (stderr,
819 "%s: Bad file: \"%s\" has invalid buffer for XIP\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530820 params.cmdname, datafile);
wdenk5b1d7132002-11-03 00:07:02 +0000821 exit (EXIT_FAILURE);
822 }
823 }
824
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530825 offset = tparams->header_size;
wdenk5b1d7132002-11-03 00:07:02 +0000826 }
827
828 size = sbuf.st_size - offset;
Mylène Josserand9e0e9a52020-07-08 11:52:50 +0200829
830 ret = write(ifd, ptr + offset, size);
831 if (ret != size) {
832 if (ret < 0)
833 fprintf (stderr, "%s: Write error on %s: %s\n",
834 params.cmdname, params.imagefile, strerror(errno));
835 else if (ret < size)
836 fprintf (stderr, "%s: Write only %d/%d bytes, "\
837 "probably no space left on the device\n",
838 params.cmdname, ret, size);
wdenk5b1d7132002-11-03 00:07:02 +0000839 exit (EXIT_FAILURE);
840 }
841
Stefano Babic6531bd92013-08-19 19:03:19 +0200842 tail = size % 4;
843 if ((pad == 1) && (tail != 0)) {
wdenk5b1d7132002-11-03 00:07:02 +0000844
845 if (write(ifd, (char *)&zero, 4-tail) != 4-tail) {
846 fprintf (stderr, "%s: Write error on %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530847 params.cmdname, params.imagefile,
848 strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000849 exit (EXIT_FAILURE);
850 }
Stefano Babic6531bd92013-08-19 19:03:19 +0200851 } else if (pad > 1) {
Simon Glass46409cc2015-08-30 16:55:22 -0600852 while (pad > 0) {
853 int todo = sizeof(zeros);
854
855 if (todo > pad)
856 todo = pad;
857 if (write(ifd, (char *)&zeros, todo) != todo) {
858 fprintf(stderr, "%s: Write error on %s: %s\n",
859 params.cmdname, params.imagefile,
860 strerror(errno));
861 exit(EXIT_FAILURE);
862 }
863 pad -= todo;
Stefano Babic6531bd92013-08-19 19:03:19 +0200864 }
wdenk5b1d7132002-11-03 00:07:02 +0000865 }
866
867 (void) munmap((void *)ptr, sbuf.st_size);
868 (void) close (dfd);
869}