lib/libc/net/getaddrinfo.c

   1 /*      $KAME: getaddrinfo.c,v 1.15 2000/07/09 04:37:24 itojun Exp $    */
   2
   3 /*-
   4  * SPDX-License-Identifier: BSD-3-Clause
   5  *
   6  * Copyright (C) 1995, 1996, 1997, and 1998 WIDE Project.
   7  * All rights reserved.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted provided that the following conditions
  11  * are met:
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in the
  16  *    documentation and/or other materials provided with the distribution.
  17  * 3. Neither the name of the project nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  */
  33
  34 /*
  35  * Issues to be discussed:
  36  * - Return values.  There are nonstandard return values defined and used
  37  *   in the source code.  This is because RFC2553 is silent about which error
  38  *   code must be returned for which situation.
  39  * - freeaddrinfo(NULL).  RFC2553 is silent about it.  XNET 5.2 says it is
  40  *   invalid.  Current code accepts NULL to be compatible with other OSes.
  41  *
  42  * Note:
  43  * - The code filters out AFs that are not supported by the kernel,
  44  *   when globbing NULL hostname (to loopback, or wildcard).  Is it the right
  45  *   thing to do?  What is the relationship with post-RFC2553 AI_ADDRCONFIG
  46  *   in ai_flags?
  47  * - (post-2553) semantics of AI_ADDRCONFIG itself is too vague.
  48  *   (1) what should we do against numeric hostname (2) what should we do
  49  *   against NULL hostname (3) what is AI_ADDRCONFIG itself.  AF not ready?
  50  *   non-loopback address configured?  global address configured?
  51  *
  52  * OS specific notes for freebsd4:
  53  * - FreeBSD supported $GAI.  The code does not.
  54  */
  55
  56 #include <sys/cdefs.h>
  57 __FBSDID("$FreeBSD$");
  58
  59 #include "namespace.h"
  60 #include <sys/param.h>
  61 #include <sys/socket.h>
  62 #include <net/if.h>
  63 #include <netinet/in.h>
  64 #include <net/if_types.h>
  65 #include <ifaddrs.h>
  66 #include <sys/queue.h>
  67 #ifdef INET6
  68 #include <sys/sysctl.h>
  69 #include <sys/ioctl.h>
  70 #include <netinet6/in6_var.h>
  71 #include <netinet6/nd6.h>
  72 #endif
  73 #include <arpa/inet.h>
  74 #include <arpa/nameser.h>
  75 #include <rpc/rpc.h>
  76 #include <rpcsvc/yp_prot.h>
  77 #include <rpcsvc/ypclnt.h>
  78 #include <netdb.h>
  79 #include <resolv.h>
  80 #include <string.h>
  81 #include <stdlib.h>
  82 #include <stddef.h>
  83 #include <ctype.h>
  84 #include <unistd.h>
  85 #include <stdio.h>
  86 #include <errno.h>
  87
  88 #include "res_config.h"
  89
  90 #ifdef DEBUG
  91 #include <syslog.h>
  92 #endif
  93
  94 #include <stdarg.h>
  95 #include <nsswitch.h>
  96 #include "un-namespace.h"
  97 #include "netdb_private.h"
  98 #include "libc_private.h"
  99 #ifdef NS_CACHING
 100 #include "nscache.h"
 101 #endif
 102
 103 #define ANY 0
 104 #define YES 1
 105 #define NO  0
 106
 107 static const char in_addrany[] = { 0, 0, 0, 0 };
 108 static const char in_loopback[] = { 127, 0, 0, 1 };
 109 #ifdef INET6
 110 static const char in6_addrany[] = {
 111         0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
 112 };
 113 static const char in6_loopback[] = {
 114         0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1
 115 };
 116 #endif
 117
 118 struct policyqueue {
 119         TAILQ_ENTRY(policyqueue) pc_entry;
 120 #ifdef INET6
 121         struct in6_addrpolicy pc_policy;
 122 #endif
 123 };
 124 TAILQ_HEAD(policyhead, policyqueue);
 125
 126 static const struct afd {
 127         int a_af;
 128         int a_addrlen;
 129         socklen_t a_socklen;
 130         int a_off;
 131         const char *a_addrany;
 132         const char *a_loopback;
 133         int a_scoped;
 134 } afdl [] = {
 135 #ifdef INET6
 136 #define N_INET6 0
 137         {PF_INET6, sizeof(struct in6_addr),
 138          sizeof(struct sockaddr_in6),
 139          offsetof(struct sockaddr_in6, sin6_addr),
 140          in6_addrany, in6_loopback, 1},
 141 #define N_INET 1
 142 #define N_LOCAL 2
 143 #else
 144 #define N_INET 0
 145 #define N_LOCAL 1
 146 #endif
 147         {PF_INET, sizeof(struct in_addr),
 148          sizeof(struct sockaddr_in),
 149          offsetof(struct sockaddr_in, sin_addr),
 150          in_addrany, in_loopback, 0},
 151 #define sizeofmember(type, member)      (sizeof(((type *)0)->member))
 152         {PF_LOCAL, sizeofmember(struct sockaddr_un, sun_path),
 153          sizeof(struct sockaddr_un),
 154          offsetof(struct sockaddr_un, sun_path),
 155          NULL, NULL, 0},
 156         {0, 0, 0, 0, NULL, NULL, 0},
 157 };
 158
 159 struct explore {
 160         int e_af;
 161         int e_socktype;
 162         int e_protocol;
 163         int e_wild;
 164 #define AF_ANY          0x01
 165 #define SOCKTYPE_ANY    0x02
 166 #define PROTOCOL_ANY    0x04
 167 #define WILD_AF(ex)             ((ex)->e_wild & AF_ANY)
 168 #define WILD_SOCKTYPE(ex)       ((ex)->e_wild & SOCKTYPE_ANY)
 169 #define WILD_PROTOCOL(ex)       ((ex)->e_wild & PROTOCOL_ANY)
 170 };
 171
 172 static const struct explore explore[] = {
 173 #ifdef INET6
 174         { PF_INET6, SOCK_DGRAM,  IPPROTO_UDP,
 175             AF_ANY | SOCKTYPE_ANY | PROTOCOL_ANY },
 176         { PF_INET6, SOCK_STREAM, IPPROTO_TCP,
 177             AF_ANY | SOCKTYPE_ANY | PROTOCOL_ANY },
 178         { PF_INET6, SOCK_STREAM, IPPROTO_SCTP,
 179             AF_ANY | SOCKTYPE_ANY },
 180         { PF_INET6, SOCK_SEQPACKET, IPPROTO_SCTP,
 181             AF_ANY | SOCKTYPE_ANY | PROTOCOL_ANY },
 182         { PF_INET6, SOCK_DGRAM, IPPROTO_UDPLITE,
 183             AF_ANY | SOCKTYPE_ANY },
 184         { PF_INET6, SOCK_RAW, ANY,
 185             AF_ANY | PROTOCOL_ANY },
 186 #endif
 187         { PF_INET, SOCK_DGRAM, IPPROTO_UDP,
 188             AF_ANY | SOCKTYPE_ANY | PROTOCOL_ANY },
 189         { PF_INET, SOCK_STREAM, IPPROTO_TCP,
 190             AF_ANY | SOCKTYPE_ANY | PROTOCOL_ANY },
 191         { PF_INET, SOCK_STREAM, IPPROTO_SCTP,
 192             AF_ANY | SOCKTYPE_ANY },
 193         { PF_INET, SOCK_SEQPACKET, IPPROTO_SCTP,
 194             AF_ANY | SOCKTYPE_ANY | PROTOCOL_ANY },
 195         { PF_INET, SOCK_DGRAM, IPPROTO_UDPLITE,
 196             AF_ANY | SOCKTYPE_ANY },
 197         { PF_INET, SOCK_RAW, ANY,
 198             AF_ANY | PROTOCOL_ANY },
 199         { PF_LOCAL, SOCK_DGRAM, ANY,
 200             AF_ANY | SOCKTYPE_ANY | PROTOCOL_ANY },
 201         { PF_LOCAL, SOCK_STREAM, ANY,
 202             AF_ANY | SOCKTYPE_ANY | PROTOCOL_ANY },
 203         { PF_LOCAL, SOCK_SEQPACKET, ANY,
 204             AF_ANY | SOCKTYPE_ANY | PROTOCOL_ANY },
 205         { -1, 0, 0, 0 },
 206 };
 207
 208 #ifdef INET6
 209 #define PTON_MAX        16
 210 #else
 211 #define PTON_MAX        4
 212 #endif
 213
 214 #define AIO_SRCFLAG_DEPRECATED  0x1
 215
 216 struct ai_order {
 217         union {
 218                 struct sockaddr_storage aiou_ss;
 219                 struct sockaddr aiou_sa;
 220         } aio_src_un;
 221 #define aio_srcsa aio_src_un.aiou_sa
 222         u_int32_t aio_srcflag;
 223         int aio_srcscope;
 224         int aio_dstscope;
 225         struct policyqueue *aio_srcpolicy;
 226         struct policyqueue *aio_dstpolicy;
 227         struct addrinfo *aio_ai;
 228         int aio_matchlen;
 229         int aio_initial_sequence;
 230 };
 231
 232 static const ns_src default_dns_files[] = {
 233         { NSSRC_FILES,  NS_SUCCESS },
 234         { NSSRC_DNS,    NS_SUCCESS },
 235         { 0 }
 236 };
 237
 238 struct res_target {
 239         struct res_target *next;
 240         const char *name;       /* domain name */
 241         int qclass, qtype;      /* class and type of query */
 242         u_char *answer;         /* buffer to put answer */
 243         int anslen;             /* size of answer buffer */
 244         int n;                  /* result length */
 245 };
 246
 247 #define MAXPACKET       (64*1024)
 248
 249 typedef union {
 250         HEADER hdr;
 251         u_char buf[MAXPACKET];
 252 } querybuf;
 253
 254 static int str2number(const char *, int *);
 255 static int explore_copy(const struct addrinfo *, const struct addrinfo *,
 256         struct addrinfo **);
 257 static int explore_null(const struct addrinfo *,
 258         const char *, struct addrinfo **);
 259 static int explore_numeric(const struct addrinfo *, const char *,
 260         const char *, struct addrinfo **, const char *);
 261 static int explore_numeric_scope(const struct addrinfo *, const char *,
 262         const char *, struct addrinfo **);
 263 static int get_canonname(const struct addrinfo *,
 264         struct addrinfo *, const char *);
 265 static struct addrinfo *get_ai(const struct addrinfo *,
 266         const struct afd *, const char *);
 267 static struct addrinfo *copy_ai(const struct addrinfo *);
 268 static int get_portmatch(const struct addrinfo *, const char *);
 269 static int get_port(struct addrinfo *, const char *, int);
 270 static const struct afd *find_afd(int);
 271 static int addrconfig(struct addrinfo *);
 272 #ifdef INET6
 273 static int is_ifdisabled(char *);
 274 #endif
 275 static void set_source(struct ai_order *, struct policyhead *);
 276 static int comp_dst(const void *, const void *);
 277 #ifdef INET6
 278 static int ip6_str2scopeid(char *, struct sockaddr_in6 *, u_int32_t *);
 279 #endif
 280 static int gai_addr2scopetype(struct sockaddr *);
 281
 282 static int explore_fqdn(const struct addrinfo *, const char *,
 283         const char *, struct addrinfo **);
 284
 285 static int reorder(struct addrinfo *);
 286 static int get_addrselectpolicy(struct policyhead *);
 287 static void free_addrselectpolicy(struct policyhead *);
 288 static struct policyqueue *match_addrselectpolicy(struct sockaddr *,
 289         struct policyhead *);
 290 static int matchlen(struct sockaddr *, struct sockaddr *);
 291
 292 static struct addrinfo *getanswer(const querybuf *, int, const char *, int,
 293         const struct addrinfo *, res_state);
 294 #if defined(RESOLVSORT)
 295 static int addr4sort(struct addrinfo *, res_state);
 296 #endif
 297 static int _dns_getaddrinfo(void *, void *, va_list);
 298 static void _sethtent(FILE **);
 299 static void _endhtent(FILE **);
 300 static struct addrinfo *_gethtent(FILE **, const char *,
 301         const struct addrinfo *);
 302 static int _files_getaddrinfo(void *, void *, va_list);
 303 #ifdef YP
 304 static struct addrinfo *_yphostent(char *, const struct addrinfo *);
 305 static int _yp_getaddrinfo(void *, void *, va_list);
 306 #endif
 307 #ifdef NS_CACHING
 308 static int addrinfo_id_func(char *, size_t *, va_list, void *);
 309 static int addrinfo_marshal_func(char *, size_t *, void *, va_list, void *);
 310 static int addrinfo_unmarshal_func(char *, size_t, void *, va_list, void *);
 311 #endif
 312
 313 static int res_queryN(const char *, struct res_target *, res_state);
 314 static int res_searchN(const char *, struct res_target *, res_state);
 315 static int res_querydomainN(const char *, const char *,
 316         struct res_target *, res_state);
 317
 318 /* XXX macros that make external reference is BAD. */
 319
 320 #define GET_AI(ai, afd, addr) \
 321 do { \
 322         /* external reference: pai, error, and label free */ \
 323         (ai) = get_ai(pai, (afd), (addr)); \
 324         if ((ai) == NULL) { \
 325                 error = EAI_MEMORY; \
 326                 goto free; \
 327         } \
 328 } while (/*CONSTCOND*/0)
 329
 330 #define GET_PORT(ai, serv) \
 331 do { \
 332         /* external reference: error and label free */ \
 333         error = get_port((ai), (serv), 0); \
 334         if (error != 0) \
 335                 goto free; \
 336 } while (/*CONSTCOND*/0)
 337
 338 #define GET_CANONNAME(ai, str) \
 339 do { \
 340         /* external reference: pai, error and label free */ \
 341         error = get_canonname(pai, (ai), (str)); \
 342         if (error != 0) \
 343                 goto free; \
 344 } while (/*CONSTCOND*/0)
 345
 346 #define ERR(err) \
 347 do { \
 348         /* external reference: error, and label bad */ \
 349         error = (err); \
 350         goto bad; \
 351         /*NOTREACHED*/ \
 352 } while (/*CONSTCOND*/0)
 353
 354 #define MATCH_FAMILY(x, y, w) \
 355         ((x) == (y) || (/*CONSTCOND*/(w) && ((x) == PF_UNSPEC || (y) == PF_UNSPEC)))
 356 #define MATCH(x, y, w) \
 357         ((x) == (y) || (/*CONSTCOND*/(w) && ((x) == ANY || (y) == ANY)))
 358
 359 void
 360 freeaddrinfo(struct addrinfo *ai)
 361 {
 362         struct addrinfo *next;
 363
 364         while (ai != NULL) {
 365                 next = ai->ai_next;
 366                 free(ai->ai_canonname);
 367                 /* no need to free(ai->ai_addr) */
 368                 free(ai);
 369                 ai = next;
 370         }
 371 }
 372
 373 static int
 374 str2number(const char *p, int *portp)
 375 {
 376         char *ep;
 377         unsigned long v;
 378
 379         if (*p == '\0')
 380                 return -1;
 381         ep = NULL;
 382         errno = 0;
 383         v = strtoul(p, &ep, 10);
 384         if (errno == 0 && ep && *ep == '\0' && v <= UINT_MAX) {
 385                 *portp = v;
 386                 return 0;
 387         } else
 388                 return -1;
 389 }
 390
 391 int
 392 getaddrinfo(const char *hostname, const char *servname,
 393     const struct addrinfo *hints, struct addrinfo **res)
 394 {
 395         struct addrinfo sentinel;
 396         struct addrinfo *cur;
 397         int error = 0;
 398         struct addrinfo ai, ai0, *afai;
 399         struct addrinfo *pai;
 400         const struct afd *afd;
 401         const struct explore *ex;
 402         struct addrinfo *afailist[nitems(afdl)];
 403         struct addrinfo *afai_unspec;
 404         int found;
 405         int numeric = 0;
 406
 407         /* ensure we return NULL on errors */
 408         *res = NULL;
 409
 410         memset(&ai, 0, sizeof(ai));
 411
 412         memset(afailist, 0, sizeof(afailist));
 413         afai_unspec = NULL;
 414
 415         memset(&sentinel, 0, sizeof(sentinel));
 416         cur = &sentinel;
 417         pai = &ai;
 418         pai->ai_flags = 0;
 419         pai->ai_family = PF_UNSPEC;
 420         pai->ai_socktype = ANY;
 421         pai->ai_protocol = ANY;
 422         pai->ai_addrlen = 0;
 423         pai->ai_canonname = NULL;
 424         pai->ai_addr = NULL;
 425         pai->ai_next = NULL;
 426
 427         if (hostname == NULL && servname == NULL)
 428                 return EAI_NONAME;
 429         if (hints) {
 430                 /* error check for hints */
 431                 if (hints->ai_addrlen || hints->ai_canonname ||
 432                     hints->ai_addr || hints->ai_next)
 433                         ERR(EAI_BADHINTS); /* xxx */
 434                 if (hints->ai_flags & ~AI_MASK)
 435                         ERR(EAI_BADFLAGS);
 436                 switch (hints->ai_family) {
 437                 case PF_UNSPEC:
 438                 case PF_LOCAL:
 439                 case PF_INET:
 440 #ifdef INET6
 441                 case PF_INET6:
 442 #endif
 443                         break;
 444                 default:
 445                         ERR(EAI_FAMILY);
 446                 }
 447                 memcpy(pai, hints, sizeof(*pai));
 448
 449                 /*
 450                  * if both socktype/protocol are specified, check if they
 451                  * are meaningful combination.
 452                  */
 453                 if (pai->ai_socktype != ANY && pai->ai_protocol != ANY) {
 454                         for (ex = explore; ex->e_af >= 0; ex++) {
 455                                 if (!MATCH_FAMILY(pai->ai_family, ex->e_af,
 456                                     WILD_AF(ex)))
 457                                         continue;
 458                                 if (!MATCH(pai->ai_socktype, ex->e_socktype,
 459                                     WILD_SOCKTYPE(ex)))
 460                                         continue;
 461                                 if (!MATCH(pai->ai_protocol, ex->e_protocol,
 462                                     WILD_PROTOCOL(ex)))
 463                                         continue;
 464
 465                                 /* matched */
 466                                 break;
 467                         }
 468
 469                         if (ex->e_af < 0)
 470                                 ERR(EAI_BADHINTS);
 471                 }
 472         }
 473
 474         /*
 475          * RFC 3493: AI_ALL and AI_V4MAPPED are effective only against
 476          * AF_INET6 query.  They need to be ignored if specified in other
 477          * occasions.
 478          */
 479         switch (pai->ai_flags & (AI_ALL | AI_V4MAPPED)) {
 480         case AI_V4MAPPED:
 481         case AI_ALL | AI_V4MAPPED:
 482 #ifdef INET6
 483                 if (pai->ai_family != AF_INET6)
 484                         pai->ai_flags &= ~(AI_ALL | AI_V4MAPPED);
 485                 break;
 486 #endif
 487         case AI_ALL:
 488                 pai->ai_flags &= ~(AI_ALL | AI_V4MAPPED);
 489                 break;
 490         }
 491
 492         /*
 493          * check for special cases.  (1) numeric servname is disallowed if
 494          * socktype/protocol are left unspecified. (2) servname is disallowed
 495          * for raw and other inet{,6} sockets.
 496          */
 497         if (MATCH_FAMILY(pai->ai_family, PF_INET, 1)
 498 #ifdef PF_INET6
 499             || MATCH_FAMILY(pai->ai_family, PF_INET6, 1)
 500 #endif
 501             ) {
 502                 ai0 = *pai;     /* backup *pai */
 503
 504                 if (pai->ai_family == PF_UNSPEC) {
 505 #ifdef PF_INET6
 506                         pai->ai_family = PF_INET6;
 507 #else
 508                         pai->ai_family = PF_INET;
 509 #endif
 510                 }
 511                 error = get_portmatch(pai, servname);
 512                 if (error)
 513                         goto bad;
 514
 515                 *pai = ai0;
 516         }
 517
 518         ai0 = *pai;
 519
 520         /*
 521          * NULL hostname, or numeric hostname.
 522          * If numeric representation of AF1 can be interpreted as FQDN
 523          * representation of AF2, we need to think again about the code below.
 524          */
 525         found = 0;
 526         for (afd = afdl; afd->a_af; afd++) {
 527                 *pai = ai0;
 528
 529                 if (!MATCH_FAMILY(pai->ai_family, afd->a_af, 1))
 530                         continue;
 531
 532                 if (pai->ai_family == PF_UNSPEC)
 533                         pai->ai_family = afd->a_af;
 534
 535                 if (hostname == NULL) {
 536                         error = explore_null(pai, servname,
 537                             &afailist[afd - afdl]);
 538
 539                         /*
 540                          * Errors from explore_null should be unexpected and
 541                          * be caught to avoid returning an incomplete result.
 542                          */
 543                         if (error != 0)
 544                                 goto bad;
 545                 } else {
 546                         error = explore_numeric_scope(pai, hostname, servname,
 547                             &afailist[afd - afdl]);
 548
 549                         /*
 550                          * explore_numeric_scope returns an error for address
 551                          * families that do not match that of hostname.
 552                          * Thus we should not catch the error at this moment.
 553                          */
 554                 }
 555
 556                 if (!error && afailist[afd - afdl])
 557                         found++;
 558         }
 559         if (found) {
 560                 numeric = 1;
 561                 goto globcopy;
 562         }
 563
 564         if (hostname == NULL)
 565                 ERR(EAI_NONAME);        /* used to be EAI_NODATA */
 566         if (pai->ai_flags & AI_NUMERICHOST)
 567                 ERR(EAI_NONAME);
 568
 569         if ((pai->ai_flags & AI_ADDRCONFIG) != 0 && !addrconfig(&ai0))
 570                 ERR(EAI_FAIL);
 571
 572         /*
 573          * hostname as alphabetical name.
 574          */
 575         *pai = ai0;
 576         error = explore_fqdn(pai, hostname, servname, &afai_unspec);
 577
 578 globcopy:
 579         for (ex = explore; ex->e_af >= 0; ex++) {
 580                 *pai = ai0;
 581
 582                 if (!MATCH_FAMILY(pai->ai_family, ex->e_af, WILD_AF(ex)))
 583                         continue;
 584                 if (!MATCH(pai->ai_socktype, ex->e_socktype,
 585                     WILD_SOCKTYPE(ex)))
 586                         continue;
 587                 if (!MATCH(pai->ai_protocol, ex->e_protocol,
 588                     WILD_PROTOCOL(ex)))
 589                         continue;
 590
 591                 if (pai->ai_family == PF_UNSPEC)
 592                         pai->ai_family = ex->e_af;
 593                 if (pai->ai_socktype == ANY && ex->e_socktype != ANY)
 594                         pai->ai_socktype = ex->e_socktype;
 595                 if (pai->ai_protocol == ANY && ex->e_protocol != ANY)
 596                         pai->ai_protocol = ex->e_protocol;
 597
 598                 /*
 599                  * if the servname does not match socktype/protocol, ignore it.
 600                  */
 601                 if (get_portmatch(pai, servname) != 0)
 602                         continue;
 603
 604                 if (afai_unspec)
 605                         afai = afai_unspec;
 606                 else {
 607                         if ((afd = find_afd(pai->ai_family)) == NULL)
 608                                 continue;
 609                         /* XXX assumes that afd points inside afdl[] */
 610                         afai = afailist[afd - afdl];
 611                 }
 612                 if (!afai)
 613                         continue;
 614
 615                 error = explore_copy(pai, afai, &cur->ai_next);
 616                 if (error != 0)
 617                         goto bad;
 618
 619                 while (cur && cur->ai_next)
 620                         cur = cur->ai_next;
 621         }
 622
 623         /*
 624          * ensure we return either:
 625          * - error == 0, non-NULL *res
 626          * - error != 0, NULL *res
 627          */
 628         if (error == 0) {
 629                 if (sentinel.ai_next) {
 630                         /*
 631                          * If the returned entry is for an active connection,
 632                          * and the given name is not numeric, reorder the
 633                          * list, so that the application would try the list
 634                          * in the most efficient order.  Since the head entry
 635                          * of the original list may contain ai_canonname and
 636                          * that entry may be moved elsewhere in the new list,
 637                          * we keep the pointer and will  restore it in the new
 638                          * head entry.  (Note that RFC3493 requires the head
 639                          * entry store it when requested by the caller).
 640                          */
 641                         if (hints == NULL || !(hints->ai_flags & AI_PASSIVE)) {
 642                                 if (!numeric) {
 643                                         char *canonname;
 644
 645                                         canonname =
 646                                             sentinel.ai_next->ai_canonname;
 647                                         sentinel.ai_next->ai_canonname = NULL;
 648                                         (void)reorder(&sentinel);
 649                                         if (sentinel.ai_next->ai_canonname ==
 650                                             NULL) {
 651                                                 sentinel.ai_next->ai_canonname
 652                                                     = canonname;
 653                                         } else if (canonname != NULL)
 654                                                 free(canonname);
 655                                 }
 656                         }
 657                         *res = sentinel.ai_next;
 658                 } else
 659                         error = EAI_FAIL;
 660         }
 661
 662 bad:
 663         if (afai_unspec)
 664                 freeaddrinfo(afai_unspec);
 665         for (afd = afdl; afd->a_af; afd++) {
 666                 if (afailist[afd - afdl])
 667                         freeaddrinfo(afailist[afd - afdl]);
 668         }
 669         if (!*res)
 670                 if (sentinel.ai_next)
 671                         freeaddrinfo(sentinel.ai_next);
 672
 673         return (error);
 674 }
 675
 676 static int
 677 reorder(struct addrinfo *sentinel)
 678 {
 679         struct addrinfo *ai, **aip;
 680         struct ai_order *aio;
 681         int i, n;
 682         struct policyhead policyhead;
 683
 684         /* count the number of addrinfo elements for sorting. */
 685         for (n = 0, ai = sentinel->ai_next; ai != NULL; ai = ai->ai_next, n++)
 686                 ;
 687
 688         /*
 689          * If the number is small enough, we can skip the reordering process.
 690          */
 691         if (n <= 1)
 692                 return(n);
 693
 694         /* allocate a temporary array for sort and initialization of it. */
 695         if ((aio = calloc(n, sizeof(*aio))) == NULL)
 696                 return(n);      /* give up reordering */
 697
 698         /* retrieve address selection policy from the kernel */
 699         TAILQ_INIT(&policyhead);
 700         if (!get_addrselectpolicy(&policyhead)) {
 701                 /* no policy is installed into kernel, we don't sort. */
 702                 free(aio);
 703                 return (n);
 704         }
 705
 706         for (i = 0, ai = sentinel->ai_next; i < n; ai = ai->ai_next, i++) {
 707                 aio[i].aio_ai = ai;
 708                 aio[i].aio_dstscope = gai_addr2scopetype(ai->ai_addr);
 709                 aio[i].aio_dstpolicy = match_addrselectpolicy(ai->ai_addr,
 710                                                               &policyhead);
 711                 set_source(&aio[i], &policyhead);
 712                 aio[i].aio_initial_sequence = i;
 713         }
 714
 715         /* perform sorting. */
 716         qsort(aio, n, sizeof(*aio), comp_dst);
 717
 718         /* reorder the addrinfo chain. */
 719         for (i = 0, aip = &sentinel->ai_next; i < n; i++) {
 720                 *aip = aio[i].aio_ai;
 721                 aip = &aio[i].aio_ai->ai_next;
 722         }
 723         *aip = NULL;
 724
 725         /* cleanup and return */
 726         free(aio);
 727         free_addrselectpolicy(&policyhead);
 728         return(n);
 729 }
 730
 731 static int
 732 get_addrselectpolicy(struct policyhead *head)
 733 {
 734 #ifdef INET6
 735         int mib[] = { CTL_NET, PF_INET6, IPPROTO_IPV6, IPV6CTL_ADDRCTLPOLICY };
 736         size_t l;
 737         char *buf;
 738         struct in6_addrpolicy *pol, *ep;
 739
 740         if (sysctl(mib, nitems(mib), NULL, &l, NULL, 0) < 0)
 741                 return (0);
 742         if (l == 0)
 743                 return (0);
 744         if ((buf = malloc(l)) == NULL)
 745                 return (0);
 746         if (sysctl(mib, nitems(mib), buf, &l, NULL, 0) < 0) {
 747                 free(buf);
 748                 return (0);
 749         }
 750
 751         ep = (struct in6_addrpolicy *)(buf + l);
 752         for (pol = (struct in6_addrpolicy *)buf; pol + 1 <= ep; pol++) {
 753                 struct policyqueue *new;
 754
 755                 if ((new = malloc(sizeof(*new))) == NULL) {
 756                         free_addrselectpolicy(head); /* make the list empty */
 757                         break;
 758                 }
 759                 new->pc_policy = *pol;
 760                 TAILQ_INSERT_TAIL(head, new, pc_entry);
 761         }
 762
 763         free(buf);
 764         return (1);
 765 #else
 766         return (0);
 767 #endif
 768 }
 769
 770 static void
 771 free_addrselectpolicy(struct policyhead *head)
 772 {
 773         struct policyqueue *ent, *nent;
 774
 775         for (ent = TAILQ_FIRST(head); ent; ent = nent) {
 776                 nent = TAILQ_NEXT(ent, pc_entry);
 777                 TAILQ_REMOVE(head, ent, pc_entry);
 778                 free(ent);
 779         }
 780 }
 781
 782 static struct policyqueue *
 783 match_addrselectpolicy(struct sockaddr *addr, struct policyhead *head)
 784 {
 785 #ifdef INET6
 786         struct policyqueue *ent, *bestent = NULL;
 787         struct in6_addrpolicy *pol;
 788         int matchlen, bestmatchlen = -1;
 789         u_char *mp, *ep, *k, *p, m;
 790         struct sockaddr_in6 key;
 791
 792         switch(addr->sa_family) {
 793         case AF_INET6:
 794                 key = *(struct sockaddr_in6 *)addr;
 795                 break;
 796         case AF_INET:
 797                 /* convert the address into IPv4-mapped IPv6 address. */
 798                 memset(&key, 0, sizeof(key));
 799                 key.sin6_family = AF_INET6;
 800                 key.sin6_len = sizeof(key);
 801                 _map_v4v6_address(
 802                     (char *)&((struct sockaddr_in *)addr)->sin_addr,
 803                     (char *)&key.sin6_addr);
 804                 break;
 805         default:
 806                 return(NULL);
 807         }
 808
 809         for (ent = TAILQ_FIRST(head); ent; ent = TAILQ_NEXT(ent, pc_entry)) {
 810                 pol = &ent->pc_policy;
 811                 matchlen = 0;
 812
 813                 mp = (u_char *)&pol->addrmask.sin6_addr;
 814                 ep = mp + 16;   /* XXX: scope field? */
 815                 k = (u_char *)&key.sin6_addr;
 816                 p = (u_char *)&pol->addr.sin6_addr;
 817                 for (; mp < ep && *mp; mp++, k++, p++) {
 818                         m = *mp;
 819                         if ((*k & m) != *p)
 820                                 goto next; /* not match */
 821                         if (m == 0xff) /* short cut for a typical case */
 822                                 matchlen += 8;
 823                         else {
 824                                 while (m >= 0x80) {
 825                                         matchlen++;
 826                                         m <<= 1;
 827                                 }
 828                         }
 829                 }
 830
 831                 /* matched.  check if this is better than the current best. */
 832                 if (matchlen > bestmatchlen) {
 833                         bestent = ent;
 834                         bestmatchlen = matchlen;
 835                 }
 836
 837           next:
 838                 continue;
 839         }
 840
 841         return(bestent);
 842 #else
 843         return(NULL);
 844 #endif
 845
 846 }
 847
 848 static void
 849 set_source(struct ai_order *aio, struct policyhead *ph)
 850 {
 851         struct addrinfo ai = *aio->aio_ai;
 852         struct sockaddr_storage ss;
 853         socklen_t srclen;
 854         int s;
 855
 856         /* set unspec ("no source is available"), just in case */
 857         aio->aio_srcsa.sa_family = AF_UNSPEC;
 858         aio->aio_srcscope = -1;
 859
 860         switch(ai.ai_family) {
 861         case AF_INET:
 862 #ifdef INET6
 863         case AF_INET6:
 864 #endif
 865                 break;
 866         default:                /* ignore unsupported AFs explicitly */
 867                 return;
 868         }
 869
 870         /* XXX: make a dummy addrinfo to call connect() */
 871         ai.ai_socktype = SOCK_DGRAM;
 872         ai.ai_protocol = IPPROTO_UDP; /* is UDP too specific? */
 873         ai.ai_next = NULL;
 874         memset(&ss, 0, sizeof(ss));
 875         memcpy(&ss, ai.ai_addr, ai.ai_addrlen);
 876         ai.ai_addr = (struct sockaddr *)&ss;
 877         get_port(&ai, "1", 0);
 878
 879         /* open a socket to get the source address for the given dst */
 880         if ((s = _socket(ai.ai_family, ai.ai_socktype | SOCK_CLOEXEC,
 881             ai.ai_protocol)) < 0)
 882                 return;         /* give up */
 883 #ifdef INET6
 884         if (ai.ai_family == AF_INET6) {
 885                 struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)ai.ai_addr;
 886                 int off = 0;
 887
 888                 if (IN6_IS_ADDR_V4MAPPED(&sin6->sin6_addr))
 889                         (void)_setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY,
 890                             (char *)&off, sizeof(off));
 891         }
 892 #endif
 893         if (_connect(s, ai.ai_addr, ai.ai_addrlen) < 0)
 894                 goto cleanup;
 895         srclen = ai.ai_addrlen;
 896         if (_getsockname(s, &aio->aio_srcsa, &srclen) < 0) {
 897                 aio->aio_srcsa.sa_family = AF_UNSPEC;
 898                 goto cleanup;
 899         }
 900         aio->aio_srcscope = gai_addr2scopetype(&aio->aio_srcsa);
 901         aio->aio_srcpolicy = match_addrselectpolicy(&aio->aio_srcsa, ph);
 902         aio->aio_matchlen = matchlen(&aio->aio_srcsa, aio->aio_ai->ai_addr);
 903 #ifdef INET6
 904         if (ai.ai_family == AF_INET6) {
 905                 struct in6_ifreq ifr6;
 906                 u_int32_t flags6;
 907
 908                 memset(&ifr6, 0, sizeof(ifr6));
 909                 memcpy(&ifr6.ifr_addr, ai.ai_addr, ai.ai_addrlen);
 910                 if (_ioctl(s, SIOCGIFAFLAG_IN6, &ifr6) == 0) {
 911                         flags6 = ifr6.ifr_ifru.ifru_flags6;
 912                         if ((flags6 & IN6_IFF_DEPRECATED))
 913                                 aio->aio_srcflag |= AIO_SRCFLAG_DEPRECATED;
 914                 }
 915         }
 916 #endif
 917
 918   cleanup:
 919         _close(s);
 920         return;
 921 }
 922
 923 static int
 924 matchlen(struct sockaddr *src, struct sockaddr *dst)
 925 {
 926         int match = 0;
 927         u_char *s, *d;
 928         u_char *lim, r;
 929         int addrlen;
 930
 931         switch (src->sa_family) {
 932 #ifdef INET6
 933         case AF_INET6:
 934                 s = (u_char *)&((struct sockaddr_in6 *)src)->sin6_addr;
 935                 d = (u_char *)&((struct sockaddr_in6 *)dst)->sin6_addr;
 936                 addrlen = sizeof(struct in6_addr);
 937                 lim = s + addrlen;
 938                 break;
 939 #endif
 940         case AF_INET:
 941                 s = (u_char *)&((struct sockaddr_in *)src)->sin_addr;
 942                 d = (u_char *)&((struct sockaddr_in *)dst)->sin_addr;
 943                 addrlen = sizeof(struct in_addr);
 944                 lim = s + addrlen;
 945                 break;
 946         default:
 947                 return(0);
 948         }
 949
 950         while (s < lim)
 951                 if ((r = (*d++ ^ *s++)) != 0) {
 952                         while ((r & 0x80) == 0) {
 953                                 match++;
 954                                 r <<= 1;
 955                         }
 956                         break;
 957                 } else
 958                         match += 8;
 959         return(match);
 960 }
 961
 962 static int
 963 comp_dst(const void *arg1, const void *arg2)
 964 {
 965         const struct ai_order *dst1 = arg1, *dst2 = arg2;
 966
 967         /*
 968          * Rule 1: Avoid unusable destinations.
 969          * XXX: we currently do not consider if an appropriate route exists.
 970          */
 971         if (dst1->aio_srcsa.sa_family != AF_UNSPEC &&
 972             dst2->aio_srcsa.sa_family == AF_UNSPEC) {
 973                 return(-1);
 974         }
 975         if (dst1->aio_srcsa.sa_family == AF_UNSPEC &&
 976             dst2->aio_srcsa.sa_family != AF_UNSPEC) {
 977                 return(1);
 978         }
 979
 980         /* Rule 2: Prefer matching scope. */
 981         if (dst1->aio_dstscope == dst1->aio_srcscope &&
 982             dst2->aio_dstscope != dst2->aio_srcscope) {
 983                 return(-1);
 984         }
 985         if (dst1->aio_dstscope != dst1->aio_srcscope &&
 986             dst2->aio_dstscope == dst2->aio_srcscope) {
 987                 return(1);
 988         }
 989
 990         /* Rule 3: Avoid deprecated addresses. */
 991         if (dst1->aio_srcsa.sa_family != AF_UNSPEC &&
 992             dst2->aio_srcsa.sa_family != AF_UNSPEC) {
 993                 if (!(dst1->aio_srcflag & AIO_SRCFLAG_DEPRECATED) &&
 994                     (dst2->aio_srcflag & AIO_SRCFLAG_DEPRECATED)) {
 995                         return(-1);
 996                 }
 997                 if ((dst1->aio_srcflag & AIO_SRCFLAG_DEPRECATED) &&
 998                     !(dst2->aio_srcflag & AIO_SRCFLAG_DEPRECATED)) {
 999                         return(1);
1000                 }
1001         }
1002
1003         /* Rule 4: Prefer home addresses. */
1004         /* XXX: not implemented yet */
1005
1006         /* Rule 5: Prefer matching label. */
1007 #ifdef INET6
1008         if (dst1->aio_srcpolicy && dst1->aio_dstpolicy &&
1009             dst1->aio_srcpolicy->pc_policy.label ==
1010             dst1->aio_dstpolicy->pc_policy.label &&
1011             (dst2->aio_srcpolicy == NULL || dst2->aio_dstpolicy == NULL ||
1012              dst2->aio_srcpolicy->pc_policy.label !=
1013              dst2->aio_dstpolicy->pc_policy.label)) {
1014                 return(-1);
1015         }
1016         if (dst2->aio_srcpolicy && dst2->aio_dstpolicy &&
1017             dst2->aio_srcpolicy->pc_policy.label ==
1018             dst2->aio_dstpolicy->pc_policy.label &&
1019             (dst1->aio_srcpolicy == NULL || dst1->aio_dstpolicy == NULL ||
1020              dst1->aio_srcpolicy->pc_policy.label !=
1021              dst1->aio_dstpolicy->pc_policy.label)) {
1022                 return(1);
1023         }
1024 #endif
1025
1026         /* Rule 6: Prefer higher precedence. */
1027 #ifdef INET6
1028         if (dst1->aio_dstpolicy &&
1029             (dst2->aio_dstpolicy == NULL ||
1030              dst1->aio_dstpolicy->pc_policy.preced >
1031              dst2->aio_dstpolicy->pc_policy.preced)) {
1032                 return(-1);
1033         }
1034         if (dst2->aio_dstpolicy &&
1035             (dst1->aio_dstpolicy == NULL ||
1036              dst2->aio_dstpolicy->pc_policy.preced >
1037              dst1->aio_dstpolicy->pc_policy.preced)) {
1038                 return(1);
1039         }
1040 #endif
1041
1042         /* Rule 7: Prefer native transport. */
1043         /* XXX: not implemented yet */
1044
1045         /* Rule 8: Prefer smaller scope. */
1046         if (dst1->aio_dstscope >= 0 &&
1047             dst1->aio_dstscope < dst2->aio_dstscope) {
1048                 return(-1);
1049         }
1050         if (dst2->aio_dstscope >= 0 &&
1051             dst2->aio_dstscope < dst1->aio_dstscope) {
1052                 return(1);
1053         }
1054
1055         /*
1056          * Rule 9: Use longest matching prefix.
1057          * We compare the match length in a same AF only.
1058          */
1059         if (dst1->aio_ai->ai_addr->sa_family ==
1060             dst2->aio_ai->ai_addr->sa_family &&
1061             dst1->aio_ai->ai_addr->sa_family != AF_INET) {
1062                 if (dst1->aio_matchlen > dst2->aio_matchlen) {
1063                         return(-1);
1064                 }
1065                 if (dst1->aio_matchlen < dst2->aio_matchlen) {
1066                         return(1);
1067                 }
1068         }
1069
1070         /* Rule 10: Otherwise, leave the order unchanged. */
1071
1072         /*
1073          * Note that qsort is unstable; so, we can't return zero and
1074          * expect the order to be unchanged.
1075          * That also means we can't depend on the current position of
1076          * dst2 being after dst1.  We must enforce the initial order
1077          * with an explicit compare on the original position.
1078          * The qsort specification requires that "When the same objects
1079          * (consisting of width bytes, irrespective of their current
1080          * positions in the array) are passed more than once to the
1081          * comparison function, the results shall be consistent with one
1082          * another."
1083          * In other words, If A < B, then we must also return B > A.
1084          */
1085         if (dst2->aio_initial_sequence < dst1->aio_initial_sequence)
1086                 return(1);
1087
1088         return(-1);
1089 }
1090
1091 /*
1092  * Copy from scope.c.
1093  * XXX: we should standardize the functions and link them as standard
1094  * library.
1095  */
1096 static int
1097 gai_addr2scopetype(struct sockaddr *sa)
1098 {
1099 #ifdef INET6
1100         struct sockaddr_in6 *sa6;
1101 #endif
1102         struct sockaddr_in *sa4;
1103
1104         switch(sa->sa_family) {
1105 #ifdef INET6
1106         case AF_INET6:
1107                 sa6 = (struct sockaddr_in6 *)sa;
1108                 if (IN6_IS_ADDR_MULTICAST(&sa6->sin6_addr)) {
1109                         /* just use the scope field of the multicast address */
1110                         return(sa6->sin6_addr.s6_addr[2] & 0x0f);
1111                 }
1112                 /*
1113                  * Unicast addresses: map scope type to corresponding scope
1114                  * value defined for multcast addresses.
1115                  * XXX: hardcoded scope type values are bad...
1116                  */
1117                 if (IN6_IS_ADDR_LOOPBACK(&sa6->sin6_addr))
1118                         return(1); /* node local scope */
1119                 if (IN6_IS_ADDR_LINKLOCAL(&sa6->sin6_addr))
1120                         return(2); /* link-local scope */
1121                 if (IN6_IS_ADDR_SITELOCAL(&sa6->sin6_addr))
1122                         return(5); /* site-local scope */
1123                 return(14);     /* global scope */
1124                 break;
1125 #endif
1126         case AF_INET:
1127                 /*
1128                  * IPv4 pseudo scoping according to RFC 3484.
1129                  */
1130                 sa4 = (struct sockaddr_in *)sa;
1131                 /* IPv4 autoconfiguration addresses have link-local scope. */
1132                 if (((u_char *)&sa4->sin_addr)[0] == 169 &&
1133                     ((u_char *)&sa4->sin_addr)[1] == 254)
1134                         return(2);
1135                 /* Private addresses have site-local scope. */
1136                 if (((u_char *)&sa4->sin_addr)[0] == 10 ||
1137                     (((u_char *)&sa4->sin_addr)[0] == 172 &&
1138                      (((u_char *)&sa4->sin_addr)[1] & 0xf0) == 16) ||
1139                     (((u_char *)&sa4->sin_addr)[0] == 192 &&
1140                      ((u_char *)&sa4->sin_addr)[1] == 168))
1141                         return(14);     /* XXX: It should be 5 unless NAT */
1142                 /* Loopback addresses have link-local scope. */
1143                 if (((u_char *)&sa4->sin_addr)[0] == 127)
1144                         return(2);
1145                 return(14);
1146                 break;
1147         default:
1148                 errno = EAFNOSUPPORT; /* is this a good error? */
1149                 return(-1);
1150         }
1151 }
1152
1153 static int
1154 explore_copy(const struct addrinfo *pai, const struct addrinfo *src0,
1155     struct addrinfo **res)
1156 {
1157         int error;
1158         struct addrinfo sentinel, *cur;
1159         const struct addrinfo *src;
1160
1161         error = 0;
1162         sentinel.ai_next = NULL;
1163         cur = &sentinel;
1164
1165         for (src = src0; src != NULL; src = src->ai_next) {
1166                 if (src->ai_family != pai->ai_family)
1167                         continue;
1168
1169                 cur->ai_next = copy_ai(src);
1170                 if (!cur->ai_next) {
1171                         error = EAI_MEMORY;
1172                         goto fail;
1173                 }
1174
1175                 cur->ai_next->ai_socktype = pai->ai_socktype;
1176                 cur->ai_next->ai_protocol = pai->ai_protocol;
1177                 cur = cur->ai_next;
1178         }
1179
1180         *res = sentinel.ai_next;
1181         return 0;
1182
1183 fail:
1184         freeaddrinfo(sentinel.ai_next);
1185         return error;
1186 }
1187
1188 /*
1189  * hostname == NULL.
1190  * passive socket -> anyaddr (0.0.0.0 or ::)
1191  * non-passive socket -> localhost (127.0.0.1 or ::1)
1192  */
1193 static int
1194 explore_null(const struct addrinfo *pai, const char *servname,
1195     struct addrinfo **res)
1196 {
1197         int s;
1198         const struct afd *afd;
1199         struct addrinfo *ai;
1200         int error;
1201
1202         *res = NULL;
1203         ai = NULL;
1204
1205         if (pai->ai_family == PF_LOCAL)
1206                 return (0);
1207
1208         /*
1209          * filter out AFs that are not supported by the kernel
1210          * XXX errno?
1211          */
1212         s = _socket(pai->ai_family, SOCK_DGRAM | SOCK_CLOEXEC, 0);
1213         if (s < 0) {
1214                 if (errno != EMFILE)
1215                         return 0;
1216         } else
1217                 _close(s);
1218
1219         afd = find_afd(pai->ai_family);
1220         if (afd == NULL)
1221                 return 0;
1222
1223         if (pai->ai_flags & AI_PASSIVE) {
1224                 GET_AI(ai, afd, afd->a_addrany);
1225                 GET_PORT(ai, servname);
1226         } else {
1227                 GET_AI(ai, afd, afd->a_loopback);
1228                 GET_PORT(ai, servname);
1229         }
1230
1231         *res = ai;
1232         return 0;
1233
1234 free:
1235         if (ai != NULL)
1236                 freeaddrinfo(ai);
1237         return error;
1238 }
1239
1240 /*
1241  * numeric hostname
1242  */
1243 static int
1244 explore_numeric(const struct addrinfo *pai, const char *hostname,
1245     const char *servname, struct addrinfo **res, const char *canonname)
1246 {
1247         const struct afd *afd;
1248         struct addrinfo *ai, ai0;
1249         int error;
1250         char pton[PTON_MAX], path[PATH_MAX], *p;
1251
1252 #ifdef CTASSERT
1253         CTASSERT(sizeofmember(struct sockaddr_un, sun_path) <= PATH_MAX);
1254 #endif
1255         *res = NULL;
1256         ai = NULL;
1257
1258         afd = find_afd(pai->ai_family);
1259         if (afd == NULL)
1260                 return 0;
1261
1262         switch (afd->a_af) {
1263         case AF_LOCAL:
1264                 if (hostname[0] != '/')
1265                         ERR(EAI_NONAME);
1266                 if (strlen(hostname) > afd->a_addrlen)
1267                         ERR(EAI_MEMORY);
1268                 /* NUL-termination does not need to be guaranteed. */
1269                 strncpy(path, hostname, afd->a_addrlen);
1270                 p = &path[0];
1271                 break;
1272         case AF_INET:
1273                 /*
1274                  * RFC3493 requires getaddrinfo() to accept AF_INET formats
1275                  * that are accepted by inet_addr() and its family.  The
1276                  * accepted forms includes the "classful" one, which inet_pton
1277                  * does not accept.  So we need to separate the case for
1278                  * AF_INET.
1279                  */
1280                 if (inet_aton(hostname, (struct in_addr *)pton) != 1 ||
1281                     hostname[strspn(hostname, "0123456789.xabcdefXABCDEF")] != '\0')
1282                         return 0;
1283                 p = pton;
1284                 break;
1285         default:
1286                 if (inet_pton(afd->a_af, hostname, pton) != 1) {
1287                         if (pai->ai_family != AF_INET6 ||
1288                             (pai->ai_flags & AI_V4MAPPED) != AI_V4MAPPED)
1289                                 return 0;
1290                         if (inet_aton(hostname, (struct in_addr *)pton) != 1)
1291                                 return 0;
1292                         afd = &afdl[N_INET];
1293                         ai0 = *pai;
1294                         ai0.ai_family = AF_INET;
1295                         pai = &ai0;
1296                 }
1297                 p = pton;
1298                 break;
1299         }
1300
1301         if (pai->ai_family == afd->a_af) {
1302                 GET_AI(ai, afd, p);
1303                 GET_PORT(ai, servname);
1304                 if ((pai->ai_family == AF_INET ||
1305                      pai->ai_family == AF_INET6) &&
1306                     (pai->ai_flags & AI_CANONNAME)) {
1307                         /*
1308                          * Set the numeric address itself as the canonical
1309                          * name, based on a clarification in RFC3493.
1310                          */
1311                         GET_CANONNAME(ai, canonname);
1312                 }
1313         } else {
1314                 /*
1315                  * XXX: This should not happen since we already matched the AF
1316                  * by find_afd.
1317                  */
1318                 ERR(EAI_FAMILY);
1319         }
1320
1321         *res = ai;
1322         return 0;
1323
1324 free:
1325 bad:
1326         if (ai != NULL)
1327                 freeaddrinfo(ai);
1328         return error;
1329 }
1330
1331 /*
1332  * numeric hostname with scope
1333  */
1334 static int
1335 explore_numeric_scope(const struct addrinfo *pai, const char *hostname,
1336     const char *servname, struct addrinfo **res)
1337 {
1338 #if !defined(SCOPE_DELIMITER) || !defined(INET6)
1339         return explore_numeric(pai, hostname, servname, res, hostname);
1340 #else
1341         const struct afd *afd;
1342         struct addrinfo *cur;
1343         int error;
1344         char *cp, *hostname2 = NULL, *scope, *addr;
1345         struct sockaddr_in6 *sin6;
1346
1347         afd = find_afd(pai->ai_family);
1348         if (afd == NULL)
1349                 return 0;
1350
1351         if (!afd->a_scoped)
1352                 return explore_numeric(pai, hostname, servname, res, hostname);
1353
1354         cp = strchr(hostname, SCOPE_DELIMITER);
1355         if (cp == NULL)
1356                 return explore_numeric(pai, hostname, servname, res, hostname);
1357
1358         /*
1359          * Handle special case of <scoped_address><delimiter><scope id>
1360          */
1361         hostname2 = strdup(hostname);
1362         if (hostname2 == NULL)
1363                 return EAI_MEMORY;
1364         /* terminate at the delimiter */
1365         hostname2[cp - hostname] = '\0';
1366         addr = hostname2;
1367         scope = cp + 1;
1368
1369         error = explore_numeric(pai, addr, servname, res, hostname);
1370         if (error == 0) {
1371                 u_int32_t scopeid;
1372
1373                 for (cur = *res; cur; cur = cur->ai_next) {
1374                         if (cur->ai_family != AF_INET6)
1375                                 continue;
1376                         sin6 = (struct sockaddr_in6 *)(void *)cur->ai_addr;
1377                         if (ip6_str2scopeid(scope, sin6, &scopeid) == -1) {
1378                                 free(hostname2);
1379                                 freeaddrinfo(*res);
1380                                 *res = NULL;
1381                                 return(EAI_NONAME); /* XXX: is return OK? */
1382                         }
1383                         sin6->sin6_scope_id = scopeid;
1384                 }
1385         }
1386
1387         free(hostname2);
1388
1389         if (error && *res) {
1390                 freeaddrinfo(*res);
1391                 *res = NULL;
1392         }
1393         return error;
1394 #endif
1395 }
1396
1397 static int
1398 get_canonname(const struct addrinfo *pai, struct addrinfo *ai, const char *str)
1399 {
1400         if ((pai->ai_flags & AI_CANONNAME) != 0) {
1401                 ai->ai_canonname = strdup(str);
1402                 if (ai->ai_canonname == NULL)
1403                         return EAI_MEMORY;
1404         }
1405         return 0;
1406 }
1407
1408 static struct addrinfo *
1409 get_ai(const struct addrinfo *pai, const struct afd *afd, const char *addr)
1410 {
1411         char *p;
1412         struct addrinfo *ai;
1413 #ifdef INET6
1414         struct in6_addr mapaddr;
1415
1416         if (afd->a_af == AF_INET && (pai->ai_flags & AI_V4MAPPED) != 0) {
1417                 afd = &afdl[N_INET6];
1418                 _map_v4v6_address(addr, (char *)&mapaddr);
1419                 addr = (char *)&mapaddr;
1420         }
1421 #endif
1422
1423         ai = (struct addrinfo *)malloc(sizeof(struct addrinfo)
1424                 + (afd->a_socklen));
1425         if (ai == NULL)
1426                 return NULL;
1427
1428         memcpy(ai, pai, sizeof(struct addrinfo));
1429         ai->ai_addr = (struct sockaddr *)(void *)(ai + 1);
1430         memset(ai->ai_addr, 0, (size_t)afd->a_socklen);
1431         ai->ai_addr->sa_len = afd->a_socklen;
1432         ai->ai_addrlen = afd->a_socklen;
1433         if (ai->ai_family == PF_LOCAL) {
1434                 size_t n = strnlen(addr, afd->a_addrlen);
1435
1436                 ai->ai_addrlen -= afd->a_addrlen - n;
1437                 ai->ai_addr->sa_len -= afd->a_addrlen - n;
1438         }
1439         ai->ai_addr->sa_family = ai->ai_family = afd->a_af;
1440         p = (char *)(void *)(ai->ai_addr);
1441         memcpy(p + afd->a_off, addr, (size_t)afd->a_addrlen);
1442         return ai;
1443 }
1444
1445 /* XXX need to malloc() the same way we do from other functions! */
1446 static struct addrinfo *
1447 copy_ai(const struct addrinfo *pai)
1448 {
1449         struct addrinfo *ai;
1450         size_t l;
1451
1452         l = sizeof(*ai) + pai->ai_addrlen;
1453         if ((ai = calloc(1, l)) == NULL)
1454                 return NULL;
1455         memcpy(ai, pai, sizeof(*ai));
1456         ai->ai_addr = (struct sockaddr *)(void *)(ai + 1);
1457         memcpy(ai->ai_addr, pai->ai_addr, pai->ai_addrlen);
1458
1459         if (pai->ai_canonname) {
1460                 l = strlen(pai->ai_canonname) + 1;
1461                 if ((ai->ai_canonname = malloc(l)) == NULL) {
1462                         free(ai);
1463                         return NULL;
1464                 }
1465                 strlcpy(ai->ai_canonname, pai->ai_canonname, l);
1466         } else {
1467                 /* just to make sure */
1468                 ai->ai_canonname = NULL;
1469         }
1470
1471         ai->ai_next = NULL;
1472
1473         return ai;
1474 }
1475
1476 static int
1477 get_portmatch(const struct addrinfo *ai, const char *servname)
1478 {
1479
1480         /* get_port does not touch first argument when matchonly == 1. */
1481         /* LINTED const cast */
1482         return get_port((struct addrinfo *)ai, servname, 1);
1483 }
1484
1485 static int
1486 get_port(struct addrinfo *ai, const char *servname, int matchonly)
1487 {
1488         const char *proto;
1489         struct servent *sp;
1490         int port, error;
1491         int allownumeric;
1492
1493         if (servname == NULL)
1494                 return 0;
1495         switch (ai->ai_family) {
1496         case AF_LOCAL:
1497                 /* AF_LOCAL ignores servname silently. */
1498                 return (0);
1499         case AF_INET:
1500 #ifdef AF_INET6
1501         case AF_INET6:
1502 #endif
1503                 break;
1504         default:
1505                 return 0;
1506         }
1507
1508         switch (ai->ai_socktype) {
1509         case SOCK_RAW:
1510                 return EAI_SERVICE;
1511         case SOCK_DGRAM:
1512         case SOCK_STREAM:
1513         case SOCK_SEQPACKET:
1514                 allownumeric = 1;
1515                 break;
1516         case ANY:
1517                 switch (ai->ai_family) {
1518                 case AF_INET:
1519 #ifdef AF_INET6
1520                 case AF_INET6:
1521 #endif
1522                         allownumeric = 1;
1523                         break;
1524                 default:
1525                         allownumeric = 0;
1526                         break;
1527                 }
1528                 break;
1529         default:
1530                 return EAI_SOCKTYPE;
1531         }
1532
1533         error = str2number(servname, &port);
1534         if (error == 0) {
1535                 if (!allownumeric)
1536                         return EAI_SERVICE;
1537                 if (port < 0 || port > 65535)
1538                         return EAI_SERVICE;
1539                 port = htons(port);
1540         } else {
1541                 if (ai->ai_flags & AI_NUMERICSERV)
1542                         return EAI_NONAME;
1543
1544                 switch (ai->ai_protocol) {
1545                 case IPPROTO_UDP:
1546                         proto = "udp";
1547                         break;
1548                 case IPPROTO_TCP:
1549                         proto = "tcp";
1550                         break;
1551                 case IPPROTO_SCTP:
1552                         proto = "sctp";
1553                         break;
1554                 case IPPROTO_UDPLITE:
1555                         proto = "udplite";
1556                         break;
1557                 default:
1558                         proto = NULL;
1559                         break;
1560                 }
1561
1562                 if ((sp = getservbyname(servname, proto)) == NULL)
1563                         return EAI_SERVICE;
1564                 port = sp->s_port;
1565         }
1566
1567         if (!matchonly) {
1568                 switch (ai->ai_family) {
1569                 case AF_INET:
1570                         ((struct sockaddr_in *)(void *)
1571                             ai->ai_addr)->sin_port = port;
1572                         break;
1573 #ifdef INET6
1574                 case AF_INET6:
1575                         ((struct sockaddr_in6 *)(void *)
1576                             ai->ai_addr)->sin6_port = port;
1577                         break;
1578 #endif
1579                 }
1580         }
1581
1582         return 0;
1583 }
1584
1585 static const struct afd *
1586 find_afd(int af)
1587 {
1588         const struct afd *afd;
1589
1590         if (af == PF_UNSPEC)
1591                 return NULL;
1592         for (afd = afdl; afd->a_af; afd++) {
1593                 if (afd->a_af == af)
1594                         return afd;
1595         }
1596         return NULL;
1597 }
1598
1599 /*
1600  * RFC 3493: AI_ADDRCONFIG check.  Determines which address families are
1601  * configured on the local system and correlates with pai->ai_family value.
1602  * If an address family is not configured on the system, it will not be
1603  * queried for.  For this purpose, loopback addresses are not considered
1604  * configured addresses.
1605  *
1606  * XXX PF_UNSPEC -> PF_INET6 + PF_INET mapping needs to be in sync with
1607  * _dns_getaddrinfo.
1608  */
1609 static int
1610 addrconfig(struct addrinfo *pai)
1611 {
1612         struct ifaddrs *ifaddrs, *ifa;
1613         struct sockaddr_in *sin;
1614 #ifdef INET6
1615         struct sockaddr_in6 *sin6;
1616 #endif
1617         int seen_inet = 0, seen_inet6 = 0;
1618
1619         if (getifaddrs(&ifaddrs) != 0)
1620                 return (0);
1621
1622         for (ifa = ifaddrs; ifa != NULL; ifa = ifa->ifa_next) {
1623                 if (ifa->ifa_addr == NULL || (ifa->ifa_flags & IFF_UP) == 0)
1624                         continue;
1625                 switch (ifa->ifa_addr->sa_family) {
1626                 case AF_INET:
1627                         if (seen_inet)
1628                                 continue;
1629                         sin = (struct sockaddr_in *)(ifa->ifa_addr);
1630                         if (htonl(sin->sin_addr.s_addr) == INADDR_LOOPBACK)
1631                                 continue;
1632                         seen_inet = 1;
1633                         break;
1634 #ifdef INET6
1635                 case AF_INET6:
1636                         if (seen_inet6)
1637                                 continue;
1638                         sin6 = (struct sockaddr_in6 *)(ifa->ifa_addr);
1639                         if (IN6_IS_ADDR_LOOPBACK(&sin6->sin6_addr))
1640                                 continue;
1641                         if ((ifa->ifa_flags & IFT_LOOP) != 0 &&
1642                             IN6_IS_ADDR_LINKLOCAL(&sin6->sin6_addr))
1643                                 continue;
1644                         if (is_ifdisabled(ifa->ifa_name))
1645                                 continue;
1646                         seen_inet6 = 1;
1647                         break;
1648 #endif
1649                 }
1650         }
1651         freeifaddrs(ifaddrs);
1652
1653         switch(pai->ai_family) {
1654         case AF_INET6:
1655                 return (seen_inet6);
1656         case AF_INET:
1657                 return (seen_inet);
1658         case AF_UNSPEC:
1659                 if (seen_inet == seen_inet6)
1660                         return (seen_inet);
1661                 pai->ai_family = seen_inet ? AF_INET : AF_INET6;
1662                 return (1);
1663         }
1664         return (1);
1665 }
1666
1667 #ifdef INET6
1668 static int
1669 is_ifdisabled(char *name)
1670 {
1671         struct in6_ndireq nd;
1672         int fd;
1673
1674         if ((fd = _socket(AF_INET6, SOCK_DGRAM | SOCK_CLOEXEC, 0)) < 0)
1675                 return (-1);
1676         memset(&nd, 0, sizeof(nd));
1677         strlcpy(nd.ifname, name, sizeof(nd.ifname));
1678         if (_ioctl(fd, SIOCGIFINFO_IN6, &nd) < 0) {
1679                 _close(fd);
1680                 return (-1);
1681         }
1682         _close(fd);
1683         return ((nd.ndi.flags & ND6_IFF_IFDISABLED) != 0);
1684 }
1685
1686 /* convert a string to a scope identifier. XXX: IPv6 specific */
1687 static int
1688 ip6_str2scopeid(char *scope, struct sockaddr_in6 *sin6, u_int32_t *scopeid)
1689 {
1690         u_long lscopeid;
1691         struct in6_addr *a6;
1692         char *ep;
1693
1694         a6 = &sin6->sin6_addr;
1695
1696         /* empty scopeid portion is invalid */
1697         if (*scope == '\0')
1698                 return -1;
1699
1700         if (IN6_IS_ADDR_LINKLOCAL(a6) || IN6_IS_ADDR_MC_LINKLOCAL(a6) ||
1701             IN6_IS_ADDR_MC_NODELOCAL(a6)) {
1702                 /*
1703                  * We currently assume a one-to-one mapping between links
1704                  * and interfaces, so we simply use interface indices for
1705                  * like-local scopes.
1706                  */
1707                 *scopeid = if_nametoindex(scope);
1708                 if (*scopeid == 0)
1709                         goto trynumeric;
1710                 return 0;
1711         }
1712
1713         /* still unclear about literal, allow numeric only - placeholder */
1714         if (IN6_IS_ADDR_SITELOCAL(a6) || IN6_IS_ADDR_MC_SITELOCAL(a6))
1715                 goto trynumeric;
1716         if (IN6_IS_ADDR_MC_ORGLOCAL(a6))
1717                 goto trynumeric;
1718         else
1719                 goto trynumeric;        /* global */
1720
1721         /* try to convert to a numeric id as a last resort */
1722   trynumeric:
1723         errno = 0;
1724         lscopeid = strtoul(scope, &ep, 10);
1725         *scopeid = (u_int32_t)(lscopeid & 0xffffffffUL);
1726         if (errno == 0 && ep && *ep == '\0' && *scopeid == lscopeid)
1727                 return 0;
1728         else
1729                 return -1;
1730 }
1731 #endif
1732
1733
1734 #ifdef NS_CACHING
1735 static int
1736 addrinfo_id_func(char *buffer, size_t *buffer_size, va_list ap,
1737     void *cache_mdata)
1738 {
1739         res_state statp;
1740         u_long res_options;
1741
1742         const int op_id = 0;    /* identifies the getaddrinfo for the cache */
1743         char *hostname;
1744         struct addrinfo *hints;
1745
1746         char *p;
1747         int ai_flags, ai_family, ai_socktype, ai_protocol;
1748         size_t desired_size, size;
1749
1750         statp = __res_state();
1751         res_options = statp->options & (RES_RECURSE | RES_DEFNAMES |
1752             RES_DNSRCH | RES_NOALIASES | RES_USE_INET6);
1753
1754         hostname = va_arg(ap, char *);
1755         hints = va_arg(ap, struct addrinfo *);
1756
1757         desired_size = sizeof(res_options) + sizeof(int) + sizeof(int) * 4;
1758         if (hostname != NULL) {
1759                 size = strlen(hostname);
1760                 desired_size += size + 1;
1761         } else
1762                 size = 0;
1763
1764         if (desired_size > *buffer_size) {
1765                 *buffer_size = desired_size;
1766                 return (NS_RETURN);
1767         }
1768
1769         if (hints == NULL)
1770                 ai_flags = ai_family = ai_socktype = ai_protocol = 0;
1771         else {
1772                 ai_flags = hints->ai_flags;
1773                 ai_family = hints->ai_family;
1774                 ai_socktype = hints->ai_socktype;
1775                 ai_protocol = hints->ai_protocol;
1776         }
1777
1778         p = buffer;
1779         memcpy(p, &res_options, sizeof(res_options));
1780         p += sizeof(res_options);
1781
1782         memcpy(p, &op_id, sizeof(int));
1783         p += sizeof(int);
1784
1785         memcpy(p, &ai_flags, sizeof(int));
1786         p += sizeof(int);
1787
1788         memcpy(p, &ai_family, sizeof(int));
1789         p += sizeof(int);
1790
1791         memcpy(p, &ai_socktype, sizeof(int));
1792         p += sizeof(int);
1793
1794         memcpy(p, &ai_protocol, sizeof(int));
1795         p += sizeof(int);
1796
1797         if (hostname != NULL)
1798                 memcpy(p, hostname, size);
1799
1800         *buffer_size = desired_size;
1801         return (NS_SUCCESS);
1802 }
1803
1804 static int
1805 addrinfo_marshal_func(char *buffer, size_t *buffer_size, void *retval,
1806     va_list ap, void *cache_mdata)
1807 {
1808         struct addrinfo *ai, *cai;
1809         char *p;
1810         size_t desired_size, size, ai_size;
1811
1812         ai = *((struct addrinfo **)retval);
1813
1814         desired_size = sizeof(size_t);
1815         ai_size = 0;
1816         for (cai = ai; cai != NULL; cai = cai->ai_next) {
1817                 desired_size += sizeof(struct addrinfo) + cai->ai_addrlen;
1818                 if (cai->ai_canonname != NULL)
1819                         desired_size += sizeof(size_t) +
1820                             strlen(cai->ai_canonname);
1821                 ++ai_size;
1822         }
1823
1824         if (desired_size > *buffer_size) {
1825                 /* this assignment is here for future use */
1826                 errno = ERANGE;
1827                 *buffer_size = desired_size;
1828                 return (NS_RETURN);
1829         }
1830
1831         memset(buffer, 0, desired_size);
1832         p = buffer;
1833
1834         memcpy(p, &ai_size, sizeof(size_t));
1835         p += sizeof(size_t);
1836         for (cai = ai; cai != NULL; cai = cai->ai_next) {
1837                 memcpy(p, cai, sizeof(struct addrinfo));
1838                 p += sizeof(struct addrinfo);
1839
1840                 memcpy(p, cai->ai_addr, cai->ai_addrlen);
1841                 p += cai->ai_addrlen;
1842
1843                 if (cai->ai_canonname != NULL) {
1844                         size = strlen(cai->ai_canonname);
1845                         memcpy(p, &size, sizeof(size_t));
1846                         p += sizeof(size_t);
1847
1848                         memcpy(p, cai->ai_canonname, size);
1849                         p += size;
1850                 }
1851         }
1852
1853         return (NS_SUCCESS);
1854 }
1855
1856 static int
1857 addrinfo_unmarshal_func(char *buffer, size_t buffer_size, void *retval,
1858     va_list ap, void *cache_mdata)
1859 {
1860         struct addrinfo new_ai, *result, *sentinel, *lasts;
1861
1862         char *p;
1863         size_t ai_size, ai_i, size;
1864
1865         p = buffer;
1866         memcpy(&ai_size, p, sizeof(size_t));
1867         p += sizeof(size_t);
1868
1869         result = NULL;
1870         lasts = NULL;
1871         for (ai_i = 0; ai_i < ai_size; ++ai_i) {
1872                 memcpy(&new_ai, p, sizeof(struct addrinfo));
1873                 p += sizeof(struct addrinfo);
1874                 size = new_ai.ai_addrlen + sizeof(struct addrinfo) +
1875                         _ALIGNBYTES;
1876
1877                 sentinel = calloc(1, size);
1878
1879                 memcpy(sentinel, &new_ai, sizeof(struct addrinfo));
1880                 sentinel->ai_addr = (struct sockaddr *)_ALIGN((char *)sentinel +
1881                     sizeof(struct addrinfo));
1882
1883                 memcpy(sentinel->ai_addr, p, new_ai.ai_addrlen);
1884                 p += new_ai.ai_addrlen;
1885
1886                 if (new_ai.ai_canonname != NULL) {
1887                         memcpy(&size, p, sizeof(size_t));
1888                         p += sizeof(size_t);
1889
1890                         sentinel->ai_canonname = calloc(1, size + 1);
1891
1892                         memcpy(sentinel->ai_canonname, p, size);
1893                         p += size;
1894                 }
1895
1896                 if (result == NULL) {
1897                         result = sentinel;
1898                         lasts = sentinel;
1899                 } else {
1900                         lasts->ai_next = sentinel;
1901                         lasts = sentinel;
1902                 }
1903         }
1904
1905         *((struct addrinfo **)retval) = result;
1906         return (NS_SUCCESS);
1907 }
1908 #endif /* NS_CACHING */
1909
1910 /*
1911  * FQDN hostname, DNS lookup
1912  */
1913 static int
1914 explore_fqdn(const struct addrinfo *pai, const char *hostname,
1915     const char *servname, struct addrinfo **res)
1916 {
1917         struct addrinfo *result;
1918         struct addrinfo *cur;
1919         int error = 0;
1920
1921 #ifdef NS_CACHING
1922         static const nss_cache_info cache_info =
1923         NS_COMMON_CACHE_INFO_INITIALIZER(
1924                 hosts, NULL, addrinfo_id_func, addrinfo_marshal_func,
1925                 addrinfo_unmarshal_func);
1926 #endif
1927         static const ns_dtab dtab[] = {
1928                 NS_FILES_CB(_files_getaddrinfo, NULL)
1929                 { NSSRC_DNS, _dns_getaddrinfo, NULL },  /* force -DHESIOD */
1930                 NS_NIS_CB(_yp_getaddrinfo, NULL)
1931 #ifdef NS_CACHING
1932                 NS_CACHE_CB(&cache_info)
1933 #endif
1934                 { 0 }
1935         };
1936
1937         result = NULL;
1938
1939         /*
1940          * if the servname does not match socktype/protocol, ignore it.
1941          */
1942         if (get_portmatch(pai, servname) != 0)
1943                 return 0;
1944
1945         switch (_nsdispatch(&result, dtab, NSDB_HOSTS, "getaddrinfo",
1946                         default_dns_files, hostname, pai)) {
1947         case NS_TRYAGAIN:
1948                 error = EAI_AGAIN;
1949                 goto free;
1950         case NS_UNAVAIL:
1951                 error = EAI_FAIL;
1952                 goto free;
1953         case NS_NOTFOUND:
1954                 error = EAI_NONAME;
1955                 goto free;
1956         case NS_ADDRFAMILY:
1957                 error = EAI_ADDRFAMILY;
1958                 goto free;
1959         case NS_SUCCESS:
1960                 error = 0;
1961                 for (cur = result; cur; cur = cur->ai_next) {
1962                         GET_PORT(cur, servname);
1963                         /* canonname should be filled already */
1964                 }
1965                 break;
1966         }
1967
1968         *res = result;
1969
1970         return 0;
1971
1972 free:
1973         if (result)
1974                 freeaddrinfo(result);
1975         return error;
1976 }
1977
1978 #ifdef DEBUG
1979 static const char AskedForGot[] =
1980         "gethostby*.getanswer: asked for \"%s\", got \"%s\"";
1981 #endif
1982
1983 static struct addrinfo *
1984 getanswer(const querybuf *answer, int anslen, const char *qname, int qtype,
1985     const struct addrinfo *pai, res_state res)
1986 {
1987         struct addrinfo sentinel, *cur;
1988         struct addrinfo ai;
1989         const struct afd *afd;
1990         char *canonname;
1991         const HEADER *hp;
1992         const u_char *cp;
1993         int n;
1994         const u_char *eom;
1995         char *bp, *ep;
1996         int type, class, ancount, qdcount;
1997         int haveanswer, had_error;
1998         char tbuf[MAXDNAME];
1999         int (*name_ok)(const char *);
2000         char hostbuf[8*1024];
2001
2002         memset(&sentinel, 0, sizeof(sentinel));
2003         cur = &sentinel;
2004
2005         canonname = NULL;
2006         eom = answer->buf + anslen;
2007         switch (qtype) {
2008         case T_A:
2009         case T_AAAA:
2010         case T_ANY:     /*use T_ANY only for T_A/T_AAAA lookup*/
2011                 name_ok = res_hnok;
2012                 break;
2013         default:
2014                 return (NULL);  /* XXX should be abort(); */
2015         }
2016         /*
2017          * find first satisfactory answer
2018          */
2019         hp = &answer->hdr;
2020         ancount = ntohs(hp->ancount);
2021         qdcount = ntohs(hp->qdcount);
2022         bp = hostbuf;
2023         ep = hostbuf + sizeof hostbuf;
2024         cp = answer->buf + HFIXEDSZ;
2025         if (qdcount != 1) {
2026                 RES_SET_H_ERRNO(res, NO_RECOVERY);
2027                 return (NULL);
2028         }
2029         n = dn_expand(answer->buf, eom, cp, bp, ep - bp);
2030         if ((n < 0) || !(*name_ok)(bp)) {
2031                 RES_SET_H_ERRNO(res, NO_RECOVERY);
2032                 return (NULL);
2033         }
2034         cp += n + QFIXEDSZ;
2035         if (qtype == T_A || qtype == T_AAAA || qtype == T_ANY) {
2036                 /* res_send() has already verified that the query name is the
2037                  * same as the one we sent; this just gets the expanded name
2038                  * (i.e., with the succeeding search-domain tacked on).
2039                  */
2040                 n = strlen(bp) + 1;             /* for the \0 */
2041                 if (n >= MAXHOSTNAMELEN) {
2042                         RES_SET_H_ERRNO(res, NO_RECOVERY);
2043                         return (NULL);
2044                 }
2045                 canonname = bp;
2046                 bp += n;
2047                 /* The qname can be abbreviated, but h_name is now absolute. */
2048                 qname = canonname;
2049         }
2050         haveanswer = 0;
2051         had_error = 0;
2052         while (ancount-- > 0 && cp < eom && !had_error) {
2053                 n = dn_expand(answer->buf, eom, cp, bp, ep - bp);
2054                 if ((n < 0) || !(*name_ok)(bp)) {
2055                         had_error++;
2056                         continue;
2057                 }
2058                 cp += n;                        /* name */
2059                 type = _getshort(cp);
2060                 cp += INT16SZ;                  /* type */
2061                 class = _getshort(cp);
2062                 cp += INT16SZ + INT32SZ;        /* class, TTL */
2063                 n = _getshort(cp);
2064                 cp += INT16SZ;                  /* len */
2065                 if (class != C_IN) {
2066                         /* XXX - debug? syslog? */
2067                         cp += n;
2068                         continue;               /* XXX - had_error++ ? */
2069                 }
2070                 if ((qtype == T_A || qtype == T_AAAA || qtype == T_ANY) &&
2071                     type == T_CNAME) {
2072                         n = dn_expand(answer->buf, eom, cp, tbuf, sizeof tbuf);
2073                         if ((n < 0) || !(*name_ok)(tbuf)) {
2074                                 had_error++;
2075                                 continue;
2076                         }
2077                         cp += n;
2078                         /* Get canonical name. */
2079                         n = strlen(tbuf) + 1;   /* for the \0 */
2080                         if (n > ep - bp || n >= MAXHOSTNAMELEN) {
2081                                 had_error++;
2082                                 continue;
2083                         }
2084                         strlcpy(bp, tbuf, ep - bp);
2085                         canonname = bp;
2086                         bp += n;
2087                         continue;
2088                 }
2089                 if (qtype == T_ANY) {
2090                         if (!(type == T_A || type == T_AAAA)) {
2091                                 cp += n;
2092                                 continue;
2093                         }
2094                 } else if (type != qtype) {
2095 #ifdef DEBUG
2096                         if (type != T_KEY && type != T_SIG &&
2097                             type != T_DNAME && type != T_RRSIG)
2098                                 syslog(LOG_NOTICE|LOG_AUTH,
2099                "gethostby*.getanswer: asked for \"%s %s %s\", got type \"%s\"",
2100                                        qname, p_class(C_IN), p_type(qtype),
2101                                        p_type(type));
2102 #endif
2103                         cp += n;
2104                         continue;               /* XXX - had_error++ ? */
2105                 }
2106                 switch (type) {
2107                 case T_A:
2108                 case T_AAAA:
2109                         if (strcasecmp(canonname, bp) != 0) {
2110 #ifdef DEBUG
2111                                 syslog(LOG_NOTICE|LOG_AUTH,
2112                                        AskedForGot, canonname, bp);
2113 #endif
2114                                 cp += n;
2115                                 continue;       /* XXX - had_error++ ? */
2116                         }
2117                         if (type == T_A && n != INADDRSZ) {
2118                                 cp += n;
2119                                 continue;
2120                         }
2121                         if (type == T_AAAA && n != IN6ADDRSZ) {
2122                                 cp += n;
2123                                 continue;
2124                         }
2125 #ifdef FILTER_V4MAPPED
2126                         if (type == T_AAAA) {
2127                                 struct in6_addr in6;
2128                                 memcpy(&in6, cp, sizeof(in6));
2129                                 if (IN6_IS_ADDR_V4MAPPED(&in6)) {
2130                                         cp += n;
2131                                         continue;
2132                                 }
2133                         }
2134 #endif
2135                         if (!haveanswer) {
2136                                 int nn;
2137
2138                                 canonname = bp;
2139                                 nn = strlen(bp) + 1;    /* for the \0 */
2140                                 bp += nn;
2141                         }
2142
2143                         /* don't overwrite pai */
2144                         ai = *pai;
2145                         ai.ai_family = (type == T_A) ? AF_INET : AF_INET6;
2146                         afd = find_afd(ai.ai_family);
2147                         if (afd == NULL) {
2148                                 cp += n;
2149                                 continue;
2150                         }
2151                         cur->ai_next = get_ai(&ai, afd, (const char *)cp);
2152                         if (cur->ai_next == NULL)
2153                                 had_error++;
2154                         while (cur && cur->ai_next)
2155                                 cur = cur->ai_next;
2156                         cp += n;
2157                         break;
2158                 default:
2159                         abort();
2160                 }
2161                 if (!had_error)
2162                         haveanswer++;
2163         }
2164         if (haveanswer) {
2165 #if defined(RESOLVSORT)
2166                 /*
2167                  * We support only IPv4 address for backward
2168                  * compatibility against gethostbyname(3).
2169                  */
2170                 if (res->nsort && qtype == T_A) {
2171                         if (addr4sort(&sentinel, res) < 0) {
2172                                 freeaddrinfo(sentinel.ai_next);
2173                                 RES_SET_H_ERRNO(res, NO_RECOVERY);
2174                                 return NULL;
2175                         }
2176                 }
2177 #endif /*RESOLVSORT*/
2178                 if (!canonname)
2179                         (void)get_canonname(pai, sentinel.ai_next, qname);
2180                 else
2181                         (void)get_canonname(pai, sentinel.ai_next, canonname);
2182                 RES_SET_H_ERRNO(res, NETDB_SUCCESS);
2183                 return sentinel.ai_next;
2184         }
2185
2186         /*
2187          * We could have walked a CNAME chain, but the ultimate target
2188          * may not have what we looked for.
2189          */
2190         RES_SET_H_ERRNO(res, ntohs(hp->ancount) > 0 ? NO_DATA : NO_RECOVERY);
2191         return NULL;
2192 }
2193
2194 #ifdef RESOLVSORT
2195 struct addr_ptr {
2196         struct addrinfo *ai;
2197         int aval;
2198 };
2199
2200 static int
2201 addr4sort(struct addrinfo *sentinel, res_state res)
2202 {
2203         struct addrinfo *ai;
2204         struct addr_ptr *addrs, addr;
2205         struct sockaddr_in *sin;
2206         int naddrs, i, j;
2207         int needsort = 0;
2208
2209         if (!sentinel)
2210                 return -1;
2211         naddrs = 0;
2212         for (ai = sentinel->ai_next; ai; ai = ai->ai_next)
2213                 naddrs++;
2214         if (naddrs < 2)
2215                 return 0;               /* We don't need sorting. */
2216         if ((addrs = malloc(sizeof(struct addr_ptr) * naddrs)) == NULL)
2217                 return -1;
2218         i = 0;
2219         for (ai = sentinel->ai_next; ai; ai = ai->ai_next) {
2220                 sin = (struct sockaddr_in *)ai->ai_addr;
2221                 for (j = 0; (unsigned)j < res->nsort; j++) {
2222                         if (res->sort_list[j].addr.s_addr ==
2223                             (sin->sin_addr.s_addr & res->sort_list[j].mask))
2224                                 break;
2225                 }
2226                 addrs[i].ai = ai;
2227                 addrs[i].aval = j;
2228                 if (needsort == 0 && i > 0 && j < addrs[i - 1].aval)
2229                         needsort = i;
2230                 i++;
2231         }
2232         if (!needsort) {
2233                 free(addrs);
2234                 return 0;
2235         }
2236
2237         while (needsort < naddrs) {
2238                 for (j = needsort - 1; j >= 0; j--) {
2239                         if (addrs[j].aval > addrs[j+1].aval) {
2240                                 addr = addrs[j];
2241                                 addrs[j] = addrs[j + 1];
2242                                 addrs[j + 1] = addr;
2243                         } else
2244                                 break;
2245                 }
2246                 needsort++;
2247         }
2248
2249         ai = sentinel;
2250         for (i = 0; i < naddrs; ++i) {
2251                 ai->ai_next = addrs[i].ai;
2252                 ai = ai->ai_next;
2253         }
2254         ai->ai_next = NULL;
2255         free(addrs);
2256         return 0;
2257 }
2258 #endif /*RESOLVSORT*/
2259
2260 /*ARGSUSED*/
2261 static int
2262 _dns_getaddrinfo(void *rv, void *cb_data, va_list ap)
2263 {
2264         struct addrinfo *ai, ai0;
2265         querybuf *buf, *buf2;
2266         const char *hostname;
2267         const struct addrinfo *pai;
2268         struct addrinfo sentinel, *cur;
2269         struct res_target q, q2;
2270         res_state res;
2271
2272         ai = NULL;
2273
2274         hostname = va_arg(ap, char *);
2275         pai = va_arg(ap, const struct addrinfo *);
2276
2277         memset(&q, 0, sizeof(q));
2278         memset(&q2, 0, sizeof(q2));
2279         memset(&sentinel, 0, sizeof(sentinel));
2280         cur = &sentinel;
2281
2282         res = __res_state();
2283
2284         buf = malloc(sizeof(*buf));
2285         if (!buf) {
2286                 RES_SET_H_ERRNO(res, NETDB_INTERNAL);
2287                 return NS_NOTFOUND;
2288         }
2289         buf2 = malloc(sizeof(*buf2));
2290         if (!buf2) {
2291                 free(buf);
2292                 RES_SET_H_ERRNO(res, NETDB_INTERNAL);
2293                 return NS_NOTFOUND;
2294         }
2295
2296         if (pai->ai_family == AF_INET6 &&
2297             (pai->ai_flags & AI_V4MAPPED) == AI_V4MAPPED) {
2298                 ai0 = *pai;
2299                 ai0.ai_family = AF_UNSPEC;
2300                 pai = &ai0;
2301         }
2302
2303         switch (pai->ai_family) {
2304         case AF_UNSPEC:
2305                 q.name = hostname;
2306                 q.qclass = C_IN;
2307                 q.qtype = T_A;
2308                 q.answer = buf->buf;
2309                 q.anslen = sizeof(buf->buf);
2310                 q.next = &q2;
2311                 q2.name = hostname;
2312                 q2.qclass = C_IN;
2313                 q2.qtype = T_AAAA;
2314                 q2.answer = buf2->buf;
2315                 q2.anslen = sizeof(buf2->buf);
2316                 break;
2317         case AF_INET:
2318                 q.name = hostname;
2319                 q.qclass = C_IN;
2320                 q.qtype = T_A;
2321                 q.answer = buf->buf;
2322                 q.anslen = sizeof(buf->buf);
2323                 break;
2324         case AF_INET6:
2325                 q.name = hostname;
2326                 q.qclass = C_IN;
2327                 q.qtype = T_AAAA;
2328                 q.answer = buf->buf;
2329                 q.anslen = sizeof(buf->buf);
2330                 break;
2331         default:
2332                 free(buf);
2333                 free(buf2);
2334                 return NS_UNAVAIL;
2335         }
2336
2337         if ((res->options & RES_INIT) == 0 && res_ninit(res) == -1) {
2338                 RES_SET_H_ERRNO(res, NETDB_INTERNAL);
2339                 free(buf);
2340                 free(buf2);
2341                 return NS_NOTFOUND;
2342         }
2343
2344         if (res_searchN(hostname, &q, res) < 0) {
2345                 free(buf);
2346                 free(buf2);
2347                 if (res->res_h_errno == NO_DATA)
2348                         return (NS_ADDRFAMILY);
2349                 return (NS_NOTFOUND);
2350         }
2351         /* prefer IPv6 */
2352         if (q.next) {
2353                 ai = getanswer(buf2, q2.n, q2.name, q2.qtype, pai, res);
2354                 if (ai != NULL) {
2355                         cur->ai_next = ai;
2356                         while (cur && cur->ai_next)
2357                                 cur = cur->ai_next;
2358                 }
2359         }
2360         if (ai == NULL || pai->ai_family != AF_UNSPEC ||
2361             (pai->ai_flags & (AI_ALL | AI_V4MAPPED)) != AI_V4MAPPED) {
2362                 ai = getanswer(buf, q.n, q.name, q.qtype, pai, res);
2363                 if (ai != NULL)
2364                         cur->ai_next = ai;
2365         }
2366         free(buf);
2367         free(buf2);
2368         if (sentinel.ai_next == NULL)
2369                 switch (res->res_h_errno) {
2370                 case HOST_NOT_FOUND:
2371                         return (NS_NOTFOUND);
2372                 case NO_DATA:
2373                         return (NS_ADDRFAMILY);
2374                 case TRY_AGAIN:
2375                         return (NS_TRYAGAIN);
2376                 default:
2377                         return (NS_UNAVAIL);
2378                 }
2379         *((struct addrinfo **)rv) = sentinel.ai_next;
2380         return (NS_SUCCESS);
2381 }
2382
2383 static void
2384 _sethtent(FILE **hostf)
2385 {
2386         if (!*hostf)
2387                 *hostf = fopen(_PATH_HOSTS, "re");
2388         else
2389                 rewind(*hostf);
2390 }
2391
2392 static void
2393 _endhtent(FILE **hostf)
2394 {
2395         if (*hostf) {
2396                 (void) fclose(*hostf);
2397                 *hostf = NULL;
2398         }
2399 }
2400
2401 static struct addrinfo *
2402 _gethtent(FILE **hostf, const char *name, const struct addrinfo *pai)
2403 {
2404         char *p;
2405         char *cp, *tname, *cname;
2406         struct addrinfo hints, *res0, *res;
2407         int error;
2408         const char *addr;
2409         char hostbuf[8*1024];
2410
2411         if (!*hostf && !(*hostf = fopen(_PATH_HOSTS, "re")))
2412                 return (NULL);
2413 again:
2414         if (!(p = fgets(hostbuf, sizeof hostbuf, *hostf)))
2415                 return (NULL);
2416         if (*p == '#')
2417                 goto again;
2418         cp = strpbrk(p, "#\n");
2419         if (cp != NULL)
2420                 *cp = '\0';
2421         if (!(cp = strpbrk(p, " \t")))
2422                 goto again;
2423         *cp++ = '\0';
2424         addr = p;
2425         cname = NULL;
2426         /* if this is not something we're looking for, skip it. */
2427         while (cp && *cp) {
2428                 if (*cp == ' ' || *cp == '\t') {
2429                         cp++;
2430                         continue;
2431                 }
2432                 tname = cp;
2433                 if (cname == NULL)
2434                         cname = cp;
2435                 if ((cp = strpbrk(cp, " \t")) != NULL)
2436                         *cp++ = '\0';
2437                 if (strcasecmp(name, tname) == 0)
2438                         goto found;
2439         }
2440         goto again;
2441
2442 found:
2443         /* we should not glob socktype/protocol here */
2444         memset(&hints, 0, sizeof(hints));
2445         hints.ai_family = pai->ai_family;
2446         hints.ai_socktype = SOCK_DGRAM;
2447         hints.ai_protocol = 0;
2448         hints.ai_flags = AI_NUMERICHOST;
2449         if (pai->ai_family == AF_INET6 &&
2450             (pai->ai_flags & AI_V4MAPPED) == AI_V4MAPPED)
2451                 hints.ai_flags |= AI_V4MAPPED;
2452         error = getaddrinfo(addr, "0", &hints, &res0);
2453         if (error)
2454                 goto again;
2455 #ifdef FILTER_V4MAPPED
2456         /* XXX should check all items in the chain */
2457         if (res0->ai_family == AF_INET6 &&
2458             IN6_IS_ADDR_V4MAPPED(&((struct sockaddr_in6 *)res0->ai_addr)->sin6_addr)) {
2459                 freeaddrinfo(res0);
2460                 goto again;
2461         }
2462 #endif
2463         for (res = res0; res; res = res->ai_next) {
2464                 /* cover it up */
2465                 res->ai_flags = pai->ai_flags;
2466                 res->ai_socktype = pai->ai_socktype;
2467                 res->ai_protocol = pai->ai_protocol;
2468
2469                 if (pai->ai_flags & AI_CANONNAME) {
2470                         if (get_canonname(pai, res, cname) != 0) {
2471                                 freeaddrinfo(res0);
2472                                 goto again;
2473                         }
2474                 }
2475         }
2476         return res0;
2477 }
2478
2479 static struct addrinfo *
2480 _getht(FILE **hostf, const char *name, const struct addrinfo *pai,
2481      struct addrinfo *cur)
2482 {
2483         struct addrinfo *p;
2484
2485         while ((p = _gethtent(hostf, name, pai)) != NULL) {
2486                 cur->ai_next = p;
2487                 while (cur && cur->ai_next)
2488                         cur = cur->ai_next;
2489         }
2490         return (cur);
2491 }
2492
2493 /*ARGSUSED*/
2494 static int
2495 _files_getaddrinfo(void *rv, void *cb_data, va_list ap)
2496 {
2497         const char *name;
2498         const struct addrinfo *pai;
2499         struct addrinfo sentinel, *cur;
2500         FILE *hostf = NULL;
2501
2502         name = va_arg(ap, char *);
2503         pai = va_arg(ap, struct addrinfo *);
2504
2505         memset(&sentinel, 0, sizeof(sentinel));
2506         cur = &sentinel;
2507
2508         _sethtent(&hostf);
2509         if (pai->ai_family == AF_INET6 &&
2510             (pai->ai_flags & (AI_ALL | AI_V4MAPPED)) == AI_V4MAPPED) {
2511                 struct addrinfo ai0 = *pai;
2512
2513                 ai0.ai_flags &= ~AI_V4MAPPED;
2514                 cur = _getht(&hostf, name, &ai0, cur);
2515                 if (sentinel.ai_next == NULL) {
2516                         _sethtent(&hostf);
2517                         ai0.ai_flags |= AI_V4MAPPED;
2518                         cur = _getht(&hostf, name, &ai0, cur);
2519                 }
2520         } else
2521                 cur = _getht(&hostf, name, pai, cur);
2522         _endhtent(&hostf);
2523
2524         *((struct addrinfo **)rv) = sentinel.ai_next;
2525         if (sentinel.ai_next == NULL)
2526                 return NS_NOTFOUND;
2527         return NS_SUCCESS;
2528 }
2529
2530 #ifdef YP
2531 /*ARGSUSED*/
2532 static struct addrinfo *
2533 _yphostent(char *line, const struct addrinfo *pai)
2534 {
2535         struct addrinfo sentinel, *cur;
2536         struct addrinfo hints, *res, *res0;
2537         int error;
2538         char *p = line;
2539         const char *addr, *canonname;
2540         char *nextline;
2541         char *cp;
2542
2543         addr = canonname = NULL;
2544
2545         memset(&sentinel, 0, sizeof(sentinel));
2546         cur = &sentinel;
2547
2548 nextline:
2549         /* terminate line */
2550         cp = strchr(p, '\n');
2551         if (cp) {
2552                 *cp++ = '\0';
2553                 nextline = cp;
2554         } else
2555                 nextline = NULL;
2556
2557         cp = strpbrk(p, " \t");
2558         if (cp == NULL) {
2559                 if (canonname == NULL)
2560                         return (NULL);
2561                 else
2562                         goto done;
2563         }
2564         *cp++ = '\0';
2565
2566         addr = p;
2567
2568         while (cp && *cp) {
2569                 if (*cp == ' ' || *cp == '\t') {
2570                         cp++;
2571                         continue;
2572                 }
2573                 if (!canonname)
2574                         canonname = cp;
2575                 if ((cp = strpbrk(cp, " \t")) != NULL)
2576                         *cp++ = '\0';
2577         }
2578
2579         hints = *pai;
2580         hints.ai_flags = AI_NUMERICHOST;
2581         if (pai->ai_family == AF_INET6 &&
2582             (pai->ai_flags & AI_V4MAPPED) == AI_V4MAPPED)
2583                 hints.ai_flags |= AI_V4MAPPED;
2584         error = getaddrinfo(addr, NULL, &hints, &res0);
2585         if (error == 0) {
2586                 for (res = res0; res; res = res->ai_next) {
2587                         /* cover it up */
2588                         res->ai_flags = pai->ai_flags;
2589
2590                         if (pai->ai_flags & AI_CANONNAME)
2591                                 (void)get_canonname(pai, res, canonname);
2592                 }
2593         } else
2594                 res0 = NULL;
2595         if (res0) {
2596                 cur->ai_next = res0;
2597                 while (cur && cur->ai_next)
2598                         cur = cur->ai_next;
2599         }
2600
2601         if (nextline) {
2602                 p = nextline;
2603                 goto nextline;
2604         }
2605
2606 done:
2607         return sentinel.ai_next;
2608 }
2609
2610 /*ARGSUSED*/
2611 static int
2612 _yp_getaddrinfo(void *rv, void *cb_data, va_list ap)
2613 {
2614         struct addrinfo sentinel, *cur;
2615         struct addrinfo *ai = NULL;
2616         char *ypbuf;
2617         int ypbuflen, r;
2618         const char *name;
2619         const struct addrinfo *pai;
2620         char *ypdomain;
2621
2622         if (_yp_check(&ypdomain) == 0)
2623                 return NS_UNAVAIL;
2624
2625         name = va_arg(ap, char *);
2626         pai = va_arg(ap, const struct addrinfo *);
2627
2628         memset(&sentinel, 0, sizeof(sentinel));
2629         cur = &sentinel;
2630
2631         /* ipnodes.byname can hold both IPv4/v6 */
2632         r = yp_match(ypdomain, "ipnodes.byname", name,
2633                 (int)strlen(name), &ypbuf, &ypbuflen);
2634         if (r == 0) {
2635                 ai = _yphostent(ypbuf, pai);
2636                 if (ai) {
2637                         cur->ai_next = ai;
2638                         while (cur && cur->ai_next)
2639                                 cur = cur->ai_next;
2640                 }
2641                 free(ypbuf);
2642         }
2643
2644         if (ai != NULL) {
2645                 struct sockaddr_in6 *sin6;
2646
2647                 switch (ai->ai_family) {
2648                 case AF_INET:
2649                         goto done;
2650                 case AF_INET6:
2651                         sin6 = (struct sockaddr_in6 *)ai->ai_addr;
2652                         if (IN6_IS_ADDR_V4MAPPED(&sin6->sin6_addr))
2653                                 goto done;
2654                         break;
2655                 }
2656         }
2657
2658         /* hosts.byname is only for IPv4 (Solaris8) */
2659         if (pai->ai_family == AF_UNSPEC || pai->ai_family == AF_INET ||
2660             ((pai->ai_family == AF_INET6 &&
2661              (pai->ai_flags & AI_V4MAPPED) == AI_V4MAPPED) &&
2662               (ai == NULL || (pai->ai_flags & AI_ALL) == AI_ALL))) {
2663                 r = yp_match(ypdomain, "hosts.byname", name,
2664                         (int)strlen(name), &ypbuf, &ypbuflen);
2665                 if (r == 0) {
2666                         struct addrinfo ai4;
2667
2668                         ai4 = *pai;
2669                         if (pai->ai_family == AF_UNSPEC)
2670                                 ai4.ai_family = AF_INET;
2671                         ai = _yphostent(ypbuf, &ai4);
2672                         if (ai) {
2673                                 cur->ai_next = ai;
2674                                 while (cur && cur->ai_next)
2675                                         cur = cur->ai_next;
2676                         }
2677                         free(ypbuf);
2678                 }
2679         }
2680
2681 done:
2682         if (sentinel.ai_next == NULL) {
2683                 RES_SET_H_ERRNO(__res_state(), HOST_NOT_FOUND);
2684                 return NS_NOTFOUND;
2685         }
2686         *((struct addrinfo **)rv) = sentinel.ai_next;
2687         return NS_SUCCESS;
2688 }
2689 #endif
2690
2691 /* resolver logic */
2692
2693 /*
2694  * Formulate a normal query, send, and await answer.
2695  * Returned answer is placed in supplied buffer "answer".
2696  * Perform preliminary check of answer, returning success only
2697  * if no error is indicated and the answer count is nonzero.
2698  * Return the size of the response on success, -1 on error.
2699  * Error number is left in h_errno.
2700  *
2701  * Caller must parse answer and determine whether it answers the question.
2702  */
2703 static int
2704 res_queryN(const char *name, struct res_target *target, res_state res)
2705 {
2706         u_char *buf;
2707         HEADER *hp;
2708         int n;
2709         u_int oflags;
2710         struct res_target *t;
2711         int rcode;
2712         int ancount;
2713
2714         rcode = NOERROR;
2715         ancount = 0;
2716
2717         buf = malloc(MAXPACKET);
2718         if (!buf) {
2719                 RES_SET_H_ERRNO(res, NETDB_INTERNAL);
2720                 return -1;
2721         }
2722
2723         for (t = target; t; t = t->next) {
2724                 int class, type;
2725                 u_char *answer;
2726                 int anslen;
2727
2728                 hp = (HEADER *)(void *)t->answer;
2729
2730                 /* make it easier... */
2731                 class = t->qclass;
2732                 type = t->qtype;
2733                 answer = t->answer;
2734                 anslen = t->anslen;
2735
2736                 oflags = res->_flags;
2737
2738 again:
2739                 hp->rcode = NOERROR;    /* default */
2740
2741 #ifdef DEBUG
2742                 if (res->options & RES_DEBUG)
2743                         printf(";; res_query(%s, %d, %d)\n", name, class, type);
2744 #endif
2745
2746                 n = res_nmkquery(res, QUERY, name, class, type, NULL, 0, NULL,
2747                     buf, MAXPACKET);
2748                 if (n > 0 && (res->_flags & RES_F_EDNS0ERR) == 0 &&
2749                     (res->options & (RES_USE_EDNS0|RES_USE_DNSSEC)) != 0U)
2750                         n = res_nopt(res, n, buf, MAXPACKET, anslen);
2751                 if (n <= 0) {
2752 #ifdef DEBUG
2753                         if (res->options & RES_DEBUG)
2754                                 printf(";; res_query: mkquery failed\n");
2755 #endif
2756                         free(buf);
2757                         RES_SET_H_ERRNO(res, NO_RECOVERY);
2758                         return (n);
2759                 }
2760                 n = res_nsend(res, buf, n, answer, anslen);
2761                 if (n < 0) {
2762                         /*
2763                          * if the query choked with EDNS0, retry
2764                          * without EDNS0
2765                          */
2766                         if ((res->options & (RES_USE_EDNS0|RES_USE_DNSSEC))
2767                             != 0U &&
2768                             ((oflags ^ res->_flags) & RES_F_EDNS0ERR) != 0) {
2769                                 res->_flags |= RES_F_EDNS0ERR;
2770                                 if (res->options & RES_DEBUG)
2771                                         printf(";; res_nquery: retry without EDNS0\n");
2772                                 goto again;
2773                         }
2774                         rcode = hp->rcode;      /* record most recent error */
2775 #ifdef DEBUG
2776                         if (res->options & RES_DEBUG)
2777                                 printf(";; res_query: send error\n");
2778 #endif
2779                         continue;
2780                 }
2781
2782                 if (n > anslen)
2783                         hp->rcode = FORMERR; /* XXX not very informative */
2784                 if (hp->rcode != NOERROR || ntohs(hp->ancount) == 0) {
2785                         rcode = hp->rcode;      /* record most recent error */
2786 #ifdef DEBUG
2787                         if (res->options & RES_DEBUG)
2788                                 printf(";; rcode = %u, ancount=%u\n", hp->rcode,
2789                                     ntohs(hp->ancount));
2790 #endif
2791                         continue;
2792                 }
2793
2794                 ancount += ntohs(hp->ancount);
2795
2796                 t->n = n;
2797         }
2798
2799         free(buf);
2800
2801         if (ancount == 0) {
2802                 switch (rcode) {
2803                 case NXDOMAIN:
2804                         RES_SET_H_ERRNO(res, HOST_NOT_FOUND);
2805                         break;
2806                 case SERVFAIL:
2807                         RES_SET_H_ERRNO(res, TRY_AGAIN);
2808                         break;
2809                 case NOERROR:
2810                         RES_SET_H_ERRNO(res, NO_DATA);
2811                         break;
2812                 case FORMERR:
2813                 case NOTIMP:
2814                 case REFUSED:
2815                 default:
2816                         RES_SET_H_ERRNO(res, NO_RECOVERY);
2817                         break;
2818                 }
2819                 return (-1);
2820         }
2821         return (ancount);
2822 }
2823
2824 /*
2825  * Formulate a normal query, send, and retrieve answer in supplied buffer.
2826  * Return the size of the response on success, -1 on error.
2827  * If enabled, implement search rules until answer or unrecoverable failure
2828  * is detected.  Error code, if any, is left in h_errno.
2829  */
2830 static int
2831 res_searchN(const char *name, struct res_target *target, res_state res)
2832 {
2833         const char *cp, * const *domain;
2834         HEADER *hp = (HEADER *)(void *)target->answer;  /*XXX*/
2835         u_int dots;
2836         int trailing_dot, ret, saved_herrno;
2837         int got_nodata = 0, got_servfail = 0, root_on_list = 0;
2838         int tried_as_is = 0;
2839         int searched = 0;
2840         char abuf[MAXDNAME];
2841
2842         errno = 0;
2843         RES_SET_H_ERRNO(res, HOST_NOT_FOUND); /* default, if we never query */
2844         dots = 0;
2845         for (cp = name; *cp; cp++)
2846                 dots += (*cp == '.');
2847         trailing_dot = 0;
2848         if (cp > name && *--cp == '.')
2849                 trailing_dot++;
2850
2851         /*
2852          * if there aren't any dots, it could be a user-level alias
2853          */
2854         if (!dots &&
2855             (cp = res_hostalias(res, name, abuf, sizeof(abuf))) != NULL)
2856                 return (res_queryN(cp, target, res));
2857
2858         /*
2859          * If there are enough dots in the name, let's just give it a
2860          * try 'as is'. The threshold can be set with the "ndots" option.
2861          * Also, query 'as is', if there is a trailing dot in the name.
2862          */
2863         saved_herrno = -1;
2864         if (dots >= res->ndots || trailing_dot) {
2865                 ret = res_querydomainN(name, NULL, target, res);
2866                 if (ret > 0 || trailing_dot)
2867                         return (ret);
2868                 if (errno == ECONNREFUSED) {
2869                         RES_SET_H_ERRNO(res, TRY_AGAIN);
2870                         return (-1);
2871                 }
2872                 switch (res->res_h_errno) {
2873                 case NO_DATA:
2874                 case HOST_NOT_FOUND:
2875                         break;
2876                 case TRY_AGAIN:
2877                         if (hp->rcode == SERVFAIL)
2878                                 break;
2879                         /* FALLTHROUGH */
2880                 default:
2881                         return (-1);
2882                 }
2883                 saved_herrno = res->res_h_errno;
2884                 tried_as_is++;
2885         }
2886
2887         /*
2888          * We do at least one level of search if
2889          *      - there is no dot and RES_DEFNAME is set, or
2890          *      - there is at least one dot, there is no trailing dot,
2891          *        and RES_DNSRCH is set.
2892          */
2893         if ((!dots && (res->options & RES_DEFNAMES)) ||
2894             (dots && !trailing_dot && (res->options & RES_DNSRCH))) {
2895                 int done = 0;
2896
2897                 for (domain = (const char * const *)res->dnsrch;
2898                    *domain && !done;
2899                    domain++) {
2900                         searched = 1;
2901
2902                         if (domain[0][0] == '\0' ||
2903                             (domain[0][0] == '.' && domain[0][1] == '\0'))
2904                                 root_on_list++;
2905
2906                         if (root_on_list && tried_as_is)
2907                                 continue;
2908
2909                         ret = res_querydomainN(name, *domain, target, res);
2910                         if (ret > 0)
2911                                 return (ret);
2912
2913                         /*
2914                          * If no server present, give up.
2915                          * If name isn't found in this domain,
2916                          * keep trying higher domains in the search list
2917                          * (if that's enabled).
2918                          * On a NO_DATA error, keep trying, otherwise
2919                          * a wildcard entry of another type could keep us
2920                          * from finding this entry higher in the domain.
2921                          * If we get some other error (negative answer or
2922                          * server failure), then stop searching up,
2923                          * but try the input name below in case it's
2924                          * fully-qualified.
2925                          */
2926                         if (errno == ECONNREFUSED) {
2927                                 RES_SET_H_ERRNO(res, TRY_AGAIN);
2928                                 return (-1);
2929                         }
2930
2931                         switch (res->res_h_errno) {
2932                         case NO_DATA:
2933                                 got_nodata++;
2934                                 /* FALLTHROUGH */
2935                         case HOST_NOT_FOUND:
2936                                 /* keep trying */
2937                                 break;
2938                         case TRY_AGAIN:
2939                                 got_servfail++;
2940                                 if (hp->rcode == SERVFAIL) {
2941                                         /* try next search element, if any */
2942                                         break;
2943                                 }
2944                                 /* FALLTHROUGH */
2945                         default:
2946                                 /* anything else implies that we're done */
2947                                 done++;
2948                         }
2949                         /*
2950                          * if we got here for some reason other than DNSRCH,
2951                          * we only wanted one iteration of the loop, so stop.
2952                          */
2953                         if (!(res->options & RES_DNSRCH))
2954                                 done++;
2955                 }
2956         }
2957
2958         switch (res->res_h_errno) {
2959         case NO_DATA:
2960         case HOST_NOT_FOUND:
2961                 break;
2962         case TRY_AGAIN:
2963                 if (hp->rcode == SERVFAIL)
2964                         break;
2965                 /* FALLTHROUGH */
2966         default:
2967                 goto giveup;
2968         }
2969
2970         /*
2971          * If the query has not already been tried as is then try it
2972          * unless RES_NOTLDQUERY is set and there were no dots.
2973          */
2974         if ((dots || !searched || !(res->options & RES_NOTLDQUERY)) &&
2975             !(tried_as_is || root_on_list)) {
2976                 ret = res_querydomainN(name, NULL, target, res);
2977                 if (ret > 0)
2978                         return (ret);
2979         }
2980
2981         /*
2982          * if we got here, we didn't satisfy the search.
2983          * if we did an initial full query, return that query's h_errno
2984          * (note that we wouldn't be here if that query had succeeded).
2985          * else if we ever got a nodata, send that back as the reason.
2986          * else send back meaningless h_errno, that being the one from
2987          * the last DNSRCH we did.
2988          */
2989 giveup:
2990         if (saved_herrno != -1)
2991                 RES_SET_H_ERRNO(res, saved_herrno);
2992         else if (got_nodata)
2993                 RES_SET_H_ERRNO(res, NO_DATA);
2994         else if (got_servfail)
2995                 RES_SET_H_ERRNO(res, TRY_AGAIN);
2996         return (-1);
2997 }
2998
2999 /*
3000  * Perform a call on res_query on the concatenation of name and domain,
3001  * removing a trailing dot from name if domain is NULL.
3002  */
3003 static int
3004 res_querydomainN(const char *name, const char *domain,
3005     struct res_target *target, res_state res)
3006 {
3007         char nbuf[MAXDNAME];
3008         const char *longname = nbuf;
3009         size_t n, d;
3010
3011 #ifdef DEBUG
3012         if (res->options & RES_DEBUG)
3013                 printf(";; res_querydomain(%s, %s)\n",
3014                         name, domain?domain:"<Nil>");
3015 #endif
3016         if (domain == NULL) {
3017                 /*
3018                  * Check for trailing '.';
3019                  * copy without '.' if present.
3020                  */
3021                 n = strlen(name);
3022                 if (n >= MAXDNAME) {
3023                         RES_SET_H_ERRNO(res, NO_RECOVERY);
3024                         return (-1);
3025                 }
3026                 if (n > 0 && name[--n] == '.') {
3027                         strncpy(nbuf, name, n);
3028                         nbuf[n] = '\0';
3029                 } else
3030                         longname = name;
3031         } else {
3032                 n = strlen(name);
3033                 d = strlen(domain);
3034                 if (n + d + 1 >= MAXDNAME) {
3035                         RES_SET_H_ERRNO(res, NO_RECOVERY);
3036                         return (-1);
3037                 }
3038                 snprintf(nbuf, sizeof(nbuf), "%s.%s", name, domain);
3039         }
3040         return (res_queryN(longname, target, res));
3041 }