tinc-vpn.org Git - tinc/blob - src/netutl.c

   1 /*
   2     netutl.c -- some supporting network utility code
   3     Copyright (C) 1998-2005 Ivo Timmermans
   4                   2000-2016 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 = NULL, hint = {0};
  37         int err;
  38
  39         hint.ai_family = addressfamily;
  40         hint.ai_socktype = socktype;
  41
  42 #if HAVE_DECL_RES_INIT
  43         // ensure glibc reloads /etc/resolv.conf.
  44         res_init();
  45 #endif
  46         err = getaddrinfo(address, service, &hint, &ai);
  47
  48         if(err) {
  49                 logger(LOG_WARNING, "Error looking up %s port %s: %s", address,
  50                        service, gai_strerror(err));
  51                 return NULL;
  52         }
  53
  54         return ai;
  55 }
  56
  57 sockaddr_t str2sockaddr(const char *address, const char *port) {
  58         struct addrinfo *ai = NULL, hint = {0};
  59         sockaddr_t result;
  60         int err;
  61
  62         hint.ai_family = AF_UNSPEC;
  63         hint.ai_flags = AI_NUMERICHOST;
  64         hint.ai_socktype = SOCK_STREAM;
  65
  66         err = getaddrinfo(address, port, &hint, &ai);
  67
  68         if(err || !ai) {
  69                 ifdebug(SCARY_THINGS)
  70                 logger(LOG_DEBUG, "Unknown type address %s port %s", address, port);
  71                 result.sa.sa_family = AF_UNKNOWN;
  72                 result.unknown.address = xstrdup(address);
  73                 result.unknown.port = xstrdup(port);
  74                 return result;
  75         }
  76
  77         memcpy(&result, ai->ai_addr, ai->ai_addrlen);
  78         freeaddrinfo(ai);
  79
  80         return result;
  81 }
  82
  83 void sockaddr2str(const sockaddr_t *sa, char **addrstr, char **portstr) {
  84         char address[NI_MAXHOST];
  85         char port[NI_MAXSERV];
  86         char *scopeid;
  87         int err;
  88
  89         if(sa->sa.sa_family == AF_UNKNOWN) {
  90                 if(addrstr) {
  91                         *addrstr = xstrdup(sa->unknown.address);
  92                 }
  93
  94                 if(portstr) {
  95                         *portstr = xstrdup(sa->unknown.port);
  96                 }
  97
  98                 return;
  99         }
 100
 101         err = getnameinfo(&sa->sa, SALEN(sa->sa), address, sizeof(address), port, sizeof(port), NI_NUMERICHOST | NI_NUMERICSERV);
 102
 103         if(err) {
 104                 logger(LOG_ERR, "Error while translating addresses: %s",
 105                        gai_strerror(err));
 106                 abort();
 107         }
 108
 109         scopeid = strchr(address, '%');
 110
 111         if(scopeid) {
 112                 *scopeid = '\0';        /* Descope. */
 113         }
 114
 115         if(addrstr) {
 116                 *addrstr = xstrdup(address);
 117         }
 118
 119         if(portstr) {
 120                 *portstr = xstrdup(port);
 121         }
 122 }
 123
 124 char *sockaddr2hostname(const sockaddr_t *sa) {
 125         char *str;
 126         char address[NI_MAXHOST] = "unknown";
 127         char port[NI_MAXSERV] = "unknown";
 128         int err;
 129
 130         if(sa->sa.sa_family == AF_UNKNOWN) {
 131                 xasprintf(&str, "%s port %s", sa->unknown.address, sa->unknown.port);
 132                 return str;
 133         }
 134
 135         err = getnameinfo(&sa->sa, SALEN(sa->sa), address, sizeof(address), port, sizeof(port),
 136                           hostnames ? 0 : (NI_NUMERICHOST | NI_NUMERICSERV));
 137
 138         if(err) {
 139                 logger(LOG_ERR, "Error while looking up hostname: %s",
 140                        gai_strerror(err));
 141         }
 142
 143         xasprintf(&str, "%s port %s", address, port);
 144
 145         return str;
 146 }
 147
 148 int sockaddrcmp_noport(const sockaddr_t *a, const sockaddr_t *b) {
 149         int result;
 150
 151         result = a->sa.sa_family - b->sa.sa_family;
 152
 153         if(result) {
 154                 return result;
 155         }
 156
 157         switch(a->sa.sa_family) {
 158         case AF_UNSPEC:
 159                 return 0;
 160
 161         case AF_UNKNOWN:
 162                 return strcmp(a->unknown.address, b->unknown.address);
 163
 164         case AF_INET:
 165                 return memcmp(&a->in.sin_addr, &b->in.sin_addr, sizeof(a->in.sin_addr));
 166
 167         case AF_INET6:
 168                 return memcmp(&a->in6.sin6_addr, &b->in6.sin6_addr, sizeof(a->in6.sin6_addr));
 169
 170         default:
 171                 logger(LOG_ERR, "sockaddrcmp() was called with unknown address family %d, exitting!",
 172                        a->sa.sa_family);
 173                 abort();
 174         }
 175 }
 176
 177 int sockaddrcmp(const sockaddr_t *a, const sockaddr_t *b) {
 178         int result;
 179
 180         result = a->sa.sa_family - b->sa.sa_family;
 181
 182         if(result) {
 183                 return result;
 184         }
 185
 186         switch(a->sa.sa_family) {
 187         case AF_UNSPEC:
 188                 return 0;
 189
 190         case AF_UNKNOWN:
 191                 result = strcmp(a->unknown.address, b->unknown.address);
 192
 193                 if(result) {
 194                         return result;
 195                 }
 196
 197                 return strcmp(a->unknown.port, b->unknown.port);
 198
 199         case AF_INET:
 200                 result = memcmp(&a->in.sin_addr, &b->in.sin_addr, sizeof(a->in.sin_addr));
 201
 202                 if(result) {
 203                         return result;
 204                 }
 205
 206                 return memcmp(&a->in.sin_port, &b->in.sin_port, sizeof(a->in.sin_port));
 207
 208         case AF_INET6:
 209                 result = memcmp(&a->in6.sin6_addr, &b->in6.sin6_addr, sizeof(a->in6.sin6_addr));
 210
 211                 if(result) {
 212                         return result;
 213                 }
 214
 215                 return memcmp(&a->in6.sin6_port, &b->in6.sin6_port, sizeof(a->in6.sin6_port));
 216
 217         default:
 218                 logger(LOG_ERR, "sockaddrcmp() was called with unknown address family %d, exitting!",
 219                        a->sa.sa_family);
 220                 abort();
 221         }
 222 }
 223
 224 void sockaddrcpy(sockaddr_t *a, const sockaddr_t *b) {
 225         if(b->sa.sa_family != AF_UNKNOWN) {
 226                 *a = *b;
 227         } else {
 228                 a->unknown.family = AF_UNKNOWN;
 229                 a->unknown.address = xstrdup(b->unknown.address);
 230                 a->unknown.port = xstrdup(b->unknown.port);
 231         }
 232 }
 233
 234 void sockaddrfree(sockaddr_t *a) {
 235         if(a->sa.sa_family == AF_UNKNOWN) {
 236                 free(a->unknown.address);
 237                 free(a->unknown.port);
 238         }
 239 }
 240
 241 void sockaddrunmap(sockaddr_t *sa) {
 242         if(sa->sa.sa_family == AF_INET6 && IN6_IS_ADDR_V4MAPPED(&sa->in6.sin6_addr)) {
 243                 sa->in.sin_addr.s_addr = ((uint32_t *) & sa->in6.sin6_addr)[3];
 244                 sa->in.sin_family = AF_INET;
 245         }
 246 }
 247
 248 void sockaddr_setport(sockaddr_t *sa, const char *port) {
 249         uint16_t portnum = htons(atoi(port));
 250
 251         if(!portnum) {
 252                 return;
 253         }
 254
 255         switch(sa->sa.sa_family) {
 256         case AF_INET:
 257                 sa->in.sin_port = portnum;
 258                 break;
 259
 260         case AF_INET6:
 261                 sa->in6.sin6_port = portnum;
 262                 break;
 263
 264         case AF_UNKNOWN:
 265                 free(sa->unknown.port);
 266                 sa->unknown.port = xstrdup(port);
 267
 268         default:
 269                 return;
 270         }
 271 }
 272
 273 /* Subnet mask handling */
 274
 275 int maskcmp(const void *va, const void *vb, int masklen) {
 276         int i, m, result;
 277         const char *a = va;
 278         const char *b = vb;
 279
 280         for(m = masklen, i = 0; m >= 8; m -= 8, i++) {
 281                 result = a[i] - b[i];
 282
 283                 if(result) {
 284                         return result;
 285                 }
 286         }
 287
 288         if(m)
 289                 return (a[i] & (0x100 - (1 << (8 - m)))) -
 290                        (b[i] & (0x100 - (1 << (8 - m))));
 291
 292         return 0;
 293 }
 294
 295 void mask(void *va, int masklen, int len) {
 296         int i;
 297         char *a = va;
 298
 299         i = masklen / 8;
 300         masklen %= 8;
 301
 302         if(masklen) {
 303                 a[i++] &= (0x100 - (1 << (8 - masklen)));
 304         }
 305
 306         for(; i < len; i++) {
 307                 a[i] = 0;
 308         }
 309 }
 310
 311 void maskcpy(void *va, const void *vb, int masklen, int len) {
 312         int i, m;
 313         char *a = va;
 314         const char *b = vb;
 315
 316         for(m = masklen, i = 0; m >= 8; m -= 8, i++) {
 317                 a[i] = b[i];
 318         }
 319
 320         if(m) {
 321                 a[i] = b[i] & (0x100 - (1 << (8 - m)));
 322                 i++;
 323         }
 324
 325         for(; i < len; i++) {
 326                 a[i] = 0;
 327         }
 328 }
 329
 330 bool maskcheck(const void *va, int masklen, int len) {
 331         int i;
 332         const char *a = va;
 333
 334         i = masklen / 8;
 335         masklen %= 8;
 336
 337         if(masklen && a[i++] & (0xff >> masklen)) {
 338                 return false;
 339         }
 340
 341         for(; i < len; i++)
 342                 if(a[i] != 0) {
 343                         return false;
 344                 }
 345
 346         return true;
 347 }