lib/getaddrinfo.c

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