tools/firmware-utils/src/mkzynfw.c

   1 /*
   2  *
   3  *  Copyright (C) 2007-2008 OpenWrt.org
   4  *  Copyright (C) 2007-2008 Gabor Juhos <juhosg at openwrt.org>
   5  *
   6  *  This code was based on the information of the ZyXEL's firmware
   7  *  image format written by Kolja Waschk, can be found at:
   8  *  http://www.ixo.de/info/zyxel_uclinux
   9  *
  10  *  This program is free software; you can redistribute it and/or modify it
  11  *  under the terms of the GNU General Public License version 2 as published
  12  *  by the Free Software Foundation.
  13  *
  14  */
  15
  16 #include <stdio.h>
  17 #include <stdlib.h>
  18 #include <stdint.h>
  19 #include <string.h>
  20 #include <unistd.h>     /* for unlink() */
  21 #include <libgen.h>
  22 #include <getopt.h>     /* for getopt() */
  23 #include <stdarg.h>
  24 #include <errno.h>
  25 #include <sys/stat.h>
  26 #include <endian.h>     /* for __BYTE_ORDER */
  27 #if defined(__CYGWIN__)
  28 #  include <byteswap.h>
  29 #endif
  30 #include <inttypes.h>
  31
  32 #include "zynos.h"
  33
  34 #if (__BYTE_ORDER == __LITTLE_ENDIAN)
  35 #  define HOST_TO_LE16(x)       (x)
  36 #  define HOST_TO_LE32(x)       (x)
  37 #  define LE16_TO_HOST(x)       (x)
  38 #  define LE32_TO_HOST(x)       (x)
  39 #  define HOST_TO_BE16(x)       bswap_16(x)
  40 #  define HOST_TO_BE32(x)       bswap_32(x)
  41 #  define BE16_TO_HOST(x)       bswap_16(x)
  42 #  define BE32_TO_HOST(x)       bswap_32(x)
  43 #else
  44 #  define HOST_TO_BE16(x)       (x)
  45 #  define HOST_TO_BE32(x)       (x)
  46 #  define BE16_TO_HOST(x)       (x)
  47 #  define BE32_TO_HOST(x)       (x)
  48 #  define HOST_TO_LE16(x)       bswap_16(x)
  49 #  define HOST_TO_LE32(x)       bswap_32(x)
  50 #  define LE16_TO_HOST(x)       bswap_16(x)
  51 #  define LE32_TO_HOST(x)       bswap_32(x)
  52 #endif
  53
  54 #define ALIGN(x,y)      (((x)+((y)-1)) & ~((y)-1))
  55
  56 #define MAX_NUM_BLOCKS  8
  57 #define MAX_ARG_COUNT   32
  58 #define MAX_ARG_LEN     1024
  59 #define FILE_BUF_LEN    (16*1024)
  60
  61
  62 struct csum_state{
  63         int             odd;
  64         uint32_t        sum;
  65         uint32_t        tmp;
  66 };
  67
  68 struct fw_block {
  69         uint32_t        align;          /* alignment of this block */
  70         char            *file_name;     /* name of the file */
  71         uint32_t        file_size;      /* length of the file */
  72         char            *mmap_name;     /* name in the MMAP table */
  73         int             type;           /* block type */
  74         uint32_t        padlen;
  75         uint8_t         padc;
  76 };
  77
  78 #define BLOCK_TYPE_BOOTEXT      0
  79 #define BLOCK_TYPE_RAW          1
  80
  81 struct fw_mmap {
  82         uint32_t        addr;
  83         uint32_t        size;
  84         uint32_t        user_addr;
  85         uint32_t        user_size;
  86 };
  87 #define MMAP_DATA_SIZE  1024
  88 #define MMAP_ALIGN      16
  89
  90 struct board_info {
  91         char *name;             /* model name */
  92         char *desc;             /* description */
  93         uint16_t vendor;        /* vendor id */
  94         uint16_t model;         /* model id */
  95         uint32_t flash_base;    /* flash base address */
  96         uint32_t flash_size;    /* board flash size */
  97         uint32_t code_start;    /* code start address */
  98         uint32_t romio_offs;    /* offset of the firmware within the flash */
  99         uint32_t bootext_size;  /* maximum size of bootext block */
 100 };
 101
 102 /*
 103  * Globals
 104  */
 105 char *progname;
 106 char *ofname = NULL;
 107 int verblevel = 0;
 108
 109 struct board_info *board = NULL;
 110
 111 struct fw_block blocks[MAX_NUM_BLOCKS];
 112 struct fw_block *bootext_block = NULL;
 113 int num_blocks = 0;
 114
 115 #define ADM5120_FLASH_BASE      0xBFC00000
 116 #define ADM5120_CODE_START      0x80008000
 117
 118 /* TODO: check values for AR7 */
 119 #define AR7_FLASH_BASE          0xB0000000
 120 #define AR7_CODE_START          0x94008000
 121
 122 #define ATHEROS_FLASH_BASE      0xBFC00000
 123 #define ATHEROS_CODE_START      0x80e00000
 124
 125 #define AR71XX_FLASH_BASE       0xBFC00000
 126 #define AR71XX_CODE_START       0x81E00000
 127
 128 #define BOARD(n, d, v, m, fb, fs, cs, fo) {             \
 129         .name = (n), .desc=(d),                         \
 130         .vendor = (v), .model = (m),                    \
 131         .flash_base = (fb), .flash_size = (fs)<<20,     \
 132         .code_start = (cs), .romio_offs = (fo),         \
 133         .bootext_size = BOOTEXT_DEF_SIZE                \
 134         }
 135
 136 #define ADMBOARD1(n, d, m, fs) BOARD(n, d, ZYNOS_VENDOR_ID_ZYXEL, m, \
 137         ADM5120_FLASH_BASE, fs, ADM5120_CODE_START, 0x8000)
 138
 139 #define ADMBOARD2(n, d, m, fs) BOARD(n, d, ZYNOS_VENDOR_ID_ZYXEL, m, \
 140         ADM5120_FLASH_BASE, fs, ADM5120_CODE_START, 0x10000)
 141
 142 #define AR7BOARD1(n, d, m, fs) BOARD(n, d, ZYNOS_VENDOR_ID_ZYXEL, m, \
 143         AR7_FLASH_BASE, fs, AR7_CODE_START, 0x8000)
 144
 145 #define ATHEROSBOARD1(n, d, m, fs) BOARD(n, d, ZYNOS_VENDOR_ID_ZYXEL, m, \
 146         ATHEROS_FLASH_BASE, fs, ATHEROS_CODE_START, 0x30000)
 147
 148 #define AR71XXBOARD1(n, d, m, fs) {             \
 149         .name = (n), .desc=(d),                         \
 150         .vendor = (ZYNOS_VENDOR_ID_ZYXEL), .model = (m),                        \
 151         .flash_base = (AR71XX_FLASH_BASE), .flash_size = (fs)<<20,      \
 152         .code_start = (AR71XX_CODE_START), .romio_offs = (0x40000),             \
 153         .bootext_size = 0x30000         \
 154         }
 155
 156
 157 static struct board_info boards[] = {
 158         /*
 159          * Infineon/ADMtek ADM5120 based boards
 160          */
 161         ADMBOARD2("ES-2024A",   "ZyXEL ES-2024A", ZYNOS_MODEL_ES_2024A, 4),
 162         ADMBOARD2("ES-2024PWR", "ZyXEL ES-2024PWR", ZYNOS_MODEL_ES_2024PWR, 4),
 163         ADMBOARD2("ES-2108",    "ZyXEL ES-2108", ZYNOS_MODEL_ES_2108, 4),
 164         ADMBOARD2("ES-2108-F",  "ZyXEL ES-2108-F", ZYNOS_MODEL_ES_2108_F, 4),
 165         ADMBOARD2("ES-2108-G",  "ZyXEL ES-2108-G", ZYNOS_MODEL_ES_2108_G, 4),
 166         ADMBOARD2("ES-2108-LC", "ZyXEL ES-2108-LC", ZYNOS_MODEL_ES_2108_LC, 4),
 167         ADMBOARD2("ES-2108PWR", "ZyXEL ES-2108PWR", ZYNOS_MODEL_ES_2108PWR, 4),
 168         ADMBOARD1("HS-100",     "ZyXEL HomeSafe 100", ZYNOS_MODEL_HS_100, 2),
 169         ADMBOARD1("HS-100W",    "ZyXEL HomeSafe 100W", ZYNOS_MODEL_HS_100W, 2),
 170         ADMBOARD1("P-334",      "ZyXEL Prestige 334", ZYNOS_MODEL_P_334, 2),
 171         ADMBOARD1("P-334U",     "ZyXEL Prestige 334U", ZYNOS_MODEL_P_334U, 4),
 172         ADMBOARD1("P-334W",     "ZyXEL Prestige 334W", ZYNOS_MODEL_P_334W, 2),
 173         ADMBOARD1("P-334WH",    "ZyXEL Prestige 334WH", ZYNOS_MODEL_P_334WH, 4),
 174         ADMBOARD1("P-334WHD",   "ZyXEL Prestige 334WHD", ZYNOS_MODEL_P_334WHD, 4),
 175         ADMBOARD1("P-334WT",    "ZyXEL Prestige 334WT", ZYNOS_MODEL_P_334WT, 4),
 176         ADMBOARD1("P-335",      "ZyXEL Prestige 335", ZYNOS_MODEL_P_335, 4),
 177         ADMBOARD1("P-335Plus",  "ZyXEL Prestige 335Plus", ZYNOS_MODEL_P_335PLUS, 4),
 178         ADMBOARD1("P-335U",     "ZyXEL Prestige 335U", ZYNOS_MODEL_P_335U, 4),
 179         ADMBOARD1("P-335WT",    "ZyXEL Prestige 335WT", ZYNOS_MODEL_P_335WT, 4),
 180
 181         {
 182                 .name           = "P-2602HW-D1A",
 183                 .desc           = "ZyXEL P-2602HW-D1A",
 184                 .vendor         = ZYNOS_VENDOR_ID_ZYXEL,
 185                 .model          = ZYNOS_MODEL_P_2602HW_D1A,
 186                 .flash_base     = AR7_FLASH_BASE,
 187                 .flash_size     = 4*1024*1024,
 188                 .code_start     = 0x94008000,
 189                 .romio_offs     = 0x20000,
 190                 .bootext_size   = BOOTEXT_DEF_SIZE,
 191         },
 192
 193 #if 0
 194         /*
 195          * Texas Instruments AR7 based boards
 196          */
 197         AR7BOARD1("P-660H-61",  "ZyXEL P-660H-61", ZYNOS_MODEL_P_660H_61, 2),
 198         AR7BOARD1("P-660H-63",  "ZyXEL P-660H-63", ZYNOS_MODEL_P_660H_63, 2),
 199         AR7BOARD1("P-660H-D1",  "ZyXEL P-660H-D1", ZYNOS_MODEL_P_660H_D1, 2),
 200         AR7BOARD1("P-660H-D3",  "ZyXEL P-660H-D3", ZYNOS_MODEL_P_660H_D3, 2),
 201         AR7BOARD1("P-660HW-61", "ZyXEL P-660HW-61", ZYNOS_MODEL_P_660HW_61, 2),
 202         AR7BOARD1("P-660HW-63", "ZyXEL P-660HW-63", ZYNOS_MODEL_P_660HW_63, 2),
 203         AR7BOARD1("P-660HW-67", "ZyXEL P-660HW-67", ZYNOS_MODEL_P_660HW_67, 2),
 204         AR7BOARD1("P-660HW-D1", "ZyXEL P-660HW-D1", ZYNOS_MODEL_P_660HW_D1, 2),
 205         AR7BOARD1("P-660HW-D3", "ZyXEL P-660HW-D3", ZYNOS_MODEL_P_660HW_D3, 2),
 206         AR7BOARD1("P-660R-61",  "ZyXEL P-660R-61", ZYNOS_MODEL_P_660R_61, 2),
 207         AR7BOARD1("P-660R-61C", "ZyXEL P-660R-61C", ZYNOS_MODEL_P_660R_61C, 2),
 208         AR7BOARD1("P-660R-63",  "ZyXEL P-660R-63", ZYNOS_MODEL_P_660R_63, 2),
 209         AR7BOARD1("P-660R-63C", "ZyXEL P-660R-63C", ZYNOS_MODEL_P_660R_63C, 2),
 210         AR7BOARD1("P-660R-67",  "ZyXEL P-660R-67", ZYNOS_MODEL_P_660R_67, 2),
 211         AR7BOARD1("P-660R-D1",  "ZyXEL P-660R-D1", ZYNOS_MODEL_P_660R_D1, 2),
 212         AR7BOARD1("P-660R-D3",  "ZyXEL P-660R-D3", ZYNOS_MODEL_P_660R_D3, 2),
 213 #endif
 214         {
 215                 .name           = "O2SURF",
 216                 .desc           = "O2 DSL Surf & Phone",
 217                 .vendor         = ZYNOS_VENDOR_ID_O2,
 218                 .model          = ZYNOS_MODEL_O2SURF,
 219                 .flash_base     = AR7_FLASH_BASE,
 220                 .flash_size     = 8*1024*1024,
 221                 .code_start     = 0x94014000,
 222                 .romio_offs     = 0x40000,
 223                 .bootext_size   = BOOTEXT_DEF_SIZE,
 224         },
 225
 226         /*
 227 :x
 228          */
 229         ATHEROSBOARD1("NBG-318S", "ZyXEL NBG-318S", ZYNOS_MODEL_NBG_318S, 4),
 230
 231         /*
 232          * Atheros ar71xx based boards
 233          */
 234         AR71XXBOARD1("NBG-460N", "ZyXEL NBG-460N", ZYNOS_MODEL_NBG_460N, 4),
 235
 236         {.name = NULL}
 237 };
 238
 239 /*
 240  * Message macros
 241  */
 242 #define ERR(fmt, ...) do { \
 243         fflush(0); \
 244         fprintf(stderr, "[%s] *** error: " fmt "\n", \
 245                         progname, ## __VA_ARGS__ ); \
 246 } while (0)
 247
 248 #define ERRS(fmt, ...) do { \
 249         int save = errno; \
 250         fflush(0); \
 251         fprintf(stderr, "[%s] *** error: " fmt ", %s\n", \
 252                         progname, ## __VA_ARGS__, strerror(save)); \
 253 } while (0)
 254
 255 #define WARN(fmt, ...) do { \
 256         fprintf(stderr, "[%s] *** warning: " fmt "\n", \
 257                         progname, ## __VA_ARGS__ ); \
 258 } while (0)
 259
 260 #define DBG(lev, fmt, ...) do { \
 261         if (verblevel < lev) \
 262                 break;\
 263         fprintf(stderr, "[%s] " fmt "\n", progname, ## __VA_ARGS__ ); \
 264 } while (0)
 265
 266 #define ERR_FATAL               -1
 267 #define ERR_INVALID_IMAGE       -2
 268
 269 /*
 270  * Helper routines
 271  */
 272 void
 273 usage(int status)
 274 {
 275         FILE *stream = (status != EXIT_SUCCESS) ? stderr : stdout;
 276         struct board_info *board;
 277
 278         fprintf(stream, "Usage: %s [OPTIONS...]\n", progname);
 279         fprintf(stream,
 280 "\n"
 281 "Options:\n"
 282 "  -B <board>      create image for the board specified with <board>.\n"
 283 "                  valid <board> values:\n"
 284         );
 285         for (board = boards; board->name != NULL; board++){
 286                 fprintf(stream,
 287 "                    %-12s= %s\n",
 288                  board->name, board->desc);
 289         };
 290         fprintf(stream,
 291 "  -b <file>[:<align>]\n"
 292 "                  add boot extension block to the image\n"
 293 "  -r <file>[:<align>]\n"
 294 "                  add raw block to the image\n"
 295 "  -o <file>       write output to the file <file>\n"
 296 "  -h              show this screen\n"
 297         );
 298
 299         exit(status);
 300 }
 301
 302
 303 /*
 304  * argument parsing
 305  */
 306 int
 307 str2u32(char *arg, uint32_t *val)
 308 {
 309         char *err = NULL;
 310         uint32_t t;
 311
 312         errno=0;
 313         t = strtoul(arg, &err, 0);
 314         if (errno || (err==arg) || ((err != NULL) && *err)) {
 315                 return -1;
 316         }
 317
 318         *val = t;
 319         return 0;
 320 }
 321
 322
 323 int
 324 str2u16(char *arg, uint16_t *val)
 325 {
 326         char *err = NULL;
 327         uint32_t t;
 328
 329         errno=0;
 330         t = strtoul(arg, &err, 0);
 331         if (errno || (err==arg) || ((err != NULL) && *err) || (t >= 0x10000)) {
 332                 return -1;
 333         }
 334
 335         *val = t & 0xFFFF;
 336         return 0;
 337 }
 338
 339 int
 340 str2u8(char *arg, uint8_t *val)
 341 {
 342         char *err = NULL;
 343         uint32_t t;
 344
 345         errno=0;
 346         t = strtoul(arg, &err, 0);
 347         if (errno || (err==arg) || ((err != NULL) && *err) || (t >= 0x100)) {
 348                 return -1;
 349         }
 350
 351         *val = t & 0xFF;
 352         return 0;
 353 }
 354
 355 int
 356 str2sig(char *arg, uint32_t *sig)
 357 {
 358         if (strlen(arg) != 4)
 359                 return -1;
 360
 361         *sig = arg[0] | (arg[1] << 8) | (arg[2] << 16) | (arg[3] << 24);
 362
 363         return 0;
 364 }
 365
 366
 367 int
 368 parse_arg(char *arg, char *buf, char *argv[])
 369 {
 370         int res = 0;
 371         size_t argl;
 372         char *tok;
 373         char **ap = &buf;
 374         int i;
 375
 376         memset(argv, 0, MAX_ARG_COUNT * sizeof(void *));
 377
 378         if ((arg == NULL)) {
 379                 /* no arguments */
 380                 return 0;
 381         }
 382
 383         argl = strlen(arg);
 384         if (argl == 0) {
 385                 /* no arguments */
 386                 return 0;
 387         }
 388
 389         if (argl >= MAX_ARG_LEN) {
 390                 /* argument is too long */
 391                 argl = MAX_ARG_LEN-1;
 392         }
 393
 394         memcpy(buf, arg, argl);
 395         buf[argl] = '\0';
 396
 397         for (i = 0; i < MAX_ARG_COUNT; i++) {
 398                 tok = strsep(ap, ":");
 399                 if (tok == NULL) {
 400                         break;
 401                 }
 402 #if 0
 403                 else if (tok[0] == '\0') {
 404                         break;
 405                 }
 406 #endif
 407                 argv[i] = tok;
 408                 res++;
 409         }
 410
 411         return res;
 412 }
 413
 414
 415 int
 416 required_arg(char c, char *arg)
 417 {
 418         if (arg == NULL || *arg != '-')
 419                 return 0;
 420
 421         ERR("option -%c requires an argument\n", c);
 422         return -1;
 423 }
 424
 425
 426 int
 427 is_empty_arg(char *arg)
 428 {
 429         int ret = 1;
 430         if (arg != NULL) {
 431                 if (*arg) ret = 0;
 432         };
 433         return ret;
 434 }
 435
 436
 437 void
 438 csum_init(struct csum_state *css)
 439 {
 440         css->odd = 0;
 441         css->sum = 0;
 442         css->tmp = 0;
 443 }
 444
 445
 446 void
 447 csum_update(uint8_t *p, uint32_t len, struct csum_state *css)
 448 {
 449         if (len == 0)
 450                 return;
 451
 452         if (css->odd) {
 453                 css->sum += (css->tmp << 8) + p[0];
 454                 if (css->sum > 0xFFFF) {
 455                         css->sum += 1;
 456                         css->sum &= 0xFFFF;
 457                 }
 458                 css->odd = 0;
 459                 len--;
 460                 p++;
 461         }
 462
 463         for ( ; len > 1; len -= 2, p +=2 ) {
 464                 css->sum  += (p[0] << 8) + p[1];
 465                 if (css->sum > 0xFFFF) {
 466                         css->sum += 1;
 467                         css->sum &= 0xFFFF;
 468                 }
 469         }
 470
 471         if (len == 1){
 472                 css->tmp = p[0];
 473                 css->odd = 1;
 474         }
 475 }
 476
 477
 478 uint16_t
 479 csum_get(struct csum_state *css)
 480 {
 481         char pad = 0;
 482
 483         csum_update(&pad, 1, css);
 484         return css->sum;
 485 }
 486
 487 uint16_t
 488 csum_buf(uint8_t *p, uint32_t len)
 489 {
 490         struct csum_state css;
 491
 492         csum_init(&css);
 493         csum_update(p, len, &css);
 494         return csum_get(&css);
 495
 496 }
 497
 498 /*
 499  * routines to write data to the output file
 500  */
 501 int
 502 write_out_data(FILE *outfile, uint8_t *data, size_t len,
 503                 struct csum_state *css)
 504 {
 505         errno = 0;
 506
 507         fwrite(data, len, 1, outfile);
 508         if (errno) {
 509                 ERR("unable to write output file");
 510                 return -1;
 511         }
 512
 513         if (css) {
 514                 csum_update(data, len, css);
 515         }
 516
 517         return 0;
 518 }
 519
 520
 521 int
 522 write_out_padding(FILE *outfile, size_t len, uint8_t padc,
 523                  struct csum_state *css)
 524 {
 525         uint8_t buf[512];
 526         size_t buflen = sizeof(buf);
 527
 528         memset(buf, padc, buflen);
 529         while (len > 0) {
 530                 if (len < buflen)
 531                         buflen = len;
 532
 533                 if (write_out_data(outfile, buf, buflen, css))
 534                         return -1;
 535
 536                 len -= buflen;
 537         }
 538
 539         return 0;
 540 }
 541
 542
 543 int
 544 write_out_data_align(FILE *outfile, uint8_t *data, size_t len, size_t align,
 545                 struct csum_state *css)
 546 {
 547         size_t padlen;
 548         int res;
 549
 550         res = write_out_data(outfile, data, len, css);
 551         if (res)
 552                 return res;
 553
 554         padlen = ALIGN(len,align) - len;
 555         res = write_out_padding(outfile, padlen, 0xFF, css);
 556
 557         return res;
 558 }
 559
 560
 561 int
 562 write_out_header(FILE *outfile, struct zyn_rombin_hdr *hdr)
 563 {
 564         struct zyn_rombin_hdr t;
 565
 566         errno = 0;
 567         if (fseek(outfile, 0, SEEK_SET) != 0) {
 568                 ERRS("fseek failed on output file");
 569                 return -1;
 570         }
 571
 572         /* setup temporary header fields */
 573         memset(&t, 0, sizeof(t));
 574         t.addr = HOST_TO_BE32(hdr->addr);
 575         memcpy(&t.sig, ROMBIN_SIGNATURE, ROMBIN_SIG_LEN);
 576         t.type = hdr->type;
 577         t.flags = hdr->flags;
 578         t.osize = HOST_TO_BE32(hdr->osize);
 579         t.csize = HOST_TO_BE32(hdr->csize);
 580         t.ocsum = HOST_TO_BE16(hdr->ocsum);
 581         t.ccsum = HOST_TO_BE16(hdr->ccsum);
 582         t.mmap_addr = HOST_TO_BE32(hdr->mmap_addr);
 583
 584         DBG(2, "hdr.addr      = 0x%08x", hdr->addr);
 585         DBG(2, "hdr.type      = 0x%02x", hdr->type);
 586         DBG(2, "hdr.osize     = 0x%08x", hdr->osize);
 587         DBG(2, "hdr.csize     = 0x%08x", hdr->csize);
 588         DBG(2, "hdr.flags     = 0x%02x", hdr->flags);
 589         DBG(2, "hdr.ocsum     = 0x%04x", hdr->ocsum);
 590         DBG(2, "hdr.ccsum     = 0x%04x", hdr->ccsum);
 591         DBG(2, "hdr.mmap_addr = 0x%08x", hdr->mmap_addr);
 592
 593         return write_out_data(outfile, (uint8_t *)&t, sizeof(t), NULL);
 594 }
 595
 596
 597 int
 598 write_out_mmap(FILE *outfile, struct fw_mmap *mmap, struct csum_state *css)
 599 {
 600         struct zyn_mmt_hdr *mh;
 601         uint8_t buf[MMAP_DATA_SIZE];
 602         uint32_t user_size;
 603         char *data;
 604         int res;
 605
 606         memset(buf, 0, sizeof(buf));
 607
 608         mh = (struct zyn_mmt_hdr *)buf;
 609
 610         /* TODO: needs to recreate the memory map too? */
 611         mh->count=0;
 612
 613         /* Build user data section */
 614         data = buf+sizeof(*mh);
 615         data += sprintf(data, "Vendor 1 %d", board->vendor);
 616         *data++ = '\0';
 617         data += sprintf(data, "Model 1 %d", BE16_TO_HOST(board->model));
 618         *data++ = '\0';
 619         /* TODO: make hardware version configurable? */
 620         data += sprintf(data, "HwVerRange 2 %d %d", 0, 0);
 621         *data++ = '\0';
 622
 623         user_size = (uint8_t *)data - buf;
 624         mh->user_start= HOST_TO_BE32(mmap->addr+sizeof(*mh));
 625         mh->user_end= HOST_TO_BE32(mmap->addr+user_size);
 626         mh->csum = HOST_TO_BE16(csum_buf(buf+sizeof(*mh), user_size));
 627
 628         res = write_out_data(outfile, buf, sizeof(buf), css);
 629
 630         return res;
 631 }
 632
 633
 634 int
 635 block_stat_file(struct fw_block *block)
 636 {
 637         struct stat st;
 638         int res;
 639
 640         if (block->file_name == NULL)
 641                 return 0;
 642
 643         res = stat(block->file_name, &st);
 644         if (res){
 645                 ERRS("stat failed on %s", block->file_name);
 646                 return res;
 647         }
 648
 649         block->file_size = st.st_size;
 650         return 0;
 651 }
 652
 653
 654 int
 655 read_magic(uint16_t *magic)
 656 {
 657         FILE *f;
 658         int res;
 659
 660         errno = 0;
 661         f = fopen(bootext_block->file_name,"r");
 662         if (errno) {
 663                 ERRS("unable to open file: %s", bootext_block->file_name);
 664                 return -1;
 665         }
 666
 667         errno = 0;
 668         fread(magic, 2, 1, f);
 669         if (errno != 0) {
 670                 ERRS("unable to read from file: %s", bootext_block->file_name);
 671                 res = -1;
 672                 goto err;
 673         }
 674
 675         res = 0;
 676
 677 err:
 678         fclose(f);
 679         return res;
 680 }
 681
 682
 683 int
 684 write_out_file(FILE *outfile, char *name, size_t len, struct csum_state *css)
 685 {
 686         char buf[FILE_BUF_LEN];
 687         size_t buflen = sizeof(buf);
 688         FILE *f;
 689         int res;
 690
 691         DBG(2, "writing out file, name=%s, len=%zu",
 692                 name, len);
 693
 694         errno = 0;
 695         f = fopen(name,"r");
 696         if (errno) {
 697                 ERRS("unable to open file: %s", name);
 698                 return -1;
 699         }
 700
 701         while (len > 0) {
 702                 if (len < buflen)
 703                         buflen = len;
 704
 705                 /* read data from source file */
 706                 errno = 0;
 707                 fread(buf, buflen, 1, f);
 708                 if (errno != 0) {
 709                         ERRS("unable to read from file: %s",name);
 710                         res = -1;
 711                         break;
 712                 }
 713
 714                 res = write_out_data(outfile, buf, buflen, css);
 715                 if (res)
 716                         break;
 717
 718                 len -= buflen;
 719         }
 720
 721         fclose(f);
 722         return res;
 723 }
 724
 725
 726 int
 727 write_out_block(FILE *outfile, struct fw_block *block, struct csum_state *css)
 728 {
 729         int res;
 730
 731         if (block == NULL)
 732                 return 0;
 733
 734         if (block->file_name == NULL)
 735                 return 0;
 736
 737         if (block->file_size == 0)
 738                 return 0;
 739
 740         res = write_out_file(outfile, block->file_name,
 741                         block->file_size, css);
 742         return res;
 743 }
 744
 745
 746 int
 747 write_out_image(FILE *outfile)
 748 {
 749         struct fw_block *block;
 750         struct fw_mmap mmap;
 751         struct zyn_rombin_hdr hdr;
 752         struct csum_state css;
 753         int i, res;
 754         uint32_t offset;
 755         uint32_t padlen;
 756         uint16_t csum;
 757         uint16_t t;
 758
 759         /* setup header fields */
 760         memset(&hdr, 0, sizeof(hdr));
 761         hdr.addr = board->code_start;
 762         hdr.type = OBJECT_TYPE_BOOTEXT;
 763         hdr.flags = ROMBIN_FLAG_OCSUM;
 764
 765         offset = board->romio_offs;
 766
 767         res = write_out_header(outfile, &hdr);
 768         if (res)
 769                 return res;
 770
 771         offset += sizeof(hdr);
 772
 773         csum_init(&css);
 774         res = write_out_block(outfile, bootext_block, &css);
 775         if (res)
 776                 return res;
 777
 778         offset += bootext_block->file_size;
 779         if (offset > (board->romio_offs + board->bootext_size)) {
 780                 ERR("bootext file '%s' is too big", bootext_block->file_name);
 781                 return -1;
 782         }
 783
 784         padlen = ALIGN(offset, MMAP_ALIGN) - offset;
 785         res = write_out_padding(outfile, padlen, 0xFF, &css);
 786         if (res)
 787                 return res;
 788
 789         offset += padlen;
 790
 791         mmap.addr = board->flash_base + offset;
 792         res = write_out_mmap(outfile, &mmap, &css);
 793         if (res)
 794                 return res;
 795
 796         offset += MMAP_DATA_SIZE;
 797
 798         if ((offset - board->romio_offs) < board->bootext_size) {
 799                 padlen = board->romio_offs + board->bootext_size - offset;
 800                 res = write_out_padding(outfile, padlen, 0xFF, &css);
 801                 if (res)
 802                         return res;
 803                 offset += padlen;
 804
 805                 DBG(2, "bootext end at %08x", offset);
 806         }
 807
 808         for (i = 0; i < num_blocks; i++) {
 809                 block = &blocks[i];
 810
 811                 if (block->type == BLOCK_TYPE_BOOTEXT)
 812                         continue;
 813
 814                 padlen = ALIGN(offset, block->align) - offset;
 815                 res = write_out_padding(outfile, padlen, 0xFF, &css);
 816                 if (res)
 817                         return res;
 818                 offset += padlen;
 819
 820                 res = write_out_block(outfile, block, &css);
 821                 if (res)
 822                         return res;
 823                 offset += block->file_size;
 824         }
 825
 826         padlen = ALIGN(offset, 4) - offset;
 827         res = write_out_padding(outfile, padlen, 0xFF, &css);
 828         if (res)
 829                 return res;
 830         offset += padlen;
 831
 832         csum = csum_get(&css);
 833         hdr.mmap_addr = mmap.addr;
 834         hdr.osize = 2;
 835
 836         res = read_magic(&hdr.ocsum);
 837         if (res)
 838                 return res;
 839         hdr.ocsum = BE16_TO_HOST(hdr.ocsum);
 840
 841         if (csum <= hdr.ocsum)
 842                 t = hdr.ocsum - csum;
 843         else
 844                 t = hdr.ocsum - csum - 1;
 845
 846         DBG(2, "ocsum=%04x, csum=%04x, fix=%04x", hdr.ocsum, csum, t);
 847
 848         t = HOST_TO_BE16(t);
 849         res = write_out_data(outfile, (uint8_t *)&t, 2, NULL);
 850         if (res)
 851                 return res;
 852
 853
 854         res = write_out_header(outfile, &hdr);
 855
 856         return res;
 857 }
 858
 859
 860 struct board_info *
 861 find_board(char *name)
 862 {
 863         struct board_info *ret;
 864         struct board_info *board;
 865
 866         ret = NULL;
 867         for (board = boards; board->name != NULL; board++){
 868                 if (strcasecmp(name, board->name) == 0) {
 869                         ret = board;
 870                         break;
 871                 }
 872         };
 873
 874         return ret;
 875 }
 876
 877
 878 int
 879 parse_opt_board(char ch, char *arg)
 880 {
 881
 882         DBG(1,"parsing board option: -%c %s", ch, arg);
 883
 884         if (board != NULL) {
 885                 ERR("only one board option allowed");
 886                 return -1;
 887         }
 888
 889         if (required_arg(ch, arg))
 890                 return -1;
 891
 892         board = find_board(arg);
 893         if (board == NULL){
 894                 ERR("invalid/unknown board specified: %s", arg);
 895                 return -1;
 896         }
 897
 898         return 0;
 899 }
 900
 901
 902 int
 903 parse_opt_ofname(char ch, char *arg)
 904 {
 905
 906         if (ofname != NULL) {
 907                 ERR("only one output file allowed");
 908                 return -1;
 909         }
 910
 911         if (required_arg(ch, arg))
 912                 return -1;
 913
 914         ofname = arg;
 915
 916         return 0;
 917 }
 918
 919
 920 int
 921 parse_opt_block(char ch, char *arg)
 922 {
 923         char buf[MAX_ARG_LEN];
 924         char *argv[MAX_ARG_COUNT];
 925         int argc;
 926         char *p;
 927         struct fw_block *block;
 928         int i;
 929
 930         if ( num_blocks >= MAX_NUM_BLOCKS ) {
 931                 ERR("too many blocks specified");
 932                 return -1;
 933         }
 934
 935         block = &blocks[num_blocks++];
 936
 937         /* setup default field values */
 938         block->padc = 0xFF;
 939
 940         switch(ch) {
 941         case 'b':
 942                 if (bootext_block) {
 943                         ERR("only one boot extension block allowed");
 944                         break;
 945                 }
 946                 block->type = BLOCK_TYPE_BOOTEXT;
 947                 bootext_block = block;
 948                 break;
 949         case 'r':
 950                 block->type = BLOCK_TYPE_RAW;
 951                 break;
 952         }
 953
 954         argc = parse_arg(arg, buf, argv);
 955
 956         i = 0;
 957         p = argv[i++];
 958         if (is_empty_arg(p)) {
 959                 ERR("no file specified in %s", arg);
 960                 return -1;
 961         } else {
 962                 block->file_name = strdup(p);
 963                 if (block->file_name == NULL) {
 964                         ERR("not enough memory");
 965                         return -1;
 966                 }
 967         }
 968
 969         if (block->type == BLOCK_TYPE_BOOTEXT)
 970                 return 0;
 971
 972         p = argv[i++];
 973         if (!is_empty_arg(p)) {
 974                 if (str2u32(p, &block->align) != 0) {
 975                         ERR("invalid block align in %s", arg);
 976                         return -1;
 977                 }
 978         }
 979
 980         return 0;
 981 }
 982
 983
 984 int
 985 calc_block_offsets(int type, uint32_t *offset)
 986 {
 987         struct fw_block *block;
 988         uint32_t next_offs;
 989         uint32_t avail;
 990         int i, res;
 991
 992         DBG(1,"calculating block offsets, starting with %" PRIu32,
 993                 *offset);
 994
 995         res = 0;
 996         for (i = 0; i < num_blocks; i++) {
 997                 block = &blocks[i];
 998
 999                 if (block->type != type)
1000                         continue;
1001
1002                 next_offs = ALIGN(*offset, block->align);
1003                 avail = board->flash_size - next_offs;
1004                 if (block->file_size > avail) {
1005                         ERR("file %s is too big, offset = %u, size=%u,"
1006                                 " avail = %u, align = %u", block->file_name,
1007                                 (unsigned)next_offs,
1008                                 (unsigned)block->file_size,
1009                                 (unsigned)avail,
1010                                 (unsigned)block->align);
1011                         res = -1;
1012                         break;
1013                 }
1014
1015                 block->padlen = next_offs - *offset;
1016                 *offset += block->file_size;
1017         }
1018
1019         return res;
1020 }
1021
1022 int
1023 process_blocks(void)
1024 {
1025         struct fw_block *block;
1026         uint32_t offset;
1027         int i;
1028         int res;
1029
1030         /* collecting file stats */
1031         for (i = 0; i < num_blocks; i++) {
1032                 block = &blocks[i];
1033                 res = block_stat_file(block);
1034                 if (res)
1035                         return res;
1036         }
1037
1038         offset = board->romio_offs + bootext_block->file_size;
1039         res = calc_block_offsets(BLOCK_TYPE_RAW, &offset);
1040
1041         return res;
1042 }
1043
1044
1045 int
1046 main(int argc, char *argv[])
1047 {
1048         int optinvalid = 0;   /* flag for invalid option */
1049         int c;
1050         int res = EXIT_FAILURE;
1051
1052         FILE *outfile;
1053
1054         progname=basename(argv[0]);
1055
1056         opterr = 0;  /* could not print standard getopt error messages */
1057         while ( 1 ) {
1058                 optinvalid = 0;
1059
1060                 c = getopt(argc, argv, "b:B:ho:r:v");
1061                 if (c == -1)
1062                         break;
1063
1064                 switch (c) {
1065                 case 'b':
1066                 case 'r':
1067                         optinvalid = parse_opt_block(c,optarg);
1068                         break;
1069                 case 'B':
1070                         optinvalid = parse_opt_board(c,optarg);
1071                         break;
1072                 case 'o':
1073                         optinvalid = parse_opt_ofname(c,optarg);
1074                         break;
1075                 case 'v':
1076                         verblevel++;
1077                         break;
1078                 case 'h':
1079                         usage(EXIT_SUCCESS);
1080                         break;
1081                 default:
1082                         optinvalid = 1;
1083                         break;
1084                 }
1085                 if (optinvalid != 0 ) {
1086                         ERR("invalid option: -%c", optopt);
1087                         goto out;
1088                 }
1089         }
1090
1091         if (board == NULL) {
1092                 ERR("no board specified");
1093                 goto out;
1094         }
1095
1096         if (ofname == NULL) {
1097                 ERR("no output file specified");
1098                 goto out;
1099         }
1100
1101         if (optind < argc) {
1102                 ERR("invalid option: %s", argv[optind]);
1103                 goto out;
1104         }
1105
1106         if (process_blocks() != 0) {
1107                 goto out;
1108         }
1109
1110         outfile = fopen(ofname, "w");
1111         if (outfile == NULL) {
1112                 ERRS("could not open \"%s\" for writing", ofname);
1113                 goto out;
1114         }
1115
1116         if (write_out_image(outfile) != 0)
1117                 goto out_flush;
1118
1119         DBG(1,"Image file %s completed.", ofname);
1120
1121         res = EXIT_SUCCESS;
1122
1123 out_flush:
1124         fflush(outfile);
1125         fclose(outfile);
1126         if (res != EXIT_SUCCESS) {
1127                 unlink(ofname);
1128         }
1129 out:
1130         return res;
1131 }