tools/firmware/ihex2fw.c

   1 // SPDX-License-Identifier: GPL-2.0-only
   2 /*
   3  * Parser/loader for IHEX formatted data.
   4  *
   5  * Copyright © 2008 David Woodhouse <dwmw2@infradead.org>
   6  * Copyright © 2005 Jan Harkes <jaharkes@cs.cmu.edu>
   7  */
   8
   9 #include <stdint.h>
  10 #include <arpa/inet.h>
  11 #include <stdio.h>
  12 #include <errno.h>
  13 #include <sys/types.h>
  14 #include <sys/stat.h>
  15 #include <sys/mman.h>
  16 #include <fcntl.h>
  17 #include <string.h>
  18 #include <unistd.h>
  19 #include <stdlib.h>
  20 #define _GNU_SOURCE
  21 #include <getopt.h>
  22
  23
  24 #define __ALIGN_KERNEL_MASK(x, mask)    (((x) + (mask)) & ~(mask))
  25 #define __ALIGN_KERNEL(x, a)            __ALIGN_KERNEL_MASK(x, (typeof(x))(a) - 1)
  26 #define ALIGN(x, a)                     __ALIGN_KERNEL((x), (a))
  27
  28 struct ihex_binrec {
  29         struct ihex_binrec *next; /* not part of the real data structure */
  30         uint32_t addr;
  31         uint16_t len;
  32         uint8_t data[];
  33 };
  34
  35 /**
  36  * nybble/hex are little helpers to parse hexadecimal numbers to a byte value
  37  **/
  38 static uint8_t nybble(const uint8_t n)
  39 {
  40         if      (n >= '0' && n <= '9') return n - '0';
  41         else if (n >= 'A' && n <= 'F') return n - ('A' - 10);
  42         else if (n >= 'a' && n <= 'f') return n - ('a' - 10);
  43         return 0;
  44 }
  45
  46 static uint8_t hex(const uint8_t *data, uint8_t *crc)
  47 {
  48         uint8_t val = (nybble(data[0]) << 4) | nybble(data[1]);
  49         *crc += val;
  50         return val;
  51 }
  52
  53 static int process_ihex(uint8_t *data, ssize_t size);
  54 static void file_record(struct ihex_binrec *record);
  55 static int output_records(int outfd);
  56
  57 static int sort_records = 0;
  58 static int wide_records = 0;
  59 static int include_jump = 0;
  60
  61 static int usage(void)
  62 {
  63         fprintf(stderr, "ihex2fw: Convert ihex files into binary "
  64                 "representation for use by Linux kernel\n");
  65         fprintf(stderr, "usage: ihex2fw [<options>] <src.HEX> <dst.fw>\n");
  66         fprintf(stderr, "       -w: wide records (16-bit length)\n");
  67         fprintf(stderr, "       -s: sort records by address\n");
  68         fprintf(stderr, "       -j: include records for CS:IP/EIP address\n");
  69         return 1;
  70 }
  71
  72 int main(int argc, char **argv)
  73 {
  74         int infd, outfd;
  75         struct stat st;
  76         uint8_t *data;
  77         int opt;
  78
  79         while ((opt = getopt(argc, argv, "wsj")) != -1) {
  80                 switch (opt) {
  81                 case 'w':
  82                         wide_records = 1;
  83                         break;
  84                 case 's':
  85                         sort_records = 1;
  86                         break;
  87                 case 'j':
  88                         include_jump = 1;
  89                         break;
  90                 default:
  91                         return usage();
  92                 }
  93         }
  94
  95         if (optind + 2 != argc)
  96                 return usage();
  97
  98         if (!strcmp(argv[optind], "-"))
  99                 infd = 0;
 100         else
 101                 infd = open(argv[optind], O_RDONLY);
 102         if (infd == -1) {
 103                 fprintf(stderr, "Failed to open source file: %s",
 104                         strerror(errno));
 105                 return usage();
 106         }
 107         if (fstat(infd, &st)) {
 108                 perror("stat");
 109                 return 1;
 110         }
 111         data = mmap(NULL, st.st_size, PROT_READ, MAP_SHARED, infd, 0);
 112         if (data == MAP_FAILED) {
 113                 perror("mmap");
 114                 return 1;
 115         }
 116
 117         if (!strcmp(argv[optind+1], "-"))
 118                 outfd = 1;
 119         else
 120                 outfd = open(argv[optind+1], O_TRUNC|O_CREAT|O_WRONLY, 0644);
 121         if (outfd == -1) {
 122                 fprintf(stderr, "Failed to open destination file: %s",
 123                         strerror(errno));
 124                 return usage();
 125         }
 126         if (process_ihex(data, st.st_size))
 127                 return 1;
 128
 129         return output_records(outfd);
 130 }
 131
 132 static int process_ihex(uint8_t *data, ssize_t size)
 133 {
 134         struct ihex_binrec *record;
 135         size_t record_size;
 136         uint32_t offset = 0;
 137         uint32_t data32;
 138         uint8_t type, crc = 0, crcbyte = 0;
 139         int i, j;
 140         int line = 1;
 141         int len;
 142
 143         i = 0;
 144 next_record:
 145         /* search for the start of record character */
 146         while (i < size) {
 147                 if (data[i] == '\n') line++;
 148                 if (data[i++] == ':') break;
 149         }
 150
 151         /* Minimum record length would be about 10 characters */
 152         if (i + 10 > size) {
 153                 fprintf(stderr, "Can't find valid record at line %d\n", line);
 154                 return -EINVAL;
 155         }
 156
 157         len = hex(data + i, &crc); i += 2;
 158         if (wide_records) {
 159                 len <<= 8;
 160                 len += hex(data + i, &crc); i += 2;
 161         }
 162         record_size = ALIGN(sizeof(*record) + len, 4);
 163         record = malloc(record_size);
 164         if (!record) {
 165                 fprintf(stderr, "out of memory for records\n");
 166                 return -ENOMEM;
 167         }
 168         memset(record, 0, record_size);
 169         record->len = len;
 170
 171         /* now check if we have enough data to read everything */
 172         if (i + 8 + (record->len * 2) > size) {
 173                 fprintf(stderr, "Not enough data to read complete record at line %d\n",
 174                         line);
 175                 return -EINVAL;
 176         }
 177
 178         record->addr  = hex(data + i, &crc) << 8; i += 2;
 179         record->addr |= hex(data + i, &crc); i += 2;
 180         type = hex(data + i, &crc); i += 2;
 181
 182         for (j = 0; j < record->len; j++, i += 2)
 183                 record->data[j] = hex(data + i, &crc);
 184
 185         /* check CRC */
 186         crcbyte = hex(data + i, &crc); i += 2;
 187         if (crc != 0) {
 188                 fprintf(stderr, "CRC failure at line %d: got 0x%X, expected 0x%X\n",
 189                         line, crcbyte, (unsigned char)(crcbyte-crc));
 190                 return -EINVAL;
 191         }
 192
 193         /* Done reading the record */
 194         switch (type) {
 195         case 0:
 196                 /* old style EOF record? */
 197                 if (!record->len)
 198                         break;
 199
 200                 record->addr += offset;
 201                 file_record(record);
 202                 goto next_record;
 203
 204         case 1: /* End-Of-File Record */
 205                 if (record->addr || record->len) {
 206                         fprintf(stderr, "Bad EOF record (type 01) format at line %d",
 207                                 line);
 208                         return -EINVAL;
 209                 }
 210                 break;
 211
 212         case 2: /* Extended Segment Address Record (HEX86) */
 213         case 4: /* Extended Linear Address Record (HEX386) */
 214                 if (record->addr || record->len != 2) {
 215                         fprintf(stderr, "Bad HEX86/HEX386 record (type %02X) at line %d\n",
 216                                 type, line);
 217                         return -EINVAL;
 218                 }
 219
 220                 /* We shouldn't really be using the offset for HEX86 because
 221                  * the wraparound case is specified quite differently. */
 222                 offset = record->data[0] << 8 | record->data[1];
 223                 offset <<= (type == 2 ? 4 : 16);
 224                 goto next_record;
 225
 226         case 3: /* Start Segment Address Record */
 227         case 5: /* Start Linear Address Record */
 228                 if (record->addr || record->len != 4) {
 229                         fprintf(stderr, "Bad Start Address record (type %02X) at line %d\n",
 230                                 type, line);
 231                         return -EINVAL;
 232                 }
 233
 234                 memcpy(&data32, &record->data[0], sizeof(data32));
 235                 data32 = htonl(data32);
 236                 memcpy(&record->data[0], &data32, sizeof(data32));
 237
 238                 /* These records contain the CS/IP or EIP where execution
 239                  * starts. If requested output this as a record. */
 240                 if (include_jump)
 241                         file_record(record);
 242                 goto next_record;
 243
 244         default:
 245                 fprintf(stderr, "Unknown record (type %02X)\n", type);
 246                 return -EINVAL;
 247         }
 248
 249         return 0;
 250 }
 251
 252 static struct ihex_binrec *records;
 253
 254 static void file_record(struct ihex_binrec *record)
 255 {
 256         struct ihex_binrec **p = &records;
 257
 258         while ((*p) && (!sort_records || (*p)->addr < record->addr))
 259                 p = &((*p)->next);
 260
 261         record->next = *p;
 262         *p = record;
 263 }
 264
 265 static uint16_t ihex_binrec_size(struct ihex_binrec *p)
 266 {
 267         return p->len + sizeof(p->addr) + sizeof(p->len);
 268 }
 269
 270 static int output_records(int outfd)
 271 {
 272         unsigned char zeroes[6] = {0, 0, 0, 0, 0, 0};
 273         struct ihex_binrec *p = records;
 274
 275         while (p) {
 276                 uint16_t writelen = ALIGN(ihex_binrec_size(p), 4);
 277
 278                 p->addr = htonl(p->addr);
 279                 p->len = htons(p->len);
 280                 if (write(outfd, &p->addr, writelen) != writelen)
 281                         return 1;
 282                 p = p->next;
 283         }
 284         /* EOF record is zero length, since we don't bother to represent
 285            the type field in the binary version */
 286         if (write(outfd, zeroes, 6) != 6)
 287                 return 1;
 288         return 0;
 289 }