diff options
Diffstat (limited to 'package/system/mtd/src/trx.c')
| -rw-r--r-- | package/system/mtd/src/trx.c | 14 |
1 files changed, 5 insertions, 9 deletions
diff --git a/package/system/mtd/src/trx.c b/package/system/mtd/src/trx.c index 5763917827..00c4d6c0b5 100644 --- a/package/system/mtd/src/trx.c +++ b/package/system/mtd/src/trx.c @@ -44,8 +44,6 @@ struct trx_header { uint32_t offsets[3]; /* Offsets of partitions from start of header */ }; -#define SEAMA_MAGIC 0x17a4a35e - #if __BYTE_ORDER == __BIG_ENDIAN #define STORE32_LE(X) ((((X) & 0x000000FF) << 24) | (((X) & 0x0000FF00) << 8) | (((X) & 0x00FF0000) >> 8) | (((X) & 0xFF000000) >> 24)) #elif __BYTE_ORDER == __LITTLE_ENDIAN @@ -114,16 +112,14 @@ trx_check(int imagefd, const char *mtd, char *buf, int *len) if (strcmp(mtd, "firmware") != 0) return 1; - *len = read(imagefd, buf, 32); if (*len < 32) { - fprintf(stdout, "Could not get image header, file too small (%d bytes)\n", *len); - return 0; + *len += read(imagefd, buf + *len, 32 - *len); + if (*len < 32) { + fprintf(stdout, "Could not get image header, file too small (%d bytes)\n", *len); + return 0; + } } - /* Allow writing Seama files to firmware without an extra validation */ - if (trx->magic == SEAMA_MAGIC) - return 1; - if (trx->magic != TRX_MAGIC || trx->len < sizeof(struct trx_header)) { if (quiet < 2) { fprintf(stderr, "Bad trx header\n"); |