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

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

代理服务器

开发平台：

Unix_Linux

tcp-banger3.c：源码内容
  1. #include "config.h"
  3. /*
  4.  * On some systems, FD_SETSIZE is set to something lower than the
  5.  * actual number of files which can be opened.  IRIX is one case,
  6.  * NetBSD is another.  So here we increase FD_SETSIZE to our
  7.  * configure-discovered maximum *before* any system includes.
  8.  */
  9. #define CHANGE_FD_SETSIZE 1
  11. /* Cannot increase FD_SETSIZE on Linux */
  12. #if defined(_SQUID_LINUX_)
  13. #undef CHANGE_FD_SETSIZE
  14. #define CHANGE_FD_SETSIZE 0
  15. #endif
  17. /* Cannot increase FD_SETSIZE on FreeBSD before 2.2.0, causes select(2)
  18.  * to return EINVAL. */
  19. /* Marian Durkovic <marian@svf.stuba.sk> */
  20. /* Peter Wemm <peter@spinner.DIALix.COM> */
  21. #if defined(_SQUID_FREEBSD_)
  22. #include <osreldate.h>
  23. #if __FreeBSD_version < 220000
  24. #undef CHANGE_FD_SETSIZE
  25. #define CHANGE_FD_SETSIZE 0
  26. #endif
  27. #endif
  29. /* Increase FD_SETSIZE if SQUID_MAXFD is bigger */
  30. #if CHANGE_FD_SETSIZE && SQUID_MAXFD > DEFAULT_FD_SETSIZE
  31. #define FD_SETSIZE SQUID_MAXFD
  32. #endif
  34. #if HAVE_UNISTD_H
  35. #include <unistd.h>
  36. #endif
  37. #if HAVE_STDLIB_H
  38. #include <stdlib.h>
  39. #endif
  40. #if HAVE_STDIO_H
  41. #include <stdio.h>
  42. #endif
  43. #if HAVE_FCNTL_H
  44. #include <fcntl.h>
  45. #endif
  46. #ifdef HAVE_STRING_H
  47. #include <string.h>
  48. #endif
  49. #ifdef HAVE_STRINGS_H
  50. #include <strings.h>
  51. #endif
  52. #if HAVE_SYS_TYPES_H
  53. #include <sys/types.h>
  54. #endif
  55. #if HAVE_SYS_SELECT_H
  56. #include <sys/select.h>
  57. #endif
  58. #if HAVE_SIGNAL_H
  59. #include <signal.h>
  60. #endif
  61. #if HAVE_TIME_H
  62. #include <time.h>
  63. #endif
  64. #if HAVE_SYS_TIME_H
  65. #include <sys/time.h>
  66. #endif
  67. #if HAVE_SYS_SOCKET_H
  68. #include <sys/socket.h>
  69. #endif
  70. #if HAVE_NETINET_IN_H
  71. #include <netinet/in.h>
  72. #endif
  73. #if HAVE_ARPA_INET_H
  74. #include <arpa/inet.h>
  75. #endif
  76. #if HAVE_ERRNO_H
  77. #include <errno.h>
  78. #endif
  79. #if HAVE_SYS_STAT_H
  80. #include <sys/stat.h>
  81. #endif
  82. #if HAVE_ASSERT_H
  83. #include <assert.h>
  84. #endif
  85. #include <netdb.h>
  86. #include <sys/wait.h>
  88. #define READ_BUF_SZ 4096
  89. #define URL_BUF_SZ 4096
  91. struct _thing {
  92.     int rfd;
  93.     int wfd;
  94.     int state;
  95.     pid_t pid;
  96.     char url[URL_BUF_SZ];
  97.     struct _thing *next;
  98. };
  99. typedef struct _thing thing;
  101. static thing *things = NULL;
  102. static fd_set R1;
  103. static int maxfd = 0;
  104. static struct timeval now;
  105. #if DEBUG
  106. static int debug = 1;
  107. #else
  108. static int debug = 0;
  109. #endif
  111. int
  112. tvSubMsec(struct timeval t1, struct timeval t2)
  113. {
  114.     return (t2.tv_sec - t1.tv_sec) * 1000 + (t2.tv_usec - t1.tv_usec) / 1000;
  115. }
  117. static int
  118. get_url(const char *url)
  119. {
  120.     char host[URL_BUF_SZ];
  121.     char path[URL_BUF_SZ];
  122.     char request[URL_BUF_SZ];
  123.     char reply[READ_BUF_SZ];
  124.     char *t;
  125.     int x;
  126.     int s;
  127.     int nr = 0;
  128.     struct hostent *h;
  129.     struct sockaddr_in S;
  130.     unsigned short port = 80;
  131.     assert(!strncmp(url, "http://", 7));
  132.     strncpy(host, url + 7, URL_BUF_SZ);
  133.     if ((t = strchr(host, '/')))
  134. *t = '';
  135.     if ((t = strchr(host, ':'))) {
  136. *t = '';
  137. port = (unsigned short) atoi(t + 1);
  138.     }
  139. #if 0
  140.     if ((int) port != 80)
  141. return 0;
  142. #endif
  143.     t = strchr(url + 7, '/');
  144.     strncpy(path, (t ? t : "/"), URL_BUF_SZ);
  145.     memset(&S, '', sizeof(S));
  146.     h = gethostbyname(host);
  147.     if (!h)
  148. return 0;
  149.     memcpy(&S.sin_addr.s_addr, h->h_addr_list[0], sizeof(S.sin_addr.s_addr));
  150.     S.sin_port = htons(port);
  151.     S.sin_family = AF_INET;
  152.     if (debug)
  153. fprintf(stderr, "%s (%s) %d %sn", host, inet_ntoa(S.sin_addr), (int) port, path);
  154.     s = socket(PF_INET, SOCK_STREAM, 0);
  155.     if (s < 0) {
  156. perror("socket");
  157. return -errno;
  158.     }
  159.     x = connect(s, (struct sockaddr *) &S, sizeof(S));
  160.     if (x < 0) {
  161. perror(host);
  162. return -errno;
  163.     }
  164.     snprintf(request, URL_BUF_SZ,
  165. "GET %s HTTP/1.1rn"
  166. "Accept: */*rn"
  167. "Host: %srn"
  168. "Connection: closern"
  169. "rn",
  170. path,
  171. host);
  172.     x = write(s, request, strlen(request));
  173.     if (x < 0) {
  174. perror("write");
  175. return -errno;
  176.     }
  177.     do {
  178. x = read(s, reply, READ_BUF_SZ);
  179. if (x > 0)
  180.     nr += x;
  181.     } while (x > 0);
  182.     close(s);
  183.     return nr;
  184. }
  186. static void
  187. child_main_loop(void)
  188. {
  189.     char buf[URL_BUF_SZ];
  190.     char *t;
  191.     int n;
  192.     struct timeval t1;
  193.     struct timeval t2;
  194.     if (debug)
  195. fprintf(stderr, "Child PID %d entering child_main_loopn", (int) getpid());
  196.     setbuf(stdin, NULL);
  197.     setbuf(stdout, NULL);
  198.     setbuf(stderr, NULL);
  199.     while (fgets(buf, URL_BUF_SZ, stdin)) {
  200. t = strchr(buf, 'n');
  201. if (t == NULL)
  202.     continue;
  203. *t = '';
  204. if (strncmp(buf, "http://", 7))
  205.     continue;
  206. gettimeofday(&t1, NULL);
  207. n = get_url(buf);
  208. gettimeofday(&t2, NULL);
  209. printf("%d %dn", n, tvSubMsec(t1, t2));
  210.     }
  211. }
  213. static thing *
  214. create_a_thing(char *argv[])
  215. {
  216.     int p2c[2];
  217.     int c2p[2];
  218.     int prfd, pwfd, crfd, cwfd;
  219.     pid_t pid;
  220.     thing *t;
  221.     if (pipe(p2c) < 0)
  222. abort();
  223.     if (pipe(c2p) < 0)
  224. abort();
  225.     prfd = p2c[0];
  226.     cwfd = p2c[1];
  227.     crfd = c2p[0];
  228.     pwfd = c2p[1];
  229.     if ((pid = fork()) < 0)
  230. abort();
  231.     if (pid > 0) { /* parent */
  232. /* close shared socket with child */
  233. close(crfd);
  234. close(cwfd);
  235. t = calloc(1, sizeof(*t));
  236. t->wfd = pwfd;
  237. t->rfd = prfd;
  238. if (pwfd > maxfd)
  239.     maxfd = pwfd;
  240. if (prfd > maxfd)
  241.     maxfd = prfd;
  242. t->pid = pid;
  243. return t;
  244.     }
  245.     /* child */
  246.     close(prfd);
  247.     close(pwfd);
  248.     dup2(crfd, 0);
  249.     dup2(cwfd, 1);
  250.     close(crfd);
  251.     close(cwfd);
  252.     child_main_loop();
  253.     exit(0);
  254. }
  256. static void
  257. create_children(char *argv[])
  258. {
  259.     thing *t;
  260.     thing **T = &things;
  261.     int i;
  262.     for (i = 0; i < 20; i++) {
  263. t = create_a_thing(argv);
  264. assert(t);
  265. if (debug)
  266.     fprintf(stderr, "Thing #%d on FD %d/%dn",
  267. i, t->rfd, t->wfd);
  268. *T = t;
  269. T = &t->next;
  270.     }
  271. }
  273. char *
  274. parent_read_url(void)
  275. {
  276.     static char buf[URL_BUF_SZ];
  277.     while (fgets(buf, URL_BUF_SZ, stdin)) {
  278. if (strncmp(buf, "http://", 7))
  279.     continue;
  280. return buf;
  281.     }
  282.     return NULL;
  283. }
  285. static thing *
  286. get_idle_thing(void)
  287. {
  288.     thing *t;
  289.     thing *n = things;
  290.     while ((t = n)) {
  291. n = t->next;
  292. if (t->state == 0)
  293.     break;
  294.     }
  295.     return t;
  296. }
  298. static void
  299. dispatch(thing * t, char *url)
  300. {
  301.     int x;
  302.     char *s;
  303.     assert(t->state == 0);
  304.     x = write(t->wfd, url, strlen(url));
  305.     if (x < 0)
  306. perror("write");
  307.     if (debug)
  308. fprintf(stderr, "dispatched URL to thing PID %d, %d bytesn", (int) t->pid, x);
  309.     strncpy(t->url, url, URL_BUF_SZ);
  310.     if ((s = strchr(t->url, 'n')))
  311. *s = '';
  312.     t->state = 1;
  313.     FD_SET(t->rfd, &R1);
  314. }
  316. static void
  317. read_reply(thing * t)
  318. {
  319.     char buf[128];
  320.     int i;
  321.     int x;
  322.     int j;
  323.     x = read(t->rfd, buf, 128);
  324.     if (x < 0) {
  325. perror("read");
  326.     } else if (2 == sscanf(buf, "%d %d", &i, &j)) {
  327. gettimeofday(&now, NULL);
  328. printf("%d.%06d %9d %9d %sn", (int) now.tv_sec, (int) now.tv_usec, i, j, t->url);
  329.     }
  330.     t->state = 0;
  331.     FD_CLR(t->rfd, &R1);
  332. }
  334. static void
  335. parent_main_loop(void)
  336. {
  337.     thing *t;
  338.     char *url;
  339.     fd_set R2;
  340.     int x;
  341.     struct timeval to;
  342.     FD_ZERO(&R1);
  343.     for (;;) {
  344. while ((t = get_idle_thing()) && (url = parent_read_url()))
  345.     dispatch(t, url);
  346. R2 = R1;
  347. to.tv_sec = 60;
  348. to.tv_usec = 0;
  349. x = select(maxfd + 1, &R2, NULL, NULL, &to);
  350. if (x < 0) {
  351.     perror("select");
  352.     continue;
  353. } else if (x == 0) {
  354.     return;
  355. }
  356. for (t = things; t; t = t->next) {
  357.     if (t->state != 1)
  358. continue;
  359.     if (!FD_ISSET(t->rfd, &R2))
  360. continue;
  361.     read_reply(t);
  362. }
  363.     }
  364. }
  366. static void
  367. sig_child(int sig)
  368. {
  369.     int status;
  370.     pid_t pid;
  371.     do {
  372. pid = waitpid(-1, &status, WNOHANG);
  373.     } while (pid > 0 || (pid < 0 && errno == EINTR));
  374.     signal(sig, sig_child);
  375. }
  378. int
  379. main(int argc, char *argv[])
  380. {
  381.     int i;
  382.     signal(SIGCHLD, sig_child);
  383.     create_children(argv);
  384.     parent_main_loop();
  385.     for (i=3;i<=maxfd; i++)
  386. close(i);
  387.     sleep(1);
  388. }