[Minor] Add shutdown calls to specify one direction for the pipe

rspamd · Jun 10, 2020 · 0d218b7 · 0d218b7
1 parent 980d8b7
commit 0d218b7
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/src/fuzzy_storage.c b/src/fuzzy_storage.c
@@ -2165,11 +2165,17 @@ fuzzy_peer_rep (struct rspamd_worker *worker,
 		cur = g_list_next (cur);
 	}
 
-	if (worker->index == 0 && ctx->peer_fd != -1) {
-		/* Listen for peer requests */
-		ctx->peer_ev.data = ctx;
-		ev_io_init (&ctx->peer_ev, rspamd_fuzzy_peer_io, ctx->peer_fd, EV_READ);
-		ev_io_start (ctx->event_loop, &ctx->peer_ev);
+	if (ctx->peer_fd != -1) {
+		if (worker->index == 0) {
+			/* Listen for peer requests */
+			shutdown (ctx->peer_fd, SHUT_WR);
+			ctx->peer_ev.data = ctx;
+			ev_io_init (&ctx->peer_ev, rspamd_fuzzy_peer_io, ctx->peer_fd, EV_READ);
+			ev_io_start (ctx->event_loop, &ctx->peer_ev);
+		}
+		else {
+			shutdown (ctx->peer_fd, SHUT_RD);
+		}
 	}
 }