Fix socket_accept usage to depend on family.

1 files changed, 15 insertions, 1 deletions

diff --git a/src/inet.c b/src/inet.c
index dfee700..52f7397 100644
--- a/src/inet.c
+++ b/src/inet.c
@@ -263,7 +263,6 @@ int inet_meth_getpeername(lua_State *L, p_socket ps, int family)
                 lua_pushliteral(L, "inet6");
                 return 3;
             }
-            return 2;
         }
         default:
             lua_pushnil(L);
@@ -424,6 +423,21 @@ const char *inet_tryconnect(p_socket ps, const char *address,
 }
 
 /*-------------------------------------------------------------------------*\
+* Tries to accept a socket
+\*-------------------------------------------------------------------------*/
+const char *inet_tryaccept(p_socket server, int family, p_socket client, p_timeout tm)
+{
+        socklen_t len;
+        t_sockaddr_storage addr;
+        if (family == PF_INET6) {
+                len = sizeof(struct sockaddr_in6);
+        } else {
+                len = sizeof(struct sockaddr_in);
+        }
+        return socket_strerror(socket_accept(server, client, (SA *) &addr, &len, tm));
+}
+
+/*-------------------------------------------------------------------------*\
 * Tries to bind socket to (address, port)
 \*-------------------------------------------------------------------------*/
 const char *inet_trybind(p_socket ps, const char *address, const char *serv,