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