tools/firmware-utils/src/trx.c

   1 /*
   2  * Copyright (C) 2004  Manuel Novoa III  <mjn3@codepoet.org>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  17  */
  18
  19 /* July 29, 2004
  20  *
  21  * This is a hacked replacement for the 'trx' utility used to create
  22  * wrt54g .trx firmware files.  It isn't pretty, but it does the job
  23  * for me.
  24  *
  25  * As an extension, you can specify a larger maximum length for the
  26  * .trx file using '-m'.  It will be rounded up to be a multiple of 4K.
  27  * NOTE: This space will be malloc()'d.
  28  *
  29  * August 16, 2004
  30  *
  31  * Sigh... Make it endian-neutral.
  32  *
  33  * TODO: Support '-b' option to specify offsets for each file.
  34  *
  35  * February 19, 2005 - mbm
  36  *
  37  * Add -a (align offset) and -b (absolute offset)
  38  *
  39  * March 24, 2010 - markus
  40  *
  41  * extend trx header struct for new version
  42  * assume v1 for as default
  43  * Add option -2 to allow v2 header
  44  */
  45
  46 #include <stdio.h>
  47 #include <stdlib.h>
  48 #include <stddef.h>
  49 #include <stdint.h>
  50 #include <string.h>
  51 #include <errno.h>
  52 #include <unistd.h>
  53
  54 #if __BYTE_ORDER == __BIG_ENDIAN
  55 #define STORE32_LE(X)           bswap_32(X)
  56 #elif __BYTE_ORDER == __LITTLE_ENDIAN
  57 #define STORE32_LE(X)           (X)
  58 #else
  59 #error unkown endianness!
  60 #endif
  61
  62 uint32_t crc32buf(char *buf, size_t len);
  63
  64 /**********************************************************************/
  65 /* from trxhdr.h */
  66
  67 #define TRX_MAGIC       0x30524448      /* "HDR0" */
  68 #define TRX_MAX_LEN     0x720000
  69 #define TRX_NO_HEADER   1               /* Do not write TRX header */
  70
  71 struct trx_header {
  72         uint32_t magic;                 /* "HDR0" */
  73         uint32_t len;                   /* Length of file including header */
  74         uint32_t crc32;                 /* 32-bit CRC from flag_version to end of file */
  75         uint32_t flag_version;  /* 0:15 flags, 16:31 version */
  76         uint32_t offsets[4];    /* Offsets of partitions from start of header */
  77 };
  78
  79 /**********************************************************************/
  80
  81 void usage(void) __attribute__ (( __noreturn__ ));
  82
  83 void usage(void)
  84 {
  85         fprintf(stderr, "Usage:\n");
  86         fprintf(stderr, " trx [-2] [-o outfile] [-m maxlen] [-a align] [-b absolute offset] [-x relative offset]\n");
  87         fprintf(stderr, "     [-f file] [-f file [-f file [-f file (v2 only)]]]\n");
  88         exit(EXIT_FAILURE);
  89 }
  90
  91 int main(int argc, char **argv)
  92 {
  93         FILE *out = stdout;
  94         FILE *in;
  95         char *ofn = NULL;
  96         char *buf;
  97         char *e;
  98         int c, i, append = 0;
  99         size_t n;
 100         ssize_t n2;
 101         uint32_t cur_len, fsmark=0;
 102         unsigned long maxlen = TRX_MAX_LEN;
 103         struct trx_header *p;
 104         char trx_version = 1;
 105         unsigned char binheader[32];
 106
 107         fprintf(stderr, "mjn3's trx replacement - v0.81.1\n");
 108
 109         if (!(buf = malloc(maxlen))) {
 110                 fprintf(stderr, "malloc failed\n");
 111                 return EXIT_FAILURE;
 112         }
 113
 114         p = (struct trx_header *) buf;
 115
 116         p->magic = STORE32_LE(TRX_MAGIC);
 117         cur_len = sizeof(struct trx_header) - 4; /* assume v1 header */
 118
 119         in = NULL;
 120         i = 0;
 121
 122         while ((c = getopt(argc, argv, "-:2o:m:a:x:b:f:A:F:")) != -1) {
 123                 switch (c) {
 124                         case '2':
 125                                 /* take care that nothing was written to buf so far */
 126                                 if (cur_len != sizeof(struct trx_header) - 4) {
 127                                         fprintf(stderr, "-2 has to be used before any other argument!\n");
 128                                 }
 129                                 else {
 130                                         trx_version = 2;
 131                                         cur_len += 4;
 132                                 }
 133                                 break;
 134                         case 'F':
 135                                 fsmark = cur_len;
 136                         case 'A':
 137                                 append = 1;
 138                                 /* fall through */
 139                         case 'f':
 140                         case 1:
 141                                 if (!append)
 142                                         p->offsets[i++] = STORE32_LE(cur_len);
 143
 144                                 if (!(in = fopen(optarg, "r"))) {
 145                                         fprintf(stderr, "can not open \"%s\" for reading\n", optarg);
 146                                         usage();
 147                                 }
 148                                 n = fread(buf + cur_len, 1, maxlen - cur_len, in);
 149                                 if (!feof(in)) {
 150                                         fprintf(stderr, "fread failure or file \"%s\" too large\n",optarg);
 151                                         fclose(in);
 152                                         return EXIT_FAILURE;
 153                                 }
 154                                 fclose(in);
 155 #undef  ROUND
 156 #define ROUND 4
 157                                 if (n & (ROUND-1)) {
 158                                         memset(buf + cur_len + n, 0, ROUND - (n & (ROUND-1)));
 159                                         n += ROUND - (n & (ROUND-1));
 160                                 }
 161                                 cur_len += n;
 162                                 append = 0;
 163
 164                                 break;
 165                         case 'o':
 166                                 ofn = optarg;
 167                                 if (ofn && !(out = fopen(ofn, "w"))) {
 168                                         fprintf(stderr, "can not open \"%s\" for writing\n", ofn);
 169                                         usage();
 170                                 }
 171
 172                                 break;
 173                         case 'm':
 174                                 errno = 0;
 175                                 maxlen = strtoul(optarg, &e, 0);
 176                                 if (errno || (e == optarg) || *e) {
 177                                         fprintf(stderr, "illegal numeric string\n");
 178                                         usage();
 179                                 }
 180 #undef  ROUND
 181 #define ROUND 0x1000
 182                                 if (maxlen & (ROUND-1)) {
 183                                         maxlen += (ROUND - (maxlen & (ROUND-1)));
 184                                 }
 185                                 if (maxlen < ROUND) {
 186                                         fprintf(stderr, "maxlen too small (or wrapped)\n");
 187                                         usage();
 188                                 }
 189                                 if (maxlen > TRX_MAX_LEN) {
 190                                         fprintf(stderr, "WARNING: maxlen exceeds default maximum!  Beware of overwriting nvram!\n");
 191                                 }
 192                                 if (!(buf = realloc(buf,maxlen))) {
 193                                         fprintf(stderr, "realloc failed");
 194                                         return EXIT_FAILURE;
 195                                 }
 196                                 p = (struct trx_header *) buf;
 197                                 break;
 198                         case 'a':
 199                                 errno = 0;
 200                                 n = strtoul(optarg, &e, 0);
 201                                 if (errno || (e == optarg) || *e) {
 202                                         fprintf(stderr, "illegal numeric string\n");
 203                                         usage();
 204                                 }
 205                                 if (cur_len & (n-1)) {
 206                                         n = n - (cur_len & (n-1));
 207                                         memset(buf + cur_len, 0, n);
 208                                         cur_len += n;
 209                                 }
 210                                 break;
 211                         case 'b':
 212                                 errno = 0;
 213                                 n = strtoul(optarg, &e, 0);
 214                                 if (errno || (e == optarg) || *e) {
 215                                         fprintf(stderr, "illegal numeric string\n");
 216                                         usage();
 217                                 }
 218                                 if (n < cur_len) {
 219                                         fprintf(stderr, "WARNING: current length exceeds -b %d offset\n",(int) n);
 220                                 } else {
 221                                         memset(buf + cur_len, 0, n - cur_len);
 222                                         cur_len = n;
 223                                 }
 224                                 break;
 225                         case 'x':
 226                                 errno = 0;
 227                                 n2 = strtol(optarg, &e, 0);
 228                                 if (errno || (e == optarg) || *e) {
 229                                         fprintf(stderr, "illegal numeric string\n");
 230                                         usage();
 231                                 }
 232                                 if (n2 < 0) {
 233                                         if (-n2 > cur_len) {
 234                                                 fprintf(stderr, "WARNING: current length smaller then -x %d offset\n",(int) n2);
 235                                                 cur_len = 0;
 236                                         } else
 237                                                 cur_len += n2;
 238                                 } else {
 239                                         memset(buf + cur_len, 0, n2);
 240                                         cur_len += n2;
 241                                 }
 242
 243                                 break;
 244                         default:
 245                                 usage();
 246                 }
 247         }
 248         p->flag_version = STORE32_LE((trx_version << 16));
 249
 250         if (!in) {
 251                 fprintf(stderr, "we require atleast one filename\n");
 252                 usage();
 253         }
 254
 255 #undef  ROUND
 256 #define ROUND 0x1000
 257         n = cur_len & (ROUND-1);
 258         if (n) {
 259                 memset(buf + cur_len, 0, ROUND - n);
 260                 cur_len += ROUND - n;
 261         }
 262
 263         /* for TRXv2 set bin-header Flags to 0xFF for CRC calculation like CFE does */
 264         if (trx_version == 2) {
 265                 if(cur_len - p->offsets[3] < sizeof(binheader)) {
 266                         fprintf(stderr, "TRXv2 binheader too small!\n");
 267                         return EXIT_FAILURE;
 268                 }
 269                 memcpy(binheader, buf + p->offsets[3], sizeof(binheader)); /* save header */
 270                 memset(buf + p->offsets[3] + 22, 0xFF, 8); /* set stable and try1-3 to 0xFF */
 271         }
 272
 273         p->crc32 = crc32buf((char *) &p->flag_version,
 274                                                 (fsmark)?fsmark:cur_len - offsetof(struct trx_header, flag_version));
 275         p->crc32 = STORE32_LE(p->crc32);
 276
 277         p->len = (fsmark)?fsmark:cur_len - offsetof(struct trx_header, flag_version);
 278
 279         /* restore TRXv2 bin-header */
 280         if (trx_version == 2) {
 281                 memcpy(buf + p->offsets[3], binheader, sizeof(binheader));
 282         }
 283
 284         if (!fwrite(buf, cur_len, 1, out) || fflush(out)) {
 285                 fprintf(stderr, "fwrite failed\n");
 286                 return EXIT_FAILURE;
 287         }
 288
 289         fclose(out);
 290
 291         return EXIT_SUCCESS;
 292 }
 293
 294 /**********************************************************************/
 295 /* The following was grabbed and tweaked from the old snippets collection
 296  * of public domain C code. */
 297
 298 /**********************************************************************\
 299 |* Demonstration program to compute the 32-bit CRC used as the frame  *|
 300 |* check sequence in ADCCP (ANSI X3.66, also known as FIPS PUB 71     *|
 301 |* and FED-STD-1003, the U.S. versions of CCITT's X.25 link-level     *|
 302 |* protocol).  The 32-bit FCS was added via the Federal Register,     *|
 303 |* 1 June 1982, p.23798.  I presume but don't know for certain that   *|
 304 |* this polynomial is or will be included in CCITT V.41, which        *|
 305 |* defines the 16-bit CRC (often called CRC-CCITT) polynomial.  FIPS  *|
 306 |* PUB 78 says that the 32-bit FCS reduces otherwise undetected       *|
 307 |* errors by a factor of 10^-5 over 16-bit FCS.                       *|
 308 \**********************************************************************/
 309
 310 /* Copyright (C) 1986 Gary S. Brown.  You may use this program, or
 311    code or tables extracted from it, as desired without restriction.*/
 312
 313 /* First, the polynomial itself and its table of feedback terms.  The  */
 314 /* polynomial is                                                       */
 315 /* X^32+X^26+X^23+X^22+X^16+X^12+X^11+X^10+X^8+X^7+X^5+X^4+X^2+X^1+X^0 */
 316 /* Note that we take it "backwards" and put the highest-order term in  */
 317 /* the lowest-order bit.  The X^32 term is "implied"; the LSB is the   */
 318 /* X^31 term, etc.  The X^0 term (usually shown as "+1") results in    */
 319 /* the MSB being 1.                                                    */
 320
 321 /* Note that the usual hardware shift register implementation, which   */
 322 /* is what we're using (we're merely optimizing it by doing eight-bit  */
 323 /* chunks at a time) shifts bits into the lowest-order term.  In our   */
 324 /* implementation, that means shifting towards the right.  Why do we   */
 325 /* do it this way?  Because the calculated CRC must be transmitted in  */
 326 /* order from highest-order term to lowest-order term.  UARTs transmit */
 327 /* characters in order from LSB to MSB.  By storing the CRC this way,  */
 328 /* we hand it to the UART in the order low-byte to high-byte; the UART */
 329 /* sends each low-bit to hight-bit; and the result is transmission bit */
 330 /* by bit from highest- to lowest-order term without requiring any bit */
 331 /* shuffling on our part.  Reception works similarly.                  */
 332
 333 /* The feedback terms table consists of 256, 32-bit entries.  Notes:   */
 334 /*                                                                     */
 335 /*  1. The table can be generated at runtime if desired; code to do so */
 336 /*     is shown later.  It might not be obvious, but the feedback      */
 337 /*     terms simply represent the results of eight shift/xor opera-    */
 338 /*     tions for all combinations of data and CRC register values.     */
 339 /*                                                                     */
 340 /*  2. The CRC accumulation logic is the same for all CRC polynomials, */
 341 /*     be they sixteen or thirty-two bits wide.  You simply choose the */
 342 /*     appropriate table.  Alternatively, because the table can be     */
 343 /*     generated at runtime, you can start by generating the table for */
 344 /*     the polynomial in question and use exactly the same "updcrc",   */
 345 /*     if your application needn't simultaneously handle two CRC       */
 346 /*     polynomials.  (Note, however, that XMODEM is strange.)          */
 347 /*                                                                     */
 348 /*  3. For 16-bit CRCs, the table entries need be only 16 bits wide;   */
 349 /*     of course, 32-bit entries work OK if the high 16 bits are zero. */
 350 /*                                                                     */
 351 /*  4. The values must be right-shifted by eight bits by the "updcrc"  */
 352 /*     logic; the shift must be unsigned (bring in zeroes).  On some   */
 353 /*     hardware you could probably optimize the shift in assembler by  */
 354 /*     using byte-swap instructions.                                   */
 355
 356 static const uint32_t crc_32_tab[] = { /* CRC polynomial 0xedb88320 */
 357 0x00000000, 0x77073096, 0xee0e612c, 0x990951ba, 0x076dc419, 0x706af48f,
 358 0xe963a535, 0x9e6495a3, 0x0edb8832, 0x79dcb8a4, 0xe0d5e91e, 0x97d2d988,
 359 0x09b64c2b, 0x7eb17cbd, 0xe7b82d07, 0x90bf1d91, 0x1db71064, 0x6ab020f2,
 360 0xf3b97148, 0x84be41de, 0x1adad47d, 0x6ddde4eb, 0xf4d4b551, 0x83d385c7,
 361 0x136c9856, 0x646ba8c0, 0xfd62f97a, 0x8a65c9ec, 0x14015c4f, 0x63066cd9,
 362 0xfa0f3d63, 0x8d080df5, 0x3b6e20c8, 0x4c69105e, 0xd56041e4, 0xa2677172,
 363 0x3c03e4d1, 0x4b04d447, 0xd20d85fd, 0xa50ab56b, 0x35b5a8fa, 0x42b2986c,
 364 0xdbbbc9d6, 0xacbcf940, 0x32d86ce3, 0x45df5c75, 0xdcd60dcf, 0xabd13d59,
 365 0x26d930ac, 0x51de003a, 0xc8d75180, 0xbfd06116, 0x21b4f4b5, 0x56b3c423,
 366 0xcfba9599, 0xb8bda50f, 0x2802b89e, 0x5f058808, 0xc60cd9b2, 0xb10be924,
 367 0x2f6f7c87, 0x58684c11, 0xc1611dab, 0xb6662d3d, 0x76dc4190, 0x01db7106,
 368 0x98d220bc, 0xefd5102a, 0x71b18589, 0x06b6b51f, 0x9fbfe4a5, 0xe8b8d433,
 369 0x7807c9a2, 0x0f00f934, 0x9609a88e, 0xe10e9818, 0x7f6a0dbb, 0x086d3d2d,
 370 0x91646c97, 0xe6635c01, 0x6b6b51f4, 0x1c6c6162, 0x856530d8, 0xf262004e,
 371 0x6c0695ed, 0x1b01a57b, 0x8208f4c1, 0xf50fc457, 0x65b0d9c6, 0x12b7e950,
 372 0x8bbeb8ea, 0xfcb9887c, 0x62dd1ddf, 0x15da2d49, 0x8cd37cf3, 0xfbd44c65,
 373 0x4db26158, 0x3ab551ce, 0xa3bc0074, 0xd4bb30e2, 0x4adfa541, 0x3dd895d7,
 374 0xa4d1c46d, 0xd3d6f4fb, 0x4369e96a, 0x346ed9fc, 0xad678846, 0xda60b8d0,
 375 0x44042d73, 0x33031de5, 0xaa0a4c5f, 0xdd0d7cc9, 0x5005713c, 0x270241aa,
 376 0xbe0b1010, 0xc90c2086, 0x5768b525, 0x206f85b3, 0xb966d409, 0xce61e49f,
 377 0x5edef90e, 0x29d9c998, 0xb0d09822, 0xc7d7a8b4, 0x59b33d17, 0x2eb40d81,
 378 0xb7bd5c3b, 0xc0ba6cad, 0xedb88320, 0x9abfb3b6, 0x03b6e20c, 0x74b1d29a,
 379 0xead54739, 0x9dd277af, 0x04db2615, 0x73dc1683, 0xe3630b12, 0x94643b84,
 380 0x0d6d6a3e, 0x7a6a5aa8, 0xe40ecf0b, 0x9309ff9d, 0x0a00ae27, 0x7d079eb1,
 381 0xf00f9344, 0x8708a3d2, 0x1e01f268, 0x6906c2fe, 0xf762575d, 0x806567cb,
 382 0x196c3671, 0x6e6b06e7, 0xfed41b76, 0x89d32be0, 0x10da7a5a, 0x67dd4acc,
 383 0xf9b9df6f, 0x8ebeeff9, 0x17b7be43, 0x60b08ed5, 0xd6d6a3e8, 0xa1d1937e,
 384 0x38d8c2c4, 0x4fdff252, 0xd1bb67f1, 0xa6bc5767, 0x3fb506dd, 0x48b2364b,
 385 0xd80d2bda, 0xaf0a1b4c, 0x36034af6, 0x41047a60, 0xdf60efc3, 0xa867df55,
 386 0x316e8eef, 0x4669be79, 0xcb61b38c, 0xbc66831a, 0x256fd2a0, 0x5268e236,
 387 0xcc0c7795, 0xbb0b4703, 0x220216b9, 0x5505262f, 0xc5ba3bbe, 0xb2bd0b28,
 388 0x2bb45a92, 0x5cb36a04, 0xc2d7ffa7, 0xb5d0cf31, 0x2cd99e8b, 0x5bdeae1d,
 389 0x9b64c2b0, 0xec63f226, 0x756aa39c, 0x026d930a, 0x9c0906a9, 0xeb0e363f,
 390 0x72076785, 0x05005713, 0x95bf4a82, 0xe2b87a14, 0x7bb12bae, 0x0cb61b38,
 391 0x92d28e9b, 0xe5d5be0d, 0x7cdcefb7, 0x0bdbdf21, 0x86d3d2d4, 0xf1d4e242,
 392 0x68ddb3f8, 0x1fda836e, 0x81be16cd, 0xf6b9265b, 0x6fb077e1, 0x18b74777,
 393 0x88085ae6, 0xff0f6a70, 0x66063bca, 0x11010b5c, 0x8f659eff, 0xf862ae69,
 394 0x616bffd3, 0x166ccf45, 0xa00ae278, 0xd70dd2ee, 0x4e048354, 0x3903b3c2,
 395 0xa7672661, 0xd06016f7, 0x4969474d, 0x3e6e77db, 0xaed16a4a, 0xd9d65adc,
 396 0x40df0b66, 0x37d83bf0, 0xa9bcae53, 0xdebb9ec5, 0x47b2cf7f, 0x30b5ffe9,
 397 0xbdbdf21c, 0xcabac28a, 0x53b39330, 0x24b4a3a6, 0xbad03605, 0xcdd70693,
 398 0x54de5729, 0x23d967bf, 0xb3667a2e, 0xc4614ab8, 0x5d681b02, 0x2a6f2b94,
 399 0xb40bbe37, 0xc30c8ea1, 0x5a05df1b, 0x2d02ef8d
 400 };
 401
 402 #define UPDC32(octet,crc) (crc_32_tab[((crc) ^ (octet)) & 0xff] ^ ((crc) >> 8))
 403
 404 uint32_t crc32buf(char *buf, size_t len)
 405 {
 406       uint32_t crc;
 407
 408       crc = 0xFFFFFFFF;
 409
 410       for ( ; len; --len, ++buf)
 411       {
 412             crc = UPDC32(*buf, crc);
 413       }
 414
 415       return crc;
 416 }