X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/741d654495cf4a35bd92ddf0df14e56681454ec6..7f459268d903c8647cfbfcc8ff0c9fda18733861:/io.c

diff --git a/io.c b/io.c
index bd1ce1ec..47b15707 100644
--- a/io.c
+++ b/io.c
@@ -510,7 +510,7 @@ static int read_timeout(int fd, char *buf, size_t len)
 
 			/* Don't write errors on a dead socket. */
 			if (fd == sock_f_in)
-				io_multiplexing_close();
+				close_multiplexing_out();
 			rsyserr(FERROR, errno, "read error");
 			exit_cleanup(RERR_STREAMIO);
 		}
@@ -895,10 +895,17 @@ static void writefd_unbuffered(int fd,char *buf,size_t len)
 
 			/* Don't try to write errors back across the stream. */
 			if (fd == sock_f_out)
-				io_multiplexing_close();
+				close_multiplexing_out();
 			rsyserr(FERROR, errno,
 				"writefd_unbuffered failed to write %ld bytes: phase \"%s\"",
 				(long)len, io_write_phase);
+			/* If the other side is sending us error messages, try
+			 * to grab any messages they sent before they died. */
+			while (fd == sock_f_out && io_multiplexing_in) {
+				io_timeout = 30;
+				readfd_unbuffered(sock_f_in, io_filesfrom_buf,
+						  sizeof io_filesfrom_buf);
+			}
 			exit_cleanup(RERR_STREAMIO);
 		}
 
@@ -1122,8 +1129,13 @@ int io_multiplex_write(enum msgcode code, char *buf, size_t len)
 	return 1;
 }
 
+void close_multiplexing_in(void)
+{
+	io_multiplexing_in = 0;
+}
+
 /** Stop output multiplexing. */
-void io_multiplexing_close(void)
+void close_multiplexing_out(void)
 {
 	io_multiplexing_out = 0;
 }