src/ipaddr/ng_ipaddr.c

   1 /*
   2  * (c) 2008 Florian Westphal <fw@strlen.de>, public domain.
   3  */
   4
   5 #include "portab.h"
   6
   7 /**
   8  * @file
   9  * Functions for AF_ agnostic ipv4/ipv6 handling.
  10  */
  11
  12 #include <assert.h>
  13 #include <stdio.h>
  14 #include <string.h>
  15
  16 #ifdef HAVE_GETADDRINFO
  17 #include <netdb.h>
  18 #include <sys/types.h>
  19 #endif
  20
  21 #include "ng_ipaddr.h"
  22
  23 GLOBAL bool
  24 ng_ipaddr_init(ng_ipaddr_t *addr, const char *ip_str, UINT16 port)
  25 {
  26 #ifdef HAVE_GETADDRINFO
  27         int ret;
  28         char portstr[64];
  29         struct addrinfo *res0;
  30         struct addrinfo hints;
  31
  32         assert(ip_str);
  33
  34         memset(&hints, 0, sizeof(hints));
  35         hints.ai_flags = AI_NUMERICHOST;
  36 #ifndef WANT_IPV6       /* do not convert ipv6 addresses */
  37         hints.ai_family = AF_INET;
  38 #endif
  39
  40         /* some getaddrinfo implementations require that ai_socktype is set. */
  41         hints.ai_socktype = SOCK_STREAM;
  42
  43         /* silly, but ngircd stores UINT16 in server config, not string */
  44         snprintf(portstr, sizeof(portstr), "%u", (unsigned int) port);
  45
  46         ret = getaddrinfo(ip_str, portstr, &hints, &res0);
  47         if (ret != 0)
  48                 return false;
  49
  50         assert(sizeof(*addr) >= res0->ai_addrlen);
  51         if (sizeof(*addr) >= res0->ai_addrlen)
  52                 memcpy(addr, res0->ai_addr, res0->ai_addrlen);
  53         else
  54                 ret = -1;
  55         freeaddrinfo(res0);
  56         return ret == 0;
  57 #else /* HAVE_GETADDRINFO */
  58         assert(ip_str);
  59         memset(addr, 0, sizeof *addr);
  60 #ifdef HAVE_sockaddr_in_len
  61         addr->sin4.sin_len = sizeof(addr->sin4);
  62 #endif
  63         addr->sin4.sin_family = AF_INET;
  64 # ifdef HAVE_INET_ATON
  65         if (inet_aton(ip_str, &addr->sin4.sin_addr) == 0)
  66                 return false;
  67 # else
  68         addr->sin4.sin_addr.s_addr = inet_addr(ip_str);
  69         if (addr->sin4.sin_addr.s_addr == (unsigned) -1)
  70                 return false;
  71 # endif
  72         ng_ipaddr_setport(addr, port);
  73         return true;
  74 #endif /* HAVE_GETADDRINFO */
  75 }
  76
  77
  78 GLOBAL void
  79 ng_ipaddr_setport(ng_ipaddr_t *a, UINT16 port)
  80 {
  81 #ifdef WANT_IPV6
  82         int af;
  83
  84         assert(a != NULL);
  85
  86         af = a->sa.sa_family;
  87
  88         assert(af == AF_INET || af == AF_INET6);
  89
  90         switch (af) {
  91         case AF_INET:
  92                 a->sin4.sin_port = htons(port);
  93                 break;
  94         case AF_INET6:
  95                 a->sin6.sin6_port = htons(port);
  96                 break;
  97         }
  98 #else /* WANT_IPV6 */
  99         assert(a != NULL);
 100         assert(a->sin4.sin_family == AF_INET);
 101         a->sin4.sin_port = htons(port);
 102 #endif /* WANT_IPV6 */
 103 }
 104
 105
 106
 107 GLOBAL bool
 108 ng_ipaddr_ipequal(const ng_ipaddr_t *a, const ng_ipaddr_t *b)
 109 {
 110         assert(a != NULL);
 111         assert(b != NULL);
 112 #ifdef WANT_IPV6
 113         if (a->sa.sa_family != b->sa.sa_family)
 114                 return false;
 115         assert(ng_ipaddr_salen(a) == ng_ipaddr_salen(b));
 116         switch (a->sa.sa_family) {
 117         case AF_INET6:
 118                 return IN6_ARE_ADDR_EQUAL(&a->sin6.sin6_addr, &b->sin6.sin6_addr);
 119         case AF_INET:
 120                 return memcmp(&a->sin4.sin_addr, &b->sin4.sin_addr, sizeof(a->sin4.sin_addr)) == 0;
 121         }
 122         return false;
 123 #else
 124         assert(a->sin4.sin_family == AF_INET);
 125         assert(b->sin4.sin_family == AF_INET);
 126         return memcmp(&a->sin4.sin_addr, &b->sin4.sin_addr, sizeof(a->sin4.sin_addr)) == 0;
 127 #endif
 128 }
 129
 130
 131 #ifdef WANT_IPV6
 132 GLOBAL const char *
 133 ng_ipaddr_tostr(const ng_ipaddr_t *addr)
 134 {
 135         static char strbuf[NG_INET_ADDRSTRLEN];
 136
 137         strbuf[0] = 0;
 138
 139         ng_ipaddr_tostr_r(addr, strbuf);
 140         return strbuf;
 141 }
 142
 143
 144 /* str must be at least NG_INET_ADDRSTRLEN bytes long */
 145 GLOBAL bool
 146 ng_ipaddr_tostr_r(const ng_ipaddr_t *addr, char *str)
 147 {
 148 #ifdef HAVE_GETNAMEINFO
 149         const struct sockaddr *sa = (const struct sockaddr *) addr;
 150         int ret;
 151
 152         *str = 0;
 153
 154         ret = getnameinfo(sa, ng_ipaddr_salen(addr),
 155                         str, NG_INET_ADDRSTRLEN, NULL, 0, NI_NUMERICHOST);
 156         /*
 157          * avoid leading ':'.
 158          * causes mis-interpretation of client host in e.g. /WHOIS
 159          */
 160         if (*str == ':') {
 161                 char tmp[NG_INET_ADDRSTRLEN] = "0";
 162                 ret = getnameinfo(sa, ng_ipaddr_salen(addr),
 163                                   tmp + 1, (socklen_t)sizeof(tmp) - 1,
 164                                   NULL, 0, NI_NUMERICHOST);
 165                 if (ret == 0)
 166                         strlcpy(str, tmp, NG_INET_ADDRSTRLEN);
 167         }
 168         assert (ret == 0);
 169         return ret == 0;
 170 #else
 171         abort(); /* WANT_IPV6 depends on HAVE_GETNAMEINFO */
 172 #endif
 173 }
 174
 175 #endif /* WANT_IPV6 */
 176
 177 /* -eof- */