Added SILC Thread Queue API
[crypto.git] / apps / irssi / src / core / network.c
index e53b01897b4938ddfe1ea3e092165160a26e9428..21235b2af8380470aac142452fa4e56993d81411 100644 (file)
@@ -21,6 +21,8 @@
 #include "module.h"
 #include "network.h"
 
+#include <sys/un.h>
+
 #ifndef INADDR_NONE
 #  define INADDR_NONE INADDR_BROADCAST
 #endif
@@ -49,6 +51,11 @@ union sockaddr_union {
 /* Cygwin need this, don't know others.. */
 /*#define BLOCKING_SOCKETS 1*/
 
+IPADDR ip4_any = {
+       AF_INET,
+       { INADDR_ANY }
+};
+
 int net_ip_compare(IPADDR *ip1, IPADDR *ip2)
 {
        if (ip1->family != ip2->family)
@@ -63,13 +70,7 @@ int net_ip_compare(IPADDR *ip1, IPADDR *ip2)
 }
 
 
-/* copy IP to sockaddr */
-#ifdef G_CAN_INLINE
-G_INLINE_FUNC
-#else
-static
-#endif
-void sin_set_ip(union sockaddr_union *so, const IPADDR *ip)
+static void sin_set_ip(union sockaddr_union *so, const IPADDR *ip)
 {
        if (ip == NULL) {
 #ifdef HAVE_IPV6
@@ -103,27 +104,17 @@ void sin_get_ip(const union sockaddr_union *so, IPADDR *ip)
                memcpy(&ip->ip, &so->sin.sin_addr, 4);
 }
 
-#ifdef G_CAN_INLINE
-G_INLINE_FUNC
-#else
-static
-#endif
-void sin_set_port(union sockaddr_union *so, int port)
+static void sin_set_port(union sockaddr_union *so, int port)
 {
 #ifdef HAVE_IPV6
        if (so->sin.sin_family == AF_INET6)
-                so->sin6.sin6_port = htons(port);
+                so->sin6.sin6_port = htons((unsigned short)port);
        else
 #endif
                so->sin.sin_port = htons((unsigned short)port);
 }
 
-#ifdef G_CAN_INLINE
-G_INLINE_FUNC
-#else
-static
-#endif
-int sin_get_port(union sockaddr_union *so)
+static int sin_get_port(union sockaddr_union *so)
 {
 #ifdef HAVE_IPV6
        if (so->sin.sin_family == AF_INET6)
@@ -199,10 +190,12 @@ GIOChannel *net_connect_ip(IPADDR *ip, int port, IPADDR *my_ip)
        /* set our own address */
        if (my_ip != NULL) {
                sin_set_ip(&so, my_ip);
-               if (bind(handle, &so.sa, SIZEOF_SOCKADDR(so)) == -1) {
-                       /* failed, set it back to INADDR_ANY */
-                       sin_set_ip(&so, NULL);
-                       bind(handle, &so.sa, SIZEOF_SOCKADDR(so));
+               if (bind(handle, &so.sa, SIZEOF_SOCKADDR(so)) < 0) {
+                       int old_errno = errno;
+
+                       close(handle);
+                       errno = old_errno;
+                       return NULL;
                }
        }
 
@@ -217,7 +210,42 @@ GIOChannel *net_connect_ip(IPADDR *ip, int port, IPADDR *my_ip)
        if (ret < 0 && WSAGetLastError() != WSAEWOULDBLOCK)
 #endif
        {
+               int old_errno = errno;
                close(handle);
+               errno = old_errno;
+               return NULL;
+       }
+
+       return g_io_channel_new(handle);
+}
+
+/* Connect to named UNIX socket */
+GIOChannel *net_connect_unix(const char *path)
+{
+       struct sockaddr_un sa;
+       int handle, ret;
+
+       /* create the socket */
+       handle = socket(PF_UNIX, SOCK_STREAM, 0);
+       if (handle == -1)
+               return NULL;
+
+       /* set socket options */
+#ifndef WIN32
+       fcntl(handle, F_SETFL, O_NONBLOCK);
+#endif
+
+       /* connect */
+       memset(&sa, 0, sizeof(sa));
+       sa.sun_family = AF_UNIX;
+       strncpy(sa.sun_path, path, sizeof(sa.sun_path)-1);
+       sa.sun_path[sizeof(sa.sun_path)-1] = '\0';
+
+       ret = connect(handle, (struct sockaddr *) &sa, sizeof(sa));
+       if (ret < 0 && errno != EINPROGRESS) {
+               int old_errno = errno;
+               close(handle);
+               errno = old_errno;
                return NULL;
        }
 
@@ -244,8 +272,8 @@ GIOChannel *net_listen(IPADDR *my_ip, int *port)
        g_return_val_if_fail(port != NULL, NULL);
 
        memset(&so, 0, sizeof(so));
-       sin_set_port(&so, *port);
        sin_set_ip(&so, my_ip);
+       sin_set_port(&so, *port);
 
        /* create the socket */
        handle = socket(so.sin.sin_family, SOCK_STREAM, 0);
@@ -318,7 +346,7 @@ GIOChannel *net_accept(GIOChannel *handle, IPADDR *addr, int *port)
 /* Read data from socket, return number of bytes read, -1 = error */
 int net_receive(GIOChannel *handle, char *buf, int len)
 {
-        unsigned int ret;
+        gsize ret;
        int err;
 
        g_return_val_if_fail(handle != NULL, -1);
@@ -337,7 +365,7 @@ int net_receive(GIOChannel *handle, char *buf, int len)
 /* Transmit data, return number of bytes sent, -1 = error */
 int net_transmit(GIOChannel *handle, const char *data, int len)
 {
-        unsigned int ret;
+        gsize ret;
        int err;
 
        g_return_val_if_fail(handle != NULL, -1);
@@ -379,9 +407,10 @@ int net_gethostbyname(const char *addr, IPADDR *ip4, IPADDR *ip6)
 #ifdef HAVE_IPV6
        union sockaddr_union *so;
        struct addrinfo hints, *ai, *ailist;
-       int ret, count;
+       int ret, count_v4, count_v6, use_v4, use_v6;
 #else
        struct hostent *hp;
+       int count;
 #endif
 
        g_return_val_if_fail(addr != NULL, -1);
@@ -398,27 +427,54 @@ int net_gethostbyname(const char *addr, IPADDR *ip4, IPADDR *ip6)
        if (ret != 0)
                return ret;
 
-        count = 0;
-       for (ai = ailist; ai != NULL && count < 2; ai = ai->ai_next) {
+       /* count IPs */
+        count_v4 = count_v6 = 0;
+       for (ai = ailist; ai != NULL; ai = ai->ai_next) {
+               if (ai->ai_family == AF_INET)
+                       count_v4++;
+               else if (ai->ai_family == AF_INET6)
+                       count_v6++;
+       }
+
+       if (count_v4 == 0 && count_v6 == 0)
+               return HOST_NOT_FOUND; /* shouldn't happen? */
+
+       /* if there are multiple addresses, return random one */
+       use_v4 = count_v4 <= 1 ? 0 : rand() % count_v4;
+       use_v6 = count_v6 <= 1 ? 0 : rand() % count_v6;
+
+       count_v4 = count_v6 = 0;
+       for (ai = ailist; ai != NULL; ai = ai->ai_next) {
                so = (union sockaddr_union *) ai->ai_addr;
 
-               if (ai->ai_family == AF_INET6 && ip6->family == 0) {
-                       sin_get_ip(so, ip6);
-                        count++;
-               } else if (ai->ai_family == AF_INET && ip4->family == 0) {
-                       sin_get_ip(so, ip4);
-                        count++;
+               if (ai->ai_family == AF_INET) {
+                       if (use_v4 == count_v4)
+                               sin_get_ip(so, ip4);
+                        count_v4++;
+               } else if (ai->ai_family == AF_INET6) {
+                       if (use_v6 == count_v6)
+                               sin_get_ip(so, ip6);
+                       count_v6++;
                }
        }
        freeaddrinfo(ailist);
-       return count > 0 ? 0 : 1;
+       return 0;
 #else
        hp = gethostbyname(addr);
        if (hp == NULL)
                return h_errno;
 
+       /* count IPs */
+       count = 0;
+       while (hp->h_addr_list[count] != NULL)
+               count++;
+
+       if (count == 0)
+               return HOST_NOT_FOUND; /* shouldn't happen? */
+
+       /* if there are multiple addresses, return random one */
        ip4->family = AF_INET;
-       memcpy(&ip4->ip, hp->h_addr, 4);
+       memcpy(&ip4->ip, hp->h_addr_list[rand() % count], 4);
 
        return 0;
 #endif
@@ -429,33 +485,31 @@ int net_gethostbyname(const char *addr, IPADDR *ip4, IPADDR *ip6)
 int net_gethostbyaddr(IPADDR *ip, char **name)
 {
 #ifdef HAVE_IPV6
-       struct addrinfo req, *ai;
+       union sockaddr_union so;
        int host_error;
+       char hostname[NI_MAXHOST];
 #else
        struct hostent *hp;
 #endif
-       char ipname[MAX_IP_LEN];
 
        g_return_val_if_fail(ip != NULL, -1);
        g_return_val_if_fail(name != NULL, -1);
 
-       net_ip2host(ip, ipname);
-
        *name = NULL;
 #ifdef HAVE_IPV6
-       memset(&req, 0, sizeof(struct addrinfo));
-       req.ai_socktype = SOCK_STREAM;
-       req.ai_flags = AI_CANONNAME;
+       memset(&so, 0, sizeof(so));
+       sin_set_ip(&so, ip);
 
        /* save error to host_error for later use */
-       host_error = getaddrinfo(ipname, NULL, &req, &ai);
-       if (host_error != 0)
-               return host_error;
-       *name = g_strdup(ai->ai_canonname);
+        host_error = getnameinfo((struct sockaddr *) &so, sizeof(so),
+                                 hostname, sizeof(hostname), NULL, 0, 0);
+        if (host_error != 0)
+                return host_error;
 
-       freeaddrinfo(ai);
+       *name = g_strdup(hostname);
 #else
-       hp = gethostbyaddr(ipname, strlen(ipname), AF_INET);
+       if (ip->family != AF_INET) return -1;
+       hp = gethostbyaddr((const char *) &ip->ip, 4, AF_INET);
        if (hp == NULL) return -1;
 
        *name = g_strdup(hp->h_name);
@@ -472,12 +526,16 @@ int net_ip2host(IPADDR *ip, char *host)
 #else
        unsigned long ip4;
 
-       ip4 = ntohl(ip->ip.s_addr);
-       g_snprintf(host, MAX_IP_LEN, "%lu.%lu.%lu.%lu",
-                  (ip4 & 0xff000000UL) >> 24,
-                  (ip4 & 0x00ff0000) >> 16,
-                  (ip4 & 0x0000ff00) >> 8,
-                  (ip4 & 0x000000ff));
+       if (ip->family != AF_INET) {
+               strcpy(host, "0.0.0.0");
+       } else {
+               ip4 = ntohl(ip->ip.s_addr);
+               g_snprintf(host, MAX_IP_LEN, "%lu.%lu.%lu.%lu",
+                          (ip4 & 0xff000000UL) >> 24,
+                          (ip4 & 0x00ff0000) >> 16,
+                          (ip4 & 0x0000ff00) >> 8,
+                          (ip4 & 0x000000ff));
+       }
 #endif
        return 0;
 }
@@ -486,15 +544,16 @@ int net_host2ip(const char *host, IPADDR *ip)
 {
        unsigned long addr;
 
-#ifdef HAVE_IPV6
        if (strchr(host, ':') != NULL) {
                /* IPv6 */
                ip->family = AF_INET6;
+#ifdef HAVE_IPV6
                if (inet_pton(AF_INET6, host, &ip->ip) == 0)
                        return -1;
-       } else
+#else
+               ip->ip.s_addr = 0;
 #endif
-       {
+       } else {
                /* IPv4 */
                ip->family = AF_INET;
 #ifdef HAVE_INET_ATON
@@ -560,7 +619,11 @@ const char *net_gethosterror(int error)
 int net_hosterror_notfound(int error)
 {
 #ifdef HAVE_IPV6
+#ifdef EAI_NODATA /* NODATA is depricated */
        return error != 1 && (error == EAI_NONAME || error == EAI_NODATA);
+#else
+       return error != 1 && (error == EAI_NONAME);
+#endif
 #else
        return error == HOST_NOT_FOUND || error == NO_ADDRESS;
 #endif