Add support for proxying through an external command.
[tinc] / src / netutl.c
1 /*
2     netutl.c -- some supporting network utility code
3     Copyright (C) 1998-2005 Ivo Timmermans
4                   2000-2011 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, hint = {0};
37         int err;
38
39         hint.ai_family = addressfamily;
40         hint.ai_socktype = socktype;
41
42         err = getaddrinfo(address, service, &hint, &ai);
43
44         if(err) {
45                 logger(LOG_WARNING, "Error looking up %s port %s: %s", address,
46                                    service, gai_strerror(err));
47                 return NULL;
48         }
49
50         return ai;
51 }
52
53 sockaddr_t str2sockaddr(const char *address, const char *port) {
54         struct addrinfo *ai, hint = {0};
55         sockaddr_t result;
56         int err;
57
58         hint.ai_family = AF_UNSPEC;
59         hint.ai_flags = AI_NUMERICHOST;
60         hint.ai_socktype = SOCK_STREAM;
61
62         err = getaddrinfo(address, port, &hint, &ai);
63
64         if(err || !ai) {
65                 ifdebug(SCARY_THINGS)
66                         logger(LOG_DEBUG, "Unknown type address %s port %s", address, port);
67                 result.sa.sa_family = AF_UNKNOWN;
68                 result.unknown.address = xstrdup(address);
69                 result.unknown.port = xstrdup(port);
70                 return result;
71         }
72
73         memcpy(&result, ai->ai_addr, ai->ai_addrlen);
74         freeaddrinfo(ai);
75
76         return result;
77 }
78
79 void sockaddr2str(const sockaddr_t *sa, char **addrstr, char **portstr) {
80         char address[NI_MAXHOST];
81         char port[NI_MAXSERV];
82         char *scopeid;
83         int err;
84
85         if(sa->sa.sa_family == AF_UNKNOWN) {
86                 if(addrstr)
87                         *addrstr = xstrdup(sa->unknown.address);
88                 if(portstr)
89                         *portstr = xstrdup(sa->unknown.port);
90                 return;
91         }
92
93         err = getnameinfo(&sa->sa, SALEN(sa->sa), address, sizeof(address), port, sizeof(port), NI_NUMERICHOST | NI_NUMERICSERV);
94
95         if(err) {
96                 logger(LOG_ERR, "Error while translating addresses: %s",
97                            gai_strerror(err));
98                 abort();
99         }
100
101         scopeid = strchr(address, '%');
102
103         if(scopeid)
104                 *scopeid = '\0';                /* Descope. */
105
106         if(addrstr)
107                 *addrstr = xstrdup(address);
108         if(portstr)
109                 *portstr = xstrdup(port);
110 }
111
112 char *sockaddr2hostname(const sockaddr_t *sa) {
113         char *str;
114         char address[NI_MAXHOST] = "unknown";
115         char port[NI_MAXSERV] = "unknown";
116         int err;
117
118         if(sa->sa.sa_family == AF_UNKNOWN) {
119                 xasprintf(&str, "%s port %s", sa->unknown.address, sa->unknown.port);
120                 return str;
121         }
122
123         err = getnameinfo(&sa->sa, SALEN(sa->sa), address, sizeof(address), port, sizeof(port),
124                                         hostnames ? 0 : (NI_NUMERICHOST | NI_NUMERICSERV));
125         if(err) {
126                 logger(LOG_ERR, "Error while looking up hostname: %s",
127                            gai_strerror(err));
128         }
129
130         xasprintf(&str, "%s port %s", address, port);
131
132         return str;
133 }
134
135 int sockaddrcmp_noport(const sockaddr_t *a, const sockaddr_t *b) {
136         int result;
137
138         result = a->sa.sa_family - b->sa.sa_family;
139
140         if(result)
141                 return result;
142
143         switch (a->sa.sa_family) {
144                 case AF_UNSPEC:
145                         return 0;
146
147                 case AF_UNKNOWN:
148                         return strcmp(a->unknown.address, b->unknown.address);
149
150                 case AF_INET:
151                         return memcmp(&a->in.sin_addr, &b->in.sin_addr, sizeof(a->in.sin_addr));
152
153                 case AF_INET6:
154                         return memcmp(&a->in6.sin6_addr, &b->in6.sin6_addr, sizeof(a->in6.sin6_addr));
155
156                 default:
157                         logger(LOG_ERR, "sockaddrcmp() was called with unknown address family %d, exitting!",
158                                    a->sa.sa_family);
159                         abort();
160         }
161 }
162
163 int sockaddrcmp(const sockaddr_t *a, const sockaddr_t *b) {
164         int result;
165
166         result = a->sa.sa_family - b->sa.sa_family;
167
168         if(result)
169                 return result;
170
171         switch (a->sa.sa_family) {
172                 case AF_UNSPEC:
173                         return 0;
174
175                 case AF_UNKNOWN:
176                         result = strcmp(a->unknown.address, b->unknown.address);
177
178                         if(result)
179                                 return result;
180
181                         return strcmp(a->unknown.port, b->unknown.port);
182
183                 case AF_INET:
184                         result = memcmp(&a->in.sin_addr, &b->in.sin_addr, sizeof(a->in.sin_addr));
185
186                         if(result)
187                                 return result;
188
189                         return memcmp(&a->in.sin_port, &b->in.sin_port, sizeof(a->in.sin_port));
190
191                 case AF_INET6:
192                         result = memcmp(&a->in6.sin6_addr, &b->in6.sin6_addr, sizeof(a->in6.sin6_addr));
193
194                         if(result)
195                                 return result;
196
197                         return memcmp(&a->in6.sin6_port, &b->in6.sin6_port, sizeof(a->in6.sin6_port));
198
199                 default:
200                         logger(LOG_ERR, "sockaddrcmp() was called with unknown address family %d, exitting!",
201                                    a->sa.sa_family);
202                         abort();
203         }
204 }
205
206 void sockaddrcpy(sockaddr_t *a, const sockaddr_t *b) {
207         if(b->sa.sa_family != AF_UNKNOWN) {
208                 *a = *b;
209         } else {
210                 a->unknown.family = AF_UNKNOWN;
211                 a->unknown.address = xstrdup(b->unknown.address);
212                 a->unknown.port = xstrdup(b->unknown.port);
213         }
214 }
215
216 void sockaddrfree(sockaddr_t *a) {
217         if(a->sa.sa_family == AF_UNKNOWN) {
218                 free(a->unknown.address);
219                 free(a->unknown.port);
220         }
221 }
222         
223 void sockaddrunmap(sockaddr_t *sa) {
224         if(sa->sa.sa_family == AF_INET6 && IN6_IS_ADDR_V4MAPPED(&sa->in6.sin6_addr)) {
225                 sa->in.sin_addr.s_addr = ((uint32_t *) & sa->in6.sin6_addr)[3];
226                 sa->in.sin_family = AF_INET;
227         }
228 }
229
230 /* Subnet mask handling */
231
232 int maskcmp(const void *va, const void *vb, int masklen) {
233         int i, m, result;
234         const char *a = va;
235         const char *b = vb;
236
237         for(m = masklen, i = 0; m >= 8; m -= 8, i++) {
238                 result = a[i] - b[i];
239                 if(result)
240                         return result;
241         }
242
243         if(m)
244                 return (a[i] & (0x100 - (1 << (8 - m)))) -
245                         (b[i] & (0x100 - (1 << (8 - m))));
246
247         return 0;
248 }
249
250 void mask(void *va, int masklen, int len) {
251         int i;
252         char *a = va;
253
254         i = masklen / 8;
255         masklen %= 8;
256
257         if(masklen)
258                 a[i++] &= (0x100 - (1 << (8 - masklen)));
259
260         for(; i < len; i++)
261                 a[i] = 0;
262 }
263
264 void maskcpy(void *va, const void *vb, int masklen, int len) {
265         int i, m;
266         char *a = va;
267         const char *b = vb;
268
269         for(m = masklen, i = 0; m >= 8; m -= 8, i++)
270                 a[i] = b[i];
271
272         if(m) {
273                 a[i] = b[i] & (0x100 - (1 << (8 - m)));
274                 i++;
275         }
276
277         for(; i < len; i++)
278                 a[i] = 0;
279 }
280
281 bool maskcheck(const void *va, int masklen, int len) {
282         int i;
283         const char *a = va;
284
285         i = masklen / 8;
286         masklen %= 8;
287
288         if(masklen && a[i++] & (0xff >> masklen))
289                 return false;
290
291         for(; i < len; i++)
292                 if(a[i] != 0)
293                         return false;
294
295         return true;
296 }