/*
* $Id$
*
- * Copyright (C) 2007 Gabor Juhos <juhosg@freemail.hu>
+ * Copyright (C) 2007 Gabor Juhos <juhosg at 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>
*
# 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 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;
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;
+ uint32_t size;
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;
};
uint32_t webp_size;
uint32_t webp_size_max;
uint32_t code_size;
-
+
uint32_t addr_code;
uint32_t addr_webp;
};
.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)
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 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),
+ 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),
/* Planet products */
BOARD_ADM("XRT-401D", "Planet XRT-401D", 2, SIG_XRT401D),
BOARD_ADM("XRT-402D", "Planet XRT-402D", 2, SIG_XRT402D),
-
+
{.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)
);
for (board = boards; board->model != NULL; board++){
fprintf(stream,
-" %-12s: %s\n",
+" %-12s: %s\n",
board->model, board->name);
};
fprintf(stream,
-" -d disable check for available space\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"
-" -b <file>[:<len>[:<padc>]]\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"
-" -r <file>:[<addr>][:<
len>[:<padc>]]\n"
-" add runtime code to the image"
-" -w [<file>:[<addr>][:<
len>[:<padc>]]]\n"
-" add webpages to the image"
-" -x <file>[:<
len>[:<padc>]]\n"
+" -r <file>:[<addr>][:<
align>[:<padc>]]\n"
+" add runtime code to the image\n"
+" -w [<file>:[<addr>][:<
align>[:<padc>]]]\n"
+" add webpages to the image\n"
+" -x <file>[:<
align>[:<padc>]]\n"
" add extra data at the end of the image\n"
" -h show this screen\n"
"Parameters:\n"
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
{
char *err = NULL;
uint32_t t;
-
+
errno=0;
t = strtoul(arg, &err, 0);
if (errno || (err==arg) || ((err != NULL) && *err)) {
if (errno || (err==arg) || ((err != NULL) && *err) || (t >= 0x10000)) {
return -1;
}
-
+
*val = t & 0xFFFF;
return 0;
}
if (errno || (err==arg) || ((err != NULL) && *err) || (t >= 0x100)) {
return -1;
}
-
+
*val = t & 0xFF;
return 0;
}
return -1;
*sig = arg[0] | (arg[1] << 8) | (arg[2] << 16) | (arg[3] << 24);
-
+
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;
}
csum8_get(struct csum_state *css)
{
uint8_t t;
-
+
t = css->val;
return ~t + 1;
}
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);
len--;
p++;
}
-
+
for ( ; len > 1; len -= 2, p +=2 ) {
t = p[0] + (p[1] << 8);
css->val += LE16_TO_HOST(t);
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;
- case 2:
+ case CSUM_SIZE_16:
csum16_update(p,len,css);
break;
}
uint16_t ret;
switch (css->size) {
- case 1:
+ case CSUM_SIZE_8:
ret = csum8_get(css);
break;
- case 2:
+ case CSUM_SIZE_16:
ret = csum16_get(css);
break;
}
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) {
{
uint8_t buf[512];
size_t buflen = sizeof(buf);
+ int err;
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;
}
block_stat_file(struct csys_block *block)
{
struct stat st;
- int res;
+ int err;
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;
/* 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->align-block->size_hdr);
- 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;
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;
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;
}
}
-int
+int
block_writeout_data(FILE *outfile, struct csys_block *block)
{
int res;
/* 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;
}
-int
+int
block_writeout_csum(FILE *outfile, struct csys_block *block)
{
uint16_t csum;
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);
{
int res;
struct csum_state css;
-
+
res = 0;
if (block == 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)
res = 0;
for (i=0; i < num_blocks; i++) {
block = &blocks[i];
-
+
if (block->type != BLOCK_TYPE_XTRA)
continue;
-
+
res = block_writeout(outfile, block);
if (res)
break;
}
-
+
return res;
}
ret = NULL;
for (board = boards; board->model != NULL; board++){
- if (strcmp(model, board->model) == 0) {
+ if (strcasecmp(model, board->model) == 0) {
ret = board;
break;
}
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) {
- errmsg(0,"only one board option allowed");
- return -1;
+ ERR("only one board option allowed");
+ return ERR_FATAL;
}
if (required_arg(ch, arg))
- return -1;
+ return ERR_FATAL;
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;
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];
-
+
/* setup default field values */
block->need_file = 1;
block->padc = 0xFF;
-
+
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;
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;
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);
- block->size_csum = 1;
+ block->size_csum = CSUM_SIZE_8;
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);
- block->size_csum = 2;
+ block->size_csum = CSUM_SIZE_16;
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);
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){
- 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) {
- errmsg(0,"invalid start address in %s", arg);
- return -1;
+ ERR("invalid start address in %s", arg);
+ return ERR_FATAL;
}
block->addr_set = 1;
}
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)) {
- errmsg(0,"invalid paddig character in %s", arg);
- return -1;
+ ERR("invalid paddig character in %s", arg);
+ return ERR_FATAL;
}
num_blocks++;
process_blocks(void)
{
struct csys_block *block;
- uint32_t size_avail;
+ uint32_t offs = 0;
int i;
int res;
+ res = 0;
/* collecting stats */
for (i = 0; i < num_blocks; i++) {
block = &blocks[i];
return res;
}
- size_avail = board->flash_size;
-
/* 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) {
- 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) {
- 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);
+
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) {
- 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);
- 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];
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++) {
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);
- res = -1;
- break;
+ res = ERR_INVALID_IMAGE;
}
}
{
int optinvalid = 0; /* flag for invalid option */
int c;
- int res = EXIT_FAILURE;
+ int res = ERR_FATAL;
FILE *outfile;
-
+
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;
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;
}
if (optinvalid != 0 ){
- errmsg(0, "invalid option: -%c", optopt);
+ ERR("invalid option: -%c", optopt);
goto out;
}
}
+ if (board == NULL) {
+ ERR("no board specified");
+ goto out;
+ }
+
if (optind == argc) {
- errmsg(0, "no output file specified");
+ ERR("no output file specified");
goto out;
}
ofname = argv[optind++];
if (optind < argc) {
- errmsg(0, "invalid option: %s", argv[optind]);
+ ERR("invalid option: %s", argv[optind]);
goto out;
}
-
- if (board == NULL) {
- errmsg(0, "no board specified");
+ res = process_blocks();
+ if (res == ERR_FATAL)
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) {
- errmsg(1, "could not open \"%s\" for writing", ofname);
+ ERRS("could not open \"%s\" for writing", ofname);
+ res = ERR_FATAL;
goto out;
}
- if (write_out_blocks(outfile) != 0)
+ if (write_out_blocks(outfile) != 0) {
+ res = ERR_FATAL;
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);
- if (res != EXIT_SUCCESS) {
+ if (res == ERR_FATAL) {
unlink(ofname);
}
out:
- return res;
+ if (res == ERR_FATAL)
+ return EXIT_FAILURE;
+
+ return EXIT_SUCCESS;
}
projects / openwrt.git / blobdiff