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
107static const char in_addrany[] = { 0, 0, 0, 0 };
108static const char in_loopback[] = { 127, 0, 0, 1 };
109#ifdef INET6
110static const char in6_addrany[] = {
111	0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
112};
113static 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
118struct policyqueue {
119	TAILQ_ENTRY(policyqueue) pc_entry;
120#ifdef INET6
121	struct in6_addrpolicy pc_policy;
122#endif
123};
124TAILQ_HEAD(policyhead, policyqueue);
125
126static 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
159struct 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
172static 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
216struct 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
232static const ns_src default_dns_files[] = {
233	{ NSSRC_FILES, 	NS_SUCCESS },
234	{ NSSRC_DNS, 	NS_SUCCESS },
235	{ 0 }
236};
237
238struct 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
249typedef union {
250	HEADER hdr;
251	u_char buf[MAXPACKET];
252} querybuf;
253
254static int str2number(const char *, int *);
255static int explore_copy(const struct addrinfo *, const struct addrinfo *,
256	struct addrinfo **);
257static int explore_null(const struct addrinfo *,
258	const char *, struct addrinfo **);
259static int explore_numeric(const struct addrinfo *, const char *,
260	const char *, struct addrinfo **, const char *);
261static int explore_numeric_scope(const struct addrinfo *, const char *,
262	const char *, struct addrinfo **);
263static int get_canonname(const struct addrinfo *,
264	struct addrinfo *, const char *);
265static struct addrinfo *get_ai(const struct addrinfo *,
266	const struct afd *, const char *);
267static struct addrinfo *copy_ai(const struct addrinfo *);
268static int get_portmatch(const struct addrinfo *, const char *);
269static int get_port(struct addrinfo *, const char *, int);
270static const struct afd *find_afd(int);
271static int addrconfig(struct addrinfo *);
272#ifdef INET6
273static int is_ifdisabled(char *);
274#endif
275static void set_source(struct ai_order *, struct policyhead *);
276static int comp_dst(const void *, const void *);
277#ifdef INET6
278static int ip6_str2scopeid(char *, struct sockaddr_in6 *, u_int32_t *);
279#endif
280static int gai_addr2scopetype(struct sockaddr *);
281
282static int explore_fqdn(const struct addrinfo *, const char *,
283	const char *, struct addrinfo **);
284
285static int reorder(struct addrinfo *);
286static int get_addrselectpolicy(struct policyhead *);
287static void free_addrselectpolicy(struct policyhead *);
288static struct policyqueue *match_addrselectpolicy(struct sockaddr *,
289	struct policyhead *);
290static int matchlen(struct sockaddr *, struct sockaddr *);
291
292static struct addrinfo *getanswer(const querybuf *, int, const char *, int,
293	const struct addrinfo *, res_state);
294#if defined(RESOLVSORT)
295static int addr4sort(struct addrinfo *, res_state);
296#endif
297static int _dns_getaddrinfo(void *, void *, va_list);
298static void _sethtent(FILE **);
299static void _endhtent(FILE **);
300static struct addrinfo *_gethtent(FILE **, const char *,
301	const struct addrinfo *);
302static int _files_getaddrinfo(void *, void *, va_list);
303#ifdef YP
304static struct addrinfo *_yphostent(char *, const struct addrinfo *);
305static int _yp_getaddrinfo(void *, void *, va_list);
306#endif
307#ifdef NS_CACHING
308static int addrinfo_id_func(char *, size_t *, va_list, void *);
309static int addrinfo_marshal_func(char *, size_t *, void *, va_list, void *);
310static int addrinfo_unmarshal_func(char *, size_t, void *, va_list, void *);
311#endif
312
313static int res_queryN(const char *, struct res_target *, res_state);
314static int res_searchN(const char *, struct res_target *, res_state);
315static 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) \
321do { \
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) \
331do { \
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) \
339do { \
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) \
347do { \
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
359void
360freeaddrinfo(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
373static int
374str2number(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
391int
392getaddrinfo(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
578globcopy:
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
662bad:
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
676static int
677reorder(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
731static int
732get_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
770static void
771free_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
782static struct policyqueue *
783match_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
848static void
849set_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
923static int
924matchlen(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
962static int
963comp_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 */
1096static int
1097gai_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
1153static int
1154explore_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
1183fail:
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 */
1193static int
1194explore_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
1234free:
1235	if (ai != NULL)
1236		freeaddrinfo(ai);
1237	return error;
1238}
1239
1240/*
1241 * numeric hostname
1242 */
1243static int
1244explore_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
1324free:
1325bad:
1326	if (ai != NULL)
1327		freeaddrinfo(ai);
1328	return error;
1329}
1330
1331/*
1332 * numeric hostname with scope
1333 */
1334static int
1335explore_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
1397static int
1398get_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
1408static struct addrinfo *
1409get_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! */
1446static struct addrinfo *
1447copy_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
1476static int
1477get_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
1485static int
1486get_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
1585static const struct afd *
1586find_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 */
1609static int
1610addrconfig(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
1668static int
1669is_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 */
1687static int
1688ip6_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
1735static int
1736addrinfo_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
1804static int
1805addrinfo_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
1856static int
1857addrinfo_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 */
1913static int
1914explore_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_SUCCESS:
1957		error = 0;
1958		for (cur = result; cur; cur = cur->ai_next) {
1959			GET_PORT(cur, servname);
1960			/* canonname should be filled already */
1961		}
1962		break;
1963	}
1964
1965	*res = result;
1966
1967	return 0;
1968
1969free:
1970	if (result)
1971		freeaddrinfo(result);
1972	return error;
1973}
1974
1975#ifdef DEBUG
1976static const char AskedForGot[] =
1977	"gethostby*.getanswer: asked for \"%s\", got \"%s\"";
1978#endif
1979
1980static struct addrinfo *
1981getanswer(const querybuf *answer, int anslen, const char *qname, int qtype,
1982    const struct addrinfo *pai, res_state res)
1983{
1984	struct addrinfo sentinel, *cur;
1985	struct addrinfo ai;
1986	const struct afd *afd;
1987	char *canonname;
1988	const HEADER *hp;
1989	const u_char *cp;
1990	int n;
1991	const u_char *eom;
1992	char *bp, *ep;
1993	int type, class, ancount, qdcount;
1994	int haveanswer, had_error;
1995	char tbuf[MAXDNAME];
1996	int (*name_ok)(const char *);
1997	char hostbuf[8*1024];
1998
1999	memset(&sentinel, 0, sizeof(sentinel));
2000	cur = &sentinel;
2001
2002	canonname = NULL;
2003	eom = answer->buf + anslen;
2004	switch (qtype) {
2005	case T_A:
2006	case T_AAAA:
2007	case T_ANY:	/*use T_ANY only for T_A/T_AAAA lookup*/
2008		name_ok = res_hnok;
2009		break;
2010	default:
2011		return (NULL);	/* XXX should be abort(); */
2012	}
2013	/*
2014	 * find first satisfactory answer
2015	 */
2016	hp = &answer->hdr;
2017	ancount = ntohs(hp->ancount);
2018	qdcount = ntohs(hp->qdcount);
2019	bp = hostbuf;
2020	ep = hostbuf + sizeof hostbuf;
2021	cp = answer->buf + HFIXEDSZ;
2022	if (qdcount != 1) {
2023		RES_SET_H_ERRNO(res, NO_RECOVERY);
2024		return (NULL);
2025	}
2026	n = dn_expand(answer->buf, eom, cp, bp, ep - bp);
2027	if ((n < 0) || !(*name_ok)(bp)) {
2028		RES_SET_H_ERRNO(res, NO_RECOVERY);
2029		return (NULL);
2030	}
2031	cp += n + QFIXEDSZ;
2032	if (qtype == T_A || qtype == T_AAAA || qtype == T_ANY) {
2033		/* res_send() has already verified that the query name is the
2034		 * same as the one we sent; this just gets the expanded name
2035		 * (i.e., with the succeeding search-domain tacked on).
2036		 */
2037		n = strlen(bp) + 1;		/* for the \0 */
2038		if (n >= MAXHOSTNAMELEN) {
2039			RES_SET_H_ERRNO(res, NO_RECOVERY);
2040			return (NULL);
2041		}
2042		canonname = bp;
2043		bp += n;
2044		/* The qname can be abbreviated, but h_name is now absolute. */
2045		qname = canonname;
2046	}
2047	haveanswer = 0;
2048	had_error = 0;
2049	while (ancount-- > 0 && cp < eom && !had_error) {
2050		n = dn_expand(answer->buf, eom, cp, bp, ep - bp);
2051		if ((n < 0) || !(*name_ok)(bp)) {
2052			had_error++;
2053			continue;
2054		}
2055		cp += n;			/* name */
2056		type = _getshort(cp);
2057 		cp += INT16SZ;			/* type */
2058		class = _getshort(cp);
2059 		cp += INT16SZ + INT32SZ;	/* class, TTL */
2060		n = _getshort(cp);
2061		cp += INT16SZ;			/* len */
2062		if (class != C_IN) {
2063			/* XXX - debug? syslog? */
2064			cp += n;
2065			continue;		/* XXX - had_error++ ? */
2066		}
2067		if ((qtype == T_A || qtype == T_AAAA || qtype == T_ANY) &&
2068		    type == T_CNAME) {
2069			n = dn_expand(answer->buf, eom, cp, tbuf, sizeof tbuf);
2070			if ((n < 0) || !(*name_ok)(tbuf)) {
2071				had_error++;
2072				continue;
2073			}
2074			cp += n;
2075			/* Get canonical name. */
2076			n = strlen(tbuf) + 1;	/* for the \0 */
2077			if (n > ep - bp || n >= MAXHOSTNAMELEN) {
2078				had_error++;
2079				continue;
2080			}
2081			strlcpy(bp, tbuf, ep - bp);
2082			canonname = bp;
2083			bp += n;
2084			continue;
2085		}
2086		if (qtype == T_ANY) {
2087			if (!(type == T_A || type == T_AAAA)) {
2088				cp += n;
2089				continue;
2090			}
2091		} else if (type != qtype) {
2092#ifdef DEBUG
2093			if (type != T_KEY && type != T_SIG &&
2094			    type != ns_t_dname)
2095				syslog(LOG_NOTICE|LOG_AUTH,
2096	       "gethostby*.getanswer: asked for \"%s %s %s\", got type \"%s\"",
2097				       qname, p_class(C_IN), p_type(qtype),
2098				       p_type(type));
2099#endif
2100			cp += n;
2101			continue;		/* XXX - had_error++ ? */
2102		}
2103		switch (type) {
2104		case T_A:
2105		case T_AAAA:
2106			if (strcasecmp(canonname, bp) != 0) {
2107#ifdef DEBUG
2108				syslog(LOG_NOTICE|LOG_AUTH,
2109				       AskedForGot, canonname, bp);
2110#endif
2111				cp += n;
2112				continue;	/* XXX - had_error++ ? */
2113			}
2114			if (type == T_A && n != INADDRSZ) {
2115				cp += n;
2116				continue;
2117			}
2118			if (type == T_AAAA && n != IN6ADDRSZ) {
2119				cp += n;
2120				continue;
2121			}
2122#ifdef FILTER_V4MAPPED
2123			if (type == T_AAAA) {
2124				struct in6_addr in6;
2125				memcpy(&in6, cp, sizeof(in6));
2126				if (IN6_IS_ADDR_V4MAPPED(&in6)) {
2127					cp += n;
2128					continue;
2129				}
2130			}
2131#endif
2132			if (!haveanswer) {
2133				int nn;
2134
2135				canonname = bp;
2136				nn = strlen(bp) + 1;	/* for the \0 */
2137				bp += nn;
2138			}
2139
2140			/* don't overwrite pai */
2141			ai = *pai;
2142			ai.ai_family = (type == T_A) ? AF_INET : AF_INET6;
2143			afd = find_afd(ai.ai_family);
2144			if (afd == NULL) {
2145				cp += n;
2146				continue;
2147			}
2148			cur->ai_next = get_ai(&ai, afd, (const char *)cp);
2149			if (cur->ai_next == NULL)
2150				had_error++;
2151			while (cur && cur->ai_next)
2152				cur = cur->ai_next;
2153			cp += n;
2154			break;
2155		default:
2156			abort();
2157		}
2158		if (!had_error)
2159			haveanswer++;
2160	}
2161	if (haveanswer) {
2162#if defined(RESOLVSORT)
2163		/*
2164		 * We support only IPv4 address for backward
2165		 * compatibility against gethostbyname(3).
2166		 */
2167		if (res->nsort && qtype == T_A) {
2168			if (addr4sort(&sentinel, res) < 0) {
2169				freeaddrinfo(sentinel.ai_next);
2170				RES_SET_H_ERRNO(res, NO_RECOVERY);
2171				return NULL;
2172			}
2173		}
2174#endif /*RESOLVSORT*/
2175		if (!canonname)
2176			(void)get_canonname(pai, sentinel.ai_next, qname);
2177		else
2178			(void)get_canonname(pai, sentinel.ai_next, canonname);
2179		RES_SET_H_ERRNO(res, NETDB_SUCCESS);
2180		return sentinel.ai_next;
2181	}
2182
2183	/*
2184	 * We could have walked a CNAME chain, but the ultimate target
2185	 * may not have what we looked for.
2186	 */
2187	RES_SET_H_ERRNO(res, ntohs(hp->ancount) > 0 ? NO_DATA : NO_RECOVERY);
2188	return NULL;
2189}
2190
2191#ifdef RESOLVSORT
2192struct addr_ptr {
2193	struct addrinfo *ai;
2194	int aval;
2195};
2196
2197static int
2198addr4sort(struct addrinfo *sentinel, res_state res)
2199{
2200	struct addrinfo *ai;
2201	struct addr_ptr *addrs, addr;
2202	struct sockaddr_in *sin;
2203	int naddrs, i, j;
2204	int needsort = 0;
2205
2206	if (!sentinel)
2207		return -1;
2208	naddrs = 0;
2209	for (ai = sentinel->ai_next; ai; ai = ai->ai_next)
2210		naddrs++;
2211	if (naddrs < 2)
2212		return 0;		/* We don't need sorting. */
2213	if ((addrs = malloc(sizeof(struct addr_ptr) * naddrs)) == NULL)
2214		return -1;
2215	i = 0;
2216	for (ai = sentinel->ai_next; ai; ai = ai->ai_next) {
2217		sin = (struct sockaddr_in *)ai->ai_addr;
2218		for (j = 0; (unsigned)j < res->nsort; j++) {
2219			if (res->sort_list[j].addr.s_addr ==
2220			    (sin->sin_addr.s_addr & res->sort_list[j].mask))
2221				break;
2222		}
2223		addrs[i].ai = ai;
2224		addrs[i].aval = j;
2225		if (needsort == 0 && i > 0 && j < addrs[i - 1].aval)
2226			needsort = i;
2227		i++;
2228	}
2229	if (!needsort) {
2230		free(addrs);
2231		return 0;
2232	}
2233
2234	while (needsort < naddrs) {
2235		for (j = needsort - 1; j >= 0; j--) {
2236			if (addrs[j].aval > addrs[j+1].aval) {
2237				addr = addrs[j];
2238				addrs[j] = addrs[j + 1];
2239				addrs[j + 1] = addr;
2240			} else
2241				break;
2242		}
2243		needsort++;
2244	}
2245
2246	ai = sentinel;
2247	for (i = 0; i < naddrs; ++i) {
2248		ai->ai_next = addrs[i].ai;
2249		ai = ai->ai_next;
2250	}
2251	ai->ai_next = NULL;
2252	free(addrs);
2253	return 0;
2254}
2255#endif /*RESOLVSORT*/
2256
2257/*ARGSUSED*/
2258static int
2259_dns_getaddrinfo(void *rv, void *cb_data, va_list ap)
2260{
2261	struct addrinfo *ai, ai0;
2262	querybuf *buf, *buf2;
2263	const char *hostname;
2264	const struct addrinfo *pai;
2265	struct addrinfo sentinel, *cur;
2266	struct res_target q, q2;
2267	res_state res;
2268
2269	ai = NULL;
2270
2271	hostname = va_arg(ap, char *);
2272	pai = va_arg(ap, const struct addrinfo *);
2273
2274	memset(&q, 0, sizeof(q));
2275	memset(&q2, 0, sizeof(q2));
2276	memset(&sentinel, 0, sizeof(sentinel));
2277	cur = &sentinel;
2278
2279	res = __res_state();
2280
2281	buf = malloc(sizeof(*buf));
2282	if (!buf) {
2283		RES_SET_H_ERRNO(res, NETDB_INTERNAL);
2284		return NS_NOTFOUND;
2285	}
2286	buf2 = malloc(sizeof(*buf2));
2287	if (!buf2) {
2288		free(buf);
2289		RES_SET_H_ERRNO(res, NETDB_INTERNAL);
2290		return NS_NOTFOUND;
2291	}
2292
2293	if (pai->ai_family == AF_INET6 &&
2294	    (pai->ai_flags & AI_V4MAPPED) == AI_V4MAPPED) {
2295		ai0 = *pai;
2296		ai0.ai_family = AF_UNSPEC;
2297		pai = &ai0;
2298	}
2299
2300	switch (pai->ai_family) {
2301	case AF_UNSPEC:
2302		q.name = hostname;
2303		q.qclass = C_IN;
2304		q.qtype = T_A;
2305		q.answer = buf->buf;
2306		q.anslen = sizeof(buf->buf);
2307		q.next = &q2;
2308		q2.name = hostname;
2309		q2.qclass = C_IN;
2310		q2.qtype = T_AAAA;
2311		q2.answer = buf2->buf;
2312		q2.anslen = sizeof(buf2->buf);
2313		break;
2314	case AF_INET:
2315		q.name = hostname;
2316		q.qclass = C_IN;
2317		q.qtype = T_A;
2318		q.answer = buf->buf;
2319		q.anslen = sizeof(buf->buf);
2320		break;
2321	case AF_INET6:
2322		q.name = hostname;
2323		q.qclass = C_IN;
2324		q.qtype = T_AAAA;
2325		q.answer = buf->buf;
2326		q.anslen = sizeof(buf->buf);
2327		break;
2328	default:
2329		free(buf);
2330		free(buf2);
2331		return NS_UNAVAIL;
2332	}
2333
2334	if ((res->options & RES_INIT) == 0 && res_ninit(res) == -1) {
2335		RES_SET_H_ERRNO(res, NETDB_INTERNAL);
2336		free(buf);
2337		free(buf2);
2338		return NS_NOTFOUND;
2339	}
2340
2341	if (res_searchN(hostname, &q, res) < 0) {
2342		free(buf);
2343		free(buf2);
2344		return NS_NOTFOUND;
2345	}
2346	/* prefer IPv6 */
2347	if (q.next) {
2348		ai = getanswer(buf2, q2.n, q2.name, q2.qtype, pai, res);
2349		if (ai != NULL) {
2350			cur->ai_next = ai;
2351			while (cur && cur->ai_next)
2352				cur = cur->ai_next;
2353		}
2354	}
2355	if (ai == NULL || pai->ai_family != AF_UNSPEC ||
2356	    (pai->ai_flags & (AI_ALL | AI_V4MAPPED)) != AI_V4MAPPED) {
2357		ai = getanswer(buf, q.n, q.name, q.qtype, pai, res);
2358		if (ai != NULL)
2359			cur->ai_next = ai;
2360	}
2361	free(buf);
2362	free(buf2);
2363	if (sentinel.ai_next == NULL)
2364		switch (res->res_h_errno) {
2365		case HOST_NOT_FOUND:
2366		case NO_DATA:
2367			return NS_NOTFOUND;
2368		case TRY_AGAIN:
2369			return NS_TRYAGAIN;
2370		default:
2371			return NS_UNAVAIL;
2372		}
2373	*((struct addrinfo **)rv) = sentinel.ai_next;
2374	return NS_SUCCESS;
2375}
2376
2377static void
2378_sethtent(FILE **hostf)
2379{
2380	if (!*hostf)
2381		*hostf = fopen(_PATH_HOSTS, "re");
2382	else
2383		rewind(*hostf);
2384}
2385
2386static void
2387_endhtent(FILE **hostf)
2388{
2389	if (*hostf) {
2390		(void) fclose(*hostf);
2391		*hostf = NULL;
2392	}
2393}
2394
2395static struct addrinfo *
2396_gethtent(FILE **hostf, const char *name, const struct addrinfo *pai)
2397{
2398	char *p;
2399	char *cp, *tname, *cname;
2400	struct addrinfo hints, *res0, *res;
2401	int error;
2402	const char *addr;
2403	char hostbuf[8*1024];
2404
2405	if (!*hostf && !(*hostf = fopen(_PATH_HOSTS, "re")))
2406		return (NULL);
2407again:
2408	if (!(p = fgets(hostbuf, sizeof hostbuf, *hostf)))
2409		return (NULL);
2410	if (*p == '#')
2411		goto again;
2412	cp = strpbrk(p, "#\n");
2413	if (cp != NULL)
2414		*cp = '\0';
2415	if (!(cp = strpbrk(p, " \t")))
2416		goto again;
2417	*cp++ = '\0';
2418	addr = p;
2419	cname = NULL;
2420	/* if this is not something we're looking for, skip it. */
2421	while (cp && *cp) {
2422		if (*cp == ' ' || *cp == '\t') {
2423			cp++;
2424			continue;
2425		}
2426		tname = cp;
2427		if (cname == NULL)
2428			cname = cp;
2429		if ((cp = strpbrk(cp, " \t")) != NULL)
2430			*cp++ = '\0';
2431		if (strcasecmp(name, tname) == 0)
2432			goto found;
2433	}
2434	goto again;
2435
2436found:
2437	/* we should not glob socktype/protocol here */
2438	memset(&hints, 0, sizeof(hints));
2439	hints.ai_family = pai->ai_family;
2440	hints.ai_socktype = SOCK_DGRAM;
2441	hints.ai_protocol = 0;
2442	hints.ai_flags = AI_NUMERICHOST;
2443	if (pai->ai_family == AF_INET6 &&
2444	    (pai->ai_flags & AI_V4MAPPED) == AI_V4MAPPED)
2445		hints.ai_flags |= AI_V4MAPPED;
2446	error = getaddrinfo(addr, "0", &hints, &res0);
2447	if (error)
2448		goto again;
2449#ifdef FILTER_V4MAPPED
2450	/* XXX should check all items in the chain */
2451	if (res0->ai_family == AF_INET6 &&
2452	    IN6_IS_ADDR_V4MAPPED(&((struct sockaddr_in6 *)res0->ai_addr)->sin6_addr)) {
2453		freeaddrinfo(res0);
2454		goto again;
2455	}
2456#endif
2457	for (res = res0; res; res = res->ai_next) {
2458		/* cover it up */
2459		res->ai_flags = pai->ai_flags;
2460		res->ai_socktype = pai->ai_socktype;
2461		res->ai_protocol = pai->ai_protocol;
2462
2463		if (pai->ai_flags & AI_CANONNAME) {
2464			if (get_canonname(pai, res, cname) != 0) {
2465				freeaddrinfo(res0);
2466				goto again;
2467			}
2468		}
2469	}
2470	return res0;
2471}
2472
2473static struct addrinfo *
2474_getht(FILE **hostf, const char *name, const struct addrinfo *pai,
2475     struct addrinfo *cur)
2476{
2477	struct addrinfo *p;
2478
2479	while ((p = _gethtent(hostf, name, pai)) != NULL) {
2480		cur->ai_next = p;
2481		while (cur && cur->ai_next)
2482			cur = cur->ai_next;
2483	}
2484	return (cur);
2485}
2486
2487/*ARGSUSED*/
2488static int
2489_files_getaddrinfo(void *rv, void *cb_data, va_list ap)
2490{
2491	const char *name;
2492	const struct addrinfo *pai;
2493	struct addrinfo sentinel, *cur;
2494	FILE *hostf = NULL;
2495
2496	name = va_arg(ap, char *);
2497	pai = va_arg(ap, struct addrinfo *);
2498
2499	memset(&sentinel, 0, sizeof(sentinel));
2500	cur = &sentinel;
2501
2502	_sethtent(&hostf);
2503	if (pai->ai_family == AF_INET6 &&
2504	    (pai->ai_flags & (AI_ALL | AI_V4MAPPED)) == AI_V4MAPPED) {
2505		struct addrinfo ai0 = *pai;
2506
2507		ai0.ai_flags &= ~AI_V4MAPPED;
2508		cur = _getht(&hostf, name, &ai0, cur);
2509		if (sentinel.ai_next == NULL) {
2510			_sethtent(&hostf);
2511			ai0.ai_flags |= AI_V4MAPPED;
2512			cur = _getht(&hostf, name, &ai0, cur);
2513		}
2514	} else
2515		cur = _getht(&hostf, name, pai, cur);
2516	_endhtent(&hostf);
2517
2518	*((struct addrinfo **)rv) = sentinel.ai_next;
2519	if (sentinel.ai_next == NULL)
2520		return NS_NOTFOUND;
2521	return NS_SUCCESS;
2522}
2523
2524#ifdef YP
2525/*ARGSUSED*/
2526static struct addrinfo *
2527_yphostent(char *line, const struct addrinfo *pai)
2528{
2529	struct addrinfo sentinel, *cur;
2530	struct addrinfo hints, *res, *res0;
2531	int error;
2532	char *p = line;
2533	const char *addr, *canonname;
2534	char *nextline;
2535	char *cp;
2536
2537	addr = canonname = NULL;
2538
2539	memset(&sentinel, 0, sizeof(sentinel));
2540	cur = &sentinel;
2541
2542nextline:
2543	/* terminate line */
2544	cp = strchr(p, '\n');
2545	if (cp) {
2546		*cp++ = '\0';
2547		nextline = cp;
2548	} else
2549		nextline = NULL;
2550
2551	cp = strpbrk(p, " \t");
2552	if (cp == NULL) {
2553		if (canonname == NULL)
2554			return (NULL);
2555		else
2556			goto done;
2557	}
2558	*cp++ = '\0';
2559
2560	addr = p;
2561
2562	while (cp && *cp) {
2563		if (*cp == ' ' || *cp == '\t') {
2564			cp++;
2565			continue;
2566		}
2567		if (!canonname)
2568			canonname = cp;
2569		if ((cp = strpbrk(cp, " \t")) != NULL)
2570			*cp++ = '\0';
2571	}
2572
2573	hints = *pai;
2574	hints.ai_flags = AI_NUMERICHOST;
2575	if (pai->ai_family == AF_INET6 &&
2576	    (pai->ai_flags & AI_V4MAPPED) == AI_V4MAPPED)
2577		hints.ai_flags |= AI_V4MAPPED;
2578	error = getaddrinfo(addr, NULL, &hints, &res0);
2579	if (error == 0) {
2580		for (res = res0; res; res = res->ai_next) {
2581			/* cover it up */
2582			res->ai_flags = pai->ai_flags;
2583
2584			if (pai->ai_flags & AI_CANONNAME)
2585				(void)get_canonname(pai, res, canonname);
2586		}
2587	} else
2588		res0 = NULL;
2589	if (res0) {
2590		cur->ai_next = res0;
2591		while (cur && cur->ai_next)
2592			cur = cur->ai_next;
2593	}
2594
2595	if (nextline) {
2596		p = nextline;
2597		goto nextline;
2598	}
2599
2600done:
2601	return sentinel.ai_next;
2602}
2603
2604/*ARGSUSED*/
2605static int
2606_yp_getaddrinfo(void *rv, void *cb_data, va_list ap)
2607{
2608	struct addrinfo sentinel, *cur;
2609	struct addrinfo *ai = NULL;
2610	char *ypbuf;
2611	int ypbuflen, r;
2612	const char *name;
2613	const struct addrinfo *pai;
2614	char *ypdomain;
2615
2616	if (_yp_check(&ypdomain) == 0)
2617		return NS_UNAVAIL;
2618
2619	name = va_arg(ap, char *);
2620	pai = va_arg(ap, const struct addrinfo *);
2621
2622	memset(&sentinel, 0, sizeof(sentinel));
2623	cur = &sentinel;
2624
2625	/* ipnodes.byname can hold both IPv4/v6 */
2626	r = yp_match(ypdomain, "ipnodes.byname", name,
2627		(int)strlen(name), &ypbuf, &ypbuflen);
2628	if (r == 0) {
2629		ai = _yphostent(ypbuf, pai);
2630		if (ai) {
2631			cur->ai_next = ai;
2632			while (cur && cur->ai_next)
2633				cur = cur->ai_next;
2634		}
2635		free(ypbuf);
2636	}
2637
2638	if (ai != NULL) {
2639		struct sockaddr_in6 *sin6;
2640
2641		switch (ai->ai_family) {
2642		case AF_INET:
2643			goto done;
2644		case AF_INET6:
2645			sin6 = (struct sockaddr_in6 *)ai->ai_addr;
2646			if (IN6_IS_ADDR_V4MAPPED(&sin6->sin6_addr))
2647				goto done;
2648			break;
2649		}
2650	}
2651
2652	/* hosts.byname is only for IPv4 (Solaris8) */
2653	if (pai->ai_family == AF_UNSPEC || pai->ai_family == AF_INET ||
2654	    ((pai->ai_family == AF_INET6 &&
2655	     (pai->ai_flags & AI_V4MAPPED) == AI_V4MAPPED) &&
2656	      (ai == NULL || (pai->ai_flags & AI_ALL) == AI_ALL))) {
2657		r = yp_match(ypdomain, "hosts.byname", name,
2658			(int)strlen(name), &ypbuf, &ypbuflen);
2659		if (r == 0) {
2660			struct addrinfo ai4;
2661
2662			ai4 = *pai;
2663			if (pai->ai_family == AF_UNSPEC)
2664				ai4.ai_family = AF_INET;
2665			ai = _yphostent(ypbuf, &ai4);
2666			if (ai) {
2667				cur->ai_next = ai;
2668				while (cur && cur->ai_next)
2669					cur = cur->ai_next;
2670			}
2671			free(ypbuf);
2672		}
2673	}
2674
2675done:
2676	if (sentinel.ai_next == NULL) {
2677		RES_SET_H_ERRNO(__res_state(), HOST_NOT_FOUND);
2678		return NS_NOTFOUND;
2679	}
2680	*((struct addrinfo **)rv) = sentinel.ai_next;
2681	return NS_SUCCESS;
2682}
2683#endif
2684
2685/* resolver logic */
2686
2687/*
2688 * Formulate a normal query, send, and await answer.
2689 * Returned answer is placed in supplied buffer "answer".
2690 * Perform preliminary check of answer, returning success only
2691 * if no error is indicated and the answer count is nonzero.
2692 * Return the size of the response on success, -1 on error.
2693 * Error number is left in h_errno.
2694 *
2695 * Caller must parse answer and determine whether it answers the question.
2696 */
2697static int
2698res_queryN(const char *name, struct res_target *target, res_state res)
2699{
2700	u_char *buf;
2701	HEADER *hp;
2702	int n;
2703	u_int oflags;
2704	struct res_target *t;
2705	int rcode;
2706	int ancount;
2707
2708	rcode = NOERROR;
2709	ancount = 0;
2710
2711	buf = malloc(MAXPACKET);
2712	if (!buf) {
2713		RES_SET_H_ERRNO(res, NETDB_INTERNAL);
2714		return -1;
2715	}
2716
2717	for (t = target; t; t = t->next) {
2718		int class, type;
2719		u_char *answer;
2720		int anslen;
2721
2722		hp = (HEADER *)(void *)t->answer;
2723
2724		/* make it easier... */
2725		class = t->qclass;
2726		type = t->qtype;
2727		answer = t->answer;
2728		anslen = t->anslen;
2729
2730		oflags = res->_flags;
2731
2732again:
2733		hp->rcode = NOERROR;	/* default */
2734
2735#ifdef DEBUG
2736		if (res->options & RES_DEBUG)
2737			printf(";; res_query(%s, %d, %d)\n", name, class, type);
2738#endif
2739
2740		n = res_nmkquery(res, QUERY, name, class, type, NULL, 0, NULL,
2741		    buf, MAXPACKET);
2742		if (n > 0 && (res->_flags & RES_F_EDNS0ERR) == 0 &&
2743		    (res->options & (RES_USE_EDNS0|RES_USE_DNSSEC)) != 0U)
2744			n = res_nopt(res, n, buf, MAXPACKET, anslen);
2745		if (n <= 0) {
2746#ifdef DEBUG
2747			if (res->options & RES_DEBUG)
2748				printf(";; res_query: mkquery failed\n");
2749#endif
2750			free(buf);
2751			RES_SET_H_ERRNO(res, NO_RECOVERY);
2752			return (n);
2753		}
2754		n = res_nsend(res, buf, n, answer, anslen);
2755		if (n < 0) {
2756			/*
2757			 * if the query choked with EDNS0, retry
2758			 * without EDNS0
2759			 */
2760			if ((res->options & (RES_USE_EDNS0|RES_USE_DNSSEC))
2761			    != 0U &&
2762			    ((oflags ^ res->_flags) & RES_F_EDNS0ERR) != 0) {
2763				res->_flags |= RES_F_EDNS0ERR;
2764				if (res->options & RES_DEBUG)
2765					printf(";; res_nquery: retry without EDNS0\n");
2766				goto again;
2767			}
2768			rcode = hp->rcode;	/* record most recent error */
2769#ifdef DEBUG
2770			if (res->options & RES_DEBUG)
2771				printf(";; res_query: send error\n");
2772#endif
2773			continue;
2774		}
2775
2776		if (n > anslen)
2777			hp->rcode = FORMERR; /* XXX not very informative */
2778		if (hp->rcode != NOERROR || ntohs(hp->ancount) == 0) {
2779			rcode = hp->rcode;	/* record most recent error */
2780#ifdef DEBUG
2781			if (res->options & RES_DEBUG)
2782				printf(";; rcode = %u, ancount=%u\n", hp->rcode,
2783				    ntohs(hp->ancount));
2784#endif
2785			continue;
2786		}
2787
2788		ancount += ntohs(hp->ancount);
2789
2790		t->n = n;
2791	}
2792
2793	free(buf);
2794
2795	if (ancount == 0) {
2796		switch (rcode) {
2797		case NXDOMAIN:
2798			RES_SET_H_ERRNO(res, HOST_NOT_FOUND);
2799			break;
2800		case SERVFAIL:
2801			RES_SET_H_ERRNO(res, TRY_AGAIN);
2802			break;
2803		case NOERROR:
2804			RES_SET_H_ERRNO(res, NO_DATA);
2805			break;
2806		case FORMERR:
2807		case NOTIMP:
2808		case REFUSED:
2809		default:
2810			RES_SET_H_ERRNO(res, NO_RECOVERY);
2811			break;
2812		}
2813		return (-1);
2814	}
2815	return (ancount);
2816}
2817
2818/*
2819 * Formulate a normal query, send, and retrieve answer in supplied buffer.
2820 * Return the size of the response on success, -1 on error.
2821 * If enabled, implement search rules until answer or unrecoverable failure
2822 * is detected.  Error code, if any, is left in h_errno.
2823 */
2824static int
2825res_searchN(const char *name, struct res_target *target, res_state res)
2826{
2827	const char *cp, * const *domain;
2828	HEADER *hp = (HEADER *)(void *)target->answer;	/*XXX*/
2829	u_int dots;
2830	int trailing_dot, ret, saved_herrno;
2831	int got_nodata = 0, got_servfail = 0, root_on_list = 0;
2832	int tried_as_is = 0;
2833	int searched = 0;
2834	char abuf[MAXDNAME];
2835
2836	errno = 0;
2837	RES_SET_H_ERRNO(res, HOST_NOT_FOUND); /* default, if we never query */
2838	dots = 0;
2839	for (cp = name; *cp; cp++)
2840		dots += (*cp == '.');
2841	trailing_dot = 0;
2842	if (cp > name && *--cp == '.')
2843		trailing_dot++;
2844
2845	/*
2846	 * if there aren't any dots, it could be a user-level alias
2847	 */
2848	if (!dots &&
2849	    (cp = res_hostalias(res, name, abuf, sizeof(abuf))) != NULL)
2850		return (res_queryN(cp, target, res));
2851
2852	/*
2853	 * If there are enough dots in the name, let's just give it a
2854	 * try 'as is'. The threshold can be set with the "ndots" option.
2855	 * Also, query 'as is', if there is a trailing dot in the name.
2856	 */
2857	saved_herrno = -1;
2858	if (dots >= res->ndots || trailing_dot) {
2859		ret = res_querydomainN(name, NULL, target, res);
2860		if (ret > 0 || trailing_dot)
2861			return (ret);
2862		if (errno == ECONNREFUSED) {
2863			RES_SET_H_ERRNO(res, TRY_AGAIN);
2864			return (-1);
2865		}
2866		switch (res->res_h_errno) {
2867		case NO_DATA:
2868		case HOST_NOT_FOUND:
2869			break;
2870		case TRY_AGAIN:
2871			if (hp->rcode == SERVFAIL)
2872				break;
2873			/* FALLTHROUGH */
2874		default:
2875			return (-1);
2876		}
2877		saved_herrno = res->res_h_errno;
2878		tried_as_is++;
2879	}
2880
2881	/*
2882	 * We do at least one level of search if
2883	 *	- there is no dot and RES_DEFNAME is set, or
2884	 *	- there is at least one dot, there is no trailing dot,
2885	 *	  and RES_DNSRCH is set.
2886	 */
2887	if ((!dots && (res->options & RES_DEFNAMES)) ||
2888	    (dots && !trailing_dot && (res->options & RES_DNSRCH))) {
2889		int done = 0;
2890
2891		for (domain = (const char * const *)res->dnsrch;
2892		   *domain && !done;
2893		   domain++) {
2894			searched = 1;
2895
2896			if (domain[0][0] == '\0' ||
2897			    (domain[0][0] == '.' && domain[0][1] == '\0'))
2898				root_on_list++;
2899
2900			if (root_on_list && tried_as_is)
2901				continue;
2902
2903			ret = res_querydomainN(name, *domain, target, res);
2904			if (ret > 0)
2905				return (ret);
2906
2907			/*
2908			 * If no server present, give up.
2909			 * If name isn't found in this domain,
2910			 * keep trying higher domains in the search list
2911			 * (if that's enabled).
2912			 * On a NO_DATA error, keep trying, otherwise
2913			 * a wildcard entry of another type could keep us
2914			 * from finding this entry higher in the domain.
2915			 * If we get some other error (negative answer or
2916			 * server failure), then stop searching up,
2917			 * but try the input name below in case it's
2918			 * fully-qualified.
2919			 */
2920			if (errno == ECONNREFUSED) {
2921				RES_SET_H_ERRNO(res, TRY_AGAIN);
2922				return (-1);
2923			}
2924
2925			switch (res->res_h_errno) {
2926			case NO_DATA:
2927				got_nodata++;
2928				/* FALLTHROUGH */
2929			case HOST_NOT_FOUND:
2930				/* keep trying */
2931				break;
2932			case TRY_AGAIN:
2933				got_servfail++;
2934				if (hp->rcode == SERVFAIL) {
2935					/* try next search element, if any */
2936					break;
2937				}
2938				/* FALLTHROUGH */
2939			default:
2940				/* anything else implies that we're done */
2941				done++;
2942			}
2943			/*
2944			 * if we got here for some reason other than DNSRCH,
2945			 * we only wanted one iteration of the loop, so stop.
2946			 */
2947			if (!(res->options & RES_DNSRCH))
2948			        done++;
2949		}
2950	}
2951
2952	switch (res->res_h_errno) {
2953	case NO_DATA:
2954	case HOST_NOT_FOUND:
2955		break;
2956	case TRY_AGAIN:
2957		if (hp->rcode == SERVFAIL)
2958			break;
2959		/* FALLTHROUGH */
2960	default:
2961		goto giveup;
2962	}
2963
2964	/*
2965	 * If the query has not already been tried as is then try it
2966	 * unless RES_NOTLDQUERY is set and there were no dots.
2967	 */
2968	if ((dots || !searched || !(res->options & RES_NOTLDQUERY)) &&
2969	    !(tried_as_is || root_on_list)) {
2970		ret = res_querydomainN(name, NULL, target, res);
2971		if (ret > 0)
2972			return (ret);
2973	}
2974
2975	/*
2976	 * if we got here, we didn't satisfy the search.
2977	 * if we did an initial full query, return that query's h_errno
2978	 * (note that we wouldn't be here if that query had succeeded).
2979	 * else if we ever got a nodata, send that back as the reason.
2980	 * else send back meaningless h_errno, that being the one from
2981	 * the last DNSRCH we did.
2982	 */
2983giveup:
2984	if (saved_herrno != -1)
2985		RES_SET_H_ERRNO(res, saved_herrno);
2986	else if (got_nodata)
2987		RES_SET_H_ERRNO(res, NO_DATA);
2988	else if (got_servfail)
2989		RES_SET_H_ERRNO(res, TRY_AGAIN);
2990	return (-1);
2991}
2992
2993/*
2994 * Perform a call on res_query on the concatenation of name and domain,
2995 * removing a trailing dot from name if domain is NULL.
2996 */
2997static int
2998res_querydomainN(const char *name, const char *domain,
2999    struct res_target *target, res_state res)
3000{
3001	char nbuf[MAXDNAME];
3002	const char *longname = nbuf;
3003	size_t n, d;
3004
3005#ifdef DEBUG
3006	if (res->options & RES_DEBUG)
3007		printf(";; res_querydomain(%s, %s)\n",
3008			name, domain?domain:"<Nil>");
3009#endif
3010	if (domain == NULL) {
3011		/*
3012		 * Check for trailing '.';
3013		 * copy without '.' if present.
3014		 */
3015		n = strlen(name);
3016		if (n >= MAXDNAME) {
3017			RES_SET_H_ERRNO(res, NO_RECOVERY);
3018			return (-1);
3019		}
3020		if (n > 0 && name[--n] == '.') {
3021			strncpy(nbuf, name, n);
3022			nbuf[n] = '\0';
3023		} else
3024			longname = name;
3025	} else {
3026		n = strlen(name);
3027		d = strlen(domain);
3028		if (n + d + 1 >= MAXDNAME) {
3029			RES_SET_H_ERRNO(res, NO_RECOVERY);
3030			return (-1);
3031		}
3032		snprintf(nbuf, sizeof(nbuf), "%s.%s", name, domain);
3033	}
3034	return (res_queryN(longname, target, res));
3035}
3036