2 netutl.c -- some supporting network utility code
3 Copyright (C) 2014-2017 Guus Sliepen <guus@meshlink.io>
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.
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.
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.
22 #include "meshlink_internal.h"
30 Turn a string into a struct addrinfo.
31 Return NULL on failure.
33 struct addrinfo *str2addrinfo(const char *address, const char *service, int socktype) {
37 struct addrinfo hint = {
38 .ai_family = AF_UNSPEC,
39 .ai_socktype = socktype,
42 err = getaddrinfo(address, service, &hint, &ai);
45 logger(NULL, MESHLINK_WARNING, "Error looking up %s port %s: %s", address, service, strerror(errno));
52 sockaddr_t str2sockaddr(const char *address, const char *port) {
57 memset(&result, 0, sizeof(result));
59 struct addrinfo hint = {
60 .ai_family = AF_UNSPEC,
61 .ai_flags = AI_NUMERICHOST | AI_NUMERICSERV,
62 .ai_socktype = SOCK_STREAM,
65 err = getaddrinfo(address, port, &hint, &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);
75 memcpy(&result, ai->ai_addr, ai->ai_addrlen);
81 sockaddr_t str2sockaddr_random(struct meshlink_handle *mesh, const char *address, const char *port) {
86 memset(&result, 0, sizeof(result));
88 struct addrinfo hint = {
89 .ai_family = AF_UNSPEC,
90 .ai_flags = AI_NUMERICHOST | AI_NUMERICSERV,
91 .ai_socktype = SOCK_STREAM,
94 err = getaddrinfo(address, port, &hint, &ai);
97 result.sa.sa_family = AF_UNKNOWN;
98 result.unknown.address = NULL;
99 result.unknown.port = NULL;
105 for(struct addrinfo *aip = ai; aip; aip = aip->ai_next) {
109 struct addrinfo *aip = ai;
111 for(count = prng(mesh, count); count--; aip = aip->ai_next);
113 memcpy(&result, aip->ai_addr, aip->ai_addrlen);
119 void sockaddr2str(const sockaddr_t *sa, char **addrstr, char **portstr) {
124 if(sa->sa.sa_family == AF_UNKNOWN) {
126 *addrstr = xstrdup(sa->unknown.address);
130 *portstr = xstrdup(sa->unknown.port);
136 switch(sa->sa.sa_family) {
138 snprintf(port, sizeof port, "%d", sa->in.sin_port);
139 success = inet_ntop(AF_INET, &sa->in.sin_addr, address, sizeof(address));
143 snprintf(port, sizeof port, "%d", sa->in6.sin6_port);
144 success = inet_ntop(AF_INET6, &sa->in6.sin6_addr, address, sizeof(address));
153 logger(NULL, MESHLINK_ERROR, "Error while translating addresses: %s", strerror(errno));
158 *addrstr = xstrdup(address);
162 *portstr = xstrdup(port);
166 char *sockaddr2hostname(const sockaddr_t *sa) {
168 char address[48] = "unknown";
169 char port[16] = "unknown";
172 if(sa->sa.sa_family == AF_UNKNOWN) {
173 xasprintf(&str, "%s port %s", sa->unknown.address, sa->unknown.port);
177 switch(sa->sa.sa_family) {
179 snprintf(port, sizeof port, "%d", sa->in.sin_port);
180 success = inet_ntop(AF_INET, &sa->in.sin_addr, address, sizeof(address));
184 snprintf(port, sizeof port, "%d", sa->in6.sin6_port);
185 success = inet_ntop(AF_INET6, &sa->in6.sin6_addr, address, sizeof(address));
194 logger(NULL, MESHLINK_ERROR, "Error while resolving address to hostname: %s", strerror(errno));
198 xasprintf(&str, "%s port %s", address, port);
203 int sockaddrcmp_noport(const sockaddr_t *a, const sockaddr_t *b) {
206 result = a->sa.sa_family - b->sa.sa_family;
212 switch(a->sa.sa_family) {
217 return strcmp(a->unknown.address, b->unknown.address);
220 return memcmp(&a->in.sin_addr, &b->in.sin_addr, sizeof(a->in.sin_addr));
223 return memcmp(&a->in6.sin6_addr, &b->in6.sin6_addr, sizeof(a->in6.sin6_addr));
226 logger(NULL, MESHLINK_ERROR, "sockaddrcmp() was called with unknown address family %d, exitting!",
232 int sockaddrcmp(const sockaddr_t *a, const sockaddr_t *b) {
235 result = a->sa.sa_family - b->sa.sa_family;
241 switch(a->sa.sa_family) {
246 result = strcmp(a->unknown.address, b->unknown.address);
252 return strcmp(a->unknown.port, b->unknown.port);
255 result = memcmp(&a->in.sin_addr, &b->in.sin_addr, sizeof(a)->in.sin_addr);
261 return memcmp(&a->in.sin_port, &b->in.sin_port, sizeof(a)->in.sin_port);
264 result = memcmp(&a->in6.sin6_addr, &b->in6.sin6_addr, sizeof(a)->in6.sin6_addr);
270 return memcmp(&a->in6.sin6_port, &b->in6.sin6_port, sizeof(a)->in6.sin6_port);
273 logger(NULL, MESHLINK_ERROR, "sockaddrcmp() was called with unknown address family %d, exitting!",
279 void sockaddrcpy(sockaddr_t *a, const sockaddr_t *b) {
280 if(b->sa.sa_family != AF_UNKNOWN) {
283 a->unknown.family = AF_UNKNOWN;
284 a->unknown.address = xstrdup(b->unknown.address);
285 a->unknown.port = xstrdup(b->unknown.port);
289 void sockaddrcpy_setport(sockaddr_t *a, const sockaddr_t *b, uint16_t port) {
292 switch(b->sa.sa_family) {
294 a->in.sin_port = htons(port);
298 a->in6.sin6_port = htons(port);
306 void sockaddrfree(sockaddr_t *a) {
307 if(a->sa.sa_family == AF_UNKNOWN) {
308 free(a->unknown.address);
309 free(a->unknown.port);
313 void packmsg_add_sockaddr(packmsg_output_t *out, const sockaddr_t *sa) {
314 switch(sa->sa.sa_family) {
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));
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));
332 packmsg_output_invalidate(out);
337 sockaddr_t packmsg_get_sockaddr(packmsg_input_t *in) {
339 memset(&sa, 0, sizeof sa);
343 uint32_t len = packmsg_get_ext_raw(in, &type, &data);
348 packmsg_input_invalidate(in);
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);
359 packmsg_input_invalidate(in);
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);
369 packmsg_input_invalidate(in);