blob: 123d0c7d938dae609c6eea050cbdbc5a6ce96075 [file] [log] [blame]
wdenk5b1d7132002-11-03 00:07:02 +00001/*
Bartlomiej Sieka9d254382008-03-11 12:34:47 +01002 * (C) Copyright 2008 Semihalf
3 *
Prafulla Wadaskar89a4d6b2009-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
wdenk0c852a22004-02-26 23:01:04 +00007 *
Wolfgang Denk1a459662013-07-08 09:37:19 +02008 * SPDX-License-Identifier: GPL-2.0+
wdenk5b1d7132002-11-03 00:07:02 +00009 */
10
Marian Balakowiczb97a2a02008-01-08 18:14:09 +010011#include "mkimage.h"
wdenk5b1d7132002-11-03 00:07:02 +000012#include <image.h>
Wolfgang Denk976b38c2011-02-12 10:37:11 +010013#include <version.h>
wdenk5b1d7132002-11-03 00:07:02 +000014
Prafulla Wadaskar89a4d6b2009-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 Wadaskar89a4d6b2009-08-19 17:36:46 +053018/* image_type_params link list to maintain registered image type supports */
19struct image_type_params *mkimage_tparams = NULL;
wdenk5b1d7132002-11-03 00:07:02 +000020
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053021/* parameters initialized by core will be used by the image type code */
Guilherme Maciel Ferreiraf86ed6a2013-12-01 12:43:10 -070022struct image_tool_params params = {
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053023 .os = IH_OS_LINUX,
24 .arch = IH_ARCH_PPC,
25 .type = IH_TYPE_KERNEL,
26 .comp = IH_COMP_GZIP,
27 .dtc = MKIMAGE_DEFAULT_DTC_OPTIONS,
Wolfgang Denk04387d22010-03-27 23:37:46 +010028 .imagename = "",
Shaohui Xie5d898a02012-08-10 02:49:35 +000029 .imagename2 = "",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053030};
wdenk5b1d7132002-11-03 00:07:02 +000031
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +053032/*
33 * mkimage_register -
34 *
35 * It is used to register respective image generation/list support to the
36 * mkimage core
37 *
38 * the input struct image_type_params is checked and appended to the link
39 * list, if the input structure is already registered, error
40 */
41void mkimage_register (struct image_type_params *tparams)
42{
43 struct image_type_params **tp;
44
45 if (!tparams) {
46 fprintf (stderr, "%s: %s: Null input\n",
47 params.cmdname, __FUNCTION__);
48 exit (EXIT_FAILURE);
49 }
50
51 /* scan the linked list, check for registry and point the last one */
52 for (tp = &mkimage_tparams; *tp != NULL; tp = &(*tp)->next) {
53 if (!strcmp((*tp)->name, tparams->name)) {
54 fprintf (stderr, "%s: %s already registered\n",
55 params.cmdname, tparams->name);
56 return;
57 }
58 }
59
60 /* add input struct entry at the end of link list */
61 *tp = tparams;
62 /* mark input entry as last entry in the link list */
63 tparams->next = NULL;
64
65 debug ("Registered %s\n", tparams->name);
66}
67
68/*
69 * mkimage_get_type -
70 *
71 * It scans all registers image type supports
72 * checks the input type_id for each supported image type
73 *
74 * if successful,
75 * returns respective image_type_params pointer if success
76 * if input type_id is not supported by any of image_type_support
77 * returns NULL
78 */
79struct image_type_params *mkimage_get_type(int type)
80{
81 struct image_type_params *curr;
82
83 for (curr = mkimage_tparams; curr != NULL; curr = curr->next) {
84 if (curr->check_image_type) {
85 if (!curr->check_image_type (type))
86 return curr;
87 }
88 }
89 return NULL;
90}
91
92/*
93 * mkimage_verify_print_header -
94 *
95 * It scans mkimage_tparams link list,
96 * verifies image_header for each supported image type
97 * if verification is successful, prints respective header
98 *
99 * returns negative if input image format does not match with any of
100 * supported image types
101 */
102int mkimage_verify_print_header (void *ptr, struct stat *sbuf)
103{
104 int retval = -1;
105 struct image_type_params *curr;
106
107 for (curr = mkimage_tparams; curr != NULL; curr = curr->next ) {
108 if (curr->verify_header) {
109 retval = curr->verify_header (
110 (unsigned char *)ptr, sbuf->st_size,
111 &params);
112
113 if (retval == 0) {
114 /*
115 * Print the image information
116 * if verify is successful
117 */
118 if (curr->print_header)
119 curr->print_header (ptr);
120 else {
121 fprintf (stderr,
122 "%s: print_header undefined for %s\n",
123 params.cmdname, curr->name);
124 }
125 break;
126 }
127 }
128 }
129 return retval;
130}
wdenk5b1d7132002-11-03 00:07:02 +0000131
132int
133main (int argc, char **argv)
134{
Bartlomiej Sieka9d254382008-03-11 12:34:47 +0100135 int ifd = -1;
wdenk5b1d7132002-11-03 00:07:02 +0000136 struct stat sbuf;
Peter Tysera2513e22010-04-04 22:36:03 -0500137 char *ptr;
Prafulla Wadaskar449609f2009-08-16 05:28:19 +0530138 int retval = 0;
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530139 struct image_type_params *tparams = NULL;
Stefano Babic9bac0bb2013-08-19 19:03:19 +0200140 int pad_len = 0;
wdenk5b1d7132002-11-03 00:07:02 +0000141
Guilherme Maciel Ferreiraf86ed6a2013-12-01 12:43:10 -0700142 /* Init all image generation/list support */
143 register_image_tool(mkimage_register);
wdenk5b1d7132002-11-03 00:07:02 +0000144
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530145 params.cmdname = *argv;
146 params.addr = params.ep = 0;
wdenk5b1d7132002-11-03 00:07:02 +0000147
148 while (--argc > 0 && **++argv == '-') {
149 while (*++*argv) {
150 switch (**argv) {
151 case 'l':
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530152 params.lflag = 1;
wdenk5b1d7132002-11-03 00:07:02 +0000153 break;
154 case 'A':
155 if ((--argc <= 0) ||
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530156 (params.arch =
157 genimg_get_arch_id (*++argv)) < 0)
wdenk5b1d7132002-11-03 00:07:02 +0000158 usage ();
159 goto NXTARG;
Simon Glass4f610422013-06-13 15:10:06 -0700160 case 'c':
161 if (--argc <= 0)
162 usage();
163 params.comment = *++argv;
164 goto NXTARG;
wdenk5b1d7132002-11-03 00:07:02 +0000165 case 'C':
166 if ((--argc <= 0) ||
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530167 (params.comp =
168 genimg_get_comp_id (*++argv)) < 0)
wdenk5b1d7132002-11-03 00:07:02 +0000169 usage ();
170 goto NXTARG;
Bartlomiej Sieka9d254382008-03-11 12:34:47 +0100171 case 'D':
172 if (--argc <= 0)
173 usage ();
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530174 params.dtc = *++argv;
Bartlomiej Sieka9d254382008-03-11 12:34:47 +0100175 goto NXTARG;
176
wdenk5b1d7132002-11-03 00:07:02 +0000177 case 'O':
178 if ((--argc <= 0) ||
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530179 (params.os =
180 genimg_get_os_id (*++argv)) < 0)
wdenk5b1d7132002-11-03 00:07:02 +0000181 usage ();
182 goto NXTARG;
183 case 'T':
184 if ((--argc <= 0) ||
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530185 (params.type =
186 genimg_get_type_id (*++argv)) < 0)
wdenk5b1d7132002-11-03 00:07:02 +0000187 usage ();
188 goto NXTARG;
189
190 case 'a':
191 if (--argc <= 0)
192 usage ();
Peter Tysera2513e22010-04-04 22:36:03 -0500193 params.addr = strtoul (*++argv, &ptr, 16);
wdenk5b1d7132002-11-03 00:07:02 +0000194 if (*ptr) {
195 fprintf (stderr,
196 "%s: invalid load address %s\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530197 params.cmdname, *argv);
wdenk5b1d7132002-11-03 00:07:02 +0000198 exit (EXIT_FAILURE);
199 }
200 goto NXTARG;
201 case 'd':
202 if (--argc <= 0)
203 usage ();
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530204 params.datafile = *++argv;
205 params.dflag = 1;
wdenk5b1d7132002-11-03 00:07:02 +0000206 goto NXTARG;
207 case 'e':
208 if (--argc <= 0)
209 usage ();
Peter Tysera2513e22010-04-04 22:36:03 -0500210 params.ep = strtoul (*++argv, &ptr, 16);
wdenk5b1d7132002-11-03 00:07:02 +0000211 if (*ptr) {
212 fprintf (stderr,
213 "%s: invalid entry point %s\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530214 params.cmdname, *argv);
wdenk5b1d7132002-11-03 00:07:02 +0000215 exit (EXIT_FAILURE);
216 }
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530217 params.eflag = 1;
wdenk5b1d7132002-11-03 00:07:02 +0000218 goto NXTARG;
Bartlomiej Sieka9d254382008-03-11 12:34:47 +0100219 case 'f':
220 if (--argc <= 0)
221 usage ();
Simon Glass95d77b42013-06-13 15:10:05 -0700222 params.datafile = *++argv;
223 /* no break */
224 case 'F':
Peter Tyser1a99de22009-11-24 16:42:08 -0600225 /*
226 * The flattened image tree (FIT) format
227 * requires a flattened device tree image type
228 */
229 params.type = IH_TYPE_FLATDT;
Peter Tyserfbc1c8f2009-12-06 01:33:24 -0600230 params.fflag = 1;
Bartlomiej Sieka9d254382008-03-11 12:34:47 +0100231 goto NXTARG;
Simon Glass80e4df82013-06-13 15:10:03 -0700232 case 'k':
233 if (--argc <= 0)
234 usage();
235 params.keydir = *++argv;
236 goto NXTARG;
Simon Glasse29495d2013-06-13 15:10:04 -0700237 case 'K':
238 if (--argc <= 0)
239 usage();
240 params.keydest = *++argv;
241 goto NXTARG;
wdenk5b1d7132002-11-03 00:07:02 +0000242 case 'n':
243 if (--argc <= 0)
244 usage ();
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530245 params.imagename = *++argv;
wdenk5b1d7132002-11-03 00:07:02 +0000246 goto NXTARG;
Simon Glass399c7442013-06-13 15:10:07 -0700247 case 'r':
248 params.require_keys = 1;
249 break;
Shaohui Xie5d898a02012-08-10 02:49:35 +0000250 case 'R':
251 if (--argc <= 0)
252 usage();
253 /*
254 * This entry is for the second configuration
255 * file, if only one is not enough.
256 */
257 params.imagename2 = *++argv;
258 goto NXTARG;
Stefano Babicf0662102011-09-15 23:50:16 +0000259 case 's':
260 params.skipcpy = 1;
261 break;
wdenk5b1d7132002-11-03 00:07:02 +0000262 case 'v':
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530263 params.vflag++;
wdenk5b1d7132002-11-03 00:07:02 +0000264 break;
Wolfgang Denk976b38c2011-02-12 10:37:11 +0100265 case 'V':
266 printf("mkimage version %s\n", PLAIN_VERSION);
267 exit(EXIT_SUCCESS);
wdenk5b1d7132002-11-03 00:07:02 +0000268 case 'x':
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530269 params.xflag++;
wdenk5b1d7132002-11-03 00:07:02 +0000270 break;
271 default:
272 usage ();
273 }
274 }
275NXTARG: ;
276 }
277
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530278 if (argc != 1)
279 usage ();
wdenk5b1d7132002-11-03 00:07:02 +0000280
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530281 /* set tparams as per input type_id */
282 tparams = mkimage_get_type(params.type);
283 if (tparams == NULL) {
284 fprintf (stderr, "%s: unsupported type %s\n",
285 params.cmdname, genimg_get_type_name(params.type));
286 exit (EXIT_FAILURE);
287 }
288
289 /*
290 * check the passed arguments parameters meets the requirements
291 * as per image type to be generated/listed
292 */
293 if (tparams->check_params)
294 if (tparams->check_params (&params))
295 usage ();
296
297 if (!params.eflag) {
298 params.ep = params.addr;
wdenk5b1d7132002-11-03 00:07:02 +0000299 /* If XIP, entry point must be after the U-Boot header */
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530300 if (params.xflag)
301 params.ep += tparams->header_size;
wdenk5b1d7132002-11-03 00:07:02 +0000302 }
303
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530304 params.imagefile = *argv;
wdenk5b1d7132002-11-03 00:07:02 +0000305
Peter Tyserc81296c2009-11-24 16:42:10 -0600306 if (params.fflag){
307 if (tparams->fflag_handle)
308 /*
309 * in some cases, some additional processing needs
310 * to be done if fflag is defined
311 *
312 * For ex. fit_handle_file for Fit file support
313 */
314 retval = tparams->fflag_handle(&params);
wdenk5b1d7132002-11-03 00:07:02 +0000315
Peter Tyserc81296c2009-11-24 16:42:10 -0600316 if (retval != EXIT_SUCCESS)
317 exit (retval);
wdenk5b1d7132002-11-03 00:07:02 +0000318 }
319
Peter Tyserc81296c2009-11-24 16:42:10 -0600320 if (params.lflag || params.fflag) {
321 ifd = open (params.imagefile, O_RDONLY|O_BINARY);
322 } else {
323 ifd = open (params.imagefile,
324 O_RDWR|O_CREAT|O_TRUNC|O_BINARY, 0666);
325 }
326
327 if (ifd < 0) {
328 fprintf (stderr, "%s: Can't open %s: %s\n",
329 params.cmdname, params.imagefile,
330 strerror(errno));
331 exit (EXIT_FAILURE);
332 }
333
334 if (params.lflag || params.fflag) {
wdenk5b1d7132002-11-03 00:07:02 +0000335 /*
336 * list header information of existing image
337 */
338 if (fstat(ifd, &sbuf) < 0) {
339 fprintf (stderr, "%s: Can't stat %s: %s\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530340 params.cmdname, params.imagefile,
341 strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000342 exit (EXIT_FAILURE);
343 }
344
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530345 if ((unsigned)sbuf.st_size < tparams->header_size) {
wdenk5b1d7132002-11-03 00:07:02 +0000346 fprintf (stderr,
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530347 "%s: Bad size: \"%s\" is not valid image\n",
348 params.cmdname, params.imagefile);
wdenk5b1d7132002-11-03 00:07:02 +0000349 exit (EXIT_FAILURE);
350 }
351
Mike Frysingerfa956fd2008-05-01 04:13:05 -0400352 ptr = mmap(0, sbuf.st_size, PROT_READ, MAP_SHARED, ifd, 0);
353 if (ptr == MAP_FAILED) {
wdenk5b1d7132002-11-03 00:07:02 +0000354 fprintf (stderr, "%s: Can't read %s: %s\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530355 params.cmdname, params.imagefile,
356 strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000357 exit (EXIT_FAILURE);
358 }
359
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530360 /*
361 * scan through mkimage registry for all supported image types
362 * and verify the input image file header for match
363 * Print the image information for matched image type
364 * Returns the error code if not matched
365 */
366 retval = mkimage_verify_print_header (ptr, &sbuf);
wdenk5b1d7132002-11-03 00:07:02 +0000367
wdenk5b1d7132002-11-03 00:07:02 +0000368 (void) munmap((void *)ptr, sbuf.st_size);
369 (void) close (ifd);
370
Prafulla Wadaskarf7644c02009-08-10 18:49:37 +0530371 exit (retval);
wdenk5b1d7132002-11-03 00:07:02 +0000372 }
373
374 /*
Stefano Babicf0662102011-09-15 23:50:16 +0000375 * In case there an header with a variable
376 * length will be added, the corresponding
377 * function is called. This is responsible to
378 * allocate memory for the header itself.
wdenk5b1d7132002-11-03 00:07:02 +0000379 */
Stefano Babicf0662102011-09-15 23:50:16 +0000380 if (tparams->vrec_header)
Stefano Babic9bac0bb2013-08-19 19:03:19 +0200381 pad_len = tparams->vrec_header(&params, tparams);
Stefano Babicf0662102011-09-15 23:50:16 +0000382 else
383 memset(tparams->hdr, 0, tparams->header_size);
wdenk5b1d7132002-11-03 00:07:02 +0000384
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530385 if (write(ifd, tparams->hdr, tparams->header_size)
386 != tparams->header_size) {
wdenk5b1d7132002-11-03 00:07:02 +0000387 fprintf (stderr, "%s: Write error on %s: %s\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530388 params.cmdname, params.imagefile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000389 exit (EXIT_FAILURE);
390 }
391
Christian Rieschd1be8f92011-12-09 09:47:38 +0000392 if (!params.skipcpy) {
393 if (params.type == IH_TYPE_MULTI ||
394 params.type == IH_TYPE_SCRIPT) {
395 char *file = params.datafile;
396 uint32_t size;
wdenk5b1d7132002-11-03 00:07:02 +0000397
Christian Rieschd1be8f92011-12-09 09:47:38 +0000398 for (;;) {
399 char *sep = NULL;
wdenk5b1d7132002-11-03 00:07:02 +0000400
Christian Rieschd1be8f92011-12-09 09:47:38 +0000401 if (file) {
402 if ((sep = strchr(file, ':')) != NULL) {
403 *sep = '\0';
404 }
405
406 if (stat (file, &sbuf) < 0) {
407 fprintf (stderr, "%s: Can't stat %s: %s\n",
408 params.cmdname, file, strerror(errno));
409 exit (EXIT_FAILURE);
410 }
411 size = cpu_to_uimage (sbuf.st_size);
412 } else {
413 size = 0;
wdenk5b1d7132002-11-03 00:07:02 +0000414 }
415
Christian Rieschd1be8f92011-12-09 09:47:38 +0000416 if (write(ifd, (char *)&size, sizeof(size)) != sizeof(size)) {
417 fprintf (stderr, "%s: Write error on %s: %s\n",
418 params.cmdname, params.imagefile,
419 strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000420 exit (EXIT_FAILURE);
421 }
Christian Rieschd1be8f92011-12-09 09:47:38 +0000422
423 if (!file) {
424 break;
425 }
426
427 if (sep) {
428 *sep = ':';
429 file = sep + 1;
430 } else {
431 file = NULL;
432 }
wdenk5b1d7132002-11-03 00:07:02 +0000433 }
434
Christian Rieschd1be8f92011-12-09 09:47:38 +0000435 file = params.datafile;
wdenk5b1d7132002-11-03 00:07:02 +0000436
Christian Rieschd1be8f92011-12-09 09:47:38 +0000437 for (;;) {
438 char *sep = strchr(file, ':');
439 if (sep) {
440 *sep = '\0';
441 copy_file (ifd, file, 1);
442 *sep++ = ':';
443 file = sep;
444 } else {
445 copy_file (ifd, file, 0);
446 break;
447 }
wdenk5b1d7132002-11-03 00:07:02 +0000448 }
Shaohui Xie5d898a02012-08-10 02:49:35 +0000449 } else if (params.type == IH_TYPE_PBLIMAGE) {
450 /* PBL has special Image format, implements its' own */
451 pbl_load_uboot(ifd, &params);
Christian Rieschd1be8f92011-12-09 09:47:38 +0000452 } else {
Stefano Babic9bac0bb2013-08-19 19:03:19 +0200453 copy_file(ifd, params.datafile, pad_len);
wdenk5b1d7132002-11-03 00:07:02 +0000454 }
wdenk5b1d7132002-11-03 00:07:02 +0000455 }
456
457 /* We're a bit of paranoid */
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530458#if defined(_POSIX_SYNCHRONIZED_IO) && \
459 !defined(__sun__) && \
460 !defined(__FreeBSD__) && \
461 !defined(__APPLE__)
wdenk5b1d7132002-11-03 00:07:02 +0000462 (void) fdatasync (ifd);
463#else
464 (void) fsync (ifd);
465#endif
466
467 if (fstat(ifd, &sbuf) < 0) {
468 fprintf (stderr, "%s: Can't stat %s: %s\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530469 params.cmdname, params.imagefile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000470 exit (EXIT_FAILURE);
471 }
472
Mike Frysingerfa956fd2008-05-01 04:13:05 -0400473 ptr = mmap(0, sbuf.st_size, PROT_READ|PROT_WRITE, MAP_SHARED, ifd, 0);
474 if (ptr == MAP_FAILED) {
wdenk5b1d7132002-11-03 00:07:02 +0000475 fprintf (stderr, "%s: Can't map %s: %s\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530476 params.cmdname, params.imagefile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000477 exit (EXIT_FAILURE);
478 }
479
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530480 /* Setup the image header as per input image type*/
481 if (tparams->set_header)
482 tparams->set_header (ptr, &sbuf, ifd, &params);
483 else {
484 fprintf (stderr, "%s: Can't set header for %s: %s\n",
485 params.cmdname, tparams->name, strerror(errno));
486 exit (EXIT_FAILURE);
487 }
wdenk5b1d7132002-11-03 00:07:02 +0000488
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530489 /* Print the image information by processing image header */
490 if (tparams->print_header)
491 tparams->print_header (ptr);
492 else {
493 fprintf (stderr, "%s: Can't print header for %s: %s\n",
494 params.cmdname, tparams->name, strerror(errno));
495 exit (EXIT_FAILURE);
496 }
wdenk5b1d7132002-11-03 00:07:02 +0000497
498 (void) munmap((void *)ptr, sbuf.st_size);
499
500 /* We're a bit of paranoid */
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530501#if defined(_POSIX_SYNCHRONIZED_IO) && \
502 !defined(__sun__) && \
503 !defined(__FreeBSD__) && \
504 !defined(__APPLE__)
wdenk5b1d7132002-11-03 00:07:02 +0000505 (void) fdatasync (ifd);
506#else
507 (void) fsync (ifd);
508#endif
509
510 if (close(ifd)) {
511 fprintf (stderr, "%s: Write error on %s: %s\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530512 params.cmdname, params.imagefile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000513 exit (EXIT_FAILURE);
514 }
515
516 exit (EXIT_SUCCESS);
517}
518
519static void
520copy_file (int ifd, const char *datafile, int pad)
521{
522 int dfd;
523 struct stat sbuf;
524 unsigned char *ptr;
525 int tail;
526 int zero = 0;
Stefano Babic9bac0bb2013-08-19 19:03:19 +0200527 uint8_t zeros[4096];
wdenk5b1d7132002-11-03 00:07:02 +0000528 int offset = 0;
529 int size;
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530530 struct image_type_params *tparams = mkimage_get_type (params.type);
wdenk5b1d7132002-11-03 00:07:02 +0000531
Stefano Babic9bac0bb2013-08-19 19:03:19 +0200532 if (pad >= sizeof(zeros)) {
533 fprintf(stderr, "%s: Can't pad to %d\n",
534 params.cmdname, pad);
535 exit(EXIT_FAILURE);
536 }
537
538 memset(zeros, 0, sizeof(zeros));
539
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530540 if (params.vflag) {
wdenk5b1d7132002-11-03 00:07:02 +0000541 fprintf (stderr, "Adding Image %s\n", datafile);
542 }
543
wdenkef1464c2003-10-08 22:14:02 +0000544 if ((dfd = open(datafile, O_RDONLY|O_BINARY)) < 0) {
wdenk5b1d7132002-11-03 00:07:02 +0000545 fprintf (stderr, "%s: Can't open %s: %s\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530546 params.cmdname, datafile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000547 exit (EXIT_FAILURE);
548 }
549
550 if (fstat(dfd, &sbuf) < 0) {
551 fprintf (stderr, "%s: Can't stat %s: %s\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530552 params.cmdname, datafile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000553 exit (EXIT_FAILURE);
554 }
555
Mike Frysingerfa956fd2008-05-01 04:13:05 -0400556 ptr = mmap(0, sbuf.st_size, PROT_READ, MAP_SHARED, dfd, 0);
557 if (ptr == MAP_FAILED) {
wdenk5b1d7132002-11-03 00:07:02 +0000558 fprintf (stderr, "%s: Can't read %s: %s\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530559 params.cmdname, datafile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000560 exit (EXIT_FAILURE);
561 }
562
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530563 if (params.xflag) {
wdenk5b1d7132002-11-03 00:07:02 +0000564 unsigned char *p = NULL;
565 /*
566 * XIP: do not append the image_header_t at the
567 * beginning of the file, but consume the space
568 * reserved for it.
569 */
570
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530571 if ((unsigned)sbuf.st_size < tparams->header_size) {
wdenk5b1d7132002-11-03 00:07:02 +0000572 fprintf (stderr,
573 "%s: Bad size: \"%s\" is too small for XIP\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530574 params.cmdname, datafile);
wdenk5b1d7132002-11-03 00:07:02 +0000575 exit (EXIT_FAILURE);
576 }
577
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530578 for (p = ptr; p < ptr + tparams->header_size; p++) {
wdenk5b1d7132002-11-03 00:07:02 +0000579 if ( *p != 0xff ) {
580 fprintf (stderr,
581 "%s: Bad file: \"%s\" has invalid buffer for XIP\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530582 params.cmdname, datafile);
wdenk5b1d7132002-11-03 00:07:02 +0000583 exit (EXIT_FAILURE);
584 }
585 }
586
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530587 offset = tparams->header_size;
wdenk5b1d7132002-11-03 00:07:02 +0000588 }
589
590 size = sbuf.st_size - offset;
591 if (write(ifd, ptr + offset, size) != size) {
592 fprintf (stderr, "%s: Write error on %s: %s\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530593 params.cmdname, params.imagefile, strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000594 exit (EXIT_FAILURE);
595 }
596
Stefano Babic9bac0bb2013-08-19 19:03:19 +0200597 tail = size % 4;
598 if ((pad == 1) && (tail != 0)) {
wdenk5b1d7132002-11-03 00:07:02 +0000599
600 if (write(ifd, (char *)&zero, 4-tail) != 4-tail) {
601 fprintf (stderr, "%s: Write error on %s: %s\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530602 params.cmdname, params.imagefile,
603 strerror(errno));
wdenk5b1d7132002-11-03 00:07:02 +0000604 exit (EXIT_FAILURE);
605 }
Stefano Babic9bac0bb2013-08-19 19:03:19 +0200606 } else if (pad > 1) {
607 if (write(ifd, (char *)&zeros, pad) != pad) {
608 fprintf(stderr, "%s: Write error on %s: %s\n",
609 params.cmdname, params.imagefile,
610 strerror(errno));
611 exit(EXIT_FAILURE);
612 }
wdenk5b1d7132002-11-03 00:07:02 +0000613 }
614
615 (void) munmap((void *)ptr, sbuf.st_size);
616 (void) close (dfd);
617}
618
Guilherme Maciel Ferreiraf1cc4582013-12-01 12:43:09 -0700619static void usage(void)
wdenk5b1d7132002-11-03 00:07:02 +0000620{
621 fprintf (stderr, "Usage: %s -l image\n"
Bartlomiej Sieka9d254382008-03-11 12:34:47 +0100622 " -l ==> list image header information\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530623 params.cmdname);
Bartlomiej Sieka9d254382008-03-11 12:34:47 +0100624 fprintf (stderr, " %s [-x] -A arch -O os -T type -C comp "
625 "-a addr -e ep -n name -d data_file[:data_file...] image\n"
626 " -A ==> set architecture to 'arch'\n"
wdenk5b1d7132002-11-03 00:07:02 +0000627 " -O ==> set operating system to 'os'\n"
628 " -T ==> set image type to 'type'\n"
629 " -C ==> set compression type 'comp'\n"
630 " -a ==> set load address to 'addr' (hex)\n"
631 " -e ==> set entry point to 'ep' (hex)\n"
632 " -n ==> set image name to 'name'\n"
633 " -d ==> use image data from 'datafile'\n"
Bartlomiej Sieka9d254382008-03-11 12:34:47 +0100634 " -x ==> set XIP (execute in place)\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530635 params.cmdname);
Simon Glass95d77b42013-06-13 15:10:05 -0700636 fprintf(stderr, " %s [-D dtc_options] [-f fit-image.its|-F] fit-image\n",
Prafulla Wadaskar89a4d6b2009-08-19 17:36:46 +0530637 params.cmdname);
Simon Glass80e4df82013-06-13 15:10:03 -0700638 fprintf(stderr, " -D => set options for device tree compiler\n"
639 " -f => input filename for FIT source\n");
640#ifdef CONFIG_FIT_SIGNATURE
Simon Glass399c7442013-06-13 15:10:07 -0700641 fprintf(stderr, "Signing / verified boot options: [-k keydir] [-K dtb] [ -c <comment>] [-r]\n"
Simon Glasse29495d2013-06-13 15:10:04 -0700642 " -k => set directory containing private keys\n"
Simon Glass95d77b42013-06-13 15:10:05 -0700643 " -K => write public keys to this .dtb file\n"
Simon Glass4f610422013-06-13 15:10:06 -0700644 " -c => add comment in signature node\n"
Simon Glass399c7442013-06-13 15:10:07 -0700645 " -F => re-sign existing FIT image\n"
646 " -r => mark keys used as 'required' in dtb\n");
Simon Glass80e4df82013-06-13 15:10:03 -0700647#else
648 fprintf(stderr, "Signing / verified boot not supported (CONFIG_FIT_SIGNATURE undefined)\n");
649#endif
Wolfgang Denk976b38c2011-02-12 10:37:11 +0100650 fprintf (stderr, " %s -V ==> print version information and exit\n",
651 params.cmdname);
Bartlomiej Sieka9d254382008-03-11 12:34:47 +0100652
wdenk5b1d7132002-11-03 00:07:02 +0000653 exit (EXIT_FAILURE);
654}