From d18c1b1a0ab57d77ba823cfedf40077d24dafbd7 Mon Sep 17 00:00:00 2001
From: Max Kellermann <max@duempel.org>
Date: Tue, 3 Aug 2010 15:55:25 +0200
Subject: fd_util: add function recvmsg_cloexec()

---
 src/fd_util.c | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

(limited to 'src/fd_util.c')

diff --git a/src/fd_util.c b/src/fd_util.c
index d8fe166d7..b9443b03d 100644
--- a/src/fd_util.c
+++ b/src/fd_util.c
@@ -246,6 +246,33 @@ accept_cloexec_nonblock(int fd, struct sockaddr *address,
 	return ret;
 }
 
+#ifndef WIN32
+
+ssize_t
+recvmsg_cloexec(int sockfd, struct msghdr *msg, int flags)
+{
+#ifdef MSG_CMSG_CLOEXEC
+	flags |= MSG_CMSG_CLOEXEC;
+#endif
+
+	ssize_t result = recvmsg(sockfd, msg, flags);
+	if (result >= 0) {
+		struct cmsghdr *cmsg = CMSG_FIRSTHDR(msg);
+		while (cmsg != NULL) {
+			if (cmsg->cmsg_type == SCM_RIGHTS) {
+				int fd = *(const int *)CMSG_DATA(cmsg);
+				fd_set_cloexec(fd, true);
+			}
+
+			cmsg = CMSG_NXTHDR(msg, cmsg);
+		}
+	}
+
+	return result;
+}
+
+#endif
+
 #ifdef HAVE_INOTIFY_INIT
 
 int
-- 
cgit v1.2.3