res_search.c
上传用户:tsgydb
上传日期:2007-04-14
资源大小:10674k
文件大小:5k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /*
  2.  * Copyright (c) 1985, 1988 Regents of the University of California.
  3.  * 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.  * 3. All advertising materials mentioning features or use of this software
  14.  *   must display the following acknowledgement:
  15.  * This product includes software developed by the University of
  16.  * California, Berkeley and its contributors.
  17.  * 4. Neither the name of the University nor the names of its contributors
  18.  *   may be used to endorse or promote products derived from this software
  19.  *   without specific prior written permission.
  20.  *
  21.  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24.  * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31.  * SUCH DAMAGE.
  32.  */
  33. #if defined(LIBC_SCCS) && !defined(lint)
  34. /*static char *sccsid = "from: @(#)res_search.c 6.45 (Berkeley) 2/24/91";*/
  35. static char *rcsid = "$Id$";
  36. #endif /* LIBC_SCCS and not lint */
  37. #include <pthread.h>
  38. #include <string.h>
  39. #include <stdio.h>
  40. #include <stdlib.h>
  41. #include <resolv.h>
  42. #include <netdb.h>
  43. #include "res_internal.h"
  44. static char *search_aliases(const char *name, char *buf, int bufsize);
  45. int res_search(const char *name, int class, int type, unsigned char *answer,
  46.    int anslen)
  47. {
  48. struct res_data *data;
  49. const char *p;
  50. int num_dots, len, result, no_data = 0, error;
  51. char buf[2 * MAXDNAME + 2], *domain, **dptr, *alias;
  52. data = _res_init();
  53. if (!data)
  54. return -1;
  55. /* Count the dots in name, and get a pointer to the end of name. */
  56. num_dots = 0;
  57. for (p = name; *p; p++) {
  58. if (*p == '.')
  59. num_dots++;
  60. }
  61. len = p - name;
  62. /* If there aren't any dots, check to see if name is an alias for
  63.  * another host.  If so, try the resolved alias as a fully-qualified
  64.  * name. */
  65. alias = search_aliases(name, buf, sizeof(buf));
  66. if (alias != NULL)
  67. return res_query(alias, class, type, answer, anslen);
  68. /* If there's a trailing dot, try to strip it off and query the name. */
  69. if (len > 0 && p[-1] == '.') {
  70. if (len > sizeof(buf)) {
  71. /* It's too long; just query the original name. */
  72. return res_query(name, class, type, answer, anslen);
  73. } else {
  74. /* Copy the name without the trailing dot and query. */
  75. memcpy(buf, name, len - 1);
  76. buf[len] = 0;
  77. return res_query(buf, class, type, answer, anslen);
  78. }
  79. }
  80. if (data->state.options & RES_DNSRCH) {
  81. /* If RES_DNSRCH is set, query all the domains until we get a
  82.  * definitive answer. */
  83. for (dptr = data->state.dnsrch; *dptr; dptr++) {
  84. domain = *dptr;
  85. sprintf(buf, "%.*s.%.*s", MAXDNAME, name, MAXDNAME, domain);
  86. result = res_query(buf, class, type, answer, anslen);
  87. if (result > 0)
  88. return result;
  89. if (data->errval == NO_DATA)
  90. no_data = 1;
  91. else if (data->errval != HOST_NOT_FOUND)
  92. break;
  93. }
  94. } else if (num_dots == 0 && data->state.options & RES_DEFNAMES) {
  95. /* If RES_DEFNAMES is set and there is no dot, query the default
  96.  * domain. */
  97. domain = data->state.defdname;
  98. sprintf(buf, "%.*s.%.%s", MAXDNAME, name, MAXDNAME, domain);
  99. result = res_query(buf, class, type, answer, anslen);
  100. if (result > 0)
  101. return result;
  102. if (data->errval == NO_DATA)
  103. no_data = 1;
  104. }
  105. /* If all the domain queries failed, try the name as fully-qualified.
  106.  * Only do this if there is at least one dot in the name. */
  107. if (num_dots > 0) {
  108. result = res_query(name, class, type, answer, anslen);
  109. if (result > 0)
  110. return result;
  111. }
  112. if (no_data)
  113. data->errval = NO_DATA;
  114. return -1;
  115. }
  116. static char *search_aliases(const char *name, char *buf, int bufsize)
  117. {
  118. FILE *fp;
  119. char *filename, *p;
  120. int len;
  121. filename = getenv("HOSTALIASES");
  122. if (filename == NULL)
  123. return NULL;
  124. fp = fopen(filename, "r");
  125. if (fp == NULL)
  126. return NULL;
  127. len = strlen(name);
  128. while (fgets(buf, bufsize, fp)) {
  129. /* Get the first word from the buffer. */
  130. p = buf;
  131. while (*p && !isspace(*p))
  132. p++;
  133. if (!*p)
  134. break;
  135. /* Null-terminate the first word and compare it with the name. */
  136. *p = 0;
  137. if (strcasecmp(buf, name) != 0)
  138. continue;
  139. p++;
  140. while (isspace(*p))
  141. p++;
  142. fclose(fp);
  143. return (*p) ? p : NULL;
  144. }
  145. fclose(fp);
  146. return NULL;
  147. }