oysteikt/mpd
0
0
Fork 0
Code Issues Pull Requests Releases Activity

fd_util: add function recvmsg_cloexec()

Browse Source
This commit is contained in:
Max Kellermann 2010-08-03 15:55:25 +02:00
parent c980fc653d
commit d18c1b1a0a
2 changed files with 49 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
src/fd_util.c
View File

@ -246,6 +246,33 @@ accept_cloexec_nonblock(int fd, struct sockaddr *address,
return ret; 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 #ifdef HAVE_INOTIFY_INIT
int int

22
src/fd_util.h
View File

@ -39,6 +39,14 @@
#include <stdbool.h> #include <stdbool.h>
#include <stddef.h> #include <stddef.h>
#ifndef WIN32
#if !defined(_GNU_SOURCE) && (defined(HAVE_PIPE2) || defined(HAVE_ACCEPT4))
#define _GNU_SOURCE
#endif
#include <sys/types.h>
#endif
struct sockaddr; struct sockaddr;
/** /**
@ -91,6 +99,20 @@ int
accept_cloexec_nonblock(int fd, struct sockaddr *address, accept_cloexec_nonblock(int fd, struct sockaddr *address,
size_t *address_length_r); size_t *address_length_r);
#ifndef WIN32
struct msghdr;
/**
* Wrapper for recvmsg(), which sets the CLOEXEC flag (atomically if
* supported by the OS).
*/
ssize_t
recvmsg_cloexec(int sockfd, struct msghdr *msg, int flags);
#endif
/** /**
* Wrapper for inotify_init(), which sets the CLOEXEC flag (atomically * Wrapper for inotify_init(), which sets the CLOEXEC flag (atomically
* if supported by the OS). * if supported by the OS).