summaryrefslogtreecommitdiff
path: root/ext/socket/socket.c
diff options
context:
space:
mode:
authorakr <akr@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>2009-01-01 15:59:10 +0000
committerakr <akr@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>2009-01-01 15:59:10 +0000
commitc6bbf28a9f139a5fbf34e54673328d3d58e8e397 (patch)
treeca298128a1041c1a1fff2e5a77513ef7e0a190ea /ext/socket/socket.c
parent31fa80daddbdeb074bd8bb4afb5e129f1c92ec82 (diff)
* ext/socket/socket.c (family_arg): accept symbols as well.
(socktype_arg): ditto. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@21257 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'ext/socket/socket.c')
-rw-r--r--ext/socket/socket.c16
1 files changed, 12 insertions, 4 deletions
diff --git a/ext/socket/socket.c b/ext/socket/socket.c
index 94244fc5ad..7b7aa416ac 100644
--- a/ext/socket/socket.c
+++ b/ext/socket/socket.c
@@ -249,9 +249,13 @@ family_arg(VALUE domain)
char *ptr;
int ret;
- tmp = rb_check_string_type(domain);
- if (!NIL_P(tmp)) {
+ if (SYMBOL_P(domain)) {
+ domain = rb_sym_to_s(domain);
+ goto str;
+ }
+ else if (!NIL_P(tmp = rb_check_string_type(domain))) {
domain = tmp;
+ str:
rb_check_safe_obj(domain);
ptr = RSTRING_PTR(domain);
if (family_to_int(ptr, RSTRING_LEN(domain), &ret) == -1)
@@ -271,9 +275,13 @@ socktype_arg(VALUE type)
char *ptr;
int ret;
- tmp = rb_check_string_type(type);
- if (!NIL_P(tmp)) {
+ if (SYMBOL_P(type)) {
+ type = rb_sym_to_s(type);
+ goto str;
+ }
+ else if (!NIL_P(tmp = rb_check_string_type(type))) {
type = tmp;
+ str:
rb_check_safe_obj(type);
ptr = RSTRING_PTR(type);
if (socktype_to_int(ptr, RSTRING_LEN(type), &ret) == -1)