X-Git-Url: https://git.rohieb.name/openwrt.git/blobdiff_plain/9505c5deb3ed1a63bfab937dbf54cf3e12b0cde2..e12cd3d5ae4c09dbf49d20c29cff8f2183946c73:/tools/firmware-utils/src/trx.c?ds=inline diff --git a/tools/firmware-utils/src/trx.c b/tools/firmware-utils/src/trx.c index 4af17cf1c..8e95d98d7 100644 --- a/tools/firmware-utils/src/trx.c +++ b/tools/firmware-utils/src/trx.c @@ -53,8 +53,10 @@ #if __BYTE_ORDER == __BIG_ENDIAN #define STORE32_LE(X) bswap_32(X) +#define LOAD32_LE(X) bswap_32(X) #elif __BYTE_ORDER == __LITTLE_ENDIAN #define STORE32_LE(X) (X) +#define LOAD32_LE(X) (X) #else #error unkown endianness! #endif @@ -262,23 +264,24 @@ int main(int argc, char **argv) /* for TRXv2 set bin-header Flags to 0xFF for CRC calculation like CFE does */ if (trx_version == 2) { - if(cur_len - p->offsets[3] < sizeof(binheader)) { + if(cur_len - LOAD32_LE(p->offsets[3]) < sizeof(binheader)) { fprintf(stderr, "TRXv2 binheader too small!\n"); return EXIT_FAILURE; } - memcpy(binheader, buf + p->offsets[3], sizeof(binheader)); /* save header */ - memset(buf + p->offsets[3] + 22, 0xFF, 8); /* set stable and try1-3 to 0xFF */ + memcpy(binheader, buf + LOAD32_LE(p->offsets[3]), sizeof(binheader)); /* save header */ + memset(buf + LOAD32_LE(p->offsets[3]) + 22, 0xFF, 8); /* set stable and try1-3 to 0xFF */ } p->crc32 = crc32buf((char *) &p->flag_version, (fsmark)?fsmark:cur_len - offsetof(struct trx_header, flag_version)); p->crc32 = STORE32_LE(p->crc32); - p->len = (fsmark)?fsmark:cur_len - offsetof(struct trx_header, flag_version); + p->len = STORE32_LE((fsmark) ? fsmark : cur_len); + p->len = STORE32_LE(p->len); /* restore TRXv2 bin-header */ if (trx_version == 2) { - memcpy(buf + p->offsets[3], binheader, sizeof(binheader)); + memcpy(buf + LOAD32_LE(p->offsets[3]), binheader, sizeof(binheader)); } if (!fwrite(buf, cur_len, 1, out) || fflush(out)) {