开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 代理服务器 > 查看源码
dnsserver.c
资源名称：squid-2.2.devel3-src.tar.gz　[点击查看]
上传用户：liugui
上传日期：2007-01-04
资源大小：822k
文件大小：8k
源码类别：

代理服务器

开发平台：

Unix_Linux

dnsserver.c：源码内容
  2. /*
  3.  * $Id: dnsserver.c,v 1.56 1999/01/13 23:24:12 wessels Exp $
  4.  *
  5.  * DEBUG: section 0     DNS Resolver
  6.  * AUTHOR: Harvest Derived
  7.  *
  8.  * SQUID Internet Object Cache  http://squid.nlanr.net/Squid/
  9.  * ----------------------------------------------------------
  10.  *
  11.  *  Squid is the result of efforts by numerous individuals from the
  12.  *  Internet community.  Development is led by Duane Wessels of the
  13.  *  National Laboratory for Applied Network Research and funded by the
  14.  *  National Science Foundation.  Squid is Copyrighted (C) 1998 by
  15.  *  Duane Wessels and the University of California San Diego.  Please
  16.  *  see the COPYRIGHT file for full details.  Squid incorporates
  17.  *  software developed and/or copyrighted by other sources.  Please see
  18.  *  the CREDITS file for full details.
  19.  *
  20.  *  This program is free software; you can redistribute it and/or modify
  21.  *  it under the terms of the GNU General Public License as published by
  22.  *  the Free Software Foundation; either version 2 of the License, or
  23.  *  (at your option) any later version.
  24.  *  
  25.  *  This program is distributed in the hope that it will be useful,
  26.  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  27.  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  28.  *  GNU General Public License for more details.
  29.  *  
  30.  *  You should have received a copy of the GNU General Public License
  31.  *  along with this program; if not, write to the Free Software
  32.  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111, USA.
  33.  *
  34.  */
  36. #include "config.h"
  38. #if HAVE_UNISTD_H
  39. #include <unistd.h>
  40. #endif
  41. #if HAVE_STDLIB_H
  42. #include <stdlib.h>
  43. #endif
  44. #if HAVE_STDIO_H
  45. #include <stdio.h>
  46. #endif
  47. #if HAVE_SYS_TYPES_H
  48. #include <sys/types.h>
  49. #endif
  50. #if HAVE_CTYPE_H
  51. #include <ctype.h>
  52. #endif
  53. #if HAVE_ERRNO_H
  54. #include <errno.h>
  55. #endif
  56. #if HAVE_FCNTL_H
  57. #include <fcntl.h>
  58. #endif
  59. #if HAVE_GRP_H
  60. #include <grp.h>
  61. #endif
  62. #if HAVE_GNUMALLOC_H
  63. #include <gnumalloc.h>
  64. #elif HAVE_MALLOC_H && !defined(_SQUID_FREEBSD_) && !defined(_SQUID_NEXT_)
  65. #include <malloc.h>
  66. #endif
  67. #if HAVE_MEMORY_H
  68. #include <memory.h>
  69. #endif
  70. #if HAVE_NETDB_H && !defined(_SQUID_NETDB_H_) /* protect NEXTSTEP */
  71. #define _SQUID_NETDB_H_
  72. #include <netdb.h>
  73. #endif
  74. #if HAVE_PWD_H
  75. #include <pwd.h>
  76. #endif
  77. #if HAVE_SIGNAL_H
  78. #include <signal.h>
  79. #endif
  80. #if HAVE_TIME_H
  81. #include <time.h>
  82. #endif
  83. #if HAVE_SYS_PARAM_H
  84. #include <sys/param.h>
  85. #endif
  86. #if HAVE_SYS_TIME_H
  87. #include <sys/time.h>
  88. #endif
  89. #if HAVE_SYS_RESOURCE_H
  90. #include <sys/resource.h> /* needs sys/time.h above it */
  91. #endif
  92. #if HAVE_SYS_SOCKET_H
  93. #include <sys/socket.h>
  94. #endif
  95. #if HAVE_NETINET_IN_H
  96. #include <netinet/in.h>
  97. #endif
  98. #if HAVE_ARPA_INET_H
  99. #include <arpa/inet.h>
  100. #endif
  101. #if HAVE_SYS_STAT_H
  102. #include <sys/stat.h>
  103. #endif
  104. #if HAVE_SYS_UN_H
  105. #include <sys/un.h>
  106. #endif
  107. #if HAVE_SYS_WAIT_H
  108. #include <sys/wait.h>
  109. #endif
  110. #if HAVE_LIBC_H
  111. #include <libc.h>
  112. #endif
  113. #ifdef HAVE_SYS_SYSCALL_H
  114. #include <sys/syscall.h>
  115. #endif
  116. #ifdef HAVE_STRING_H
  117. #include <string.h>
  118. #endif
  119. #ifdef HAVE_STRINGS_H
  120. #include <strings.h>
  121. #endif
  122. #if HAVE_BSTRING_H
  123. #include <bstring.h>
  124. #endif
  125. #ifdef HAVE_CRYPT_H
  126. #include <crypt.h>
  127. #endif
  128. #if HAVE_SYS_SELECT_H
  129. #include <sys/select.h>
  130. #endif
  132. #if HAVE_ARPA_NAMESER_H
  133. #include <arpa/nameser.h>
  134. #endif
  135. #if HAVE_RESOLV_H
  136. #include <resolv.h>
  137. #endif
  139. #include "util.h"
  140. #include "snprintf.h"
  142. #if !defined(_SQUID_AIX_)
  143. extern int h_errno;
  144. #endif
  146. #if LIBRESOLV_DNS_TTL_HACK
  147. extern int _dns_ttl_; /* this is a really *dirty* hack - bne */
  148. #endif
  150. #ifdef _SQUID_NEXT_
  151. /* This is a really bloody hack. frank@langen.bull.de
  152.  * Workaround bug in gethostbyname which sets h_errno wrong
  153.  * WARNING: This hack queries only the resolver and not NetInfo or YP
  154.  */
  155. struct hostent *_res_gethostbyname(char *name);
  156. #define gethostbyname _res_gethostbyname
  157. #endif /* _SQUID_NEXT_ */
  159. static struct in_addr no_addr;
  161. #ifdef _SQUID_OS2_
  162. struct state _res =
  163. {0}; /* it's not in any of the libraries */
  164. #endif
  166. /* error messages from gethostbyname() */
  167. static char *
  168. my_h_msgs(int x)
  169. {
  170.     if (x == HOST_NOT_FOUND)
  171. return "Host not found (authoritative)";
  172.     else if (x == TRY_AGAIN)
  173. return "Host not found (non-authoritative)";
  174.     else if (x == NO_RECOVERY)
  175. return "Non recoverable errors";
  176.     else if (x == NO_DATA || x == NO_ADDRESS)
  177. return "Valid name, no data record of requested type";
  178.     else
  179. return "Unknown DNS problem";
  180. }
  182. #define REQ_SZ 512
  184. static void
  185. lookup(const char *buf)
  186. {
  187.     const struct hostent *result = NULL;
  188.     int reverse = 0;
  189.     int ttl = 0;
  190.     int retry = 0;
  191.     int i;
  192.     struct in_addr addr;
  193.     if (0 == strcmp(buf, "$shutdown"))
  194. exit(0);
  195.     if (0 == strcmp(buf, "$hello")) {
  196. printf("$aliven");
  197. return;
  198.     }
  199.     /* check if it's already an IP address in text form. */
  200.     for (;;) {
  201. if (safe_inet_addr(buf, &addr)) {
  202.     reverse = 1;
  203.     result = gethostbyaddr((char *) &addr.s_addr, 4, AF_INET);
  204. } else {
  205.     result = gethostbyname(buf);
  206. }
  207. if (NULL != result)
  208.     break;
  209. if (h_errno != TRY_AGAIN)
  210.     break;
  211. if (++retry == 3)
  212.     break;
  213. sleep(1);
  214.     }
  215.     if (NULL == result) {
  216. if (h_errno == TRY_AGAIN) {
  217.     printf("$fail Name Server for domain '%s' is unavailable.n", buf);
  218. } else {
  219.     printf("$fail DNS Domain '%s' is invalid: %s.n",
  220. buf, my_h_msgs(h_errno));
  221. }
  222. return;
  223.     }
  224. #if LIBRESOLV_DNS_TTL_HACK
  225.     /* DNS TTL handling - bne@CareNet.hu
  226.      * for first try it's a dirty hack, by hacking getanswer
  227.      * to place the ttl in a global variable */
  228.     if (_dns_ttl_ > -1)
  229. ttl = _dns_ttl_;
  230. #endif
  231.     if (reverse) {
  232. printf("$name %d %sn", ttl, result->h_name);
  233. return;
  234.     }
  235.     printf("$addr %d", ttl);
  236.     for (i = 0; NULL != result->h_addr_list[i]; i++) {
  237. if (32 == i)
  238.     break;
  239. xmemcpy(&addr, result->h_addr_list[i], sizeof(addr));
  240. printf(" %s", inet_ntoa(addr));
  241.     }
  242.     printf("n");
  243. }
  245. static void
  246. usage(void)
  247. {
  248.     fprintf(stderr, "usage: dnsserver -Dhv -s nameservern"
  249. "t-D             Enable resolver RES_DEFNAMES and RES_DNSRCH optionsn"
  250. "t-h             Helpn"
  251. "t-v             Versionn"
  252. "t-s nameserver  Specify alternate name server(s).  'nameserver'n"
  253. "t               must be an IP address, -s option may be repeatedn");
  254. }
  256. int
  257. main(int argc, char *argv[])
  258. {
  259.     char request[512];
  260.     char *t = NULL;
  261.     int c;
  262.     int opt_s = 0;
  263.     extern char *optarg;
  265.     safe_inet_addr("255.255.255.255", &no_addr);
  267. #if HAVE_RES_INIT
  268.     res_init();
  269. #ifdef RES_DEFAULT
  270.     _res.options = RES_DEFAULT;
  271. #endif
  272. #ifdef RES_DEFNAMES
  273.     _res.options &= ~RES_DEFNAMES;
  274. #endif
  275. #ifdef RES_DNSRCH
  276.     _res.options &= ~RES_DNSRCH;
  277. #endif
  278. #endif
  280.     while ((c = getopt(argc, argv, "Dhs:v")) != -1) {
  281. switch (c) {
  282. case 'D':
  283. #ifdef RES_DEFNAMES
  284.     _res.options |= RES_DEFNAMES;
  285. #endif
  286. #ifdef RES_DNSRCH
  287.     _res.options |= RES_DNSRCH;
  288. #endif
  289.     break;
  290. case 's':
  291. #if HAVE_RES_INIT
  292.     if (opt_s == 0) {
  293. _res.nscount = 0;
  294. /*
  295.  * Setting RES_INIT here causes coredumps when -s is
  296.  * used with -D option.  It looks to me like setting
  297.  * RES_INIT is wrong.  The resolver code sets RES_INIT
  298.  * after calling res_init().  When we change the _res
  299.  * structure and set RES_INIT, some internal resolver
  300.  * structures get confused.             -DW 2.1.p1
  301.  */
  302. #if SEEMS_WRONG
  303. _res.options |= RES_INIT;
  304. #endif
  305. opt_s = 1;
  306.     } else if (_res.nscount == MAXNS) {
  307. fprintf(stderr, "Too many -s options, only %d are allowedn",
  308.     MAXNS);
  309. break;
  310.     }
  311. #if HAVE_RES_NSADDR_LIST
  312.     _res.nsaddr_list[_res.nscount] = _res.nsaddr_list[0];
  313.     safe_inet_addr(optarg, &_res.nsaddr_list[_res.nscount++].sin_addr);
  314. #elif HAVE_RES_NS_LIST
  315.     _res.ns_list[_res.nscount] = _res.ns_list[0];
  316.     safe_inet_addr(optarg, &_res.ns_list[_res.nscount++].addr.sin_addr);
  317. #else /* Unknown NS list format */
  318.     fprintf(stderr, "-s is not supported on this resolvern");
  319. #endif
  320. #else /* !HAVE_RES_INIT */
  321.     fprintf(stderr, "-s is not supported on this resolvern");
  322. #endif /* HAVE_RES_INIT */
  323.     break;
  324. case 'v':
  325.     printf("dnsserver version %sn", SQUID_VERSION);
  326.     exit(0);
  327.     break;
  328. case 'h':
  329. default:
  330.     usage();
  331.     exit(1);
  332.     break;
  333. }
  334.     }
  336.     for (;;) {
  337. memset(request, '', REQ_SZ);
  338. if (fgets(request, REQ_SZ, stdin) == NULL)
  339.     exit(1);
  340. t = strrchr(request, 'n');
  341. if (t == NULL) /* Ignore if no newline */
  342.     continue;
  343. *t = ''; /* strip NL */
  344. if ((t = strrchr(request, 'r')) != NULL)
  345.     *t = ''; /* strip CR */
  346. lookup(request);
  347. fflush(stdout);
  348.     }
  349.     /* NOTREACHED */
  350.     return 0;
  351. }