diff --git a/src/transport/tcp.c b/src/transport/tcp.c index 761433c..6964c7e 100644 --- a/src/transport/tcp.c +++ b/src/transport/tcp.c @@ -42,6 +42,11 @@ # define close(fd) closesocket(fd) #endif +/* for tcp4 and tcp6 */ +#ifndef TCP_FAMILY +# define TCP_FAMILY AF_INET +#endif + /* TCP */ /* private */ @@ -200,7 +205,7 @@ static int _init_address(char const * name, struct sockaddr_in * sa) /* check for errors */ if(l < 0) return -1; - sa->sin_family = AF_INET; + sa->sin_family = TCP_FAMILY; sa->sin_port = htons(l); memcpy(&sa->sin_addr, he->h_addr_list[0], sizeof(sa->sin_addr)); return 0; diff --git a/src/transport/tcp4.c b/src/transport/tcp4.c index fc9a3d0..45225b6 100644 --- a/src/transport/tcp4.c +++ b/src/transport/tcp4.c @@ -15,4 +15,5 @@ +#define TCP_FAMILY AF_INET #include "tcp.c" diff --git a/src/transport/tcp6.c b/src/transport/tcp6.c index fc9a3d0..2b747c4 100644 --- a/src/transport/tcp6.c +++ b/src/transport/tcp6.c @@ -15,4 +15,5 @@ +#define TCP_FAMILY AF_INET6 #include "tcp.c"