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

   1 /*
   2     netutl.c -- some supporting network utility code
   3     Copyright (C) 2014 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, hint = {0};
  36         int err;
  37
  38         hint.ai_family = addressfamily;
  39         hint.ai_socktype = socktype;
  40
  41         err = getaddrinfo(address, service, &hint, &ai);
  42
  43         if(err) {
  44                 logger(DEBUG_ALWAYS, LOG_WARNING, "Error looking up %s port %s: %s", address, service, err == EAI_SYSTEM ? strerror(errno) : gai_strerror(err));
  45                 return NULL;
  46         }
  47
  48         return ai;
  49 }
  50
  51 sockaddr_t str2sockaddr(const char *address, const char *port) {
  52         struct addrinfo *ai, hint = {0};
  53         sockaddr_t result = {{0}};
  54         int err;
  55
  56         hint.ai_family = AF_UNSPEC;
  57         hint.ai_flags = AI_NUMERICHOST;
  58         hint.ai_socktype = SOCK_STREAM;
  59
  60         err = getaddrinfo(address, port, &hint, &ai);
  61
  62         if(err || !ai) {
  63                 logger(DEBUG_SCARY_THINGS, LOG_DEBUG, "Unknown type address %s port %s", address, port);
  64                 result.sa.sa_family = AF_UNKNOWN;
  65                 result.unknown.address = xstrdup(address);
  66                 result.unknown.port = xstrdup(port);
  67                 return result;
  68         }
  69
  70         memcpy(&result, ai->ai_addr, ai->ai_addrlen);
  71         freeaddrinfo(ai);
  72
  73         return result;
  74 }
  75
  76 void sockaddr2str(const sockaddr_t *sa, char **addrstr, char **portstr) {
  77         char address[NI_MAXHOST];
  78         char port[NI_MAXSERV];
  79         char *scopeid;
  80         int err;
  81
  82         if(sa->sa.sa_family == AF_UNKNOWN) {
  83                 if(addrstr)
  84                         *addrstr = xstrdup(sa->unknown.address);
  85                 if(portstr)
  86                         *portstr = xstrdup(sa->unknown.port);
  87                 return;
  88         }
  89
  90         err = getnameinfo(&sa->sa, SALEN(sa->sa), address, sizeof address, port, sizeof port, NI_NUMERICHOST | NI_NUMERICSERV);
  91
  92         if(err) {
  93                 logger(DEBUG_ALWAYS, LOG_ERR, "Error while translating addresses: %s", err == EAI_SYSTEM ? strerror(errno) : gai_strerror(err));
  94                 abort();
  95         }
  96
  97         scopeid = strchr(address, '%');
  98
  99         if(scopeid)
 100                 *scopeid = '\0'; /* Descope. */
 101
 102         if(addrstr)
 103                 *addrstr = xstrdup(address);
 104         if(portstr)
 105                 *portstr = xstrdup(port);
 106 }
 107
 108 char *sockaddr2hostname(const sockaddr_t *sa) {
 109         char *str;
 110         char address[NI_MAXHOST] = "unknown";
 111         char port[NI_MAXSERV] = "unknown";
 112         int err;
 113
 114         if(sa->sa.sa_family == AF_UNKNOWN) {
 115                 xasprintf(&str, "%s port %s", sa->unknown.address, sa->unknown.port);
 116                 return str;
 117         }
 118
 119         err = getnameinfo(&sa->sa, SALEN(sa->sa), address, sizeof address, port, sizeof port,
 120                                         hostnames ? 0 : (NI_NUMERICHOST | NI_NUMERICSERV));
 121         if(err) {
 122                 logger(DEBUG_ALWAYS, LOG_ERR, "Error while looking up hostname: %s", err == EAI_SYSTEM ? strerror(errno) : gai_strerror(err));
 123         }
 124
 125         xasprintf(&str, "%s port %s", address, port);
 126
 127         return str;
 128 }
 129
 130 int sockaddrcmp_noport(const sockaddr_t *a, const sockaddr_t *b) {
 131         int result;
 132
 133         result = a->sa.sa_family - b->sa.sa_family;
 134
 135         if(result)
 136                 return result;
 137
 138         switch (a->sa.sa_family) {
 139                 case AF_UNSPEC:
 140                         return 0;
 141
 142                 case AF_UNKNOWN:
 143                         return strcmp(a->unknown.address, b->unknown.address);
 144
 145                 case AF_INET:
 146                         return memcmp(&a->in.sin_addr, &b->in.sin_addr, sizeof(a->in.sin_addr));
 147
 148                 case AF_INET6:
 149                         return memcmp(&a->in6.sin6_addr, &b->in6.sin6_addr, sizeof(a->in6.sin6_addr));
 150
 151                 default:
 152                         logger(DEBUG_ALWAYS, LOG_ERR, "sockaddrcmp() was called with unknown address family %d, exitting!",
 153                                    a->sa.sa_family);
 154                         abort();
 155         }
 156 }
 157
 158 int sockaddrcmp(const sockaddr_t *a, const sockaddr_t *b) {
 159         int result;
 160
 161         result = a->sa.sa_family - b->sa.sa_family;
 162
 163         if(result)
 164                 return result;
 165
 166         switch (a->sa.sa_family) {
 167                 case AF_UNSPEC:
 168                         return 0;
 169
 170                 case AF_UNKNOWN:
 171                         result = strcmp(a->unknown.address, b->unknown.address);
 172
 173                         if(result)
 174                                 return result;
 175
 176                         return strcmp(a->unknown.port, b->unknown.port);
 177
 178                 case AF_INET:
 179                         result = memcmp(&a->in.sin_addr, &b->in.sin_addr, sizeof a->in.sin_addr);
 180
 181                         if(result)
 182                                 return result;
 183
 184                         return memcmp(&a->in.sin_port, &b->in.sin_port, sizeof a->in.sin_port);
 185
 186                 case AF_INET6:
 187                         result = memcmp(&a->in6.sin6_addr, &b->in6.sin6_addr, sizeof a->in6.sin6_addr);
 188
 189                         if(result)
 190                                 return result;
 191
 192                         return memcmp(&a->in6.sin6_port, &b->in6.sin6_port, sizeof a->in6.sin6_port);
 193
 194                 default:
 195                         logger(DEBUG_ALWAYS, LOG_ERR, "sockaddrcmp() was called with unknown address family %d, exitting!",
 196                                    a->sa.sa_family);
 197                         abort();
 198         }
 199 }
 200
 201 void sockaddrcpy(sockaddr_t *a, const sockaddr_t *b) {
 202         if(b->sa.sa_family != AF_UNKNOWN) {
 203                 *a = *b;
 204         } else {
 205                 a->unknown.family = AF_UNKNOWN;
 206                 a->unknown.address = xstrdup(b->unknown.address);
 207                 a->unknown.port = xstrdup(b->unknown.port);
 208         }
 209 }
 210
 211 void sockaddrfree(sockaddr_t *a) {
 212         if(a->sa.sa_family == AF_UNKNOWN) {
 213                 free(a->unknown.address);
 214                 free(a->unknown.port);
 215         }
 216 }
 217
 218 void sockaddrunmap(sockaddr_t *sa) {
 219         if(sa->sa.sa_family == AF_INET6 && IN6_IS_ADDR_V4MAPPED(&sa->in6.sin6_addr)) {
 220                 sa->in.sin_addr.s_addr = ((uint32_t *) & sa->in6.sin6_addr)[3];
 221                 sa->in.sin_family = AF_INET;
 222         }
 223 }