From: wessels <>
Date: Tue, 14 Oct 1997 04:08:06 +0000 (+0000)
Subject: adding
X-Git-Tag: SQUID_3_0_PRE1~4802
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=735246e86fc1bb9e29e2e9701ee08a586ab438ea;p=thirdparty%2Fsquid.git

adding
---

diff --git a/test-suite/tcp-banger2.c b/test-suite/tcp-banger2.c
new file mode 100644
index 0000000000..b8661df39e
--- /dev/null
+++ b/test-suite/tcp-banger2.c
@@ -0,0 +1,246 @@
+#define FD_SETSIZE 1024
+
+#include <stdio.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <fcntl.h>
+#include <string.h>
+#include <strings.h>
+#include <sys/types.h>
+#include <sys/select.h>
+#include <signal.h>
+#include <time.h>
+#include <sys/time.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#include <errno.h>
+
+#define PROXY_PORT 3128
+#define PROXY_ADDR "127.0.0.1"
+#define MAX_FDS 1024
+#define READ_BUF_SZ 4096
+
+static int proxy_port = PROXY_PORT;
+static char *proxy_addr = PROXY_ADDR;
+static char *progname;
+static int reqpersec;
+static int nrequests;
+static int opt_ims = 0;
+static int max_connections = 64;
+
+typedef void (CB) (int, void *);
+
+struct _f {
+    CB *cb;
+    void *data;
+};
+
+struct _f FD[MAX_FDS];
+int nfds = 0;
+int maxfd = 0;
+
+
+char *
+mkrfc850(t)
+    time_t *t;
+{
+    static char buf[128];
+    struct tm *gmt = gmtime(t);
+    buf[0] = '\0';
+    (void) strftime(buf, 127, "%A, %d-%b-%y %H:%M:%S GMT", gmt);
+    return buf;
+}
+
+void
+fd_close(int fd)
+{
+    close(fd);
+    FD[fd].cb = NULL;
+    FD[fd].data = NULL;
+    nfds--;
+    if (fd == maxfd) {
+	while (FD[fd].cb == NULL)
+		fd--;
+	maxfd = fd;
+    }
+}
+
+void
+fd_open(int fd, CB *cb, void *data)
+{
+    FD[fd].cb = cb;
+    FD[fd].data = data;
+    if (fd > maxfd)
+	maxfd = fd;
+    nfds++;
+}
+
+void
+sig_intr(int sig)
+{
+	fd_close(0);
+	printf ("\rWaiting for open connections to finish...\n");
+	signal(sig, SIG_DFL);
+}
+
+void 
+read_reply(int fd, void *data)
+{
+	static char buf[READ_BUF_SZ];
+	if (read(fd, buf, READ_BUF_SZ) <= 0) {
+		fd_close(fd);
+		reqpersec++;
+		nrequests++;
+	}
+}
+
+int
+request(url)
+    char *url;
+{
+    int s;
+    char buf[4096];
+    int len;
+    time_t w;
+    struct sockaddr_in S;
+    if ((s = socket(PF_INET, SOCK_STREAM, 0)) < 0) {
+	perror("socket");
+	return -1;
+    }
+    memset(&S, '\0', sizeof(struct sockaddr_in));
+    S.sin_family = AF_INET;
+    S.sin_port = htons(proxy_port);
+    S.sin_addr.s_addr = inet_addr(proxy_addr);
+    if (connect(s, (struct sockaddr *) & S, sizeof(S)) < 0) {
+	close(s);
+	perror("connect");
+	return -1;
+    }
+    buf[0] = '\0';
+    strcat(buf, "GET ");
+    strcat(buf, url);
+    strcat(buf, " HTTP/1.0\r\n");
+    strcat(buf, "Accept: */*\r\n");
+    if (opt_ims && (lrand48() & 0x03) == 0) {
+	w = time(NULL) - (lrand48() & 0x3FFFF);
+	strcat(buf, "If-Modified-Since: ");
+	strcat(buf, mkrfc850(&w));
+	strcat(buf, "\r\n");
+    }
+    strcat(buf, "\r\n");
+    len = strlen(buf);
+    if (write(s, buf, len) < 0) {
+	close(s);
+	perror("write");
+	return -1;
+    }
+/*
+    if (fcntl(s, F_SETFL, O_NDELAY) < 0)
+	perror("fcntl O_NDELAY");
+*/
+    return s;
+}
+
+void
+read_url(int fd, void *junk)
+{
+    static char buf[8192];
+    char *t;
+    int s;
+    if (fgets(buf, 8191, stdin) == NULL) {
+	printf("Done Reading URLS\n");
+	fd_close(0);
+	return;
+    }
+    if ((t = strchr(buf, '\n')))
+	*t = '\0';
+    s = request(buf);
+    if (s < 0) {
+	max_connections = nfds-1;
+	printf("NOTE: max_connections set at %d\n", max_connections);
+    }
+    fd_open(s, read_reply, NULL);
+}
+
+void
+usage(void)
+{
+	fprintf(stderr, "usage: %s: -p port -h host -n max\n", progname);
+}
+
+int 
+main(argc, argv)
+    int argc;
+    char *argv[];
+{
+    int i;
+    int c;
+    int dt;
+    fd_set R;
+    struct timeval start;
+    struct timeval now;
+    struct timeval last;
+    struct timeval to;
+    setbuf(stdout, NULL);
+    setbuf(stderr, NULL);
+    progname = strdup(argv[0]);
+    gettimeofday(&start, NULL);
+    last = start;
+    while ((c = getopt(argc, argv, "p:h:n:i")) != -1) {
+        switch (c) {
+	case 'p':
+		proxy_port = atoi(optarg);
+		break;
+	case 'h':
+		proxy_addr = strdup(optarg);
+		break;
+	case 'n':
+		max_connections = atoi(optarg);
+		break;
+	case 'i':
+		opt_ims = 1;
+		break;
+	default:
+		usage();
+		return 1;
+        }
+    }
+    fd_open(0, read_url, NULL);
+    signal(SIGINT, sig_intr);
+    while (nfds) {
+	FD_ZERO(&R);
+	to.tv_sec = 0;
+	to.tv_usec = 100000;
+	if (nfds < max_connections && FD[0].cb)
+	    FD_SET(0, &R);
+	for (i = 1; i <= maxfd; i++) {
+	    if (FD[i].cb == NULL)
+		continue;
+	    FD_SET(i, &R);
+	}
+	if (select(maxfd + 1, &R, NULL, NULL, &to) < 0) {
+	    if (errno != EINTR)
+	        perror("select");
+	    continue;
+	}
+	for (i = 0; i <= maxfd; i++) {
+	    if (!FD_ISSET(i, &R))
+		continue;
+	    FD[i].cb(i, FD[i].data);
+	}
+	gettimeofday(&now, NULL);
+	if (now.tv_sec > last.tv_sec) {
+		last = now;
+		dt = (int) (now.tv_sec - start.tv_sec);
+		printf ("T+ %6d: %9d req (%+4d), %4d conn, %3d/sec avg\n",
+			dt,
+			nrequests,
+			reqpersec,
+			nfds,
+			(int) (nrequests / dt));
+		reqpersec = 0;
+	}
+    }
+    return 0;
+}