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