pogmommy/nmrpflash
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add -c switch to execute command before (or instead of) TFTP upload

Browse source
This commit is contained in:
Joseph C. Lehner 2016-02-12 12:22:44 +01:00
parent 936219b598
commit 32914803f3
2 changed files with 21 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
main.c
View file

@ -30,8 +30,9 @@ void usage(FILE *fp)
fprintf(fp,
"Usage: nmrpflash [OPTIONS...]\n"
"\n"
"Options (-a, -i and -f are mandatory):\n"
"Options (-a, -i and -f and/or -c are mandatory):\n"
" -a <ipaddr> IP address to assign to target device\n"
" -c <command> Command to run before (or instead of) TFTP upload\n"
" -f <firmware> Firmware file\n"
" -i <interface> Network interface directly connected to device\n"
" -m <mac> MAC address of target device (xx:xx:xx:xx:xx:xx)\n"
@ -91,12 +92,15 @@ int main(int argc, char **argv)
opterr = 0;
while ((c = getopt(argc, argv, "a:f:i:m:M:p:t:T:hLVvU")) != -1) {
while ((c = getopt(argc, argv, "a:c:f:i:m:M:p:t:T:hLVvU")) != -1) {
max = 0x7fffffff;
switch (c) {
case 'a':
args.ipaddr = optarg;
break;
case 'c':
args.tftpcmd = optarg;
break;
case 'f':
args.filename = optarg;
break;
@ -157,7 +161,7 @@ int main(int argc, char **argv)
}
}
if (!args.filename || !args.intf || !args.ipaddr) {
if ((!args.filename && !args.tftpcmd) || !args.intf || !args.ipaddr) {
usage(stderr);
return 1;
}

19
nmrp.c
View file

@ -417,14 +417,23 @@ int nmrp_do(struct nmrpd_args *args)
break;
}
if (!args->tftpcmd) {
err = 0;
if (args->tftpcmd) {
printf("Executing '%s' ... ", args->tftpcmd);
fflush(stdout);
err = system(args->tftpcmd);
if (!err) {
printf("OK\n");
} else {
printf("ERR\n");
}
}
if (!err && args->filename) {
printf("Uploading %s ... ", args->filename);
fflush(stdout);
err = tftp_put(args);
} else {
printf("Running %s ... ", args->tftpcmd);
fflush(stdout);
err = system(args->tftpcmd);
}
if (!err) {