projects / openwrt.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | inline | side by side
firmware-utils/mktplinkfw: add support for TL-WR703N v1
[openwrt.git] / tools / firmware-utils / src / mkcsysimg.c
diff --git a/tools/firmware-utils/src/mkcsysimg.c b/tools/firmware-utils/src/mkcsysimg.c
index 993ac07..4f2352a 100644 (file)
--- a/tools/firmware-utils/src/mkcsysimg.c
+++ b/tools/firmware-utils/src/mkcsysimg.c
@@ -1,9 +1,8 @@
 /*
 /*
- *  $Id$
  *
  *
- *  Copyright (C) 2007 Gabor Juhos <juhosg@freemail.hu>
+ *  Copyright (C) 2007-2009 Gabor Juhos <juhosg@openwrt.org>
  *
  *
- *  This program was based on the code found in various Linux 
+ *  This program was based on the code found in various Linux
  *  source tarballs released by Edimax for it's devices.
  *  Original author: David Hsu <davidhsu@realtek.com.tw>
  *
  *  source tarballs released by Edimax for it's devices.
  *  Original author: David Hsu <davidhsu@realtek.com.tw>
  *
@@ -52,8 +51,6 @@
 #  define LE32_TO_HOST(x)      bswap_32(x)
 #endif
 
 #  define LE32_TO_HOST(x)      bswap_32(x)
 #endif
 
-#define ALIGN(x,y)     ((x)+((y)-1)) & ~((y)-1)
-
 #define MAX_NUM_BLOCKS 8
 #define MAX_ARG_COUNT  32
 #define MAX_ARG_LEN    1024
 #define MAX_NUM_BLOCKS 8
 #define MAX_ARG_COUNT  32
 #define MAX_ARG_LEN    1024
@@ -66,6 +63,12 @@
 #define BLOCK_TYPE_CODE        3
 #define BLOCK_TYPE_XTRA        4
 
 #define BLOCK_TYPE_CODE        3
 #define BLOCK_TYPE_XTRA        4
 
+#define DEFAULT_BLOCK_ALIGN    0x10000U
+
+#define CSUM_SIZE_NONE 0
+#define CSUM_SIZE_8    1
+#define CSUM_SIZE_16   2
+
 
 struct csum_state{
        int     size;
 
 struct csum_state{
        int     size;
@@ -82,17 +85,18 @@ struct csys_block {
        char            *file_name;     /* name of the file */
        uint32_t        file_size;      /* length of the file */
 
        char            *file_name;     /* name of the file */
        uint32_t        file_size;      /* length of the file */
 
-       uint32_t        size;
-       int             size_set;
+       unsigned char   sig[SIG_LEN];
+       uint32_t        addr;
+       int             addr_set;
+       uint32_t        align;
+       int             align_set;
        uint8_t         padc;
 
        uint8_t         padc;
 
+       uint32_t        size;
        uint32_t        size_hdr;
        uint32_t        size_csum;
        uint32_t        size_avail;
 
        uint32_t        size_hdr;
        uint32_t        size_csum;
        uint32_t        size_avail;
 
-       unsigned char   sig[SIG_LEN];
-       uint32_t        addr;
-       int             addr_set;
        struct csum_state *css;
 };
 
        struct csum_state *css;
 };
 
@@ -111,7 +115,7 @@ struct board_info {
        uint32_t webp_size;
        uint32_t webp_size_max;
        uint32_t code_size;
        uint32_t webp_size;
        uint32_t webp_size_max;
        uint32_t code_size;
-       
+
        uint32_t addr_code;
        uint32_t addr_webp;
 };
        uint32_t addr_code;
        uint32_t addr_webp;
 };
@@ -123,7 +127,7 @@ struct board_info {
        .webp_size = ws, .webp_size_max = 3*0x10000, \
        .addr_code = ac, .addr_webp = aw \
        }
        .webp_size = ws, .webp_size_max = 3*0x10000, \
        .addr_code = ac, .addr_webp = aw \
        }
-       
+
 #define BOARD_ADM(m,n,f, sigw) BOARD(m,n,f, ADM_BOOT_SIG, sigw, \
        ADM_BOOT_SIZE, ADM_CONF_SIZE, ADM_WEBP_SIZE, \
        ADM_CODE_ADDR, ADM_WEBP_ADDR)
 #define BOARD_ADM(m,n,f, sigw) BOARD(m,n,f, ADM_BOOT_SIG, sigw, \
        ADM_BOOT_SIZE, ADM_CONF_SIZE, ADM_WEBP_SIZE, \
        ADM_CODE_ADDR, ADM_WEBP_ADDR)
@@ -135,7 +139,8 @@ struct board_info {
 char *progname;
 char *ofname = NULL;
 int verblevel = 0;
 char *progname;
 char *ofname = NULL;
 int verblevel = 0;
-int size_check = 1;
+int invalid_causes_error = 1;
+int keep_invalid_images = 0;
 
 struct board_info *board = NULL;
 
 
 struct board_info *board = NULL;
 
@@ -147,11 +152,11 @@ struct csys_block *code_block = NULL;
 struct csys_block blocks[MAX_NUM_BLOCKS];
 int num_blocks = 0;
 
 struct csys_block blocks[MAX_NUM_BLOCKS];
 int num_blocks = 0;
 
-
 static struct board_info boards[] = {
        /* The original Edimax products */
        BOARD_ADM("BR-6104K", "Edimax BR-6104K", 2, SIG_BR6104K),
        BOARD_ADM("BR-6104KP", "Edimax BR-6104KP", 2, SIG_BR6104KP),
 static struct board_info boards[] = {
        /* The original Edimax products */
        BOARD_ADM("BR-6104K", "Edimax BR-6104K", 2, SIG_BR6104K),
        BOARD_ADM("BR-6104KP", "Edimax BR-6104KP", 2, SIG_BR6104KP),
+       BOARD_ADM("BR-6104Wg", "Edimax BR-6104Wg", 2, SIG_BR6104Wg),
        BOARD_ADM("BR-6114WG", "Edimax BR-6114WG", 2, SIG_BR6114WG),
        BOARD_ADM("BR-6524K", "Edimax BR-6524K", 2, SIG_BR6524K),
        BOARD_ADM("BR-6524KP", "Edimax BR-6524KP", 2, SIG_BR6524KP),
        BOARD_ADM("BR-6114WG", "Edimax BR-6114WG", 2, SIG_BR6114WG),
        BOARD_ADM("BR-6524K", "Edimax BR-6524K", 2, SIG_BR6524K),
        BOARD_ADM("BR-6524KP", "Edimax BR-6524KP", 2, SIG_BR6524KP),
@@ -172,49 +177,43 @@ static struct board_info boards[] = {
        /* Planet products */
        BOARD_ADM("XRT-401D", "Planet XRT-401D", 2, SIG_XRT401D),
        BOARD_ADM("XRT-402D", "Planet XRT-402D", 2, SIG_XRT402D),
        /* Planet products */
        BOARD_ADM("XRT-401D", "Planet XRT-401D", 2, SIG_XRT401D),
        BOARD_ADM("XRT-402D", "Planet XRT-402D", 2, SIG_XRT402D),
-       
+
+       /* Conceptronic products */
+       BOARD_ADM("C54BSR4", "Conceptronic C54BSR4", 2, SIG_C54BSR4),
+
+       /* OSBRiDGE products */
+       BOARD_ADM("5GXi", "OSBDRiDGE 5GXi", 2, SIG_5GXI),
+
        {.model = NULL}
 };
 
 /*
        {.model = NULL}
 };
 
 /*
- * Helper routines
+ * Message macros
  */
  */
-void
-errmsgv(int syserr, const char *fmt, va_list arg_ptr)
-{
-       int save = errno;
-
-       fflush(0);
-       fprintf(stderr, "%s: error, ", progname);
-       vfprintf(stderr, fmt, arg_ptr);
-       if (syserr != 0) {
-               fprintf(stderr, " (%s)", strerror(save));
-       }
-       fprintf(stderr, "\n");
-}
-
-void
-errmsg(int syserr, const char *fmt, ...)
-{
-       va_list arg_ptr;
-       va_start(arg_ptr, fmt);
-       errmsgv(syserr, fmt, arg_ptr);
-       va_end(arg_ptr);
-}
-
-void
-dbgmsg(int level, const char *fmt, ...)
-{
-       va_list arg_ptr;
-       if (verblevel >= level) {
-               fflush(0);
-               va_start(arg_ptr, fmt);
-               vfprintf(stderr, fmt, arg_ptr);
-               fprintf(stderr, "\n");
-               va_end(arg_ptr);
-       }
-}
-
+#define ERR(fmt, ...) do { \
+       fflush(0); \
+       fprintf(stderr, "[%s] *** error: " fmt "\n", progname, ## __VA_ARGS__ ); \
+} while (0)
+
+#define ERRS(fmt, ...) do { \
+       int save = errno; \
+       fflush(0); \
+       fprintf(stderr, "[%s] *** error: " fmt "\n", progname, ## __VA_ARGS__ \
+               , strerror(save)); \
+} while (0)
+
+#define WARN(fmt, ...) do { \
+       fprintf(stderr, "[%s] *** warning: " fmt "\n", progname, ## __VA_ARGS__ ); \
+} while (0)
+
+#define DBG(lev, fmt, ...) do { \
+       if (verblevel < lev) \
+               break;\
+       fprintf(stderr, "[%s] " fmt "\n", progname, ## __VA_ARGS__ ); \
+} while (0)
+
+#define ERR_FATAL              -1
+#define ERR_INVALID_IMAGE      -2
 
 void
 usage(int status)
 
 void
 usage(int status)
@@ -235,17 +234,17 @@ usage(int status)
                 board->model, board->name);
        };
        fprintf(stream,
                 board->model, board->name);
        };
        fprintf(stream,
-"  -d              disable check for available space\n"
-"                  add boot code to the image\n"
-"  -b <file>[:<len>[:<padc>]]\n"
+"  -d              don't throw error on invalid images\n"
+"  -k              keep invalid images\n"
+"  -b <file>[:<align>[:<padc>]]\n"
 "                  add boot code to the image\n"
 "                  add boot code to the image\n"
-"  -c <file>[:<len>[:<padc>]]\n"
+"  -c <file>[:<align>[:<padc>]]\n"
 "                  add configuration settings to the image\n"
 "                  add configuration settings to the image\n"
-"  -r <file>:[<addr>][:<len>[:<padc>]]\n"
+"  -r <file>:[<addr>][:<align>[:<padc>]]\n"
 "                  add runtime code to the image\n"
 "                  add runtime code to the image\n"
-"  -w [<file>:[<addr>][:<len>[:<padc>]]]\n"
+"  -w [<file>:[<addr>][:<align>[:<padc>]]]\n"
 "                  add webpages to the image\n"
 "                  add webpages to the image\n"
-"  -x <file>[:<len>[:<padc>]]\n"
+"  -x <file>[:<align>[:<padc>]]\n"
 "                  add extra data at the end of the image\n"
 "  -h              show this screen\n"
 "Parameters:\n"
 "                  add extra data at the end of the image\n"
 "  -h              show this screen\n"
 "Parameters:\n"
@@ -255,6 +254,19 @@ usage(int status)
        exit(status);
 }
 
        exit(status);
 }
 
+static inline uint32_t align(uint32_t base, uint32_t alignment)
+{
+       uint32_t ret;
+
+       if (alignment) {
+               ret = (base + alignment - 1);
+               ret &= ~(alignment-1);
+       } else {
+               ret = base;
+       }
+
+       return ret;
+}
 
 /*
  * argument parsing
 
 /*
  * argument parsing
@@ -264,7 +276,7 @@ str2u32(char *arg, uint32_t *val)
 {
        char *err = NULL;
        uint32_t t;
 {
        char *err = NULL;
        uint32_t t;
-       
+
        errno=0;
        t = strtoul(arg, &err, 0);
        if (errno || (err==arg) || ((err != NULL) && *err)) {
        errno=0;
        t = strtoul(arg, &err, 0);
        if (errno || (err==arg) || ((err != NULL) && *err)) {
@@ -287,7 +299,7 @@ str2u16(char *arg, uint16_t *val)
        if (errno || (err==arg) || ((err != NULL) && *err) || (t >= 0x10000)) {
                return -1;
        }
        if (errno || (err==arg) || ((err != NULL) && *err) || (t >= 0x10000)) {
                return -1;
        }
-       
+
        *val = t & 0xFFFF;
        return 0;
 }
        *val = t & 0xFFFF;
        return 0;
 }
@@ -303,7 +315,7 @@ str2u8(char *arg, uint8_t *val)
        if (errno || (err==arg) || ((err != NULL) && *err) || (t >= 0x100)) {
                return -1;
        }
        if (errno || (err==arg) || ((err != NULL) && *err) || (t >= 0x100)) {
                return -1;
        }
-       
+
        *val = t & 0xFF;
        return 0;
 }
        *val = t & 0xFF;
        return 0;
 }
@@ -315,7 +327,7 @@ str2sig(char *arg, uint32_t *sig)
                return -1;
 
        *sig = arg[0] | (arg[1] << 8) | (arg[2] << 16) | (arg[3] << 24);
                return -1;
 
        *sig = arg[0] | (arg[1] << 8) | (arg[2] << 16) | (arg[3] << 24);
-       
+
        return 0;
 }
 
        return 0;
 }
 
@@ -374,8 +386,8 @@ required_arg(char c, char *arg)
        if (arg == NULL || *arg != '-')
                return 0;
 
        if (arg == NULL || *arg != '-')
                return 0;
 
-       errmsg(0,"option -%c requires an argument\n", c);
-       return -1;
+       ERR("option -%c requires an argument\n", c);
+       return ERR_FATAL;
 }
 
 
 }
 
 
@@ -403,7 +415,7 @@ uint16_t
 csum8_get(struct csum_state *css)
 {
        uint8_t t;
 csum8_get(struct csum_state *css)
 {
        uint8_t t;
-       
+
        t = css->val;
        return ~t + 1;
 }
        t = css->val;
        return ~t + 1;
 }
@@ -413,7 +425,7 @@ void
 csum16_update(uint8_t *p, uint32_t len, struct csum_state *css)
 {
        uint16_t t;
 csum16_update(uint8_t *p, uint32_t len, struct csum_state *css)
 {
        uint16_t t;
-       
+
        if (css->odd) {
                t = css->tmp + (p[0]<<8);
                css->val += LE16_TO_HOST(t);
        if (css->odd) {
                t = css->tmp + (p[0]<<8);
                css->val += LE16_TO_HOST(t);
@@ -421,7 +433,7 @@ csum16_update(uint8_t *p, uint32_t len, struct csum_state *css)
                len--;
                p++;
        }
                len--;
                p++;
        }
-       
+
        for ( ; len > 1; len -= 2, p +=2 ) {
                t = p[0] + (p[1] << 8);
                css->val += LE16_TO_HOST(t);
        for ( ; len > 1; len -= 2, p +=2 ) {
                t = p[0] + (p[1] << 8);
                css->val += LE16_TO_HOST(t);
@@ -458,10 +470,10 @@ void
 csum_update(uint8_t *p, uint32_t len, struct csum_state *css)
 {
        switch (css->size) {
 csum_update(uint8_t *p, uint32_t len, struct csum_state *css)
 {
        switch (css->size) {
-       case 1:
+       case CSUM_SIZE_8:
                csum8_update(p,len,css);
                break;
                csum8_update(p,len,css);
                break;
-       case 2:
+       case CSUM_SIZE_16:
                csum16_update(p,len,css);
                break;
        }
                csum16_update(p,len,css);
                break;
        }
@@ -474,10 +486,10 @@ csum_get(struct csum_state *css)
        uint16_t ret;
 
        switch (css->size) {
        uint16_t ret;
 
        switch (css->size) {
-       case 1:
+       case CSUM_SIZE_8:
                ret = csum8_get(css);
                break;
                ret = csum8_get(css);
                break;
-       case 2:
+       case CSUM_SIZE_16:
                ret = csum16_get(css);
                break;
        }
                ret = csum16_get(css);
                break;
        }
@@ -497,8 +509,8 @@ write_out_data(FILE *outfile, uint8_t *data, size_t len,
 
        fwrite(data, len, 1, outfile);
        if (errno) {
 
        fwrite(data, len, 1, outfile);
        if (errno) {
-               errmsg(1,"unable to write output file");
-               return -1;
+               ERRS("unable to write output file");
+               return ERR_FATAL;
        }
 
        if (css) {
        }
 
        if (css) {
@@ -515,15 +527,17 @@ write_out_padding(FILE *outfile, size_t len, uint8_t padc,
 {
        uint8_t buf[512];
        size_t buflen = sizeof(buf);
 {
        uint8_t buf[512];
        size_t buflen = sizeof(buf);
+       int err;
 
        memset(buf, padc, buflen);
        while (len > 0) {
                if (len < buflen)
                        buflen = len;
 
 
        memset(buf, padc, buflen);
        while (len > 0) {
                if (len < buflen)
                        buflen = len;
 
-               if (write_out_data(outfile, buf, buflen, css))
-                       return -1;
-                       
+               err = write_out_data(outfile, buf, buflen, css);
+               if (err)
+                       return err;
+
                len -= buflen;
        }
 
                len -= buflen;
        }
 
@@ -535,15 +549,15 @@ int
 block_stat_file(struct csys_block *block)
 {
        struct stat st;
 block_stat_file(struct csys_block *block)
 {
        struct stat st;
-       int res;
+       int err;
 
        if (block->file_name == NULL)
                return 0;
 
 
        if (block->file_name == NULL)
                return 0;
 
-       res = stat(block->file_name, &st);
-       if (res){
-               errmsg(1, "stat failed on %s", block->file_name);
-               return res;
+       err = stat(block->file_name, &st);
+       if (err){
+               ERRS("stat failed on %s", block->file_name);
+               return ERR_FATAL;
        }
 
        block->file_size = st.st_size;
        }
 
        block->file_size = st.st_size;
@@ -563,9 +577,9 @@ block_writeout_hdr(FILE *outfile, struct csys_block *block)
        /* setup header fields */
        memcpy(hdr.sig, block->sig, 4);
        hdr.addr = HOST_TO_LE32(block->addr);
        /* setup header fields */
        memcpy(hdr.sig, block->sig, 4);
        hdr.addr = HOST_TO_LE32(block->addr);
-       hdr.size = HOST_TO_LE32(block->size-block->size_hdr);
+       hdr.size = HOST_TO_LE32(block->size - block->size_hdr - block->size_csum);
 
 
-       dbgmsg(1,"writing header for block");
+       DBG(1,"writing header for block");
        res = write_out_data(outfile, (uint8_t *)&hdr, sizeof(hdr),NULL);
        return res;
 
        res = write_out_data(outfile, (uint8_t *)&hdr, sizeof(hdr),NULL);
        return res;
 
@@ -590,8 +604,8 @@ block_writeout_file(FILE *outfile, struct csys_block *block)
        errno = 0;
        f = fopen(block->file_name,"r");
        if (errno) {
        errno = 0;
        f = fopen(block->file_name,"r");
        if (errno) {
-               errmsg(1,"unable to open file: %s", block->file_name);
-               return -1;
+               ERRS("unable to open file: %s", block->file_name);
+               return ERR_FATAL;
        }
 
        len = block->file_size;
        }
 
        len = block->file_size;
@@ -603,9 +617,8 @@ block_writeout_file(FILE *outfile, struct csys_block *block)
                errno = 0;
                fread(buf, buflen, 1, f);
                if (errno != 0) {
                errno = 0;
                fread(buf, buflen, 1, f);
                if (errno != 0) {
-                       errmsg(1,"unable to read from file: %s",
-                               block->file_name);
-                       res = -1;
+                       ERRS("unable to read from file: %s", block->file_name);
+                       res = ERR_FATAL;
                        break;
                }
 
                        break;
                }
 
@@ -621,7 +634,7 @@ block_writeout_file(FILE *outfile, struct csys_block *block)
 }
 
 
 }
 
 
-int 
+int
 block_writeout_data(FILE *outfile, struct csys_block *block)
 {
        int res;
 block_writeout_data(FILE *outfile, struct csys_block *block)
 {
        int res;
@@ -633,14 +646,14 @@ block_writeout_data(FILE *outfile, struct csys_block *block)
 
        /* write padding data if neccesary */
        padlen = block->size_avail - block->file_size;
 
        /* write padding data if neccesary */
        padlen = block->size_avail - block->file_size;
-       dbgmsg(1,"padding block, length=%d", padlen);
+       DBG(1,"padding block, length=%d", padlen);
        res = write_out_padding(outfile, padlen, block->padc, block->css);
 
        return res;
 }
 
 
        res = write_out_padding(outfile, padlen, block->padc, block->css);
 
        return res;
 }
 
 
-int 
+int
 block_writeout_csum(FILE *outfile, struct csys_block *block)
 {
        uint16_t csum;
 block_writeout_csum(FILE *outfile, struct csys_block *block)
 {
        uint16_t csum;
@@ -649,7 +662,7 @@ block_writeout_csum(FILE *outfile, struct csys_block *block)
        if (block->size_csum == 0)
                return 0;
 
        if (block->size_csum == 0)
                return 0;
 
-       dbgmsg(1,"writing checksum for block");
+       DBG(1,"writing checksum for block");
        csum = HOST_TO_LE16(csum_get(block->css));
        res = write_out_data(outfile, (uint8_t *)&csum, block->size_csum, NULL);
 
        csum = HOST_TO_LE16(csum_get(block->css));
        res = write_out_data(outfile, (uint8_t *)&csum, block->size_csum, NULL);
 
@@ -662,7 +675,7 @@ block_writeout(FILE *outfile, struct csys_block *block)
 {
        int res;
        struct csum_state css;
 {
        int res;
        struct csum_state css;
-       
+
        res = 0;
 
        if (block == NULL)
        res = 0;
 
        if (block == NULL)
@@ -670,7 +683,7 @@ block_writeout(FILE *outfile, struct csys_block *block)
 
        block->css = NULL;
 
 
        block->css = NULL;
 
-       dbgmsg(2, "writing block, file=%s, file_size=%d, space=%d",
+       DBG(2, "writing block, file=%s, file_size=%d, space=%d",
                block->file_name, block->file_size, block->size_avail);
        res = block_writeout_hdr(outfile, block);
        if (res)
                block->file_name, block->file_size, block->size_avail);
        res = block_writeout_hdr(outfile, block);
        if (res)
@@ -718,15 +731,15 @@ write_out_blocks(FILE *outfile)
        res = 0;
        for (i=0; i < num_blocks; i++) {
                block = &blocks[i];
        res = 0;
        for (i=0; i < num_blocks; i++) {
                block = &blocks[i];
-               
+
                if (block->type != BLOCK_TYPE_XTRA)
                        continue;
                if (block->type != BLOCK_TYPE_XTRA)
                        continue;
-                       
+
                res = block_writeout(outfile, block);
                if (res)
                        break;
        }
                res = block_writeout(outfile, block);
                if (res)
                        break;
        }
-       
+
        return res;
 }
 
        return res;
 }
 
@@ -739,7 +752,7 @@ find_board(char *model)
 
        ret = NULL;
        for (board = boards; board->model != NULL; board++){
 
        ret = NULL;
        for (board = boards; board->model != NULL; board++){
-               if (strcmp(model, board->model) == 0) {
+               if (strcasecmp(model, board->model) == 0) {
                        ret = board;
                        break;
                }
                        ret = board;
                        break;
                }
@@ -753,20 +766,20 @@ int
 parse_opt_board(char ch, char *arg)
 {
 
 parse_opt_board(char ch, char *arg)
 {
 
-       dbgmsg(1,"parsing board option: -%c %s", ch, arg);
+       DBG(1,"parsing board option: -%c %s", ch, arg);
 
        if (board != NULL) {
 
        if (board != NULL) {
-               errmsg(0,"only one board option allowed");
-               return -1;
+               ERR("only one board option allowed");
+               return ERR_FATAL;
        }
 
        if (required_arg(ch, arg))
        }
 
        if (required_arg(ch, arg))
-               return -1;
+               return ERR_FATAL;
 
        board = find_board(arg);
        if (board == NULL){
 
        board = find_board(arg);
        if (board == NULL){
-               errmsg(0,"invalid/unknown board specified: %s", arg);
-               return -1;
+               ERR("invalid/unknown board specified: %s", arg);
+               return ERR_FATAL;
        }
 
        return 0;
        }
 
        return 0;
@@ -784,20 +797,20 @@ parse_opt_block(char ch, char *arg)
        int i;
 
        if ( num_blocks > MAX_NUM_BLOCKS ) {
        int i;
 
        if ( num_blocks > MAX_NUM_BLOCKS ) {
-               errmsg(0,"too many blocks specified");
-               return -1;
+               ERR("too many blocks specified");
+               return ERR_FATAL;
        }
 
        block = &blocks[num_blocks];
        }
 
        block = &blocks[num_blocks];
-       
+
        /* setup default field values */
        block->need_file = 1;
        block->padc = 0xFF;
        /* setup default field values */
        block->need_file = 1;
        block->padc = 0xFF;
-       
+
        switch (ch) {
        case 'b':
                if (boot_block) {
        switch (ch) {
        case 'b':
                if (boot_block) {
-                       errmsg(0,"only one boot block allowed");
+                       WARN("only one boot block allowed");
                        break;
                }
                block->type = BLOCK_TYPE_BOOT;
                        break;
                }
                block->type = BLOCK_TYPE_BOOT;
@@ -805,7 +818,7 @@ parse_opt_block(char ch, char *arg)
                break;
        case 'c':
                if (conf_block) {
                break;
        case 'c':
                if (conf_block) {
-                       errmsg(0,"only one config block allowed");
+                       WARN("only one config block allowed");
                        break;
                }
                block->type = BLOCK_TYPE_CONF;
                        break;
                }
                block->type = BLOCK_TYPE_CONF;
@@ -813,31 +826,31 @@ parse_opt_block(char ch, char *arg)
                break;
        case 'w':
                if (webp_block) {
                break;
        case 'w':
                if (webp_block) {
-                       errmsg(0,"only one web block allowed");
+                       WARN("only one web block allowed");
                        break;
                }
                block->type = BLOCK_TYPE_WEBP;
                block->size_hdr = sizeof(struct csys_header);
                        break;
                }
                block->type = BLOCK_TYPE_WEBP;
                block->size_hdr = sizeof(struct csys_header);
-               block->size_csum = 1;
+               block->size_csum = CSUM_SIZE_8;
                block->need_file = 0;
                webp_block = block;
                break;
        case 'r':
                if (code_block) {
                block->need_file = 0;
                webp_block = block;
                break;
        case 'r':
                if (code_block) {
-                       errmsg(0,"only one runtime block allowed");
+                       WARN("only one runtime block allowed");
                        break;
                }
                block->type = BLOCK_TYPE_CODE;
                block->size_hdr = sizeof(struct csys_header);
                        break;
                }
                block->type = BLOCK_TYPE_CODE;
                block->size_hdr = sizeof(struct csys_header);
-               block->size_csum = 2;
+               block->size_csum = CSUM_SIZE_16;
                code_block = block;
                break;
        case 'x':
                block->type = BLOCK_TYPE_XTRA;
                break;
        default:
                code_block = block;
                break;
        case 'x':
                block->type = BLOCK_TYPE_XTRA;
                break;
        default:
-               errmsg(0,"unknown block type \"%c\"", ch);
-               return -1;
+               ERR("unknown block type \"%c\"", ch);
+               return ERR_FATAL;
        }
 
        argc = parse_arg(arg, buf, argv);
        }
 
        argc = parse_arg(arg, buf, argv);
@@ -847,20 +860,20 @@ parse_opt_block(char ch, char *arg)
        if (!is_empty_arg(p)) {
                block->file_name = strdup(p);
                if (block->file_name == NULL) {
        if (!is_empty_arg(p)) {
                block->file_name = strdup(p);
                if (block->file_name == NULL) {
-                       errmsg(0,"not enough memory");
-                       return -1;
+                       ERR("not enough memory");
+                       return ERR_FATAL;
                }
        } else if (block->need_file){
                }
        } else if (block->need_file){
-               errmsg(0,"no file specified in %s", arg);
-               return -1;
+               ERR("no file specified in %s", arg);
+               return ERR_FATAL;
        }
 
        if (block->size_hdr) {
                p = argv[i++];
                if (!is_empty_arg(p)) {
                        if (str2u32(p, &block->addr) != 0) {
        }
 
        if (block->size_hdr) {
                p = argv[i++];
                if (!is_empty_arg(p)) {
                        if (str2u32(p, &block->addr) != 0) {
-                               errmsg(0,"invalid start address in %s", arg);
-                               return -1;
+                               ERR("invalid start address in %s", arg);
+                               return ERR_FATAL;
                        }
                        block->addr_set = 1;
                }
                        }
                        block->addr_set = 1;
                }
@@ -868,17 +881,17 @@ parse_opt_block(char ch, char *arg)
 
        p = argv[i++];
        if (!is_empty_arg(p)) {
 
        p = argv[i++];
        if (!is_empty_arg(p)) {
-               if (str2u32(p, &block->size) != 0) {
-                       errmsg(0,"invalid block size in %s", arg);
-                       return -1;
+               if (str2u32(p, &block->align) != 0) {
+                       ERR("invalid alignment value in %s", arg);
+                       return ERR_FATAL;
                }
                }
-               block->size_set = 1;
+               block->align_set = 1;
        }
 
        p = argv[i++];
        if (!is_empty_arg(p) && (str2u8(p, &block->padc) != 0)) {
        }
 
        p = argv[i++];
        if (!is_empty_arg(p) && (str2u8(p, &block->padc) != 0)) {
-               errmsg(0,"invalid paddig character in %s", arg);
-               return -1;
+               ERR("invalid paddig character in %s", arg);
+               return ERR_FATAL;
        }
 
        num_blocks++;
        }
 
        num_blocks++;
@@ -891,10 +904,11 @@ int
 process_blocks(void)
 {
        struct csys_block *block;
 process_blocks(void)
 {
        struct csys_block *block;
-       uint32_t size_avail;
+       uint32_t offs = 0;
        int i;
        int res;
 
        int i;
        int res;
 
+       res = 0;
        /* collecting stats */
        for (i = 0; i < num_blocks; i++) {
                block = &blocks[i];
        /* collecting stats */
        for (i = 0; i < num_blocks; i++) {
                block = &blocks[i];
@@ -903,73 +917,82 @@ process_blocks(void)
                        return res;
        }
 
                        return res;
        }
 
-       size_avail = board->flash_size;
-
        /* bootloader */
        block = boot_block;
        if (block) {
        /* bootloader */
        block = boot_block;
        if (block) {
-               if (block->size_set) {
-                       board->boot_size= block->size;
-               } else {
-                       block->size = board->boot_size;
-               }
-               if (block->size > size_avail) {
-                       errmsg(0,"boot block is too big");
-                       return -1;
+               block->size = board->boot_size;
+               if (block->file_size > board->boot_size) {
+                       WARN("boot block is too big");
+                       res = ERR_INVALID_IMAGE;
                }
        }
                }
        }
-       size_avail -= board->boot_size;
+       offs += board->boot_size;
 
        /* configuration data */
        block = conf_block;
        if (block) {
 
        /* configuration data */
        block = conf_block;
        if (block) {
-               if (block->size_set) {
-                       board->conf_size = block->size;
-               } else {
-                       block->size = board->conf_size;
+               block->size = board->conf_size;
+               if (block->file_size > board->conf_size) {
+                       WARN("config block is too big");
+                       res = ERR_INVALID_IMAGE;
                }
                }
-               if (block->size > size_avail) {
-                       errmsg(0,"config block is too big");
-                       return -1;
-               }
-
        }
        }
-       size_avail -= board->conf_size;
+       offs += board->conf_size;
 
        /* webpages */
        block = webp_block;
        if (block) {
 
        /* webpages */
        block = webp_block;
        if (block) {
-               if (block->size_set == 0)
-                       block->size = board->webp_size;
-               board->webp_size = block->size;
-               if (block->size > board->webp_size_max) {
-                       errmsg(0,"webpages block is too big");
-                       return -1;
-               }
+
                memcpy(block->sig, board->sig_webp, 4);
                memcpy(block->sig, board->sig_webp, 4);
+
                if (block->addr_set == 0)
                        block->addr = board->addr_webp;
                if (block->addr_set == 0)
                        block->addr = board->addr_webp;
+
+               if (block->align_set == 0)
+                       block->align = DEFAULT_BLOCK_ALIGN;
+
+               block->size = align(offs + block->file_size + block->size_hdr +
+                               block->size_csum, block->align) - offs;
+
+               if (block->size > board->webp_size_max) {
+                       WARN("webpages block is too big");
+                       res = ERR_INVALID_IMAGE;
+               }
+
+               DBG(2,"webpages start at %08x, size=%08x", offs,
+                               block->size);
+
+               offs += block->size;
+               if (offs > board->flash_size) {
+                       WARN("webp block is too big");
+                       res = ERR_INVALID_IMAGE;
+               }
        }
        }
-       size_avail -= board->webp_size;
 
        /* runtime code */
        block = code_block;
        if (block) {
 
        /* runtime code */
        block = code_block;
        if (block) {
-               if (block->size_set == 0) {
-                       block->size =ALIGN(block->file_size+ block->size_hdr +
-                               block->size_csum, 0x10000);
-               }
-               board->code_size = block->size;
-               if (board->code_size > size_avail) {
-                       errmsg(0,"code block is too big");
-                       if (size_check)
-                               return -1;
-               }
                memcpy(code_block->sig, SIG_CSYS, 4);
                memcpy(code_block->sig, SIG_CSYS, 4);
-               if (code_block->addr_set == 0)
-                       code_block->addr = board->addr_code;
+
+               if (block->addr_set == 0)
+                       block->addr = board->addr_code;
+
+               if (block->align_set == 0)
+                       block->align = DEFAULT_BLOCK_ALIGN;
+
+               block->size = align(offs + block->file_size +
+                               block->size_hdr + block->size_csum,
+                               block->align) - offs;
+
+               DBG(2,"code block start at %08x, size=%08x", offs,
+                               block->size);
+
+               offs += block->size;
+               if (offs > board->flash_size) {
+                       WARN("code block is too big");
+                       res = ERR_INVALID_IMAGE;
+               }
        }
        }
-       size_avail -= board->code_size;
 
        for (i = 0; i < num_blocks; i++) {
                block = &blocks[i];
 
        for (i = 0; i < num_blocks; i++) {
                block = &blocks[i];
@@ -977,17 +1000,22 @@ process_blocks(void)
                if (block->type != BLOCK_TYPE_XTRA)
                        continue;
 
                if (block->type != BLOCK_TYPE_XTRA)
                        continue;
 
-               if (block->size_set == 0)
-                       block->size = ALIGN(block->file_size, 0x10000);
+               if (block->align_set == 0)
+                       block->align = DEFAULT_BLOCK_ALIGN;
 
 
-               if (block->size > size_avail) {
-                       errmsg(0,"no space for file %s",
-                               block->file_name);
-                       if (size_check)
-                               return -1;
-               }
+               block->size = align(offs + block->file_size,
+                               block->align) - offs;
 
 
-               size_avail -= block->size;
+               DBG(2,"file %s start at %08x, size=%08x, align=%08x",
+                       block->file_name, offs, block->size, block->align);
+
+               offs += block->size;
+               if (offs > board->flash_size) {
+                       WARN("file %s is too big, size=%d, avail=%d",
+                               block->file_name, block->file_size,
+                               board->flash_size - offs);
+                       res = ERR_INVALID_IMAGE;
+               }
        }
 
        for (i = 0; i < num_blocks; i++) {
        }
 
        for (i = 0; i < num_blocks; i++) {
@@ -997,11 +1025,10 @@ process_blocks(void)
                        block->size_csum;
 
                if (block->size_avail < block->file_size) {
                        block->size_csum;
 
                if (block->size_avail < block->file_size) {
-                       errmsg(0,"file %s is too big, size=%d, avail=%d",
-                               block->file_name, block->file_size, 
+                       WARN("file %s is too big, size=%d, avail=%d",
+                               block->file_name, block->file_size,
                                block->size_avail);
                                block->size_avail);
-                       res = -1;
-                       break;
+                       res = ERR_INVALID_IMAGE;
                }
        }
 
                }
        }
 
@@ -1014,17 +1041,17 @@ main(int argc, char *argv[])
 {
        int optinvalid = 0;   /* flag for invalid option */
        int c;
 {
        int optinvalid = 0;   /* flag for invalid option */
        int c;
-       int res = EXIT_FAILURE;
+       int res = ERR_FATAL;
 
        FILE *outfile;
 
        FILE *outfile;
-       
+
        progname=basename(argv[0]);
 
        opterr = 0;  /* could not print standard getopt error messages */
        while ( 1 ) {
                optinvalid = 0;
        progname=basename(argv[0]);
 
        opterr = 0;  /* could not print standard getopt error messages */
        while ( 1 ) {
                optinvalid = 0;
-               
-               c = getopt(argc, argv, "b:c:dB:hr:vw:x:");
+
+               c = getopt(argc, argv, "b:B:c:dhkr:vw:x:");
                if (c == -1)
                        break;
 
                if (c == -1)
                        break;
 
@@ -1044,7 +1071,10 @@ main(int argc, char *argv[])
                        optinvalid = parse_opt_block(c,optarg);
                        break;
                case 'd':
                        optinvalid = parse_opt_block(c,optarg);
                        break;
                case 'd':
-                       size_check = 0;
+                       invalid_causes_error = 0;
+                       break;
+               case 'k':
+                       keep_invalid_images = 1;
                        break;
                case 'B':
                        optinvalid = parse_opt_board(c,optarg);
                        break;
                case 'B':
                        optinvalid = parse_opt_board(c,optarg);
@@ -1060,52 +1090,67 @@ main(int argc, char *argv[])
                        break;
                }
                if (optinvalid != 0 ){
                        break;
                }
                if (optinvalid != 0 ){
-                       errmsg(0, "invalid option: -%c", optopt);
+                       ERR("invalid option: -%c", optopt);
                        goto out;
                }
        }
 
                        goto out;
                }
        }
 
+       if (board == NULL) {
+               ERR("no board specified");
+               goto out;
+       }
+
        if (optind == argc) {
        if (optind == argc) {
-               errmsg(0, "no output file specified");
+               ERR("no output file specified");
                goto out;
        }
 
        ofname = argv[optind++];
 
        if (optind < argc) {
                goto out;
        }
 
        ofname = argv[optind++];
 
        if (optind < argc) {
-               errmsg(0, "invalid option: %s", argv[optind]);
+               ERR("invalid option: %s", argv[optind]);
                goto out;
        }
                goto out;
        }
-       
 
 
-       if (board == NULL) {
-               errmsg(0, "no board specified");
+       res = process_blocks();
+       if (res == ERR_FATAL)
                goto out;
                goto out;
-       }
 
 
-       if (process_blocks() != 0) {
-               goto out;
+       if (res == ERR_INVALID_IMAGE) {
+               if (invalid_causes_error)
+                       res = ERR_FATAL;
+
+               if (keep_invalid_images == 0) {
+                       WARN("generation of invalid images disabled", ofname);
+                       goto out;
+               }
+
+               WARN("generating invalid image", ofname);
        }
 
        outfile = fopen(ofname, "w");
        if (outfile == NULL) {
        }
 
        outfile = fopen(ofname, "w");
        if (outfile == NULL) {
-               errmsg(1, "could not open \"%s\" for writing", ofname);
+               ERRS("could not open \"%s\" for writing", ofname);
+               res = ERR_FATAL;
                goto out;
        }
 
                goto out;
        }
 
-       if (write_out_blocks(outfile) != 0)
+       if (write_out_blocks(outfile) != 0) {
+               res = ERR_FATAL;
                goto out_flush;
                goto out_flush;
+       }
 
 
-       dbgmsg(1,"Image file %s completed.", ofname);
-
-       res = EXIT_SUCCESS;
+       DBG(1,"Image file %s completed.", ofname);
 
 out_flush:
        fflush(outfile);
        fclose(outfile);
 
 out_flush:
        fflush(outfile);
        fclose(outfile);
-       if (res != EXIT_SUCCESS) {
+       if (res == ERR_FATAL) {
                unlink(ofname);
        }
 out:
                unlink(ofname);
        }
 out:
-       return res;
+       if (res == ERR_FATAL)
+               return EXIT_FAILURE;
+
+       return EXIT_SUCCESS;
 }
 }
Personal OpenWRT clone
This page took 0.053942 seconds and 4 git commands to generate.