hak5
/
openwrt-owl
mirror of https://github.com/hak5/openwrt-owl.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add -f option to skip trx check 

SVN-Revision: 1809
owl
Waldemar Brodkorb 2005-08-31 23:05:37 +00:00
parent 28326be623
commit e747ff8032
1 changed files with 28 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
openwrt/package/mtd/mtd.c
View File

@ -55,7 +55,7 @@ struct trx_header {
};
int
trx_check(const char *trxfile, const char *mtd)
trx_check(const char *trxfile, const char *mtd, int force)
{
struct mtd_info_user mtdInfo;
int trxfd, fd;
@ -65,33 +65,35 @@ trx_check(const char *trxfile, const char *mtd)
trxfd = open(trxfile,O_RDONLY);
if(trxfd < 0) {
fprintf(stderr, "Could not open trx image: %s\n", trxfile);
fprintf(stderr, "Could not open image: %s\n", trxfile);
exit(1);
}
if (fstat(trxfd,&trxstat) < 0) {
fprintf(stderr, "Could not get trx image file status: %s\n", trxfile);
fprintf(stderr, "Could not get 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 (force == 0) {
count = read(trxfd, &trx, sizeof(struct trx_header));
if (count < sizeof(struct trx_header)) {
fprintf(stderr, "Could not get 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);
}
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);
lseek(trxfd, 0, SEEK_SET);
}
/* check if image fits to mtd device */
@ -283,6 +285,7 @@ void usage(void)
" write <imagefile> write imagefile to device\n"
"Following options are available:\n"
" -r reboot after successful command\n"
" -f force write without trx checks\n"
" -e <device> erase <device> before executing the command\n\n"
"Example: To write linux.trx to mtd4 labeled as linux and reboot afterwards\n"
" mtd -r write linux.trx linux\n\n");
@ -291,7 +294,7 @@ void usage(void)
int main (int argc, char **argv)
{
int ch, i, boot, unlock, trxfd;
int ch, i, boot, unlock, trxfd, force;
char *erase[MAX_ARGS], *device;
enum {
CMD_ERASE,
@ -301,9 +304,13 @@ int main (int argc, char **argv)
erase[0] = NULL;
boot = 0;
force = 0;
while ((ch = getopt(argc, argv, "re:")) != -1)
while ((ch = getopt(argc, argv, "fre:")) != -1)
switch (ch) {
case 'f':
force = 1;
break;
case 'r':
boot = 1;
break;
@ -336,7 +343,7 @@ int main (int argc, char **argv)
cmd = CMD_WRITE;
device = argv[2];
/* check trx file before erasing or writing anything */
trxfd = trx_check(argv[1], device);
trxfd = trx_check(argv[1], device, force);
} else {
usage();
}