From d933fb2296e4ace25c5eb558be56cd23a8fcc88f Mon Sep 17 00:00:00 2001 From: akr Date: Sat, 6 Apr 2013 02:39:44 +0000 Subject: * ext/socket: Improve socket exception message to show socket address. [ruby-core:45617] [Feature #6583] proposed Eric Hodel. * ext/socket/rubysocket.h (rsock_sys_fail_host_port): Declared. (rsock_sys_fail_path): Ditto. (rsock_sys_fail_sockaddr): Ditto. * ext/socket/udpsocket.c (udp_connect): Use rsock_sys_fail_host_port. (udp_bind): Ditto. (udp_send): Ditto. * ext/socket/init.c (rsock_init_sock): Specify a string for rb_sys_fail argument. (make_fd_nonblock): Ditto. (rsock_s_accept): Ditto. * ext/socket/ipsocket.c (init_inetsock_internal): Use rsock_sys_fail_host_port. * ext/socket/socket.c (rsock_sys_fail_host_port): Defined. (rsock_sys_fail_path): Ditto. (rsock_sys_fail_sockaddr): Ditto. (setup_domain_and_type): Use rsock_sys_fail_sockaddr. (sock_connect_nonblock): Ditto. (sock_bind): Ditto. (sock_gethostname): Specify a string for rb_sys_fail argument. (socket_s_ip_address_list): Ditto. * ext/socket/basicsocket.c (bsock_shutdown): Specify a string for rb_sys_fail argument. (bsock_setsockopt): Use rsock_sys_fail_path. (bsock_getsockopt): Ditto. (bsock_getpeereid): Refine the argument for rb_sys_fail. * ext/socket/unixsocket.c (rsock_init_unixsock): Use rsock_sys_fail_path. (unix_path): Ditto. (unix_send_io): Ditto. (unix_recv_io): Ditto. (unix_addr): Ditto. (unix_peeraddr): Ditto. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@40149 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- ext/socket/basicsocket.c | 10 +++++----- ext/socket/init.c | 8 ++++---- ext/socket/ipsocket.c | 15 +++++++++++++-- ext/socket/rubysocket.h | 4 ++++ ext/socket/socket.c | 46 ++++++++++++++++++++++++++++++++++++++++------ ext/socket/udpsocket.c | 8 +++++--- ext/socket/unixsocket.c | 16 ++++++++-------- 7 files changed, 79 insertions(+), 28 deletions(-) (limited to 'ext/socket') diff --git a/ext/socket/basicsocket.c b/ext/socket/basicsocket.c index 40867434fc..94c34daf32 100644 --- a/ext/socket/basicsocket.c +++ b/ext/socket/basicsocket.c @@ -80,7 +80,7 @@ bsock_shutdown(int argc, VALUE *argv, VALUE sock) } GetOpenFile(sock, fptr); if (shutdown(fptr->fd, how) == -1) - rb_sys_fail(0); + rb_sys_fail("shutdown(2)"); return INT2FIX(0); } @@ -249,7 +249,7 @@ bsock_setsockopt(int argc, VALUE *argv, VALUE sock) rb_io_check_closed(fptr); if (setsockopt(fptr->fd, level, option, v, vlen) < 0) - rb_sys_fail_path(fptr->pathv); + rsock_sys_fail_path("setsockopt(2)", fptr->pathv); return INT2FIX(0); } @@ -330,7 +330,7 @@ bsock_getsockopt(VALUE sock, VALUE lev, VALUE optname) rb_io_check_closed(fptr); if (getsockopt(fptr->fd, level, option, buf, &len) < 0) - rb_sys_fail_path(fptr->pathv); + rsock_sys_fail_path("getsockopt(2)", fptr->pathv); return rsock_sockopt_new(family, level, option, rb_str_new(buf, len)); } @@ -429,7 +429,7 @@ bsock_getpeereid(VALUE self) gid_t egid; GetOpenFile(self, fptr); if (getpeereid(fptr->fd, &euid, &egid) == -1) - rb_sys_fail("getpeereid"); + rb_sys_fail("getpeereid(3)"); return rb_assoc_new(UIDT2NUM(euid), GIDT2NUM(egid)); #elif defined(SO_PEERCRED) /* GNU/Linux */ rb_io_t *fptr; @@ -445,7 +445,7 @@ bsock_getpeereid(VALUE self) VALUE ret; GetOpenFile(self, fptr); if (getpeerucred(fptr->fd, &uc) == -1) - rb_sys_fail("getpeerucred"); + rb_sys_fail("getpeerucred(3C)"); ret = rb_assoc_new(UIDT2NUM(ucred_geteuid(uc)), GIDT2NUM(ucred_getegid(uc))); ucred_free(uc); return ret; diff --git a/ext/socket/init.c b/ext/socket/init.c index 33ddc05fa4..3fda607bee 100644 --- a/ext/socket/init.c +++ b/ext/socket/init.c @@ -47,7 +47,7 @@ rsock_init_sock(VALUE sock, int fd) struct stat sbuf; if (fstat(fd, &sbuf) < 0) - rb_sys_fail(0); + rb_sys_fail("fstat(2)"); rb_update_max_fd(fd); if (!S_ISSOCK(sbuf.st_mode)) rb_raise(rb_eArgError, "not a socket file descriptor"); @@ -475,14 +475,14 @@ make_fd_nonblock(int fd) #ifdef F_GETFL flags = fcntl(fd, F_GETFL); if (flags == -1) { - rb_sys_fail(0); + rb_sys_fail("fnctl(2)"); } #else flags = 0; #endif flags |= O_NONBLOCK; if (fcntl(fd, F_SETFL, flags) == -1) { - rb_sys_fail(0); + rb_sys_fail("fnctl(2)"); } } @@ -590,7 +590,7 @@ rsock_s_accept(VALUE klass, int fd, struct sockaddr *sockaddr, socklen_t *len) retry = 0; goto retry; } - rb_sys_fail(0); + rb_sys_fail("accept(2)"); } rb_update_max_fd(fd2); if (!klass) return INT2NUM(fd2); diff --git a/ext/socket/ipsocket.c b/ext/socket/ipsocket.c index 84bc897055..c3b989730d 100644 --- a/ext/socket/ipsocket.c +++ b/ext/socket/ipsocket.c @@ -43,7 +43,7 @@ init_inetsock_internal(struct inetsock_arg *arg) { int type = arg->type; struct addrinfo *res; - int fd, status = 0; + int fd, status = 0, local = 0; const char *syscall = 0; arg->remote.res = rsock_addrinfo(arg->remote.host, arg->remote.serv, SOCK_STREAM, @@ -81,6 +81,7 @@ init_inetsock_internal(struct inetsock_arg *arg) else { if (arg->local.res) { status = bind(fd, arg->local.res->ai_addr, arg->local.res->ai_addrlen); + local = status; syscall = "bind(2)"; } @@ -99,7 +100,17 @@ init_inetsock_internal(struct inetsock_arg *arg) break; } if (status < 0) { - rb_sys_fail(syscall); + VALUE host, port; + + if (local < 0) { + host = arg->local.host; + port = arg->local.serv; + } else { + host = arg->remote.host; + port = arg->remote.serv; + } + + rsock_sys_fail_host_port(syscall, host, port); } arg->fd = -1; diff --git a/ext/socket/rubysocket.h b/ext/socket/rubysocket.h index 10aa630602..bc6e3b07e2 100644 --- a/ext/socket/rubysocket.h +++ b/ext/socket/rubysocket.h @@ -333,4 +333,8 @@ void rsock_init_addrinfo(void); void rsock_init_sockopt(void); void rsock_init_socket_init(void); +void rsock_sys_fail_host_port(const char *, VALUE, VALUE); +void rsock_sys_fail_path(const char *, VALUE); +void rsock_sys_fail_sockaddr(const char *, VALUE, VALUE); + #endif diff --git a/ext/socket/socket.c b/ext/socket/socket.c index 32548944b2..10412be719 100644 --- a/ext/socket/socket.c +++ b/ext/socket/socket.c @@ -10,6 +10,40 @@ #include "rubysocket.h" +static VALUE sock_s_unpack_sockaddr_in(VALUE, VALUE); + +void +rsock_sys_fail_host_port(const char *mesg, VALUE host, VALUE port) +{ + VALUE message; + + port = rb_String(port); + + message = rb_sprintf("%s for \"%s\" port %s", + mesg, StringValueCStr(host), StringValueCStr(port)); + + rb_sys_fail_str(message); +} + +void +rsock_sys_fail_path(const char *mesg, VALUE path) +{ + VALUE message = rb_sprintf("%s for \"%s\"", + mesg, StringValueCStr(path)); + + rb_sys_fail_str(message); +} + +void +rsock_sys_fail_sockaddr(const char *mesg, VALUE sock, VALUE addr) +{ + VALUE host_port = sock_s_unpack_sockaddr_in(sock, addr); + + rsock_sys_fail_host_port(mesg, + RARRAY_PTR(host_port)[1], + RARRAY_PTR(host_port)[0]); +} + static void setup_domain_and_type(VALUE domain, int *dv, VALUE type, int *tv) { @@ -313,7 +347,7 @@ sock_connect(VALUE sock, VALUE addr) fd = fptr->fd; n = rsock_connect(fd, (struct sockaddr*)RSTRING_PTR(addr), RSTRING_LENINT(addr), 0); if (n < 0) { - rb_sys_fail("connect(2)"); + rsock_sys_fail_sockaddr("connect(2)", sock, addr); } return INT2FIX(n); @@ -375,7 +409,7 @@ sock_connect_nonblock(VALUE sock, VALUE addr) if (n < 0) { if (errno == EINPROGRESS) rb_mod_sys_fail(rb_mWaitWritable, "connect(2) would block"); - rb_sys_fail("connect(2)"); + rsock_sys_fail_sockaddr("connect(2)", sock, addr); } return INT2FIX(n); @@ -475,7 +509,7 @@ sock_bind(VALUE sock, VALUE addr) SockAddrStringValue(addr); GetOpenFile(sock, fptr); if (bind(fptr->fd, (struct sockaddr*)RSTRING_PTR(addr), RSTRING_LENINT(addr)) < 0) - rb_sys_fail("bind(2)"); + rsock_sys_fail_sockaddr("bind(2)", sock, addr); return INT2FIX(0); } @@ -910,7 +944,7 @@ sock_gethostname(VALUE obj) rb_secure(3); if (gethostname(buf, (int)sizeof buf - 1) < 0) - rb_sys_fail("gethostname"); + rb_sys_fail("gethostname(3)"); buf[sizeof buf - 1] = '\0'; return rb_str_new2(buf); @@ -1608,7 +1642,7 @@ socket_s_ip_address_list(VALUE self) fd = socket(AF_INET, SOCK_DGRAM, 0); if (fd == -1) - rb_sys_fail("socket"); + rb_sys_fail("socket(2)"); memset(&ln, 0, sizeof(ln)); ln.lifn_family = AF_UNSPEC; @@ -1678,7 +1712,7 @@ socket_s_ip_address_list(VALUE self) fd = socket(AF_INET, SOCK_DGRAM, 0); if (fd == -1) - rb_sys_fail("socket"); + rb_sys_fail("socket(2)"); bufsize = sizeof(initbuf); buf = initbuf; diff --git a/ext/socket/udpsocket.c b/ext/socket/udpsocket.c index 0ba4371f1a..4ea9db6583 100644 --- a/ext/socket/udpsocket.c +++ b/ext/socket/udpsocket.c @@ -93,7 +93,7 @@ udp_connect(VALUE sock, VALUE host, VALUE port) arg.fd = fptr->fd; ret = rb_ensure(udp_connect_internal, (VALUE)&arg, rsock_freeaddrinfo, (VALUE)arg.res); - if (!ret) rb_sys_fail("connect(2)"); + if (!ret) rsock_sys_fail_host_port("connect(2)", host, port); return INT2FIX(0); } @@ -126,7 +126,9 @@ udp_bind(VALUE sock, VALUE host, VALUE port) return INT2FIX(0); } freeaddrinfo(res0); - rb_sys_fail("bind(2)"); + + rsock_sys_fail_host_port("bind(2)", host, port); + return INT2FIX(0); } @@ -187,7 +189,7 @@ udp_send(int argc, VALUE *argv, VALUE sock) } } freeaddrinfo(res0); - rb_sys_fail("sendto(2)"); + rsock_sys_fail_host_port("sendto(2)", host, port); return INT2FIX(n); } diff --git a/ext/socket/unixsocket.c b/ext/socket/unixsocket.c index 6e4e0fc05a..3d3fc889e1 100644 --- a/ext/socket/unixsocket.c +++ b/ext/socket/unixsocket.c @@ -36,7 +36,7 @@ rsock_init_unixsock(VALUE sock, VALUE path, int server) SafeStringValue(path); fd = rsock_socket(AF_UNIX, SOCK_STREAM, 0); if (fd < 0) { - rb_sys_fail("socket(2)"); + rsock_sys_fail_path("socket(2)", path); } INIT_SOCKADDR_UN(&sockaddr, sizeof(struct sockaddr_un)); @@ -65,13 +65,13 @@ rsock_init_unixsock(VALUE sock, VALUE path, int server) if (status < 0) { close(fd); - rb_sys_fail_str(rb_inspect(path)); + rsock_sys_fail_path("connect(2)", path); } if (server) { if (listen(fd, SOMAXCONN) < 0) { close(fd); - rb_sys_fail("listen(2)"); + rsock_sys_fail_path("listen(2)", path); } } @@ -121,7 +121,7 @@ unix_path(VALUE sock) socklen_t len = (socklen_t)sizeof(addr); socklen_t len0 = len; if (getsockname(fptr->fd, (struct sockaddr*)&addr, &len) < 0) - rb_sys_fail(0); + rsock_sys_fail_path("getsockname(2)", fptr->pathv); if (len0 < len) len = len0; fptr->pathv = rb_obj_freeze(rsock_unixpath_str(&addr, len)); } @@ -255,7 +255,7 @@ unix_send_io(VALUE sock, VALUE val) arg.fd = fptr->fd; while ((int)BLOCKING_REGION_FD(sendmsg_blocking, &arg) == -1) { if (!rb_io_wait_writable(arg.fd)) - rb_sys_fail("sendmsg(2)"); + rsock_sys_fail_path("sendmsg(2)", fptr->pathv); } return Qnil; @@ -343,7 +343,7 @@ unix_recv_io(int argc, VALUE *argv, VALUE sock) arg.fd = fptr->fd; while ((int)BLOCKING_REGION_FD(recvmsg_blocking, &arg) == -1) { if (!rb_io_wait_readable(arg.fd)) - rb_sys_fail("recvmsg(2)"); + rsock_sys_fail_path("recvmsg(2)", fptr->pathv); } #if FD_PASSING_BY_MSG_CONTROL @@ -430,7 +430,7 @@ unix_addr(VALUE sock) GetOpenFile(sock, fptr); if (getsockname(fptr->fd, (struct sockaddr*)&addr, &len) < 0) - rb_sys_fail("getsockname(2)"); + rsock_sys_fail_path("getsockname(2)", fptr->pathv); if (len0 < len) len = len0; return rsock_unixaddr(&addr, len); } @@ -458,7 +458,7 @@ unix_peeraddr(VALUE sock) GetOpenFile(sock, fptr); if (getpeername(fptr->fd, (struct sockaddr*)&addr, &len) < 0) - rb_sys_fail("getpeername(2)"); + rsock_sys_fail_path("getpeername(2)", fptr->pathv); if (len0 < len) len = len0; return rsock_unixaddr(&addr, len); } -- cgit v1.2.3