From c02ce88762eeda44002b82f4176e9fdbc0dd7e3c Mon Sep 17 00:00:00 2001 From: akr Date: Sun, 8 Feb 2009 15:12:27 +0000 Subject: * ext/socket/rubysocket.h (sockopt_new): add family argument. * ext/socket/option.c (sockopt_initialize): add vfamily argument. (sockopt_new): add family argument and record it in the object. (sockopt_family): new method. (sockopt_s_int): add vfamily argument. (sockopt_inspect): show family. * ext/socket/basicsocket.c (bsock_getsockopt): check address family using getsockname. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@22135 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- ext/socket/basicsocket.c | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) (limited to 'ext/socket/basicsocket.c') diff --git a/ext/socket/basicsocket.c b/ext/socket/basicsocket.c index da8ab106a3..65c19d00ee 100644 --- a/ext/socket/basicsocket.c +++ b/ext/socket/basicsocket.c @@ -290,6 +290,8 @@ bsock_getsockopt(VALUE sock, VALUE lev, VALUE optname) socklen_t len; char *buf; rb_io_t *fptr; + struct sockaddr_storage ss; + socklen_t sslen = sizeof(ss); level = level_arg(lev); option = optname_arg(level, optname); @@ -297,10 +299,16 @@ bsock_getsockopt(VALUE sock, VALUE lev, VALUE optname) buf = ALLOCA_N(char,len); GetOpenFile(sock, fptr); + + if (getsockname(fptr->fd, (struct sockaddr*)&ss, &sslen) < 0) + rb_sys_fail("getsockname(2)"); + if (getsockopt(fptr->fd, level, option, buf, &len) < 0) rb_sys_fail_path(fptr->pathv); - return sockopt_new(level, option, rb_str_new(buf, len)); + if (sslen < (char*)&ss.ss_family + sizeof(ss.ss_family) - (char*)&ss) + ss.ss_family = AF_UNSPEC; + return sockopt_new(ss.ss_family, level, option, rb_str_new(buf, len)); #else rb_notimplement(); #endif -- cgit v1.2.3