2003-03-15 Roland McGrath <roland@redhat.com>
[kopensolaris-gnu/glibc.git] / resolv / res_query.c
index 7ee6be7..ab4e02a 100644 (file)
@@ -1,6 +1,4 @@
 /*
- * ++Copyright++ 1988, 1993
- * -
  * Copyright (c) 1988, 1993
  *    The Regents of the University of California.  All rights reserved.
  *
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
  *    documentation and/or other materials provided with the distribution.
- * 3. All advertising materials mentioning features or use of this software
- *    must display the following acknowledgement:
- *     This product includes software developed by the University of
- *     California, Berkeley and its contributors.
  * 4. Neither the name of the University nor the names of its contributors
  *    may be used to endorse or promote products derived from this software
  *    without specific prior written permission.
@@ -31,7 +25,9 @@
  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  * SUCH DAMAGE.
- * -
+ */
+
+/*
  * Portions Copyright (c) 1993 by Digital Equipment Corporation.
  *
  * Permission to use, copy, modify, and distribute this software for any
  * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
  * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
  * SOFTWARE.
- * -
- * --Copyright--
+ */
+
+/*
+ * Portions Copyright (c) 1996-1999 by Internet Software Consortium.
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SOFTWARE CONSORTIUM DISCLAIMS
+ * ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL INTERNET SOFTWARE
+ * CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
+ * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
+ * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
+ * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
+ * SOFTWARE.
  */
 
 #if defined(LIBC_SCCS) && !defined(lint)
-static char sccsid[] = "@(#)res_query.c        8.1 (Berkeley) 6/4/93";
-static char rcsid[] = "$Id$";
+static const char sccsid[] = "@(#)res_query.c  8.1 (Berkeley) 6/4/93";
+static const char rcsid[] = "$BINDId: res_query.c,v 8.20 2000/02/29 05:39:12 vixie Exp $";
 #endif /* LIBC_SCCS and not lint */
 
 #include <sys/types.h>
@@ -63,31 +74,29 @@ static char rcsid[] = "$Id$";
 #include <netinet/in.h>
 #include <arpa/inet.h>
 #include <arpa/nameser.h>
-
-#include <stdio.h>
-#include <netdb.h>
-#include <resolv.h>
 #include <ctype.h>
 #include <errno.h>
-#if defined(BSD) && (BSD >= 199306)
-# include <stdlib.h>
-# include <string.h>
-#else
-# include "../conf/portability.h"
-#endif
+#include <netdb.h>
+#include <resolv.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
 
-#if defined(USE_OPTIONS_H)
-# include <../conf/options.h>
-#endif
+/* Options.  Leave them on. */
+/* #undef DEBUG */
 
-#if PACKETSZ > 1024
+#if PACKETSZ > 65536
 #define MAXPACKET      PACKETSZ
 #else
-#define MAXPACKET      1024
+#define MAXPACKET      65536
 #endif
 
-char *__hostalias __P((const char *));
-int h_errno;
+#define QUERYSIZE      (HFIXEDSZ + QFIXEDSZ + MAXCDNAME + 1)
+
+static int
+__libc_res_nquerydomain(res_state statp, const char *name, const char *domain,
+                       int class, int type, u_char *answer, int anslen,
+                       u_char **answerp);
 
 /*
  * Formulate a normal query, send, and await answer.
@@ -95,73 +104,86 @@ int h_errno;
  * Perform preliminary check of answer, returning success only
  * if no error is indicated and the answer count is nonzero.
  * Return the size of the response on success, -1 on error.
- * Error number is left in h_errno.
+ * Error number is left in H_ERRNO.
  *
  * Caller must parse answer and determine whether it answers the question.
  */
 int
-res_query(name, class, type, answer, anslen)
-       const char *name;       /* domain name */
-       int class, type;        /* class and type of query */
-       u_char *answer;         /* buffer to put answer */
-       int anslen;             /* size of answer buffer */
+__libc_res_nquery(res_state statp,
+                 const char *name,     /* domain name */
+                 int class, int type,  /* class and type of query */
+                 u_char *answer,       /* buffer to put answer */
+                 int anslen,           /* size of answer buffer */
+                 u_char **answerp)     /* if buffer needs to be enlarged */
 {
-       u_char buf[MAXPACKET];
-       register HEADER *hp = (HEADER *) answer;
-       int n;
+       u_char *buf;
+       HEADER *hp = (HEADER *) answer;
+       int n, use_malloc = 0;
 
        hp->rcode = NOERROR;    /* default */
 
-       if ((_res.options & RES_INIT) == 0 && res_init() == -1) {
-               h_errno = NETDB_INTERNAL;
-               return (-1);
-       }
+       buf = alloca (QUERYSIZE);
+
 #ifdef DEBUG
-       if (_res.options & RES_DEBUG)
+       if (statp->options & RES_DEBUG)
                printf(";; res_query(%s, %d, %d)\n", name, class, type);
 #endif
 
-       n = res_mkquery(QUERY, name, class, type, NULL, 0, NULL,
-                       buf, sizeof(buf));
-       if (n <= 0) {
+       n = res_nmkquery(statp, QUERY, name, class, type, NULL, 0, NULL,
+                        buf, QUERYSIZE);
+       if (__builtin_expect (n <= 0, 0)) {
+               /* Retry just in case res_nmkquery failed because of too
+                  short buffer.  Shouldn't happen.  */
+               buf = malloc (MAXPACKET);
+               if (buf != NULL) {
+                       use_malloc = 1;
+                       n = res_nmkquery(statp, QUERY, name, class, type, NULL,
+                                        0, NULL, buf, MAXPACKET);
+               }               
+       }
+       if (__builtin_expect (n <= 0, 0)) {
 #ifdef DEBUG
-               if (_res.options & RES_DEBUG)
+               if (statp->options & RES_DEBUG)
                        printf(";; res_query: mkquery failed\n");
 #endif
-               h_errno = NO_RECOVERY;
+               RES_SET_H_ERRNO(statp, NO_RECOVERY);
+               if (use_malloc)
+                       free (buf);
                return (n);
        }
-       n = res_send(buf, n, answer, anslen);
+       n = __libc_res_nsend(statp, buf, n, answer, anslen, answerp);
+       if (use_malloc)
+               free (buf);
        if (n < 0) {
 #ifdef DEBUG
-               if (_res.options & RES_DEBUG)
+               if (statp->options & RES_DEBUG)
                        printf(";; res_query: send error\n");
 #endif
-               h_errno = TRY_AGAIN;
+               RES_SET_H_ERRNO(statp, TRY_AGAIN);
                return (n);
        }
 
        if (hp->rcode != NOERROR || ntohs(hp->ancount) == 0) {
 #ifdef DEBUG
-               if (_res.options & RES_DEBUG)
+               if (statp->options & RES_DEBUG)
                        printf(";; rcode = %d, ancount=%d\n", hp->rcode,
                            ntohs(hp->ancount));
 #endif
                switch (hp->rcode) {
                case NXDOMAIN:
-                       h_errno = HOST_NOT_FOUND;
+                       RES_SET_H_ERRNO(statp, HOST_NOT_FOUND);
                        break;
                case SERVFAIL:
-                       h_errno = TRY_AGAIN;
+                       RES_SET_H_ERRNO(statp, TRY_AGAIN);
                        break;
                case NOERROR:
-                       h_errno = NO_DATA;
+                       RES_SET_H_ERRNO(statp, NO_DATA);
                        break;
                case FORMERR:
                case NOTIMP:
                case REFUSED:
                default:
-                       h_errno = NO_RECOVERY;
+                       RES_SET_H_ERRNO(statp, NO_RECOVERY);
                        break;
                }
                return (-1);
@@ -169,55 +191,71 @@ res_query(name, class, type, answer, anslen)
        return (n);
 }
 
+int
+res_nquery(res_state statp,
+          const char *name,    /* domain name */
+          int class, int type, /* class and type of query */
+          u_char *answer,      /* buffer to put answer */
+          int anslen)          /* size of answer buffer */
+{
+       return __libc_res_nquery(statp, name, class, type, answer, anslen,
+                                NULL);
+}
+
 /*
  * Formulate a normal query, send, and retrieve answer in supplied buffer.
  * Return the size of the response on success, -1 on error.
  * If enabled, implement search rules until answer or unrecoverable failure
- * is detected.  Error code, if any, is left in h_errno.
+ * is detected.  Error code, if any, is left in H_ERRNO.
  */
 int
-res_search(name, class, type, answer, anslen)
-       const char *name;       /* domain name */
-       int class, type;        /* class and type of query */
-       u_char *answer;         /* buffer to put answer */
-       int anslen;             /* size of answer */
+__libc_res_nsearch(res_state statp,
+           const char *name,   /* domain name */
+           int class, int type,        /* class and type of query */
+           u_char *answer,     /* buffer to put answer */
+           int anslen,         /* size of answer */
+           u_char **answerp)
 {
-       register const char *cp, * const *domain;
+       const char *cp, * const *domain;
        HEADER *hp = (HEADER *) answer;
+       char tmp[NS_MAXDNAME];
        u_int dots;
        int trailing_dot, ret, saved_herrno;
-       int got_nodata = 0, got_servfail = 0, tried_as_is = 0;
+       int got_nodata = 0, got_servfail = 0, root_on_list = 0;
+       int tried_as_is = 0;
+
+       __set_errno (0);
+       RES_SET_H_ERRNO(statp, HOST_NOT_FOUND);  /* True if we never query. */
 
-       if ((_res.options & RES_INIT) == 0 && res_init() == -1) {
-               h_errno = NETDB_INTERNAL;
-               return (-1);
-       }
-       errno = 0;
-       h_errno = HOST_NOT_FOUND;       /* default, if we never query */
        dots = 0;
-       for (cp = name; *cp; cp++)
+       for (cp = name; *cp != '\0'; cp++)
                dots += (*cp == '.');
        trailing_dot = 0;
        if (cp > name && *--cp == '.')
                trailing_dot++;
 
-       /*
-        * if there aren't any dots, it could be a user-level alias
-        */
-       if (!dots && (cp = __hostalias(name)) != NULL)
-               return (res_query(cp, class, type, answer, anslen));
+       /* If there aren't any dots, it could be a user-level alias. */
+       if (!dots && (cp = res_hostalias(statp, name, tmp, sizeof tmp))!= NULL)
+               return (__libc_res_nquery(statp, cp, class, type, answer,
+                                         anslen, answerp));
 
        /*
-        * If there are dots in the name already, let's just give it a try
-        * 'as is'.  The threshold can be set with the "ndots" option.
+        * If there are enough dots in the name, let's just give it a
+        * try 'as is'. The threshold can be set with the "ndots" option.
+        * Also, query 'as is', if there is a trailing dot in the name.
         */
        saved_herrno = -1;
-       if (dots >= _res.ndots) {
-               ret = res_querydomain(name, NULL, class, type, answer, anslen);
-               if (ret > 0)
+       if (dots >= statp->ndots || trailing_dot) {
+               ret = __libc_res_nquerydomain(statp, name, NULL, class, type,
+                                             answer, anslen, answerp);
+               if (ret > 0 || trailing_dot)
                        return (ret);
                saved_herrno = h_errno;
                tried_as_is++;
+               if (answerp && *answerp != answer) {
+                       answer = *answerp;
+                       anslen = MAXPACKET;
+               }
        }
 
        /*
@@ -226,19 +264,29 @@ res_search(name, class, type, answer, anslen)
         *      - there is at least one dot, there is no trailing dot,
         *        and RES_DNSRCH is set.
         */
-       if ((!dots && (_res.options & RES_DEFNAMES)) ||
-           (dots && !trailing_dot && (_res.options & RES_DNSRCH))) {
+       if ((!dots && (statp->options & RES_DEFNAMES) != 0) ||
+           (dots && !trailing_dot && (statp->options & RES_DNSRCH) != 0)) {
                int done = 0;
 
-               for (domain = (const char * const *)_res.dnsrch;
+               for (domain = (const char * const *)statp->dnsrch;
                     *domain && !done;
                     domain++) {
 
-                       ret = res_querydomain(name, *domain, class, type,
-                                             answer, anslen);
+                       if (domain[0][0] == '\0' ||
+                           (domain[0][0] == '.' && domain[0][1] == '\0'))
+                               root_on_list++;
+
+                       ret = __libc_res_nquerydomain(statp, name, *domain,
+                                                     class, type,
+                                                     answer, anslen, answerp);
                        if (ret > 0)
                                return (ret);
 
+                       if (answerp && *answerp != answer) {
+                               answer = *answerp;
+                               anslen = MAXPACKET;
+                       }
+
                        /*
                         * If no server present, give up.
                         * If name isn't found in this domain,
@@ -253,11 +301,11 @@ res_search(name, class, type, answer, anslen)
                         * fully-qualified.
                         */
                        if (errno == ECONNREFUSED) {
-                               h_errno = TRY_AGAIN;
+                               RES_SET_H_ERRNO(statp, TRY_AGAIN);
                                return (-1);
                        }
 
-                       switch (h_errno) {
+                       switch (statp->res_h_errno) {
                        case NO_DATA:
                                got_nodata++;
                                /* FALLTHROUGH */
@@ -279,59 +327,70 @@ res_search(name, class, type, answer, anslen)
                        /* if we got here for some reason other than DNSRCH,
                         * we only wanted one iteration of the loop, so stop.
                         */
-                       if (!(_res.options & RES_DNSRCH))
+                       if ((statp->options & RES_DNSRCH) == 0)
                                done++;
                }
        }
 
-       /* if we have not already tried the name "as is", do that now.
-        * note that we do this regardless of how many dots were in the
-        * name or whether it ends with a dot.
+       /*
+        * If the name has any dots at all, and no earlier 'as-is' query
+        * for the name, and "." is not on the search list, then try an as-is
+        * query now.
         */
-       if (!tried_as_is) {
-               ret = res_querydomain(name, NULL, class, type, answer, anslen);
+       if (statp->ndots && !(tried_as_is || root_on_list)) {
+               ret = __libc_res_nquerydomain(statp, name, NULL, class, type,
+                                             answer, anslen, answerp);
                if (ret > 0)
                        return (ret);
        }
 
        /* if we got here, we didn't satisfy the search.
-        * if we did an initial full query, return that query's h_errno
+        * if we did an initial full query, return that query's H_ERRNO
         * (note that we wouldn't be here if that query had succeeded).
         * else if we ever got a nodata, send that back as the reason.
-        * else send back meaningless h_errno, that being the one from
+        * else send back meaningless H_ERRNO, that being the one from
         * the last DNSRCH we did.
         */
        if (saved_herrno != -1)
-               h_errno = saved_herrno;
+               RES_SET_H_ERRNO(statp, saved_herrno);
        else if (got_nodata)
-               h_errno = NO_DATA;
+               RES_SET_H_ERRNO(statp, NO_DATA);
        else if (got_servfail)
-               h_errno = TRY_AGAIN;
+               RES_SET_H_ERRNO(statp, TRY_AGAIN);
        return (-1);
 }
 
+int
+res_nsearch(res_state statp,
+           const char *name,   /* domain name */
+           int class, int type,        /* class and type of query */
+           u_char *answer,     /* buffer to put answer */
+           int anslen)         /* size of answer */
+{
+       return __libc_res_nsearch(statp, name, class, type, answer,
+                                 anslen, NULL);
+}
+
 /*
  * Perform a call on res_query on the concatenation of name and domain,
  * removing a trailing dot from name if domain is NULL.
  */
-int
-res_querydomain(name, domain, class, type, answer, anslen)
-       const char *name, *domain;
-       int class, type;        /* class and type of query */
-       u_char *answer;         /* buffer to put answer */
-       int anslen;             /* size of answer */
+static int
+__libc_res_nquerydomain(res_state statp,
+           const char *name,
+           const char *domain,
+           int class, int type,        /* class and type of query */
+           u_char *answer,             /* buffer to put answer */
+           int anslen,                 /* size of answer */
+           u_char **answerp)
 {
-       char nbuf[2*MAXDNAME+2];
+       char nbuf[MAXDNAME];
        const char *longname = nbuf;
-       int n;
+       int n, d;
 
-       if ((_res.options & RES_INIT) == 0 && res_init() == -1) {
-               h_errno = NETDB_INTERNAL;
-               return (-1);
-       }
 #ifdef DEBUG
-       if (_res.options & RES_DEBUG)
-               printf(";; res_querydomain(%s, %s, %d, %d)\n",
+       if (statp->options & RES_DEBUG)
+               printf(";; res_nquerydomain(%s, %s, %d, %d)\n",
                       name, domain?domain:"<Nil>", class, type);
 #endif
        if (domain == NULL) {
@@ -339,29 +398,49 @@ res_querydomain(name, domain, class, type, answer, anslen)
                 * Check for trailing '.';
                 * copy without '.' if present.
                 */
-               n = strlen(name) - 1;
-               if (n != (0 - 1) && name[n] == '.' && n < sizeof(nbuf) - 1) {
-                       bcopy(name, nbuf, n);
+               n = strlen(name);
+               if (n >= MAXDNAME) {
+                       RES_SET_H_ERRNO(statp, NO_RECOVERY);
+                       return (-1);
+               }
+               n--;
+               if (n >= 0 && name[n] == '.') {
+                       strncpy(nbuf, name, n);
                        nbuf[n] = '\0';
                } else
                        longname = name;
-       } else
-               sprintf(nbuf, "%.*s.%.*s", MAXDNAME, name, MAXDNAME, domain);
-
-       return (res_query(longname, class, type, answer, anslen));
+       } else {
+               n = strlen(name);
+               d = strlen(domain);
+               if (n + d + 1 >= MAXDNAME) {
+                       RES_SET_H_ERRNO(statp, NO_RECOVERY);
+                       return (-1);
+               }
+               sprintf(nbuf, "%s.%s", name, domain);
+       }
+       return (__libc_res_nquery(statp, longname, class, type, answer,
+                                 anslen, answerp));
 }
 
-char *
-__hostalias(name)
-       register const char *name;
+int
+res_nquerydomain(res_state statp,
+           const char *name,
+           const char *domain,
+           int class, int type,        /* class and type of query */
+           u_char *answer,             /* buffer to put answer */
+           int anslen)         /* size of answer */
 {
-       register char *cp1, *cp2;
-       FILE *fp;
-       char *file;
+       return __libc_res_nquerydomain(statp, name, domain, class, type,
+                                      answer, anslen, NULL);
+}
+
+const char *
+res_hostalias(const res_state statp, const char *name, char *dst, size_t siz) {
+       char *file, *cp1, *cp2;
        char buf[BUFSIZ];
-       static char abuf[MAXDNAME];
+       FILE *fp;
 
-       if (_res.options & RES_NOALIASES)
+       if (statp->options & RES_NOALIASES)
                return (NULL);
        file = getenv("HOSTALIASES");
        if (file == NULL || (fp = fopen(file, "r")) == NULL)
@@ -374,17 +453,18 @@ __hostalias(name)
                if (!*cp1)
                        break;
                *cp1 = '\0';
-               if (!strcasecmp(buf, name)) {
+               if (ns_samename(buf, name) == 1) {
                        while (isspace(*++cp1))
                                ;
                        if (!*cp1)
                                break;
                        for (cp2 = cp1 + 1; *cp2 && !isspace(*cp2); ++cp2)
                                ;
-                       abuf[sizeof(abuf) - 1] = *cp2 = '\0';
-                       strncpy(abuf, cp1, sizeof(abuf) - 1);
+                       *cp2 = '\0';
+                       strncpy(dst, cp1, siz - 1);
+                       dst[siz - 1] = '\0';
                        fclose(fp);
-                       return (abuf);
+                       return (dst);
                }
        }
        fclose(fp);