ext/socket: BasicSocket#shutdown(how) accepts symbol or string as how.

MacRuby · Dec 17, 2011 · 7b43577 · 7b43577
1 parent efbe159
commit 7b43577
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ext/socket/socket.c b/ext/socket/socket.c
@@ -359,11 +359,11 @@ bsock_shutdown(VALUE sock, SEL sel, int argc, VALUE *argv)
     }
     rb_scan_args(argc, argv, "01", &howto);
     if (howto == Qnil)
-	how = 2;
+	how = SHUT_RDWR;
     else {
-	how = NUM2INT(howto);
-	if (how < 0 || 2 < how) {
-	    rb_raise(rb_eArgError, "`how' should be either 0, 1, 2");
+	how = shutdown_how_arg(howto);
+	if (how != SHUT_WR && how != SHUT_RD && how != SHUT_RDWR) {
+	    rb_raise(rb_eArgError, "`how' should be either :SHUT_RD, :SHUT_WR, :SHUT_RDWR");
 	}
     }
     GetOpenFile(sock, fptr);