blob: 8f8b6df37cd112a6788c21e1d75c9ab401bdb870 [file] [log] [blame]
wdenk5b1d7132002-11-03 00:07:02 +00001/*
Bartlomiej Siekad7816fb2008-03-11 12:34:47 +01002 * (C) Copyright 2008 Semihalf
3 *
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +05304 * (C) Copyright 2000-2009
wdenk5b1d7132002-11-03 00:07:02 +00005 * DENX Software Engineering
6 * Wolfgang Denk, wd@denx.de
wdenkdb82c8e2004-02-26 23:01:04 +00007 *
Wolfgang Denkd79de1d2013-07-08 09:37:19 +02008 * SPDX-License-Identifier: GPL-2.0+
wdenk5b1d7132002-11-03 00:07:02 +00009 */
10
Marian Balakowicz41d71ed2008-01-08 18:14:09 +010011#include "mkimage.h"
wdenk5b1d7132002-11-03 00:07:02 +000012#include <image.h>
Wolfgang Denke97ab722011-02-12 10:37:11 +010013#include <version.h>
wdenk5b1d7132002-11-03 00:07:02 +000014
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +053015static void copy_file(int, const char *, int);
16static void usage(void);
wdenk5b1d7132002-11-03 00:07:02 +000017
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +053018/* parameters initialized by core will be used by the image type code */
Guilherme Maciel Ferreira8ed4d1c2013-12-01 12:43:10 -070019struct image_tool_params params = {
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +053020 .os = IH_OS_LINUX,
21 .arch = IH_ARCH_PPC,
22 .type = IH_TYPE_KERNEL,
23 .comp = IH_COMP_GZIP,
24 .dtc = MKIMAGE_DEFAULT_DTC_OPTIONS,
Wolfgang Denk23416b82010-03-27 23:37:46 +010025 .imagename = "",
Shaohui Xieea65fd82012-08-10 02:49:35 +000026 .imagename2 = "",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +053027};
wdenk5b1d7132002-11-03 00:07:02 +000028
Simon Glass434a31b2015-06-23 15:38:25 -060029static int h_compare_image_name(const void *vtype1, const void *vtype2)
30{
31 const int *type1 = vtype1;
32 const int *type2 = vtype2;
33 const char *name1 = genimg_get_type_short_name(*type1);
34 const char *name2 = genimg_get_type_short_name(*type2);
35
36 return strcmp(name1, name2);
37}
38
39/* Show all image types supported by mkimage */
40static void show_image_types(void)
41{
42 struct image_type_params *tparams;
43 int order[IH_TYPE_COUNT];
44 int count;
45 int type;
46 int i;
47
48 /* Sort the names in order of short name for easier reading */
49 memset(order, '\0', sizeof(order));
50 for (count = 0, type = 0; type < IH_TYPE_COUNT; type++) {
51 tparams = imagetool_get_type(type);
52 if (tparams)
53 order[count++] = type;
54 }
55 qsort(order, count, sizeof(int), h_compare_image_name);
56
57 fprintf(stderr, "\nInvalid image type. Supported image types:\n");
58 for (i = 0; i < count; i++) {
59 type = order[i];
60 tparams = imagetool_get_type(type);
61 if (tparams) {
62 fprintf(stderr, "\t%-15s %s\n",
63 genimg_get_type_short_name(type),
64 genimg_get_type_name(type));
65 }
66 }
67 fprintf(stderr, "\n");
68}
69
70int main(int argc, char **argv)
wdenk5b1d7132002-11-03 00:07:02 +000071{
Bartlomiej Siekad7816fb2008-03-11 12:34:47 +010072 int ifd = -1;
wdenk5b1d7132002-11-03 00:07:02 +000073 struct stat sbuf;
Peter Tyser632d9ca2010-04-04 22:36:03 -050074 char *ptr;
Prafulla Wadaskar4d4d67d2009-08-16 05:28:19 +053075 int retval = 0;
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +053076 struct image_type_params *tparams = NULL;
Stefano Babic6531bd92013-08-19 19:03:19 +020077 int pad_len = 0;
Simon Glass139651f2015-06-23 15:39:12 -060078 int dfd;
wdenk5b1d7132002-11-03 00:07:02 +000079
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +053080 params.cmdname = *argv;
81 params.addr = params.ep = 0;
wdenk5b1d7132002-11-03 00:07:02 +000082
83 while (--argc > 0 && **++argv == '-') {
84 while (*++*argv) {
85 switch (**argv) {
86 case 'l':
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +053087 params.lflag = 1;
wdenk5b1d7132002-11-03 00:07:02 +000088 break;
89 case 'A':
90 if ((--argc <= 0) ||
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +053091 (params.arch =
92 genimg_get_arch_id (*++argv)) < 0)
wdenk5b1d7132002-11-03 00:07:02 +000093 usage ();
94 goto NXTARG;
Simon Glassbf27d3d2013-06-13 15:10:06 -070095 case 'c':
96 if (--argc <= 0)
97 usage();
98 params.comment = *++argv;
99 goto NXTARG;
wdenk5b1d7132002-11-03 00:07:02 +0000100 case 'C':
101 if ((--argc <= 0) ||
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530102 (params.comp =
103 genimg_get_comp_id (*++argv)) < 0)
wdenk5b1d7132002-11-03 00:07:02 +0000104 usage ();
Bartlomiej Siekad7816fb2008-03-11 12:34:47 +0100105 goto NXTARG;
106 case 'D':
107 if (--argc <= 0)
108 usage ();
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530109 params.dtc = *++argv;
wdenk5b1d7132002-11-03 00:07:02 +0000110 goto NXTARG;
Bartlomiej Siekad7816fb2008-03-11 12:34:47 +0100111
wdenk5b1d7132002-11-03 00:07:02 +0000112 case 'O':
113 if ((--argc <= 0) ||
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530114 (params.os =
115 genimg_get_os_id (*++argv)) < 0)
wdenk5b1d7132002-11-03 00:07:02 +0000116 usage ();
117 goto NXTARG;
118 case 'T':
Simon Glass434a31b2015-06-23 15:38:25 -0600119 params.type = -1;
120 if (--argc >= 0 && argv[1]) {
121 params.type =
122 genimg_get_type_id(*++argv);
123 }
124 if (params.type < 0) {
125 show_image_types();
126 usage();
127 }
wdenk5b1d7132002-11-03 00:07:02 +0000128 goto NXTARG;
wdenk5b1d7132002-11-03 00:07:02 +0000129 case 'a':
130 if (--argc <= 0)
131 usage ();
Peter Tyser632d9ca2010-04-04 22:36:03 -0500132 params.addr = strtoul (*++argv, &ptr, 16);
wdenk5b1d7132002-11-03 00:07:02 +0000133 if (*ptr) {
134 fprintf (stderr,
135 "%s: invalid load address %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530136 params.cmdname, *argv);
wdenk5b1d7132002-11-03 00:07:02 +0000137 exit (EXIT_FAILURE);
138 }
139 goto NXTARG;
140 case 'd':
141 if (--argc <= 0)
142 usage ();
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530143 params.datafile = *++argv;
144 params.dflag = 1;
wdenk5b1d7132002-11-03 00:07:02 +0000145 goto NXTARG;
146 case 'e':
147 if (--argc <= 0)
148 usage ();
Peter Tyser632d9ca2010-04-04 22:36:03 -0500149 params.ep = strtoul (*++argv, &ptr, 16);
wdenk5b1d7132002-11-03 00:07:02 +0000150 if (*ptr) {
151 fprintf (stderr,
152 "%s: invalid entry point %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530153 params.cmdname, *argv);
wdenk5b1d7132002-11-03 00:07:02 +0000154 exit (EXIT_FAILURE);
155 }
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530156 params.eflag = 1;
wdenk5b1d7132002-11-03 00:07:02 +0000157 goto NXTARG;
Bartlomiej Siekad7816fb2008-03-11 12:34:47 +0100158 case 'f':
159 if (--argc <= 0)
160 usage ();
Simon Glassce8c3ca2013-06-13 15:10:05 -0700161 params.datafile = *++argv;
162 /* no break */
163 case 'F':
Peter Tyser94734e62009-11-24 16:42:08 -0600164 /*
165 * The flattened image tree (FIT) format
166 * requires a flattened device tree image type
167 */
168 params.type = IH_TYPE_FLATDT;
Peter Tyser54cd5392009-12-06 01:33:24 -0600169 params.fflag = 1;
Bartlomiej Siekad7816fb2008-03-11 12:34:47 +0100170 goto NXTARG;
Simon Glassb1489742013-06-13 15:10:03 -0700171 case 'k':
172 if (--argc <= 0)
173 usage();
174 params.keydir = *++argv;
175 goto NXTARG;
Simon Glassb4d8b092013-06-13 15:10:04 -0700176 case 'K':
177 if (--argc <= 0)
178 usage();
179 params.keydest = *++argv;
180 goto NXTARG;
wdenk5b1d7132002-11-03 00:07:02 +0000181 case 'n':
182 if (--argc <= 0)
183 usage ();
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530184 params.imagename = *++argv;
wdenk5b1d7132002-11-03 00:07:02 +0000185 goto NXTARG;
Simon Glass817278a2013-06-13 15:10:07 -0700186 case 'r':
187 params.require_keys = 1;
188 break;
Shaohui Xieea65fd82012-08-10 02:49:35 +0000189 case 'R':
190 if (--argc <= 0)
191 usage();
192 /*
193 * This entry is for the second configuration
194 * file, if only one is not enough.
195 */
196 params.imagename2 = *++argv;
197 goto NXTARG;
Stefano Babic4a963302011-09-15 23:50:16 +0000198 case 's':
199 params.skipcpy = 1;
200 break;
wdenk5b1d7132002-11-03 00:07:02 +0000201 case 'v':
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530202 params.vflag++;
wdenk5b1d7132002-11-03 00:07:02 +0000203 break;
Wolfgang Denke97ab722011-02-12 10:37:11 +0100204 case 'V':
205 printf("mkimage version %s\n", PLAIN_VERSION);
206 exit(EXIT_SUCCESS);
wdenk5b1d7132002-11-03 00:07:02 +0000207 case 'x':
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530208 params.xflag++;
wdenk5b1d7132002-11-03 00:07:02 +0000209 break;
210 default:
211 usage ();
212 }
213 }
214NXTARG: ;
215 }
216
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530217 if (argc != 1)
218 usage ();
219
220 /* set tparams as per input type_id */
Guilherme Maciel Ferreira28be1cf2015-01-15 02:48:07 -0200221 tparams = imagetool_get_type(params.type);
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530222 if (tparams == NULL) {
223 fprintf (stderr, "%s: unsupported type %s\n",
224 params.cmdname, genimg_get_type_name(params.type));
225 exit (EXIT_FAILURE);
226 }
227
228 /*
229 * check the passed arguments parameters meets the requirements
230 * as per image type to be generated/listed
231 */
232 if (tparams->check_params)
233 if (tparams->check_params (&params))
234 usage ();
wdenk5b1d7132002-11-03 00:07:02 +0000235
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530236 if (!params.eflag) {
237 params.ep = params.addr;
wdenk5b1d7132002-11-03 00:07:02 +0000238 /* If XIP, entry point must be after the U-Boot header */
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530239 if (params.xflag)
240 params.ep += tparams->header_size;
wdenk5b1d7132002-11-03 00:07:02 +0000241 }
242
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530243 params.imagefile = *argv;
wdenk5b1d7132002-11-03 00:07:02 +0000244
Peter Tyserb2f8b942009-11-24 16:42:10 -0600245 if (params.fflag){
246 if (tparams->fflag_handle)
247 /*
248 * in some cases, some additional processing needs
249 * to be done if fflag is defined
250 *
251 * For ex. fit_handle_file for Fit file support
252 */
253 retval = tparams->fflag_handle(&params);
wdenk5b1d7132002-11-03 00:07:02 +0000254
Peter Tyserb2f8b942009-11-24 16:42:10 -0600255 if (retval != EXIT_SUCCESS)
256 exit (retval);
wdenk5b1d7132002-11-03 00:07:02 +0000257 }
258
Peter Tyserb2f8b942009-11-24 16:42:10 -0600259 if (params.lflag || params.fflag) {
260 ifd = open (params.imagefile, O_RDONLY|O_BINARY);
261 } else {
262 ifd = open (params.imagefile,
263 O_RDWR|O_CREAT|O_TRUNC|O_BINARY, 0666);
264 }
265
266 if (ifd < 0) {
267 fprintf (stderr, "%s: Can't open %s: %s\n",
268 params.cmdname, params.imagefile,
269 strerror(errno));
270 exit (EXIT_FAILURE);
271 }
272
273 if (params.lflag || params.fflag) {
wdenk5b1d7132002-11-03 00:07:02 +0000274 /*
275 * list header information of existing image
276 */
277 if (fstat(ifd, &sbuf) < 0) {
278 fprintf (stderr, "%s: Can't stat %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530279 params.cmdname, params.imagefile,
280 strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000281 exit (EXIT_FAILURE);
282 }
283
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530284 if ((unsigned)sbuf.st_size < tparams->header_size) {
wdenk5b1d7132002-11-03 00:07:02 +0000285 fprintf (stderr,
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530286 "%s: Bad size: \"%s\" is not valid image\n",
287 params.cmdname, params.imagefile);
wdenk5b1d7132002-11-03 00:07:02 +0000288 exit (EXIT_FAILURE);
289 }
290
Mike Frysingercd7e1a32008-05-01 04:13:05 -0400291 ptr = mmap(0, sbuf.st_size, PROT_READ, MAP_SHARED, ifd, 0);
292 if (ptr == MAP_FAILED) {
wdenk5b1d7132002-11-03 00:07:02 +0000293 fprintf (stderr, "%s: Can't read %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530294 params.cmdname, params.imagefile,
295 strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000296 exit (EXIT_FAILURE);
297 }
298
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530299 /*
300 * scan through mkimage registry for all supported image types
301 * and verify the input image file header for match
302 * Print the image information for matched image type
303 * Returns the error code if not matched
304 */
Guilherme Maciel Ferreira0b338032015-01-15 02:48:05 -0200305 retval = imagetool_verify_print_header(ptr, &sbuf,
306 tparams, &params);
wdenk5b1d7132002-11-03 00:07:02 +0000307
wdenk5b1d7132002-11-03 00:07:02 +0000308 (void) munmap((void *)ptr, sbuf.st_size);
309 (void) close (ifd);
310
Prafulla Wadaskar139cb082009-08-10 18:49:37 +0530311 exit (retval);
wdenk5b1d7132002-11-03 00:07:02 +0000312 }
313
Marek Vasut4cc86272015-12-07 18:01:54 +0100314 if ((params.type != IH_TYPE_MULTI) && (params.type != IH_TYPE_SCRIPT)) {
Philippe De Swert3edd0412015-12-04 00:11:23 +0200315 dfd = open(params.datafile, O_RDONLY | O_BINARY);
316 if (dfd < 0) {
317 fprintf(stderr, "%s: Can't open %s: %s\n",
318 params.cmdname, params.datafile,
319 strerror(errno));
320 exit(EXIT_FAILURE);
321 }
Simon Glass139651f2015-06-23 15:39:12 -0600322
Philippe De Swert3edd0412015-12-04 00:11:23 +0200323 if (fstat(dfd, &sbuf) < 0) {
324 fprintf(stderr, "%s: Can't stat %s: %s\n",
325 params.cmdname, params.datafile,
326 strerror(errno));
327 exit(EXIT_FAILURE);
328 }
Simon Glass139651f2015-06-23 15:39:12 -0600329
Philippe De Swert3edd0412015-12-04 00:11:23 +0200330 params.file_size = sbuf.st_size + tparams->header_size;
331 close(dfd);
332 }
Simon Glass139651f2015-06-23 15:39:12 -0600333
wdenk5b1d7132002-11-03 00:07:02 +0000334 /*
Stefano Babic4a963302011-09-15 23:50:16 +0000335 * In case there an header with a variable
336 * length will be added, the corresponding
337 * function is called. This is responsible to
338 * allocate memory for the header itself.
wdenk5b1d7132002-11-03 00:07:02 +0000339 */
Stefano Babic4a963302011-09-15 23:50:16 +0000340 if (tparams->vrec_header)
Stefano Babic6531bd92013-08-19 19:03:19 +0200341 pad_len = tparams->vrec_header(&params, tparams);
Stefano Babic4a963302011-09-15 23:50:16 +0000342 else
343 memset(tparams->hdr, 0, tparams->header_size);
wdenk5b1d7132002-11-03 00:07:02 +0000344
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530345 if (write(ifd, tparams->hdr, tparams->header_size)
346 != tparams->header_size) {
wdenk5b1d7132002-11-03 00:07:02 +0000347 fprintf (stderr, "%s: Write error on %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530348 params.cmdname, params.imagefile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000349 exit (EXIT_FAILURE);
350 }
351
Christian Rieschc14f2812011-12-09 09:47:38 +0000352 if (!params.skipcpy) {
353 if (params.type == IH_TYPE_MULTI ||
354 params.type == IH_TYPE_SCRIPT) {
355 char *file = params.datafile;
356 uint32_t size;
wdenk5b1d7132002-11-03 00:07:02 +0000357
Christian Rieschc14f2812011-12-09 09:47:38 +0000358 for (;;) {
359 char *sep = NULL;
wdenk5b1d7132002-11-03 00:07:02 +0000360
Christian Rieschc14f2812011-12-09 09:47:38 +0000361 if (file) {
362 if ((sep = strchr(file, ':')) != NULL) {
363 *sep = '\0';
364 }
365
366 if (stat (file, &sbuf) < 0) {
367 fprintf (stderr, "%s: Can't stat %s: %s\n",
368 params.cmdname, file, strerror(errno));
369 exit (EXIT_FAILURE);
370 }
371 size = cpu_to_uimage (sbuf.st_size);
372 } else {
373 size = 0;
wdenk5b1d7132002-11-03 00:07:02 +0000374 }
375
Christian Rieschc14f2812011-12-09 09:47:38 +0000376 if (write(ifd, (char *)&size, sizeof(size)) != sizeof(size)) {
377 fprintf (stderr, "%s: Write error on %s: %s\n",
378 params.cmdname, params.imagefile,
379 strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000380 exit (EXIT_FAILURE);
381 }
wdenk5b1d7132002-11-03 00:07:02 +0000382
Christian Rieschc14f2812011-12-09 09:47:38 +0000383 if (!file) {
384 break;
385 }
wdenk5b1d7132002-11-03 00:07:02 +0000386
Christian Rieschc14f2812011-12-09 09:47:38 +0000387 if (sep) {
388 *sep = ':';
389 file = sep + 1;
390 } else {
391 file = NULL;
392 }
wdenk5b1d7132002-11-03 00:07:02 +0000393 }
wdenk5b1d7132002-11-03 00:07:02 +0000394
Christian Rieschc14f2812011-12-09 09:47:38 +0000395 file = params.datafile;
wdenk5b1d7132002-11-03 00:07:02 +0000396
Christian Rieschc14f2812011-12-09 09:47:38 +0000397 for (;;) {
398 char *sep = strchr(file, ':');
399 if (sep) {
400 *sep = '\0';
401 copy_file (ifd, file, 1);
402 *sep++ = ':';
403 file = sep;
404 } else {
405 copy_file (ifd, file, 0);
406 break;
407 }
wdenk5b1d7132002-11-03 00:07:02 +0000408 }
Shaohui Xieea65fd82012-08-10 02:49:35 +0000409 } else if (params.type == IH_TYPE_PBLIMAGE) {
410 /* PBL has special Image format, implements its' own */
411 pbl_load_uboot(ifd, &params);
Christian Rieschc14f2812011-12-09 09:47:38 +0000412 } else {
Stefano Babic6531bd92013-08-19 19:03:19 +0200413 copy_file(ifd, params.datafile, pad_len);
wdenk5b1d7132002-11-03 00:07:02 +0000414 }
wdenk5b1d7132002-11-03 00:07:02 +0000415 }
416
417 /* We're a bit of paranoid */
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530418#if defined(_POSIX_SYNCHRONIZED_IO) && \
419 !defined(__sun__) && \
420 !defined(__FreeBSD__) && \
Luka Perkov98c689b2014-08-09 18:17:47 +0200421 !defined(__OpenBSD__) && \
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530422 !defined(__APPLE__)
wdenk5b1d7132002-11-03 00:07:02 +0000423 (void) fdatasync (ifd);
424#else
425 (void) fsync (ifd);
426#endif
427
428 if (fstat(ifd, &sbuf) < 0) {
429 fprintf (stderr, "%s: Can't stat %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530430 params.cmdname, params.imagefile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000431 exit (EXIT_FAILURE);
432 }
Simon Glass139651f2015-06-23 15:39:12 -0600433 params.file_size = sbuf.st_size;
wdenk5b1d7132002-11-03 00:07:02 +0000434
Mike Frysingercd7e1a32008-05-01 04:13:05 -0400435 ptr = mmap(0, sbuf.st_size, PROT_READ|PROT_WRITE, MAP_SHARED, ifd, 0);
436 if (ptr == MAP_FAILED) {
wdenk5b1d7132002-11-03 00:07:02 +0000437 fprintf (stderr, "%s: Can't map %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530438 params.cmdname, params.imagefile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000439 exit (EXIT_FAILURE);
440 }
441
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530442 /* Setup the image header as per input image type*/
443 if (tparams->set_header)
444 tparams->set_header (ptr, &sbuf, ifd, &params);
445 else {
446 fprintf (stderr, "%s: Can't set header for %s: %s\n",
447 params.cmdname, tparams->name, strerror(errno));
448 exit (EXIT_FAILURE);
449 }
wdenk5b1d7132002-11-03 00:07:02 +0000450
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530451 /* Print the image information by processing image header */
452 if (tparams->print_header)
453 tparams->print_header (ptr);
454 else {
455 fprintf (stderr, "%s: Can't print header for %s: %s\n",
456 params.cmdname, tparams->name, strerror(errno));
457 exit (EXIT_FAILURE);
458 }
wdenk5b1d7132002-11-03 00:07:02 +0000459
460 (void) munmap((void *)ptr, sbuf.st_size);
461
462 /* We're a bit of paranoid */
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530463#if defined(_POSIX_SYNCHRONIZED_IO) && \
464 !defined(__sun__) && \
465 !defined(__FreeBSD__) && \
Luka Perkov98c689b2014-08-09 18:17:47 +0200466 !defined(__OpenBSD__) && \
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530467 !defined(__APPLE__)
wdenk5b1d7132002-11-03 00:07:02 +0000468 (void) fdatasync (ifd);
469#else
470 (void) fsync (ifd);
471#endif
472
473 if (close(ifd)) {
474 fprintf (stderr, "%s: Write error on %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530475 params.cmdname, params.imagefile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000476 exit (EXIT_FAILURE);
477 }
478
479 exit (EXIT_SUCCESS);
480}
481
482static void
483copy_file (int ifd, const char *datafile, int pad)
484{
485 int dfd;
486 struct stat sbuf;
487 unsigned char *ptr;
488 int tail;
489 int zero = 0;
Stefano Babic6531bd92013-08-19 19:03:19 +0200490 uint8_t zeros[4096];
wdenk5b1d7132002-11-03 00:07:02 +0000491 int offset = 0;
492 int size;
Guilherme Maciel Ferreira28be1cf2015-01-15 02:48:07 -0200493 struct image_type_params *tparams = imagetool_get_type(params.type);
wdenk5b1d7132002-11-03 00:07:02 +0000494
Stefano Babic6531bd92013-08-19 19:03:19 +0200495 memset(zeros, 0, sizeof(zeros));
496
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530497 if (params.vflag) {
wdenk5b1d7132002-11-03 00:07:02 +0000498 fprintf (stderr, "Adding Image %s\n", datafile);
499 }
500
wdenk1ee550e2003-10-08 22:14:02 +0000501 if ((dfd = open(datafile, O_RDONLY|O_BINARY)) < 0) {
wdenk5b1d7132002-11-03 00:07:02 +0000502 fprintf (stderr, "%s: Can't open %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530503 params.cmdname, datafile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000504 exit (EXIT_FAILURE);
505 }
506
507 if (fstat(dfd, &sbuf) < 0) {
508 fprintf (stderr, "%s: Can't stat %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530509 params.cmdname, datafile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000510 exit (EXIT_FAILURE);
511 }
512
Mike Frysingercd7e1a32008-05-01 04:13:05 -0400513 ptr = mmap(0, sbuf.st_size, PROT_READ, MAP_SHARED, dfd, 0);
514 if (ptr == MAP_FAILED) {
wdenk5b1d7132002-11-03 00:07:02 +0000515 fprintf (stderr, "%s: Can't read %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530516 params.cmdname, datafile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000517 exit (EXIT_FAILURE);
518 }
519
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530520 if (params.xflag) {
wdenk5b1d7132002-11-03 00:07:02 +0000521 unsigned char *p = NULL;
522 /*
523 * XIP: do not append the image_header_t at the
524 * beginning of the file, but consume the space
525 * reserved for it.
526 */
527
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530528 if ((unsigned)sbuf.st_size < tparams->header_size) {
wdenk5b1d7132002-11-03 00:07:02 +0000529 fprintf (stderr,
530 "%s: Bad size: \"%s\" is too small for XIP\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530531 params.cmdname, datafile);
wdenk5b1d7132002-11-03 00:07:02 +0000532 exit (EXIT_FAILURE);
533 }
534
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530535 for (p = ptr; p < ptr + tparams->header_size; p++) {
wdenk5b1d7132002-11-03 00:07:02 +0000536 if ( *p != 0xff ) {
537 fprintf (stderr,
538 "%s: Bad file: \"%s\" has invalid buffer for XIP\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530539 params.cmdname, datafile);
wdenk5b1d7132002-11-03 00:07:02 +0000540 exit (EXIT_FAILURE);
541 }
542 }
543
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530544 offset = tparams->header_size;
wdenk5b1d7132002-11-03 00:07:02 +0000545 }
546
547 size = sbuf.st_size - offset;
548 if (write(ifd, ptr + offset, size) != size) {
549 fprintf (stderr, "%s: Write error on %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530550 params.cmdname, params.imagefile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000551 exit (EXIT_FAILURE);
552 }
553
Stefano Babic6531bd92013-08-19 19:03:19 +0200554 tail = size % 4;
555 if ((pad == 1) && (tail != 0)) {
wdenk5b1d7132002-11-03 00:07:02 +0000556
557 if (write(ifd, (char *)&zero, 4-tail) != 4-tail) {
558 fprintf (stderr, "%s: Write error on %s: %s\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530559 params.cmdname, params.imagefile,
560 strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000561 exit (EXIT_FAILURE);
562 }
Stefano Babic6531bd92013-08-19 19:03:19 +0200563 } else if (pad > 1) {
Simon Glass46409cc2015-08-30 16:55:22 -0600564 while (pad > 0) {
565 int todo = sizeof(zeros);
566
567 if (todo > pad)
568 todo = pad;
569 if (write(ifd, (char *)&zeros, todo) != todo) {
570 fprintf(stderr, "%s: Write error on %s: %s\n",
571 params.cmdname, params.imagefile,
572 strerror(errno));
573 exit(EXIT_FAILURE);
574 }
575 pad -= todo;
Stefano Babic6531bd92013-08-19 19:03:19 +0200576 }
wdenk5b1d7132002-11-03 00:07:02 +0000577 }
578
579 (void) munmap((void *)ptr, sbuf.st_size);
580 (void) close (dfd);
581}
582
Guilherme Maciel Ferreiraff0fa612013-12-01 12:43:09 -0700583static void usage(void)
wdenk5b1d7132002-11-03 00:07:02 +0000584{
585 fprintf (stderr, "Usage: %s -l image\n"
Bartlomiej Siekad7816fb2008-03-11 12:34:47 +0100586 " -l ==> list image header information\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530587 params.cmdname);
Bartlomiej Siekad7816fb2008-03-11 12:34:47 +0100588 fprintf (stderr, " %s [-x] -A arch -O os -T type -C comp "
589 "-a addr -e ep -n name -d data_file[:data_file...] image\n"
590 " -A ==> set architecture to 'arch'\n"
wdenk5b1d7132002-11-03 00:07:02 +0000591 " -O ==> set operating system to 'os'\n"
592 " -T ==> set image type to 'type'\n"
593 " -C ==> set compression type 'comp'\n"
594 " -a ==> set load address to 'addr' (hex)\n"
595 " -e ==> set entry point to 'ep' (hex)\n"
596 " -n ==> set image name to 'name'\n"
597 " -d ==> use image data from 'datafile'\n"
Bartlomiej Siekad7816fb2008-03-11 12:34:47 +0100598 " -x ==> set XIP (execute in place)\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530599 params.cmdname);
Simon Glassce8c3ca2013-06-13 15:10:05 -0700600 fprintf(stderr, " %s [-D dtc_options] [-f fit-image.its|-F] fit-image\n",
Prafulla Wadaskarfabf3cf2009-08-19 17:36:46 +0530601 params.cmdname);
Tom Rinie140a462015-08-26 15:21:23 -0400602 fprintf(stderr, " -D => set all options for device tree compiler\n"
Simon Glassb1489742013-06-13 15:10:03 -0700603 " -f => input filename for FIT source\n");
604#ifdef CONFIG_FIT_SIGNATURE
Simon Glass817278a2013-06-13 15:10:07 -0700605 fprintf(stderr, "Signing / verified boot options: [-k keydir] [-K dtb] [ -c <comment>] [-r]\n"
Simon Glassb4d8b092013-06-13 15:10:04 -0700606 " -k => set directory containing private keys\n"
Simon Glassce8c3ca2013-06-13 15:10:05 -0700607 " -K => write public keys to this .dtb file\n"
Simon Glassbf27d3d2013-06-13 15:10:06 -0700608 " -c => add comment in signature node\n"
Simon Glass817278a2013-06-13 15:10:07 -0700609 " -F => re-sign existing FIT image\n"
610 " -r => mark keys used as 'required' in dtb\n");
Simon Glassb1489742013-06-13 15:10:03 -0700611#else
612 fprintf(stderr, "Signing / verified boot not supported (CONFIG_FIT_SIGNATURE undefined)\n");
613#endif
Wolfgang Denke97ab722011-02-12 10:37:11 +0100614 fprintf (stderr, " %s -V ==> print version information and exit\n",
615 params.cmdname);
Simon Glass434a31b2015-06-23 15:38:25 -0600616 fprintf(stderr, "Use -T to see a list of available image types\n");
Bartlomiej Siekad7816fb2008-03-11 12:34:47 +0100617
wdenk5b1d7132002-11-03 00:07:02 +0000618 exit (EXIT_FAILURE);
619}