]> git.meshlink.io Git - meshlink/blob - src/netutl.c
Add a test for changing ports and handling of port conflicts.
[meshlink] / 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, err == EAI_SYSTEM ? strerror(errno) : gai_strerror(err));
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[NI_MAXHOST];
121         char port[NI_MAXSERV];
122         char *scopeid;
123         int err;
124
125         if(sa->sa.sa_family == AF_UNKNOWN) {
126                 if(addrstr) {
127                         *addrstr = xstrdup(sa->unknown.address);
128                 }
129
130                 if(portstr) {
131                         *portstr = xstrdup(sa->unknown.port);
132                 }
133
134                 return;
135         }
136
137         err = getnameinfo(&sa->sa, SALEN(sa->sa), address, sizeof(address), port, sizeof(port), NI_NUMERICHOST | NI_NUMERICSERV);
138
139         if(err) {
140                 logger(NULL, MESHLINK_ERROR, "Error while translating addresses: %s", err == EAI_SYSTEM ? strerror(errno) : gai_strerror(err));
141                 abort();
142         }
143
144         scopeid = strchr(address, '%');
145
146         if(scopeid) {
147                 *scopeid = '\0';        /* Descope. */
148         }
149
150         if(addrstr) {
151                 *addrstr = xstrdup(address);
152         }
153
154         if(portstr) {
155                 *portstr = xstrdup(port);
156         }
157 }
158
159 char *sockaddr2hostname(const sockaddr_t *sa) {
160         char *str;
161         char address[NI_MAXHOST] = "unknown";
162         char port[NI_MAXSERV] = "unknown";
163         int err;
164
165         if(sa->sa.sa_family == AF_UNKNOWN) {
166                 xasprintf(&str, "%s port %s", sa->unknown.address, sa->unknown.port);
167                 return str;
168         }
169
170         err = getnameinfo(&sa->sa, SALEN(sa->sa), address, sizeof(address), port, sizeof(port), NI_NUMERICHOST | NI_NUMERICSERV);
171
172         if(err) {
173                 logger(NULL, MESHLINK_ERROR, "Error while looking up hostname: %s", err == EAI_SYSTEM ? strerror(errno) : gai_strerror(err));
174                 abort();
175         }
176
177         xasprintf(&str, "%s port %s", address, port);
178
179         return str;
180 }
181
182 int sockaddrcmp_noport(const sockaddr_t *a, const sockaddr_t *b) {
183         int result;
184
185         result = a->sa.sa_family - b->sa.sa_family;
186
187         if(result) {
188                 return result;
189         }
190
191         switch(a->sa.sa_family) {
192         case AF_UNSPEC:
193                 return 0;
194
195         case AF_UNKNOWN:
196                 return strcmp(a->unknown.address, b->unknown.address);
197
198         case AF_INET:
199                 return memcmp(&a->in.sin_addr, &b->in.sin_addr, sizeof(a->in.sin_addr));
200
201         case AF_INET6:
202                 return memcmp(&a->in6.sin6_addr, &b->in6.sin6_addr, sizeof(a->in6.sin6_addr));
203
204         default:
205                 logger(NULL, MESHLINK_ERROR, "sockaddrcmp() was called with unknown address family %d, exitting!",
206                        a->sa.sa_family);
207                 abort();
208         }
209 }
210
211 int sockaddrcmp(const sockaddr_t *a, const sockaddr_t *b) {
212         int result;
213
214         result = a->sa.sa_family - b->sa.sa_family;
215
216         if(result) {
217                 return result;
218         }
219
220         switch(a->sa.sa_family) {
221         case AF_UNSPEC:
222                 return 0;
223
224         case AF_UNKNOWN:
225                 result = strcmp(a->unknown.address, b->unknown.address);
226
227                 if(result) {
228                         return result;
229                 }
230
231                 return strcmp(a->unknown.port, b->unknown.port);
232
233         case AF_INET:
234                 result = memcmp(&a->in.sin_addr, &b->in.sin_addr, sizeof(a)->in.sin_addr);
235
236                 if(result) {
237                         return result;
238                 }
239
240                 return memcmp(&a->in.sin_port, &b->in.sin_port, sizeof(a)->in.sin_port);
241
242         case AF_INET6:
243                 result = memcmp(&a->in6.sin6_addr, &b->in6.sin6_addr, sizeof(a)->in6.sin6_addr);
244
245                 if(result) {
246                         return result;
247                 }
248
249                 return memcmp(&a->in6.sin6_port, &b->in6.sin6_port, sizeof(a)->in6.sin6_port);
250
251         default:
252                 logger(NULL, MESHLINK_ERROR, "sockaddrcmp() was called with unknown address family %d, exitting!",
253                        a->sa.sa_family);
254                 abort();
255         }
256 }
257
258 void sockaddrcpy(sockaddr_t *a, const sockaddr_t *b) {
259         if(b->sa.sa_family != AF_UNKNOWN) {
260                 *a = *b;
261         } else {
262                 a->unknown.family = AF_UNKNOWN;
263                 a->unknown.address = xstrdup(b->unknown.address);
264                 a->unknown.port = xstrdup(b->unknown.port);
265         }
266 }
267
268 void sockaddrcpy_setport(sockaddr_t *a, const sockaddr_t *b, uint16_t port) {
269         sockaddrcpy(a, b);
270
271         switch(b->sa.sa_family) {
272         case AF_INET:
273                 a->in.sin_port = htons(port);
274                 break;
275
276         case AF_INET6:
277                 a->in6.sin6_port = htons(port);
278                 break;
279
280         default:
281                 break;
282         }
283 }
284
285 void sockaddrfree(sockaddr_t *a) {
286         if(a->sa.sa_family == AF_UNKNOWN) {
287                 free(a->unknown.address);
288                 free(a->unknown.port);
289         }
290 }
291
292 void sockaddrunmap(sockaddr_t *sa) {
293         if(sa->sa.sa_family == AF_INET6 && IN6_IS_ADDR_V4MAPPED(&sa->in6.sin6_addr)) {
294                 sa->in.sin_addr.s_addr = ((uint32_t *) & sa->in6.sin6_addr)[3];
295                 sa->in.sin_family = AF_INET;
296         }
297 }
298
299 void packmsg_add_sockaddr(packmsg_output_t *out, const sockaddr_t *sa) {
300         switch(sa->sa.sa_family) {
301         case AF_INET: {
302                 uint8_t buf[6];
303                 memcpy(buf + 0, &sa->in.sin_port, 2);
304                 memcpy(buf + 2, &sa->in.sin_addr, 4);
305                 packmsg_add_ext(out, 4, buf, sizeof(buf));
306                 break;
307         }
308
309         case AF_INET6: {
310                 uint8_t buf[18];
311                 memcpy(buf + 0, &sa->in6.sin6_port, 2);
312                 memcpy(buf + 2, &sa->in6.sin6_addr, 16);
313                 packmsg_add_ext(out, 6, buf, sizeof(buf));
314                 break;
315         }
316
317         default:
318                 packmsg_output_invalidate(out);
319                 break;
320         }
321 }
322
323 sockaddr_t packmsg_get_sockaddr(packmsg_input_t *in) {
324         sockaddr_t sa;
325         memset(&sa, 0, sizeof sa);
326
327         int8_t type;
328         const void *data;
329         uint32_t len = packmsg_get_ext_raw(in, &type, &data);
330
331         switch(type) {
332         case 4:
333                 if(len != 6) {
334                         packmsg_input_invalidate(in);
335                         return sa;
336                 }
337
338                 sa.sa.sa_family = AF_INET;
339                 memcpy(&sa.in.sin_port, (uint8_t *)data + 0, 2);
340                 memcpy(&sa.in.sin_addr, (uint8_t *)data + 2, 4);
341                 break;
342
343         case 6:
344                 if(len != 18) {
345                         packmsg_input_invalidate(in);
346                         return sa;
347                 }
348
349                 sa.sa.sa_family = AF_INET6;
350                 memcpy(&sa.in6.sin6_port, (uint8_t *)data + 0, 2);
351                 memcpy(&sa.in6.sin6_addr, (uint8_t *)data + 2, 16);
352                 break;
353
354         default:
355                 packmsg_input_invalidate(in);
356                 return sa;
357         }
358
359         return sa;
360 }