git.meshlink.io Git - meshlink/blob - src/netutl.c

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