git.meshlink.io Git - meshlink-tiny/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 "meshlink_internal.h"
  23 #include "net.h"
  24 #include "netutl.h"
  25 #include "logger.h"
  26 #include "utils.h"
  27 #include "xalloc.h"
  28
  29 /*
  30   Turn a string into a struct addrinfo.
  31   Return NULL on failure.
  32 */
  33 struct addrinfo *str2addrinfo(const char *address, const char *service, int socktype) {
  34         struct addrinfo *ai;
  35         int err;
  36
  37         struct addrinfo hint = {
  38                 .ai_family = AF_UNSPEC,
  39                 .ai_socktype = socktype,
  40         };
  41
  42         err = getaddrinfo(address, service, &hint, &ai);
  43
  44         if(err) {
  45                 logger(NULL, MESHLINK_WARNING, "Error looking up %s port %s: %s", address, service, strerror(errno));
  46                 return NULL;
  47         }
  48
  49         return ai;
  50 }
  51
  52 sockaddr_t str2sockaddr(const char *address, const char *port) {
  53         struct addrinfo *ai;
  54         sockaddr_t result;
  55         int err;
  56
  57         memset(&result, 0, sizeof(result));
  58
  59         struct addrinfo hint = {
  60                 .ai_family = AF_UNSPEC,
  61                 .ai_flags = AI_NUMERICHOST | AI_NUMERICSERV,
  62                 .ai_socktype = SOCK_STREAM,
  63         };
  64
  65         err = getaddrinfo(address, port, &hint, &ai);
  66
  67         if(err || !ai) {
  68                 logger(NULL, MESHLINK_DEBUG, "Unknown type address %s port %s", address, port);
  69                 result.sa.sa_family = AF_UNKNOWN;
  70                 result.unknown.address = xstrdup(address);
  71                 result.unknown.port = xstrdup(port);
  72                 return result;
  73         }
  74
  75         memcpy(&result, ai->ai_addr, ai->ai_addrlen);
  76         freeaddrinfo(ai);
  77
  78         return result;
  79 }
  80
  81 sockaddr_t str2sockaddr_random(struct meshlink_handle *mesh, const char *address, const char *port) {
  82         struct addrinfo *ai;
  83         sockaddr_t result;
  84         int err;
  85
  86         memset(&result, 0, sizeof(result));
  87
  88         struct addrinfo hint = {
  89                 .ai_family = AF_UNSPEC,
  90                 .ai_flags = AI_NUMERICHOST | AI_NUMERICSERV,
  91                 .ai_socktype = SOCK_STREAM,
  92         };
  93
  94         err = getaddrinfo(address, port, &hint, &ai);
  95
  96         if(err || !ai) {
  97                 result.sa.sa_family = AF_UNKNOWN;
  98                 result.unknown.address = NULL;
  99                 result.unknown.port = NULL;
 100                 return result;
 101         }
 102
 103         int count = 0;
 104
 105         for(struct addrinfo *aip = ai; aip; aip = aip->ai_next) {
 106                 count++;
 107         }
 108
 109         struct addrinfo *aip = ai;
 110
 111         for(count = prng(mesh, count); count--; aip = aip->ai_next);
 112
 113         memcpy(&result, aip->ai_addr, aip->ai_addrlen);
 114         freeaddrinfo(ai);
 115
 116         return result;
 117 }
 118
 119 void sockaddr2str(const sockaddr_t *sa, char **addrstr, char **portstr) {
 120         char address[48];
 121         char port[16];
 122         bool success;
 123
 124         if(sa->sa.sa_family == AF_UNKNOWN) {
 125                 if(addrstr) {
 126                         *addrstr = xstrdup(sa->unknown.address);
 127                 }
 128
 129                 if(portstr) {
 130                         *portstr = xstrdup(sa->unknown.port);
 131                 }
 132
 133                 return;
 134         }
 135
 136         switch(sa->sa.sa_family) {
 137         case AF_INET:
 138                 snprintf(port, sizeof port, "%d", sa->in.sin_port);
 139                 success = inet_ntop(AF_INET, &sa->in.sin_addr, address, sizeof(address));
 140                 break;
 141
 142         case AF_INET6:
 143                 snprintf(port, sizeof port, "%d", sa->in6.sin6_port);
 144                 success = inet_ntop(AF_INET6, &sa->in6.sin6_addr, address, sizeof(address));
 145                 break;
 146
 147         default:
 148                 success = false;
 149                 break;
 150         }
 151
 152         if(!success) {
 153                 logger(NULL, MESHLINK_ERROR, "Error while translating addresses: %s", strerror(errno));
 154                 abort();
 155         }
 156
 157         if(addrstr) {
 158                 *addrstr = xstrdup(address);
 159         }
 160
 161         if(portstr) {
 162                 *portstr = xstrdup(port);
 163         }
 164 }
 165
 166 char *sockaddr2hostname(const sockaddr_t *sa) {
 167         char *str;
 168         char address[48] = "unknown";
 169         char port[16] = "unknown";
 170         bool success;
 171
 172         if(sa->sa.sa_family == AF_UNKNOWN) {
 173                 xasprintf(&str, "%s port %s", sa->unknown.address, sa->unknown.port);
 174                 return str;
 175         }
 176
 177         switch(sa->sa.sa_family) {
 178         case AF_INET:
 179                 snprintf(port, sizeof port, "%d", sa->in.sin_port);
 180                 success = inet_ntop(AF_INET, &sa->in.sin_addr, address, sizeof(address));
 181                 break;
 182
 183         case AF_INET6:
 184                 snprintf(port, sizeof port, "%d", sa->in6.sin6_port);
 185                 success = inet_ntop(AF_INET6, &sa->in6.sin6_addr, address, sizeof(address));
 186                 break;
 187
 188         default:
 189                 success = false;
 190                 break;
 191         }
 192
 193         if(!success) {
 194                 logger(NULL, MESHLINK_ERROR, "Error while resolving address to hostname: %s", strerror(errno));
 195                 abort();
 196         }
 197
 198         xasprintf(&str, "%s port %s", address, port);
 199
 200         return str;
 201 }
 202
 203 int sockaddrcmp_noport(const sockaddr_t *a, const sockaddr_t *b) {
 204         int result;
 205
 206         result = a->sa.sa_family - b->sa.sa_family;
 207
 208         if(result) {
 209                 return result;
 210         }
 211
 212         switch(a->sa.sa_family) {
 213         case AF_UNSPEC:
 214                 return 0;
 215
 216         case AF_UNKNOWN:
 217                 return strcmp(a->unknown.address, b->unknown.address);
 218
 219         case AF_INET:
 220                 return memcmp(&a->in.sin_addr, &b->in.sin_addr, sizeof(a->in.sin_addr));
 221
 222         case AF_INET6:
 223                 return memcmp(&a->in6.sin6_addr, &b->in6.sin6_addr, sizeof(a->in6.sin6_addr));
 224
 225         default:
 226                 logger(NULL, MESHLINK_ERROR, "sockaddrcmp() was called with unknown address family %d, exitting!",
 227                        a->sa.sa_family);
 228                 abort();
 229         }
 230 }
 231
 232 int sockaddrcmp(const sockaddr_t *a, const sockaddr_t *b) {
 233         int result;
 234
 235         result = a->sa.sa_family - b->sa.sa_family;
 236
 237         if(result) {
 238                 return result;
 239         }
 240
 241         switch(a->sa.sa_family) {
 242         case AF_UNSPEC:
 243                 return 0;
 244
 245         case AF_UNKNOWN:
 246                 result = strcmp(a->unknown.address, b->unknown.address);
 247
 248                 if(result) {
 249                         return result;
 250                 }
 251
 252                 return strcmp(a->unknown.port, b->unknown.port);
 253
 254         case AF_INET:
 255                 result = memcmp(&a->in.sin_addr, &b->in.sin_addr, sizeof(a)->in.sin_addr);
 256
 257                 if(result) {
 258                         return result;
 259                 }
 260
 261                 return memcmp(&a->in.sin_port, &b->in.sin_port, sizeof(a)->in.sin_port);
 262
 263         case AF_INET6:
 264                 result = memcmp(&a->in6.sin6_addr, &b->in6.sin6_addr, sizeof(a)->in6.sin6_addr);
 265
 266                 if(result) {
 267                         return result;
 268                 }
 269
 270                 return memcmp(&a->in6.sin6_port, &b->in6.sin6_port, sizeof(a)->in6.sin6_port);
 271
 272         default:
 273                 logger(NULL, MESHLINK_ERROR, "sockaddrcmp() was called with unknown address family %d, exitting!",
 274                        a->sa.sa_family);
 275                 abort();
 276         }
 277 }
 278
 279 void sockaddrcpy(sockaddr_t *a, const sockaddr_t *b) {
 280         if(b->sa.sa_family != AF_UNKNOWN) {
 281                 *a = *b;
 282         } else {
 283                 a->unknown.family = AF_UNKNOWN;
 284                 a->unknown.address = xstrdup(b->unknown.address);
 285                 a->unknown.port = xstrdup(b->unknown.port);
 286         }
 287 }
 288
 289 void sockaddrcpy_setport(sockaddr_t *a, const sockaddr_t *b, uint16_t port) {
 290         sockaddrcpy(a, b);
 291
 292         switch(b->sa.sa_family) {
 293         case AF_INET:
 294                 a->in.sin_port = htons(port);
 295                 break;
 296
 297         case AF_INET6:
 298                 a->in6.sin6_port = htons(port);
 299                 break;
 300
 301         default:
 302                 break;
 303         }
 304 }
 305
 306 void sockaddrfree(sockaddr_t *a) {
 307         if(a->sa.sa_family == AF_UNKNOWN) {
 308                 free(a->unknown.address);
 309                 free(a->unknown.port);
 310         }
 311 }
 312
 313 void packmsg_add_sockaddr(packmsg_output_t *out, const sockaddr_t *sa) {
 314         switch(sa->sa.sa_family) {
 315         case AF_INET: {
 316                 uint8_t buf[6];
 317                 memcpy(buf + 0, &sa->in.sin_port, 2);
 318                 memcpy(buf + 2, &sa->in.sin_addr, 4);
 319                 packmsg_add_ext(out, 4, buf, sizeof(buf));
 320                 break;
 321         }
 322
 323         case AF_INET6: {
 324                 uint8_t buf[18];
 325                 memcpy(buf + 0, &sa->in6.sin6_port, 2);
 326                 memcpy(buf + 2, &sa->in6.sin6_addr, 16);
 327                 packmsg_add_ext(out, 6, buf, sizeof(buf));
 328                 break;
 329         }
 330
 331         default:
 332                 packmsg_output_invalidate(out);
 333                 break;
 334         }
 335 }
 336
 337 sockaddr_t packmsg_get_sockaddr(packmsg_input_t *in) {
 338         sockaddr_t sa;
 339         memset(&sa, 0, sizeof sa);
 340
 341         int8_t type;
 342         const void *data;
 343         uint32_t len = packmsg_get_ext_raw(in, &type, &data);
 344
 345         switch(type) {
 346         case 4:
 347                 if(len != 6) {
 348                         packmsg_input_invalidate(in);
 349                         return sa;
 350                 }
 351
 352                 sa.sa.sa_family = AF_INET;
 353                 memcpy(&sa.in.sin_port, (uint8_t *)data + 0, 2);
 354                 memcpy(&sa.in.sin_addr, (uint8_t *)data + 2, 4);
 355                 break;
 356
 357         case 6:
 358                 if(len != 18) {
 359                         packmsg_input_invalidate(in);
 360                         return sa;
 361                 }
 362
 363                 sa.sa.sa_family = AF_INET6;
 364                 memcpy(&sa.in6.sin6_port, (uint8_t *)data + 0, 2);
 365                 memcpy(&sa.in6.sin6_addr, (uint8_t *)data + 2, 16);
 366                 break;
 367
 368         default:
 369                 packmsg_input_invalidate(in);
 370                 return sa;
 371         }
 372
 373         return sa;
 374 }