Home | History | Annotate | Line # | Download | only in net
getaddrinfo.c revision 1.109
      1  1.109  christos /*	$NetBSD: getaddrinfo.c,v 1.109 2015/09/10 14:05:06 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.109  christos __RCSID("$NetBSD: getaddrinfo.c,v 1.109 2015/09/10 14:05:06 christos Exp $");
     59   1.39    itojun #endif /* LIBC_SCCS and not lint */
     60    1.1    itojun 
     61  1.107     ozaki #ifndef RUMP_ACTION
     62   1.38    itojun #include "namespace.h"
     63  1.107     ozaki #endif
     64   1.14    itojun #include <sys/types.h>
     65    1.1    itojun #include <sys/param.h>
     66    1.1    itojun #include <sys/socket.h>
     67   1.14    itojun #include <net/if.h>
     68    1.1    itojun #include <netinet/in.h>
     69    1.1    itojun #include <arpa/inet.h>
     70    1.1    itojun #include <arpa/nameser.h>
     71   1.51     lukem #include <assert.h>
     72   1.51     lukem #include <ctype.h>
     73   1.51     lukem #include <errno.h>
     74    1.1    itojun #include <netdb.h>
     75    1.1    itojun #include <resolv.h>
     76   1.51     lukem #include <stddef.h>
     77   1.51     lukem #include <stdio.h>
     78   1.51     lukem #include <stdlib.h>
     79   1.14    itojun #include <string.h>
     80    1.1    itojun #include <unistd.h>
     81   1.98  christos #include <ifaddrs.h>
     82   1.14    itojun 
     83   1.32    itojun #include <syslog.h>
     84   1.32    itojun #include <stdarg.h>
     85   1.32    itojun #include <nsswitch.h>
     86   1.32    itojun 
     87   1.32    itojun #ifdef YP
     88   1.32    itojun #include <rpc/rpc.h>
     89   1.32    itojun #include <rpcsvc/yp_prot.h>
     90   1.32    itojun #include <rpcsvc/ypclnt.h>
     91   1.38    itojun #endif
     92   1.38    itojun 
     93   1.84  christos #include "servent.h"
     94   1.84  christos 
     95  1.107     ozaki #ifndef RUMP_ACTION
     96   1.38    itojun #ifdef __weak_alias
     97   1.38    itojun __weak_alias(getaddrinfo,_getaddrinfo)
     98  1.109  christos __weak_alias(allocaddrinfo,_allocaddrinfo)
     99   1.38    itojun __weak_alias(freeaddrinfo,_freeaddrinfo)
    100   1.38    itojun __weak_alias(gai_strerror,_gai_strerror)
    101   1.32    itojun #endif
    102  1.107     ozaki #endif
    103   1.32    itojun 
    104    1.1    itojun #define SUCCESS 0
    105    1.1    itojun #define ANY 0
    106    1.1    itojun #define YES 1
    107    1.1    itojun #define NO  0
    108    1.1    itojun 
    109    1.1    itojun static const char in_addrany[] = { 0, 0, 0, 0 };
    110   1.49     lukem static const char in_loopback[] = { 127, 0, 0, 1 };
    111   1.49     lukem #ifdef INET6
    112    1.1    itojun static const char in6_addrany[] = {
    113    1.1    itojun 	0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
    114    1.1    itojun };
    115    1.1    itojun static const char in6_loopback[] = {
    116    1.1    itojun 	0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1
    117    1.1    itojun };
    118   1.49     lukem #endif
    119    1.1    itojun 
    120   1.14    itojun static const struct afd {
    121    1.1    itojun 	int a_af;
    122    1.1    itojun 	int a_addrlen;
    123    1.1    itojun 	int a_socklen;
    124    1.1    itojun 	int a_off;
    125    1.1    itojun 	const char *a_addrany;
    126  1.102  christos 	const char *a_loopback;
    127   1.14    itojun 	int a_scoped;
    128    1.1    itojun } afdl [] = {
    129    1.1    itojun #ifdef INET6
    130    1.1    itojun 	{PF_INET6, sizeof(struct in6_addr),
    131    1.1    itojun 	 sizeof(struct sockaddr_in6),
    132    1.1    itojun 	 offsetof(struct sockaddr_in6, sin6_addr),
    133   1.14    itojun 	 in6_addrany, in6_loopback, 1},
    134    1.1    itojun #endif
    135    1.1    itojun 	{PF_INET, sizeof(struct in_addr),
    136    1.1    itojun 	 sizeof(struct sockaddr_in),
    137    1.1    itojun 	 offsetof(struct sockaddr_in, sin_addr),
    138   1.14    itojun 	 in_addrany, in_loopback, 0},
    139   1.14    itojun 	{0, 0, 0, 0, NULL, NULL, 0},
    140   1.14    itojun };
    141   1.14    itojun 
    142   1.14    itojun struct explore {
    143   1.14    itojun 	int e_af;
    144   1.14    itojun 	int e_socktype;
    145   1.14    itojun 	int e_protocol;
    146   1.14    itojun 	const char *e_protostr;
    147   1.14    itojun 	int e_wild;
    148   1.14    itojun #define WILD_AF(ex)		((ex)->e_wild & 0x01)
    149   1.14    itojun #define WILD_SOCKTYPE(ex)	((ex)->e_wild & 0x02)
    150   1.14    itojun #define WILD_PROTOCOL(ex)	((ex)->e_wild & 0x04)
    151   1.14    itojun };
    152   1.14    itojun 
    153   1.14    itojun static const struct explore explore[] = {
    154   1.14    itojun #if 0
    155   1.14    itojun 	{ PF_LOCAL, 0, ANY, ANY, NULL, 0x01 },
    156   1.14    itojun #endif
    157   1.14    itojun #ifdef INET6
    158   1.14    itojun 	{ PF_INET6, SOCK_DGRAM, IPPROTO_UDP, "udp", 0x07 },
    159   1.14    itojun 	{ PF_INET6, SOCK_STREAM, IPPROTO_TCP, "tcp", 0x07 },
    160   1.14    itojun 	{ PF_INET6, SOCK_RAW, ANY, NULL, 0x05 },
    161   1.14    itojun #endif
    162   1.14    itojun 	{ PF_INET, SOCK_DGRAM, IPPROTO_UDP, "udp", 0x07 },
    163   1.14    itojun 	{ PF_INET, SOCK_STREAM, IPPROTO_TCP, "tcp", 0x07 },
    164   1.14    itojun 	{ PF_INET, SOCK_RAW, ANY, NULL, 0x05 },
    165   1.32    itojun 	{ PF_UNSPEC, SOCK_DGRAM, IPPROTO_UDP, "udp", 0x07 },
    166   1.32    itojun 	{ PF_UNSPEC, SOCK_STREAM, IPPROTO_TCP, "tcp", 0x07 },
    167   1.32    itojun 	{ PF_UNSPEC, SOCK_RAW, ANY, NULL, 0x05 },
    168   1.14    itojun 	{ -1, 0, 0, NULL, 0 },
    169    1.1    itojun };
    170    1.1    itojun 
    171    1.1    itojun #ifdef INET6
    172    1.1    itojun #define PTON_MAX	16
    173    1.1    itojun #else
    174    1.1    itojun #define PTON_MAX	4
    175    1.1    itojun #endif
    176    1.1    itojun 
    177   1.32    itojun static const ns_src default_dns_files[] = {
    178  1.102  christos 	{ NSSRC_FILES,	NS_SUCCESS },
    179  1.102  christos 	{ NSSRC_DNS,	NS_SUCCESS },
    180   1.87  christos 	{ 0, 0 }
    181   1.32    itojun };
    182   1.32    itojun 
    183   1.63    itojun #define MAXPACKET	(64*1024)
    184   1.32    itojun 
    185   1.32    itojun typedef union {
    186   1.32    itojun 	HEADER hdr;
    187   1.32    itojun 	u_char buf[MAXPACKET];
    188   1.32    itojun } querybuf;
    189   1.32    itojun 
    190   1.32    itojun struct res_target {
    191   1.32    itojun 	struct res_target *next;
    192   1.32    itojun 	const char *name;	/* domain name */
    193   1.43    itojun 	int qclass, qtype;	/* class and type of query */
    194   1.32    itojun 	u_char *answer;		/* buffer to put answer */
    195   1.32    itojun 	int anslen;		/* size of answer buffer */
    196   1.32    itojun 	int n;			/* result length */
    197   1.32    itojun };
    198    1.1    itojun 
    199  1.103  christos struct srvinfo {
    200  1.103  christos        struct srvinfo *next;
    201  1.103  christos        char name[MAXDNAME];
    202  1.103  christos        int port, pri, weight;
    203  1.103  christos };
    204  1.103  christos 
    205  1.103  christos static int gai_srvok(const char *);
    206   1.70  christos static int str2number(const char *);
    207   1.70  christos static int explore_fqdn(const struct addrinfo *, const char *,
    208  1.102  christos     const char *, struct addrinfo **, struct servent_data *);
    209   1.70  christos static int explore_null(const struct addrinfo *,
    210  1.102  christos     const char *, struct addrinfo **, struct servent_data *);
    211   1.70  christos static int explore_numeric(const struct addrinfo *, const char *,
    212  1.102  christos     const char *, struct addrinfo **, const char *, struct servent_data *);
    213   1.70  christos static int explore_numeric_scope(const struct addrinfo *, const char *,
    214  1.102  christos     const char *, struct addrinfo **, struct servent_data *);
    215   1.70  christos static int get_canonname(const struct addrinfo *,
    216  1.102  christos     struct addrinfo *, const char *);
    217   1.70  christos static struct addrinfo *get_ai(const struct addrinfo *,
    218  1.102  christos     const struct afd *, const char *);
    219   1.89  christos static int get_portmatch(const struct addrinfo *, const char *,
    220   1.89  christos     struct servent_data *);
    221   1.89  christos static int get_port(const struct addrinfo *, const char *, int,
    222   1.89  christos     struct servent_data *);
    223   1.70  christos static const struct afd *find_afd(int);
    224   1.98  christos static int addrconfig(uint64_t *);
    225   1.27    itojun #ifdef INET6
    226   1.70  christos static int ip6_str2scopeid(char *, struct sockaddr_in6 *, u_int32_t *);
    227   1.42    itojun #endif
    228   1.14    itojun 
    229   1.70  christos static struct addrinfo *getanswer(const querybuf *, int, const char *, int,
    230  1.102  christos     const struct addrinfo *);
    231   1.73    tsarna static void aisort(struct addrinfo *s, res_state res);
    232  1.103  christos static struct addrinfo * _dns_query(struct res_target *,
    233  1.103  christos     const struct addrinfo *, res_state, int);
    234  1.103  christos static struct addrinfo * _dns_srv_lookup(const char *, const char *,
    235  1.103  christos     const struct addrinfo *);
    236  1.103  christos static struct addrinfo * _dns_host_lookup(const char *,
    237  1.103  christos     const struct addrinfo *);
    238   1.70  christos static int _dns_getaddrinfo(void *, void *, va_list);
    239   1.70  christos static void _sethtent(FILE **);
    240   1.70  christos static void _endhtent(FILE **);
    241   1.70  christos static struct addrinfo *_gethtent(FILE **, const char *,
    242   1.70  christos     const struct addrinfo *);
    243   1.70  christos static int _files_getaddrinfo(void *, void *, va_list);
    244   1.32    itojun #ifdef YP
    245   1.70  christos static struct addrinfo *_yphostent(char *, const struct addrinfo *);
    246   1.70  christos static int _yp_getaddrinfo(void *, void *, va_list);
    247   1.32    itojun #endif
    248   1.32    itojun 
    249   1.70  christos static int res_queryN(const char *, struct res_target *, res_state);
    250   1.73    tsarna static int res_searchN(const char *, struct res_target *, res_state);
    251   1.70  christos static int res_querydomainN(const char *, const char *,
    252  1.102  christos     struct res_target *, res_state);
    253   1.32    itojun 
    254   1.53  jdolecek static const char * const ai_errlist[] = {
    255    1.7     lukem 	"Success",
    256    1.7     lukem 	"Address family for hostname not supported",	/* EAI_ADDRFAMILY */
    257  1.102  christos 	"Temporary failure in name resolution",		/* EAI_AGAIN	  */
    258  1.102  christos 	"Invalid value for ai_flags",			/* EAI_BADFLAGS	  */
    259  1.102  christos 	"Non-recoverable failure in name resolution",	/* EAI_FAIL	  */
    260  1.102  christos 	"ai_family not supported",			/* EAI_FAMILY	  */
    261  1.102  christos 	"Memory allocation failure",			/* EAI_MEMORY	  */
    262  1.102  christos 	"No address associated with hostname",		/* EAI_NODATA	  */
    263  1.102  christos 	"hostname nor servname provided, or not known", /* EAI_NONAME	  */
    264  1.102  christos 	"servname not supported for ai_socktype",	/* EAI_SERVICE	  */
    265  1.102  christos 	"ai_socktype not supported",			/* EAI_SOCKTYPE	  */
    266  1.102  christos 	"System error returned in errno",		/* EAI_SYSTEM	  */
    267    1.7     lukem 	"Invalid value for hints",			/* EAI_BADHINTS	  */
    268  1.102  christos 	"Resolved protocol is unknown",			/* EAI_PROTOCOL	  */
    269  1.102  christos 	"Argument buffer overflow",			/* EAI_OVERFLOW	  */
    270  1.102  christos 	"Unknown error",				/* EAI_MAX	  */
    271    1.1    itojun };
    272    1.1    itojun 
    273   1.14    itojun /* XXX macros that make external reference is BAD. */
    274   1.14    itojun 
    275  1.102  christos #define GET_AI(ai, afd, addr)					\
    276  1.102  christos do {								\
    277  1.102  christos 	/* external reference: pai, error, and label free */	\
    278  1.102  christos 	(ai) = get_ai(pai, (afd), (addr));			\
    279  1.102  christos 	if ((ai) == NULL) {					\
    280  1.102  christos 		error = EAI_MEMORY;				\
    281  1.102  christos 		goto free;					\
    282  1.102  christos 	}							\
    283   1.20   mycroft } while (/*CONSTCOND*/0)
    284   1.14    itojun 
    285  1.102  christos #define GET_PORT(ai, serv, svd)					\
    286  1.102  christos do {								\
    287  1.102  christos 	/* external reference: error and label free */		\
    288  1.102  christos 	error = get_port((ai), (serv), 0, (svd));		\
    289  1.102  christos 	if (error != 0)						\
    290  1.102  christos 		goto free;					\
    291   1.20   mycroft } while (/*CONSTCOND*/0)
    292   1.14    itojun 
    293  1.102  christos #define GET_CANONNAME(ai, str)					\
    294  1.102  christos do {								\
    295  1.102  christos 	/* external reference: pai, error and label free */	\
    296  1.102  christos 	error = get_canonname(pai, (ai), (str));		\
    297  1.102  christos 	if (error != 0)						\
    298  1.102  christos 		goto free;					\
    299   1.20   mycroft } while (/*CONSTCOND*/0)
    300   1.14    itojun 
    301  1.102  christos #define ERR(err)						\
    302  1.102  christos do {								\
    303  1.102  christos 	/* external reference: error, and label bad */		\
    304  1.102  christos 	error = (err);						\
    305  1.102  christos 	goto bad;						\
    306  1.102  christos 	/*NOTREACHED*/						\
    307   1.20   mycroft } while (/*CONSTCOND*/0)
    308   1.14    itojun 
    309  1.102  christos #define MATCH_FAMILY(x, y, w)						\
    310  1.102  christos 	((x) == (y) || (/*CONSTCOND*/(w) && ((x) == PF_UNSPEC ||	\
    311  1.102  christos 	    (y) == PF_UNSPEC)))
    312  1.102  christos #define MATCH(x, y, w)							\
    313   1.20   mycroft 	((x) == (y) || (/*CONSTCOND*/(w) && ((x) == ANY || (y) == ANY)))
    314    1.1    itojun 
    315   1.70  christos const char *
    316   1.70  christos gai_strerror(int ecode)
    317    1.1    itojun {
    318    1.1    itojun 	if (ecode < 0 || ecode > EAI_MAX)
    319    1.1    itojun 		ecode = EAI_MAX;
    320   1.70  christos 	return ai_errlist[ecode];
    321    1.1    itojun }
    322    1.1    itojun 
    323    1.1    itojun void
    324   1.70  christos freeaddrinfo(struct addrinfo *ai)
    325    1.1    itojun {
    326    1.1    itojun 	struct addrinfo *next;
    327    1.1    itojun 
    328   1.51     lukem 	_DIAGASSERT(ai != NULL);
    329   1.51     lukem 
    330   1.26    itojun 	do {
    331    1.1    itojun 		next = ai->ai_next;
    332    1.1    itojun 		if (ai->ai_canonname)
    333    1.1    itojun 			free(ai->ai_canonname);
    334    1.1    itojun 		/* no need to free(ai->ai_addr) */
    335    1.1    itojun 		free(ai);
    336   1.27    itojun 		ai = next;
    337   1.27    itojun 	} while (ai);
    338    1.1    itojun }
    339    1.1    itojun 
    340  1.103  christos /*
    341  1.103  christos  * We don't want localization to affect us
    342  1.103  christos  */
    343  1.103  christos #define PERIOD '.'
    344  1.103  christos #define hyphenchar(c) ((c) == '-')
    345  1.103  christos #define periodchar(c) ((c) == PERIOD)
    346  1.103  christos #define underschar(c) ((c) == '_')
    347  1.103  christos #define alphachar(c) (((c) >= 'a' && (c) <= 'z') || ((c) >= 'A' && (c) <= 'Z'))
    348  1.103  christos #define digitchar(c) ((c) >= '0' && (c) <= '9')
    349  1.103  christos 
    350  1.103  christos #define firstchar(c)  (alphachar(c) || digitchar(c) || underschar(c))
    351  1.103  christos #define lastchar(c)   (alphachar(c) || digitchar(c))
    352  1.103  christos #define middlechar(c) (lastchar(c) || hyphenchar(c))
    353  1.103  christos 
    354  1.103  christos static int
    355  1.103  christos gai_srvok(const char *dn)
    356  1.103  christos {
    357  1.103  christos 	int nch, pch, ch;
    358  1.103  christos 
    359  1.103  christos 	for (pch = PERIOD, nch = ch = *dn++; ch != '\0'; pch = ch, ch = nch) {
    360  1.103  christos 		if (periodchar(ch))
    361  1.103  christos 			continue;
    362  1.103  christos 		if (periodchar(pch)) {
    363  1.103  christos 			if (!firstchar(ch))
    364  1.103  christos 				return 0;
    365  1.103  christos 		} else if (periodchar(nch) || nch == '\0') {
    366  1.103  christos 			if (!lastchar(ch))
    367  1.103  christos 				return 0;
    368  1.103  christos 		} else if (!middlechar(ch))
    369  1.103  christos 			return 0;
    370  1.103  christos        }
    371  1.103  christos        return 1;
    372  1.103  christos }
    373  1.103  christos 
    374  1.103  christos static in_port_t *
    375  1.103  christos getport(struct addrinfo *ai) {
    376  1.103  christos 	static in_port_t p;
    377  1.103  christos 
    378  1.103  christos 	switch (ai->ai_family) {
    379  1.103  christos 	case AF_INET:
    380  1.103  christos 		return &((struct sockaddr_in *)(void *)ai->ai_addr)->sin_port;
    381  1.103  christos #ifdef INET6
    382  1.103  christos 	case AF_INET6:
    383  1.103  christos 		return &((struct sockaddr_in6 *)(void *)ai->ai_addr)->sin6_port;
    384  1.103  christos #endif
    385  1.103  christos 	default:
    386  1.103  christos 		p = 0;
    387  1.103  christos 		/* XXX: abort()? */
    388  1.103  christos 		return &p;
    389  1.103  christos 	}
    390  1.103  christos }
    391  1.103  christos 
    392    1.1    itojun static int
    393   1.70  christos str2number(const char *p)
    394    1.1    itojun {
    395   1.45    itojun 	char *ep;
    396   1.67    itojun 	unsigned long v;
    397   1.45    itojun 
    398   1.51     lukem 	_DIAGASSERT(p != NULL);
    399   1.51     lukem 
    400   1.46    itojun 	if (*p == '\0')
    401   1.67    itojun 		return -1;
    402   1.45    itojun 	ep = NULL;
    403   1.59    itojun 	errno = 0;
    404   1.67    itojun 	v = strtoul(p, &ep, 10);
    405   1.97  christos 	if (errno == 0 && ep && *ep == '\0' && v <= INT_MAX)
    406   1.97  christos 		return (int)v;
    407   1.45    itojun 	else
    408   1.67    itojun 		return -1;
    409    1.1    itojun }
    410    1.1    itojun 
    411    1.1    itojun int
    412   1.70  christos getaddrinfo(const char *hostname, const char *servname,
    413   1.70  christos     const struct addrinfo *hints, struct addrinfo **res)
    414    1.1    itojun {
    415    1.1    itojun 	struct addrinfo sentinel;
    416    1.1    itojun 	struct addrinfo *cur;
    417   1.14    itojun 	int error = 0;
    418    1.1    itojun 	struct addrinfo ai;
    419   1.14    itojun 	struct addrinfo ai0;
    420    1.1    itojun 	struct addrinfo *pai;
    421   1.14    itojun 	const struct explore *ex;
    422   1.89  christos 	struct servent_data svd;
    423   1.98  christos 	uint64_t mask = (uint64_t)~0ULL;
    424    1.1    itojun 
    425   1.51     lukem 	/* hostname is allowed to be NULL */
    426   1.51     lukem 	/* servname is allowed to be NULL */
    427   1.51     lukem 	/* hints is allowed to be NULL */
    428   1.51     lukem 	_DIAGASSERT(res != NULL);
    429   1.51     lukem 
    430   1.89  christos 	(void)memset(&svd, 0, sizeof(svd));
    431   1.32    itojun 	memset(&sentinel, 0, sizeof(sentinel));
    432    1.1    itojun 	cur = &sentinel;
    433   1.86  christos 	memset(&ai, 0, sizeof(ai));
    434    1.1    itojun 	pai = &ai;
    435    1.1    itojun 	pai->ai_flags = 0;
    436    1.1    itojun 	pai->ai_family = PF_UNSPEC;
    437    1.1    itojun 	pai->ai_socktype = ANY;
    438    1.1    itojun 	pai->ai_protocol = ANY;
    439    1.1    itojun 	pai->ai_addrlen = 0;
    440    1.1    itojun 	pai->ai_canonname = NULL;
    441    1.1    itojun 	pai->ai_addr = NULL;
    442    1.1    itojun 	pai->ai_next = NULL;
    443  1.102  christos 
    444    1.1    itojun 	if (hostname == NULL && servname == NULL)
    445    1.1    itojun 		return EAI_NONAME;
    446    1.1    itojun 	if (hints) {
    447    1.1    itojun 		/* error check for hints */
    448    1.1    itojun 		if (hints->ai_addrlen || hints->ai_canonname ||
    449    1.1    itojun 		    hints->ai_addr || hints->ai_next)
    450    1.1    itojun 			ERR(EAI_BADHINTS); /* xxx */
    451    1.1    itojun 		if (hints->ai_flags & ~AI_MASK)
    452    1.1    itojun 			ERR(EAI_BADFLAGS);
    453    1.1    itojun 		switch (hints->ai_family) {
    454    1.1    itojun 		case PF_UNSPEC:
    455    1.1    itojun 		case PF_INET:
    456    1.1    itojun #ifdef INET6
    457    1.1    itojun 		case PF_INET6:
    458    1.1    itojun #endif
    459    1.1    itojun 			break;
    460    1.1    itojun 		default:
    461    1.1    itojun 			ERR(EAI_FAMILY);
    462    1.1    itojun 		}
    463    1.1    itojun 		memcpy(pai, hints, sizeof(*pai));
    464   1.14    itojun 
    465   1.14    itojun 		/*
    466   1.14    itojun 		 * if both socktype/protocol are specified, check if they
    467   1.14    itojun 		 * are meaningful combination.
    468   1.14    itojun 		 */
    469   1.14    itojun 		if (pai->ai_socktype != ANY && pai->ai_protocol != ANY) {
    470   1.14    itojun 			for (ex = explore; ex->e_af >= 0; ex++) {
    471   1.14    itojun 				if (pai->ai_family != ex->e_af)
    472   1.14    itojun 					continue;
    473   1.14    itojun 				if (ex->e_socktype == ANY)
    474   1.14    itojun 					continue;
    475   1.14    itojun 				if (ex->e_protocol == ANY)
    476   1.14    itojun 					continue;
    477   1.14    itojun 				if (pai->ai_socktype == ex->e_socktype
    478   1.14    itojun 				 && pai->ai_protocol != ex->e_protocol) {
    479   1.14    itojun 					ERR(EAI_BADHINTS);
    480   1.14    itojun 				}
    481    1.1    itojun 			}
    482    1.1    itojun 		}
    483    1.1    itojun 	}
    484    1.1    itojun 
    485   1.98  christos 	if ((pai->ai_flags & AI_ADDRCONFIG) != 0 && addrconfig(&mask) == -1)
    486   1.98  christos 		ERR(EAI_FAIL);
    487   1.98  christos 
    488    1.1    itojun 	/*
    489   1.14    itojun 	 * check for special cases.  (1) numeric servname is disallowed if
    490   1.14    itojun 	 * socktype/protocol are left unspecified. (2) servname is disallowed
    491   1.14    itojun 	 * for raw and other inet{,6} sockets.
    492    1.1    itojun 	 */
    493   1.14    itojun 	if (MATCH_FAMILY(pai->ai_family, PF_INET, 1)
    494   1.27    itojun #ifdef PF_INET6
    495   1.27    itojun 	 || MATCH_FAMILY(pai->ai_family, PF_INET6, 1)
    496   1.27    itojun #endif
    497   1.27    itojun 	    ) {
    498   1.23    itojun 		ai0 = *pai;	/* backup *pai */
    499   1.21   mycroft 
    500   1.27    itojun 		if (pai->ai_family == PF_UNSPEC) {
    501   1.27    itojun #ifdef PF_INET6
    502   1.14    itojun 			pai->ai_family = PF_INET6;
    503   1.27    itojun #else
    504   1.27    itojun 			pai->ai_family = PF_INET;
    505   1.27    itojun #endif
    506   1.27    itojun 		}
    507   1.89  christos 		error = get_portmatch(pai, servname, &svd);
    508   1.14    itojun 		if (error)
    509  1.101    martin 			goto bad;
    510   1.21   mycroft 
    511   1.21   mycroft 		*pai = ai0;
    512   1.14    itojun 	}
    513   1.14    itojun 
    514   1.14    itojun 	ai0 = *pai;
    515   1.14    itojun 
    516   1.14    itojun 	/* NULL hostname, or numeric hostname */
    517   1.14    itojun 	for (ex = explore; ex->e_af >= 0; ex++) {
    518   1.14    itojun 		*pai = ai0;
    519   1.14    itojun 
    520   1.98  christos 		/* ADDRCONFIG check */
    521   1.98  christos 		if ((((uint64_t)1 << ex->e_af) & mask) == 0)
    522   1.98  christos 			continue;
    523   1.98  christos 
    524   1.32    itojun 		/* PF_UNSPEC entries are prepared for DNS queries only */
    525   1.32    itojun 		if (ex->e_af == PF_UNSPEC)
    526   1.32    itojun 			continue;
    527   1.32    itojun 
    528   1.14    itojun 		if (!MATCH_FAMILY(pai->ai_family, ex->e_af, WILD_AF(ex)))
    529   1.14    itojun 			continue;
    530   1.14    itojun 		if (!MATCH(pai->ai_socktype, ex->e_socktype, WILD_SOCKTYPE(ex)))
    531   1.14    itojun 			continue;
    532   1.14    itojun 		if (!MATCH(pai->ai_protocol, ex->e_protocol, WILD_PROTOCOL(ex)))
    533   1.14    itojun 			continue;
    534   1.14    itojun 		if (pai->ai_family == PF_UNSPEC)
    535   1.14    itojun 			pai->ai_family = ex->e_af;
    536   1.14    itojun 		if (pai->ai_socktype == ANY && ex->e_socktype != ANY)
    537   1.14    itojun 			pai->ai_socktype = ex->e_socktype;
    538   1.14    itojun 		if (pai->ai_protocol == ANY && ex->e_protocol != ANY)
    539   1.14    itojun 			pai->ai_protocol = ex->e_protocol;
    540   1.14    itojun 
    541   1.14    itojun 		if (hostname == NULL)
    542   1.89  christos 			error = explore_null(pai, servname, &cur->ai_next,
    543   1.89  christos 			    &svd);
    544   1.14    itojun 		else
    545   1.65    itojun 			error = explore_numeric_scope(pai, hostname, servname,
    546   1.89  christos 			    &cur->ai_next, &svd);
    547   1.14    itojun 
    548   1.14    itojun 		if (error)
    549   1.14    itojun 			goto free;
    550   1.14    itojun 
    551   1.79  christos 		while (cur->ai_next)
    552   1.14    itojun 			cur = cur->ai_next;
    553   1.14    itojun 	}
    554   1.14    itojun 
    555   1.14    itojun 	/*
    556   1.14    itojun 	 * XXX
    557   1.76  ginsbach 	 * If numeric representation of AF1 can be interpreted as FQDN
    558   1.14    itojun 	 * representation of AF2, we need to think again about the code below.
    559   1.14    itojun 	 */
    560   1.14    itojun 	if (sentinel.ai_next)
    561   1.14    itojun 		goto good;
    562   1.14    itojun 
    563   1.14    itojun 	if (hostname == NULL)
    564   1.43    itojun 		ERR(EAI_NODATA);
    565   1.64    itojun 	if (pai->ai_flags & AI_NUMERICHOST)
    566   1.64    itojun 		ERR(EAI_NONAME);
    567   1.14    itojun 
    568   1.14    itojun 	/*
    569   1.14    itojun 	 * hostname as alphabetical name.
    570   1.14    itojun 	 * we would like to prefer AF_INET6 than AF_INET, so we'll make a
    571   1.14    itojun 	 * outer loop by AFs.
    572   1.14    itojun 	 */
    573   1.32    itojun 	for (ex = explore; ex->e_af >= 0; ex++) {
    574   1.14    itojun 		*pai = ai0;
    575   1.14    itojun 
    576  1.100    kardel 
    577   1.98  christos 		/* ADDRCONFIG check */
    578  1.100    kardel 		/* PF_UNSPEC entries are prepared for DNS queries only */
    579  1.100    kardel 		if (ex->e_af != PF_UNSPEC &&
    580  1.100    kardel 		    (((uint64_t)1 << ex->e_af) & mask) == 0)
    581   1.98  christos 			continue;
    582   1.98  christos 
    583   1.32    itojun 		/* require exact match for family field */
    584   1.32    itojun 		if (pai->ai_family != ex->e_af)
    585   1.14    itojun 			continue;
    586   1.14    itojun 
    587   1.32    itojun 		if (!MATCH(pai->ai_socktype, ex->e_socktype,
    588   1.32    itojun 				WILD_SOCKTYPE(ex))) {
    589   1.32    itojun 			continue;
    590   1.32    itojun 		}
    591   1.32    itojun 		if (!MATCH(pai->ai_protocol, ex->e_protocol,
    592   1.32    itojun 				WILD_PROTOCOL(ex))) {
    593   1.32    itojun 			continue;
    594   1.32    itojun 		}
    595   1.14    itojun 
    596   1.32    itojun 		if (pai->ai_socktype == ANY && ex->e_socktype != ANY)
    597   1.32    itojun 			pai->ai_socktype = ex->e_socktype;
    598   1.32    itojun 		if (pai->ai_protocol == ANY && ex->e_protocol != ANY)
    599   1.32    itojun 			pai->ai_protocol = ex->e_protocol;
    600    1.1    itojun 
    601   1.89  christos 		error = explore_fqdn(pai, hostname, servname, &cur->ai_next,
    602   1.89  christos 		    &svd);
    603   1.14    itojun 
    604   1.32    itojun 		while (cur && cur->ai_next)
    605   1.32    itojun 			cur = cur->ai_next;
    606    1.1    itojun 	}
    607   1.14    itojun 
    608   1.14    itojun 	/* XXX */
    609   1.14    itojun 	if (sentinel.ai_next)
    610   1.14    itojun 		error = 0;
    611   1.14    itojun 
    612   1.14    itojun 	if (error)
    613   1.14    itojun 		goto free;
    614   1.89  christos 
    615   1.89  christos 	if (sentinel.ai_next) {
    616   1.14    itojun  good:
    617   1.89  christos 		endservent_r(&svd);
    618   1.89  christos 		*res = sentinel.ai_next;
    619   1.89  christos 		return SUCCESS;
    620   1.89  christos 	} else
    621   1.89  christos 		error = EAI_FAIL;
    622   1.14    itojun  free:
    623   1.14    itojun  bad:
    624   1.89  christos 	endservent_r(&svd);
    625   1.14    itojun 	if (sentinel.ai_next)
    626   1.14    itojun 		freeaddrinfo(sentinel.ai_next);
    627   1.14    itojun 	*res = NULL;
    628   1.14    itojun 	return error;
    629   1.14    itojun }
    630   1.14    itojun 
    631   1.14    itojun /*
    632   1.14    itojun  * FQDN hostname, DNS lookup
    633   1.14    itojun  */
    634   1.14    itojun static int
    635   1.70  christos explore_fqdn(const struct addrinfo *pai, const char *hostname,
    636   1.89  christos     const char *servname, struct addrinfo **res, struct servent_data *svd)
    637   1.14    itojun {
    638   1.32    itojun 	struct addrinfo *result;
    639   1.32    itojun 	struct addrinfo *cur;
    640   1.27    itojun 	int error = 0;
    641   1.32    itojun 	static const ns_dtab dtab[] = {
    642   1.32    itojun 		NS_FILES_CB(_files_getaddrinfo, NULL)
    643   1.32    itojun 		{ NSSRC_DNS, _dns_getaddrinfo, NULL },	/* force -DHESIOD */
    644   1.32    itojun 		NS_NIS_CB(_yp_getaddrinfo, NULL)
    645   1.87  christos 		NS_NULL_CB
    646   1.32    itojun 	};
    647   1.14    itojun 
    648   1.51     lukem 	_DIAGASSERT(pai != NULL);
    649   1.51     lukem 	/* hostname may be NULL */
    650   1.51     lukem 	/* servname may be NULL */
    651   1.51     lukem 	_DIAGASSERT(res != NULL);
    652   1.51     lukem 
    653   1.32    itojun 	result = NULL;
    654   1.14    itojun 
    655   1.14    itojun 	/*
    656   1.14    itojun 	 * if the servname does not match socktype/protocol, ignore it.
    657   1.14    itojun 	 */
    658   1.89  christos 	if (get_portmatch(pai, servname, svd) != 0)
    659   1.14    itojun 		return 0;
    660   1.14    itojun 
    661   1.32    itojun 	switch (nsdispatch(&result, dtab, NSDB_HOSTS, "getaddrinfo",
    662  1.103  christos 	    default_dns_files, hostname, pai, servname)) {
    663   1.32    itojun 	case NS_TRYAGAIN:
    664   1.32    itojun 		error = EAI_AGAIN;
    665   1.32    itojun 		goto free;
    666   1.32    itojun 	case NS_UNAVAIL:
    667   1.14    itojun 		error = EAI_FAIL;
    668   1.14    itojun 		goto free;
    669   1.32    itojun 	case NS_NOTFOUND:
    670   1.32    itojun 		error = EAI_NODATA;
    671   1.15    itojun 		goto free;
    672   1.32    itojun 	case NS_SUCCESS:
    673   1.32    itojun 		error = 0;
    674   1.32    itojun 		for (cur = result; cur; cur = cur->ai_next) {
    675  1.103  christos 			/* Check for already filled port. */
    676  1.103  christos 			if (*getport(cur))
    677  1.103  christos 				continue;
    678   1.89  christos 			GET_PORT(cur, servname, svd);
    679   1.32    itojun 			/* canonname should be filled already */
    680   1.16    itojun 		}
    681   1.32    itojun 		break;
    682   1.16    itojun 	}
    683   1.15    itojun 
    684   1.32    itojun 	*res = result;
    685   1.14    itojun 
    686   1.14    itojun 	return 0;
    687   1.14    itojun 
    688   1.14    itojun free:
    689   1.34    itojun 	if (result)
    690   1.34    itojun 		freeaddrinfo(result);
    691   1.14    itojun 	return error;
    692   1.14    itojun }
    693   1.14    itojun 
    694   1.14    itojun /*
    695   1.14    itojun  * hostname == NULL.
    696   1.14    itojun  * passive socket -> anyaddr (0.0.0.0 or ::)
    697   1.14    itojun  * non-passive socket -> localhost (127.0.0.1 or ::1)
    698   1.14    itojun  */
    699   1.14    itojun static int
    700   1.70  christos explore_null(const struct addrinfo *pai, const char *servname,
    701   1.89  christos     struct addrinfo **res, struct servent_data *svd)
    702   1.14    itojun {
    703   1.14    itojun 	int s;
    704   1.14    itojun 	const struct afd *afd;
    705   1.14    itojun 	struct addrinfo *cur;
    706   1.14    itojun 	struct addrinfo sentinel;
    707   1.14    itojun 	int error;
    708   1.14    itojun 
    709   1.51     lukem 	_DIAGASSERT(pai != NULL);
    710   1.51     lukem 	/* servname may be NULL */
    711   1.51     lukem 	_DIAGASSERT(res != NULL);
    712   1.51     lukem 
    713   1.14    itojun 	*res = NULL;
    714   1.14    itojun 	sentinel.ai_next = NULL;
    715   1.14    itojun 	cur = &sentinel;
    716   1.14    itojun 
    717   1.14    itojun 	/*
    718   1.14    itojun 	 * filter out AFs that are not supported by the kernel
    719   1.14    itojun 	 * XXX errno?
    720   1.14    itojun 	 */
    721   1.14    itojun 	s = socket(pai->ai_family, SOCK_DGRAM, 0);
    722   1.14    itojun 	if (s < 0) {
    723   1.14    itojun 		if (errno != EMFILE)
    724   1.14    itojun 			return 0;
    725   1.14    itojun 	} else
    726   1.14    itojun 		close(s);
    727   1.14    itojun 
    728   1.14    itojun 	/*
    729   1.14    itojun 	 * if the servname does not match socktype/protocol, ignore it.
    730   1.14    itojun 	 */
    731   1.89  christos 	if (get_portmatch(pai, servname, svd) != 0)
    732   1.14    itojun 		return 0;
    733   1.14    itojun 
    734   1.14    itojun 	afd = find_afd(pai->ai_family);
    735   1.42    itojun 	if (afd == NULL)
    736   1.42    itojun 		return 0;
    737   1.14    itojun 
    738   1.14    itojun 	if (pai->ai_flags & AI_PASSIVE) {
    739   1.14    itojun 		GET_AI(cur->ai_next, afd, afd->a_addrany);
    740   1.14    itojun 		/* xxx meaningless?
    741   1.14    itojun 		 * GET_CANONNAME(cur->ai_next, "anyaddr");
    742   1.14    itojun 		 */
    743   1.89  christos 		GET_PORT(cur->ai_next, servname, svd);
    744   1.14    itojun 	} else {
    745   1.14    itojun 		GET_AI(cur->ai_next, afd, afd->a_loopback);
    746   1.14    itojun 		/* xxx meaningless?
    747   1.14    itojun 		 * GET_CANONNAME(cur->ai_next, "localhost");
    748   1.14    itojun 		 */
    749   1.89  christos 		GET_PORT(cur->ai_next, servname, svd);
    750   1.14    itojun 	}
    751   1.14    itojun 	cur = cur->ai_next;
    752   1.14    itojun 
    753   1.14    itojun 	*res = sentinel.ai_next;
    754   1.14    itojun 	return 0;
    755   1.14    itojun 
    756   1.14    itojun free:
    757   1.14    itojun 	if (sentinel.ai_next)
    758   1.14    itojun 		freeaddrinfo(sentinel.ai_next);
    759   1.14    itojun 	return error;
    760   1.14    itojun }
    761   1.14    itojun 
    762   1.14    itojun /*
    763   1.14    itojun  * numeric hostname
    764   1.14    itojun  */
    765   1.14    itojun static int
    766   1.70  christos explore_numeric(const struct addrinfo *pai, const char *hostname,
    767   1.89  christos     const char *servname, struct addrinfo **res, const char *canonname,
    768   1.89  christos     struct servent_data *svd)
    769   1.14    itojun {
    770   1.14    itojun 	const struct afd *afd;
    771   1.14    itojun 	struct addrinfo *cur;
    772   1.14    itojun 	struct addrinfo sentinel;
    773   1.14    itojun 	int error;
    774   1.14    itojun 	char pton[PTON_MAX];
    775   1.14    itojun 
    776   1.51     lukem 	_DIAGASSERT(pai != NULL);
    777   1.51     lukem 	/* hostname may be NULL */
    778   1.51     lukem 	/* servname may be NULL */
    779   1.51     lukem 	_DIAGASSERT(res != NULL);
    780   1.51     lukem 
    781   1.14    itojun 	*res = NULL;
    782   1.14    itojun 	sentinel.ai_next = NULL;
    783   1.14    itojun 	cur = &sentinel;
    784   1.14    itojun 
    785   1.14    itojun 	/*
    786   1.14    itojun 	 * if the servname does not match socktype/protocol, ignore it.
    787   1.14    itojun 	 */
    788   1.89  christos 	if (get_portmatch(pai, servname, svd) != 0)
    789   1.14    itojun 		return 0;
    790   1.14    itojun 
    791   1.14    itojun 	afd = find_afd(pai->ai_family);
    792   1.42    itojun 	if (afd == NULL)
    793   1.42    itojun 		return 0;
    794   1.14    itojun 
    795   1.27    itojun 	switch (afd->a_af) {
    796   1.27    itojun #if 0 /*X/Open spec*/
    797   1.27    itojun 	case AF_INET:
    798   1.27    itojun 		if (inet_aton(hostname, (struct in_addr *)pton) == 1) {
    799   1.27    itojun 			if (pai->ai_family == afd->a_af ||
    800   1.27    itojun 			    pai->ai_family == PF_UNSPEC /*?*/) {
    801   1.27    itojun 				GET_AI(cur->ai_next, afd, pton);
    802   1.89  christos 				GET_PORT(cur->ai_next, servname, svd);
    803   1.65    itojun 				if ((pai->ai_flags & AI_CANONNAME)) {
    804   1.65    itojun 					/*
    805   1.65    itojun 					 * Set the numeric address itself as
    806   1.65    itojun 					 * the canonical name, based on a
    807   1.65    itojun 					 * clarification in rfc2553bis-03.
    808   1.65    itojun 					 */
    809   1.65    itojun 					GET_CANONNAME(cur->ai_next, canonname);
    810   1.65    itojun 				}
    811   1.27    itojun 				while (cur && cur->ai_next)
    812   1.27    itojun 					cur = cur->ai_next;
    813   1.42    itojun 			} else
    814   1.27    itojun 				ERR(EAI_FAMILY);	/*xxx*/
    815   1.27    itojun 		}
    816   1.27    itojun 		break;
    817   1.27    itojun #endif
    818   1.27    itojun 	default:
    819   1.27    itojun 		if (inet_pton(afd->a_af, hostname, pton) == 1) {
    820   1.27    itojun 			if (pai->ai_family == afd->a_af ||
    821   1.27    itojun 			    pai->ai_family == PF_UNSPEC /*?*/) {
    822   1.27    itojun 				GET_AI(cur->ai_next, afd, pton);
    823   1.89  christos 				GET_PORT(cur->ai_next, servname, svd);
    824   1.65    itojun 				if ((pai->ai_flags & AI_CANONNAME)) {
    825   1.65    itojun 					/*
    826   1.65    itojun 					 * Set the numeric address itself as
    827   1.65    itojun 					 * the canonical name, based on a
    828   1.65    itojun 					 * clarification in rfc2553bis-03.
    829   1.65    itojun 					 */
    830   1.65    itojun 					GET_CANONNAME(cur->ai_next, canonname);
    831   1.65    itojun 				}
    832   1.81  christos 				while (cur->ai_next)
    833   1.27    itojun 					cur = cur->ai_next;
    834   1.42    itojun 			} else
    835   1.27    itojun 				ERR(EAI_FAMILY);	/*xxx*/
    836   1.27    itojun 		}
    837   1.27    itojun 		break;
    838    1.1    itojun 	}
    839    1.1    itojun 
    840   1.14    itojun 	*res = sentinel.ai_next;
    841   1.14    itojun 	return 0;
    842   1.14    itojun 
    843   1.14    itojun free:
    844   1.14    itojun bad:
    845   1.14    itojun 	if (sentinel.ai_next)
    846   1.14    itojun 		freeaddrinfo(sentinel.ai_next);
    847   1.14    itojun 	return error;
    848   1.14    itojun }
    849   1.14    itojun 
    850   1.14    itojun /*
    851   1.14    itojun  * numeric hostname with scope
    852   1.14    itojun  */
    853   1.14    itojun static int
    854   1.70  christos explore_numeric_scope(const struct addrinfo *pai, const char *hostname,
    855   1.89  christos     const char *servname, struct addrinfo **res, struct servent_data *svd)
    856   1.14    itojun {
    857   1.29    itojun #if !defined(SCOPE_DELIMITER) || !defined(INET6)
    858   1.90  christos 	return explore_numeric(pai, hostname, servname, res, hostname, svd);
    859   1.14    itojun #else
    860   1.14    itojun 	const struct afd *afd;
    861   1.14    itojun 	struct addrinfo *cur;
    862   1.14    itojun 	int error;
    863   1.30    itojun 	char *cp, *hostname2 = NULL, *scope, *addr;
    864   1.14    itojun 	struct sockaddr_in6 *sin6;
    865   1.14    itojun 
    866   1.51     lukem 	_DIAGASSERT(pai != NULL);
    867   1.51     lukem 	/* hostname may be NULL */
    868   1.51     lukem 	/* servname may be NULL */
    869   1.51     lukem 	_DIAGASSERT(res != NULL);
    870   1.51     lukem 
    871   1.14    itojun 	/*
    872   1.14    itojun 	 * if the servname does not match socktype/protocol, ignore it.
    873   1.14    itojun 	 */
    874   1.89  christos 	if (get_portmatch(pai, servname, svd) != 0)
    875   1.14    itojun 		return 0;
    876   1.14    itojun 
    877   1.14    itojun 	afd = find_afd(pai->ai_family);
    878   1.42    itojun 	if (afd == NULL)
    879   1.42    itojun 		return 0;
    880   1.42    itojun 
    881   1.14    itojun 	if (!afd->a_scoped)
    882   1.89  christos 		return explore_numeric(pai, hostname, servname, res, hostname,
    883   1.89  christos 		    svd);
    884   1.14    itojun 
    885   1.14    itojun 	cp = strchr(hostname, SCOPE_DELIMITER);
    886   1.14    itojun 	if (cp == NULL)
    887   1.89  christos 		return explore_numeric(pai, hostname, servname, res, hostname,
    888   1.89  christos 		    svd);
    889   1.14    itojun 
    890   1.30    itojun 	/*
    891   1.30    itojun 	 * Handle special case of <scoped_address><delimiter><scope id>
    892   1.30    itojun 	 */
    893   1.30    itojun 	hostname2 = strdup(hostname);
    894   1.30    itojun 	if (hostname2 == NULL)
    895   1.30    itojun 		return EAI_MEMORY;
    896   1.30    itojun 	/* terminate at the delimiter */
    897   1.30    itojun 	hostname2[cp - hostname] = '\0';
    898   1.30    itojun 	addr = hostname2;
    899   1.30    itojun 	scope = cp + 1;
    900    1.1    itojun 
    901   1.89  christos 	error = explore_numeric(pai, addr, servname, res, hostname, svd);
    902    1.1    itojun 	if (error == 0) {
    903   1.59    itojun 		u_int32_t scopeid;
    904   1.27    itojun 
    905   1.14    itojun 		for (cur = *res; cur; cur = cur->ai_next) {
    906   1.14    itojun 			if (cur->ai_family != AF_INET6)
    907   1.14    itojun 				continue;
    908   1.36  christos 			sin6 = (struct sockaddr_in6 *)(void *)cur->ai_addr;
    909   1.60    itojun 			if (ip6_str2scopeid(scope, sin6, &scopeid) == -1) {
    910   1.27    itojun 				free(hostname2);
    911   1.43    itojun 				return(EAI_NODATA); /* XXX: is return OK? */
    912   1.27    itojun 			}
    913   1.27    itojun 			sin6->sin6_scope_id = scopeid;
    914   1.14    itojun 		}
    915    1.1    itojun 	}
    916   1.14    itojun 
    917   1.14    itojun 	free(hostname2);
    918   1.14    itojun 
    919    1.1    itojun 	return error;
    920   1.14    itojun #endif
    921    1.1    itojun }
    922    1.1    itojun 
    923    1.1    itojun static int
    924   1.70  christos get_canonname(const struct addrinfo *pai, struct addrinfo *ai, const char *str)
    925    1.1    itojun {
    926   1.51     lukem 
    927   1.51     lukem 	_DIAGASSERT(pai != NULL);
    928   1.51     lukem 	_DIAGASSERT(ai != NULL);
    929   1.51     lukem 	_DIAGASSERT(str != NULL);
    930   1.51     lukem 
    931   1.14    itojun 	if ((pai->ai_flags & AI_CANONNAME) != 0) {
    932   1.66    itojun 		ai->ai_canonname = strdup(str);
    933   1.14    itojun 		if (ai->ai_canonname == NULL)
    934   1.14    itojun 			return EAI_MEMORY;
    935   1.14    itojun 	}
    936   1.14    itojun 	return 0;
    937   1.14    itojun }
    938    1.1    itojun 
    939  1.109  christos struct addrinfo *
    940   1.93    tsarna allocaddrinfo(socklen_t addrlen)
    941   1.93    tsarna {
    942   1.93    tsarna 	struct addrinfo *ai;
    943   1.93    tsarna 
    944   1.93    tsarna 	ai = calloc(sizeof(struct addrinfo) + addrlen, 1);
    945   1.93    tsarna 	if (ai) {
    946   1.93    tsarna 		ai->ai_addr = (void *)(ai+1);
    947   1.93    tsarna 		ai->ai_addrlen = ai->ai_addr->sa_len = addrlen;
    948   1.93    tsarna 	}
    949   1.93    tsarna 
    950   1.93    tsarna 	return ai;
    951   1.93    tsarna }
    952   1.93    tsarna 
    953   1.14    itojun static struct addrinfo *
    954   1.70  christos get_ai(const struct addrinfo *pai, const struct afd *afd, const char *addr)
    955   1.14    itojun {
    956   1.14    itojun 	char *p;
    957   1.14    itojun 	struct addrinfo *ai;
    958   1.93    tsarna 	struct sockaddr *save;
    959   1.12     lukem 
    960   1.51     lukem 	_DIAGASSERT(pai != NULL);
    961   1.51     lukem 	_DIAGASSERT(afd != NULL);
    962   1.51     lukem 	_DIAGASSERT(addr != NULL);
    963   1.51     lukem 
    964   1.93    tsarna 	ai = allocaddrinfo((socklen_t)afd->a_socklen);
    965   1.14    itojun 	if (ai == NULL)
    966   1.14    itojun 		return NULL;
    967   1.14    itojun 
    968  1.102  christos 	save = ai->ai_addr;
    969   1.14    itojun 	memcpy(ai, pai, sizeof(struct addrinfo));
    970   1.93    tsarna 
    971  1.102  christos 	/* since we just overwrote all of ai, we have
    972  1.102  christos 	   to restore ai_addr and ai_addrlen */
    973  1.102  christos 	ai->ai_addr = save;
    974  1.102  christos 	ai->ai_addrlen = (socklen_t)afd->a_socklen;
    975  1.102  christos 
    976   1.14    itojun 	ai->ai_addr->sa_family = ai->ai_family = afd->a_af;
    977   1.36  christos 	p = (char *)(void *)(ai->ai_addr);
    978   1.20   mycroft 	memcpy(p + afd->a_off, addr, (size_t)afd->a_addrlen);
    979   1.14    itojun 	return ai;
    980   1.14    itojun }
    981    1.1    itojun 
    982   1.14    itojun static int
    983   1.89  christos get_portmatch(const struct addrinfo *ai, const char *servname,
    984   1.89  christos     struct servent_data *svd)
    985   1.14    itojun {
    986    1.4    itojun 
    987   1.51     lukem 	_DIAGASSERT(ai != NULL);
    988   1.51     lukem 	/* servname may be NULL */
    989   1.51     lukem 
    990   1.89  christos 	return get_port(ai, servname, 1, svd);
    991    1.1    itojun }
    992    1.1    itojun 
    993    1.1    itojun static int
    994   1.89  christos get_port(const struct addrinfo *ai, const char *servname, int matchonly,
    995   1.89  christos     struct servent_data *svd)
    996    1.1    itojun {
    997   1.14    itojun 	const char *proto;
    998   1.14    itojun 	struct servent *sp;
    999   1.14    itojun 	int port;
   1000   1.14    itojun 	int allownumeric;
   1001   1.12     lukem 
   1002   1.51     lukem 	_DIAGASSERT(ai != NULL);
   1003   1.51     lukem 	/* servname may be NULL */
   1004   1.51     lukem 
   1005   1.14    itojun 	if (servname == NULL)
   1006   1.14    itojun 		return 0;
   1007   1.23    itojun 	switch (ai->ai_family) {
   1008   1.23    itojun 	case AF_INET:
   1009   1.27    itojun #ifdef AF_INET6
   1010   1.23    itojun 	case AF_INET6:
   1011   1.27    itojun #endif
   1012   1.23    itojun 		break;
   1013   1.23    itojun 	default:
   1014   1.14    itojun 		return 0;
   1015   1.23    itojun 	}
   1016    1.1    itojun 
   1017   1.14    itojun 	switch (ai->ai_socktype) {
   1018   1.14    itojun 	case SOCK_RAW:
   1019   1.14    itojun 		return EAI_SERVICE;
   1020   1.14    itojun 	case SOCK_DGRAM:
   1021   1.14    itojun 	case SOCK_STREAM:
   1022   1.14    itojun 		allownumeric = 1;
   1023   1.14    itojun 		break;
   1024   1.14    itojun 	case ANY:
   1025   1.92       tls 		/*
   1026  1.102  christos 		 * This was 0.	It is now 1 so that queries specifying
   1027   1.92       tls 		 * a NULL hint, or hint without socktype (but, hopefully,
   1028   1.92       tls 		 * with protocol) and numeric address actually work.
   1029   1.92       tls 		 */
   1030   1.92       tls 		allownumeric = 1;
   1031   1.14    itojun 		break;
   1032   1.14    itojun 	default:
   1033   1.14    itojun 		return EAI_SOCKTYPE;
   1034    1.1    itojun 	}
   1035   1.14    itojun 
   1036   1.67    itojun 	port = str2number(servname);
   1037   1.67    itojun 	if (port >= 0) {
   1038   1.14    itojun 		if (!allownumeric)
   1039   1.14    itojun 			return EAI_SERVICE;
   1040   1.14    itojun 		if (port < 0 || port > 65535)
   1041   1.14    itojun 			return EAI_SERVICE;
   1042   1.58    itojun 		port = htons(port);
   1043   1.14    itojun 	} else {
   1044   1.84  christos 		struct servent sv;
   1045   1.67    itojun 		if (ai->ai_flags & AI_NUMERICSERV)
   1046   1.67    itojun 			return EAI_NONAME;
   1047   1.67    itojun 
   1048   1.14    itojun 		switch (ai->ai_socktype) {
   1049   1.14    itojun 		case SOCK_DGRAM:
   1050   1.14    itojun 			proto = "udp";
   1051    1.1    itojun 			break;
   1052   1.14    itojun 		case SOCK_STREAM:
   1053   1.14    itojun 			proto = "tcp";
   1054    1.1    itojun 			break;
   1055    1.1    itojun 		default:
   1056   1.14    itojun 			proto = NULL;
   1057    1.1    itojun 			break;
   1058    1.1    itojun 		}
   1059   1.14    itojun 
   1060   1.89  christos 		sp = getservbyname_r(servname, proto, &sv, svd);
   1061   1.85  christos 		if (sp == NULL)
   1062   1.14    itojun 			return EAI_SERVICE;
   1063   1.14    itojun 		port = sp->s_port;
   1064    1.1    itojun 	}
   1065    1.1    itojun 
   1066  1.103  christos 	if (!matchonly)
   1067  1.103  christos 		*getport(__UNCONST(ai)) = port;
   1068   1.14    itojun 	return 0;
   1069   1.14    itojun }
   1070   1.14    itojun 
   1071   1.14    itojun static const struct afd *
   1072   1.70  christos find_afd(int af)
   1073   1.14    itojun {
   1074   1.14    itojun 	const struct afd *afd;
   1075    1.1    itojun 
   1076   1.14    itojun 	if (af == PF_UNSPEC)
   1077   1.14    itojun 		return NULL;
   1078   1.14    itojun 	for (afd = afdl; afd->a_af; afd++) {
   1079   1.14    itojun 		if (afd->a_af == af)
   1080   1.14    itojun 			return afd;
   1081    1.1    itojun 	}
   1082   1.14    itojun 	return NULL;
   1083    1.1    itojun }
   1084   1.27    itojun 
   1085   1.98  christos /*
   1086   1.98  christos  * AI_ADDRCONFIG check: Build a mask containing a bit set for each address
   1087   1.98  christos  * family configured in the system.
   1088   1.98  christos  *
   1089   1.98  christos  */
   1090   1.98  christos static int
   1091   1.98  christos addrconfig(uint64_t *mask)
   1092   1.98  christos {
   1093   1.98  christos 	struct ifaddrs *ifaddrs, *ifa;
   1094   1.98  christos 
   1095   1.98  christos 	if (getifaddrs(&ifaddrs) == -1)
   1096   1.98  christos 		return -1;
   1097   1.98  christos 
   1098   1.99    kardel 	*mask = 0;
   1099   1.98  christos 	for (ifa = ifaddrs; ifa != NULL; ifa = ifa->ifa_next)
   1100   1.98  christos 		if (ifa->ifa_addr && (ifa->ifa_flags & IFF_UP)) {
   1101   1.98  christos 			_DIAGASSERT(ifa->ifa_addr->sa_family < 64);
   1102   1.98  christos 			*mask |= (uint64_t)1 << ifa->ifa_addr->sa_family;
   1103   1.98  christos 		}
   1104   1.98  christos 
   1105   1.98  christos 	freeifaddrs(ifaddrs);
   1106   1.98  christos 	return 0;
   1107   1.98  christos }
   1108   1.98  christos 
   1109   1.27    itojun #ifdef INET6
   1110   1.27    itojun /* convert a string to a scope identifier. XXX: IPv6 specific */
   1111   1.60    itojun static int
   1112   1.70  christos ip6_str2scopeid(char *scope, struct sockaddr_in6 *sin6, u_int32_t *scopeid)
   1113   1.27    itojun {
   1114   1.59    itojun 	u_long lscopeid;
   1115   1.51     lukem 	struct in6_addr *a6;
   1116   1.27    itojun 	char *ep;
   1117   1.46    itojun 
   1118   1.51     lukem 	_DIAGASSERT(scope != NULL);
   1119   1.51     lukem 	_DIAGASSERT(sin6 != NULL);
   1120   1.60    itojun 	_DIAGASSERT(scopeid != NULL);
   1121   1.51     lukem 
   1122   1.51     lukem 	a6 = &sin6->sin6_addr;
   1123   1.51     lukem 
   1124   1.46    itojun 	/* empty scopeid portion is invalid */
   1125   1.46    itojun 	if (*scope == '\0')
   1126   1.46    itojun 		return -1;
   1127   1.27    itojun 
   1128   1.27    itojun 	if (IN6_IS_ADDR_LINKLOCAL(a6) || IN6_IS_ADDR_MC_LINKLOCAL(a6)) {
   1129   1.27    itojun 		/*
   1130   1.95       wiz 		 * We currently assume a one-to-one mapping between links
   1131   1.27    itojun 		 * and interfaces, so we simply use interface indices for
   1132   1.27    itojun 		 * like-local scopes.
   1133   1.27    itojun 		 */
   1134   1.60    itojun 		*scopeid = if_nametoindex(scope);
   1135   1.60    itojun 		if (*scopeid == 0)
   1136   1.27    itojun 			goto trynumeric;
   1137   1.60    itojun 		return 0;
   1138   1.27    itojun 	}
   1139   1.27    itojun 
   1140   1.27    itojun 	/* still unclear about literal, allow numeric only - placeholder */
   1141   1.27    itojun 	if (IN6_IS_ADDR_SITELOCAL(a6) || IN6_IS_ADDR_MC_SITELOCAL(a6))
   1142   1.27    itojun 		goto trynumeric;
   1143   1.27    itojun 	if (IN6_IS_ADDR_MC_ORGLOCAL(a6))
   1144   1.27    itojun 		goto trynumeric;
   1145   1.27    itojun 	else
   1146   1.27    itojun 		goto trynumeric;	/* global */
   1147   1.27    itojun 
   1148   1.27    itojun 	/* try to convert to a numeric id as a last resort */
   1149   1.27    itojun   trynumeric:
   1150   1.59    itojun 	errno = 0;
   1151   1.59    itojun 	lscopeid = strtoul(scope, &ep, 10);
   1152   1.61    itojun 	*scopeid = (u_int32_t)(lscopeid & 0xffffffffUL);
   1153   1.60    itojun 	if (errno == 0 && ep && *ep == '\0' && *scopeid == lscopeid)
   1154   1.60    itojun 		return 0;
   1155   1.27    itojun 	else
   1156   1.27    itojun 		return -1;
   1157   1.27    itojun }
   1158   1.42    itojun #endif
   1159   1.32    itojun 
   1160   1.32    itojun /* code duplicate with gethnamaddr.c */
   1161   1.32    itojun 
   1162   1.32    itojun static const char AskedForGot[] =
   1163   1.32    itojun 	"gethostby*.getanswer: asked for \"%s\", got \"%s\"";
   1164   1.32    itojun 
   1165   1.32    itojun static struct addrinfo *
   1166   1.70  christos getanswer(const querybuf *answer, int anslen, const char *qname, int qtype,
   1167   1.70  christos     const struct addrinfo *pai)
   1168   1.32    itojun {
   1169   1.32    itojun 	struct addrinfo sentinel, *cur;
   1170  1.103  christos 	struct addrinfo ai, *aip;
   1171   1.32    itojun 	const struct afd *afd;
   1172   1.32    itojun 	char *canonname;
   1173   1.32    itojun 	const HEADER *hp;
   1174   1.32    itojun 	const u_char *cp;
   1175   1.32    itojun 	int n;
   1176   1.32    itojun 	const u_char *eom;
   1177   1.56    itojun 	char *bp, *ep;
   1178   1.56    itojun 	int type, class, ancount, qdcount;
   1179   1.32    itojun 	int haveanswer, had_error;
   1180   1.32    itojun 	char tbuf[MAXDNAME];
   1181   1.70  christos 	int (*name_ok) (const char *);
   1182   1.32    itojun 	char hostbuf[8*1024];
   1183  1.103  christos 	int port, pri, weight;
   1184  1.103  christos 	struct srvinfo *srvlist, *srv, *csrv;
   1185   1.32    itojun 
   1186   1.51     lukem 	_DIAGASSERT(answer != NULL);
   1187   1.51     lukem 	_DIAGASSERT(qname != NULL);
   1188   1.51     lukem 	_DIAGASSERT(pai != NULL);
   1189   1.51     lukem 
   1190   1.32    itojun 	memset(&sentinel, 0, sizeof(sentinel));
   1191   1.32    itojun 	cur = &sentinel;
   1192   1.32    itojun 
   1193   1.32    itojun 	canonname = NULL;
   1194   1.32    itojun 	eom = answer->buf + anslen;
   1195   1.32    itojun 	switch (qtype) {
   1196   1.32    itojun 	case T_A:
   1197   1.32    itojun 	case T_AAAA:
   1198   1.32    itojun 	case T_ANY:	/*use T_ANY only for T_A/T_AAAA lookup*/
   1199   1.32    itojun 		name_ok = res_hnok;
   1200   1.32    itojun 		break;
   1201  1.103  christos 	case T_SRV:
   1202  1.103  christos 		name_ok = gai_srvok;
   1203  1.103  christos 		break;
   1204   1.32    itojun 	default:
   1205   1.70  christos 		return NULL;	/* XXX should be abort(); */
   1206   1.32    itojun 	}
   1207   1.32    itojun 	/*
   1208   1.32    itojun 	 * find first satisfactory answer
   1209   1.32    itojun 	 */
   1210   1.32    itojun 	hp = &answer->hdr;
   1211   1.32    itojun 	ancount = ntohs(hp->ancount);
   1212   1.32    itojun 	qdcount = ntohs(hp->qdcount);
   1213   1.32    itojun 	bp = hostbuf;
   1214   1.56    itojun 	ep = hostbuf + sizeof hostbuf;
   1215   1.32    itojun 	cp = answer->buf + HFIXEDSZ;
   1216   1.32    itojun 	if (qdcount != 1) {
   1217   1.32    itojun 		h_errno = NO_RECOVERY;
   1218   1.32    itojun 		return (NULL);
   1219   1.32    itojun 	}
   1220   1.97  christos 	n = dn_expand(answer->buf, eom, cp, bp, (int)(ep - bp));
   1221   1.32    itojun 	if ((n < 0) || !(*name_ok)(bp)) {
   1222   1.32    itojun 		h_errno = NO_RECOVERY;
   1223   1.32    itojun 		return (NULL);
   1224   1.32    itojun 	}
   1225   1.32    itojun 	cp += n + QFIXEDSZ;
   1226   1.32    itojun 	if (qtype == T_A || qtype == T_AAAA || qtype == T_ANY) {
   1227   1.32    itojun 		/* res_send() has already verified that the query name is the
   1228   1.32    itojun 		 * same as the one we sent; this just gets the expanded name
   1229   1.32    itojun 		 * (i.e., with the succeeding search-domain tacked on).
   1230   1.32    itojun 		 */
   1231   1.97  christos 		n = (int)strlen(bp) + 1;		/* for the \0 */
   1232   1.32    itojun 		if (n >= MAXHOSTNAMELEN) {
   1233   1.32    itojun 			h_errno = NO_RECOVERY;
   1234   1.32    itojun 			return (NULL);
   1235   1.32    itojun 		}
   1236   1.32    itojun 		canonname = bp;
   1237   1.32    itojun 		bp += n;
   1238   1.32    itojun 		/* The qname can be abbreviated, but h_name is now absolute. */
   1239   1.32    itojun 		qname = canonname;
   1240   1.32    itojun 	}
   1241   1.32    itojun 	haveanswer = 0;
   1242   1.32    itojun 	had_error = 0;
   1243  1.103  christos 	srvlist = NULL;
   1244   1.32    itojun 	while (ancount-- > 0 && cp < eom && !had_error) {
   1245   1.97  christos 		n = dn_expand(answer->buf, eom, cp, bp, (int)(ep - bp));
   1246   1.32    itojun 		if ((n < 0) || !(*name_ok)(bp)) {
   1247   1.32    itojun 			had_error++;
   1248   1.32    itojun 			continue;
   1249   1.32    itojun 		}
   1250   1.32    itojun 		cp += n;			/* name */
   1251   1.32    itojun 		type = _getshort(cp);
   1252  1.102  christos 		cp += INT16SZ;			/* type */
   1253   1.32    itojun 		class = _getshort(cp);
   1254  1.102  christos 		cp += INT16SZ + INT32SZ;	/* class, TTL */
   1255   1.32    itojun 		n = _getshort(cp);
   1256   1.32    itojun 		cp += INT16SZ;			/* len */
   1257   1.32    itojun 		if (class != C_IN) {
   1258   1.32    itojun 			/* XXX - debug? syslog? */
   1259   1.32    itojun 			cp += n;
   1260   1.32    itojun 			continue;		/* XXX - had_error++ ? */
   1261   1.32    itojun 		}
   1262   1.32    itojun 		if ((qtype == T_A || qtype == T_AAAA || qtype == T_ANY) &&
   1263   1.32    itojun 		    type == T_CNAME) {
   1264   1.97  christos 			n = dn_expand(answer->buf, eom, cp, tbuf, (int)sizeof tbuf);
   1265   1.32    itojun 			if ((n < 0) || !(*name_ok)(tbuf)) {
   1266   1.32    itojun 				had_error++;
   1267   1.32    itojun 				continue;
   1268   1.32    itojun 			}
   1269   1.32    itojun 			cp += n;
   1270   1.32    itojun 			/* Get canonical name. */
   1271   1.97  christos 			n = (int)strlen(tbuf) + 1;	/* for the \0 */
   1272   1.56    itojun 			if (n > ep - bp || n >= MAXHOSTNAMELEN) {
   1273   1.32    itojun 				had_error++;
   1274   1.32    itojun 				continue;
   1275   1.32    itojun 			}
   1276   1.66    itojun 			strlcpy(bp, tbuf, (size_t)(ep - bp));
   1277   1.32    itojun 			canonname = bp;
   1278   1.32    itojun 			bp += n;
   1279   1.32    itojun 			continue;
   1280   1.32    itojun 		}
   1281   1.32    itojun 		if (qtype == T_ANY) {
   1282   1.32    itojun 			if (!(type == T_A || type == T_AAAA)) {
   1283   1.32    itojun 				cp += n;
   1284   1.32    itojun 				continue;
   1285   1.32    itojun 			}
   1286   1.32    itojun 		} else if (type != qtype) {
   1287   1.88  christos 			if (type != T_KEY && type != T_SIG) {
   1288   1.88  christos 				struct syslog_data sd = SYSLOG_DATA_INIT;
   1289   1.88  christos 				syslog_r(LOG_NOTICE|LOG_AUTH, &sd,
   1290   1.32    itojun 	       "gethostby*.getanswer: asked for \"%s %s %s\", got type \"%s\"",
   1291   1.32    itojun 				       qname, p_class(C_IN), p_type(qtype),
   1292   1.32    itojun 				       p_type(type));
   1293   1.88  christos 			}
   1294   1.32    itojun 			cp += n;
   1295   1.32    itojun 			continue;		/* XXX - had_error++ ? */
   1296   1.32    itojun 		}
   1297   1.32    itojun 		switch (type) {
   1298   1.32    itojun 		case T_A:
   1299   1.32    itojun 		case T_AAAA:
   1300   1.32    itojun 			if (strcasecmp(canonname, bp) != 0) {
   1301   1.88  christos 				struct syslog_data sd = SYSLOG_DATA_INIT;
   1302   1.88  christos 				syslog_r(LOG_NOTICE|LOG_AUTH, &sd,
   1303   1.32    itojun 				       AskedForGot, canonname, bp);
   1304   1.32    itojun 				cp += n;
   1305   1.32    itojun 				continue;	/* XXX - had_error++ ? */
   1306   1.32    itojun 			}
   1307   1.32    itojun 			if (type == T_A && n != INADDRSZ) {
   1308   1.32    itojun 				cp += n;
   1309   1.32    itojun 				continue;
   1310   1.32    itojun 			}
   1311   1.32    itojun 			if (type == T_AAAA && n != IN6ADDRSZ) {
   1312   1.32    itojun 				cp += n;
   1313   1.32    itojun 				continue;
   1314   1.62    itojun 			}
   1315   1.62    itojun 			if (type == T_AAAA) {
   1316   1.62    itojun 				struct in6_addr in6;
   1317   1.62    itojun 				memcpy(&in6, cp, IN6ADDRSZ);
   1318   1.62    itojun 				if (IN6_IS_ADDR_V4MAPPED(&in6)) {
   1319   1.62    itojun 					cp += n;
   1320   1.62    itojun 					continue;
   1321   1.62    itojun 				}
   1322   1.32    itojun 			}
   1323   1.32    itojun 			if (!haveanswer) {
   1324   1.32    itojun 				int nn;
   1325   1.32    itojun 
   1326   1.32    itojun 				canonname = bp;
   1327   1.97  christos 				nn = (int)strlen(bp) + 1;	/* for the \0 */
   1328   1.32    itojun 				bp += nn;
   1329   1.32    itojun 			}
   1330   1.32    itojun 
   1331   1.32    itojun 			/* don't overwrite pai */
   1332   1.32    itojun 			ai = *pai;
   1333   1.32    itojun 			ai.ai_family = (type == T_A) ? AF_INET : AF_INET6;
   1334   1.32    itojun 			afd = find_afd(ai.ai_family);
   1335   1.32    itojun 			if (afd == NULL) {
   1336   1.32    itojun 				cp += n;
   1337   1.32    itojun 				continue;
   1338   1.32    itojun 			}
   1339   1.36  christos 			cur->ai_next = get_ai(&ai, afd, (const char *)cp);
   1340   1.32    itojun 			if (cur->ai_next == NULL)
   1341   1.32    itojun 				had_error++;
   1342   1.32    itojun 			while (cur && cur->ai_next)
   1343   1.32    itojun 				cur = cur->ai_next;
   1344   1.32    itojun 			cp += n;
   1345   1.32    itojun 			break;
   1346  1.103  christos 		case T_SRV:
   1347  1.103  christos 			/* Add to SRV list. Insertion sort on priority. */
   1348  1.103  christos 			pri = _getshort(cp);
   1349  1.103  christos 			cp += INT16SZ;
   1350  1.103  christos 			weight = _getshort(cp);
   1351  1.103  christos 			cp += INT16SZ;
   1352  1.103  christos 			port = _getshort(cp);
   1353  1.103  christos 			cp += INT16SZ;
   1354  1.103  christos 			n = dn_expand(answer->buf, eom, cp, tbuf,
   1355  1.103  christos 			    (int)sizeof(tbuf));
   1356  1.103  christos 			if ((n < 0) || !res_hnok(tbuf)) {
   1357  1.103  christos 				had_error++;
   1358  1.103  christos 				continue;
   1359  1.103  christos 			}
   1360  1.103  christos 			cp += n;
   1361  1.103  christos 			if (strlen(tbuf) + 1 >= MAXDNAME) {
   1362  1.103  christos 				had_error++;
   1363  1.103  christos 				continue;
   1364  1.103  christos 			}
   1365  1.103  christos 			srv = malloc(sizeof(*srv));
   1366  1.103  christos 			if (!srv) {
   1367  1.103  christos 				had_error++;
   1368  1.103  christos 				continue;
   1369  1.103  christos 			}
   1370  1.103  christos 			strlcpy(srv->name, tbuf, sizeof(srv->name));
   1371  1.103  christos 			srv->pri = pri;
   1372  1.103  christos 			srv->weight = weight;
   1373  1.103  christos 			srv->port = port;
   1374  1.103  christos 			/* Weight 0 is sorted before other weights. */
   1375  1.103  christos 			if (!srvlist
   1376  1.103  christos 			    || srv->pri < srvlist->pri
   1377  1.103  christos 			    || (srv->pri == srvlist->pri &&
   1378  1.103  christos 			    (!srv->weight || srvlist->weight))) {
   1379  1.103  christos 				srv->next = srvlist;
   1380  1.103  christos 				srvlist = srv;
   1381  1.103  christos 			} else {
   1382  1.103  christos 				for (csrv = srvlist;
   1383  1.103  christos 				    csrv->next && csrv->next->pri <= srv->pri;
   1384  1.103  christos 				    csrv = csrv->next) {
   1385  1.103  christos 					if (csrv->next->pri == srv->pri
   1386  1.103  christos 					    && (!srv->weight ||
   1387  1.103  christos 					    csrv->next->weight))
   1388  1.103  christos 						break;
   1389  1.103  christos 				}
   1390  1.103  christos 				srv->next = csrv->next;
   1391  1.103  christos 				csrv->next = srv;
   1392  1.103  christos 			}
   1393  1.103  christos 			continue; /* Don't add to haveanswer yet. */
   1394   1.32    itojun 		default:
   1395   1.32    itojun 			abort();
   1396   1.32    itojun 		}
   1397   1.32    itojun 		if (!had_error)
   1398   1.32    itojun 			haveanswer++;
   1399   1.32    itojun 	}
   1400  1.103  christos 
   1401  1.103  christos 	if (srvlist) {
   1402  1.103  christos 		res_state res;
   1403  1.103  christos 		/*
   1404  1.103  christos 		 * Check for explicit rejection.
   1405  1.103  christos 		 */
   1406  1.103  christos 		if (!srvlist->next && !srvlist->name[0]) {
   1407  1.103  christos 			free(srvlist);
   1408  1.103  christos 			h_errno = HOST_NOT_FOUND;
   1409  1.103  christos 			return NULL;
   1410  1.103  christos 		}
   1411  1.103  christos 		res = __res_get_state();
   1412  1.103  christos 		if (res == NULL) {
   1413  1.104  christos 			while (srvlist != NULL) {
   1414  1.104  christos 				srv = srvlist;
   1415  1.104  christos 				srvlist = srvlist->next;
   1416  1.104  christos 				free(srv);
   1417  1.104  christos 			}
   1418  1.103  christos 			h_errno = NETDB_INTERNAL;
   1419  1.103  christos 			return NULL;
   1420  1.103  christos 		}
   1421  1.103  christos 
   1422  1.103  christos 		while (srvlist) {
   1423  1.103  christos 			struct res_target q, q2;
   1424  1.103  christos 
   1425  1.103  christos 			srv = srvlist;
   1426  1.103  christos 			srvlist = srvlist->next;
   1427  1.103  christos 
   1428  1.103  christos 			/*
   1429  1.103  christos 			 * Since res_* doesn't give the additional
   1430  1.103  christos 			 * section, we always look up.
   1431  1.103  christos 			 */
   1432  1.103  christos 			memset(&q, 0, sizeof(q));
   1433  1.103  christos 			memset(&q2, 0, sizeof(q2));
   1434  1.103  christos 
   1435  1.103  christos 			q.name = srv->name;
   1436  1.103  christos 			q.qclass = C_IN;
   1437  1.103  christos 			q.qtype = T_AAAA;
   1438  1.103  christos 			q.next = &q2;
   1439  1.103  christos 			q2.name = srv->name;
   1440  1.103  christos 			q2.qclass = C_IN;
   1441  1.103  christos 			q2.qtype = T_A;
   1442  1.103  christos 
   1443  1.103  christos 			aip = _dns_query(&q, pai, res, 0);
   1444  1.103  christos 
   1445  1.103  christos 			if (aip != NULL) {
   1446  1.103  christos 				cur->ai_next = aip;
   1447  1.103  christos 				while (cur && cur->ai_next) {
   1448  1.103  christos 					cur = cur->ai_next;
   1449  1.103  christos 					*getport(cur) = htons(srv->port);
   1450  1.103  christos 					haveanswer++;
   1451  1.103  christos 				}
   1452  1.103  christos 			}
   1453  1.103  christos 			free(srv);
   1454  1.103  christos 		}
   1455  1.103  christos 		__res_put_state(res);
   1456  1.103  christos 	}
   1457   1.32    itojun 	if (haveanswer) {
   1458  1.103  christos 		if (!sentinel.ai_next->ai_canonname)
   1459  1.103  christos 		       (void)get_canonname(pai, sentinel.ai_next,
   1460  1.103  christos 			   canonname ? canonname : qname);
   1461   1.32    itojun 		h_errno = NETDB_SUCCESS;
   1462   1.32    itojun 		return sentinel.ai_next;
   1463   1.32    itojun 	}
   1464   1.32    itojun 
   1465   1.32    itojun 	h_errno = NO_RECOVERY;
   1466   1.32    itojun 	return NULL;
   1467   1.32    itojun }
   1468   1.32    itojun 
   1469   1.73    tsarna #define SORTEDADDR(p)	(((struct sockaddr_in *)(void *)(p->ai_next->ai_addr))->sin_addr.s_addr)
   1470   1.73    tsarna #define SORTMATCH(p, s) ((SORTEDADDR(p) & (s).mask) == (s).addr.s_addr)
   1471   1.73    tsarna 
   1472   1.73    tsarna static void
   1473   1.73    tsarna aisort(struct addrinfo *s, res_state res)
   1474   1.73    tsarna {
   1475   1.73    tsarna 	struct addrinfo head, *t, *p;
   1476   1.73    tsarna 	int i;
   1477   1.73    tsarna 
   1478   1.73    tsarna 	head.ai_next = NULL;
   1479   1.73    tsarna 	t = &head;
   1480   1.73    tsarna 
   1481   1.73    tsarna 	for (i = 0; i < res->nsort; i++) {
   1482   1.73    tsarna 		p = s;
   1483   1.73    tsarna 		while (p->ai_next) {
   1484   1.73    tsarna 			if ((p->ai_next->ai_family != AF_INET)
   1485   1.73    tsarna 			|| SORTMATCH(p, res->sort_list[i])) {
   1486   1.73    tsarna 				t->ai_next = p->ai_next;
   1487   1.73    tsarna 				t = t->ai_next;
   1488   1.73    tsarna 				p->ai_next = p->ai_next->ai_next;
   1489   1.73    tsarna 			} else {
   1490   1.73    tsarna 				p = p->ai_next;
   1491   1.73    tsarna 			}
   1492   1.73    tsarna 		}
   1493   1.73    tsarna 	}
   1494   1.73    tsarna 
   1495   1.73    tsarna 	/* add rest of list and reset s to the new list*/
   1496   1.73    tsarna 	t->ai_next = s->ai_next;
   1497   1.73    tsarna 	s->ai_next = head.ai_next;
   1498   1.73    tsarna }
   1499   1.73    tsarna 
   1500  1.103  christos static struct addrinfo *
   1501  1.103  christos _dns_query(struct res_target *q, const struct addrinfo *pai,
   1502  1.103  christos     res_state res, int dosearch)
   1503  1.103  christos {
   1504  1.103  christos 	struct res_target *q2 = q->next;
   1505  1.103  christos  	querybuf *buf, *buf2;
   1506  1.103  christos 	struct addrinfo sentinel, *cur, *ai;
   1507  1.103  christos 
   1508  1.103  christos #ifdef DNS_DEBUG
   1509  1.103  christos 	struct res_target *iter;
   1510  1.103  christos 	for (iter = q; iter; iter = iter->next)
   1511  1.103  christos 		printf("Query type %d for %s\n", iter->qtype, iter->name);
   1512  1.103  christos #endif
   1513  1.103  christos 
   1514  1.103  christos  	buf = malloc(sizeof(*buf));
   1515  1.103  christos  	if (buf == NULL) {
   1516  1.103  christos  		h_errno = NETDB_INTERNAL;
   1517  1.103  christos 		return NULL;
   1518  1.103  christos  	}
   1519  1.103  christos  	buf2 = malloc(sizeof(*buf2));
   1520  1.103  christos  	if (buf2 == NULL) {
   1521  1.103  christos  		free(buf);
   1522  1.103  christos  		h_errno = NETDB_INTERNAL;
   1523  1.103  christos 		return NULL;
   1524  1.103  christos 	}
   1525  1.103  christos 
   1526  1.103  christos 	memset(&sentinel, 0, sizeof(sentinel));
   1527  1.103  christos 	cur = &sentinel;
   1528  1.103  christos 
   1529  1.103  christos 	q->answer = buf->buf;
   1530  1.103  christos 	q->anslen = sizeof(buf->buf);
   1531  1.103  christos 	if (q2) {
   1532  1.103  christos 		q2->answer = buf2->buf;
   1533  1.103  christos 		q2->anslen = sizeof(buf2->buf);
   1534  1.103  christos 	}
   1535  1.103  christos 
   1536  1.103  christos 	if (dosearch) {
   1537  1.103  christos 		if (res_searchN(q->name, q, res) < 0)
   1538  1.103  christos 			goto out;
   1539  1.103  christos 	} else {
   1540  1.103  christos 		if (res_queryN(q->name, q, res) < 0)
   1541  1.103  christos 			goto out;
   1542  1.103  christos 	}
   1543  1.103  christos 
   1544  1.103  christos 	ai = getanswer(buf, q->n, q->name, q->qtype, pai);
   1545  1.103  christos 	if (ai) {
   1546  1.103  christos 		cur->ai_next = ai;
   1547  1.103  christos 		while (cur && cur->ai_next)
   1548  1.103  christos 			cur = cur->ai_next;
   1549  1.103  christos 	}
   1550  1.103  christos 	if (q2) {
   1551  1.103  christos 		ai = getanswer(buf2, q2->n, q2->name, q2->qtype, pai);
   1552  1.103  christos 		if (ai)
   1553  1.103  christos 			cur->ai_next = ai;
   1554  1.103  christos  	}
   1555  1.103  christos 	free(buf);
   1556  1.103  christos 	free(buf2);
   1557  1.103  christos 	return sentinel.ai_next;
   1558  1.103  christos out:
   1559  1.103  christos 	free(buf);
   1560  1.103  christos 	free(buf2);
   1561  1.103  christos 	return NULL;
   1562  1.103  christos }
   1563  1.103  christos 
   1564   1.32    itojun /*ARGSUSED*/
   1565  1.103  christos static struct addrinfo *
   1566  1.103  christos _dns_srv_lookup(const char *name, const char *servname,
   1567  1.103  christos     const struct addrinfo *pai)
   1568   1.32    itojun {
   1569  1.103  christos 	static const char * const srvprotos[] = { "tcp", "udp" };
   1570  1.103  christos 	static const int srvnottype[] = { SOCK_DGRAM, SOCK_STREAM };
   1571  1.103  christos 	static const int nsrvprotos = 2;
   1572  1.103  christos 	struct addrinfo sentinel, *cur, *ai;
   1573  1.103  christos 	struct servent *serv, sv;
   1574  1.103  christos 	struct servent_data svd;
   1575  1.103  christos 	struct res_target q;
   1576   1.73    tsarna 	res_state res;
   1577  1.103  christos 	char *tname;
   1578  1.103  christos 	int i;
   1579   1.32    itojun 
   1580  1.103  christos 	res = __res_get_state();
   1581  1.103  christos 	if (res == NULL)
   1582  1.103  christos 		return NULL;
   1583   1.32    itojun 
   1584  1.103  christos 	memset(&svd, 0, sizeof(svd));
   1585   1.32    itojun 	memset(&sentinel, 0, sizeof(sentinel));
   1586   1.32    itojun 	cur = &sentinel;
   1587   1.32    itojun 
   1588  1.103  christos 	/*
   1589  1.103  christos 	 * Iterate over supported SRV protocols.
   1590  1.103  christos 	 * (currently UDP and TCP only)
   1591  1.103  christos 	 */
   1592  1.103  christos 	for (i = 0; i < nsrvprotos; i++) {
   1593  1.103  christos 		/*
   1594  1.103  christos 		 * Check that the caller didn't specify a hint
   1595  1.103  christos 		 * which precludes this protocol.
   1596  1.103  christos 		 */
   1597  1.103  christos 		if (pai->ai_socktype == srvnottype[i])
   1598  1.103  christos 			continue;
   1599  1.103  christos 		/*
   1600  1.103  christos 		 * If the caller specified a port,
   1601  1.103  christos 		 * then lookup the database for the
   1602  1.103  christos 		 * official service name.
   1603  1.103  christos 		 */
   1604  1.103  christos 		serv = getservbyname_r(servname, srvprotos[i], &sv, &svd);
   1605  1.103  christos 		if (serv == NULL)
   1606  1.103  christos 			continue;
   1607  1.103  christos 
   1608  1.103  christos 		/*
   1609  1.103  christos 		 * Construct service DNS name.
   1610  1.103  christos 		 */
   1611  1.103  christos 		if (asprintf(&tname, "_%s._%s.%s", serv->s_name, serv->s_proto,
   1612  1.103  christos 		    name) < 0)
   1613  1.103  christos 			continue;
   1614  1.103  christos 
   1615  1.103  christos 		memset(&q, 0, sizeof(q));
   1616  1.103  christos 		q.name = tname;
   1617  1.103  christos 		q.qclass = C_IN;
   1618  1.103  christos 		q.qtype = T_SRV;
   1619  1.103  christos 
   1620  1.103  christos 		/*
   1621  1.103  christos 		 * Do SRV query.
   1622  1.103  christos 		 */
   1623  1.103  christos 		ai = _dns_query(&q, pai, res, 1);
   1624  1.103  christos 		if (ai) {
   1625  1.103  christos 			cur->ai_next = ai;
   1626  1.103  christos 			while (cur && cur->ai_next)
   1627  1.103  christos 				cur = cur->ai_next;
   1628  1.103  christos 		}
   1629  1.103  christos 		free(tname);
   1630   1.63    itojun 	}
   1631  1.103  christos 
   1632  1.103  christos 	if (res->nsort)
   1633  1.103  christos 		aisort(&sentinel, res);
   1634  1.103  christos 
   1635  1.103  christos 	__res_put_state(res);
   1636  1.103  christos 
   1637  1.103  christos 	return sentinel.ai_next;
   1638  1.103  christos }
   1639  1.103  christos 
   1640  1.103  christos /*ARGSUSED*/
   1641  1.103  christos static struct addrinfo *
   1642  1.103  christos _dns_host_lookup(const char *name, const struct addrinfo *pai)
   1643  1.103  christos {
   1644  1.103  christos 	struct res_target q, q2;
   1645  1.103  christos 	struct addrinfo sentinel, *ai;
   1646  1.103  christos 	res_state res;
   1647  1.103  christos 
   1648  1.103  christos 	res = __res_get_state();
   1649  1.103  christos 	if (res == NULL)
   1650  1.103  christos 		return NULL;
   1651  1.103  christos 
   1652  1.103  christos 	memset(&q, 0, sizeof(q2));
   1653  1.103  christos 	memset(&q2, 0, sizeof(q2));
   1654   1.63    itojun 
   1655   1.32    itojun 	switch (pai->ai_family) {
   1656   1.32    itojun 	case AF_UNSPEC:
   1657   1.32    itojun 		/* prefer IPv6 */
   1658   1.52    itojun 		q.name = name;
   1659   1.43    itojun 		q.qclass = C_IN;
   1660   1.43    itojun 		q.qtype = T_AAAA;
   1661   1.32    itojun 		q.next = &q2;
   1662   1.52    itojun 		q2.name = name;
   1663   1.43    itojun 		q2.qclass = C_IN;
   1664   1.43    itojun 		q2.qtype = T_A;
   1665   1.32    itojun 		break;
   1666   1.32    itojun 	case AF_INET:
   1667   1.52    itojun 		q.name = name;
   1668   1.43    itojun 		q.qclass = C_IN;
   1669   1.43    itojun 		q.qtype = T_A;
   1670   1.32    itojun 		break;
   1671   1.32    itojun 	case AF_INET6:
   1672   1.52    itojun 		q.name = name;
   1673   1.43    itojun 		q.qclass = C_IN;
   1674   1.43    itojun 		q.qtype = T_AAAA;
   1675   1.32    itojun 		break;
   1676   1.32    itojun 	default:
   1677  1.105  christos 		__res_put_state(res);
   1678  1.103  christos 		h_errno = NETDB_INTERNAL;
   1679  1.103  christos 		return NULL;
   1680   1.32    itojun 	}
   1681   1.73    tsarna 
   1682  1.103  christos 	ai = _dns_query(&q, pai, res, 1);
   1683   1.73    tsarna 
   1684  1.103  christos 	memset(&sentinel, 0, sizeof(sentinel));
   1685  1.103  christos 	sentinel.ai_next = ai;
   1686   1.73    tsarna 
   1687  1.103  christos 	if (ai != NULL && res->nsort)
   1688   1.73    tsarna 		aisort(&sentinel, res);
   1689   1.73    tsarna 
   1690   1.73    tsarna 	__res_put_state(res);
   1691   1.73    tsarna 
   1692  1.103  christos 	return sentinel.ai_next;
   1693  1.103  christos }
   1694  1.103  christos 
   1695  1.103  christos /*ARGSUSED*/
   1696  1.103  christos static int
   1697  1.103  christos _dns_getaddrinfo(void *rv, void *cb_data, va_list ap)
   1698  1.103  christos {
   1699  1.103  christos 	struct addrinfo *ai = NULL;
   1700  1.103  christos 	const char *name, *servname;
   1701  1.103  christos 	const struct addrinfo *pai;
   1702  1.103  christos 
   1703  1.103  christos 	name = va_arg(ap, char *);
   1704  1.103  christos 	pai = va_arg(ap, const struct addrinfo *);
   1705  1.103  christos 	servname = va_arg(ap, char *);
   1706  1.103  christos 
   1707  1.103  christos 	/*
   1708  1.103  christos 	 * Try doing SRV lookup on service first.
   1709  1.103  christos 	 */
   1710  1.103  christos 	if (servname
   1711  1.103  christos #ifdef AI_SRV
   1712  1.103  christos 	    && (pai->ai_flags & AI_SRV)
   1713  1.103  christos #endif
   1714  1.103  christos 	    && !(pai->ai_flags & AI_NUMERICSERV)
   1715  1.103  christos 	    && str2number(servname) == -1) {
   1716  1.103  christos 
   1717  1.103  christos #ifdef DNS_DEBUG
   1718  1.103  christos 		printf("%s: try SRV lookup\n", __func__);
   1719  1.103  christos #endif
   1720  1.103  christos 		ai = _dns_srv_lookup(name, servname, pai);
   1721  1.103  christos 	}
   1722  1.103  christos 
   1723  1.103  christos 	/*
   1724  1.103  christos 	 * Do lookup on name.
   1725  1.103  christos 	 */
   1726  1.103  christos 	if (ai == NULL) {
   1727  1.103  christos 
   1728  1.103  christos #ifdef DNS_DEBUG
   1729  1.103  christos 		printf("%s: try HOST lookup\n", __func__);
   1730  1.103  christos #endif
   1731  1.103  christos 		ai = _dns_host_lookup(name, pai);
   1732  1.103  christos 
   1733  1.103  christos 		if (ai == NULL) {
   1734  1.103  christos 			switch (h_errno) {
   1735  1.103  christos 			case HOST_NOT_FOUND:
   1736  1.103  christos 				return NS_NOTFOUND;
   1737  1.103  christos 			case TRY_AGAIN:
   1738  1.103  christos 				return NS_TRYAGAIN;
   1739  1.103  christos 			default:
   1740  1.103  christos 				return NS_UNAVAIL;
   1741  1.103  christos 			}
   1742  1.103  christos 		}
   1743  1.103  christos 	}
   1744  1.103  christos 
   1745  1.103  christos 	*((struct addrinfo **)rv) = ai;
   1746   1.32    itojun 	return NS_SUCCESS;
   1747   1.32    itojun }
   1748   1.32    itojun 
   1749   1.32    itojun static void
   1750   1.70  christos _sethtent(FILE **hostf)
   1751   1.32    itojun {
   1752   1.51     lukem 
   1753   1.70  christos 	if (!*hostf)
   1754   1.96  christos 		*hostf = fopen(_PATH_HOSTS, "re");
   1755   1.32    itojun 	else
   1756   1.70  christos 		rewind(*hostf);
   1757   1.32    itojun }
   1758   1.32    itojun 
   1759   1.32    itojun static void
   1760   1.70  christos _endhtent(FILE **hostf)
   1761   1.32    itojun {
   1762   1.51     lukem 
   1763   1.70  christos 	if (*hostf) {
   1764   1.70  christos 		(void) fclose(*hostf);
   1765   1.70  christos 		*hostf = NULL;
   1766   1.32    itojun 	}
   1767   1.32    itojun }
   1768   1.32    itojun 
   1769   1.32    itojun static struct addrinfo *
   1770   1.70  christos _gethtent(FILE **hostf, const char *name, const struct addrinfo *pai)
   1771   1.32    itojun {
   1772   1.32    itojun 	char *p;
   1773   1.48    itojun 	char *cp, *tname, *cname;
   1774   1.32    itojun 	struct addrinfo hints, *res0, *res;
   1775   1.32    itojun 	int error;
   1776   1.32    itojun 	const char *addr;
   1777   1.32    itojun 	char hostbuf[8*1024];
   1778   1.32    itojun 
   1779   1.51     lukem 	_DIAGASSERT(name != NULL);
   1780   1.51     lukem 	_DIAGASSERT(pai != NULL);
   1781   1.51     lukem 
   1782   1.96  christos 	if (!*hostf && !(*hostf = fopen(_PATH_HOSTS, "re")))
   1783   1.32    itojun 		return (NULL);
   1784   1.32    itojun  again:
   1785   1.97  christos 	if (!(p = fgets(hostbuf, (int)sizeof hostbuf, *hostf)))
   1786   1.32    itojun 		return (NULL);
   1787   1.32    itojun 	if (*p == '#')
   1788   1.32    itojun 		goto again;
   1789   1.32    itojun 	if (!(cp = strpbrk(p, "#\n")))
   1790   1.32    itojun 		goto again;
   1791   1.32    itojun 	*cp = '\0';
   1792   1.32    itojun 	if (!(cp = strpbrk(p, " \t")))
   1793   1.32    itojun 		goto again;
   1794   1.32    itojun 	*cp++ = '\0';
   1795   1.32    itojun 	addr = p;
   1796   1.32    itojun 	/* if this is not something we're looking for, skip it. */
   1797   1.48    itojun 	cname = NULL;
   1798   1.32    itojun 	while (cp && *cp) {
   1799   1.32    itojun 		if (*cp == ' ' || *cp == '\t') {
   1800   1.32    itojun 			cp++;
   1801   1.32    itojun 			continue;
   1802   1.32    itojun 		}
   1803   1.48    itojun 		if (!cname)
   1804   1.48    itojun 			cname = cp;
   1805   1.32    itojun 		tname = cp;
   1806   1.32    itojun 		if ((cp = strpbrk(cp, " \t")) != NULL)
   1807   1.32    itojun 			*cp++ = '\0';
   1808   1.32    itojun 		if (strcasecmp(name, tname) == 0)
   1809   1.32    itojun 			goto found;
   1810   1.32    itojun 	}
   1811   1.32    itojun 	goto again;
   1812   1.32    itojun 
   1813   1.32    itojun found:
   1814   1.32    itojun 	hints = *pai;
   1815   1.32    itojun 	hints.ai_flags = AI_NUMERICHOST;
   1816   1.32    itojun 	error = getaddrinfo(addr, NULL, &hints, &res0);
   1817   1.32    itojun 	if (error)
   1818   1.32    itojun 		goto again;
   1819   1.32    itojun 	for (res = res0; res; res = res->ai_next) {
   1820   1.32    itojun 		/* cover it up */
   1821   1.32    itojun 		res->ai_flags = pai->ai_flags;
   1822   1.32    itojun 
   1823   1.32    itojun 		if (pai->ai_flags & AI_CANONNAME) {
   1824   1.48    itojun 			if (get_canonname(pai, res, cname) != 0) {
   1825   1.32    itojun 				freeaddrinfo(res0);
   1826   1.32    itojun 				goto again;
   1827   1.32    itojun 			}
   1828   1.32    itojun 		}
   1829   1.32    itojun 	}
   1830   1.32    itojun 	return res0;
   1831   1.32    itojun }
   1832   1.32    itojun 
   1833   1.32    itojun /*ARGSUSED*/
   1834   1.32    itojun static int
   1835   1.70  christos _files_getaddrinfo(void *rv, void *cb_data, va_list ap)
   1836   1.32    itojun {
   1837   1.32    itojun 	const char *name;
   1838   1.32    itojun 	const struct addrinfo *pai;
   1839   1.32    itojun 	struct addrinfo sentinel, *cur;
   1840   1.32    itojun 	struct addrinfo *p;
   1841   1.70  christos #ifndef _REENTRANT
   1842   1.70  christos 	static
   1843   1.70  christos #endif
   1844   1.70  christos 	FILE *hostf = NULL;
   1845   1.32    itojun 
   1846   1.32    itojun 	name = va_arg(ap, char *);
   1847   1.91     lukem 	pai = va_arg(ap, const struct addrinfo *);
   1848   1.32    itojun 
   1849   1.32    itojun 	memset(&sentinel, 0, sizeof(sentinel));
   1850   1.32    itojun 	cur = &sentinel;
   1851   1.32    itojun 
   1852   1.70  christos 	_sethtent(&hostf);
   1853   1.70  christos 	while ((p = _gethtent(&hostf, name, pai)) != NULL) {
   1854   1.32    itojun 		cur->ai_next = p;
   1855   1.32    itojun 		while (cur && cur->ai_next)
   1856   1.32    itojun 			cur = cur->ai_next;
   1857   1.32    itojun 	}
   1858   1.70  christos 	_endhtent(&hostf);
   1859   1.32    itojun 
   1860   1.32    itojun 	*((struct addrinfo **)rv) = sentinel.ai_next;
   1861   1.32    itojun 	if (sentinel.ai_next == NULL)
   1862   1.32    itojun 		return NS_NOTFOUND;
   1863   1.32    itojun 	return NS_SUCCESS;
   1864   1.32    itojun }
   1865   1.32    itojun 
   1866   1.32    itojun #ifdef YP
   1867   1.32    itojun /*ARGSUSED*/
   1868   1.32    itojun static struct addrinfo *
   1869   1.70  christos _yphostent(char *line, const struct addrinfo *pai)
   1870   1.32    itojun {
   1871   1.32    itojun 	struct addrinfo sentinel, *cur;
   1872   1.32    itojun 	struct addrinfo hints, *res, *res0;
   1873   1.32    itojun 	int error;
   1874   1.51     lukem 	char *p;
   1875   1.32    itojun 	const char *addr, *canonname;
   1876   1.32    itojun 	char *nextline;
   1877   1.32    itojun 	char *cp;
   1878   1.32    itojun 
   1879   1.51     lukem 	_DIAGASSERT(line != NULL);
   1880   1.51     lukem 	_DIAGASSERT(pai != NULL);
   1881   1.51     lukem 
   1882   1.51     lukem 	p = line;
   1883   1.32    itojun 	addr = canonname = NULL;
   1884   1.32    itojun 
   1885   1.36  christos 	memset(&sentinel, 0, sizeof(sentinel));
   1886   1.36  christos 	cur = &sentinel;
   1887   1.36  christos 
   1888   1.32    itojun nextline:
   1889   1.32    itojun 	/* terminate line */
   1890   1.32    itojun 	cp = strchr(p, '\n');
   1891   1.32    itojun 	if (cp) {
   1892   1.32    itojun 		*cp++ = '\0';
   1893   1.32    itojun 		nextline = cp;
   1894   1.32    itojun 	} else
   1895   1.32    itojun 		nextline = NULL;
   1896   1.32    itojun 
   1897   1.32    itojun 	cp = strpbrk(p, " \t");
   1898   1.32    itojun 	if (cp == NULL) {
   1899   1.32    itojun 		if (canonname == NULL)
   1900   1.32    itojun 			return (NULL);
   1901   1.32    itojun 		else
   1902   1.32    itojun 			goto done;
   1903   1.32    itojun 	}
   1904   1.32    itojun 	*cp++ = '\0';
   1905   1.32    itojun 
   1906   1.32    itojun 	addr = p;
   1907   1.32    itojun 
   1908   1.32    itojun 	while (cp && *cp) {
   1909   1.32    itojun 		if (*cp == ' ' || *cp == '\t') {
   1910   1.32    itojun 			cp++;
   1911   1.32    itojun 			continue;
   1912   1.32    itojun 		}
   1913   1.32    itojun 		if (!canonname)
   1914   1.32    itojun 			canonname = cp;
   1915   1.32    itojun 		if ((cp = strpbrk(cp, " \t")) != NULL)
   1916   1.32    itojun 			*cp++ = '\0';
   1917   1.32    itojun 	}
   1918   1.32    itojun 
   1919   1.32    itojun 	hints = *pai;
   1920   1.32    itojun 	hints.ai_flags = AI_NUMERICHOST;
   1921   1.32    itojun 	error = getaddrinfo(addr, NULL, &hints, &res0);
   1922   1.32    itojun 	if (error == 0) {
   1923   1.32    itojun 		for (res = res0; res; res = res->ai_next) {
   1924   1.32    itojun 			/* cover it up */
   1925   1.32    itojun 			res->ai_flags = pai->ai_flags;
   1926   1.32    itojun 
   1927   1.32    itojun 			if (pai->ai_flags & AI_CANONNAME)
   1928   1.32    itojun 				(void)get_canonname(pai, res, canonname);
   1929   1.32    itojun 		}
   1930   1.37    itojun 	} else
   1931   1.37    itojun 		res0 = NULL;
   1932   1.32    itojun 	if (res0) {
   1933   1.32    itojun 		cur->ai_next = res0;
   1934   1.80  christos 		while (cur->ai_next)
   1935   1.32    itojun 			cur = cur->ai_next;
   1936   1.32    itojun 	}
   1937   1.32    itojun 
   1938   1.32    itojun 	if (nextline) {
   1939   1.32    itojun 		p = nextline;
   1940   1.32    itojun 		goto nextline;
   1941   1.32    itojun 	}
   1942   1.32    itojun 
   1943   1.32    itojun done:
   1944   1.32    itojun 	return sentinel.ai_next;
   1945   1.32    itojun }
   1946   1.32    itojun 
   1947   1.32    itojun /*ARGSUSED*/
   1948   1.32    itojun static int
   1949   1.70  christos _yp_getaddrinfo(void *rv, void *cb_data, va_list ap)
   1950   1.32    itojun {
   1951   1.32    itojun 	struct addrinfo sentinel, *cur;
   1952   1.32    itojun 	struct addrinfo *ai = NULL;
   1953   1.72  christos 	char *ypbuf;
   1954   1.72  christos 	int ypbuflen, r;
   1955   1.32    itojun 	const char *name;
   1956   1.32    itojun 	const struct addrinfo *pai;
   1957   1.72  christos 	char *ypdomain;
   1958   1.72  christos 
   1959   1.72  christos 	if (_yp_check(&ypdomain) == 0)
   1960   1.72  christos 		return NS_UNAVAIL;
   1961   1.32    itojun 
   1962   1.32    itojun 	name = va_arg(ap, char *);
   1963   1.32    itojun 	pai = va_arg(ap, const struct addrinfo *);
   1964   1.32    itojun 
   1965   1.32    itojun 	memset(&sentinel, 0, sizeof(sentinel));
   1966   1.32    itojun 	cur = &sentinel;
   1967   1.32    itojun 
   1968   1.32    itojun 	/* hosts.byname is only for IPv4 (Solaris8) */
   1969   1.33    itojun 	if (pai->ai_family == PF_UNSPEC || pai->ai_family == PF_INET) {
   1970   1.72  christos 		r = yp_match(ypdomain, "hosts.byname", name,
   1971   1.72  christos 			(int)strlen(name), &ypbuf, &ypbuflen);
   1972   1.33    itojun 		if (r == 0) {
   1973   1.33    itojun 			struct addrinfo ai4;
   1974   1.33    itojun 
   1975   1.33    itojun 			ai4 = *pai;
   1976   1.33    itojun 			ai4.ai_family = AF_INET;
   1977   1.72  christos 			ai = _yphostent(ypbuf, &ai4);
   1978   1.33    itojun 			if (ai) {
   1979   1.33    itojun 				cur->ai_next = ai;
   1980   1.33    itojun 				while (cur && cur->ai_next)
   1981   1.33    itojun 					cur = cur->ai_next;
   1982   1.33    itojun 			}
   1983   1.32    itojun 		}
   1984   1.72  christos 		free(ypbuf);
   1985   1.32    itojun 	}
   1986   1.32    itojun 
   1987   1.32    itojun 	/* ipnodes.byname can hold both IPv4/v6 */
   1988   1.72  christos 	r = yp_match(ypdomain, "ipnodes.byname", name,
   1989   1.72  christos 		(int)strlen(name), &ypbuf, &ypbuflen);
   1990   1.32    itojun 	if (r == 0) {
   1991   1.72  christos 		ai = _yphostent(ypbuf, pai);
   1992   1.72  christos 		if (ai)
   1993   1.32    itojun 			cur->ai_next = ai;
   1994   1.72  christos 		free(ypbuf);
   1995   1.32    itojun 	}
   1996   1.32    itojun 
   1997   1.32    itojun 	if (sentinel.ai_next == NULL) {
   1998   1.32    itojun 		h_errno = HOST_NOT_FOUND;
   1999   1.32    itojun 		return NS_NOTFOUND;
   2000   1.32    itojun 	}
   2001   1.32    itojun 	*((struct addrinfo **)rv) = sentinel.ai_next;
   2002   1.32    itojun 	return NS_SUCCESS;
   2003   1.32    itojun }
   2004   1.32    itojun #endif
   2005   1.32    itojun 
   2006   1.32    itojun /* resolver logic */
   2007   1.32    itojun 
   2008   1.32    itojun /*
   2009   1.32    itojun  * Formulate a normal query, send, and await answer.
   2010   1.32    itojun  * Returned answer is placed in supplied buffer "answer".
   2011   1.32    itojun  * Perform preliminary check of answer, returning success only
   2012   1.32    itojun  * if no error is indicated and the answer count is nonzero.
   2013   1.32    itojun  * Return the size of the response on success, -1 on error.
   2014   1.32    itojun  * Error number is left in h_errno.
   2015   1.32    itojun  *
   2016   1.32    itojun  * Caller must parse answer and determine whether it answers the question.
   2017   1.32    itojun  */
   2018   1.32    itojun static int
   2019   1.70  christos res_queryN(const char *name, /* domain name */ struct res_target *target,
   2020   1.70  christos     res_state res)
   2021   1.32    itojun {
   2022   1.32    itojun 	u_char buf[MAXPACKET];
   2023   1.32    itojun 	HEADER *hp;
   2024   1.32    itojun 	int n;
   2025   1.32    itojun 	struct res_target *t;
   2026   1.32    itojun 	int rcode;
   2027   1.32    itojun 	int ancount;
   2028   1.32    itojun 
   2029   1.51     lukem 	_DIAGASSERT(name != NULL);
   2030   1.51     lukem 	/* XXX: target may be NULL??? */
   2031   1.51     lukem 
   2032   1.32    itojun 	rcode = NOERROR;
   2033   1.32    itojun 	ancount = 0;
   2034   1.32    itojun 
   2035   1.32    itojun 	for (t = target; t; t = t->next) {
   2036   1.32    itojun 		int class, type;
   2037   1.32    itojun 		u_char *answer;
   2038   1.32    itojun 		int anslen;
   2039   1.32    itojun 
   2040   1.32    itojun 		hp = (HEADER *)(void *)t->answer;
   2041   1.32    itojun 		hp->rcode = NOERROR;	/* default */
   2042   1.32    itojun 
   2043   1.32    itojun 		/* make it easier... */
   2044   1.43    itojun 		class = t->qclass;
   2045   1.43    itojun 		type = t->qtype;
   2046   1.32    itojun 		answer = t->answer;
   2047   1.32    itojun 		anslen = t->anslen;
   2048   1.32    itojun #ifdef DEBUG
   2049   1.70  christos 		if (res->options & RES_DEBUG)
   2050   1.70  christos 			printf(";; res_nquery(%s, %d, %d)\n", name, class, type);
   2051   1.32    itojun #endif
   2052   1.32    itojun 
   2053   1.70  christos 		n = res_nmkquery(res, QUERY, name, class, type, NULL, 0, NULL,
   2054   1.97  christos 		    buf, (int)sizeof(buf));
   2055   1.47    itojun #ifdef RES_USE_EDNS0
   2056   1.70  christos 		if (n > 0 && (res->options & RES_USE_EDNS0) != 0)
   2057   1.97  christos 			n = res_nopt(res, n, buf, (int)sizeof(buf), anslen);
   2058   1.47    itojun #endif
   2059   1.32    itojun 		if (n <= 0) {
   2060   1.32    itojun #ifdef DEBUG
   2061   1.70  christos 			if (res->options & RES_DEBUG)
   2062   1.70  christos 				printf(";; res_nquery: mkquery failed\n");
   2063   1.32    itojun #endif
   2064   1.32    itojun 			h_errno = NO_RECOVERY;
   2065   1.70  christos 			return n;
   2066   1.32    itojun 		}
   2067   1.70  christos 		n = res_nsend(res, buf, n, answer, anslen);
   2068   1.35    itojun #if 0
   2069   1.32    itojun 		if (n < 0) {
   2070   1.32    itojun #ifdef DEBUG
   2071   1.70  christos 			if (res->options & RES_DEBUG)
   2072   1.32    itojun 				printf(";; res_query: send error\n");
   2073   1.32    itojun #endif
   2074   1.32    itojun 			h_errno = TRY_AGAIN;
   2075   1.70  christos 			return n;
   2076   1.32    itojun 		}
   2077   1.35    itojun #endif
   2078   1.32    itojun 
   2079   1.35    itojun 		if (n < 0 || hp->rcode != NOERROR || ntohs(hp->ancount) == 0) {
   2080   1.32    itojun 			rcode = hp->rcode;	/* record most recent error */
   2081   1.32    itojun #ifdef DEBUG
   2082   1.70  christos 			if (res->options & RES_DEBUG)
   2083   1.57    itojun 				printf(";; rcode = %u, ancount=%u\n", hp->rcode,
   2084   1.32    itojun 				    ntohs(hp->ancount));
   2085   1.32    itojun #endif
   2086   1.32    itojun 			continue;
   2087   1.32    itojun 		}
   2088   1.32    itojun 
   2089   1.32    itojun 		ancount += ntohs(hp->ancount);
   2090   1.32    itojun 
   2091   1.32    itojun 		t->n = n;
   2092   1.32    itojun 	}
   2093   1.32    itojun 
   2094   1.32    itojun 	if (ancount == 0) {
   2095   1.32    itojun 		switch (rcode) {
   2096   1.32    itojun 		case NXDOMAIN:
   2097   1.32    itojun 			h_errno = HOST_NOT_FOUND;
   2098   1.32    itojun 			break;
   2099   1.32    itojun 		case SERVFAIL:
   2100   1.32    itojun 			h_errno = TRY_AGAIN;
   2101   1.32    itojun 			break;
   2102   1.32    itojun 		case NOERROR:
   2103   1.32    itojun 			h_errno = NO_DATA;
   2104   1.32    itojun 			break;
   2105   1.32    itojun 		case FORMERR:
   2106   1.32    itojun 		case NOTIMP:
   2107   1.32    itojun 		case REFUSED:
   2108   1.32    itojun 		default:
   2109   1.32    itojun 			h_errno = NO_RECOVERY;
   2110   1.32    itojun 			break;
   2111   1.32    itojun 		}
   2112   1.70  christos 		return -1;
   2113   1.32    itojun 	}
   2114   1.70  christos 	return ancount;
   2115   1.32    itojun }
   2116   1.32    itojun 
   2117   1.32    itojun /*
   2118   1.32    itojun  * Formulate a normal query, send, and retrieve answer in supplied buffer.
   2119   1.32    itojun  * Return the size of the response on success, -1 on error.
   2120   1.32    itojun  * If enabled, implement search rules until answer or unrecoverable failure
   2121  1.102  christos  * is detected.	 Error code, if any, is left in h_errno.
   2122   1.32    itojun  */
   2123   1.32    itojun static int
   2124   1.73    tsarna res_searchN(const char *name, struct res_target *target, res_state res)
   2125   1.32    itojun {
   2126   1.32    itojun 	const char *cp, * const *domain;
   2127   1.51     lukem 	HEADER *hp;
   2128   1.32    itojun 	u_int dots;
   2129  1.106  christos 	char buf[MAXHOSTNAMELEN];
   2130   1.32    itojun 	int trailing_dot, ret, saved_herrno;
   2131   1.32    itojun 	int got_nodata = 0, got_servfail = 0, tried_as_is = 0;
   2132   1.71  christos 
   2133   1.51     lukem 	_DIAGASSERT(name != NULL);
   2134   1.51     lukem 	_DIAGASSERT(target != NULL);
   2135   1.51     lukem 
   2136   1.51     lukem 	hp = (HEADER *)(void *)target->answer;	/*XXX*/
   2137   1.51     lukem 
   2138   1.32    itojun 	errno = 0;
   2139   1.32    itojun 	h_errno = HOST_NOT_FOUND;	/* default, if we never query */
   2140   1.32    itojun 	dots = 0;
   2141   1.32    itojun 	for (cp = name; *cp; cp++)
   2142   1.32    itojun 		dots += (*cp == '.');
   2143   1.32    itojun 	trailing_dot = 0;
   2144   1.32    itojun 	if (cp > name && *--cp == '.')
   2145   1.32    itojun 		trailing_dot++;
   2146   1.32    itojun 
   2147   1.32    itojun 	/*
   2148   1.95       wiz 	 * if there aren't any dots, it could be a user-level alias
   2149   1.32    itojun 	 */
   2150  1.106  christos 	if (!dots && (cp = res_hostalias(res, name, buf, sizeof(buf))) != NULL) {
   2151   1.70  christos 		ret = res_queryN(cp, target, res);
   2152   1.70  christos 		return ret;
   2153   1.70  christos 	}
   2154   1.32    itojun 
   2155   1.32    itojun 	/*
   2156   1.32    itojun 	 * If there are dots in the name already, let's just give it a try
   2157   1.32    itojun 	 * 'as is'.  The threshold can be set with the "ndots" option.
   2158   1.32    itojun 	 */
   2159   1.32    itojun 	saved_herrno = -1;
   2160   1.70  christos 	if (dots >= res->ndots) {
   2161   1.70  christos 		ret = res_querydomainN(name, NULL, target, res);
   2162   1.73    tsarna 		if (ret > 0)
   2163   1.32    itojun 			return (ret);
   2164   1.32    itojun 		saved_herrno = h_errno;
   2165   1.32    itojun 		tried_as_is++;
   2166   1.32    itojun 	}
   2167   1.32    itojun 
   2168   1.32    itojun 	/*
   2169   1.32    itojun 	 * We do at least one level of search if
   2170   1.32    itojun 	 *	- there is no dot and RES_DEFNAME is set, or
   2171   1.32    itojun 	 *	- there is at least one dot, there is no trailing dot,
   2172   1.32    itojun 	 *	  and RES_DNSRCH is set.
   2173   1.32    itojun 	 */
   2174   1.70  christos 	if ((!dots && (res->options & RES_DEFNAMES)) ||
   2175   1.70  christos 	    (dots && !trailing_dot && (res->options & RES_DNSRCH))) {
   2176   1.32    itojun 		int done = 0;
   2177   1.32    itojun 
   2178   1.70  christos 		for (domain = (const char * const *)res->dnsrch;
   2179   1.32    itojun 		   *domain && !done;
   2180   1.32    itojun 		   domain++) {
   2181   1.32    itojun 
   2182   1.70  christos 			ret = res_querydomainN(name, *domain, target, res);
   2183   1.73    tsarna 			if (ret > 0)
   2184   1.70  christos 				return ret;
   2185   1.32    itojun 
   2186   1.32    itojun 			/*
   2187   1.32    itojun 			 * If no server present, give up.
   2188   1.32    itojun 			 * If name isn't found in this domain,
   2189   1.32    itojun 			 * keep trying higher domains in the search list
   2190   1.32    itojun 			 * (if that's enabled).
   2191   1.32    itojun 			 * On a NO_DATA error, keep trying, otherwise
   2192   1.32    itojun 			 * a wildcard entry of another type could keep us
   2193   1.32    itojun 			 * from finding this entry higher in the domain.
   2194   1.32    itojun 			 * If we get some other error (negative answer or
   2195   1.32    itojun 			 * server failure), then stop searching up,
   2196   1.32    itojun 			 * but try the input name below in case it's
   2197   1.32    itojun 			 * fully-qualified.
   2198   1.32    itojun 			 */
   2199   1.32    itojun 			if (errno == ECONNREFUSED) {
   2200   1.32    itojun 				h_errno = TRY_AGAIN;
   2201   1.70  christos 				return -1;
   2202   1.32    itojun 			}
   2203   1.32    itojun 
   2204   1.32    itojun 			switch (h_errno) {
   2205   1.32    itojun 			case NO_DATA:
   2206   1.32    itojun 				got_nodata++;
   2207   1.32    itojun 				/* FALLTHROUGH */
   2208   1.32    itojun 			case HOST_NOT_FOUND:
   2209   1.32    itojun 				/* keep trying */
   2210   1.32    itojun 				break;
   2211   1.32    itojun 			case TRY_AGAIN:
   2212   1.32    itojun 				if (hp->rcode == SERVFAIL) {
   2213   1.32    itojun 					/* try next search element, if any */
   2214   1.32    itojun 					got_servfail++;
   2215   1.32    itojun 					break;
   2216   1.32    itojun 				}
   2217   1.32    itojun 				/* FALLTHROUGH */
   2218   1.32    itojun 			default:
   2219   1.32    itojun 				/* anything else implies that we're done */
   2220   1.32    itojun 				done++;
   2221   1.32    itojun 			}
   2222   1.32    itojun 			/*
   2223   1.32    itojun 			 * if we got here for some reason other than DNSRCH,
   2224   1.32    itojun 			 * we only wanted one iteration of the loop, so stop.
   2225   1.32    itojun 			 */
   2226   1.70  christos 			if (!(res->options & RES_DNSRCH))
   2227  1.102  christos 				done++;
   2228   1.32    itojun 		}
   2229   1.32    itojun 	}
   2230   1.32    itojun 
   2231   1.32    itojun 	/*
   2232   1.32    itojun 	 * if we have not already tried the name "as is", do that now.
   2233   1.32    itojun 	 * note that we do this regardless of how many dots were in the
   2234   1.32    itojun 	 * name or whether it ends with a dot.
   2235   1.32    itojun 	 */
   2236   1.32    itojun 	if (!tried_as_is) {
   2237   1.70  christos 		ret = res_querydomainN(name, NULL, target, res);
   2238   1.73    tsarna 		if (ret > 0)
   2239   1.70  christos 			return ret;
   2240   1.32    itojun 	}
   2241   1.32    itojun 
   2242   1.32    itojun 	/*
   2243   1.32    itojun 	 * if we got here, we didn't satisfy the search.
   2244   1.32    itojun 	 * if we did an initial full query, return that query's h_errno
   2245   1.32    itojun 	 * (note that we wouldn't be here if that query had succeeded).
   2246   1.32    itojun 	 * else if we ever got a nodata, send that back as the reason.
   2247   1.32    itojun 	 * else send back meaningless h_errno, that being the one from
   2248   1.32    itojun 	 * the last DNSRCH we did.
   2249   1.32    itojun 	 */
   2250   1.32    itojun 	if (saved_herrno != -1)
   2251   1.32    itojun 		h_errno = saved_herrno;
   2252   1.32    itojun 	else if (got_nodata)
   2253   1.32    itojun 		h_errno = NO_DATA;
   2254   1.32    itojun 	else if (got_servfail)
   2255   1.32    itojun 		h_errno = TRY_AGAIN;
   2256   1.70  christos 	return -1;
   2257   1.32    itojun }
   2258   1.32    itojun 
   2259   1.32    itojun /*
   2260   1.32    itojun  * Perform a call on res_query on the concatenation of name and domain,
   2261   1.32    itojun  * removing a trailing dot from name if domain is NULL.
   2262   1.32    itojun  */
   2263   1.32    itojun static int
   2264   1.73    tsarna res_querydomainN(const char *name, const char *domain,
   2265   1.70  christos     struct res_target *target, res_state res)
   2266   1.32    itojun {
   2267   1.32    itojun 	char nbuf[MAXDNAME];
   2268   1.32    itojun 	const char *longname = nbuf;
   2269   1.32    itojun 	size_t n, d;
   2270   1.51     lukem 
   2271   1.51     lukem 	_DIAGASSERT(name != NULL);
   2272   1.51     lukem 	/* XXX: target may be NULL??? */
   2273   1.32    itojun 
   2274   1.32    itojun #ifdef DEBUG
   2275   1.70  christos 	if (res->options & RES_DEBUG)
   2276   1.32    itojun 		printf(";; res_querydomain(%s, %s)\n",
   2277   1.32    itojun 			name, domain?domain:"<Nil>");
   2278   1.32    itojun #endif
   2279   1.32    itojun 	if (domain == NULL) {
   2280   1.32    itojun 		/*
   2281   1.32    itojun 		 * Check for trailing '.';
   2282   1.32    itojun 		 * copy without '.' if present.
   2283   1.32    itojun 		 */
   2284   1.32    itojun 		n = strlen(name);
   2285   1.63    itojun 		if (n + 1 > sizeof(nbuf)) {
   2286   1.32    itojun 			h_errno = NO_RECOVERY;
   2287   1.70  christos 			return -1;
   2288   1.32    itojun 		}
   2289   1.40    itojun 		if (n > 0 && name[--n] == '.') {
   2290   1.32    itojun 			strncpy(nbuf, name, n);
   2291   1.32    itojun 			nbuf[n] = '\0';
   2292   1.32    itojun 		} else
   2293   1.32    itojun 			longname = name;
   2294   1.32    itojun 	} else {
   2295   1.32    itojun 		n = strlen(name);
   2296   1.32    itojun 		d = strlen(domain);
   2297   1.63    itojun 		if (n + 1 + d + 1 > sizeof(nbuf)) {
   2298   1.32    itojun 			h_errno = NO_RECOVERY;
   2299   1.70  christos 			return -1;
   2300   1.32    itojun 		}
   2301   1.54    itojun 		snprintf(nbuf, sizeof(nbuf), "%s.%s", name, domain);
   2302   1.32    itojun 	}
   2303   1.70  christos 	return res_queryN(longname, target, res);
   2304   1.32    itojun }
   2305  1.103  christos 
   2306  1.103  christos #ifdef TEST
   2307  1.103  christos int
   2308  1.103  christos main(int argc, char *argv[]) {
   2309  1.103  christos 	struct addrinfo *ai, *sai;
   2310  1.103  christos 	int i, e;
   2311  1.103  christos 	char buf[1024];
   2312  1.103  christos 
   2313  1.103  christos 	for (i = 1; i < argc; i++) {
   2314  1.103  christos 		if ((e = getaddrinfo(argv[i], NULL, NULL, &sai)) != 0)
   2315  1.103  christos 			warnx("%s: %s", argv[i], gai_strerror(e));
   2316  1.103  christos 		for (ai = sai; ai; ai = ai->ai_next) {
   2317  1.103  christos 			sockaddr_snprintf(buf, sizeof(buf), "%a", ai->ai_addr);
   2318  1.103  christos              		printf("flags=0x%x family=%d socktype=%d protocol=%d "
   2319  1.103  christos 			    "addrlen=%zu addr=%s canonname=%s next=%p\n",
   2320  1.103  christos 			    ai->ai_flags,
   2321  1.103  christos              		    ai->ai_family,
   2322  1.103  christos              		    ai->ai_socktype,
   2323  1.103  christos              		    ai->ai_protocol,
   2324  1.103  christos              		    (size_t)ai->ai_addrlen,
   2325  1.103  christos 			    buf,
   2326  1.103  christos 			    ai->ai_canonname,
   2327  1.103  christos 			    ai->ai_next);
   2328  1.103  christos 		}
   2329  1.103  christos 		if (sai)
   2330  1.103  christos 			freeaddrinfo(sai);
   2331  1.103  christos 	}
   2332  1.103  christos 	return 0;
   2333  1.103  christos }
   2334  1.103  christos #endif
   2335