From 7a6136f81be3141916355ae6cc1e74fcba8f0549 Mon Sep 17 00:00:00 2001
From: Timo Sirainen <tss@iki.fi>
Date: Sat, 16 Feb 2013 07:53:17 +0200
Subject: [PATCH] dsync: Don't log read() failed: EOF when remote intentionally
 fails early.

---
 src/doveadm/dsync/dsync-ibc-stream.c | 29 +++++++++++++++++++++++++---
 1 file changed, 26 insertions(+), 3 deletions(-)

diff --git a/src/doveadm/dsync/dsync-ibc-stream.c b/src/doveadm/dsync/dsync-ibc-stream.c
index ba5267f1cd..64b4e34baa 100644
--- a/src/doveadm/dsync/dsync-ibc-stream.c
+++ b/src/doveadm/dsync/dsync-ibc-stream.c
@@ -32,6 +32,7 @@
 
 enum item_type {
 	ITEM_NONE,
+	ITEM_DONE,
 
 	ITEM_HANDSHAKE,
 	ITEM_MAILBOX_STATE,
@@ -58,6 +59,10 @@ static const struct {
 	const char *optional_keys;
 } items[ITEM_END_OF_LIST+1] = {
 	{ NULL, '\0', NULL, NULL },
+	{ .name = "done",
+	  .chr = 'X',
+	  .optional_keys = ""
+	},
 	{ .name = "handshake",
 	  .chr = 'H',
 	  .optional_keys = "sync_ns_prefix sync_box sync_type debug sync_all_namespaces "
@@ -132,10 +137,12 @@ struct dsync_ibc_stream {
 	unsigned int version_received:1;
 	unsigned int handshake_received:1;
 	unsigned int has_pending_data:1;
+	unsigned int stopped:1;
 };
 
 static void dsync_ibc_stream_stop(struct dsync_ibc_stream *ibc)
 {
+	ibc->stopped = TRUE;
 	i_stream_close(ibc->input);
 	o_stream_close(ibc->output);
 	io_loop_stop(current_ioloop);
@@ -280,7 +287,7 @@ static void dsync_ibc_stream_init(struct dsync_ibc_stream *ibc)
 	o_stream_nsend_str(ibc->output, DSYNC_HANDSHAKE_VERSION);
 
 	/* initialize serializers and send their headers to remote */
-	for (i = 1; i < ITEM_END_OF_LIST; i++) T_BEGIN {
+	for (i = ITEM_DONE + 1; i < ITEM_END_OF_LIST; i++) T_BEGIN {
 		const char *keys;
 
 		keys = items[i].required_keys == NULL ? items[i].optional_keys :
@@ -308,6 +315,13 @@ static void dsync_ibc_stream_deinit(struct dsync_ibc *_ibc)
 		dsync_deserializer_decode_finish(&ibc->cur_decoder);
 	if (ibc->mail_output != NULL)
 		i_stream_unref(&ibc->mail_output);
+	else {
+		/* notify remote that we're closing. this is mainly to avoid
+		   "read() failed: EOF" errors on failing dsyncs */
+		o_stream_nsend_str(ibc->output,
+			t_strdup_printf("%c\n", items[ITEM_DONE].chr));
+		o_stream_nfinish(ibc->output);
+	}
 
 	timeout_remove(&ibc->to);
 	if (ibc->io != NULL)
@@ -330,9 +344,10 @@ static int dsync_ibc_stream_next_line(struct dsync_ibc_stream *ibc,
 		*line_r = line;
 		return 1;
 	}
-
 	/* try reading some */
 	if (i_stream_read(ibc->input) == -1) {
+		if (ibc->stopped)
+			return -1;
 		if (ibc->input->stream_errno != 0) {
 			errno = ibc->input->stream_errno;
 			i_error("read(%s) failed: %m", ibc->name);
@@ -387,7 +402,7 @@ dsync_ibc_check_missing_deserializers(struct dsync_ibc_stream *ibc)
 	unsigned int i;
 	int ret = 0;
 
-	for (i = 1; i < ITEM_END_OF_LIST; i++) {
+	for (i = ITEM_DONE + 1; i < ITEM_END_OF_LIST; i++) {
 		if (ibc->deserializers[i] == NULL &&
 		    (items[i].required_keys != NULL ||
 		     items[i].optional_keys != NULL)) {
@@ -473,6 +488,14 @@ dsync_ibc_stream_input_next(struct dsync_ibc_stream *ibc, enum item_type item,
 		/* end of this list */
 		return DSYNC_IBC_RECV_RET_FINISHED;
 	}
+	if (line[0] == items[ITEM_DONE].chr) {
+		/* remote cleanly closed the connection, possibly because of
+		   some failure (which it should have logged). we don't want to
+		   log any stream errors anyway after this. */
+		dsync_ibc_stream_stop(ibc);
+		return DSYNC_IBC_RECV_RET_TRYAGAIN;
+
+	}
 	for (i = 1; i < ITEM_END_OF_LIST; i++) {
 		if (*line == items[i].chr) {
 			line_item = i;
-- 
2.47.3