]> CyberLeo.Net >> Repos - FreeBSD/stable/10.git/blob - lib/libc/resolv/res_init.c
MFC r301592
[FreeBSD/stable/10.git] / lib / libc / resolv / res_init.c
1 /*
2  * Copyright (c) 1985, 1989, 1993
3  *    The Regents of the University of California.  All rights reserved.
4  * 
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 4. Neither the name of the University nor the names of its contributors
14  *    may be used to endorse or promote products derived from this software
15  *    without specific prior written permission.
16  * 
17  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  */
29
30 /*
31  * Portions Copyright (c) 1993 by Digital Equipment Corporation.
32  * 
33  * Permission to use, copy, modify, and distribute this software for any
34  * purpose with or without fee is hereby granted, provided that the above
35  * copyright notice and this permission notice appear in all copies, and that
36  * the name of Digital Equipment Corporation not be used in advertising or
37  * publicity pertaining to distribution of the document or software without
38  * specific, written prior permission.
39  * 
40  * THE SOFTWARE IS PROVIDED "AS IS" AND DIGITAL EQUIPMENT CORP. DISCLAIMS ALL
41  * WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES
42  * OF MERCHANTABILITY AND FITNESS.   IN NO EVENT SHALL DIGITAL EQUIPMENT
43  * CORPORATION BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
44  * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
45  * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
46  * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
47  * SOFTWARE.
48  */
49
50 /*
51  * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
52  * Portions Copyright (c) 1996-1999 by Internet Software Consortium.
53  *
54  * Permission to use, copy, modify, and distribute this software for any
55  * purpose with or without fee is hereby granted, provided that the above
56  * copyright notice and this permission notice appear in all copies.
57  *
58  * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
59  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
60  * MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR
61  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
62  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
63  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
64  * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
65  */
66
67 #if defined(LIBC_SCCS) && !defined(lint)
68 static const char sccsid[] = "@(#)res_init.c    8.1 (Berkeley) 6/7/93";
69 static const char rcsid[] = "$Id: res_init.c,v 1.26 2008/12/11 09:59:00 marka Exp $";
70 #endif /* LIBC_SCCS and not lint */
71 #include <sys/cdefs.h>
72 __FBSDID("$FreeBSD$");
73
74 #include "port_before.h"
75
76 #include "namespace.h"
77
78 #include <sys/types.h>
79 #include <sys/param.h>
80 #include <sys/socket.h>
81 #include <sys/stat.h>
82 #include <sys/time.h>
83
84 #include <netinet/in.h>
85 #include <arpa/inet.h>
86 #include <arpa/nameser.h>
87
88 #include <ctype.h>
89 #include <stdio.h>
90 #include <stdlib.h>
91 #include <string.h>
92 #include <unistd.h>
93 #include <netdb.h>
94
95 #ifndef HAVE_MD5
96 # include "../dst/md5.h"
97 #else
98 # ifdef SOLARIS2
99 #  include <sys/md5.h>
100 # elif _LIBC
101 # include <md5.h>
102 # endif
103 #endif
104 #ifndef _MD5_H_
105 # define _MD5_H_ 1      /*%< make sure we do not include rsaref md5.h file */
106 #endif
107
108 #include "un-namespace.h"
109
110 #include "port_after.h"
111
112 /* ensure that sockaddr_in6 and IN6ADDR_ANY_INIT are declared / defined */
113 #include <resolv.h>
114
115 #include "res_private.h"
116
117 /*% Options.  Should all be left alone. */
118 #define RESOLVSORT
119 #ifndef DEBUG
120 #define DEBUG
121 #endif
122
123 #ifdef SOLARIS2
124 #include <sys/systeminfo.h>
125 #endif
126
127 static void res_setoptions(res_state, const char *, const char *);
128
129 #ifdef RESOLVSORT
130 static const char sort_mask[] = "/&";
131 #define ISSORTMASK(ch) (strchr(sort_mask, ch) != NULL)
132 static u_int32_t net_mask(struct in_addr);
133 #endif
134
135 #if !defined(isascii)   /*%< XXX - could be a function */
136 # define isascii(c) (!(c & 0200))
137 #endif
138
139 /*
140  * Resolver state default settings.
141  */
142
143 /*%
144  * Set up default settings.  If the configuration file exist, the values
145  * there will have precedence.  Otherwise, the server address is set to
146  * INADDR_ANY and the default domain name comes from the gethostname().
147  *
148  * An interrim version of this code (BIND 4.9, pre-4.4BSD) used 127.0.0.1
149  * rather than INADDR_ANY ("0.0.0.0") as the default name server address
150  * since it was noted that INADDR_ANY actually meant ``the first interface
151  * you "ifconfig"'d at boot time'' and if this was a SLIP or PPP interface,
152  * it had to be "up" in order for you to reach your own name server.  It
153  * was later decided that since the recommended practice is to always 
154  * install local static routes through 127.0.0.1 for all your network
155  * interfaces, that we could solve this problem without a code change.
156  *
157  * The configuration file should always be used, since it is the only way
158  * to specify a default domain.  If you are running a server on your local
159  * machine, you should say "nameserver 0.0.0.0" or "nameserver 127.0.0.1"
160  * in the configuration file.
161  *
162  * Return 0 if completes successfully, -1 on error
163  */
164 int
165 res_ninit(res_state statp) {
166         extern int __res_vinit(res_state, int);
167
168         return (__res_vinit(statp, 0));
169 }
170
171 /*% This function has to be reachable by res_data.c but not publically. */
172 int
173 __res_vinit(res_state statp, int preinit) {
174         FILE *fp;
175         char *cp, **pp;
176         int n;
177         char buf[BUFSIZ];
178         int nserv = 0;    /*%< number of nameserver records read from file */
179         int haveenv = 0;
180         int havesearch = 0;
181 #ifdef RESOLVSORT
182         int nsort = 0;
183         char *net;
184 #endif
185         int dots;
186         union res_sockaddr_union u[2];
187         int maxns = MAXNS;
188
189         RES_SET_H_ERRNO(statp, 0);
190         if (statp->_u._ext.ext != NULL)
191                 res_ndestroy(statp);
192
193         if (!preinit) {
194                 statp->retrans = RES_TIMEOUT;
195                 statp->retry = RES_DFLRETRY;
196                 statp->options = RES_DEFAULT;
197         }
198
199         statp->_rnd = malloc(16);
200         res_rndinit(statp);
201         statp->id = res_nrandomid(statp);
202
203         memset(u, 0, sizeof(u));
204 #ifdef USELOOPBACK
205         u[nserv].sin.sin_addr = inet_makeaddr(IN_LOOPBACKNET, 1);
206 #else
207         u[nserv].sin.sin_addr.s_addr = INADDR_ANY;
208 #endif
209         u[nserv].sin.sin_family = AF_INET;
210         u[nserv].sin.sin_port = htons(NAMESERVER_PORT);
211 #ifdef HAVE_SA_LEN
212         u[nserv].sin.sin_len = sizeof(struct sockaddr_in);
213 #endif
214         nserv++;
215 #ifdef HAS_INET6_STRUCTS
216 #ifdef USELOOPBACK
217         u[nserv].sin6.sin6_addr = in6addr_loopback;
218 #else
219         u[nserv].sin6.sin6_addr = in6addr_any;
220 #endif
221         u[nserv].sin6.sin6_family = AF_INET6;
222         u[nserv].sin6.sin6_port = htons(NAMESERVER_PORT);
223 #ifdef HAVE_SA_LEN
224         u[nserv].sin6.sin6_len = sizeof(struct sockaddr_in6);
225 #endif
226         nserv++;
227 #endif
228         statp->nscount = 0;
229         statp->ndots = 1;
230         statp->pfcode = 0;
231         statp->_vcsock = -1;
232         statp->_flags = 0;
233         statp->qhook = NULL;
234         statp->rhook = NULL;
235         statp->_u._ext.nscount = 0;
236         statp->_u._ext.ext = malloc(sizeof(*statp->_u._ext.ext));
237         if (statp->_u._ext.ext != NULL) {
238                 memset(statp->_u._ext.ext, 0, sizeof(*statp->_u._ext.ext));
239                 statp->_u._ext.ext->nsaddrs[0].sin = statp->nsaddr;
240                 strcpy(statp->_u._ext.ext->nsuffix, "ip6.arpa");
241                 strcpy(statp->_u._ext.ext->nsuffix2, "ip6.int");
242                 statp->_u._ext.ext->reload_period = 2;
243         } else {
244                 /*
245                  * Historically res_init() rarely, if at all, failed.
246                  * Examples and applications exist which do not check
247                  * our return code.  Furthermore several applications
248                  * simply call us to get the systems domainname.  So
249                  * rather then immediately fail here we store the
250                  * failure, which is returned later, in h_errno.  And
251                  * prevent the collection of 'nameserver' information
252                  * by setting maxns to 0.  Thus applications that fail
253                  * to check our return code wont be able to make
254                  * queries anyhow.
255                  */
256                 RES_SET_H_ERRNO(statp, NETDB_INTERNAL);
257                 maxns = 0;
258         }
259 #ifdef RESOLVSORT
260         statp->nsort = 0;
261 #endif
262         res_setservers(statp, u, nserv);
263
264 #ifdef  SOLARIS2
265         /*
266          * The old libresolv derived the defaultdomain from NIS/NIS+.
267          * We want to keep this behaviour
268          */
269         {
270                 char buf[sizeof(statp->defdname)], *cp;
271                 int ret;
272
273                 if ((ret = sysinfo(SI_SRPC_DOMAIN, buf, sizeof(buf))) > 0 &&
274                         (unsigned int)ret <= sizeof(buf)) {
275                         if (buf[0] == '+')
276                                 buf[0] = '.';
277                         cp = strchr(buf, '.');
278                         cp = (cp == NULL) ? buf : (cp + 1);
279                         strncpy(statp->defdname, cp,
280                                 sizeof(statp->defdname) - 1);
281                         statp->defdname[sizeof(statp->defdname) - 1] = '\0';
282                 }
283         }
284 #endif  /* SOLARIS2 */
285
286         /* Allow user to override the local domain definition */
287         if (issetugid() == 0 && (cp = getenv("LOCALDOMAIN")) != NULL) {
288                 (void)strncpy(statp->defdname, cp, sizeof(statp->defdname) - 1);
289                 statp->defdname[sizeof(statp->defdname) - 1] = '\0';
290                 haveenv++;
291
292                 /*
293                  * Set search list to be blank-separated strings
294                  * from rest of env value.  Permits users of LOCALDOMAIN
295                  * to still have a search list, and anyone to set the
296                  * one that they want to use as an individual (even more
297                  * important now that the rfc1535 stuff restricts searches)
298                  */
299                 cp = statp->defdname;
300                 pp = statp->dnsrch;
301                 *pp++ = cp;
302                 for (n = 0; *cp && pp < statp->dnsrch + MAXDNSRCH; cp++) {
303                         if (*cp == '\n')        /*%< silly backwards compat */
304                                 break;
305                         else if (*cp == ' ' || *cp == '\t') {
306                                 *cp = 0;
307                                 n = 1;
308                         } else if (n) {
309                                 *pp++ = cp;
310                                 n = 0;
311                                 havesearch = 1;
312                         }
313                 }
314                 /* null terminate last domain if there are excess */
315                 while (*cp != '\0' && *cp != ' ' && *cp != '\t' && *cp != '\n')
316                         cp++;
317                 *cp = '\0';
318                 *pp++ = 0;
319         }
320
321 #define MATCH(line, name) \
322         (!strncmp(line, name, sizeof(name) - 1) && \
323         (line[sizeof(name) - 1] == ' ' || \
324          line[sizeof(name) - 1] == '\t'))
325
326         nserv = 0;
327         if ((fp = fopen(_PATH_RESCONF, "re")) != NULL) {
328             struct stat sb;
329             struct timespec now;
330
331             if (statp->_u._ext.ext != NULL) {
332                 if (_fstat(fileno(fp), &sb) == 0) {
333                     statp->_u._ext.ext->conf_mtim = sb.st_mtim;
334                     if (clock_gettime(CLOCK_MONOTONIC_FAST, &now) == 0) {
335                         statp->_u._ext.ext->conf_stat = now.tv_sec;
336                     }
337                 }
338             }
339
340             /* read the config file */
341             while (fgets(buf, sizeof(buf), fp) != NULL) {
342                 /* skip comments */
343                 if (*buf == ';' || *buf == '#')
344                         continue;
345                 /* read default domain name */
346                 if (MATCH(buf, "domain")) {
347                     if (haveenv)        /*%< skip if have from environ */
348                             continue;
349                     cp = buf + sizeof("domain") - 1;
350                     while (*cp == ' ' || *cp == '\t')
351                             cp++;
352                     if ((*cp == '\0') || (*cp == '\n'))
353                             continue;
354                     strncpy(statp->defdname, cp, sizeof(statp->defdname) - 1);
355                     statp->defdname[sizeof(statp->defdname) - 1] = '\0';
356                     if ((cp = strpbrk(statp->defdname, " \t\n")) != NULL)
357                             *cp = '\0';
358                     havesearch = 0;
359                     continue;
360                 }
361                 /* set search list */
362                 if (MATCH(buf, "search")) {
363                     if (haveenv)        /*%< skip if have from environ */
364                             continue;
365                     cp = buf + sizeof("search") - 1;
366                     while (*cp == ' ' || *cp == '\t')
367                             cp++;
368                     if ((*cp == '\0') || (*cp == '\n'))
369                             continue;
370                     strncpy(statp->defdname, cp, sizeof(statp->defdname) - 1);
371                     statp->defdname[sizeof(statp->defdname) - 1] = '\0';
372                     if ((cp = strchr(statp->defdname, '\n')) != NULL)
373                             *cp = '\0';
374                     /*
375                      * Set search list to be blank-separated strings
376                      * on rest of line.
377                      */
378                     cp = statp->defdname;
379                     pp = statp->dnsrch;
380                     *pp++ = cp;
381                     for (n = 0; *cp && pp < statp->dnsrch + MAXDNSRCH; cp++) {
382                             if (*cp == ' ' || *cp == '\t') {
383                                     *cp = 0;
384                                     n = 1;
385                             } else if (n) {
386                                     *pp++ = cp;
387                                     n = 0;
388                             }
389                     }
390                     /* null terminate last domain if there are excess */
391                     while (*cp != '\0' && *cp != ' ' && *cp != '\t')
392                             cp++;
393                     *cp = '\0';
394                     *pp++ = 0;
395                     havesearch = 1;
396                     continue;
397                 }
398                 /* read nameservers to query */
399                 if (MATCH(buf, "nameserver") && nserv < maxns) {
400                     struct addrinfo hints, *ai;
401                     char sbuf[NI_MAXSERV];
402                     const size_t minsiz =
403                         sizeof(statp->_u._ext.ext->nsaddrs[0]);
404
405                     cp = buf + sizeof("nameserver") - 1;
406                     while (*cp == ' ' || *cp == '\t')
407                         cp++;
408                     cp[strcspn(cp, ";# \t\n")] = '\0';
409                     if ((*cp != '\0') && (*cp != '\n')) {
410                         memset(&hints, 0, sizeof(hints));
411                         hints.ai_family = PF_UNSPEC;
412                         hints.ai_socktype = SOCK_DGRAM; /*dummy*/
413                         hints.ai_flags = AI_NUMERICHOST;
414                         sprintf(sbuf, "%u", NAMESERVER_PORT);
415                         if (getaddrinfo(cp, sbuf, &hints, &ai) == 0) {
416                             if (ai->ai_addrlen <= minsiz) {
417                                 if (statp->_u._ext.ext != NULL) {
418                                     memcpy(&statp->_u._ext.ext->nsaddrs[nserv],
419                                         ai->ai_addr, ai->ai_addrlen);
420                                 }
421                                 if (ai->ai_addrlen <=
422                                     sizeof(statp->nsaddr_list[nserv])) {
423                                     memcpy(&statp->nsaddr_list[nserv],
424                                         ai->ai_addr, ai->ai_addrlen);
425                                 } else
426                                     statp->nsaddr_list[nserv].sin_family = 0;
427                                 nserv++;
428                             }
429                             freeaddrinfo(ai);
430                         }
431                     }
432                     continue;
433                 }
434 #ifdef RESOLVSORT
435                 if (MATCH(buf, "sortlist")) {
436                     struct in_addr a;
437                     struct in6_addr a6;
438                     int m, i;
439                     u_char *u;
440                     struct __res_state_ext *ext = statp->_u._ext.ext;
441
442                     cp = buf + sizeof("sortlist") - 1;
443                     while (nsort < MAXRESOLVSORT) {
444                         while (*cp == ' ' || *cp == '\t')
445                             cp++;
446                         if (*cp == '\0' || *cp == '\n' || *cp == ';')
447                             break;
448                         net = cp;
449                         while (*cp && !ISSORTMASK(*cp) && *cp != ';' &&
450                                isascii(*cp) && !isspace((unsigned char)*cp))
451                                 cp++;
452                         n = *cp;
453                         *cp = 0;
454                         if (inet_aton(net, &a)) {
455                             statp->sort_list[nsort].addr = a;
456                             if (ISSORTMASK(n)) {
457                                 *cp++ = n;
458                                 net = cp;
459                                 while (*cp && *cp != ';' &&
460                                         isascii(*cp) &&
461                                         !isspace((unsigned char)*cp))
462                                     cp++;
463                                 n = *cp;
464                                 *cp = 0;
465                                 if (inet_aton(net, &a)) {
466                                     statp->sort_list[nsort].mask = a.s_addr;
467                                 } else {
468                                     statp->sort_list[nsort].mask = 
469                                         net_mask(statp->sort_list[nsort].addr);
470                                 }
471                             } else {
472                                 statp->sort_list[nsort].mask = 
473                                     net_mask(statp->sort_list[nsort].addr);
474                             }
475                             ext->sort_list[nsort].af = AF_INET;
476                             ext->sort_list[nsort].addr.ina =
477                                 statp->sort_list[nsort].addr;
478                             ext->sort_list[nsort].mask.ina.s_addr =
479                                 statp->sort_list[nsort].mask;
480                             nsort++;
481                         }
482                         else if (inet_pton(AF_INET6, net, &a6) == 1) {
483
484                             ext->sort_list[nsort].af = AF_INET6;
485                             ext->sort_list[nsort].addr.in6a = a6;
486                             u = (u_char *)&ext->sort_list[nsort].mask.in6a;
487                             *cp++ = n;
488                             net = cp;
489                             while (*cp && *cp != ';' &&
490                                     isascii(*cp) && !isspace(*cp))
491                                 cp++;
492                             m = n;
493                             n = *cp;
494                             *cp = 0;
495                             switch (m) {
496                             case '/':
497                                 m = atoi(net);
498                                 break;
499                             case '&':
500                                 if (inet_pton(AF_INET6, net, u) == 1) {
501                                     m = -1;
502                                     break;
503                                 }
504                                 /*FALLTHROUGH*/
505                             default:
506                                 m = sizeof(struct in6_addr) * CHAR_BIT;
507                                 break;
508                             }
509                             if (m >= 0) {
510                                 for (i = 0; i < sizeof(struct in6_addr); i++) {
511                                     if (m <= 0) {
512                                         *u = 0;
513                                     } else {
514                                         m -= CHAR_BIT;
515                                         *u = (u_char)~0;
516                                         if (m < 0)
517                                             *u <<= -m;
518                                     }
519                                     u++;
520                                 }
521                             }
522                             statp->sort_list[nsort].addr.s_addr =
523                                 (u_int32_t)0xffffffff;
524                             statp->sort_list[nsort].mask =
525                                 (u_int32_t)0xffffffff;
526                             nsort++;
527                         }
528                         *cp = n;
529                     }
530                     continue;
531                 }
532 #endif
533                 if (MATCH(buf, "options")) {
534                     res_setoptions(statp, buf + sizeof("options") - 1, "conf");
535                     continue;
536                 }
537             }
538             if (nserv > 0) 
539                 statp->nscount = nserv;
540 #ifdef RESOLVSORT
541             statp->nsort = nsort;
542 #endif
543             (void) fclose(fp);
544         }
545 /*
546  * Last chance to get a nameserver.  This should not normally
547  * be necessary
548  */
549 #ifdef NO_RESOLV_CONF
550         if(nserv == 0)
551                 nserv = get_nameservers(statp);
552 #endif
553
554         if (statp->defdname[0] == 0 &&
555             gethostname(buf, sizeof(statp->defdname) - 1) == 0 &&
556             (cp = strchr(buf, '.')) != NULL)
557                 strcpy(statp->defdname, cp + 1);
558
559         /* find components of local domain that might be searched */
560         if (havesearch == 0) {
561                 pp = statp->dnsrch;
562                 *pp++ = statp->defdname;
563                 *pp = NULL;
564
565                 dots = 0;
566                 for (cp = statp->defdname; *cp; cp++)
567                         dots += (*cp == '.');
568
569                 cp = statp->defdname;
570                 while (pp < statp->dnsrch + MAXDFLSRCH) {
571                         if (dots < LOCALDOMAINPARTS)
572                                 break;
573                         cp = strchr(cp, '.') + 1;    /*%< we know there is one */
574                         *pp++ = cp;
575                         dots--;
576                 }
577                 *pp = NULL;
578 #ifdef DEBUG
579                 if (statp->options & RES_DEBUG) {
580                         printf(";; res_init()... default dnsrch list:\n");
581                         for (pp = statp->dnsrch; *pp; pp++)
582                                 printf(";;\t%s\n", *pp);
583                         printf(";;\t..END..\n");
584                 }
585 #endif
586         }
587
588         if (issetugid())
589                 statp->options |= RES_NOALIASES;
590         else if ((cp = getenv("RES_OPTIONS")) != NULL)
591                 res_setoptions(statp, cp, "env");
592         statp->options |= RES_INIT;
593         return (statp->res_h_errno);
594 }
595
596 static void
597 res_setoptions(res_state statp, const char *options, const char *source)
598 {
599         const char *cp = options;
600         int i;
601         struct __res_state_ext *ext = statp->_u._ext.ext;
602
603 #ifdef DEBUG
604         if (statp->options & RES_DEBUG)
605                 printf(";; res_setoptions(\"%s\", \"%s\")...\n",
606                        options, source);
607 #endif
608         while (*cp) {
609                 /* skip leading and inner runs of spaces */
610                 while (*cp == ' ' || *cp == '\t')
611                         cp++;
612                 /* search for and process individual options */
613                 if (!strncmp(cp, "ndots:", sizeof("ndots:") - 1)) {
614                         i = atoi(cp + sizeof("ndots:") - 1);
615                         if (i <= RES_MAXNDOTS)
616                                 statp->ndots = i;
617                         else
618                                 statp->ndots = RES_MAXNDOTS;
619 #ifdef DEBUG
620                         if (statp->options & RES_DEBUG)
621                                 printf(";;\tndots=%d\n", statp->ndots);
622 #endif
623                 } else if (!strncmp(cp, "timeout:", sizeof("timeout:") - 1)) {
624                         i = atoi(cp + sizeof("timeout:") - 1);
625                         if (i <= RES_MAXRETRANS)
626                                 statp->retrans = i;
627                         else
628                                 statp->retrans = RES_MAXRETRANS;
629 #ifdef DEBUG
630                         if (statp->options & RES_DEBUG)
631                                 printf(";;\ttimeout=%d\n", statp->retrans);
632 #endif
633 #ifdef  SOLARIS2
634                 } else if (!strncmp(cp, "retrans:", sizeof("retrans:") - 1)) {
635                         /*
636                          * For backward compatibility, 'retrans' is
637                          * supported as an alias for 'timeout', though
638                          * without an imposed maximum.
639                          */
640                         statp->retrans = atoi(cp + sizeof("retrans:") - 1);
641                 } else if (!strncmp(cp, "retry:", sizeof("retry:") - 1)){
642                         /*
643                          * For backward compatibility, 'retry' is
644                          * supported as an alias for 'attempts', though
645                          * without an imposed maximum.
646                          */
647                         statp->retry = atoi(cp + sizeof("retry:") - 1);
648 #endif  /* SOLARIS2 */
649                 } else if (!strncmp(cp, "attempts:", sizeof("attempts:") - 1)){
650                         i = atoi(cp + sizeof("attempts:") - 1);
651                         if (i <= RES_MAXRETRY)
652                                 statp->retry = i;
653                         else
654                                 statp->retry = RES_MAXRETRY;
655 #ifdef DEBUG
656                         if (statp->options & RES_DEBUG)
657                                 printf(";;\tattempts=%d\n", statp->retry);
658 #endif
659                 } else if (!strncmp(cp, "debug", sizeof("debug") - 1)) {
660 #ifdef DEBUG
661                         if (!(statp->options & RES_DEBUG)) {
662                                 printf(";; res_setoptions(\"%s\", \"%s\")..\n",
663                                        options, source);
664                                 statp->options |= RES_DEBUG;
665                         }
666                         printf(";;\tdebug\n");
667 #endif
668                 } else if (!strncmp(cp, "no_tld_query",
669                                     sizeof("no_tld_query") - 1) ||
670                            !strncmp(cp, "no-tld-query",
671                                     sizeof("no-tld-query") - 1)) {
672                         statp->options |= RES_NOTLDQUERY;
673                 } else if (!strncmp(cp, "inet6", sizeof("inet6") - 1)) {
674                         statp->options |= RES_USE_INET6;
675                 } else if (!strncmp(cp, "insecure1", sizeof("insecure1") - 1)) {
676                        statp->options |= RES_INSECURE1;
677                 } else if (!strncmp(cp, "insecure2", sizeof("insecure2") - 1)) {
678                        statp->options |= RES_INSECURE2;
679                 } else if (!strncmp(cp, "rotate", sizeof("rotate") - 1)) {
680                         statp->options |= RES_ROTATE;
681                 } else if (!strncmp(cp, "no-check-names",
682                                     sizeof("no-check-names") - 1)) {
683                         statp->options |= RES_NOCHECKNAME;
684                 } else if (!strncmp(cp, "reload-period:",
685                                     sizeof("reload-period:") - 1)) {
686                         if (ext != NULL) {
687                                 ext->reload_period = (u_short)
688                                     atoi(cp + sizeof("reload-period:") - 1);
689                         }
690                 }
691 #ifdef RES_USE_EDNS0
692                 else if (!strncmp(cp, "edns0", sizeof("edns0") - 1)) {
693                         statp->options |= RES_USE_EDNS0;
694                 }
695 #endif
696 #ifndef _LIBC
697                 else if (!strncmp(cp, "dname", sizeof("dname") - 1)) {
698                         statp->options |= RES_USE_DNAME;
699                 }
700                 else if (!strncmp(cp, "nibble:", sizeof("nibble:") - 1)) {
701                         if (ext == NULL)
702                                 goto skip;
703                         cp += sizeof("nibble:") - 1;
704                         i = MIN(strcspn(cp, " \t"), sizeof(ext->nsuffix) - 1);
705                         strncpy(ext->nsuffix, cp, i);
706                         ext->nsuffix[i] = '\0';
707                 }
708                 else if (!strncmp(cp, "nibble2:", sizeof("nibble2:") - 1)) {
709                         if (ext == NULL)
710                                 goto skip;
711                         cp += sizeof("nibble2:") - 1;
712                         i = MIN(strcspn(cp, " \t"), sizeof(ext->nsuffix2) - 1);
713                         strncpy(ext->nsuffix2, cp, i);
714                         ext->nsuffix2[i] = '\0';
715                 }
716                 else if (!strncmp(cp, "v6revmode:", sizeof("v6revmode:") - 1)) {
717                         cp += sizeof("v6revmode:") - 1;
718                         /* "nibble" and "bitstring" used to be valid */
719                         if (!strncmp(cp, "single", sizeof("single") - 1)) {
720                                 statp->options |= RES_NO_NIBBLE2;
721                         } else if (!strncmp(cp, "both", sizeof("both") - 1)) {
722                                 statp->options &=
723                                          ~RES_NO_NIBBLE2;
724                         }
725                 }
726 #endif
727                 else {
728                         /* XXX - print a warning here? */
729                 }
730 #ifndef _LIBC
731    skip:
732 #endif
733                 /* skip to next run of spaces */
734                 while (*cp && *cp != ' ' && *cp != '\t')
735                         cp++;
736         }
737 }
738
739 #ifdef RESOLVSORT
740 /* XXX - should really support CIDR which means explicit masks always. */
741 static u_int32_t
742 net_mask(in)            /*!< XXX - should really use system's version of this  */
743         struct in_addr in;
744 {
745         u_int32_t i = ntohl(in.s_addr);
746
747         if (IN_CLASSA(i))
748                 return (htonl(IN_CLASSA_NET));
749         else if (IN_CLASSB(i))
750                 return (htonl(IN_CLASSB_NET));
751         return (htonl(IN_CLASSC_NET));
752 }
753 #endif
754
755 static u_char srnd[16];
756
757 void
758 res_rndinit(res_state statp)
759 {
760         struct timeval now;
761         u_int32_t u32;
762         u_int16_t u16;
763         u_char *rnd = statp->_rnd == NULL ? srnd : statp->_rnd;
764
765         gettimeofday(&now, NULL);
766         u32 = now.tv_sec;
767         memcpy(rnd, &u32, 4);
768         u32 = now.tv_usec;
769         memcpy(rnd + 4, &u32, 4);
770         u32 += now.tv_sec;
771         memcpy(rnd + 8, &u32, 4);
772         u16 = getpid();
773         memcpy(rnd + 12, &u16, 2);
774 }
775
776 u_int
777 res_nrandomid(res_state statp) {
778         struct timeval now;
779         u_int16_t u16;
780         MD5_CTX ctx;
781         u_char *rnd = statp->_rnd == NULL ? srnd : statp->_rnd;
782
783         gettimeofday(&now, NULL);
784         u16 = (u_int16_t) (now.tv_sec ^ now.tv_usec);
785         memcpy(rnd + 14, &u16, 2);
786 #ifndef HAVE_MD5
787         MD5_Init(&ctx);
788         MD5_Update(&ctx, rnd, 16);
789         MD5_Final(rnd, &ctx);
790 #else
791         MD5Init(&ctx);
792         MD5Update(&ctx, rnd, 16);
793         MD5Final(rnd, &ctx);
794 #endif
795         memcpy(&u16, rnd + 14, 2);
796         return ((u_int) u16);
797 }
798
799 /*%
800  * This routine is for closing the socket if a virtual circuit is used and
801  * the program wants to close it.  This provides support for endhostent()
802  * which expects to close the socket.
803  *
804  * This routine is not expected to be user visible.
805  */
806 void
807 res_nclose(res_state statp) {
808         int ns;
809
810         if (statp->_vcsock >= 0) { 
811                 (void) _close(statp->_vcsock);
812                 statp->_vcsock = -1;
813                 statp->_flags &= ~(RES_F_VC | RES_F_CONN);
814         }
815         for (ns = 0; ns < statp->_u._ext.nscount; ns++) {
816                 if (statp->_u._ext.nssocks[ns] != -1) {
817                         (void) _close(statp->_u._ext.nssocks[ns]);
818                         statp->_u._ext.nssocks[ns] = -1;
819                 }
820         }
821 }
822
823 void
824 res_ndestroy(res_state statp) {
825         res_nclose(statp);
826         if (statp->_u._ext.ext != NULL) {
827                 free(statp->_u._ext.ext);
828                 statp->_u._ext.ext = NULL;
829         }
830         if (statp->_rnd != NULL) {
831                 free(statp->_rnd);
832                 statp->_rnd = NULL;
833         }
834         statp->options &= ~RES_INIT;
835 }
836
837 #ifndef _LIBC
838 const char *
839 res_get_nibblesuffix(res_state statp) {
840         if (statp->_u._ext.ext)
841                 return (statp->_u._ext.ext->nsuffix);
842         return ("ip6.arpa");
843 }
844
845 const char *
846 res_get_nibblesuffix2(res_state statp) {
847         if (statp->_u._ext.ext)
848                 return (statp->_u._ext.ext->nsuffix2);
849         return ("ip6.int");
850 }
851 #endif
852
853 void
854 res_setservers(res_state statp, const union res_sockaddr_union *set, int cnt) {
855         int i, nserv;
856         size_t size;
857
858         /* close open servers */
859         res_nclose(statp);
860
861         /* cause rtt times to be forgotten */
862         statp->_u._ext.nscount = 0;
863
864         nserv = 0;
865         for (i = 0; i < cnt && nserv < MAXNS; i++) {
866                 switch (set->sin.sin_family) {
867                 case AF_INET:
868                         size = sizeof(set->sin);
869                         if (statp->_u._ext.ext)
870                                 memcpy(&statp->_u._ext.ext->nsaddrs[nserv],
871                                         &set->sin, size);
872                         if (size <= sizeof(statp->nsaddr_list[nserv]))
873                                 memcpy(&statp->nsaddr_list[nserv],
874                                         &set->sin, size);
875                         else
876                                 statp->nsaddr_list[nserv].sin_family = 0;
877                         nserv++;
878                         break;
879
880 #ifdef HAS_INET6_STRUCTS
881                 case AF_INET6:
882                         size = sizeof(set->sin6);
883                         if (statp->_u._ext.ext)
884                                 memcpy(&statp->_u._ext.ext->nsaddrs[nserv],
885                                         &set->sin6, size);
886                         if (size <= sizeof(statp->nsaddr_list[nserv]))
887                                 memcpy(&statp->nsaddr_list[nserv],
888                                         &set->sin6, size);
889                         else
890                                 statp->nsaddr_list[nserv].sin_family = 0;
891                         nserv++;
892                         break;
893 #endif
894
895                 default:
896                         break;
897                 }
898                 set++;
899         }
900         statp->nscount = nserv;
901         
902 }
903
904 int
905 res_getservers(res_state statp, union res_sockaddr_union *set, int cnt) {
906         int i;
907         size_t size;
908         u_int16_t family;
909
910         for (i = 0; i < statp->nscount && i < cnt; i++) {
911                 if (statp->_u._ext.ext)
912                         family = statp->_u._ext.ext->nsaddrs[i].sin.sin_family;
913                 else 
914                         family = statp->nsaddr_list[i].sin_family;
915
916                 switch (family) {
917                 case AF_INET:
918                         size = sizeof(set->sin);
919                         if (statp->_u._ext.ext)
920                                 memcpy(&set->sin,
921                                        &statp->_u._ext.ext->nsaddrs[i],
922                                        size);
923                         else
924                                 memcpy(&set->sin, &statp->nsaddr_list[i],
925                                        size);
926                         break;
927
928 #ifdef HAS_INET6_STRUCTS
929                 case AF_INET6:
930                         size = sizeof(set->sin6);
931                         if (statp->_u._ext.ext)
932                                 memcpy(&set->sin6,
933                                        &statp->_u._ext.ext->nsaddrs[i],
934                                        size);
935                         else
936                                 memcpy(&set->sin6, &statp->nsaddr_list[i],
937                                        size);
938                         break;
939 #endif
940
941                 default:
942                         set->sin.sin_family = 0;
943                         break;
944                 }
945                 set++;
946         }
947         return (statp->nscount);
948 }
949
950 /*! \file */