]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - lib/libc/resolv/res_init.c
Integrate tools/test/posixshm and tools/regression/posixshm into the FreeBSD
[FreeBSD/FreeBSD.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 #define DEBUG
120
121 #ifdef SOLARIS2
122 #include <sys/systeminfo.h>
123 #endif
124
125 static void res_setoptions(res_state, const char *, const char *);
126
127 #ifdef RESOLVSORT
128 static const char sort_mask[] = "/&";
129 #define ISSORTMASK(ch) (strchr(sort_mask, ch) != NULL)
130 static u_int32_t net_mask(struct in_addr);
131 #endif
132
133 #if !defined(isascii)   /*%< XXX - could be a function */
134 # define isascii(c) (!(c & 0200))
135 #endif
136
137 /*
138  * Resolver state default settings.
139  */
140
141 /*%
142  * Set up default settings.  If the configuration file exist, the values
143  * there will have precedence.  Otherwise, the server address is set to
144  * INADDR_ANY and the default domain name comes from the gethostname().
145  *
146  * An interrim version of this code (BIND 4.9, pre-4.4BSD) used 127.0.0.1
147  * rather than INADDR_ANY ("0.0.0.0") as the default name server address
148  * since it was noted that INADDR_ANY actually meant ``the first interface
149  * you "ifconfig"'d at boot time'' and if this was a SLIP or PPP interface,
150  * it had to be "up" in order for you to reach your own name server.  It
151  * was later decided that since the recommended practice is to always 
152  * install local static routes through 127.0.0.1 for all your network
153  * interfaces, that we could solve this problem without a code change.
154  *
155  * The configuration file should always be used, since it is the only way
156  * to specify a default domain.  If you are running a server on your local
157  * machine, you should say "nameserver 0.0.0.0" or "nameserver 127.0.0.1"
158  * in the configuration file.
159  *
160  * Return 0 if completes successfully, -1 on error
161  */
162 int
163 res_ninit(res_state statp) {
164         extern int __res_vinit(res_state, int);
165
166         return (__res_vinit(statp, 0));
167 }
168
169 /*% This function has to be reachable by res_data.c but not publically. */
170 int
171 __res_vinit(res_state statp, int preinit) {
172         FILE *fp;
173         char *cp, **pp;
174         int n;
175         char buf[BUFSIZ];
176         int nserv = 0;    /*%< number of nameserver records read from file */
177         int haveenv = 0;
178         int havesearch = 0;
179 #ifdef RESOLVSORT
180         int nsort = 0;
181         char *net;
182 #endif
183         int dots;
184         union res_sockaddr_union u[2];
185         int maxns = MAXNS;
186
187         RES_SET_H_ERRNO(statp, 0);
188         if (statp->_u._ext.ext != NULL)
189                 res_ndestroy(statp);
190
191         if (!preinit) {
192                 statp->retrans = RES_TIMEOUT;
193                 statp->retry = RES_DFLRETRY;
194                 statp->options = RES_DEFAULT;
195         }
196
197         statp->_rnd = malloc(16);
198         res_rndinit(statp);
199         statp->id = res_nrandomid(statp);
200
201         memset(u, 0, sizeof(u));
202 #ifdef USELOOPBACK
203         u[nserv].sin.sin_addr = inet_makeaddr(IN_LOOPBACKNET, 1);
204 #else
205         u[nserv].sin.sin_addr.s_addr = INADDR_ANY;
206 #endif
207         u[nserv].sin.sin_family = AF_INET;
208         u[nserv].sin.sin_port = htons(NAMESERVER_PORT);
209 #ifdef HAVE_SA_LEN
210         u[nserv].sin.sin_len = sizeof(struct sockaddr_in);
211 #endif
212         nserv++;
213 #ifdef HAS_INET6_STRUCTS
214 #ifdef USELOOPBACK
215         u[nserv].sin6.sin6_addr = in6addr_loopback;
216 #else
217         u[nserv].sin6.sin6_addr = in6addr_any;
218 #endif
219         u[nserv].sin6.sin6_family = AF_INET6;
220         u[nserv].sin6.sin6_port = htons(NAMESERVER_PORT);
221 #ifdef HAVE_SA_LEN
222         u[nserv].sin6.sin6_len = sizeof(struct sockaddr_in6);
223 #endif
224         nserv++;
225 #endif
226         statp->nscount = 0;
227         statp->ndots = 1;
228         statp->pfcode = 0;
229         statp->_vcsock = -1;
230         statp->_flags = 0;
231         statp->reload_period = 2;
232         statp->qhook = NULL;
233         statp->rhook = NULL;
234         statp->_u._ext.nscount = 0;
235         statp->_u._ext.ext = malloc(sizeof(*statp->_u._ext.ext));
236         if (statp->_u._ext.ext != NULL) {
237                 memset(statp->_u._ext.ext, 0, sizeof(*statp->_u._ext.ext));
238                 statp->_u._ext.ext->nsaddrs[0].sin = statp->nsaddr;
239                 strcpy(statp->_u._ext.ext->nsuffix, "ip6.arpa");
240                 strcpy(statp->_u._ext.ext->nsuffix2, "ip6.int");
241         } else {
242                 /*
243                  * Historically res_init() rarely, if at all, failed.
244                  * Examples and applications exist which do not check
245                  * our return code.  Furthermore several applications
246                  * simply call us to get the systems domainname.  So
247                  * rather then immediately fail here we store the
248                  * failure, which is returned later, in h_errno.  And
249                  * prevent the collection of 'nameserver' information
250                  * by setting maxns to 0.  Thus applications that fail
251                  * to check our return code wont be able to make
252                  * queries anyhow.
253                  */
254                 RES_SET_H_ERRNO(statp, NETDB_INTERNAL);
255                 maxns = 0;
256         }
257 #ifdef RESOLVSORT
258         statp->nsort = 0;
259 #endif
260         res_setservers(statp, u, nserv);
261
262 #ifdef  SOLARIS2
263         /*
264          * The old libresolv derived the defaultdomain from NIS/NIS+.
265          * We want to keep this behaviour
266          */
267         {
268                 char buf[sizeof(statp->defdname)], *cp;
269                 int ret;
270
271                 if ((ret = sysinfo(SI_SRPC_DOMAIN, buf, sizeof(buf))) > 0 &&
272                         (unsigned int)ret <= sizeof(buf)) {
273                         if (buf[0] == '+')
274                                 buf[0] = '.';
275                         cp = strchr(buf, '.');
276                         cp = (cp == NULL) ? buf : (cp + 1);
277                         strncpy(statp->defdname, cp,
278                                 sizeof(statp->defdname) - 1);
279                         statp->defdname[sizeof(statp->defdname) - 1] = '\0';
280                 }
281         }
282 #endif  /* SOLARIS2 */
283
284         /* Allow user to override the local domain definition */
285         if (issetugid() == 0 && (cp = getenv("LOCALDOMAIN")) != NULL) {
286                 (void)strncpy(statp->defdname, cp, sizeof(statp->defdname) - 1);
287                 statp->defdname[sizeof(statp->defdname) - 1] = '\0';
288                 haveenv++;
289
290                 /*
291                  * Set search list to be blank-separated strings
292                  * from rest of env value.  Permits users of LOCALDOMAIN
293                  * to still have a search list, and anyone to set the
294                  * one that they want to use as an individual (even more
295                  * important now that the rfc1535 stuff restricts searches)
296                  */
297                 cp = statp->defdname;
298                 pp = statp->dnsrch;
299                 *pp++ = cp;
300                 for (n = 0; *cp && pp < statp->dnsrch + MAXDNSRCH; cp++) {
301                         if (*cp == '\n')        /*%< silly backwards compat */
302                                 break;
303                         else if (*cp == ' ' || *cp == '\t') {
304                                 *cp = 0;
305                                 n = 1;
306                         } else if (n) {
307                                 *pp++ = cp;
308                                 n = 0;
309                                 havesearch = 1;
310                         }
311                 }
312                 /* null terminate last domain if there are excess */
313                 while (*cp != '\0' && *cp != ' ' && *cp != '\t' && *cp != '\n')
314                         cp++;
315                 *cp = '\0';
316                 *pp++ = 0;
317         }
318
319 #define MATCH(line, name) \
320         (!strncmp(line, name, sizeof(name) - 1) && \
321         (line[sizeof(name) - 1] == ' ' || \
322          line[sizeof(name) - 1] == '\t'))
323
324         nserv = 0;
325         if ((fp = fopen(_PATH_RESCONF, "re")) != NULL) {
326             struct stat sb;
327             struct timespec now;
328
329             if (_fstat(fileno(fp), &sb) == 0) {
330                 statp->conf_mtim = sb.st_mtim;
331                 if (clock_gettime(CLOCK_MONOTONIC_FAST, &now) == 0) {
332                     statp->conf_stat = now.tv_sec;
333                 } else {
334                     statp->conf_stat = 0;
335                 }
336             } else {
337                 statp->conf_mtim.tv_sec = 0;
338                 statp->conf_mtim.tv_nsec = 0;
339                 statp->conf_stat = 0;
340             }
341
342             /* read the config file */
343             while (fgets(buf, sizeof(buf), fp) != NULL) {
344                 /* skip comments */
345                 if (*buf == ';' || *buf == '#')
346                         continue;
347                 /* read default domain name */
348                 if (MATCH(buf, "domain")) {
349                     if (haveenv)        /*%< skip if have from environ */
350                             continue;
351                     cp = buf + sizeof("domain") - 1;
352                     while (*cp == ' ' || *cp == '\t')
353                             cp++;
354                     if ((*cp == '\0') || (*cp == '\n'))
355                             continue;
356                     strncpy(statp->defdname, cp, sizeof(statp->defdname) - 1);
357                     statp->defdname[sizeof(statp->defdname) - 1] = '\0';
358                     if ((cp = strpbrk(statp->defdname, " \t\n")) != NULL)
359                             *cp = '\0';
360                     havesearch = 0;
361                     continue;
362                 }
363                 /* set search list */
364                 if (MATCH(buf, "search")) {
365                     if (haveenv)        /*%< skip if have from environ */
366                             continue;
367                     cp = buf + sizeof("search") - 1;
368                     while (*cp == ' ' || *cp == '\t')
369                             cp++;
370                     if ((*cp == '\0') || (*cp == '\n'))
371                             continue;
372                     strncpy(statp->defdname, cp, sizeof(statp->defdname) - 1);
373                     statp->defdname[sizeof(statp->defdname) - 1] = '\0';
374                     if ((cp = strchr(statp->defdname, '\n')) != NULL)
375                             *cp = '\0';
376                     /*
377                      * Set search list to be blank-separated strings
378                      * on rest of line.
379                      */
380                     cp = statp->defdname;
381                     pp = statp->dnsrch;
382                     *pp++ = cp;
383                     for (n = 0; *cp && pp < statp->dnsrch + MAXDNSRCH; cp++) {
384                             if (*cp == ' ' || *cp == '\t') {
385                                     *cp = 0;
386                                     n = 1;
387                             } else if (n) {
388                                     *pp++ = cp;
389                                     n = 0;
390                             }
391                     }
392                     /* null terminate last domain if there are excess */
393                     while (*cp != '\0' && *cp != ' ' && *cp != '\t')
394                             cp++;
395                     *cp = '\0';
396                     *pp++ = 0;
397                     havesearch = 1;
398                     continue;
399                 }
400                 /* read nameservers to query */
401                 if (MATCH(buf, "nameserver") && nserv < maxns) {
402                     struct addrinfo hints, *ai;
403                     char sbuf[NI_MAXSERV];
404                     const size_t minsiz =
405                         sizeof(statp->_u._ext.ext->nsaddrs[0]);
406
407                     cp = buf + sizeof("nameserver") - 1;
408                     while (*cp == ' ' || *cp == '\t')
409                         cp++;
410                     cp[strcspn(cp, ";# \t\n")] = '\0';
411                     if ((*cp != '\0') && (*cp != '\n')) {
412                         memset(&hints, 0, sizeof(hints));
413                         hints.ai_family = PF_UNSPEC;
414                         hints.ai_socktype = SOCK_DGRAM; /*dummy*/
415                         hints.ai_flags = AI_NUMERICHOST;
416                         sprintf(sbuf, "%u", NAMESERVER_PORT);
417                         if (getaddrinfo(cp, sbuf, &hints, &ai) == 0 &&
418                             ai->ai_addrlen <= minsiz) {
419                             if (statp->_u._ext.ext != NULL) {
420                                 memcpy(&statp->_u._ext.ext->nsaddrs[nserv],
421                                     ai->ai_addr, ai->ai_addrlen);
422                             }
423                             if (ai->ai_addrlen <=
424                                 sizeof(statp->nsaddr_list[nserv])) {
425                                 memcpy(&statp->nsaddr_list[nserv],
426                                     ai->ai_addr, ai->ai_addrlen);
427                             } else
428                                 statp->nsaddr_list[nserv].sin_family = 0;
429                             freeaddrinfo(ai);
430                             nserv++;
431                         }
432                     }
433                     continue;
434                 }
435 #ifdef RESOLVSORT
436                 if (MATCH(buf, "sortlist")) {
437                     struct in_addr a;
438                     struct in6_addr a6;
439                     int m, i;
440                     u_char *u;
441                     struct __res_state_ext *ext = statp->_u._ext.ext;
442
443                     cp = buf + sizeof("sortlist") - 1;
444                     while (nsort < MAXRESOLVSORT) {
445                         while (*cp == ' ' || *cp == '\t')
446                             cp++;
447                         if (*cp == '\0' || *cp == '\n' || *cp == ';')
448                             break;
449                         net = cp;
450                         while (*cp && !ISSORTMASK(*cp) && *cp != ';' &&
451                                isascii(*cp) && !isspace((unsigned char)*cp))
452                                 cp++;
453                         n = *cp;
454                         *cp = 0;
455                         if (inet_aton(net, &a)) {
456                             statp->sort_list[nsort].addr = a;
457                             if (ISSORTMASK(n)) {
458                                 *cp++ = n;
459                                 net = cp;
460                                 while (*cp && *cp != ';' &&
461                                         isascii(*cp) &&
462                                         !isspace((unsigned char)*cp))
463                                     cp++;
464                                 n = *cp;
465                                 *cp = 0;
466                                 if (inet_aton(net, &a)) {
467                                     statp->sort_list[nsort].mask = a.s_addr;
468                                 } else {
469                                     statp->sort_list[nsort].mask = 
470                                         net_mask(statp->sort_list[nsort].addr);
471                                 }
472                             } else {
473                                 statp->sort_list[nsort].mask = 
474                                     net_mask(statp->sort_list[nsort].addr);
475                             }
476                             ext->sort_list[nsort].af = AF_INET;
477                             ext->sort_list[nsort].addr.ina =
478                                 statp->sort_list[nsort].addr;
479                             ext->sort_list[nsort].mask.ina.s_addr =
480                                 statp->sort_list[nsort].mask;
481                             nsort++;
482                         }
483                         else if (inet_pton(AF_INET6, net, &a6) == 1) {
484
485                             ext->sort_list[nsort].af = AF_INET6;
486                             ext->sort_list[nsort].addr.in6a = a6;
487                             u = (u_char *)&ext->sort_list[nsort].mask.in6a;
488                             *cp++ = n;
489                             net = cp;
490                             while (*cp && *cp != ';' &&
491                                     isascii(*cp) && !isspace(*cp))
492                                 cp++;
493                             m = n;
494                             n = *cp;
495                             *cp = 0;
496                             switch (m) {
497                             case '/':
498                                 m = atoi(net);
499                                 break;
500                             case '&':
501                                 if (inet_pton(AF_INET6, net, u) == 1) {
502                                     m = -1;
503                                     break;
504                                 }
505                                 /*FALLTHROUGH*/
506                             default:
507                                 m = sizeof(struct in6_addr) * CHAR_BIT;
508                                 break;
509                             }
510                             if (m >= 0) {
511                                 for (i = 0; i < sizeof(struct in6_addr); i++) {
512                                     if (m <= 0) {
513                                         *u = 0;
514                                     } else {
515                                         m -= CHAR_BIT;
516                                         *u = (u_char)~0;
517                                         if (m < 0)
518                                             *u <<= -m;
519                                     }
520                                     u++;
521                                 }
522                             }
523                             statp->sort_list[nsort].addr.s_addr =
524                                 (u_int32_t)0xffffffff;
525                             statp->sort_list[nsort].mask =
526                                 (u_int32_t)0xffffffff;
527                             nsort++;
528                         }
529                         *cp = n;
530                     }
531                     continue;
532                 }
533 #endif
534                 if (MATCH(buf, "options")) {
535                     res_setoptions(statp, buf + sizeof("options") - 1, "conf");
536                     continue;
537                 }
538             }
539             if (nserv > 0) 
540                 statp->nscount = nserv;
541 #ifdef RESOLVSORT
542             statp->nsort = nsort;
543 #endif
544             (void) fclose(fp);
545         }
546 /*
547  * Last chance to get a nameserver.  This should not normally
548  * be necessary
549  */
550 #ifdef NO_RESOLV_CONF
551         if(nserv == 0)
552                 nserv = get_nameservers(statp);
553 #endif
554
555         if (statp->defdname[0] == 0 &&
556             gethostname(buf, sizeof(statp->defdname) - 1) == 0 &&
557             (cp = strchr(buf, '.')) != NULL)
558                 strcpy(statp->defdname, cp + 1);
559
560         /* find components of local domain that might be searched */
561         if (havesearch == 0) {
562                 pp = statp->dnsrch;
563                 *pp++ = statp->defdname;
564                 *pp = NULL;
565
566                 dots = 0;
567                 for (cp = statp->defdname; *cp; cp++)
568                         dots += (*cp == '.');
569
570                 cp = statp->defdname;
571                 while (pp < statp->dnsrch + MAXDFLSRCH) {
572                         if (dots < LOCALDOMAINPARTS)
573                                 break;
574                         cp = strchr(cp, '.') + 1;    /*%< we know there is one */
575                         *pp++ = cp;
576                         dots--;
577                 }
578                 *pp = NULL;
579 #ifdef DEBUG
580                 if (statp->options & RES_DEBUG) {
581                         printf(";; res_init()... default dnsrch list:\n");
582                         for (pp = statp->dnsrch; *pp; pp++)
583                                 printf(";;\t%s\n", *pp);
584                         printf(";;\t..END..\n");
585                 }
586 #endif
587         }
588
589         if (issetugid())
590                 statp->options |= RES_NOALIASES;
591         else if ((cp = getenv("RES_OPTIONS")) != NULL)
592                 res_setoptions(statp, cp, "env");
593         statp->options |= RES_INIT;
594         return (statp->res_h_errno);
595 }
596
597 static void
598 res_setoptions(res_state statp, const char *options, const char *source)
599 {
600         const char *cp = options;
601         int i;
602 #ifndef _LIBC
603         struct __res_state_ext *ext = statp->_u._ext.ext;
604 #endif
605
606 #ifdef DEBUG
607         if (statp->options & RES_DEBUG)
608                 printf(";; res_setoptions(\"%s\", \"%s\")...\n",
609                        options, source);
610 #endif
611         while (*cp) {
612                 /* skip leading and inner runs of spaces */
613                 while (*cp == ' ' || *cp == '\t')
614                         cp++;
615                 /* search for and process individual options */
616                 if (!strncmp(cp, "ndots:", sizeof("ndots:") - 1)) {
617                         i = atoi(cp + sizeof("ndots:") - 1);
618                         if (i <= RES_MAXNDOTS)
619                                 statp->ndots = i;
620                         else
621                                 statp->ndots = RES_MAXNDOTS;
622 #ifdef DEBUG
623                         if (statp->options & RES_DEBUG)
624                                 printf(";;\tndots=%d\n", statp->ndots);
625 #endif
626                 } else if (!strncmp(cp, "timeout:", sizeof("timeout:") - 1)) {
627                         i = atoi(cp + sizeof("timeout:") - 1);
628                         if (i <= RES_MAXRETRANS)
629                                 statp->retrans = i;
630                         else
631                                 statp->retrans = RES_MAXRETRANS;
632 #ifdef DEBUG
633                         if (statp->options & RES_DEBUG)
634                                 printf(";;\ttimeout=%d\n", statp->retrans);
635 #endif
636 #ifdef  SOLARIS2
637                 } else if (!strncmp(cp, "retrans:", sizeof("retrans:") - 1)) {
638                         /*
639                          * For backward compatibility, 'retrans' is
640                          * supported as an alias for 'timeout', though
641                          * without an imposed maximum.
642                          */
643                         statp->retrans = atoi(cp + sizeof("retrans:") - 1);
644                 } else if (!strncmp(cp, "retry:", sizeof("retry:") - 1)){
645                         /*
646                          * For backward compatibility, 'retry' is
647                          * supported as an alias for 'attempts', though
648                          * without an imposed maximum.
649                          */
650                         statp->retry = atoi(cp + sizeof("retry:") - 1);
651 #endif  /* SOLARIS2 */
652                 } else if (!strncmp(cp, "attempts:", sizeof("attempts:") - 1)){
653                         i = atoi(cp + sizeof("attempts:") - 1);
654                         if (i <= RES_MAXRETRY)
655                                 statp->retry = i;
656                         else
657                                 statp->retry = RES_MAXRETRY;
658 #ifdef DEBUG
659                         if (statp->options & RES_DEBUG)
660                                 printf(";;\tattempts=%d\n", statp->retry);
661 #endif
662                 } else if (!strncmp(cp, "debug", sizeof("debug") - 1)) {
663 #ifdef DEBUG
664                         if (!(statp->options & RES_DEBUG)) {
665                                 printf(";; res_setoptions(\"%s\", \"%s\")..\n",
666                                        options, source);
667                                 statp->options |= RES_DEBUG;
668                         }
669                         printf(";;\tdebug\n");
670 #endif
671                 } else if (!strncmp(cp, "no_tld_query",
672                                     sizeof("no_tld_query") - 1) ||
673                            !strncmp(cp, "no-tld-query",
674                                     sizeof("no-tld-query") - 1)) {
675                         statp->options |= RES_NOTLDQUERY;
676                 } else if (!strncmp(cp, "inet6", sizeof("inet6") - 1)) {
677                         statp->options |= RES_USE_INET6;
678                 } else if (!strncmp(cp, "insecure1", sizeof("insecure1") - 1)) {
679                        statp->options |= RES_INSECURE1;
680                 } else if (!strncmp(cp, "insecure2", sizeof("insecure2") - 1)) {
681                        statp->options |= RES_INSECURE2;
682                 } else if (!strncmp(cp, "rotate", sizeof("rotate") - 1)) {
683                         statp->options |= RES_ROTATE;
684                 } else if (!strncmp(cp, "no-check-names",
685                                     sizeof("no-check-names") - 1)) {
686                         statp->options |= RES_NOCHECKNAME;
687                 } else if (!strncmp(cp, "reload-period:",
688                                     sizeof("reload-period:") - 1)) {
689                         statp->reload_period = (u_short)
690                                 atoi(cp + sizeof("reload-period:") - 1);
691                 }
692 #ifdef RES_USE_EDNS0
693                 else if (!strncmp(cp, "edns0", sizeof("edns0") - 1)) {
694                         statp->options |= RES_USE_EDNS0;
695                 }
696 #endif
697 #ifndef _LIBC
698                 else if (!strncmp(cp, "dname", sizeof("dname") - 1)) {
699                         statp->options |= RES_USE_DNAME;
700                 }
701                 else if (!strncmp(cp, "nibble:", sizeof("nibble:") - 1)) {
702                         if (ext == NULL)
703                                 goto skip;
704                         cp += sizeof("nibble:") - 1;
705                         i = MIN(strcspn(cp, " \t"), sizeof(ext->nsuffix) - 1);
706                         strncpy(ext->nsuffix, cp, i);
707                         ext->nsuffix[i] = '\0';
708                 }
709                 else if (!strncmp(cp, "nibble2:", sizeof("nibble2:") - 1)) {
710                         if (ext == NULL)
711                                 goto skip;
712                         cp += sizeof("nibble2:") - 1;
713                         i = MIN(strcspn(cp, " \t"), sizeof(ext->nsuffix2) - 1);
714                         strncpy(ext->nsuffix2, cp, i);
715                         ext->nsuffix2[i] = '\0';
716                 }
717                 else if (!strncmp(cp, "v6revmode:", sizeof("v6revmode:") - 1)) {
718                         cp += sizeof("v6revmode:") - 1;
719                         /* "nibble" and "bitstring" used to be valid */
720                         if (!strncmp(cp, "single", sizeof("single") - 1)) {
721                                 statp->options |= RES_NO_NIBBLE2;
722                         } else if (!strncmp(cp, "both", sizeof("both") - 1)) {
723                                 statp->options &=
724                                          ~RES_NO_NIBBLE2;
725                         }
726                 }
727 #endif
728                 else {
729                         /* XXX - print a warning here? */
730                 }
731 #ifndef _LIBC
732    skip:
733 #endif
734                 /* skip to next run of spaces */
735                 while (*cp && *cp != ' ' && *cp != '\t')
736                         cp++;
737         }
738 }
739
740 #ifdef RESOLVSORT
741 /* XXX - should really support CIDR which means explicit masks always. */
742 static u_int32_t
743 net_mask(struct in_addr in)             /*!< XXX - should really use system's version of this  */
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 */