Patch from Ben Low <ben@titr.uow.edu.au> to allow tftp to work

with stdin as well as stdout.

1 files changed, 7 insertions, 11 deletions

diff --git a/networking/tftp.c b/networking/tftp.c
index a82415c60..ec30725d2 100644
--- a/networking/tftp.c
+++ b/networking/tftp.c
@@ -548,17 +548,13 @@ int tftp_main(int argc, char **argv)
 	if ((cmd == 0) || (optind == argc)) {
 		show_usage();
 	}
-	if(cmd == tftp_cmd_get) {
-	    if(localfile == NULL)
-		localfile = remotefile;
-	    if(localfile && strcmp(localfile, "-") == 0) {
-		fd = fileno(stdout);
-	    }
+	if(localfile && strcmp(localfile, "-") == 0) {
+	    fd = fileno((cmd==tftp_cmd_get)? stdin : stdout);
 	}
-	if(cmd == tftp_cmd_put)
-	    if(remotefile == NULL)
-		remotefile = localfile;
-
+	if(localfile == NULL)
+	    localfile = remotefile;
+	if(remotefile == NULL)
+	    remotefile = localfile;
 	if (fd==-1) {
 	    fd = open(localfile, flags, 0644);
 	}
@@ -582,7 +578,7 @@ int tftp_main(int argc, char **argv)
 	result = tftp(cmd, host, remotefile, fd, port, blocksize);
 
 #ifdef CONFIG_FEATURE_CLEAN_UP
-	if (fd!=fileno(stdout)) {
+	if (!(fd == fileno(stdout) || fd == fileno(stdin))) {
 	    close(fd);
 	}
 #endif