From 7b08b7b712c51a99e1b76483c4c202488d987da4 Mon Sep 17 00:00:00 2001 From: Runxi Yu Date: Tue, 18 Feb 2025 08:33:32 +0800 Subject: hooks: Fix the race condition that causes EPIPE The hooks handler in the main daemon didn't wait for the hook client to write fully, and sometimes prematurely closes the connection, causing the hook client's splice to return EPIPE (or SIGPIPE if the signal handler wasn't installed). To remedy this, we call shutdown(sock, SHUT_WR) in the client, so that attempts to read on the server side return EOF. Then we can simply use io.Copy(&buf, conn) on the server side to fetch all of the data into a buffer. --- git_hooks_client/git_hooks_client.c | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'git_hooks_client') diff --git a/git_hooks_client/git_hooks_client.c b/git_hooks_client/git_hooks_client.c index e00ab3a..6471aa8 100644 --- a/git_hooks_client/git_hooks_client.c +++ b/git_hooks_client/git_hooks_client.c @@ -162,6 +162,16 @@ int main(int argc, char *argv[]) { return EXIT_FAILURE; } + /* + * The sending part of the UNIX socket should be shut down, to let + * io.Copy on the Go side return. + */ + if (shutdown(sock, SHUT_WR) == -1) { + perror("shutdown internal socket"); + close(sock); + return EXIT_FAILURE; + } + /* * The first byte of the response from the UNIX domain socket is the * status code. We read it and record it as our return value. -- cgit v1.2.3