1
0
mirror of git://projects.qi-hardware.com/openwrt-xburst.git synced 2024-10-06 01:45:26 +03:00

check if firmware is in trx format, before writing it to the flash

git-svn-id: svn://svn.openwrt.org/openwrt/trunk/openwrt@1431 3c298f89-4303-0410-b956-a3cf2f4a3e73
This commit is contained in:
wbx 2005-07-14 01:25:08 +00:00
parent de97538baa
commit 08c5754785

View File

@ -42,25 +42,18 @@
#include <linux/mtd/mtd.h> #include <linux/mtd/mtd.h>
/* trx header */
#define TRX_MAGIC 0x30524448 /* "HDR0" */ #define TRX_MAGIC 0x30524448 /* "HDR0" */
#define TRX_VERSION 1 #define BUFSIZE (10 * 1024)
#define TRX_MAX_LEN 0x3A0000
#define TRX_NO_HEADER 1 /* Do not write TRX header */
#define MAX_ARGS 8 #define MAX_ARGS 8
struct trx_header { struct trx_header {
uint32_t magic; /* "HDR0" */ uint32_t magic; /* "HDR0" */
uint32_t len; /* Length of file including header */ uint32_t len; /* Length of file including header */
uint32_t crc32; /* 32-bit CRC from flag_version to end of file */ uint32_t crc32; /* 32-bit CRC from flag_version to end of file */
uint32_t flag_version; /* 0:15 flags, 16:31 version */ uint32_t flag_version; /* 0:15 flags, 16:31 version */
uint32_t offsets[3]; /* Offsets of partitions from start of header */ uint32_t offsets[3]; /* Offsets of partitions from start of header */
}; };
#define BUFSIZE (10 * 1024)
int int
mtd_unlock(const char *mtd) mtd_unlock(const char *mtd)
{ {
@ -84,7 +77,6 @@ mtd_unlock(const char *mtd)
mtdLockInfo.start = 0; mtdLockInfo.start = 0;
mtdLockInfo.length = mtdInfo.size; mtdLockInfo.length = mtdInfo.size;
if(ioctl(fd, MEMUNLOCK, &mtdLockInfo)) { if(ioctl(fd, MEMUNLOCK, &mtdLockInfo)) {
fprintf(stderr, "Could not unlock MTD device: %s\n", mtd);
close(fd); close(fd);
return 0; return 0;
} }
@ -156,15 +148,44 @@ mtd_erase(const char *mtd)
int int
mtd_write(const char *trxfile, const char *mtd) mtd_write(const char *trxfile, const char *mtd)
{ {
int fd; int fd,trxfd,i;
int trxfd; struct trx_header trx;
int i; size_t count,result,size,written;
size_t result,size,written;
struct mtd_info_user mtdInfo; struct mtd_info_user mtdInfo;
struct erase_info_user mtdEraseInfo; struct erase_info_user mtdEraseInfo;
struct stat trxstat; struct stat trxstat;
unsigned char src[BUFSIZE],dest[BUFSIZE]; unsigned char src[BUFSIZE],dest[BUFSIZE];
trxfd = open(trxfile,O_RDONLY);
if(trxfd < 0) {
fprintf(stderr, "Could not open trx image: %s\n", trxfile);
exit(1);
}
if (fstat(trxfd,&trxstat) < 0) {
fprintf(stderr, "Could not get trx image file status: %s\n", trxfile);
close(trxfd);
exit(1);
}
count = read(trxfd, &trx, sizeof(struct trx_header));
if (count < sizeof(struct trx_header)) {
fprintf(stderr, "Could not trx header, file too small (%ld bytes)\n", count);
close(trxfd);
exit(1);
}
if (trx.magic != TRX_MAGIC || trx.len < sizeof(struct trx_header)) {
fprintf(stderr, "Bad trx header\n");
fprintf(stderr, "If this is a firmware in bin format, like some of the\n"
"original firmware files are, use following command to convert to trx:\n"
"dd if=firmware.bin of=firmware.trx bs=32 skip=1\n");
close(trxfd);
exit(1);
}
lseek(trxfd, 0, SEEK_SET);
fd = mtd_open(mtd, O_RDWR); fd = mtd_open(mtd, O_RDWR);
if(fd < 0) { if(fd < 0) {
fprintf(stderr, "Could not open mtd device: %s\n", mtd); fprintf(stderr, "Could not open mtd device: %s\n", mtd);
@ -176,18 +197,6 @@ mtd_write(const char *trxfile, const char *mtd)
close(fd); close(fd);
exit(1); exit(1);
} }
trxfd = open(trxfile,O_RDONLY);
if(trxfd < 0) {
fprintf(stderr, "Could not open trx image: %s\n", trxfile);
exit(1);
}
if (fstat (trxfd,&trxstat) < 0) {
fprintf(stderr, "Could not get trx image file status: %s\n", trxfile);
close(trxfd);
exit(1);
}
if(mtdInfo.size < trxstat.st_size) { if(mtdInfo.size < trxstat.st_size) {
fprintf(stderr, "Image too big for partition: %s\n", mtd); fprintf(stderr, "Image too big for partition: %s\n", mtd);