src/netutl.c

   1 /*
   2     netutl.c -- some supporting network utility code
   3     Copyright (C) 1998-2005 Ivo Timmermans
   4                   2000-2013 Guus Sliepen <guus@tinc-vpn.org>
   5
   6     This program is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     This program is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License along
  17     with this program; if not, write to the Free Software Foundation, Inc.,
  18     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19 */
  20
  21 #include "system.h"
  22
  23 #include "net.h"
  24 #include "netutl.h"
  25 #include "logger.h"
  26 #include "utils.h"
  27 #include "xalloc.h"
  28
  29 bool hostnames = false;
  30
  31 /*
  32   Turn a string into a struct addrinfo.
  33   Return NULL on failure.
  34 */
  35 struct addrinfo *str2addrinfo(const char *address, const char *service, int socktype) {
  36         struct addrinfo *ai, hint = {0};
  37         int err;
  38
  39         hint.ai_family = addressfamily;
  40         hint.ai_socktype = socktype;
  41
  42 #if HAVE_DECL_RES_INIT
  43         res_init();
  44 #endif
  45         err = getaddrinfo(address, service, &hint, &ai);
  46
  47         if(err) {
  48                 logger(DEBUG_ALWAYS, LOG_WARNING, "Error looking up %s port %s: %s", address, service, err == EAI_SYSTEM ? strerror(errno) : gai_strerror(err));
  49                 return NULL;
  50         }
  51
  52         return ai;
  53 }
  54
  55 sockaddr_t str2sockaddr(const char *address, const char *port) {
  56         struct addrinfo *ai, hint = {0};
  57         sockaddr_t result = {{0}};
  58         int err;
  59
  60         hint.ai_family = AF_UNSPEC;
  61         hint.ai_flags = AI_NUMERICHOST;
  62         hint.ai_socktype = SOCK_STREAM;
  63
  64         err = getaddrinfo(address, port, &hint, &ai);
  65
  66         if(err || !ai) {
  67                 logger(DEBUG_SCARY_THINGS, LOG_DEBUG, "Unknown type address %s port %s", address, port);
  68                 result.sa.sa_family = AF_UNKNOWN;
  69                 result.unknown.address = xstrdup(address);
  70                 result.unknown.port = xstrdup(port);
  71                 return result;
  72         }
  73
  74         memcpy(&result, ai->ai_addr, ai->ai_addrlen);
  75         freeaddrinfo(ai);
  76
  77         return result;
  78 }
  79
  80 void sockaddr2str(const sockaddr_t *sa, char **addrstr, char **portstr) {
  81         char address[NI_MAXHOST];
  82         char port[NI_MAXSERV];
  83         char *scopeid;
  84         int err;
  85
  86         if(sa->sa.sa_family == AF_UNSPEC) {
  87                 if(addrstr) {
  88                         *addrstr = xstrdup("unspec");
  89                 }
  90
  91                 if(portstr) {
  92                         *portstr = xstrdup("unspec");
  93                 }
  94
  95                 return;
  96         } else if(sa->sa.sa_family == AF_UNKNOWN) {
  97                 if(addrstr) {
  98                         *addrstr = xstrdup(sa->unknown.address);
  99                 }
 100
 101                 if(portstr) {
 102                         *portstr = xstrdup(sa->unknown.port);
 103                 }
 104
 105                 return;
 106         }
 107
 108         err = getnameinfo(&sa->sa, SALEN(sa->sa), address, sizeof(address), port, sizeof port, NI_NUMERICHOST | NI_NUMERICSERV);
 109
 110         if(err) {
 111                 logger(DEBUG_ALWAYS, LOG_ERR, "Error while translating addresses: %s", err == EAI_SYSTEM ? strerror(errno) : gai_strerror(err));
 112                 abort();
 113         }
 114
 115         scopeid = strchr(address, '%');
 116
 117         if(scopeid) {
 118                 *scopeid = '\0';        /* Descope. */
 119         }
 120
 121         if(addrstr) {
 122                 *addrstr = xstrdup(address);
 123         }
 124
 125         if(portstr) {
 126                 *portstr = xstrdup(port);
 127         }
 128 }
 129
 130 char *sockaddr2hostname(const sockaddr_t *sa) {
 131         char *str;
 132         char address[NI_MAXHOST] = "unknown";
 133         char port[NI_MAXSERV] = "unknown";
 134         int err;
 135
 136         if(sa->sa.sa_family == AF_UNSPEC) {
 137                 xasprintf(&str, "unspec port unspec");
 138                 return str;
 139         } else if(sa->sa.sa_family == AF_UNKNOWN) {
 140                 xasprintf(&str, "%s port %s", sa->unknown.address, sa->unknown.port);
 141                 return str;
 142         }
 143
 144         err = getnameinfo(&sa->sa, SALEN(sa->sa), address, sizeof(address), port, sizeof port,
 145                           hostnames ? 0 : (NI_NUMERICHOST | NI_NUMERICSERV));
 146
 147         if(err) {
 148                 logger(DEBUG_ALWAYS, LOG_ERR, "Error while looking up hostname: %s", err == EAI_SYSTEM ? strerror(errno) : gai_strerror(err));
 149         }
 150
 151         xasprintf(&str, "%s port %s", address, port);
 152
 153         return str;
 154 }
 155
 156 int sockaddrcmp_noport(const sockaddr_t *a, const sockaddr_t *b) {
 157         int result;
 158
 159         result = a->sa.sa_family - b->sa.sa_family;
 160
 161         if(result) {
 162                 return result;
 163         }
 164
 165         switch(a->sa.sa_family) {
 166         case AF_UNSPEC:
 167                 return 0;
 168
 169         case AF_UNKNOWN:
 170                 return strcmp(a->unknown.address, b->unknown.address);
 171
 172         case AF_INET:
 173                 return memcmp(&a->in.sin_addr, &b->in.sin_addr, sizeof(a->in.sin_addr));
 174
 175         case AF_INET6:
 176                 return memcmp(&a->in6.sin6_addr, &b->in6.sin6_addr, sizeof(a->in6.sin6_addr));
 177
 178         default:
 179                 logger(DEBUG_ALWAYS, LOG_ERR, "sockaddrcmp() was called with unknown address family %d, exitting!",
 180                        a->sa.sa_family);
 181                 abort();
 182         }
 183 }
 184
 185 int sockaddrcmp(const sockaddr_t *a, const sockaddr_t *b) {
 186         int result;
 187
 188         result = a->sa.sa_family - b->sa.sa_family;
 189
 190         if(result) {
 191                 return result;
 192         }
 193
 194         switch(a->sa.sa_family) {
 195         case AF_UNSPEC:
 196                 return 0;
 197
 198         case AF_UNKNOWN:
 199                 result = strcmp(a->unknown.address, b->unknown.address);
 200
 201                 if(result) {
 202                         return result;
 203                 }
 204
 205                 return strcmp(a->unknown.port, b->unknown.port);
 206
 207         case AF_INET:
 208                 result = memcmp(&a->in.sin_addr, &b->in.sin_addr, sizeof(a->in.sin_addr));
 209
 210                 if(result) {
 211                         return result;
 212                 }
 213
 214                 return memcmp(&a->in.sin_port, &b->in.sin_port, sizeof(a->in.sin_port));
 215
 216         case AF_INET6:
 217                 result = memcmp(&a->in6.sin6_addr, &b->in6.sin6_addr, sizeof(a->in6.sin6_addr));
 218
 219                 if(result) {
 220                         return result;
 221                 }
 222
 223                 return memcmp(&a->in6.sin6_port, &b->in6.sin6_port, sizeof(a->in6.sin6_port));
 224
 225         default:
 226                 logger(DEBUG_ALWAYS, LOG_ERR, "sockaddrcmp() was called with unknown address family %d, exitting!",
 227                        a->sa.sa_family);
 228                 abort();
 229         }
 230 }
 231
 232 void sockaddrcpy(sockaddr_t *a, const sockaddr_t *b) {
 233         if(b->sa.sa_family != AF_UNKNOWN) {
 234                 *a = *b;
 235         } else {
 236                 a->unknown.family = AF_UNKNOWN;
 237                 a->unknown.address = xstrdup(b->unknown.address);
 238                 a->unknown.port = xstrdup(b->unknown.port);
 239         }
 240 }
 241
 242 void sockaddrfree(sockaddr_t *a) {
 243         if(a->sa.sa_family == AF_UNKNOWN) {
 244                 free(a->unknown.address);
 245                 free(a->unknown.port);
 246         }
 247 }
 248
 249 void sockaddrunmap(sockaddr_t *sa) {
 250         if(sa->sa.sa_family == AF_INET6 && IN6_IS_ADDR_V4MAPPED(&sa->in6.sin6_addr)) {
 251                 sa->in.sin_addr.s_addr = ((uint32_t *) & sa->in6.sin6_addr)[3];
 252                 sa->in.sin_family = AF_INET;
 253         }
 254 }
 255
 256 void sockaddr_setport(sockaddr_t *sa, const char *port) {
 257         uint16_t portnum = htons(atoi(port));
 258
 259         if(!portnum) {
 260                 return;
 261         }
 262
 263         switch(sa->sa.sa_family) {
 264         case AF_INET:
 265                 sa->in.sin_port = portnum;
 266                 break;
 267
 268         case AF_INET6:
 269                 sa->in6.sin6_port = portnum;
 270                 break;
 271
 272         case AF_UNKNOWN:
 273                 free(sa->unknown.port);
 274                 sa->unknown.port = xstrdup(port);
 275
 276         default:
 277                 return;
 278         }
 279 }