diff options
author | akr <akr@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2009-02-18 13:47:53 +0000 |
---|---|---|
committer | akr <akr@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2009-02-18 13:47:53 +0000 |
commit | ca8b2801d2def392c15e637187defd90dc6ed19e (patch) | |
tree | be121e820eddb8096525be08ae5a32c03d6ec491 /ext | |
parent | 577689150718955df5b3d990dc4e88cab2e1c818 (diff) | |
download | ruby-ca8b2801d2def392c15e637187defd90dc6ed19e.tar.gz ruby-ca8b2801d2def392c15e637187defd90dc6ed19e.tar.xz ruby-ca8b2801d2def392c15e637187defd90dc6ed19e.zip |
* ext/socket/ancdata.c (discard_cmsg_resource): new function to close
file descriptors in control message.
(bsock_recvmsg_internal): call discard_cmsg_resource before retrying
recvmsg.
git-svn-id: http://svn.ruby-lang.org/repos/ruby/trunk@22420 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'ext')
-rw-r--r-- | ext/socket/ancdata.c | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/ext/socket/ancdata.c b/ext/socket/ancdata.c index 2ab4f2010..239d93445 100644 --- a/ext/socket/ancdata.c +++ b/ext/socket/ancdata.c @@ -1099,6 +1099,28 @@ rb_recvmsg(int fd, struct msghdr *msg, int flags) return rb_thread_blocking_region(nogvl_recvmsg_func, &args, RUBY_UBF_IO, 0); } +static void +discard_cmsg_resource(struct msghdr *mh) +{ +#if defined(HAVE_ST_MSG_CONTROL) + struct cmsghdr *cmh; + + if (mh->msg_controllen == 0) + return; + + for (cmh = CMSG_FIRSTHDR(mh); cmh != NULL; cmh = CMSG_NXTHDR(mh, cmh)) { + if (cmh->cmsg_level == SOL_SOCKET && cmh->cmsg_type == SCM_RIGHTS) { + int *fdp = (int *)CMSG_DATA(cmh); + int *end = (int *)((char *)cmh + cmh->cmsg_len); + while (fdp < end) { + close(*fdp); + fdp++; + } + } + } +#endif +} + static VALUE bsock_recvmsg_internal(int argc, VALUE *argv, VALUE sock, int nonblock) { @@ -1232,12 +1254,14 @@ bsock_recvmsg_internal(int argc, VALUE *argv, VALUE sock, int nonblock) } #endif if (grown) { + discard_cmsg_resource(&mh); goto retry; } else { grow_buffer = 0; if (flags != orig_flags) { flags = orig_flags; + discard_cmsg_resource(&mh); goto retry; } } |