1 /* Get address information (partial implementation).
2 Copyright (C) 1997, 2001-2002, 2004-2020 Free Software Foundation, Inc.
3 Contributed by Simon Josefsson <simon@josefsson.org>.
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 3, or (at your option)
8 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
16 along with this program; if not, see <https://www.gnu.org/licenses/>. */
17
18 /* Don't use __attribute__ __nonnull__ in this compilation unit. Otherwise gcc
19 optimizes away the sa == NULL test below. */
20 #define _GL_ARG_NONNULL(params)
21
22 #include <config.h>
23
24 #include <netdb.h>
25
26 #if HAVE_NETINET_IN_H
27 # include <netinet/in.h>
28 #endif
29
30 /* Get inet_ntop. */
31 #include <arpa/inet.h>
32
33 /* Get calloc. */
34 #include <stdlib.h>
35
36 /* Get memcpy, strdup. */
37 #include <string.h>
38
39 /* Get snprintf. */
40 #include <stdio.h>
41
42 #include <stdbool.h>
43
44 #include "gettext.h"
45 #define _(String) gettext (String)
46 #define N_(String) String
47
48 /* BeOS has AF_INET, but not PF_INET. */
49 #ifndef PF_INET
50 # define PF_INET AF_INET
51 #endif
52 /* BeOS also lacks PF_UNSPEC. */
53 #ifndef PF_UNSPEC
54 # define PF_UNSPEC 0
55 #endif
56
57 #if HAVE_GETADDRINFO
58
59 /* Override with cdecl calling convention. */
60
61 int
getaddrinfo(const char * restrict nodename,const char * restrict servname,const struct addrinfo * restrict hints,struct addrinfo ** restrict res)62 getaddrinfo (const char *restrict nodename,
63 const char *restrict servname,
64 const struct addrinfo *restrict hints,
65 struct addrinfo **restrict res)
66 # undef getaddrinfo
67 {
68 return getaddrinfo (nodename, servname, hints, res);
69 }
70
71 void
freeaddrinfo(struct addrinfo * ai)72 freeaddrinfo (struct addrinfo *ai)
73 # undef freeaddrinfo
74 {
75 freeaddrinfo (ai);
76 }
77
78 #else
79
80 # if defined _WIN32 && !defined __CYGWIN__
81 # define WINDOWS_NATIVE
82 # endif
83
84 /* gl_sockets_startup */
85 # include "sockets.h"
86
87 # ifdef WINDOWS_NATIVE
88
89 /* Avoid warnings from gcc -Wcast-function-type. */
90 # define GetProcAddress \
91 (void *) GetProcAddress
92
93 typedef int (WSAAPI *getaddrinfo_func) (const char*, const char*,
94 const struct addrinfo*,
95 struct addrinfo**);
96 typedef void (WSAAPI *freeaddrinfo_func) (struct addrinfo*);
97 typedef int (WSAAPI *getnameinfo_func) (const struct sockaddr*,
98 socklen_t, char*, DWORD,
99 char*, DWORD, int);
100
101 static getaddrinfo_func getaddrinfo_ptr = NULL;
102 static freeaddrinfo_func freeaddrinfo_ptr = NULL;
103 static getnameinfo_func getnameinfo_ptr = NULL;
104
105 static int
use_win32_p(void)106 use_win32_p (void)
107 {
108 static int done = 0;
109 HMODULE h;
110
111 if (done)
112 return getaddrinfo_ptr ? 1 : 0;
113
114 done = 1;
115
116 h = GetModuleHandle ("ws2_32.dll");
117
118 if (h)
119 {
120 getaddrinfo_ptr = (getaddrinfo_func) GetProcAddress (h, "getaddrinfo");
121 freeaddrinfo_ptr = (freeaddrinfo_func) GetProcAddress (h, "freeaddrinfo");
122 getnameinfo_ptr = (getnameinfo_func) GetProcAddress (h, "getnameinfo");
123 }
124
125 /* If either is missing, something is odd. */
126 if (!getaddrinfo_ptr || !freeaddrinfo_ptr || !getnameinfo_ptr)
127 {
128 getaddrinfo_ptr = NULL;
129 freeaddrinfo_ptr = NULL;
130 getnameinfo_ptr = NULL;
131 return 0;
132 }
133
134 gl_sockets_startup (SOCKETS_1_1);
135
136 return 1;
137 }
138 # endif
139
140 static bool
validate_family(int family)141 validate_family (int family)
142 {
143 /* FIXME: Support more families. */
144 # if HAVE_IPV4
145 if (family == PF_INET)
146 return true;
147 # endif
148 # if HAVE_IPV6
149 if (family == PF_INET6)
150 return true;
151 # endif
152 if (family == PF_UNSPEC)
153 return true;
154 return false;
155 }
156
157 /* Translate name of a service location and/or a service name to set of
158 socket addresses. */
159 int
getaddrinfo(const char * restrict nodename,const char * restrict servname,const struct addrinfo * restrict hints,struct addrinfo ** restrict res)160 getaddrinfo (const char *restrict nodename,
161 const char *restrict servname,
162 const struct addrinfo *restrict hints,
163 struct addrinfo **restrict res)
164 {
165 struct addrinfo *tmp;
166 int port = 0;
167 struct hostent *he;
168 void *storage;
169 size_t size;
170 # if HAVE_IPV6
171 struct v6_pair {
172 struct addrinfo addrinfo;
173 struct sockaddr_in6 sockaddr_in6;
174 };
175 # endif
176 # if HAVE_IPV4
177 struct v4_pair {
178 struct addrinfo addrinfo;
179 struct sockaddr_in sockaddr_in;
180 };
181 # endif
182
183 # ifdef WINDOWS_NATIVE
184 if (use_win32_p ())
185 return getaddrinfo_ptr (nodename, servname, hints, res);
186 # endif
187
188 if (hints && (hints->ai_flags & ~(AI_CANONNAME|AI_PASSIVE)))
189 /* FIXME: Support more flags. */
190 return EAI_BADFLAGS;
191
192 if (hints && !validate_family (hints->ai_family))
193 return EAI_FAMILY;
194
195 if (hints &&
196 hints->ai_socktype != SOCK_STREAM && hints->ai_socktype != SOCK_DGRAM)
197 /* FIXME: Support other socktype. */
198 return EAI_SOCKTYPE; /* FIXME: Better return code? */
199
200 if (!nodename)
201 {
202 if (!(hints->ai_flags & AI_PASSIVE))
203 return EAI_NONAME;
204
205 # ifdef HAVE_IPV6
206 nodename = (hints->ai_family == AF_INET6) ? "::" : "0.0.0.0";
207 # else
208 nodename = "0.0.0.0";
209 # endif
210 }
211
212 if (servname)
213 {
214 struct servent *se = NULL;
215 const char *proto =
216 (hints && hints->ai_socktype == SOCK_DGRAM) ? "udp" : "tcp";
217
218 if (hints == NULL || !(hints->ai_flags & AI_NUMERICSERV))
219 /* FIXME: Use getservbyname_r if available. */
220 se = getservbyname (servname, proto);
221
222 if (!se)
223 {
224 char *c;
225 if (!(*servname >= '0' && *servname <= '9'))
226 return EAI_NONAME;
227 port = strtoul (servname, &c, 10);
228 if (*c || port > 0xffff)
229 return EAI_NONAME;
230 port = htons (port);
231 }
232 else
233 port = se->s_port;
234 }
235
236 /* FIXME: Use gethostbyname_r if available. */
237 he = gethostbyname (nodename);
238 if (!he || he->h_addr_list[0] == NULL)
239 return EAI_NONAME;
240
241 switch (he->h_addrtype)
242 {
243 # if HAVE_IPV6
244 case PF_INET6:
245 size = sizeof (struct v6_pair);
246 break;
247 # endif
248
249 # if HAVE_IPV4
250 case PF_INET:
251 size = sizeof (struct v4_pair);
252 break;
253 # endif
254
255 default:
256 return EAI_NODATA;
257 }
258
259 storage = calloc (1, size);
260 if (!storage)
261 return EAI_MEMORY;
262
263 switch (he->h_addrtype)
264 {
265 # if HAVE_IPV6
266 case PF_INET6:
267 {
268 struct v6_pair *p = storage;
269 struct sockaddr_in6 *sinp = &p->sockaddr_in6;
270 tmp = &p->addrinfo;
271
272 if (port)
273 sinp->sin6_port = port;
274
275 if (he->h_length != sizeof (sinp->sin6_addr))
276 {
277 free (storage);
278 return EAI_SYSTEM; /* FIXME: Better return code? Set errno? */
279 }
280
281 memcpy (&sinp->sin6_addr, he->h_addr_list[0], sizeof sinp->sin6_addr);
282
283 tmp->ai_addr = (struct sockaddr *) sinp;
284 tmp->ai_addrlen = sizeof *sinp;
285 }
286 break;
287 # endif
288
289 # if HAVE_IPV4
290 case PF_INET:
291 {
292 struct v4_pair *p = storage;
293 struct sockaddr_in *sinp = &p->sockaddr_in;
294 tmp = &p->addrinfo;
295
296 if (port)
297 sinp->sin_port = port;
298
299 if (he->h_length != sizeof (sinp->sin_addr))
300 {
301 free (storage);
302 return EAI_SYSTEM; /* FIXME: Better return code? Set errno? */
303 }
304
305 memcpy (&sinp->sin_addr, he->h_addr_list[0], sizeof sinp->sin_addr);
306
307 tmp->ai_addr = (struct sockaddr *) sinp;
308 tmp->ai_addrlen = sizeof *sinp;
309 }
310 break;
311 # endif
312
313 default:
314 free (storage);
315 return EAI_NODATA;
316 }
317
318 if (hints && hints->ai_flags & AI_CANONNAME)
319 {
320 const char *cn;
321 if (he->h_name)
322 cn = he->h_name;
323 else
324 cn = nodename;
325
326 tmp->ai_canonname = strdup (cn);
327 if (!tmp->ai_canonname)
328 {
329 free (storage);
330 return EAI_MEMORY;
331 }
332 }
333
334 tmp->ai_protocol = (hints) ? hints->ai_protocol : 0;
335 tmp->ai_socktype = (hints) ? hints->ai_socktype : 0;
336 tmp->ai_addr->sa_family = he->h_addrtype;
337 tmp->ai_family = he->h_addrtype;
338
339 # ifdef HAVE_STRUCT_SOCKADDR_SA_LEN
340 switch (he->h_addrtype)
341 {
342 # if HAVE_IPV4
343 case AF_INET:
344 tmp->ai_addr->sa_len = sizeof (struct sockaddr_in);
345 break;
346 # endif
347 # if HAVE_IPV6
348 case AF_INET6:
349 tmp->ai_addr->sa_len = sizeof (struct sockaddr_in6);
350 break;
351 # endif
352 }
353 # endif
354
355 /* FIXME: If more than one address, create linked list of addrinfo's. */
356
357 *res = tmp;
358
359 return 0;
360 }
361
362 /* Free 'addrinfo' structure AI including associated storage. */
363 void
freeaddrinfo(struct addrinfo * ai)364 freeaddrinfo (struct addrinfo *ai)
365 {
366 # ifdef WINDOWS_NATIVE
367 if (use_win32_p ())
368 {
369 freeaddrinfo_ptr (ai);
370 return;
371 }
372 # endif
373
374 while (ai)
375 {
376 struct addrinfo *cur;
377
378 cur = ai;
379 ai = ai->ai_next;
380
381 free (cur->ai_canonname);
382 free (cur);
383 }
384 }
385
386 int
getnameinfo(const struct sockaddr * restrict sa,socklen_t salen,char * restrict node,socklen_t nodelen,char * restrict service,socklen_t servicelen,int flags)387 getnameinfo (const struct sockaddr *restrict sa, socklen_t salen,
388 char *restrict node, socklen_t nodelen,
389 char *restrict service, socklen_t servicelen,
390 int flags)
391 {
392 # ifdef WINDOWS_NATIVE
393 if (use_win32_p ())
394 return getnameinfo_ptr (sa, salen, node, nodelen,
395 service, servicelen, flags);
396 # endif
397
398 /* FIXME: Support other flags. */
399 if ((node && nodelen > 0 && !(flags & NI_NUMERICHOST)) ||
400 (service && servicelen > 0 && !(flags & NI_NUMERICHOST)) ||
401 (flags & ~(NI_NUMERICHOST|NI_NUMERICSERV)))
402 return EAI_BADFLAGS;
403
404 if (sa == NULL || salen < sizeof (sa->sa_family))
405 return EAI_FAMILY;
406
407 switch (sa->sa_family)
408 {
409 # if HAVE_IPV4
410 case AF_INET:
411 if (salen < sizeof (struct sockaddr_in))
412 return EAI_FAMILY;
413 break;
414 # endif
415 # if HAVE_IPV6
416 case AF_INET6:
417 if (salen < sizeof (struct sockaddr_in6))
418 return EAI_FAMILY;
419 break;
420 # endif
421 default:
422 return EAI_FAMILY;
423 }
424
425 if (node && nodelen > 0 && flags & NI_NUMERICHOST)
426 {
427 switch (sa->sa_family)
428 {
429 # if HAVE_IPV4
430 case AF_INET:
431 if (!inet_ntop (AF_INET,
432 &(((const struct sockaddr_in *) sa)->sin_addr),
433 node, nodelen))
434 return EAI_SYSTEM;
435 break;
436 # endif
437
438 # if HAVE_IPV6
439 case AF_INET6:
440 if (!inet_ntop (AF_INET6,
441 &(((const struct sockaddr_in6 *) sa)->sin6_addr),
442 node, nodelen))
443 return EAI_SYSTEM;
444 break;
445 # endif
446
447 default:
448 return EAI_FAMILY;
449 }
450 }
451
452 if (service && servicelen > 0 && flags & NI_NUMERICSERV)
453 switch (sa->sa_family)
454 {
455 # if HAVE_IPV4
456 case AF_INET:
457 # endif
458 # if HAVE_IPV6
459 case AF_INET6:
460 # endif
461 {
462 unsigned short int port
463 = ntohs (((const struct sockaddr_in *) sa)->sin_port);
464 if (servicelen <= snprintf (service, servicelen, "%u", port))
465 return EAI_OVERFLOW;
466 }
467 break;
468 }
469
470 return 0;
471 }
472
473 #endif
474