From 75a5be3717d15c468beccab2ba618b01f44d56ec Mon Sep 17 00:00:00 2001 From: akr Date: Sat, 16 Feb 2013 11:44:42 +0000 Subject: * ext/socket/ancdata.c (rsock_recvmsg): ignore truncated part of socket address returned from recvmsg(). * ext/socket/init.c (recvfrom_blocking): ignore truncated part of socket address returned from recvfrom(). (rsock_s_recvfrom_nonblock): ditto. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@39278 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- ChangeLog | 9 +++++++++ ext/socket/ancdata.c | 8 +++++++- ext/socket/init.c | 13 +++++++++++-- 3 files changed, 27 insertions(+), 3 deletions(-) diff --git a/ChangeLog b/ChangeLog index 5deb9a27b8..066fa33778 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,12 @@ +Sat Feb 16 20:43:20 2013 Tanaka Akira + + * ext/socket/ancdata.c (rsock_recvmsg): ignore truncated part of + socket address returned from recvmsg(). + + * ext/socket/init.c (recvfrom_blocking): ignore truncated part of + socket address returned from recvfrom(). + (rsock_s_recvfrom_nonblock): ditto. + Sat Feb 16 20:05:26 2013 Ayumu AIZAWA * test/ruby/test_thread.rb: fixed typo diff --git a/ext/socket/ancdata.c b/ext/socket/ancdata.c index fa794eb182..34a6185b9c 100644 --- a/ext/socket/ancdata.c +++ b/ext/socket/ancdata.c @@ -1362,11 +1362,17 @@ struct recvmsg_args_struct { ssize_t rsock_recvmsg(int socket, struct msghdr *message, int flags) { + ssize_t ret; + socklen_t len0; #ifdef MSG_CMSG_CLOEXEC /* MSG_CMSG_CLOEXEC is available since Linux 2.6.23. Linux 2.6.18 silently ignore it. */ flags |= MSG_CMSG_CLOEXEC; #endif - return recvmsg(socket, message, flags); + len0 = message->msg_namelen; + ret = recvmsg(socket, message, flags); + if (ret != -1 && len0 < message->msg_namelen) + message->msg_namelen = len0; + return ret; } static void * diff --git a/ext/socket/init.c b/ext/socket/init.c index ab3fcf2b89..fc3ab49307 100644 --- a/ext/socket/init.c +++ b/ext/socket/init.c @@ -98,8 +98,13 @@ static VALUE recvfrom_blocking(void *data) { struct recvfrom_arg *arg = data; - return (VALUE)recvfrom(arg->fd, RSTRING_PTR(arg->str), RSTRING_LEN(arg->str), - arg->flags, (struct sockaddr*)&arg->buf, &arg->alen); + socklen_t len0 = arg->alen; + ssize_t ret; + ret = recvfrom(arg->fd, RSTRING_PTR(arg->str), RSTRING_LEN(arg->str), + arg->flags, (struct sockaddr*)&arg->buf, &arg->alen); + if (ret != -1 && len0 < arg->alen) + arg->alen = len0; + return (VALUE)ret; } VALUE @@ -182,6 +187,7 @@ rsock_s_recvfrom_nonblock(VALUE sock, int argc, VALUE *argv, enum sock_recv_type long slen; int fd, flags; VALUE addr = Qnil; + socklen_t len0; rb_scan_args(argc, argv, "11", &len, &flg); @@ -205,7 +211,10 @@ rsock_s_recvfrom_nonblock(VALUE sock, int argc, VALUE *argv, enum sock_recv_type rb_io_check_closed(fptr); rb_io_set_nonblock(fptr); + len0 = alen; slen = recvfrom(fd, RSTRING_PTR(str), buflen, flags, (struct sockaddr*)&buf, &alen); + if (slen != -1 && len0 < alen) + alen = len0; if (slen < 0) { switch (errno) { -- cgit v1.2.3