kotovalexarian-likes-github/ruby--ruby
1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
Code Releases Activity

socket/option.c: helper functions

Browse source 
* ext/socket/option.c (sockopt_pack_byte, sockopt_pack_int):
  extract helper functions to pack integers in a String.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@44658 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
nobu 2014-01-19 14:56:05 +00:00
parent 5c179cd66c
commit 5ecbe189af
1 changed files with 26 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
ext/socket/option.c
View file

@ -2,6 +2,22 @@
VALUE rb_cSockOpt; VALUE rb_cSockOpt;
#define pack_var(v) rb_str_new((const char *)&(v), sizeof(v))
static VALUE
sockopt_pack_byte(VALUE value)
{
char i = NUM2CHR(rb_to_int(value));
return pack_var(i);
}
static VALUE
sockopt_pack_int(VALUE value)
{
int i = NUM2INT(rb_to_int(value));
return pack_var(i);
}
static VALUE static VALUE
constant_to_sym(int constant, ID (*intern_const)(int)) constant_to_sym(int constant, ID (*intern_const)(int))
{ {
@ -157,8 +173,7 @@ sockopt_s_byte(VALUE klass, VALUE vfamily, VALUE vlevel, VALUE voptname, VALUE v
int family = rsock_family_arg(vfamily); int family = rsock_family_arg(vfamily);
int level = rsock_level_arg(family, vlevel); int level = rsock_level_arg(family, vlevel);
int optname = rsock_optname_arg(family, level, voptname); int optname = rsock_optname_arg(family, level, voptname);
unsigned char i = (unsigned char)NUM2CHR(vint); return rsock_sockopt_new(family, level, optname, sockopt_pack_byte(vint));
return rsock_sockopt_new(family, level, optname, rb_str_new((char*)&i, sizeof(i)));
} }
/* /*
@ -199,8 +214,7 @@ sockopt_s_int(VALUE klass, VALUE vfamily, VALUE vlevel, VALUE voptname, VALUE vi
int family = rsock_family_arg(vfamily); int family = rsock_family_arg(vfamily);
int level = rsock_level_arg(family, vlevel); int level = rsock_level_arg(family, vlevel);
int optname = rsock_optname_arg(family, level, voptname); int optname = rsock_optname_arg(family, level, voptname);
int i = NUM2INT(vint); return rsock_sockopt_new(family, level, optname, sockopt_pack_int(vint));
return rsock_sockopt_new(family, level, optname, rb_str_new((char*)&i, sizeof(i)));
} }
/* /*
@ -248,7 +262,7 @@ sockopt_s_bool(VALUE klass, VALUE vfamily, VALUE vlevel, VALUE voptname, VALUE v
int level = rsock_level_arg(family, vlevel); int level = rsock_level_arg(family, vlevel);
int optname = rsock_optname_arg(family, level, voptname); int optname = rsock_optname_arg(family, level, voptname);
int i = RTEST(vbool) ? 1 : 0; int i = RTEST(vbool) ? 1 : 0;
return rsock_sockopt_new(family, level, optname, rb_str_new((char*)&i, sizeof(i))); return rsock_sockopt_new(family, level, optname, pack_var(i));
} }
/* /*
@ -298,7 +312,7 @@ sockopt_s_linger(VALUE klass, VALUE vonoff, VALUE vsecs)
else else
l.l_onoff = RTEST(vonoff) ? 1 : 0; l.l_onoff = RTEST(vonoff) ? 1 : 0;
l.l_linger = NUM2INT(vsecs); l.l_linger = NUM2INT(vsecs);
return rsock_sockopt_new(AF_UNSPEC, SOL_SOCKET, SO_LINGER, rb_str_new((char*)&l, sizeof(l))); return rsock_sockopt_new(AF_UNSPEC, SOL_SOCKET, SO_LINGER, pack_var(l));
} }
/* /*
@ -354,12 +368,12 @@ sockopt_s_ipv4_multicast_loop(VALUE klass, VALUE value)
{ {
#if defined(IPPROTO_IP) && defined(IP_MULTICAST_LOOP) #if defined(IPPROTO_IP) && defined(IP_MULTICAST_LOOP)
# if defined(__NetBSD__) || defined(__OpenBSD__) # if defined(__NetBSD__) || defined(__OpenBSD__)
unsigned char i = NUM2CHR(rb_to_int(value)); VALUE o = sockopt_pack_byte(value);
# else # else
int i = NUM2INT(rb_to_int(value)); VALUE o = sockopt_pack_int(value);
# endif # endif
return rsock_sockopt_new(AF_INET, IPPROTO_IP, IP_MULTICAST_LOOP, return rsock_sockopt_new(AF_INET, IPPROTO_IP, IP_MULTICAST_LOOP,
rb_str_new((char*)&i, sizeof(i))); o);
#else #else
# error IPPROTO_IP or IP_MULTICAST_LOOP is not implemented # error IPPROTO_IP or IP_MULTICAST_LOOP is not implemented
#endif #endif
@ -417,12 +431,12 @@ sockopt_s_ipv4_multicast_ttl(VALUE klass, VALUE value)
{ {
#if defined(IPPROTO_IP) && defined(IP_MULTICAST_TTL) #if defined(IPPROTO_IP) && defined(IP_MULTICAST_TTL)
# if defined(__NetBSD__) || defined(__OpenBSD__) # if defined(__NetBSD__) || defined(__OpenBSD__)
unsigned char i = NUM2CHR(rb_to_int(value)); VALUE o = sockopt_pack_byte(value);
# else # else
int i = NUM2INT(rb_to_int(value)); VALUE o = sockopt_pack_int(value);
# endif # endif
return rsock_sockopt_new(AF_INET, IPPROTO_IP, IP_MULTICAST_TTL, return rsock_sockopt_new(AF_INET, IPPROTO_IP, IP_MULTICAST_TTL,
rb_str_new((char*)&i, sizeof(i))); o);
#else #else
# error IPPROTO_IP or IP_MULTICAST_TTL is not implemented # error IPPROTO_IP or IP_MULTICAST_TTL is not implemented
#endif #endif