Home | History | Annotate | Line # | Download | only in net
getaddrinfo.c revision 1.90
      1  1.90  christos /*	$NetBSD: getaddrinfo.c,v 1.90 2007/06/29 15:53:21 christos Exp $	*/
      2  1.48    itojun /*	$KAME: getaddrinfo.c,v 1.29 2000/08/31 17:26:57 itojun Exp $	*/
      3   1.6    itojun 
      4   1.1    itojun /*
      5   1.1    itojun  * Copyright (C) 1995, 1996, 1997, and 1998 WIDE Project.
      6   1.1    itojun  * All rights reserved.
      7  1.42    itojun  *
      8   1.1    itojun  * Redistribution and use in source and binary forms, with or without
      9   1.1    itojun  * modification, are permitted provided that the following conditions
     10   1.1    itojun  * are met:
     11   1.1    itojun  * 1. Redistributions of source code must retain the above copyright
     12   1.1    itojun  *    notice, this list of conditions and the following disclaimer.
     13   1.1    itojun  * 2. Redistributions in binary form must reproduce the above copyright
     14   1.1    itojun  *    notice, this list of conditions and the following disclaimer in the
     15   1.1    itojun  *    documentation and/or other materials provided with the distribution.
     16   1.1    itojun  * 3. Neither the name of the project nor the names of its contributors
     17   1.1    itojun  *    may be used to endorse or promote products derived from this software
     18   1.1    itojun  *    without specific prior written permission.
     19  1.42    itojun  *
     20   1.1    itojun  * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
     21   1.1    itojun  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     22   1.1    itojun  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     23   1.1    itojun  * ARE DISCLAIMED.  IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
     24   1.1    itojun  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     25   1.1    itojun  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     26   1.1    itojun  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     27   1.1    itojun  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     28   1.1    itojun  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     29   1.1    itojun  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     30   1.1    itojun  * SUCH DAMAGE.
     31   1.1    itojun  */
     32   1.1    itojun 
     33   1.1    itojun /*
     34   1.1    itojun  * Issues to be discussed:
     35   1.1    itojun  * - Return values.  There are nonstandard return values defined and used
     36  1.14    itojun  *   in the source code.  This is because RFC2553 is silent about which error
     37   1.1    itojun  *   code must be returned for which situation.
     38  1.27    itojun  * - IPv4 classful (shortened) form.  RFC2553 is silent about it.  XNET 5.2
     39  1.27    itojun  *   says to use inet_aton() to convert IPv4 numeric to binary (alows
     40  1.27    itojun  *   classful form as a result).
     41  1.27    itojun  *   current code - disallow classful form for IPv4 (due to use of inet_pton).
     42  1.27    itojun  * - freeaddrinfo(NULL).  RFC2553 is silent about it.  XNET 5.2 says it is
     43  1.27    itojun  *   invalid.
     44  1.27    itojun  *   current code - SEGV on freeaddrinfo(NULL)
     45  1.14    itojun  * Note:
     46  1.14    itojun  * - The code filters out AFs that are not supported by the kernel,
     47  1.22    itojun  *   when globbing NULL hostname (to loopback, or wildcard).  Is it the right
     48  1.14    itojun  *   thing to do?  What is the relationship with post-RFC2553 AI_ADDRCONFIG
     49  1.14    itojun  *   in ai_flags?
     50  1.30    itojun  * - (post-2553) semantics of AI_ADDRCONFIG itself is too vague.
     51  1.30    itojun  *   (1) what should we do against numeric hostname (2) what should we do
     52  1.30    itojun  *   against NULL hostname (3) what is AI_ADDRCONFIG itself.  AF not ready?
     53  1.30    itojun  *   non-loopback address configured?  global address configured?
     54   1.1    itojun  */
     55  1.39    itojun 
     56  1.39    itojun #include <sys/cdefs.h>
     57  1.39    itojun #if defined(LIBC_SCCS) && !defined(lint)
     58  1.90  christos __RCSID("$NetBSD: getaddrinfo.c,v 1.90 2007/06/29 15:53:21 christos Exp $");
     59  1.39    itojun #endif /* LIBC_SCCS and not lint */
     60   1.1    itojun 
     61  1.38    itojun #include "namespace.h"
     62  1.14    itojun #include <sys/types.h>
     63   1.1    itojun #include <sys/param.h>
     64   1.1    itojun #include <sys/socket.h>
     65  1.14    itojun #include <net/if.h>
     66   1.1    itojun #include <netinet/in.h>
     67   1.1    itojun #include <arpa/inet.h>
     68   1.1    itojun #include <arpa/nameser.h>
     69  1.51     lukem #include <assert.h>
     70  1.51     lukem #include <ctype.h>
     71  1.51     lukem #include <errno.h>
     72   1.1    itojun #include <netdb.h>
     73   1.1    itojun #include <resolv.h>
     74  1.51     lukem #include <stddef.h>
     75  1.51     lukem #include <stdio.h>
     76  1.51     lukem #include <stdlib.h>
     77  1.14    itojun #include <string.h>
     78   1.1    itojun #include <unistd.h>
     79  1.14    itojun 
     80  1.32    itojun #include <syslog.h>
     81  1.32    itojun #include <stdarg.h>
     82  1.32    itojun #include <nsswitch.h>
     83  1.32    itojun 
     84  1.32    itojun #ifdef YP
     85  1.32    itojun #include <rpc/rpc.h>
     86  1.32    itojun #include <rpcsvc/yp_prot.h>
     87  1.32    itojun #include <rpcsvc/ypclnt.h>
     88  1.38    itojun #endif
     89  1.38    itojun 
     90  1.84  christos #include "servent.h"
     91  1.84  christos 
     92  1.38    itojun #ifdef __weak_alias
     93  1.38    itojun __weak_alias(getaddrinfo,_getaddrinfo)
     94  1.38    itojun __weak_alias(freeaddrinfo,_freeaddrinfo)
     95  1.38    itojun __weak_alias(gai_strerror,_gai_strerror)
     96  1.32    itojun #endif
     97  1.32    itojun 
     98   1.1    itojun #define SUCCESS 0
     99   1.1    itojun #define ANY 0
    100   1.1    itojun #define YES 1
    101   1.1    itojun #define NO  0
    102   1.1    itojun 
    103   1.1    itojun static const char in_addrany[] = { 0, 0, 0, 0 };
    104  1.49     lukem static const char in_loopback[] = { 127, 0, 0, 1 };
    105  1.49     lukem #ifdef INET6
    106   1.1    itojun static const char in6_addrany[] = {
    107   1.1    itojun 	0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
    108   1.1    itojun };
    109   1.1    itojun static const char in6_loopback[] = {
    110   1.1    itojun 	0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1
    111   1.1    itojun };
    112  1.49     lukem #endif
    113   1.1    itojun 
    114  1.14    itojun static const struct afd {
    115   1.1    itojun 	int a_af;
    116   1.1    itojun 	int a_addrlen;
    117   1.1    itojun 	int a_socklen;
    118   1.1    itojun 	int a_off;
    119   1.1    itojun 	const char *a_addrany;
    120   1.1    itojun 	const char *a_loopback;
    121  1.14    itojun 	int a_scoped;
    122   1.1    itojun } afdl [] = {
    123   1.1    itojun #ifdef INET6
    124   1.1    itojun 	{PF_INET6, sizeof(struct in6_addr),
    125   1.1    itojun 	 sizeof(struct sockaddr_in6),
    126   1.1    itojun 	 offsetof(struct sockaddr_in6, sin6_addr),
    127  1.14    itojun 	 in6_addrany, in6_loopback, 1},
    128   1.1    itojun #endif
    129   1.1    itojun 	{PF_INET, sizeof(struct in_addr),
    130   1.1    itojun 	 sizeof(struct sockaddr_in),
    131   1.1    itojun 	 offsetof(struct sockaddr_in, sin_addr),
    132  1.14    itojun 	 in_addrany, in_loopback, 0},
    133  1.14    itojun 	{0, 0, 0, 0, NULL, NULL, 0},
    134  1.14    itojun };
    135  1.14    itojun 
    136  1.14    itojun struct explore {
    137  1.14    itojun 	int e_af;
    138  1.14    itojun 	int e_socktype;
    139  1.14    itojun 	int e_protocol;
    140  1.14    itojun 	const char *e_protostr;
    141  1.14    itojun 	int e_wild;
    142  1.14    itojun #define WILD_AF(ex)		((ex)->e_wild & 0x01)
    143  1.14    itojun #define WILD_SOCKTYPE(ex)	((ex)->e_wild & 0x02)
    144  1.14    itojun #define WILD_PROTOCOL(ex)	((ex)->e_wild & 0x04)
    145  1.14    itojun };
    146  1.14    itojun 
    147  1.14    itojun static const struct explore explore[] = {
    148  1.14    itojun #if 0
    149  1.14    itojun 	{ PF_LOCAL, 0, ANY, ANY, NULL, 0x01 },
    150  1.14    itojun #endif
    151  1.14    itojun #ifdef INET6
    152  1.14    itojun 	{ PF_INET6, SOCK_DGRAM, IPPROTO_UDP, "udp", 0x07 },
    153  1.14    itojun 	{ PF_INET6, SOCK_STREAM, IPPROTO_TCP, "tcp", 0x07 },
    154  1.14    itojun 	{ PF_INET6, SOCK_RAW, ANY, NULL, 0x05 },
    155  1.14    itojun #endif
    156  1.14    itojun 	{ PF_INET, SOCK_DGRAM, IPPROTO_UDP, "udp", 0x07 },
    157  1.14    itojun 	{ PF_INET, SOCK_STREAM, IPPROTO_TCP, "tcp", 0x07 },
    158  1.14    itojun 	{ PF_INET, SOCK_RAW, ANY, NULL, 0x05 },
    159  1.32    itojun 	{ PF_UNSPEC, SOCK_DGRAM, IPPROTO_UDP, "udp", 0x07 },
    160  1.32    itojun 	{ PF_UNSPEC, SOCK_STREAM, IPPROTO_TCP, "tcp", 0x07 },
    161  1.32    itojun 	{ PF_UNSPEC, SOCK_RAW, ANY, NULL, 0x05 },
    162  1.14    itojun 	{ -1, 0, 0, NULL, 0 },
    163   1.1    itojun };
    164   1.1    itojun 
    165   1.1    itojun #ifdef INET6
    166   1.1    itojun #define PTON_MAX	16
    167   1.1    itojun #else
    168   1.1    itojun #define PTON_MAX	4
    169   1.1    itojun #endif
    170   1.1    itojun 
    171  1.32    itojun static const ns_src default_dns_files[] = {
    172  1.32    itojun 	{ NSSRC_FILES, 	NS_SUCCESS },
    173  1.32    itojun 	{ NSSRC_DNS, 	NS_SUCCESS },
    174  1.87  christos 	{ 0, 0 }
    175  1.32    itojun };
    176  1.32    itojun 
    177  1.63    itojun #define MAXPACKET	(64*1024)
    178  1.32    itojun 
    179  1.32    itojun typedef union {
    180  1.32    itojun 	HEADER hdr;
    181  1.32    itojun 	u_char buf[MAXPACKET];
    182  1.32    itojun } querybuf;
    183  1.32    itojun 
    184  1.32    itojun struct res_target {
    185  1.32    itojun 	struct res_target *next;
    186  1.32    itojun 	const char *name;	/* domain name */
    187  1.43    itojun 	int qclass, qtype;	/* class and type of query */
    188  1.32    itojun 	u_char *answer;		/* buffer to put answer */
    189  1.32    itojun 	int anslen;		/* size of answer buffer */
    190  1.32    itojun 	int n;			/* result length */
    191  1.32    itojun };
    192   1.1    itojun 
    193  1.70  christos static int str2number(const char *);
    194  1.70  christos static int explore_fqdn(const struct addrinfo *, const char *,
    195  1.89  christos 	const char *, struct addrinfo **, struct servent_data *);
    196  1.70  christos static int explore_null(const struct addrinfo *,
    197  1.89  christos 	const char *, struct addrinfo **, struct servent_data *);
    198  1.70  christos static int explore_numeric(const struct addrinfo *, const char *,
    199  1.89  christos 	const char *, struct addrinfo **, const char *, struct servent_data *);
    200  1.70  christos static int explore_numeric_scope(const struct addrinfo *, const char *,
    201  1.89  christos 	const char *, struct addrinfo **, struct servent_data *);
    202  1.70  christos static int get_canonname(const struct addrinfo *,
    203  1.70  christos 	struct addrinfo *, const char *);
    204  1.70  christos static struct addrinfo *get_ai(const struct addrinfo *,
    205  1.70  christos 	const struct afd *, const char *);
    206  1.89  christos static int get_portmatch(const struct addrinfo *, const char *,
    207  1.89  christos     struct servent_data *);
    208  1.89  christos static int get_port(const struct addrinfo *, const char *, int,
    209  1.89  christos     struct servent_data *);
    210  1.70  christos static const struct afd *find_afd(int);
    211  1.27    itojun #ifdef INET6
    212  1.70  christos static int ip6_str2scopeid(char *, struct sockaddr_in6 *, u_int32_t *);
    213  1.42    itojun #endif
    214  1.14    itojun 
    215  1.70  christos static struct addrinfo *getanswer(const querybuf *, int, const char *, int,
    216  1.70  christos 	const struct addrinfo *);
    217  1.73    tsarna static void aisort(struct addrinfo *s, res_state res);
    218  1.70  christos static int _dns_getaddrinfo(void *, void *, va_list);
    219  1.70  christos static void _sethtent(FILE **);
    220  1.70  christos static void _endhtent(FILE **);
    221  1.70  christos static struct addrinfo *_gethtent(FILE **, const char *,
    222  1.70  christos     const struct addrinfo *);
    223  1.70  christos static int _files_getaddrinfo(void *, void *, va_list);
    224  1.32    itojun #ifdef YP
    225  1.70  christos static struct addrinfo *_yphostent(char *, const struct addrinfo *);
    226  1.70  christos static int _yp_getaddrinfo(void *, void *, va_list);
    227  1.32    itojun #endif
    228  1.32    itojun 
    229  1.70  christos static int res_queryN(const char *, struct res_target *, res_state);
    230  1.73    tsarna static int res_searchN(const char *, struct res_target *, res_state);
    231  1.70  christos static int res_querydomainN(const char *, const char *,
    232  1.70  christos 	struct res_target *, res_state);
    233  1.32    itojun 
    234  1.53  jdolecek static const char * const ai_errlist[] = {
    235   1.7     lukem 	"Success",
    236   1.7     lukem 	"Address family for hostname not supported",	/* EAI_ADDRFAMILY */
    237   1.7     lukem 	"Temporary failure in name resolution",		/* EAI_AGAIN      */
    238   1.7     lukem 	"Invalid value for ai_flags",		       	/* EAI_BADFLAGS   */
    239   1.7     lukem 	"Non-recoverable failure in name resolution", 	/* EAI_FAIL       */
    240   1.2     lukem 	"ai_family not supported",			/* EAI_FAMILY     */
    241   1.7     lukem 	"Memory allocation failure", 			/* EAI_MEMORY     */
    242   1.7     lukem 	"No address associated with hostname", 		/* EAI_NODATA     */
    243  1.14    itojun 	"hostname nor servname provided, or not known",	/* EAI_NONAME     */
    244   1.2     lukem 	"servname not supported for ai_socktype",	/* EAI_SERVICE    */
    245   1.2     lukem 	"ai_socktype not supported", 			/* EAI_SOCKTYPE   */
    246   1.7     lukem 	"System error returned in errno", 		/* EAI_SYSTEM     */
    247   1.7     lukem 	"Invalid value for hints",			/* EAI_BADHINTS	  */
    248   1.7     lukem 	"Resolved protocol is unknown",			/* EAI_PROTOCOL   */
    249  1.77  ginsbach 	"Argument buffer overflow",			/* EAI_OVERFLOW   */
    250   1.7     lukem 	"Unknown error", 				/* EAI_MAX        */
    251   1.1    itojun };
    252   1.1    itojun 
    253  1.14    itojun /* XXX macros that make external reference is BAD. */
    254  1.14    itojun 
    255  1.82    rpaulo #define GET_AI(ai, afd, addr) 					\
    256  1.82    rpaulo do { 								\
    257  1.82    rpaulo 	/* external reference: pai, error, and label free */ 	\
    258  1.82    rpaulo 	(ai) = get_ai(pai, (afd), (addr)); 			\
    259  1.82    rpaulo 	if ((ai) == NULL) { 					\
    260  1.82    rpaulo 		error = EAI_MEMORY; 				\
    261  1.82    rpaulo 		goto free; 					\
    262  1.82    rpaulo 	} 							\
    263  1.20   mycroft } while (/*CONSTCOND*/0)
    264  1.14    itojun 
    265  1.89  christos #define GET_PORT(ai, serv, svd) 				\
    266  1.82    rpaulo do { 								\
    267  1.82    rpaulo 	/* external reference: error and label free */ 		\
    268  1.89  christos 	error = get_port((ai), (serv), 0, (svd)); 		\
    269  1.82    rpaulo 	if (error != 0) 					\
    270  1.82    rpaulo 		goto free; 					\
    271  1.20   mycroft } while (/*CONSTCOND*/0)
    272  1.14    itojun 
    273  1.82    rpaulo #define GET_CANONNAME(ai, str) 					\
    274  1.82    rpaulo do { 								\
    275  1.82    rpaulo 	/* external reference: pai, error and label free */ 	\
    276  1.82    rpaulo 	error = get_canonname(pai, (ai), (str)); 		\
    277  1.82    rpaulo 	if (error != 0) 					\
    278  1.82    rpaulo 		goto free; 					\
    279  1.20   mycroft } while (/*CONSTCOND*/0)
    280  1.14    itojun 
    281  1.82    rpaulo #define ERR(err) 						\
    282  1.82    rpaulo do { 								\
    283  1.82    rpaulo 	/* external reference: error, and label bad */ 		\
    284  1.82    rpaulo 	error = (err); 						\
    285  1.82    rpaulo 	goto bad; 						\
    286  1.82    rpaulo 	/*NOTREACHED*/ 						\
    287  1.20   mycroft } while (/*CONSTCOND*/0)
    288  1.14    itojun 
    289  1.82    rpaulo #define MATCH_FAMILY(x, y, w) 						\
    290  1.82    rpaulo 	((x) == (y) || (/*CONSTCOND*/(w) && ((x) == PF_UNSPEC || 	\
    291  1.82    rpaulo 	    (y) == PF_UNSPEC)))
    292  1.82    rpaulo #define MATCH(x, y, w) 							\
    293  1.20   mycroft 	((x) == (y) || (/*CONSTCOND*/(w) && ((x) == ANY || (y) == ANY)))
    294   1.1    itojun 
    295  1.70  christos const char *
    296  1.70  christos gai_strerror(int ecode)
    297   1.1    itojun {
    298   1.1    itojun 	if (ecode < 0 || ecode > EAI_MAX)
    299   1.1    itojun 		ecode = EAI_MAX;
    300  1.70  christos 	return ai_errlist[ecode];
    301   1.1    itojun }
    302   1.1    itojun 
    303   1.1    itojun void
    304  1.70  christos freeaddrinfo(struct addrinfo *ai)
    305   1.1    itojun {
    306   1.1    itojun 	struct addrinfo *next;
    307   1.1    itojun 
    308  1.51     lukem 	_DIAGASSERT(ai != NULL);
    309  1.51     lukem 
    310  1.26    itojun 	do {
    311   1.1    itojun 		next = ai->ai_next;
    312   1.1    itojun 		if (ai->ai_canonname)
    313   1.1    itojun 			free(ai->ai_canonname);
    314   1.1    itojun 		/* no need to free(ai->ai_addr) */
    315   1.1    itojun 		free(ai);
    316  1.27    itojun 		ai = next;
    317  1.27    itojun 	} while (ai);
    318   1.1    itojun }
    319   1.1    itojun 
    320   1.1    itojun static int
    321  1.70  christos str2number(const char *p)
    322   1.1    itojun {
    323  1.45    itojun 	char *ep;
    324  1.67    itojun 	unsigned long v;
    325  1.45    itojun 
    326  1.51     lukem 	_DIAGASSERT(p != NULL);
    327  1.51     lukem 
    328  1.46    itojun 	if (*p == '\0')
    329  1.67    itojun 		return -1;
    330  1.45    itojun 	ep = NULL;
    331  1.59    itojun 	errno = 0;
    332  1.67    itojun 	v = strtoul(p, &ep, 10);
    333  1.67    itojun 	if (errno == 0 && ep && *ep == '\0' && v <= UINT_MAX)
    334  1.67    itojun 		return v;
    335  1.45    itojun 	else
    336  1.67    itojun 		return -1;
    337   1.1    itojun }
    338   1.1    itojun 
    339   1.1    itojun int
    340  1.70  christos getaddrinfo(const char *hostname, const char *servname,
    341  1.70  christos     const struct addrinfo *hints, struct addrinfo **res)
    342   1.1    itojun {
    343   1.1    itojun 	struct addrinfo sentinel;
    344   1.1    itojun 	struct addrinfo *cur;
    345  1.14    itojun 	int error = 0;
    346   1.1    itojun 	struct addrinfo ai;
    347  1.14    itojun 	struct addrinfo ai0;
    348   1.1    itojun 	struct addrinfo *pai;
    349  1.14    itojun 	const struct explore *ex;
    350  1.89  christos 	struct servent_data svd;
    351   1.1    itojun 
    352  1.51     lukem 	/* hostname is allowed to be NULL */
    353  1.51     lukem 	/* servname is allowed to be NULL */
    354  1.51     lukem 	/* hints is allowed to be NULL */
    355  1.51     lukem 	_DIAGASSERT(res != NULL);
    356  1.51     lukem 
    357  1.89  christos 	(void)memset(&svd, 0, sizeof(svd));
    358  1.32    itojun 	memset(&sentinel, 0, sizeof(sentinel));
    359   1.1    itojun 	cur = &sentinel;
    360  1.86  christos 	memset(&ai, 0, sizeof(ai));
    361   1.1    itojun 	pai = &ai;
    362   1.1    itojun 	pai->ai_flags = 0;
    363   1.1    itojun 	pai->ai_family = PF_UNSPEC;
    364   1.1    itojun 	pai->ai_socktype = ANY;
    365   1.1    itojun 	pai->ai_protocol = ANY;
    366   1.1    itojun 	pai->ai_addrlen = 0;
    367   1.1    itojun 	pai->ai_canonname = NULL;
    368   1.1    itojun 	pai->ai_addr = NULL;
    369   1.1    itojun 	pai->ai_next = NULL;
    370   1.1    itojun 
    371   1.1    itojun 	if (hostname == NULL && servname == NULL)
    372   1.1    itojun 		return EAI_NONAME;
    373   1.1    itojun 	if (hints) {
    374   1.1    itojun 		/* error check for hints */
    375   1.1    itojun 		if (hints->ai_addrlen || hints->ai_canonname ||
    376   1.1    itojun 		    hints->ai_addr || hints->ai_next)
    377   1.1    itojun 			ERR(EAI_BADHINTS); /* xxx */
    378   1.1    itojun 		if (hints->ai_flags & ~AI_MASK)
    379   1.1    itojun 			ERR(EAI_BADFLAGS);
    380   1.1    itojun 		switch (hints->ai_family) {
    381   1.1    itojun 		case PF_UNSPEC:
    382   1.1    itojun 		case PF_INET:
    383   1.1    itojun #ifdef INET6
    384   1.1    itojun 		case PF_INET6:
    385   1.1    itojun #endif
    386   1.1    itojun 			break;
    387   1.1    itojun 		default:
    388   1.1    itojun 			ERR(EAI_FAMILY);
    389   1.1    itojun 		}
    390   1.1    itojun 		memcpy(pai, hints, sizeof(*pai));
    391  1.14    itojun 
    392  1.14    itojun 		/*
    393  1.14    itojun 		 * if both socktype/protocol are specified, check if they
    394  1.14    itojun 		 * are meaningful combination.
    395  1.14    itojun 		 */
    396  1.14    itojun 		if (pai->ai_socktype != ANY && pai->ai_protocol != ANY) {
    397  1.14    itojun 			for (ex = explore; ex->e_af >= 0; ex++) {
    398  1.14    itojun 				if (pai->ai_family != ex->e_af)
    399  1.14    itojun 					continue;
    400  1.14    itojun 				if (ex->e_socktype == ANY)
    401  1.14    itojun 					continue;
    402  1.14    itojun 				if (ex->e_protocol == ANY)
    403  1.14    itojun 					continue;
    404  1.14    itojun 				if (pai->ai_socktype == ex->e_socktype
    405  1.14    itojun 				 && pai->ai_protocol != ex->e_protocol) {
    406  1.14    itojun 					ERR(EAI_BADHINTS);
    407  1.14    itojun 				}
    408   1.1    itojun 			}
    409   1.1    itojun 		}
    410   1.1    itojun 	}
    411   1.1    itojun 
    412   1.1    itojun 	/*
    413  1.14    itojun 	 * check for special cases.  (1) numeric servname is disallowed if
    414  1.14    itojun 	 * socktype/protocol are left unspecified. (2) servname is disallowed
    415  1.14    itojun 	 * for raw and other inet{,6} sockets.
    416   1.1    itojun 	 */
    417  1.14    itojun 	if (MATCH_FAMILY(pai->ai_family, PF_INET, 1)
    418  1.27    itojun #ifdef PF_INET6
    419  1.27    itojun 	 || MATCH_FAMILY(pai->ai_family, PF_INET6, 1)
    420  1.27    itojun #endif
    421  1.27    itojun 	    ) {
    422  1.23    itojun 		ai0 = *pai;	/* backup *pai */
    423  1.21   mycroft 
    424  1.27    itojun 		if (pai->ai_family == PF_UNSPEC) {
    425  1.27    itojun #ifdef PF_INET6
    426  1.14    itojun 			pai->ai_family = PF_INET6;
    427  1.27    itojun #else
    428  1.27    itojun 			pai->ai_family = PF_INET;
    429  1.27    itojun #endif
    430  1.27    itojun 		}
    431  1.89  christos 		error = get_portmatch(pai, servname, &svd);
    432  1.14    itojun 		if (error)
    433  1.14    itojun 			ERR(error);
    434  1.21   mycroft 
    435  1.21   mycroft 		*pai = ai0;
    436  1.14    itojun 	}
    437  1.14    itojun 
    438  1.14    itojun 	ai0 = *pai;
    439  1.14    itojun 
    440  1.14    itojun 	/* NULL hostname, or numeric hostname */
    441  1.14    itojun 	for (ex = explore; ex->e_af >= 0; ex++) {
    442  1.14    itojun 		*pai = ai0;
    443  1.14    itojun 
    444  1.32    itojun 		/* PF_UNSPEC entries are prepared for DNS queries only */
    445  1.32    itojun 		if (ex->e_af == PF_UNSPEC)
    446  1.32    itojun 			continue;
    447  1.32    itojun 
    448  1.14    itojun 		if (!MATCH_FAMILY(pai->ai_family, ex->e_af, WILD_AF(ex)))
    449  1.14    itojun 			continue;
    450  1.14    itojun 		if (!MATCH(pai->ai_socktype, ex->e_socktype, WILD_SOCKTYPE(ex)))
    451  1.14    itojun 			continue;
    452  1.14    itojun 		if (!MATCH(pai->ai_protocol, ex->e_protocol, WILD_PROTOCOL(ex)))
    453  1.14    itojun 			continue;
    454  1.14    itojun 
    455  1.14    itojun 		if (pai->ai_family == PF_UNSPEC)
    456  1.14    itojun 			pai->ai_family = ex->e_af;
    457  1.14    itojun 		if (pai->ai_socktype == ANY && ex->e_socktype != ANY)
    458  1.14    itojun 			pai->ai_socktype = ex->e_socktype;
    459  1.14    itojun 		if (pai->ai_protocol == ANY && ex->e_protocol != ANY)
    460  1.14    itojun 			pai->ai_protocol = ex->e_protocol;
    461  1.14    itojun 
    462  1.14    itojun 		if (hostname == NULL)
    463  1.89  christos 			error = explore_null(pai, servname, &cur->ai_next,
    464  1.89  christos 			    &svd);
    465  1.14    itojun 		else
    466  1.65    itojun 			error = explore_numeric_scope(pai, hostname, servname,
    467  1.89  christos 			    &cur->ai_next, &svd);
    468  1.14    itojun 
    469  1.14    itojun 		if (error)
    470  1.14    itojun 			goto free;
    471  1.14    itojun 
    472  1.79  christos 		while (cur->ai_next)
    473  1.14    itojun 			cur = cur->ai_next;
    474  1.14    itojun 	}
    475  1.14    itojun 
    476  1.14    itojun 	/*
    477  1.14    itojun 	 * XXX
    478  1.76  ginsbach 	 * If numeric representation of AF1 can be interpreted as FQDN
    479  1.14    itojun 	 * representation of AF2, we need to think again about the code below.
    480  1.14    itojun 	 */
    481  1.14    itojun 	if (sentinel.ai_next)
    482  1.14    itojun 		goto good;
    483  1.14    itojun 
    484  1.14    itojun 	if (hostname == NULL)
    485  1.43    itojun 		ERR(EAI_NODATA);
    486  1.64    itojun 	if (pai->ai_flags & AI_NUMERICHOST)
    487  1.64    itojun 		ERR(EAI_NONAME);
    488  1.14    itojun 
    489  1.14    itojun 	/*
    490  1.14    itojun 	 * hostname as alphabetical name.
    491  1.14    itojun 	 * we would like to prefer AF_INET6 than AF_INET, so we'll make a
    492  1.14    itojun 	 * outer loop by AFs.
    493  1.14    itojun 	 */
    494  1.32    itojun 	for (ex = explore; ex->e_af >= 0; ex++) {
    495  1.14    itojun 		*pai = ai0;
    496  1.14    itojun 
    497  1.32    itojun 		/* require exact match for family field */
    498  1.32    itojun 		if (pai->ai_family != ex->e_af)
    499  1.14    itojun 			continue;
    500  1.14    itojun 
    501  1.32    itojun 		if (!MATCH(pai->ai_socktype, ex->e_socktype,
    502  1.32    itojun 				WILD_SOCKTYPE(ex))) {
    503  1.32    itojun 			continue;
    504  1.32    itojun 		}
    505  1.32    itojun 		if (!MATCH(pai->ai_protocol, ex->e_protocol,
    506  1.32    itojun 				WILD_PROTOCOL(ex))) {
    507  1.32    itojun 			continue;
    508  1.32    itojun 		}
    509  1.14    itojun 
    510  1.32    itojun 		if (pai->ai_socktype == ANY && ex->e_socktype != ANY)
    511  1.32    itojun 			pai->ai_socktype = ex->e_socktype;
    512  1.32    itojun 		if (pai->ai_protocol == ANY && ex->e_protocol != ANY)
    513  1.32    itojun 			pai->ai_protocol = ex->e_protocol;
    514   1.1    itojun 
    515  1.89  christos 		error = explore_fqdn(pai, hostname, servname, &cur->ai_next,
    516  1.89  christos 		    &svd);
    517  1.14    itojun 
    518  1.32    itojun 		while (cur && cur->ai_next)
    519  1.32    itojun 			cur = cur->ai_next;
    520   1.1    itojun 	}
    521  1.14    itojun 
    522  1.14    itojun 	/* XXX */
    523  1.14    itojun 	if (sentinel.ai_next)
    524  1.14    itojun 		error = 0;
    525  1.14    itojun 
    526  1.14    itojun 	if (error)
    527  1.14    itojun 		goto free;
    528  1.89  christos 
    529  1.89  christos 	if (sentinel.ai_next) {
    530  1.14    itojun  good:
    531  1.89  christos 		endservent_r(&svd);
    532  1.89  christos 		*res = sentinel.ai_next;
    533  1.89  christos 		return SUCCESS;
    534  1.89  christos 	} else
    535  1.89  christos 		error = EAI_FAIL;
    536  1.14    itojun  free:
    537  1.14    itojun  bad:
    538  1.89  christos 	endservent_r(&svd);
    539  1.14    itojun 	if (sentinel.ai_next)
    540  1.14    itojun 		freeaddrinfo(sentinel.ai_next);
    541  1.14    itojun 	*res = NULL;
    542  1.14    itojun 	return error;
    543  1.14    itojun }
    544  1.14    itojun 
    545  1.14    itojun /*
    546  1.14    itojun  * FQDN hostname, DNS lookup
    547  1.14    itojun  */
    548  1.14    itojun static int
    549  1.70  christos explore_fqdn(const struct addrinfo *pai, const char *hostname,
    550  1.89  christos     const char *servname, struct addrinfo **res, struct servent_data *svd)
    551  1.14    itojun {
    552  1.32    itojun 	struct addrinfo *result;
    553  1.32    itojun 	struct addrinfo *cur;
    554  1.27    itojun 	int error = 0;
    555  1.32    itojun 	static const ns_dtab dtab[] = {
    556  1.32    itojun 		NS_FILES_CB(_files_getaddrinfo, NULL)
    557  1.32    itojun 		{ NSSRC_DNS, _dns_getaddrinfo, NULL },	/* force -DHESIOD */
    558  1.32    itojun 		NS_NIS_CB(_yp_getaddrinfo, NULL)
    559  1.87  christos 		NS_NULL_CB
    560  1.32    itojun 	};
    561  1.14    itojun 
    562  1.51     lukem 	_DIAGASSERT(pai != NULL);
    563  1.51     lukem 	/* hostname may be NULL */
    564  1.51     lukem 	/* servname may be NULL */
    565  1.51     lukem 	_DIAGASSERT(res != NULL);
    566  1.51     lukem 
    567  1.32    itojun 	result = NULL;
    568  1.14    itojun 
    569  1.14    itojun 	/*
    570  1.14    itojun 	 * if the servname does not match socktype/protocol, ignore it.
    571  1.14    itojun 	 */
    572  1.89  christos 	if (get_portmatch(pai, servname, svd) != 0)
    573  1.14    itojun 		return 0;
    574  1.14    itojun 
    575  1.32    itojun 	switch (nsdispatch(&result, dtab, NSDB_HOSTS, "getaddrinfo",
    576  1.32    itojun 			default_dns_files, hostname, pai)) {
    577  1.32    itojun 	case NS_TRYAGAIN:
    578  1.32    itojun 		error = EAI_AGAIN;
    579  1.32    itojun 		goto free;
    580  1.32    itojun 	case NS_UNAVAIL:
    581  1.14    itojun 		error = EAI_FAIL;
    582  1.14    itojun 		goto free;
    583  1.32    itojun 	case NS_NOTFOUND:
    584  1.32    itojun 		error = EAI_NODATA;
    585  1.15    itojun 		goto free;
    586  1.32    itojun 	case NS_SUCCESS:
    587  1.32    itojun 		error = 0;
    588  1.32    itojun 		for (cur = result; cur; cur = cur->ai_next) {
    589  1.89  christos 			GET_PORT(cur, servname, svd);
    590  1.32    itojun 			/* canonname should be filled already */
    591  1.16    itojun 		}
    592  1.32    itojun 		break;
    593  1.16    itojun 	}
    594  1.15    itojun 
    595  1.32    itojun 	*res = result;
    596  1.14    itojun 
    597  1.14    itojun 	return 0;
    598  1.14    itojun 
    599  1.14    itojun free:
    600  1.34    itojun 	if (result)
    601  1.34    itojun 		freeaddrinfo(result);
    602  1.14    itojun 	return error;
    603  1.14    itojun }
    604  1.14    itojun 
    605  1.14    itojun /*
    606  1.14    itojun  * hostname == NULL.
    607  1.14    itojun  * passive socket -> anyaddr (0.0.0.0 or ::)
    608  1.14    itojun  * non-passive socket -> localhost (127.0.0.1 or ::1)
    609  1.14    itojun  */
    610  1.14    itojun static int
    611  1.70  christos explore_null(const struct addrinfo *pai, const char *servname,
    612  1.89  christos     struct addrinfo **res, struct servent_data *svd)
    613  1.14    itojun {
    614  1.14    itojun 	int s;
    615  1.14    itojun 	const struct afd *afd;
    616  1.14    itojun 	struct addrinfo *cur;
    617  1.14    itojun 	struct addrinfo sentinel;
    618  1.14    itojun 	int error;
    619  1.14    itojun 
    620  1.51     lukem 	_DIAGASSERT(pai != NULL);
    621  1.51     lukem 	/* servname may be NULL */
    622  1.51     lukem 	_DIAGASSERT(res != NULL);
    623  1.51     lukem 
    624  1.14    itojun 	*res = NULL;
    625  1.14    itojun 	sentinel.ai_next = NULL;
    626  1.14    itojun 	cur = &sentinel;
    627  1.14    itojun 
    628  1.14    itojun 	/*
    629  1.14    itojun 	 * filter out AFs that are not supported by the kernel
    630  1.14    itojun 	 * XXX errno?
    631  1.14    itojun 	 */
    632  1.14    itojun 	s = socket(pai->ai_family, SOCK_DGRAM, 0);
    633  1.14    itojun 	if (s < 0) {
    634  1.14    itojun 		if (errno != EMFILE)
    635  1.14    itojun 			return 0;
    636  1.14    itojun 	} else
    637  1.14    itojun 		close(s);
    638  1.14    itojun 
    639  1.14    itojun 	/*
    640  1.14    itojun 	 * if the servname does not match socktype/protocol, ignore it.
    641  1.14    itojun 	 */
    642  1.89  christos 	if (get_portmatch(pai, servname, svd) != 0)
    643  1.14    itojun 		return 0;
    644  1.14    itojun 
    645  1.14    itojun 	afd = find_afd(pai->ai_family);
    646  1.42    itojun 	if (afd == NULL)
    647  1.42    itojun 		return 0;
    648  1.14    itojun 
    649  1.14    itojun 	if (pai->ai_flags & AI_PASSIVE) {
    650  1.14    itojun 		GET_AI(cur->ai_next, afd, afd->a_addrany);
    651  1.14    itojun 		/* xxx meaningless?
    652  1.14    itojun 		 * GET_CANONNAME(cur->ai_next, "anyaddr");
    653  1.14    itojun 		 */
    654  1.89  christos 		GET_PORT(cur->ai_next, servname, svd);
    655  1.14    itojun 	} else {
    656  1.14    itojun 		GET_AI(cur->ai_next, afd, afd->a_loopback);
    657  1.14    itojun 		/* xxx meaningless?
    658  1.14    itojun 		 * GET_CANONNAME(cur->ai_next, "localhost");
    659  1.14    itojun 		 */
    660  1.89  christos 		GET_PORT(cur->ai_next, servname, svd);
    661  1.14    itojun 	}
    662  1.14    itojun 	cur = cur->ai_next;
    663  1.14    itojun 
    664  1.14    itojun 	*res = sentinel.ai_next;
    665  1.14    itojun 	return 0;
    666  1.14    itojun 
    667  1.14    itojun free:
    668  1.14    itojun 	if (sentinel.ai_next)
    669  1.14    itojun 		freeaddrinfo(sentinel.ai_next);
    670  1.14    itojun 	return error;
    671  1.14    itojun }
    672  1.14    itojun 
    673  1.14    itojun /*
    674  1.14    itojun  * numeric hostname
    675  1.14    itojun  */
    676  1.14    itojun static int
    677  1.70  christos explore_numeric(const struct addrinfo *pai, const char *hostname,
    678  1.89  christos     const char *servname, struct addrinfo **res, const char *canonname,
    679  1.89  christos     struct servent_data *svd)
    680  1.14    itojun {
    681  1.14    itojun 	const struct afd *afd;
    682  1.14    itojun 	struct addrinfo *cur;
    683  1.14    itojun 	struct addrinfo sentinel;
    684  1.14    itojun 	int error;
    685  1.14    itojun 	char pton[PTON_MAX];
    686  1.14    itojun 
    687  1.51     lukem 	_DIAGASSERT(pai != NULL);
    688  1.51     lukem 	/* hostname may be NULL */
    689  1.51     lukem 	/* servname may be NULL */
    690  1.51     lukem 	_DIAGASSERT(res != NULL);
    691  1.51     lukem 
    692  1.14    itojun 	*res = NULL;
    693  1.14    itojun 	sentinel.ai_next = NULL;
    694  1.14    itojun 	cur = &sentinel;
    695  1.14    itojun 
    696  1.14    itojun 	/*
    697  1.14    itojun 	 * if the servname does not match socktype/protocol, ignore it.
    698  1.14    itojun 	 */
    699  1.89  christos 	if (get_portmatch(pai, servname, svd) != 0)
    700  1.14    itojun 		return 0;
    701  1.14    itojun 
    702  1.14    itojun 	afd = find_afd(pai->ai_family);
    703  1.42    itojun 	if (afd == NULL)
    704  1.42    itojun 		return 0;
    705  1.14    itojun 
    706  1.27    itojun 	switch (afd->a_af) {
    707  1.27    itojun #if 0 /*X/Open spec*/
    708  1.27    itojun 	case AF_INET:
    709  1.27    itojun 		if (inet_aton(hostname, (struct in_addr *)pton) == 1) {
    710  1.27    itojun 			if (pai->ai_family == afd->a_af ||
    711  1.27    itojun 			    pai->ai_family == PF_UNSPEC /*?*/) {
    712  1.27    itojun 				GET_AI(cur->ai_next, afd, pton);
    713  1.89  christos 				GET_PORT(cur->ai_next, servname, svd);
    714  1.65    itojun 				if ((pai->ai_flags & AI_CANONNAME)) {
    715  1.65    itojun 					/*
    716  1.65    itojun 					 * Set the numeric address itself as
    717  1.65    itojun 					 * the canonical name, based on a
    718  1.65    itojun 					 * clarification in rfc2553bis-03.
    719  1.65    itojun 					 */
    720  1.65    itojun 					GET_CANONNAME(cur->ai_next, canonname);
    721  1.65    itojun 				}
    722  1.27    itojun 				while (cur && cur->ai_next)
    723  1.27    itojun 					cur = cur->ai_next;
    724  1.42    itojun 			} else
    725  1.27    itojun 				ERR(EAI_FAMILY);	/*xxx*/
    726  1.27    itojun 		}
    727  1.27    itojun 		break;
    728  1.27    itojun #endif
    729  1.27    itojun 	default:
    730  1.27    itojun 		if (inet_pton(afd->a_af, hostname, pton) == 1) {
    731  1.27    itojun 			if (pai->ai_family == afd->a_af ||
    732  1.27    itojun 			    pai->ai_family == PF_UNSPEC /*?*/) {
    733  1.27    itojun 				GET_AI(cur->ai_next, afd, pton);
    734  1.89  christos 				GET_PORT(cur->ai_next, servname, svd);
    735  1.65    itojun 				if ((pai->ai_flags & AI_CANONNAME)) {
    736  1.65    itojun 					/*
    737  1.65    itojun 					 * Set the numeric address itself as
    738  1.65    itojun 					 * the canonical name, based on a
    739  1.65    itojun 					 * clarification in rfc2553bis-03.
    740  1.65    itojun 					 */
    741  1.65    itojun 					GET_CANONNAME(cur->ai_next, canonname);
    742  1.65    itojun 				}
    743  1.81  christos 				while (cur->ai_next)
    744  1.27    itojun 					cur = cur->ai_next;
    745  1.42    itojun 			} else
    746  1.27    itojun 				ERR(EAI_FAMILY);	/*xxx*/
    747  1.27    itojun 		}
    748  1.27    itojun 		break;
    749   1.1    itojun 	}
    750   1.1    itojun 
    751  1.14    itojun 	*res = sentinel.ai_next;
    752  1.14    itojun 	return 0;
    753  1.14    itojun 
    754  1.14    itojun free:
    755  1.14    itojun bad:
    756  1.14    itojun 	if (sentinel.ai_next)
    757  1.14    itojun 		freeaddrinfo(sentinel.ai_next);
    758  1.14    itojun 	return error;
    759  1.14    itojun }
    760  1.14    itojun 
    761  1.14    itojun /*
    762  1.14    itojun  * numeric hostname with scope
    763  1.14    itojun  */
    764  1.14    itojun static int
    765  1.70  christos explore_numeric_scope(const struct addrinfo *pai, const char *hostname,
    766  1.89  christos     const char *servname, struct addrinfo **res, struct servent_data *svd)
    767  1.14    itojun {
    768  1.29    itojun #if !defined(SCOPE_DELIMITER) || !defined(INET6)
    769  1.90  christos 	return explore_numeric(pai, hostname, servname, res, hostname, svd);
    770  1.14    itojun #else
    771  1.14    itojun 	const struct afd *afd;
    772  1.14    itojun 	struct addrinfo *cur;
    773  1.14    itojun 	int error;
    774  1.30    itojun 	char *cp, *hostname2 = NULL, *scope, *addr;
    775  1.14    itojun 	struct sockaddr_in6 *sin6;
    776  1.14    itojun 
    777  1.51     lukem 	_DIAGASSERT(pai != NULL);
    778  1.51     lukem 	/* hostname may be NULL */
    779  1.51     lukem 	/* servname may be NULL */
    780  1.51     lukem 	_DIAGASSERT(res != NULL);
    781  1.51     lukem 
    782  1.14    itojun 	/*
    783  1.14    itojun 	 * if the servname does not match socktype/protocol, ignore it.
    784  1.14    itojun 	 */
    785  1.89  christos 	if (get_portmatch(pai, servname, svd) != 0)
    786  1.14    itojun 		return 0;
    787  1.14    itojun 
    788  1.14    itojun 	afd = find_afd(pai->ai_family);
    789  1.42    itojun 	if (afd == NULL)
    790  1.42    itojun 		return 0;
    791  1.42    itojun 
    792  1.14    itojun 	if (!afd->a_scoped)
    793  1.89  christos 		return explore_numeric(pai, hostname, servname, res, hostname,
    794  1.89  christos 		    svd);
    795  1.14    itojun 
    796  1.14    itojun 	cp = strchr(hostname, SCOPE_DELIMITER);
    797  1.14    itojun 	if (cp == NULL)
    798  1.89  christos 		return explore_numeric(pai, hostname, servname, res, hostname,
    799  1.89  christos 		    svd);
    800  1.14    itojun 
    801  1.30    itojun 	/*
    802  1.30    itojun 	 * Handle special case of <scoped_address><delimiter><scope id>
    803  1.30    itojun 	 */
    804  1.30    itojun 	hostname2 = strdup(hostname);
    805  1.30    itojun 	if (hostname2 == NULL)
    806  1.30    itojun 		return EAI_MEMORY;
    807  1.30    itojun 	/* terminate at the delimiter */
    808  1.30    itojun 	hostname2[cp - hostname] = '\0';
    809  1.30    itojun 	addr = hostname2;
    810  1.30    itojun 	scope = cp + 1;
    811   1.1    itojun 
    812  1.89  christos 	error = explore_numeric(pai, addr, servname, res, hostname, svd);
    813   1.1    itojun 	if (error == 0) {
    814  1.59    itojun 		u_int32_t scopeid;
    815  1.27    itojun 
    816  1.14    itojun 		for (cur = *res; cur; cur = cur->ai_next) {
    817  1.14    itojun 			if (cur->ai_family != AF_INET6)
    818  1.14    itojun 				continue;
    819  1.36  christos 			sin6 = (struct sockaddr_in6 *)(void *)cur->ai_addr;
    820  1.60    itojun 			if (ip6_str2scopeid(scope, sin6, &scopeid) == -1) {
    821  1.27    itojun 				free(hostname2);
    822  1.43    itojun 				return(EAI_NODATA); /* XXX: is return OK? */
    823  1.27    itojun 			}
    824  1.27    itojun 			sin6->sin6_scope_id = scopeid;
    825  1.14    itojun 		}
    826   1.1    itojun 	}
    827  1.14    itojun 
    828  1.14    itojun 	free(hostname2);
    829  1.14    itojun 
    830   1.1    itojun 	return error;
    831  1.14    itojun #endif
    832   1.1    itojun }
    833   1.1    itojun 
    834   1.1    itojun static int
    835  1.70  christos get_canonname(const struct addrinfo *pai, struct addrinfo *ai, const char *str)
    836   1.1    itojun {
    837  1.51     lukem 
    838  1.51     lukem 	_DIAGASSERT(pai != NULL);
    839  1.51     lukem 	_DIAGASSERT(ai != NULL);
    840  1.51     lukem 	_DIAGASSERT(str != NULL);
    841  1.51     lukem 
    842  1.14    itojun 	if ((pai->ai_flags & AI_CANONNAME) != 0) {
    843  1.66    itojun 		ai->ai_canonname = strdup(str);
    844  1.14    itojun 		if (ai->ai_canonname == NULL)
    845  1.14    itojun 			return EAI_MEMORY;
    846  1.14    itojun 	}
    847  1.14    itojun 	return 0;
    848  1.14    itojun }
    849   1.1    itojun 
    850  1.14    itojun static struct addrinfo *
    851  1.70  christos get_ai(const struct addrinfo *pai, const struct afd *afd, const char *addr)
    852  1.14    itojun {
    853  1.14    itojun 	char *p;
    854  1.14    itojun 	struct addrinfo *ai;
    855  1.12     lukem 
    856  1.51     lukem 	_DIAGASSERT(pai != NULL);
    857  1.51     lukem 	_DIAGASSERT(afd != NULL);
    858  1.51     lukem 	_DIAGASSERT(addr != NULL);
    859  1.51     lukem 
    860  1.14    itojun 	ai = (struct addrinfo *)malloc(sizeof(struct addrinfo)
    861  1.14    itojun 		+ (afd->a_socklen));
    862  1.14    itojun 	if (ai == NULL)
    863  1.14    itojun 		return NULL;
    864  1.14    itojun 
    865  1.14    itojun 	memcpy(ai, pai, sizeof(struct addrinfo));
    866  1.36  christos 	ai->ai_addr = (struct sockaddr *)(void *)(ai + 1);
    867  1.20   mycroft 	memset(ai->ai_addr, 0, (size_t)afd->a_socklen);
    868  1.14    itojun 	ai->ai_addr->sa_len = afd->a_socklen;
    869  1.14    itojun 	ai->ai_addrlen = afd->a_socklen;
    870  1.14    itojun 	ai->ai_addr->sa_family = ai->ai_family = afd->a_af;
    871  1.36  christos 	p = (char *)(void *)(ai->ai_addr);
    872  1.20   mycroft 	memcpy(p + afd->a_off, addr, (size_t)afd->a_addrlen);
    873  1.14    itojun 	return ai;
    874  1.14    itojun }
    875   1.1    itojun 
    876  1.14    itojun static int
    877  1.89  christos get_portmatch(const struct addrinfo *ai, const char *servname,
    878  1.89  christos     struct servent_data *svd)
    879  1.14    itojun {
    880   1.4    itojun 
    881  1.51     lukem 	_DIAGASSERT(ai != NULL);
    882  1.51     lukem 	/* servname may be NULL */
    883  1.51     lukem 
    884  1.89  christos 	return get_port(ai, servname, 1, svd);
    885   1.1    itojun }
    886   1.1    itojun 
    887   1.1    itojun static int
    888  1.89  christos get_port(const struct addrinfo *ai, const char *servname, int matchonly,
    889  1.89  christos     struct servent_data *svd)
    890   1.1    itojun {
    891  1.14    itojun 	const char *proto;
    892  1.14    itojun 	struct servent *sp;
    893  1.14    itojun 	int port;
    894  1.14    itojun 	int allownumeric;
    895  1.12     lukem 
    896  1.51     lukem 	_DIAGASSERT(ai != NULL);
    897  1.51     lukem 	/* servname may be NULL */
    898  1.51     lukem 
    899  1.14    itojun 	if (servname == NULL)
    900  1.14    itojun 		return 0;
    901  1.23    itojun 	switch (ai->ai_family) {
    902  1.23    itojun 	case AF_INET:
    903  1.27    itojun #ifdef AF_INET6
    904  1.23    itojun 	case AF_INET6:
    905  1.27    itojun #endif
    906  1.23    itojun 		break;
    907  1.23    itojun 	default:
    908  1.14    itojun 		return 0;
    909  1.23    itojun 	}
    910   1.1    itojun 
    911  1.14    itojun 	switch (ai->ai_socktype) {
    912  1.14    itojun 	case SOCK_RAW:
    913  1.14    itojun 		return EAI_SERVICE;
    914  1.14    itojun 	case SOCK_DGRAM:
    915  1.14    itojun 	case SOCK_STREAM:
    916  1.14    itojun 		allownumeric = 1;
    917  1.14    itojun 		break;
    918  1.14    itojun 	case ANY:
    919  1.14    itojun 		allownumeric = 0;
    920  1.14    itojun 		break;
    921  1.14    itojun 	default:
    922  1.14    itojun 		return EAI_SOCKTYPE;
    923   1.1    itojun 	}
    924  1.14    itojun 
    925  1.67    itojun 	port = str2number(servname);
    926  1.67    itojun 	if (port >= 0) {
    927  1.14    itojun 		if (!allownumeric)
    928  1.14    itojun 			return EAI_SERVICE;
    929  1.14    itojun 		if (port < 0 || port > 65535)
    930  1.14    itojun 			return EAI_SERVICE;
    931  1.58    itojun 		port = htons(port);
    932  1.14    itojun 	} else {
    933  1.84  christos 		struct servent sv;
    934  1.67    itojun 		if (ai->ai_flags & AI_NUMERICSERV)
    935  1.67    itojun 			return EAI_NONAME;
    936  1.67    itojun 
    937  1.14    itojun 		switch (ai->ai_socktype) {
    938  1.14    itojun 		case SOCK_DGRAM:
    939  1.14    itojun 			proto = "udp";
    940   1.1    itojun 			break;
    941  1.14    itojun 		case SOCK_STREAM:
    942  1.14    itojun 			proto = "tcp";
    943   1.1    itojun 			break;
    944   1.1    itojun 		default:
    945  1.14    itojun 			proto = NULL;
    946   1.1    itojun 			break;
    947   1.1    itojun 		}
    948  1.14    itojun 
    949  1.89  christos 		sp = getservbyname_r(servname, proto, &sv, svd);
    950  1.85  christos 		if (sp == NULL)
    951  1.14    itojun 			return EAI_SERVICE;
    952  1.14    itojun 		port = sp->s_port;
    953   1.1    itojun 	}
    954   1.1    itojun 
    955  1.14    itojun 	if (!matchonly) {
    956  1.14    itojun 		switch (ai->ai_family) {
    957   1.1    itojun 		case AF_INET:
    958  1.36  christos 			((struct sockaddr_in *)(void *)
    959  1.36  christos 			    ai->ai_addr)->sin_port = port;
    960   1.1    itojun 			break;
    961   1.1    itojun #ifdef INET6
    962  1.14    itojun 		case AF_INET6:
    963  1.36  christos 			((struct sockaddr_in6 *)(void *)
    964  1.36  christos 			    ai->ai_addr)->sin6_port = port;
    965   1.1    itojun 			break;
    966   1.1    itojun #endif
    967   1.1    itojun 		}
    968  1.14    itojun 	}
    969  1.14    itojun 
    970  1.14    itojun 	return 0;
    971  1.14    itojun }
    972  1.14    itojun 
    973  1.14    itojun static const struct afd *
    974  1.70  christos find_afd(int af)
    975  1.14    itojun {
    976  1.14    itojun 	const struct afd *afd;
    977   1.1    itojun 
    978  1.14    itojun 	if (af == PF_UNSPEC)
    979  1.14    itojun 		return NULL;
    980  1.14    itojun 	for (afd = afdl; afd->a_af; afd++) {
    981  1.14    itojun 		if (afd->a_af == af)
    982  1.14    itojun 			return afd;
    983   1.1    itojun 	}
    984  1.14    itojun 	return NULL;
    985   1.1    itojun }
    986  1.27    itojun 
    987  1.27    itojun #ifdef INET6
    988  1.27    itojun /* convert a string to a scope identifier. XXX: IPv6 specific */
    989  1.60    itojun static int
    990  1.70  christos ip6_str2scopeid(char *scope, struct sockaddr_in6 *sin6, u_int32_t *scopeid)
    991  1.27    itojun {
    992  1.59    itojun 	u_long lscopeid;
    993  1.51     lukem 	struct in6_addr *a6;
    994  1.27    itojun 	char *ep;
    995  1.46    itojun 
    996  1.51     lukem 	_DIAGASSERT(scope != NULL);
    997  1.51     lukem 	_DIAGASSERT(sin6 != NULL);
    998  1.60    itojun 	_DIAGASSERT(scopeid != NULL);
    999  1.51     lukem 
   1000  1.51     lukem 	a6 = &sin6->sin6_addr;
   1001  1.51     lukem 
   1002  1.46    itojun 	/* empty scopeid portion is invalid */
   1003  1.46    itojun 	if (*scope == '\0')
   1004  1.46    itojun 		return -1;
   1005  1.27    itojun 
   1006  1.27    itojun 	if (IN6_IS_ADDR_LINKLOCAL(a6) || IN6_IS_ADDR_MC_LINKLOCAL(a6)) {
   1007  1.27    itojun 		/*
   1008  1.27    itojun 		 * We currently assume a one-to-one mapping between links
   1009  1.27    itojun 		 * and interfaces, so we simply use interface indices for
   1010  1.27    itojun 		 * like-local scopes.
   1011  1.27    itojun 		 */
   1012  1.60    itojun 		*scopeid = if_nametoindex(scope);
   1013  1.60    itojun 		if (*scopeid == 0)
   1014  1.27    itojun 			goto trynumeric;
   1015  1.60    itojun 		return 0;
   1016  1.27    itojun 	}
   1017  1.27    itojun 
   1018  1.27    itojun 	/* still unclear about literal, allow numeric only - placeholder */
   1019  1.27    itojun 	if (IN6_IS_ADDR_SITELOCAL(a6) || IN6_IS_ADDR_MC_SITELOCAL(a6))
   1020  1.27    itojun 		goto trynumeric;
   1021  1.27    itojun 	if (IN6_IS_ADDR_MC_ORGLOCAL(a6))
   1022  1.27    itojun 		goto trynumeric;
   1023  1.27    itojun 	else
   1024  1.27    itojun 		goto trynumeric;	/* global */
   1025  1.27    itojun 
   1026  1.27    itojun 	/* try to convert to a numeric id as a last resort */
   1027  1.27    itojun   trynumeric:
   1028  1.59    itojun 	errno = 0;
   1029  1.59    itojun 	lscopeid = strtoul(scope, &ep, 10);
   1030  1.61    itojun 	*scopeid = (u_int32_t)(lscopeid & 0xffffffffUL);
   1031  1.60    itojun 	if (errno == 0 && ep && *ep == '\0' && *scopeid == lscopeid)
   1032  1.60    itojun 		return 0;
   1033  1.27    itojun 	else
   1034  1.27    itojun 		return -1;
   1035  1.27    itojun }
   1036  1.42    itojun #endif
   1037  1.32    itojun 
   1038  1.32    itojun /* code duplicate with gethnamaddr.c */
   1039  1.32    itojun 
   1040  1.32    itojun static const char AskedForGot[] =
   1041  1.32    itojun 	"gethostby*.getanswer: asked for \"%s\", got \"%s\"";
   1042  1.32    itojun 
   1043  1.32    itojun static struct addrinfo *
   1044  1.70  christos getanswer(const querybuf *answer, int anslen, const char *qname, int qtype,
   1045  1.70  christos     const struct addrinfo *pai)
   1046  1.32    itojun {
   1047  1.32    itojun 	struct addrinfo sentinel, *cur;
   1048  1.32    itojun 	struct addrinfo ai;
   1049  1.32    itojun 	const struct afd *afd;
   1050  1.32    itojun 	char *canonname;
   1051  1.32    itojun 	const HEADER *hp;
   1052  1.32    itojun 	const u_char *cp;
   1053  1.32    itojun 	int n;
   1054  1.32    itojun 	const u_char *eom;
   1055  1.56    itojun 	char *bp, *ep;
   1056  1.56    itojun 	int type, class, ancount, qdcount;
   1057  1.32    itojun 	int haveanswer, had_error;
   1058  1.32    itojun 	char tbuf[MAXDNAME];
   1059  1.70  christos 	int (*name_ok) (const char *);
   1060  1.32    itojun 	char hostbuf[8*1024];
   1061  1.32    itojun 
   1062  1.51     lukem 	_DIAGASSERT(answer != NULL);
   1063  1.51     lukem 	_DIAGASSERT(qname != NULL);
   1064  1.51     lukem 	_DIAGASSERT(pai != NULL);
   1065  1.51     lukem 
   1066  1.32    itojun 	memset(&sentinel, 0, sizeof(sentinel));
   1067  1.32    itojun 	cur = &sentinel;
   1068  1.32    itojun 
   1069  1.32    itojun 	canonname = NULL;
   1070  1.32    itojun 	eom = answer->buf + anslen;
   1071  1.32    itojun 	switch (qtype) {
   1072  1.32    itojun 	case T_A:
   1073  1.32    itojun 	case T_AAAA:
   1074  1.32    itojun 	case T_ANY:	/*use T_ANY only for T_A/T_AAAA lookup*/
   1075  1.32    itojun 		name_ok = res_hnok;
   1076  1.32    itojun 		break;
   1077  1.32    itojun 	default:
   1078  1.70  christos 		return NULL;	/* XXX should be abort(); */
   1079  1.32    itojun 	}
   1080  1.32    itojun 	/*
   1081  1.32    itojun 	 * find first satisfactory answer
   1082  1.32    itojun 	 */
   1083  1.32    itojun 	hp = &answer->hdr;
   1084  1.32    itojun 	ancount = ntohs(hp->ancount);
   1085  1.32    itojun 	qdcount = ntohs(hp->qdcount);
   1086  1.32    itojun 	bp = hostbuf;
   1087  1.56    itojun 	ep = hostbuf + sizeof hostbuf;
   1088  1.32    itojun 	cp = answer->buf + HFIXEDSZ;
   1089  1.32    itojun 	if (qdcount != 1) {
   1090  1.32    itojun 		h_errno = NO_RECOVERY;
   1091  1.32    itojun 		return (NULL);
   1092  1.32    itojun 	}
   1093  1.56    itojun 	n = dn_expand(answer->buf, eom, cp, bp, ep - bp);
   1094  1.32    itojun 	if ((n < 0) || !(*name_ok)(bp)) {
   1095  1.32    itojun 		h_errno = NO_RECOVERY;
   1096  1.32    itojun 		return (NULL);
   1097  1.32    itojun 	}
   1098  1.32    itojun 	cp += n + QFIXEDSZ;
   1099  1.32    itojun 	if (qtype == T_A || qtype == T_AAAA || qtype == T_ANY) {
   1100  1.32    itojun 		/* res_send() has already verified that the query name is the
   1101  1.32    itojun 		 * same as the one we sent; this just gets the expanded name
   1102  1.32    itojun 		 * (i.e., with the succeeding search-domain tacked on).
   1103  1.32    itojun 		 */
   1104  1.32    itojun 		n = strlen(bp) + 1;		/* for the \0 */
   1105  1.32    itojun 		if (n >= MAXHOSTNAMELEN) {
   1106  1.32    itojun 			h_errno = NO_RECOVERY;
   1107  1.32    itojun 			return (NULL);
   1108  1.32    itojun 		}
   1109  1.32    itojun 		canonname = bp;
   1110  1.32    itojun 		bp += n;
   1111  1.32    itojun 		/* The qname can be abbreviated, but h_name is now absolute. */
   1112  1.32    itojun 		qname = canonname;
   1113  1.32    itojun 	}
   1114  1.32    itojun 	haveanswer = 0;
   1115  1.32    itojun 	had_error = 0;
   1116  1.32    itojun 	while (ancount-- > 0 && cp < eom && !had_error) {
   1117  1.56    itojun 		n = dn_expand(answer->buf, eom, cp, bp, ep - bp);
   1118  1.32    itojun 		if ((n < 0) || !(*name_ok)(bp)) {
   1119  1.32    itojun 			had_error++;
   1120  1.32    itojun 			continue;
   1121  1.32    itojun 		}
   1122  1.32    itojun 		cp += n;			/* name */
   1123  1.32    itojun 		type = _getshort(cp);
   1124  1.32    itojun  		cp += INT16SZ;			/* type */
   1125  1.32    itojun 		class = _getshort(cp);
   1126  1.32    itojun  		cp += INT16SZ + INT32SZ;	/* class, TTL */
   1127  1.32    itojun 		n = _getshort(cp);
   1128  1.32    itojun 		cp += INT16SZ;			/* len */
   1129  1.32    itojun 		if (class != C_IN) {
   1130  1.32    itojun 			/* XXX - debug? syslog? */
   1131  1.32    itojun 			cp += n;
   1132  1.32    itojun 			continue;		/* XXX - had_error++ ? */
   1133  1.32    itojun 		}
   1134  1.32    itojun 		if ((qtype == T_A || qtype == T_AAAA || qtype == T_ANY) &&
   1135  1.32    itojun 		    type == T_CNAME) {
   1136  1.32    itojun 			n = dn_expand(answer->buf, eom, cp, tbuf, sizeof tbuf);
   1137  1.32    itojun 			if ((n < 0) || !(*name_ok)(tbuf)) {
   1138  1.32    itojun 				had_error++;
   1139  1.32    itojun 				continue;
   1140  1.32    itojun 			}
   1141  1.32    itojun 			cp += n;
   1142  1.32    itojun 			/* Get canonical name. */
   1143  1.32    itojun 			n = strlen(tbuf) + 1;	/* for the \0 */
   1144  1.56    itojun 			if (n > ep - bp || n >= MAXHOSTNAMELEN) {
   1145  1.32    itojun 				had_error++;
   1146  1.32    itojun 				continue;
   1147  1.32    itojun 			}
   1148  1.66    itojun 			strlcpy(bp, tbuf, (size_t)(ep - bp));
   1149  1.32    itojun 			canonname = bp;
   1150  1.32    itojun 			bp += n;
   1151  1.32    itojun 			continue;
   1152  1.32    itojun 		}
   1153  1.32    itojun 		if (qtype == T_ANY) {
   1154  1.32    itojun 			if (!(type == T_A || type == T_AAAA)) {
   1155  1.32    itojun 				cp += n;
   1156  1.32    itojun 				continue;
   1157  1.32    itojun 			}
   1158  1.32    itojun 		} else if (type != qtype) {
   1159  1.88  christos 			if (type != T_KEY && type != T_SIG) {
   1160  1.88  christos 				struct syslog_data sd = SYSLOG_DATA_INIT;
   1161  1.88  christos 				syslog_r(LOG_NOTICE|LOG_AUTH, &sd,
   1162  1.32    itojun 	       "gethostby*.getanswer: asked for \"%s %s %s\", got type \"%s\"",
   1163  1.32    itojun 				       qname, p_class(C_IN), p_type(qtype),
   1164  1.32    itojun 				       p_type(type));
   1165  1.88  christos 			}
   1166  1.32    itojun 			cp += n;
   1167  1.32    itojun 			continue;		/* XXX - had_error++ ? */
   1168  1.32    itojun 		}
   1169  1.32    itojun 		switch (type) {
   1170  1.32    itojun 		case T_A:
   1171  1.32    itojun 		case T_AAAA:
   1172  1.32    itojun 			if (strcasecmp(canonname, bp) != 0) {
   1173  1.88  christos 				struct syslog_data sd = SYSLOG_DATA_INIT;
   1174  1.88  christos 				syslog_r(LOG_NOTICE|LOG_AUTH, &sd,
   1175  1.32    itojun 				       AskedForGot, canonname, bp);
   1176  1.32    itojun 				cp += n;
   1177  1.32    itojun 				continue;	/* XXX - had_error++ ? */
   1178  1.32    itojun 			}
   1179  1.32    itojun 			if (type == T_A && n != INADDRSZ) {
   1180  1.32    itojun 				cp += n;
   1181  1.32    itojun 				continue;
   1182  1.32    itojun 			}
   1183  1.32    itojun 			if (type == T_AAAA && n != IN6ADDRSZ) {
   1184  1.32    itojun 				cp += n;
   1185  1.32    itojun 				continue;
   1186  1.62    itojun 			}
   1187  1.62    itojun 			if (type == T_AAAA) {
   1188  1.62    itojun 				struct in6_addr in6;
   1189  1.62    itojun 				memcpy(&in6, cp, IN6ADDRSZ);
   1190  1.62    itojun 				if (IN6_IS_ADDR_V4MAPPED(&in6)) {
   1191  1.62    itojun 					cp += n;
   1192  1.62    itojun 					continue;
   1193  1.62    itojun 				}
   1194  1.32    itojun 			}
   1195  1.32    itojun 			if (!haveanswer) {
   1196  1.32    itojun 				int nn;
   1197  1.32    itojun 
   1198  1.32    itojun 				canonname = bp;
   1199  1.32    itojun 				nn = strlen(bp) + 1;	/* for the \0 */
   1200  1.32    itojun 				bp += nn;
   1201  1.32    itojun 			}
   1202  1.32    itojun 
   1203  1.32    itojun 			/* don't overwrite pai */
   1204  1.32    itojun 			ai = *pai;
   1205  1.32    itojun 			ai.ai_family = (type == T_A) ? AF_INET : AF_INET6;
   1206  1.32    itojun 			afd = find_afd(ai.ai_family);
   1207  1.32    itojun 			if (afd == NULL) {
   1208  1.32    itojun 				cp += n;
   1209  1.32    itojun 				continue;
   1210  1.32    itojun 			}
   1211  1.36  christos 			cur->ai_next = get_ai(&ai, afd, (const char *)cp);
   1212  1.32    itojun 			if (cur->ai_next == NULL)
   1213  1.32    itojun 				had_error++;
   1214  1.32    itojun 			while (cur && cur->ai_next)
   1215  1.32    itojun 				cur = cur->ai_next;
   1216  1.32    itojun 			cp += n;
   1217  1.32    itojun 			break;
   1218  1.32    itojun 		default:
   1219  1.32    itojun 			abort();
   1220  1.32    itojun 		}
   1221  1.32    itojun 		if (!had_error)
   1222  1.32    itojun 			haveanswer++;
   1223  1.32    itojun 	}
   1224  1.32    itojun 	if (haveanswer) {
   1225  1.32    itojun 		if (!canonname)
   1226  1.32    itojun 			(void)get_canonname(pai, sentinel.ai_next, qname);
   1227  1.32    itojun 		else
   1228  1.32    itojun 			(void)get_canonname(pai, sentinel.ai_next, canonname);
   1229  1.32    itojun 		h_errno = NETDB_SUCCESS;
   1230  1.32    itojun 		return sentinel.ai_next;
   1231  1.32    itojun 	}
   1232  1.32    itojun 
   1233  1.32    itojun 	h_errno = NO_RECOVERY;
   1234  1.32    itojun 	return NULL;
   1235  1.32    itojun }
   1236  1.32    itojun 
   1237  1.73    tsarna #define SORTEDADDR(p)	(((struct sockaddr_in *)(void *)(p->ai_next->ai_addr))->sin_addr.s_addr)
   1238  1.73    tsarna #define SORTMATCH(p, s) ((SORTEDADDR(p) & (s).mask) == (s).addr.s_addr)
   1239  1.73    tsarna 
   1240  1.73    tsarna static void
   1241  1.73    tsarna aisort(struct addrinfo *s, res_state res)
   1242  1.73    tsarna {
   1243  1.73    tsarna 	struct addrinfo head, *t, *p;
   1244  1.73    tsarna 	int i;
   1245  1.73    tsarna 
   1246  1.73    tsarna 	head.ai_next = NULL;
   1247  1.73    tsarna 	t = &head;
   1248  1.73    tsarna 
   1249  1.73    tsarna 	for (i = 0; i < res->nsort; i++) {
   1250  1.73    tsarna 		p = s;
   1251  1.73    tsarna 		while (p->ai_next) {
   1252  1.73    tsarna 			if ((p->ai_next->ai_family != AF_INET)
   1253  1.73    tsarna 			|| SORTMATCH(p, res->sort_list[i])) {
   1254  1.73    tsarna 				t->ai_next = p->ai_next;
   1255  1.73    tsarna 				t = t->ai_next;
   1256  1.73    tsarna 				p->ai_next = p->ai_next->ai_next;
   1257  1.73    tsarna 			} else {
   1258  1.73    tsarna 				p = p->ai_next;
   1259  1.73    tsarna 			}
   1260  1.73    tsarna 		}
   1261  1.73    tsarna 	}
   1262  1.73    tsarna 
   1263  1.73    tsarna 	/* add rest of list and reset s to the new list*/
   1264  1.73    tsarna 	t->ai_next = s->ai_next;
   1265  1.73    tsarna 	s->ai_next = head.ai_next;
   1266  1.73    tsarna }
   1267  1.73    tsarna 
   1268  1.32    itojun /*ARGSUSED*/
   1269  1.32    itojun static int
   1270  1.70  christos _dns_getaddrinfo(void *rv, void	*cb_data, va_list ap)
   1271  1.32    itojun {
   1272  1.32    itojun 	struct addrinfo *ai;
   1273  1.63    itojun 	querybuf *buf, *buf2;
   1274  1.32    itojun 	const char *name;
   1275  1.32    itojun 	const struct addrinfo *pai;
   1276  1.32    itojun 	struct addrinfo sentinel, *cur;
   1277  1.32    itojun 	struct res_target q, q2;
   1278  1.73    tsarna 	res_state res;
   1279  1.32    itojun 
   1280  1.32    itojun 	name = va_arg(ap, char *);
   1281  1.32    itojun 	pai = va_arg(ap, const struct addrinfo *);
   1282  1.32    itojun 
   1283  1.78       mjl 	memset(&q, 0, sizeof(q));
   1284  1.32    itojun 	memset(&q2, 0, sizeof(q2));
   1285  1.32    itojun 	memset(&sentinel, 0, sizeof(sentinel));
   1286  1.32    itojun 	cur = &sentinel;
   1287  1.32    itojun 
   1288  1.63    itojun 	buf = malloc(sizeof(*buf));
   1289  1.63    itojun 	if (buf == NULL) {
   1290  1.63    itojun 		h_errno = NETDB_INTERNAL;
   1291  1.63    itojun 		return NS_NOTFOUND;
   1292  1.63    itojun 	}
   1293  1.63    itojun 	buf2 = malloc(sizeof(*buf2));
   1294  1.63    itojun 	if (buf2 == NULL) {
   1295  1.63    itojun 		free(buf);
   1296  1.63    itojun 		h_errno = NETDB_INTERNAL;
   1297  1.63    itojun 		return NS_NOTFOUND;
   1298  1.63    itojun 	}
   1299  1.63    itojun 
   1300  1.32    itojun 	switch (pai->ai_family) {
   1301  1.32    itojun 	case AF_UNSPEC:
   1302  1.32    itojun 		/* prefer IPv6 */
   1303  1.52    itojun 		q.name = name;
   1304  1.43    itojun 		q.qclass = C_IN;
   1305  1.43    itojun 		q.qtype = T_AAAA;
   1306  1.63    itojun 		q.answer = buf->buf;
   1307  1.63    itojun 		q.anslen = sizeof(buf->buf);
   1308  1.32    itojun 		q.next = &q2;
   1309  1.52    itojun 		q2.name = name;
   1310  1.43    itojun 		q2.qclass = C_IN;
   1311  1.43    itojun 		q2.qtype = T_A;
   1312  1.63    itojun 		q2.answer = buf2->buf;
   1313  1.63    itojun 		q2.anslen = sizeof(buf2->buf);
   1314  1.32    itojun 		break;
   1315  1.32    itojun 	case AF_INET:
   1316  1.52    itojun 		q.name = name;
   1317  1.43    itojun 		q.qclass = C_IN;
   1318  1.43    itojun 		q.qtype = T_A;
   1319  1.63    itojun 		q.answer = buf->buf;
   1320  1.63    itojun 		q.anslen = sizeof(buf->buf);
   1321  1.32    itojun 		break;
   1322  1.32    itojun 	case AF_INET6:
   1323  1.52    itojun 		q.name = name;
   1324  1.43    itojun 		q.qclass = C_IN;
   1325  1.43    itojun 		q.qtype = T_AAAA;
   1326  1.63    itojun 		q.answer = buf->buf;
   1327  1.63    itojun 		q.anslen = sizeof(buf->buf);
   1328  1.32    itojun 		break;
   1329  1.32    itojun 	default:
   1330  1.63    itojun 		free(buf);
   1331  1.63    itojun 		free(buf2);
   1332  1.32    itojun 		return NS_UNAVAIL;
   1333  1.32    itojun 	}
   1334  1.73    tsarna 
   1335  1.73    tsarna 	res = __res_get_state();
   1336  1.73    tsarna 	if (res == NULL) {
   1337  1.73    tsarna 		free(buf);
   1338  1.73    tsarna 		free(buf2);
   1339  1.73    tsarna 		return NS_NOTFOUND;
   1340  1.73    tsarna 	}
   1341  1.73    tsarna 
   1342  1.73    tsarna 	if (res_searchN(name, &q, res) < 0) {
   1343  1.73    tsarna 		__res_put_state(res);
   1344  1.63    itojun 		free(buf);
   1345  1.63    itojun 		free(buf2);
   1346  1.32    itojun 		return NS_NOTFOUND;
   1347  1.63    itojun 	}
   1348  1.63    itojun 	ai = getanswer(buf, q.n, q.name, q.qtype, pai);
   1349  1.32    itojun 	if (ai) {
   1350  1.32    itojun 		cur->ai_next = ai;
   1351  1.32    itojun 		while (cur && cur->ai_next)
   1352  1.32    itojun 			cur = cur->ai_next;
   1353  1.32    itojun 	}
   1354  1.32    itojun 	if (q.next) {
   1355  1.63    itojun 		ai = getanswer(buf2, q2.n, q2.name, q2.qtype, pai);
   1356  1.32    itojun 		if (ai)
   1357  1.32    itojun 			cur->ai_next = ai;
   1358  1.32    itojun 	}
   1359  1.63    itojun 	free(buf);
   1360  1.63    itojun 	free(buf2);
   1361  1.73    tsarna 	if (sentinel.ai_next == NULL) {
   1362  1.73    tsarna 		__res_put_state(res);
   1363  1.32    itojun 		switch (h_errno) {
   1364  1.32    itojun 		case HOST_NOT_FOUND:
   1365  1.32    itojun 			return NS_NOTFOUND;
   1366  1.32    itojun 		case TRY_AGAIN:
   1367  1.32    itojun 			return NS_TRYAGAIN;
   1368  1.32    itojun 		default:
   1369  1.32    itojun 			return NS_UNAVAIL;
   1370  1.32    itojun 		}
   1371  1.73    tsarna 	}
   1372  1.73    tsarna 
   1373  1.73    tsarna 	if (res->nsort)
   1374  1.73    tsarna 		aisort(&sentinel, res);
   1375  1.73    tsarna 
   1376  1.73    tsarna 	__res_put_state(res);
   1377  1.73    tsarna 
   1378  1.32    itojun 	*((struct addrinfo **)rv) = sentinel.ai_next;
   1379  1.32    itojun 	return NS_SUCCESS;
   1380  1.32    itojun }
   1381  1.32    itojun 
   1382  1.32    itojun static void
   1383  1.70  christos _sethtent(FILE **hostf)
   1384  1.32    itojun {
   1385  1.51     lukem 
   1386  1.70  christos 	if (!*hostf)
   1387  1.70  christos 		*hostf = fopen(_PATH_HOSTS, "r" );
   1388  1.32    itojun 	else
   1389  1.70  christos 		rewind(*hostf);
   1390  1.32    itojun }
   1391  1.32    itojun 
   1392  1.32    itojun static void
   1393  1.70  christos _endhtent(FILE **hostf)
   1394  1.32    itojun {
   1395  1.51     lukem 
   1396  1.70  christos 	if (*hostf) {
   1397  1.70  christos 		(void) fclose(*hostf);
   1398  1.70  christos 		*hostf = NULL;
   1399  1.32    itojun 	}
   1400  1.32    itojun }
   1401  1.32    itojun 
   1402  1.32    itojun static struct addrinfo *
   1403  1.70  christos _gethtent(FILE **hostf, const char *name, const struct addrinfo *pai)
   1404  1.32    itojun {
   1405  1.32    itojun 	char *p;
   1406  1.48    itojun 	char *cp, *tname, *cname;
   1407  1.32    itojun 	struct addrinfo hints, *res0, *res;
   1408  1.32    itojun 	int error;
   1409  1.32    itojun 	const char *addr;
   1410  1.32    itojun 	char hostbuf[8*1024];
   1411  1.32    itojun 
   1412  1.51     lukem 	_DIAGASSERT(name != NULL);
   1413  1.51     lukem 	_DIAGASSERT(pai != NULL);
   1414  1.51     lukem 
   1415  1.70  christos 	if (!*hostf && !(*hostf = fopen(_PATH_HOSTS, "r" )))
   1416  1.32    itojun 		return (NULL);
   1417  1.32    itojun  again:
   1418  1.70  christos 	if (!(p = fgets(hostbuf, sizeof hostbuf, *hostf)))
   1419  1.32    itojun 		return (NULL);
   1420  1.32    itojun 	if (*p == '#')
   1421  1.32    itojun 		goto again;
   1422  1.32    itojun 	if (!(cp = strpbrk(p, "#\n")))
   1423  1.32    itojun 		goto again;
   1424  1.32    itojun 	*cp = '\0';
   1425  1.32    itojun 	if (!(cp = strpbrk(p, " \t")))
   1426  1.32    itojun 		goto again;
   1427  1.32    itojun 	*cp++ = '\0';
   1428  1.32    itojun 	addr = p;
   1429  1.32    itojun 	/* if this is not something we're looking for, skip it. */
   1430  1.48    itojun 	cname = NULL;
   1431  1.32    itojun 	while (cp && *cp) {
   1432  1.32    itojun 		if (*cp == ' ' || *cp == '\t') {
   1433  1.32    itojun 			cp++;
   1434  1.32    itojun 			continue;
   1435  1.32    itojun 		}
   1436  1.48    itojun 		if (!cname)
   1437  1.48    itojun 			cname = cp;
   1438  1.32    itojun 		tname = cp;
   1439  1.32    itojun 		if ((cp = strpbrk(cp, " \t")) != NULL)
   1440  1.32    itojun 			*cp++ = '\0';
   1441  1.32    itojun 		if (strcasecmp(name, tname) == 0)
   1442  1.32    itojun 			goto found;
   1443  1.32    itojun 	}
   1444  1.32    itojun 	goto again;
   1445  1.32    itojun 
   1446  1.32    itojun found:
   1447  1.32    itojun 	hints = *pai;
   1448  1.32    itojun 	hints.ai_flags = AI_NUMERICHOST;
   1449  1.32    itojun 	error = getaddrinfo(addr, NULL, &hints, &res0);
   1450  1.32    itojun 	if (error)
   1451  1.32    itojun 		goto again;
   1452  1.32    itojun 	for (res = res0; res; res = res->ai_next) {
   1453  1.32    itojun 		/* cover it up */
   1454  1.32    itojun 		res->ai_flags = pai->ai_flags;
   1455  1.32    itojun 
   1456  1.32    itojun 		if (pai->ai_flags & AI_CANONNAME) {
   1457  1.48    itojun 			if (get_canonname(pai, res, cname) != 0) {
   1458  1.32    itojun 				freeaddrinfo(res0);
   1459  1.32    itojun 				goto again;
   1460  1.32    itojun 			}
   1461  1.32    itojun 		}
   1462  1.32    itojun 	}
   1463  1.32    itojun 	return res0;
   1464  1.32    itojun }
   1465  1.32    itojun 
   1466  1.32    itojun /*ARGSUSED*/
   1467  1.32    itojun static int
   1468  1.70  christos _files_getaddrinfo(void *rv, void *cb_data, va_list ap)
   1469  1.32    itojun {
   1470  1.32    itojun 	const char *name;
   1471  1.32    itojun 	const struct addrinfo *pai;
   1472  1.32    itojun 	struct addrinfo sentinel, *cur;
   1473  1.32    itojun 	struct addrinfo *p;
   1474  1.70  christos #ifndef _REENTRANT
   1475  1.70  christos 	static
   1476  1.70  christos #endif
   1477  1.70  christos 	FILE *hostf = NULL;
   1478  1.32    itojun 
   1479  1.32    itojun 	name = va_arg(ap, char *);
   1480  1.32    itojun 	pai = va_arg(ap, struct addrinfo *);
   1481  1.32    itojun 
   1482  1.32    itojun 	memset(&sentinel, 0, sizeof(sentinel));
   1483  1.32    itojun 	cur = &sentinel;
   1484  1.32    itojun 
   1485  1.70  christos 	_sethtent(&hostf);
   1486  1.70  christos 	while ((p = _gethtent(&hostf, name, pai)) != NULL) {
   1487  1.32    itojun 		cur->ai_next = p;
   1488  1.32    itojun 		while (cur && cur->ai_next)
   1489  1.32    itojun 			cur = cur->ai_next;
   1490  1.32    itojun 	}
   1491  1.70  christos 	_endhtent(&hostf);
   1492  1.32    itojun 
   1493  1.32    itojun 	*((struct addrinfo **)rv) = sentinel.ai_next;
   1494  1.32    itojun 	if (sentinel.ai_next == NULL)
   1495  1.32    itojun 		return NS_NOTFOUND;
   1496  1.32    itojun 	return NS_SUCCESS;
   1497  1.32    itojun }
   1498  1.32    itojun 
   1499  1.32    itojun #ifdef YP
   1500  1.32    itojun /*ARGSUSED*/
   1501  1.32    itojun static struct addrinfo *
   1502  1.70  christos _yphostent(char *line, const struct addrinfo *pai)
   1503  1.32    itojun {
   1504  1.32    itojun 	struct addrinfo sentinel, *cur;
   1505  1.32    itojun 	struct addrinfo hints, *res, *res0;
   1506  1.32    itojun 	int error;
   1507  1.51     lukem 	char *p;
   1508  1.32    itojun 	const char *addr, *canonname;
   1509  1.32    itojun 	char *nextline;
   1510  1.32    itojun 	char *cp;
   1511  1.32    itojun 
   1512  1.51     lukem 	_DIAGASSERT(line != NULL);
   1513  1.51     lukem 	_DIAGASSERT(pai != NULL);
   1514  1.51     lukem 
   1515  1.51     lukem 	p = line;
   1516  1.32    itojun 	addr = canonname = NULL;
   1517  1.32    itojun 
   1518  1.36  christos 	memset(&sentinel, 0, sizeof(sentinel));
   1519  1.36  christos 	cur = &sentinel;
   1520  1.36  christos 
   1521  1.32    itojun nextline:
   1522  1.32    itojun 	/* terminate line */
   1523  1.32    itojun 	cp = strchr(p, '\n');
   1524  1.32    itojun 	if (cp) {
   1525  1.32    itojun 		*cp++ = '\0';
   1526  1.32    itojun 		nextline = cp;
   1527  1.32    itojun 	} else
   1528  1.32    itojun 		nextline = NULL;
   1529  1.32    itojun 
   1530  1.32    itojun 	cp = strpbrk(p, " \t");
   1531  1.32    itojun 	if (cp == NULL) {
   1532  1.32    itojun 		if (canonname == NULL)
   1533  1.32    itojun 			return (NULL);
   1534  1.32    itojun 		else
   1535  1.32    itojun 			goto done;
   1536  1.32    itojun 	}
   1537  1.32    itojun 	*cp++ = '\0';
   1538  1.32    itojun 
   1539  1.32    itojun 	addr = p;
   1540  1.32    itojun 
   1541  1.32    itojun 	while (cp && *cp) {
   1542  1.32    itojun 		if (*cp == ' ' || *cp == '\t') {
   1543  1.32    itojun 			cp++;
   1544  1.32    itojun 			continue;
   1545  1.32    itojun 		}
   1546  1.32    itojun 		if (!canonname)
   1547  1.32    itojun 			canonname = cp;
   1548  1.32    itojun 		if ((cp = strpbrk(cp, " \t")) != NULL)
   1549  1.32    itojun 			*cp++ = '\0';
   1550  1.32    itojun 	}
   1551  1.32    itojun 
   1552  1.32    itojun 	hints = *pai;
   1553  1.32    itojun 	hints.ai_flags = AI_NUMERICHOST;
   1554  1.32    itojun 	error = getaddrinfo(addr, NULL, &hints, &res0);
   1555  1.32    itojun 	if (error == 0) {
   1556  1.32    itojun 		for (res = res0; res; res = res->ai_next) {
   1557  1.32    itojun 			/* cover it up */
   1558  1.32    itojun 			res->ai_flags = pai->ai_flags;
   1559  1.32    itojun 
   1560  1.32    itojun 			if (pai->ai_flags & AI_CANONNAME)
   1561  1.32    itojun 				(void)get_canonname(pai, res, canonname);
   1562  1.32    itojun 		}
   1563  1.37    itojun 	} else
   1564  1.37    itojun 		res0 = NULL;
   1565  1.32    itojun 	if (res0) {
   1566  1.32    itojun 		cur->ai_next = res0;
   1567  1.80  christos 		while (cur->ai_next)
   1568  1.32    itojun 			cur = cur->ai_next;
   1569  1.32    itojun 	}
   1570  1.32    itojun 
   1571  1.32    itojun 	if (nextline) {
   1572  1.32    itojun 		p = nextline;
   1573  1.32    itojun 		goto nextline;
   1574  1.32    itojun 	}
   1575  1.32    itojun 
   1576  1.32    itojun done:
   1577  1.32    itojun 	return sentinel.ai_next;
   1578  1.32    itojun }
   1579  1.32    itojun 
   1580  1.32    itojun /*ARGSUSED*/
   1581  1.32    itojun static int
   1582  1.70  christos _yp_getaddrinfo(void *rv, void *cb_data, va_list ap)
   1583  1.32    itojun {
   1584  1.32    itojun 	struct addrinfo sentinel, *cur;
   1585  1.32    itojun 	struct addrinfo *ai = NULL;
   1586  1.72  christos 	char *ypbuf;
   1587  1.72  christos 	int ypbuflen, r;
   1588  1.32    itojun 	const char *name;
   1589  1.32    itojun 	const struct addrinfo *pai;
   1590  1.72  christos 	char *ypdomain;
   1591  1.72  christos 
   1592  1.72  christos 	if (_yp_check(&ypdomain) == 0)
   1593  1.72  christos 		return NS_UNAVAIL;
   1594  1.32    itojun 
   1595  1.32    itojun 	name = va_arg(ap, char *);
   1596  1.32    itojun 	pai = va_arg(ap, const struct addrinfo *);
   1597  1.32    itojun 
   1598  1.32    itojun 	memset(&sentinel, 0, sizeof(sentinel));
   1599  1.32    itojun 	cur = &sentinel;
   1600  1.32    itojun 
   1601  1.32    itojun 	/* hosts.byname is only for IPv4 (Solaris8) */
   1602  1.33    itojun 	if (pai->ai_family == PF_UNSPEC || pai->ai_family == PF_INET) {
   1603  1.72  christos 		r = yp_match(ypdomain, "hosts.byname", name,
   1604  1.72  christos 			(int)strlen(name), &ypbuf, &ypbuflen);
   1605  1.33    itojun 		if (r == 0) {
   1606  1.33    itojun 			struct addrinfo ai4;
   1607  1.33    itojun 
   1608  1.33    itojun 			ai4 = *pai;
   1609  1.33    itojun 			ai4.ai_family = AF_INET;
   1610  1.72  christos 			ai = _yphostent(ypbuf, &ai4);
   1611  1.33    itojun 			if (ai) {
   1612  1.33    itojun 				cur->ai_next = ai;
   1613  1.33    itojun 				while (cur && cur->ai_next)
   1614  1.33    itojun 					cur = cur->ai_next;
   1615  1.33    itojun 			}
   1616  1.32    itojun 		}
   1617  1.72  christos 		free(ypbuf);
   1618  1.32    itojun 	}
   1619  1.32    itojun 
   1620  1.32    itojun 	/* ipnodes.byname can hold both IPv4/v6 */
   1621  1.72  christos 	r = yp_match(ypdomain, "ipnodes.byname", name,
   1622  1.72  christos 		(int)strlen(name), &ypbuf, &ypbuflen);
   1623  1.32    itojun 	if (r == 0) {
   1624  1.72  christos 		ai = _yphostent(ypbuf, pai);
   1625  1.72  christos 		if (ai)
   1626  1.32    itojun 			cur->ai_next = ai;
   1627  1.72  christos 		free(ypbuf);
   1628  1.32    itojun 	}
   1629  1.32    itojun 
   1630  1.32    itojun 	if (sentinel.ai_next == NULL) {
   1631  1.32    itojun 		h_errno = HOST_NOT_FOUND;
   1632  1.32    itojun 		return NS_NOTFOUND;
   1633  1.32    itojun 	}
   1634  1.32    itojun 	*((struct addrinfo **)rv) = sentinel.ai_next;
   1635  1.32    itojun 	return NS_SUCCESS;
   1636  1.32    itojun }
   1637  1.32    itojun #endif
   1638  1.32    itojun 
   1639  1.32    itojun /* resolver logic */
   1640  1.32    itojun 
   1641  1.32    itojun /*
   1642  1.32    itojun  * Formulate a normal query, send, and await answer.
   1643  1.32    itojun  * Returned answer is placed in supplied buffer "answer".
   1644  1.32    itojun  * Perform preliminary check of answer, returning success only
   1645  1.32    itojun  * if no error is indicated and the answer count is nonzero.
   1646  1.32    itojun  * Return the size of the response on success, -1 on error.
   1647  1.32    itojun  * Error number is left in h_errno.
   1648  1.32    itojun  *
   1649  1.32    itojun  * Caller must parse answer and determine whether it answers the question.
   1650  1.32    itojun  */
   1651  1.32    itojun static int
   1652  1.70  christos res_queryN(const char *name, /* domain name */ struct res_target *target,
   1653  1.70  christos     res_state res)
   1654  1.32    itojun {
   1655  1.32    itojun 	u_char buf[MAXPACKET];
   1656  1.32    itojun 	HEADER *hp;
   1657  1.32    itojun 	int n;
   1658  1.32    itojun 	struct res_target *t;
   1659  1.32    itojun 	int rcode;
   1660  1.32    itojun 	int ancount;
   1661  1.32    itojun 
   1662  1.51     lukem 	_DIAGASSERT(name != NULL);
   1663  1.51     lukem 	/* XXX: target may be NULL??? */
   1664  1.51     lukem 
   1665  1.32    itojun 	rcode = NOERROR;
   1666  1.32    itojun 	ancount = 0;
   1667  1.32    itojun 
   1668  1.32    itojun 	for (t = target; t; t = t->next) {
   1669  1.32    itojun 		int class, type;
   1670  1.32    itojun 		u_char *answer;
   1671  1.32    itojun 		int anslen;
   1672  1.32    itojun 
   1673  1.32    itojun 		hp = (HEADER *)(void *)t->answer;
   1674  1.32    itojun 		hp->rcode = NOERROR;	/* default */
   1675  1.32    itojun 
   1676  1.32    itojun 		/* make it easier... */
   1677  1.43    itojun 		class = t->qclass;
   1678  1.43    itojun 		type = t->qtype;
   1679  1.32    itojun 		answer = t->answer;
   1680  1.32    itojun 		anslen = t->anslen;
   1681  1.32    itojun #ifdef DEBUG
   1682  1.70  christos 		if (res->options & RES_DEBUG)
   1683  1.70  christos 			printf(";; res_nquery(%s, %d, %d)\n", name, class, type);
   1684  1.32    itojun #endif
   1685  1.32    itojun 
   1686  1.70  christos 		n = res_nmkquery(res, QUERY, name, class, type, NULL, 0, NULL,
   1687  1.32    itojun 		    buf, sizeof(buf));
   1688  1.47    itojun #ifdef RES_USE_EDNS0
   1689  1.70  christos 		if (n > 0 && (res->options & RES_USE_EDNS0) != 0)
   1690  1.70  christos 			n = res_nopt(res, n, buf, sizeof(buf), anslen);
   1691  1.47    itojun #endif
   1692  1.32    itojun 		if (n <= 0) {
   1693  1.32    itojun #ifdef DEBUG
   1694  1.70  christos 			if (res->options & RES_DEBUG)
   1695  1.70  christos 				printf(";; res_nquery: mkquery failed\n");
   1696  1.32    itojun #endif
   1697  1.32    itojun 			h_errno = NO_RECOVERY;
   1698  1.70  christos 			return n;
   1699  1.32    itojun 		}
   1700  1.70  christos 		n = res_nsend(res, buf, n, answer, anslen);
   1701  1.35    itojun #if 0
   1702  1.32    itojun 		if (n < 0) {
   1703  1.32    itojun #ifdef DEBUG
   1704  1.70  christos 			if (res->options & RES_DEBUG)
   1705  1.32    itojun 				printf(";; res_query: send error\n");
   1706  1.32    itojun #endif
   1707  1.32    itojun 			h_errno = TRY_AGAIN;
   1708  1.70  christos 			return n;
   1709  1.32    itojun 		}
   1710  1.35    itojun #endif
   1711  1.32    itojun 
   1712  1.35    itojun 		if (n < 0 || hp->rcode != NOERROR || ntohs(hp->ancount) == 0) {
   1713  1.32    itojun 			rcode = hp->rcode;	/* record most recent error */
   1714  1.32    itojun #ifdef DEBUG
   1715  1.70  christos 			if (res->options & RES_DEBUG)
   1716  1.57    itojun 				printf(";; rcode = %u, ancount=%u\n", hp->rcode,
   1717  1.32    itojun 				    ntohs(hp->ancount));
   1718  1.32    itojun #endif
   1719  1.32    itojun 			continue;
   1720  1.32    itojun 		}
   1721  1.32    itojun 
   1722  1.32    itojun 		ancount += ntohs(hp->ancount);
   1723  1.32    itojun 
   1724  1.32    itojun 		t->n = n;
   1725  1.32    itojun 	}
   1726  1.32    itojun 
   1727  1.32    itojun 	if (ancount == 0) {
   1728  1.32    itojun 		switch (rcode) {
   1729  1.32    itojun 		case NXDOMAIN:
   1730  1.32    itojun 			h_errno = HOST_NOT_FOUND;
   1731  1.32    itojun 			break;
   1732  1.32    itojun 		case SERVFAIL:
   1733  1.32    itojun 			h_errno = TRY_AGAIN;
   1734  1.32    itojun 			break;
   1735  1.32    itojun 		case NOERROR:
   1736  1.32    itojun 			h_errno = NO_DATA;
   1737  1.32    itojun 			break;
   1738  1.32    itojun 		case FORMERR:
   1739  1.32    itojun 		case NOTIMP:
   1740  1.32    itojun 		case REFUSED:
   1741  1.32    itojun 		default:
   1742  1.32    itojun 			h_errno = NO_RECOVERY;
   1743  1.32    itojun 			break;
   1744  1.32    itojun 		}
   1745  1.70  christos 		return -1;
   1746  1.32    itojun 	}
   1747  1.70  christos 	return ancount;
   1748  1.32    itojun }
   1749  1.32    itojun 
   1750  1.32    itojun /*
   1751  1.32    itojun  * Formulate a normal query, send, and retrieve answer in supplied buffer.
   1752  1.32    itojun  * Return the size of the response on success, -1 on error.
   1753  1.32    itojun  * If enabled, implement search rules until answer or unrecoverable failure
   1754  1.32    itojun  * is detected.  Error code, if any, is left in h_errno.
   1755  1.32    itojun  */
   1756  1.32    itojun static int
   1757  1.73    tsarna res_searchN(const char *name, struct res_target *target, res_state res)
   1758  1.32    itojun {
   1759  1.32    itojun 	const char *cp, * const *domain;
   1760  1.51     lukem 	HEADER *hp;
   1761  1.32    itojun 	u_int dots;
   1762  1.32    itojun 	int trailing_dot, ret, saved_herrno;
   1763  1.32    itojun 	int got_nodata = 0, got_servfail = 0, tried_as_is = 0;
   1764  1.71  christos 
   1765  1.51     lukem 	_DIAGASSERT(name != NULL);
   1766  1.51     lukem 	_DIAGASSERT(target != NULL);
   1767  1.51     lukem 
   1768  1.51     lukem 	hp = (HEADER *)(void *)target->answer;	/*XXX*/
   1769  1.51     lukem 
   1770  1.32    itojun 	errno = 0;
   1771  1.32    itojun 	h_errno = HOST_NOT_FOUND;	/* default, if we never query */
   1772  1.32    itojun 	dots = 0;
   1773  1.32    itojun 	for (cp = name; *cp; cp++)
   1774  1.32    itojun 		dots += (*cp == '.');
   1775  1.32    itojun 	trailing_dot = 0;
   1776  1.32    itojun 	if (cp > name && *--cp == '.')
   1777  1.32    itojun 		trailing_dot++;
   1778  1.32    itojun 
   1779  1.32    itojun 	/*
   1780  1.32    itojun 	 * if there aren't any dots, it could be a user-level alias
   1781  1.32    itojun 	 */
   1782  1.70  christos 	if (!dots && (cp = __hostalias(name)) != NULL) {
   1783  1.70  christos 		ret = res_queryN(cp, target, res);
   1784  1.70  christos 		return ret;
   1785  1.70  christos 	}
   1786  1.32    itojun 
   1787  1.32    itojun 	/*
   1788  1.32    itojun 	 * If there are dots in the name already, let's just give it a try
   1789  1.32    itojun 	 * 'as is'.  The threshold can be set with the "ndots" option.
   1790  1.32    itojun 	 */
   1791  1.32    itojun 	saved_herrno = -1;
   1792  1.70  christos 	if (dots >= res->ndots) {
   1793  1.70  christos 		ret = res_querydomainN(name, NULL, target, res);
   1794  1.73    tsarna 		if (ret > 0)
   1795  1.32    itojun 			return (ret);
   1796  1.32    itojun 		saved_herrno = h_errno;
   1797  1.32    itojun 		tried_as_is++;
   1798  1.32    itojun 	}
   1799  1.32    itojun 
   1800  1.32    itojun 	/*
   1801  1.32    itojun 	 * We do at least one level of search if
   1802  1.32    itojun 	 *	- there is no dot and RES_DEFNAME is set, or
   1803  1.32    itojun 	 *	- there is at least one dot, there is no trailing dot,
   1804  1.32    itojun 	 *	  and RES_DNSRCH is set.
   1805  1.32    itojun 	 */
   1806  1.70  christos 	if ((!dots && (res->options & RES_DEFNAMES)) ||
   1807  1.70  christos 	    (dots && !trailing_dot && (res->options & RES_DNSRCH))) {
   1808  1.32    itojun 		int done = 0;
   1809  1.32    itojun 
   1810  1.70  christos 		for (domain = (const char * const *)res->dnsrch;
   1811  1.32    itojun 		   *domain && !done;
   1812  1.32    itojun 		   domain++) {
   1813  1.32    itojun 
   1814  1.70  christos 			ret = res_querydomainN(name, *domain, target, res);
   1815  1.73    tsarna 			if (ret > 0)
   1816  1.70  christos 				return ret;
   1817  1.32    itojun 
   1818  1.32    itojun 			/*
   1819  1.32    itojun 			 * If no server present, give up.
   1820  1.32    itojun 			 * If name isn't found in this domain,
   1821  1.32    itojun 			 * keep trying higher domains in the search list
   1822  1.32    itojun 			 * (if that's enabled).
   1823  1.32    itojun 			 * On a NO_DATA error, keep trying, otherwise
   1824  1.32    itojun 			 * a wildcard entry of another type could keep us
   1825  1.32    itojun 			 * from finding this entry higher in the domain.
   1826  1.32    itojun 			 * If we get some other error (negative answer or
   1827  1.32    itojun 			 * server failure), then stop searching up,
   1828  1.32    itojun 			 * but try the input name below in case it's
   1829  1.32    itojun 			 * fully-qualified.
   1830  1.32    itojun 			 */
   1831  1.32    itojun 			if (errno == ECONNREFUSED) {
   1832  1.32    itojun 				h_errno = TRY_AGAIN;
   1833  1.70  christos 				return -1;
   1834  1.32    itojun 			}
   1835  1.32    itojun 
   1836  1.32    itojun 			switch (h_errno) {
   1837  1.32    itojun 			case NO_DATA:
   1838  1.32    itojun 				got_nodata++;
   1839  1.32    itojun 				/* FALLTHROUGH */
   1840  1.32    itojun 			case HOST_NOT_FOUND:
   1841  1.32    itojun 				/* keep trying */
   1842  1.32    itojun 				break;
   1843  1.32    itojun 			case TRY_AGAIN:
   1844  1.32    itojun 				if (hp->rcode == SERVFAIL) {
   1845  1.32    itojun 					/* try next search element, if any */
   1846  1.32    itojun 					got_servfail++;
   1847  1.32    itojun 					break;
   1848  1.32    itojun 				}
   1849  1.32    itojun 				/* FALLTHROUGH */
   1850  1.32    itojun 			default:
   1851  1.32    itojun 				/* anything else implies that we're done */
   1852  1.32    itojun 				done++;
   1853  1.32    itojun 			}
   1854  1.32    itojun 			/*
   1855  1.32    itojun 			 * if we got here for some reason other than DNSRCH,
   1856  1.32    itojun 			 * we only wanted one iteration of the loop, so stop.
   1857  1.32    itojun 			 */
   1858  1.70  christos 			if (!(res->options & RES_DNSRCH))
   1859  1.32    itojun 			        done++;
   1860  1.32    itojun 		}
   1861  1.32    itojun 	}
   1862  1.32    itojun 
   1863  1.32    itojun 	/*
   1864  1.32    itojun 	 * if we have not already tried the name "as is", do that now.
   1865  1.32    itojun 	 * note that we do this regardless of how many dots were in the
   1866  1.32    itojun 	 * name or whether it ends with a dot.
   1867  1.32    itojun 	 */
   1868  1.32    itojun 	if (!tried_as_is) {
   1869  1.70  christos 		ret = res_querydomainN(name, NULL, target, res);
   1870  1.73    tsarna 		if (ret > 0)
   1871  1.70  christos 			return ret;
   1872  1.32    itojun 	}
   1873  1.32    itojun 
   1874  1.32    itojun 	/*
   1875  1.32    itojun 	 * if we got here, we didn't satisfy the search.
   1876  1.32    itojun 	 * if we did an initial full query, return that query's h_errno
   1877  1.32    itojun 	 * (note that we wouldn't be here if that query had succeeded).
   1878  1.32    itojun 	 * else if we ever got a nodata, send that back as the reason.
   1879  1.32    itojun 	 * else send back meaningless h_errno, that being the one from
   1880  1.32    itojun 	 * the last DNSRCH we did.
   1881  1.32    itojun 	 */
   1882  1.32    itojun 	if (saved_herrno != -1)
   1883  1.32    itojun 		h_errno = saved_herrno;
   1884  1.32    itojun 	else if (got_nodata)
   1885  1.32    itojun 		h_errno = NO_DATA;
   1886  1.32    itojun 	else if (got_servfail)
   1887  1.32    itojun 		h_errno = TRY_AGAIN;
   1888  1.70  christos 	return -1;
   1889  1.32    itojun }
   1890  1.32    itojun 
   1891  1.32    itojun /*
   1892  1.32    itojun  * Perform a call on res_query on the concatenation of name and domain,
   1893  1.32    itojun  * removing a trailing dot from name if domain is NULL.
   1894  1.32    itojun  */
   1895  1.32    itojun static int
   1896  1.73    tsarna res_querydomainN(const char *name, const char *domain,
   1897  1.70  christos     struct res_target *target, res_state res)
   1898  1.32    itojun {
   1899  1.32    itojun 	char nbuf[MAXDNAME];
   1900  1.32    itojun 	const char *longname = nbuf;
   1901  1.32    itojun 	size_t n, d;
   1902  1.51     lukem 
   1903  1.51     lukem 	_DIAGASSERT(name != NULL);
   1904  1.51     lukem 	/* XXX: target may be NULL??? */
   1905  1.32    itojun 
   1906  1.32    itojun #ifdef DEBUG
   1907  1.70  christos 	if (res->options & RES_DEBUG)
   1908  1.32    itojun 		printf(";; res_querydomain(%s, %s)\n",
   1909  1.32    itojun 			name, domain?domain:"<Nil>");
   1910  1.32    itojun #endif
   1911  1.32    itojun 	if (domain == NULL) {
   1912  1.32    itojun 		/*
   1913  1.32    itojun 		 * Check for trailing '.';
   1914  1.32    itojun 		 * copy without '.' if present.
   1915  1.32    itojun 		 */
   1916  1.32    itojun 		n = strlen(name);
   1917  1.63    itojun 		if (n + 1 > sizeof(nbuf)) {
   1918  1.32    itojun 			h_errno = NO_RECOVERY;
   1919  1.70  christos 			return -1;
   1920  1.32    itojun 		}
   1921  1.40    itojun 		if (n > 0 && name[--n] == '.') {
   1922  1.32    itojun 			strncpy(nbuf, name, n);
   1923  1.32    itojun 			nbuf[n] = '\0';
   1924  1.32    itojun 		} else
   1925  1.32    itojun 			longname = name;
   1926  1.32    itojun 	} else {
   1927  1.32    itojun 		n = strlen(name);
   1928  1.32    itojun 		d = strlen(domain);
   1929  1.63    itojun 		if (n + 1 + d + 1 > sizeof(nbuf)) {
   1930  1.32    itojun 			h_errno = NO_RECOVERY;
   1931  1.70  christos 			return -1;
   1932  1.32    itojun 		}
   1933  1.54    itojun 		snprintf(nbuf, sizeof(nbuf), "%s.%s", name, domain);
   1934  1.32    itojun 	}
   1935  1.70  christos 	return res_queryN(longname, target, res);
   1936  1.32    itojun }
   1937