From 9e457481438cf8b4dbe033b6fbd5ea612c4529c1 Mon Sep 17 00:00:00 2001 From: akr Date: Sun, 8 Feb 2009 15:37:55 +0000 Subject: * ext/socket/extconf.rb: check sys/param.h and sys/ucred.h. * ext/socket/rubysocket.h: include sys/param.h and sys/ucred.h. * ext/socket/option.c (inspect_local_peercred): new function to show LOCAL_PEERCRED socket option on FreeBSD. (sockopt_inspect): show as LOCAL_* socket option if AF_UNIX and level is 0. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@22137 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- ext/socket/extconf.rb | 3 +++ ext/socket/mkconstants.rb | 1 + ext/socket/option.c | 67 +++++++++++++++++++++++++++++++++++++++-------- ext/socket/rubysocket.h | 7 +++++ 4 files changed, 67 insertions(+), 11 deletions(-) (limited to 'ext') diff --git a/ext/socket/extconf.rb b/ext/socket/extconf.rb index c7d07d0a4d..f3f4529902 100644 --- a/ext/socket/extconf.rb +++ b/ext/socket/extconf.rb @@ -293,6 +293,9 @@ have_header("sys/ioctl.h") have_header("sys/sockio.h") have_header("net/if.h", headers) +have_header("sys/param.h", headers) +have_header("sys/ucred.h", headers) + unless have_type("socklen_t", headers) $defs << "-Dsocklen_t=int" end diff --git a/ext/socket/mkconstants.rb b/ext/socket/mkconstants.rb index 37079ba8f9..d30c4bbf0d 100644 --- a/ext/socket/mkconstants.rb +++ b/ext/socket/mkconstants.rb @@ -262,6 +262,7 @@ def_intern('intern_ipv6_optname', /\AIPV6_/, "IPV6_") def_intern('intern_tcp_optname', /\ATCP_/, "TCP_") def_intern('intern_udp_optname', /\AUDP_/, "UDP_") def_intern('intern_scm_optname', /\ASCM_/, "SCM_") +def_intern('intern_local_optname', /\ALOCAL_/, "LOCAL_") result = ERB.new(<<'EOS', nil, '%').result(binding) /* autogenerated file */ diff --git a/ext/socket/option.c b/ext/socket/option.c index c07779e796..c79c1f529f 100644 --- a/ext/socket/option.c +++ b/ext/socket/option.c @@ -256,6 +256,32 @@ inspect_peercred(int level, int optname, VALUE data, VALUE ret) } #endif +#if defined(LOCAL_PEERCRED) /* FreeBSD */ +static int +inspect_local_peercred(int level, int optname, VALUE data, VALUE ret) +{ + if (RSTRING_LEN(data) == sizeof(struct xucred)) { + struct xucred cred; + memcpy(&cred, RSTRING_PTR(data), sizeof(struct xucred)); + rb_str_catf(ret, " version=%u", cred.cr_version); + rb_str_catf(ret, " uid=%u", cred.cr_uid); + if (cred.cr_ngroups) { + int i; + char *sep = " groups="; + for (i = 0; i < cred.cr_ngroups; i++) { + rb_str_catf(ret, "%s%u", sep, cred.cr_groups[i]); + sep = ","; + } + } + rb_str_cat2(ret, " (xucred)"); + return 0; + } + else { + return -1; + } +} +#endif + static VALUE sockopt_inspect(VALUE self) { @@ -264,7 +290,7 @@ sockopt_inspect(VALUE self) int optname = NUM2INT(sockopt_optname(self)); VALUE data = sockopt_data(self); VALUE v, ret; - ID family_id, level_id; + ID family_id, level_id, optname_id; StringValue(data); @@ -276,17 +302,35 @@ sockopt_inspect(VALUE self) else rb_str_catf(ret, "family:%d", family); - level_id = intern_level(level); - if (level_id) - rb_str_catf(ret, " %s", rb_id2name(level_id)); - else - rb_str_catf(ret, " level:%d", level); + if (family == AF_UNIX && level == 0) { + rb_str_catf(ret, " level:%d", level); - v = optname_to_sym(level, optname); - if (SYMBOL_P(v)) - rb_str_catf(ret, " %s", rb_id2name(SYM2ID(v))); - else - rb_str_catf(ret, " optname:%d", optname); + optname_id = intern_local_optname(optname); + if (optname_id) + rb_str_catf(ret, " %s", rb_id2name(optname_id)); + else + rb_str_catf(ret, " optname:%d", optname); + } + else { + level_id = intern_level(level); + if (level_id) + rb_str_catf(ret, " %s", rb_id2name(level_id)); + else + rb_str_catf(ret, " level:%d", level); + + v = optname_to_sym(level, optname); + if (SYMBOL_P(v)) + rb_str_catf(ret, " %s", rb_id2name(SYM2ID(v))); + else + rb_str_catf(ret, " optname:%d", optname); + } + + if (family == AF_UNIX && level == 0) { + if (optname == LOCAL_PEERCRED) { + if (inspect_local_peercred(level, optname, data, ret) == -1) goto dump; + goto finish; + } + } switch (level) { # if defined(SOL_SOCKET) @@ -391,6 +435,7 @@ sockopt_inspect(VALUE self) rb_str_catf(ret, " %s", StringValueCStr(data)); } + finish: rb_str_cat2(ret, ">"); return ret; diff --git a/ext/socket/rubysocket.h b/ext/socket/rubysocket.h index 99028443e9..9e3692f7f1 100644 --- a/ext/socket/rubysocket.h +++ b/ext/socket/rubysocket.h @@ -73,6 +73,13 @@ #include #endif +#ifdef HAVE_SYS_PARAM_H +#include +#endif +#ifdef HAVE_SYS_UCRED_H +#include +#endif + #ifndef EWOULDBLOCK #define EWOULDBLOCK EAGAIN #endif -- cgit v1.2.3