ext/socket: add error handling of listen

MacRuby · Dec 7, 2011 · 6873fae · 6873fae
1 parent e94a3ca
commit 6873fae
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/ext/socket/socket.c b/ext/socket/socket.c
@@ -1303,8 +1303,13 @@ init_inetsock_internal(struct inetsock_arg *arg)
 
     arg->fd = -1;
 
-    if (type == INET_SERVER)
-	listen(fd, 5);
+    if (type == INET_SERVER) {
+	status = listen(fd, 5);
+	if (status < 0) {
+	    close(fd);
+	    syscall = "listen(2)";
+	}
+    }
 
     /* create new instance */
     return init_sock(arg->sock, fd);
@@ -1634,7 +1639,12 @@ init_unixsock(VALUE sock, VALUE path, int server)
 	rb_sys_fail(sockaddr.sun_path);
     }
 
-    if (server) listen(fd, 5);
+    if (server) {
+	if (listen(fd, 5) < 0) {
+	    close(fd);
+	    rb_sys_fail("listen(2)");
+	}
+    }
 
     init_sock(sock, fd);
     if (server) {