X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/8e6b4ddbe8c4811b443d5463d54f2bf20eeef896..84c11e85a4c4a12ecacba24afe9617222e4361e6:/io.c

diff --git a/io.c b/io.c
index b1748628..e7cb76e3 100644
--- a/io.c
+++ b/io.c
@@ -76,6 +76,9 @@ int kluge_around_eof = 0;
 int sock_f_in = -1;
 int sock_f_out = -1;
 
+int64 total_data_read = 0;
+int64 total_data_written = 0;
+
 static struct {
 	xbuf in, out, msg;
 	int in_fd;
@@ -1261,7 +1264,7 @@ void stop_flist_forward(void)
 static void read_a_msg(void)
 {
 	char *data, line[BIGPATHBUFLEN];
-	int tag;
+	int tag, val;
 	size_t msg_bytes;
 
 	data = perform_io(4, PIO_INPUT_AND_CONSUME);
@@ -1303,6 +1306,17 @@ static void read_a_msg(void)
 		if (!am_generator)
 			send_msg(MSG_IO_ERROR, data, 4, 0);
 		break;
+	case MSG_IO_TIMEOUT:
+		if (msg_bytes != 4 || am_server || am_generator)
+			goto invalid_msg;
+		data = perform_io(4, PIO_INPUT_AND_CONSUME);
+		val = IVAL(data, 0);
+		if (!io_timeout || io_timeout > val) {
+			if (INFO_GTE(MISC, 2))
+				rprintf(FINFO, "Setting --timeout=%d to match server\n", val);
+			io_timeout = val;
+		}
+		break;
 	case MSG_NOOP:
 		if (am_sender)
 			maybe_send_keepalive();
@@ -1578,6 +1592,7 @@ void read_buf(int f, char *buf, size_t len)
 
 	if (!IN_MULTIPLEXED) {
 		memcpy(buf, perform_io(len, PIO_INPUT_AND_CONSUME), len);
+		total_data_read += len;
 		if (forward_flist_data)
 			write_buf(iobuf.out_fd, buf, len);
 	  batch_copy:
@@ -1601,6 +1616,7 @@ void read_buf(int f, char *buf, size_t len)
 		/* The bytes at the "data" pointer will survive long
 		 * enough to make a copy, but not past future I/O. */
 		memcpy(buf, data, siz);
+		total_data_read += siz;
 
 		if (forward_flist_data)
 			write_buf(iobuf.out_fd, buf, siz);
@@ -1873,6 +1889,7 @@ void write_buf(int f, const char *buf, size_t len)
 		memcpy(iobuf.out.buf + pos, buf, len);
 
 	iobuf.out.len += len;
+	total_data_written += len;
 
   batch_copy:
 	if (f == write_batch_monitor_out)