Fixed handling of bad checksums with "mkimage -l"
[kernel/u-boot.git] / tools / mkimage.c
1 /*
2  * (C) Copyright 2000-2004
3  * DENX Software Engineering
4  * Wolfgang Denk, wd@denx.de
5  * All rights reserved.
6  *
7  * This program is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU General Public License as
9  * published by the Free Software Foundation; either version 2 of
10  * the License, or (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, write to the Free Software
19  * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
20  * MA 02111-1307 USA
21  */
22
23 #include <errno.h>
24 #include <fcntl.h>
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #ifndef __WIN32__
29 #include <netinet/in.h>         /* for host / network byte order conversions    */
30 #endif
31 #include <sys/mman.h>
32 #include <sys/stat.h>
33 #include <time.h>
34 #include <unistd.h>
35
36 #if defined(__BEOS__) || defined(__NetBSD__) || defined(__APPLE__)
37 #include <inttypes.h>
38 #endif
39
40 #ifdef __WIN32__
41 typedef unsigned int __u32;
42
43 #define SWAP_LONG(x) \
44         ((__u32)( \
45                 (((__u32)(x) & (__u32)0x000000ffUL) << 24) | \
46                 (((__u32)(x) & (__u32)0x0000ff00UL) <<  8) | \
47                 (((__u32)(x) & (__u32)0x00ff0000UL) >>  8) | \
48                 (((__u32)(x) & (__u32)0xff000000UL) >> 24) ))
49 typedef         unsigned char   uint8_t;
50 typedef         unsigned short  uint16_t;
51 typedef         unsigned int    uint32_t;
52
53 #define     ntohl(a)    SWAP_LONG(a)
54 #define     htonl(a)    SWAP_LONG(a)
55 #endif  /* __WIN32__ */
56
57 #ifndef O_BINARY                /* should be define'd on __WIN32__ */
58 #define O_BINARY        0
59 #endif
60
61 #include <image.h>
62
63 extern int errno;
64
65 #ifndef MAP_FAILED
66 #define MAP_FAILED (-1)
67 #endif
68
69 char *cmdname;
70
71 extern unsigned long crc32 (unsigned long crc, const char *buf, unsigned int len);
72
73 typedef struct table_entry {
74         int     val;            /* as defined in image.h        */
75         char    *sname;         /* short (input) name           */
76         char    *lname;         /* long (output) name           */
77 } table_entry_t;
78
79 table_entry_t arch_name[] = {
80     {   IH_CPU_INVALID,         NULL,           "Invalid CPU",  },
81     {   IH_CPU_ALPHA,           "alpha",        "Alpha",        },
82     {   IH_CPU_ARM,             "arm",          "ARM",          },
83     {   IH_CPU_I386,            "x86",          "Intel x86",    },
84     {   IH_CPU_IA64,            "ia64",         "IA64",         },
85     {   IH_CPU_M68K,            "m68k",         "MC68000",      },
86     {   IH_CPU_MICROBLAZE,      "microblaze",   "MicroBlaze",   },
87     {   IH_CPU_MIPS,            "mips",         "MIPS",         },
88     {   IH_CPU_MIPS64,          "mips64",       "MIPS 64 Bit",  },
89     {   IH_CPU_NIOS,            "nios",         "NIOS",         },
90     {   IH_CPU_NIOS2,           "nios2",        "NIOS II",      },
91     {   IH_CPU_PPC,             "ppc",          "PowerPC",      },
92     {   IH_CPU_S390,            "s390",         "IBM S390",     },
93     {   IH_CPU_SH,              "sh",           "SuperH",       },
94     {   IH_CPU_SPARC,           "sparc",        "SPARC",        },
95     {   IH_CPU_SPARC64,         "sparc64",      "SPARC 64 Bit", },
96     {   IH_CPU_BLACKFIN,        "blackfin",     "Blackfin",     },
97     {   -1,                     "",             "",             },
98 };
99
100 table_entry_t os_name[] = {
101     {   IH_OS_INVALID,  NULL,           "Invalid OS",           },
102     {   IH_OS_4_4BSD,   "4_4bsd",       "4_4BSD",               },
103     {   IH_OS_ARTOS,    "artos",        "ARTOS",                },
104     {   IH_OS_DELL,     "dell",         "Dell",                 },
105     {   IH_OS_ESIX,     "esix",         "Esix",                 },
106     {   IH_OS_FREEBSD,  "freebsd",      "FreeBSD",              },
107     {   IH_OS_IRIX,     "irix",         "Irix",                 },
108     {   IH_OS_LINUX,    "linux",        "Linux",                },
109     {   IH_OS_LYNXOS,   "lynxos",       "LynxOS",               },
110     {   IH_OS_NCR,      "ncr",          "NCR",                  },
111     {   IH_OS_NETBSD,   "netbsd",       "NetBSD",               },
112     {   IH_OS_OPENBSD,  "openbsd",      "OpenBSD",              },
113     {   IH_OS_PSOS,     "psos",         "pSOS",                 },
114     {   IH_OS_QNX,      "qnx",          "QNX",                  },
115     {   IH_OS_RTEMS,    "rtems",        "RTEMS",                },
116     {   IH_OS_SCO,      "sco",          "SCO",                  },
117     {   IH_OS_SOLARIS,  "solaris",      "Solaris",              },
118     {   IH_OS_SVR4,     "svr4",         "SVR4",                 },
119     {   IH_OS_U_BOOT,   "u-boot",       "U-Boot",               },
120     {   IH_OS_VXWORKS,  "vxworks",      "VxWorks",              },
121     {   -1,             "",             "",                     },
122 };
123
124 table_entry_t type_name[] = {
125     {   IH_TYPE_INVALID,    NULL,         "Invalid Image",      },
126     {   IH_TYPE_FILESYSTEM, "filesystem", "Filesystem Image",   },
127     {   IH_TYPE_FIRMWARE,   "firmware",   "Firmware",           },
128     {   IH_TYPE_KERNEL,     "kernel",     "Kernel Image",       },
129     {   IH_TYPE_MULTI,      "multi",      "Multi-File Image",   },
130     {   IH_TYPE_RAMDISK,    "ramdisk",    "RAMDisk Image",      },
131     {   IH_TYPE_SCRIPT,     "script",     "Script",             },
132     {   IH_TYPE_STANDALONE, "standalone", "Standalone Program", },
133     {   -1,                 "",           "",                   },
134 };
135
136 table_entry_t comp_name[] = {
137     {   IH_COMP_NONE,   "none",         "uncompressed",         },
138     {   IH_COMP_BZIP2,  "bzip2",        "bzip2 compressed",     },
139     {   IH_COMP_GZIP,   "gzip",         "gzip compressed",      },
140     {   -1,             "",             "",                     },
141 };
142
143 static  void    copy_file (int, const char *, int);
144 static  void    usage   (void);
145 static  void    print_header (image_header_t *);
146 static  void    print_type (image_header_t *);
147 static  char    *put_table_entry (table_entry_t *, char *, int);
148 static  char    *put_arch (int);
149 static  char    *put_type (int);
150 static  char    *put_os   (int);
151 static  char    *put_comp (int);
152 static  int     get_table_entry (table_entry_t *, char *, char *);
153 static  int     get_arch(char *);
154 static  int     get_comp(char *);
155 static  int     get_os  (char *);
156 static  int     get_type(char *);
157
158
159 char    *datafile;
160 char    *imagefile;
161
162 int dflag    = 0;
163 int eflag    = 0;
164 int lflag    = 0;
165 int vflag    = 0;
166 int xflag    = 0;
167 int opt_os   = IH_OS_LINUX;
168 int opt_arch = IH_CPU_PPC;
169 int opt_type = IH_TYPE_KERNEL;
170 int opt_comp = IH_COMP_GZIP;
171
172 image_header_t header;
173 image_header_t *hdr = &header;
174
175 int
176 main (int argc, char **argv)
177 {
178         int ifd;
179         uint32_t checksum;
180         uint32_t addr;
181         uint32_t ep;
182         struct stat sbuf;
183         unsigned char *ptr;
184         char *name = "";
185
186         cmdname = *argv;
187
188         addr = ep = 0;
189
190         while (--argc > 0 && **++argv == '-') {
191                 while (*++*argv) {
192                         switch (**argv) {
193                         case 'l':
194                                 lflag = 1;
195                                 break;
196                         case 'A':
197                                 if ((--argc <= 0) ||
198                                     (opt_arch = get_arch(*++argv)) < 0)
199                                         usage ();
200                                 goto NXTARG;
201                         case 'C':
202                                 if ((--argc <= 0) ||
203                                     (opt_comp = get_comp(*++argv)) < 0)
204                                         usage ();
205                                 goto NXTARG;
206                         case 'O':
207                                 if ((--argc <= 0) ||
208                                     (opt_os = get_os(*++argv)) < 0)
209                                         usage ();
210                                 goto NXTARG;
211                         case 'T':
212                                 if ((--argc <= 0) ||
213                                     (opt_type = get_type(*++argv)) < 0)
214                                         usage ();
215                                 goto NXTARG;
216
217                         case 'a':
218                                 if (--argc <= 0)
219                                         usage ();
220                                 addr = strtoul (*++argv, (char **)&ptr, 16);
221                                 if (*ptr) {
222                                         fprintf (stderr,
223                                                 "%s: invalid load address %s\n",
224                                                 cmdname, *argv);
225                                         exit (EXIT_FAILURE);
226                                 }
227                                 goto NXTARG;
228                         case 'd':
229                                 if (--argc <= 0)
230                                         usage ();
231                                 datafile = *++argv;
232                                 dflag = 1;
233                                 goto NXTARG;
234                         case 'e':
235                                 if (--argc <= 0)
236                                         usage ();
237                                 ep = strtoul (*++argv, (char **)&ptr, 16);
238                                 if (*ptr) {
239                                         fprintf (stderr,
240                                                 "%s: invalid entry point %s\n",
241                                                 cmdname, *argv);
242                                         exit (EXIT_FAILURE);
243                                 }
244                                 eflag = 1;
245                                 goto NXTARG;
246                         case 'n':
247                                 if (--argc <= 0)
248                                         usage ();
249                                 name = *++argv;
250                                 goto NXTARG;
251                         case 'v':
252                                 vflag++;
253                                 break;
254                         case 'x':
255                                 xflag++;
256                                 break;
257                         default:
258                                 usage ();
259                         }
260                 }
261 NXTARG:         ;
262         }
263
264         if ((argc != 1) || ((lflag ^ dflag) == 0))
265                 usage();
266
267         if (!eflag) {
268                 ep = addr;
269                 /* If XIP, entry point must be after the U-Boot header */
270                 if (xflag)
271                         ep += sizeof(image_header_t);
272         }
273
274         /*
275          * If XIP, ensure the entry point is equal to the load address plus
276          * the size of the U-Boot header.
277          */
278         if (xflag) {
279                 if (ep != addr + sizeof(image_header_t)) {
280                         fprintf (stderr,
281                                 "%s: For XIP, the entry point must be the load addr + %lu\n",
282                                 cmdname,
283                                 (unsigned long)sizeof(image_header_t));
284                         exit (EXIT_FAILURE);
285                 }
286         }
287
288         imagefile = *argv;
289
290         if (lflag) {
291                 ifd = open(imagefile, O_RDONLY|O_BINARY);
292         } else {
293                 ifd = open(imagefile, O_RDWR|O_CREAT|O_TRUNC|O_BINARY, 0666);
294         }
295
296         if (ifd < 0) {
297                 fprintf (stderr, "%s: Can't open %s: %s\n",
298                         cmdname, imagefile, strerror(errno));
299                 exit (EXIT_FAILURE);
300         }
301
302         if (lflag) {
303                 int len;
304                 char *data;
305                 /*
306                  * list header information of existing image
307                  */
308                 if (fstat(ifd, &sbuf) < 0) {
309                         fprintf (stderr, "%s: Can't stat %s: %s\n",
310                                 cmdname, imagefile, strerror(errno));
311                         exit (EXIT_FAILURE);
312                 }
313
314                 if ((unsigned)sbuf.st_size < sizeof(image_header_t)) {
315                         fprintf (stderr,
316                                 "%s: Bad size: \"%s\" is no valid image\n",
317                                 cmdname, imagefile);
318                         exit (EXIT_FAILURE);
319                 }
320
321                 ptr = (unsigned char *)mmap(0, sbuf.st_size,
322                                             PROT_READ, MAP_SHARED, ifd, 0);
323                 if ((caddr_t)ptr == (caddr_t)-1) {
324                         fprintf (stderr, "%s: Can't read %s: %s\n",
325                                 cmdname, imagefile, strerror(errno));
326                         exit (EXIT_FAILURE);
327                 }
328
329                 /*
330                  * create copy of header so that we can blank out the
331                  * checksum field for checking - this can't be done
332                  * on the PROT_READ mapped data.
333                  */
334                 memcpy (hdr, ptr, sizeof(image_header_t));
335
336                 if (ntohl(hdr->ih_magic) != IH_MAGIC) {
337                         fprintf (stderr,
338                                 "%s: Bad Magic Number: \"%s\" is no valid image\n",
339                                 cmdname, imagefile);
340                         exit (EXIT_FAILURE);
341                 }
342
343                 data = (char *)hdr;
344                 len  = sizeof(image_header_t);
345
346                 checksum = ntohl(hdr->ih_hcrc);
347                 hdr->ih_hcrc = htonl(0);        /* clear for re-calculation */
348
349                 if (crc32 (0, data, len) != checksum) {
350                         fprintf (stderr,
351                                 "%s: ERROR: \"%s\" has bad header checksum!\n",
352                                 cmdname, imagefile);
353                         exit (EXIT_FAILURE);
354                 }
355
356                 data = (char *)(ptr + sizeof(image_header_t));
357                 len  = sbuf.st_size - sizeof(image_header_t) ;
358
359                 if (crc32 (0, data, len) != ntohl(hdr->ih_dcrc)) {
360                         fprintf (stderr,
361                                 "%s: ERROR: \"%s\" has corrupted data!\n",
362                                 cmdname, imagefile);
363                         exit (EXIT_FAILURE);
364                 }
365
366                 /* for multi-file images we need the data part, too */
367                 print_header ((image_header_t *)ptr);
368
369                 (void) munmap((void *)ptr, sbuf.st_size);
370                 (void) close (ifd);
371
372                 exit (EXIT_SUCCESS);
373         }
374
375         /*
376          * Must be -w then:
377          *
378          * write dummy header, to be fixed later
379          */
380         memset (hdr, 0, sizeof(image_header_t));
381
382         if (write(ifd, hdr, sizeof(image_header_t)) != sizeof(image_header_t)) {
383                 fprintf (stderr, "%s: Write error on %s: %s\n",
384                         cmdname, imagefile, strerror(errno));
385                 exit (EXIT_FAILURE);
386         }
387
388         if (opt_type == IH_TYPE_MULTI || opt_type == IH_TYPE_SCRIPT) {
389                 char *file = datafile;
390                 uint32_t size;
391
392                 for (;;) {
393                         char *sep = NULL;
394
395                         if (file) {
396                                 if ((sep = strchr(file, ':')) != NULL) {
397                                         *sep = '\0';
398                                 }
399
400                                 if (stat (file, &sbuf) < 0) {
401                                         fprintf (stderr, "%s: Can't stat %s: %s\n",
402                                                 cmdname, file, strerror(errno));
403                                         exit (EXIT_FAILURE);
404                                 }
405                                 size = htonl(sbuf.st_size);
406                         } else {
407                                 size = 0;
408                         }
409
410                         if (write(ifd, (char *)&size, sizeof(size)) != sizeof(size)) {
411                                 fprintf (stderr, "%s: Write error on %s: %s\n",
412                                         cmdname, imagefile, strerror(errno));
413                                 exit (EXIT_FAILURE);
414                         }
415
416                         if (!file) {
417                                 break;
418                         }
419
420                         if (sep) {
421                                 *sep = ':';
422                                 file = sep + 1;
423                         } else {
424                                 file = NULL;
425                         }
426                 }
427
428                 file = datafile;
429
430                 for (;;) {
431                         char *sep = strchr(file, ':');
432                         if (sep) {
433                                 *sep = '\0';
434                                 copy_file (ifd, file, 1);
435                                 *sep++ = ':';
436                                 file = sep;
437                         } else {
438                                 copy_file (ifd, file, 0);
439                                 break;
440                         }
441                 }
442         } else {
443                 copy_file (ifd, datafile, 0);
444         }
445
446         /* We're a bit of paranoid */
447 #if defined(_POSIX_SYNCHRONIZED_IO) && !defined(__sun__) && !defined(__FreeBSD__)
448         (void) fdatasync (ifd);
449 #else
450         (void) fsync (ifd);
451 #endif
452
453         if (fstat(ifd, &sbuf) < 0) {
454                 fprintf (stderr, "%s: Can't stat %s: %s\n",
455                         cmdname, imagefile, strerror(errno));
456                 exit (EXIT_FAILURE);
457         }
458
459         ptr = (unsigned char *)mmap(0, sbuf.st_size,
460                                     PROT_READ|PROT_WRITE, MAP_SHARED, ifd, 0);
461         if (ptr == (unsigned char *)MAP_FAILED) {
462                 fprintf (stderr, "%s: Can't map %s: %s\n",
463                         cmdname, imagefile, strerror(errno));
464                 exit (EXIT_FAILURE);
465         }
466
467         hdr = (image_header_t *)ptr;
468
469         checksum = crc32 (0,
470                           (const char *)(ptr + sizeof(image_header_t)),
471                           sbuf.st_size - sizeof(image_header_t)
472                          );
473
474         /* Build new header */
475         hdr->ih_magic = htonl(IH_MAGIC);
476         hdr->ih_time  = htonl(sbuf.st_mtime);
477         hdr->ih_size  = htonl(sbuf.st_size - sizeof(image_header_t));
478         hdr->ih_load  = htonl(addr);
479         hdr->ih_ep    = htonl(ep);
480         hdr->ih_dcrc  = htonl(checksum);
481         hdr->ih_os    = opt_os;
482         hdr->ih_arch  = opt_arch;
483         hdr->ih_type  = opt_type;
484         hdr->ih_comp  = opt_comp;
485
486         strncpy((char *)hdr->ih_name, name, IH_NMLEN);
487
488         checksum = crc32(0,(const char *)hdr,sizeof(image_header_t));
489
490         hdr->ih_hcrc = htonl(checksum);
491
492         print_header (hdr);
493
494         (void) munmap((void *)ptr, sbuf.st_size);
495
496         /* We're a bit of paranoid */
497 #if defined(_POSIX_SYNCHRONIZED_IO) && !defined(__sun__) && !defined(__FreeBSD__)
498         (void) fdatasync (ifd);
499 #else
500         (void) fsync (ifd);
501 #endif
502
503         if (close(ifd)) {
504                 fprintf (stderr, "%s: Write error on %s: %s\n",
505                         cmdname, imagefile, strerror(errno));
506                 exit (EXIT_FAILURE);
507         }
508
509         exit (EXIT_SUCCESS);
510 }
511
512 static void
513 copy_file (int ifd, const char *datafile, int pad)
514 {
515         int dfd;
516         struct stat sbuf;
517         unsigned char *ptr;
518         int tail;
519         int zero = 0;
520         int offset = 0;
521         int size;
522
523         if (vflag) {
524                 fprintf (stderr, "Adding Image %s\n", datafile);
525         }
526
527         if ((dfd = open(datafile, O_RDONLY|O_BINARY)) < 0) {
528                 fprintf (stderr, "%s: Can't open %s: %s\n",
529                         cmdname, datafile, strerror(errno));
530                 exit (EXIT_FAILURE);
531         }
532
533         if (fstat(dfd, &sbuf) < 0) {
534                 fprintf (stderr, "%s: Can't stat %s: %s\n",
535                         cmdname, datafile, strerror(errno));
536                 exit (EXIT_FAILURE);
537         }
538
539         ptr = (unsigned char *)mmap(0, sbuf.st_size,
540                                     PROT_READ, MAP_SHARED, dfd, 0);
541         if (ptr == (unsigned char *)MAP_FAILED) {
542                 fprintf (stderr, "%s: Can't read %s: %s\n",
543                         cmdname, datafile, strerror(errno));
544                 exit (EXIT_FAILURE);
545         }
546
547         if (xflag) {
548                 unsigned char *p = NULL;
549                 /*
550                  * XIP: do not append the image_header_t at the
551                  * beginning of the file, but consume the space
552                  * reserved for it.
553                  */
554
555                 if ((unsigned)sbuf.st_size < sizeof(image_header_t)) {
556                         fprintf (stderr,
557                                 "%s: Bad size: \"%s\" is too small for XIP\n",
558                                 cmdname, datafile);
559                         exit (EXIT_FAILURE);
560                 }
561
562                 for (p=ptr; p < ptr+sizeof(image_header_t); p++) {
563                         if ( *p != 0xff ) {
564                                 fprintf (stderr,
565                                         "%s: Bad file: \"%s\" has invalid buffer for XIP\n",
566                                         cmdname, datafile);
567                                 exit (EXIT_FAILURE);
568                         }
569                 }
570
571                 offset = sizeof(image_header_t);
572         }
573
574         size = sbuf.st_size - offset;
575         if (write(ifd, ptr + offset, size) != size) {
576                 fprintf (stderr, "%s: Write error on %s: %s\n",
577                         cmdname, imagefile, strerror(errno));
578                 exit (EXIT_FAILURE);
579         }
580
581         if (pad && ((tail = size % 4) != 0)) {
582
583                 if (write(ifd, (char *)&zero, 4-tail) != 4-tail) {
584                         fprintf (stderr, "%s: Write error on %s: %s\n",
585                                 cmdname, imagefile, strerror(errno));
586                         exit (EXIT_FAILURE);
587                 }
588         }
589
590         (void) munmap((void *)ptr, sbuf.st_size);
591         (void) close (dfd);
592 }
593
594 void
595 usage ()
596 {
597         fprintf (stderr, "Usage: %s -l image\n"
598                          "          -l ==> list image header information\n"
599                          "       %s [-x] -A arch -O os -T type -C comp "
600                          "-a addr -e ep -n name -d data_file[:data_file...] image\n",
601                 cmdname, cmdname);
602         fprintf (stderr, "          -A ==> set architecture to 'arch'\n"
603                          "          -O ==> set operating system to 'os'\n"
604                          "          -T ==> set image type to 'type'\n"
605                          "          -C ==> set compression type 'comp'\n"
606                          "          -a ==> set load address to 'addr' (hex)\n"
607                          "          -e ==> set entry point to 'ep' (hex)\n"
608                          "          -n ==> set image name to 'name'\n"
609                          "          -d ==> use image data from 'datafile'\n"
610                          "          -x ==> set XIP (execute in place)\n"
611                 );
612         exit (EXIT_FAILURE);
613 }
614
615 static void
616 print_header (image_header_t *hdr)
617 {
618         time_t timestamp;
619         uint32_t size;
620
621         timestamp = (time_t)ntohl(hdr->ih_time);
622         size = ntohl(hdr->ih_size);
623
624         printf ("Image Name:   %.*s\n", IH_NMLEN, hdr->ih_name);
625         printf ("Created:      %s", ctime(&timestamp));
626         printf ("Image Type:   "); print_type(hdr);
627         printf ("Data Size:    %d Bytes = %.2f kB = %.2f MB\n",
628                 size, (double)size / 1.024e3, (double)size / 1.048576e6 );
629         printf ("Load Address: 0x%08X\n", ntohl(hdr->ih_load));
630         printf ("Entry Point:  0x%08X\n", ntohl(hdr->ih_ep));
631
632         if (hdr->ih_type == IH_TYPE_MULTI || hdr->ih_type == IH_TYPE_SCRIPT) {
633                 int i, ptrs;
634                 uint32_t pos;
635                 unsigned long *len_ptr = (unsigned long *) (
636                                         (unsigned long)hdr + sizeof(image_header_t)
637                                 );
638
639                 /* determine number of images first (to calculate image offsets) */
640                 for (i=0; len_ptr[i]; ++i)      /* null pointer terminates list */
641                         ;
642                 ptrs = i;               /* null pointer terminates list */
643
644                 pos = sizeof(image_header_t) + ptrs * sizeof(long);
645                 printf ("Contents:\n");
646                 for (i=0; len_ptr[i]; ++i) {
647                         size = ntohl(len_ptr[i]);
648
649                         printf ("   Image %d: %8d Bytes = %4d kB = %d MB\n",
650                                 i, size, size>>10, size>>20);
651                         if (hdr->ih_type == IH_TYPE_SCRIPT && i > 0) {
652                                 /*
653                                  * the user may need to know offsets
654                                  * if planning to do something with
655                                  * multiple files
656                                  */
657                                 printf ("    Offset = %08X\n", pos);
658                         }
659                         /* copy_file() will pad the first files to even word align */
660                         size += 3;
661                         size &= ~3;
662                         pos += size;
663                 }
664         }
665 }
666
667
668 static void
669 print_type (image_header_t *hdr)
670 {
671         printf ("%s %s %s (%s)\n",
672                 put_arch (hdr->ih_arch),
673                 put_os   (hdr->ih_os  ),
674                 put_type (hdr->ih_type),
675                 put_comp (hdr->ih_comp)
676         );
677 }
678
679 static char *put_arch (int arch)
680 {
681         return (put_table_entry(arch_name, "Unknown Architecture", arch));
682 }
683
684 static char *put_os (int os)
685 {
686         return (put_table_entry(os_name, "Unknown OS", os));
687 }
688
689 static char *put_type (int type)
690 {
691         return (put_table_entry(type_name, "Unknown Image", type));
692 }
693
694 static char *put_comp (int comp)
695 {
696         return (put_table_entry(comp_name, "Unknown Compression", comp));
697 }
698
699 static char *put_table_entry (table_entry_t *table, char *msg, int type)
700 {
701         for (; table->val>=0; ++table) {
702                 if (table->val == type)
703                         return (table->lname);
704         }
705         return (msg);
706 }
707
708 static int get_arch(char *name)
709 {
710         return (get_table_entry(arch_name, "CPU", name));
711 }
712
713
714 static int get_comp(char *name)
715 {
716         return (get_table_entry(comp_name, "Compression", name));
717 }
718
719
720 static int get_os (char *name)
721 {
722         return (get_table_entry(os_name, "OS", name));
723 }
724
725
726 static int get_type(char *name)
727 {
728         return (get_table_entry(type_name, "Image", name));
729 }
730
731 static int get_table_entry (table_entry_t *table, char *msg, char *name)
732 {
733         table_entry_t *t;
734         int first = 1;
735
736         for (t=table; t->val>=0; ++t) {
737                 if (t->sname && strcasecmp(t->sname, name)==0)
738                         return (t->val);
739         }
740         fprintf (stderr, "\nInvalid %s Type - valid names are", msg);
741         for (t=table; t->val>=0; ++t) {
742                 if (t->sname == NULL)
743                         continue;
744                 fprintf (stderr, "%c %s", (first) ? ':' : ',', t->sname);
745                 first = 0;
746         }
747         fprintf (stderr, "\n");
748         return (-1);
749 }