kris/httperf
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Initial re-write of idleconn program to use libevent library

Browse Source
This commit is contained in:
tedbullock 2007-12-10 03:46:03 +00:00
parent c43251e5fe
commit c56788ae9e
1 changed files with 161 additions and 191 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

306
httperf/src/idleconn.c
View File

@ -1,64 +1,81 @@
/* /*
httperf -- a tool for measuring web server performance * Copyright (C) 2007 Ted Bullock <tbullock@canada.com>
Copyright 2000-2007 Hewlett-Packard Company and Contributors listed in *
AUTHORS file. Originally contributed by David Mosberger-Tang * This file is part of httperf, a web server performance measurment tool.
*
This file is part of httperf, a web server performance measurment * This program is free software; you can redistribute it and/or modify it
tool. * under the terms of the GNU General Public License as published by the Free
* Software Foundation; either version 2 of the License, or (at your option)
This program is free software; you can redistribute it and/or * any later version.
modify it under the terms of the GNU General Public License as *
published by the Free Software Foundation; either version 2 of the * In addition, as a special exception, the copyright holders give permission
License, or (at your option) any later version. * to link the code of this work with the OpenSSL project's "OpenSSL" library
* (or with modified versions of it that use the same license as the "OpenSSL"
In addition, as a special exception, the copyright holders give * library), and distribute linked combinations including the two. You must
permission to link the code of this work with the OpenSSL project's * obey the GNU General Public License in all respects for all of the code
"OpenSSL" library (or with modified versions of it that use the same * used other than "OpenSSL". If you modify this file, you may extend this
license as the "OpenSSL" library), and distribute linked combinations * exception to your version of the file, but you are not obligated to do so.
including the two. You must obey the GNU General Public License in * If you do not wish to do so, delete this exception statement from your
all respects for all of the code used other than "OpenSSL". If you * version.
modify this file, you may extend this exception to your version of the *
file, but you are not obligated to do so. If you do not wish to do * This program is distributed in the hope that it will be useful, but WITHOUT
so, delete this exception statement from your version. * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
This program is distributed in the hope that it will be useful, * more details.
but WITHOUT ANY WARRANTY; without even the implied warranty of *
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * You should have received a copy of the GNU General Public License along
General Public License for more details. * with this program; if not, write to the Free Software Foundation, Inc., 51
* Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
02110-1301, USA
*/ */
#define _BSD_SOURCE #include "config.h"
#include <errno.h>
#include <netdb.h>
#include <signal.h>
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h> /* For strrchr() */
#include <signal.h>
#include <sys/types.h>
#include <unistd.h> #include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/time.h> #include <sys/time.h>
#include <sys/resource.h> #include <event.h>
#include <evdns.h>
const char *prog_name; #include <generic_types.h>
unsigned long num_conn, num_closed; #include <list.h>
struct timeval start_time;
/*
* Events allocated onto the heap
*/
static struct List *active_events = NULL;
static struct List *inactive_events = NULL;
static const char *prog_name = NULL;
static unsigned long num_conn = 0, num_closed = 0;
static struct timeval start_time;
static char *server = NULL;
static int desired = 0;
static int port = 0;
static void
cleanup()
{
if (active_events != NULL) {
while (!is_list_empty(active_events)) {
Any_Type a = list_pop(active_events);
free(a.vp);
}
list_free(active_events);
}
}
void void
sigint_handler (int signal) sigint_exit(int fd, short event, void *arg)
{ {
struct event *signal_int = arg;
struct timeval stop_time; struct timeval stop_time;
double delta_t; double delta_t = 0;
gettimeofday(&stop_time, NULL); gettimeofday(&stop_time, NULL);
@ -67,20 +84,64 @@ sigint_handler (int signal)
printf("%s: total # conn. created = %lu, close() rate = %g conn/sec\n", printf("%s: total # conn. created = %lu, close() rate = %g conn/sec\n",
prog_name, num_conn, num_closed / delta_t); prog_name, num_conn, num_closed / delta_t);
exit (0);
printf("%s: caught SIGINT... Exiting.\n", __func__);
evdns_shutdown(0);
event_del(signal_int);
cleanup();
} }
int void
main (int argc, char **argv) dns_lookup_callback(int result, char type, int count, int ttl, void *addresses,
void *arg)
{ {
int desired, current = 0, port, sd, max_sd = 0, n, i; uint8_t i;
struct sockaddr_in sin, server_addr;
fd_set readable, rdfds;
struct rlimit rlimit;
struct hostent *he;
char *server;
signal (SIGINT, sigint_handler); if (result != DNS_ERR_NONE) {
printf("DNS Lookup: result(%d)\n",
evdns_err_to_string(result));
exit(EXIT_FAILURE);
}
printf("Resolved %s\n", server);
for (i = 0; i < count; ++i) {
uint32_t *address_list = (uint32_t *) addresses;
uint8_t oct[4];
oct[0] = (uint8_t) (address_list[i] & 0x000000ff);
oct[1] = (uint8_t) ((address_list[i] & 0x0000ff00) >> 8);
oct[2] = (uint8_t) ((address_list[i] & 0x00ff0000) >> 16);
oct[3] = (uint8_t) ((address_list[i] & 0xff000000) >> 24);
printf("\t(%u.%u.%u.%u)\n", oct[0], oct[1], oct[2], oct[3]);
}
}
void
main(int argc, char *argv[])
{
struct event signal_int;
active_events = list_create();
if (active_events == NULL)
goto init_failure;
inactive_events = list_create();
if (inactive_events == NULL)
goto init_failure;
event_init();
evdns_init();
/*
* Initalize one event
*/
event_set(&signal_int, SIGINT, EV_SIGNAL, sigint_exit, &signal_int);
event_add(&signal_int, NULL);
prog_name = strrchr(argv[0], '/'); prog_name = strrchr(argv[0], '/');
if (prog_name) if (prog_name)
@ -88,127 +149,36 @@ main (int argc, char **argv)
else else
prog_name = argv[0]; prog_name = argv[0];
memset (&rdfds, 0, sizeof (rdfds)); if (argc != 4) {
fprintf(stderr, "Usage: `%s server port numidle'\n",
if (argc != 4) prog_name);
{ goto init_failure;
fprintf (stderr, "Usage: %s server port numidle\n", prog_name);
exit (-1);
} }
server = argv[1]; server = argv[1];
port = atoi(argv[2]); port = atoi(argv[2]);
desired = atoi(argv[3]); desired = atoi(argv[3]);
/* boost open file limit to the max: */ printf("%s: Using libevent-%s for %s event notification system.\n"
if (getrlimit (RLIMIT_NOFILE, &rlimit) < 0) "Control-c to exit\n\n", prog_name, event_get_version(),
{ event_get_method());
fprintf (stderr, "%s: failed to get number of open file limit: %s",
prog_name, strerror (errno));
exit (1);
}
if (rlimit.rlim_max > FD_SETSIZE)
{
fprintf (stderr, "%s: warning: open file limit > FD_SETSIZE; "
"limiting max. # of open files to FD_SETSIZE\n", prog_name);
rlimit.rlim_max = FD_SETSIZE;
}
rlimit.rlim_cur = rlimit.rlim_max;
if (setrlimit (RLIMIT_NOFILE, &rlimit) < 0)
{
fprintf (stderr, "%s: failed to increase number of open file limit: %s",
prog_name, strerror (errno));
exit (1);
}
printf ("%s: creating and maintaining %d idle connections\n",
prog_name, desired);
memset (&server_addr, 0, sizeof (server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons (port);
he = gethostbyname (server);
if (he)
{
if (he->h_addrtype != AF_INET || he->h_length != sizeof (sin.sin_addr))
{
perror (server);
exit (-1);
}
memcpy (&server_addr.sin_addr, he->h_addr_list[0],
sizeof (server_addr.sin_addr));
}
else
if (!inet_aton (server, &server_addr.sin_addr))
{
fprintf (stderr, "%s: invalid server address %s\n", prog_name, server);
exit (-1);
}
gettimeofday(&start_time, NULL); gettimeofday(&start_time, NULL);
while (1) evdns_resolve_ipv4(server, 0, dns_lookup_callback, NULL);
{
while (current < desired)
{
/* create more idle connections */
sd = socket (AF_INET, SOCK_STREAM, 0);
if (sd < 0)
{
perror ("socket");
exit (-1);
}
sin = server_addr; event_dispatch();
if (connect (sd, (struct sockaddr *) &sin, sizeof (sin)) < 0)
{
printf("connect: %s\n", strerror (errno));
switch (errno)
{
case ECONNREFUSED:
/* wait for server to start up... */
sleep (1);
case ETIMEDOUT:
close (sd);
continue;
default: /*
perror ("connect"); * Event loop will only exit upon receiving SIGINT. Make sure we pass
exit (-1); * this on to the parent process
} */
} if (signal(SIGINT, SIG_DFL) < 0)
perror("signal");
else
kill(getpid(), SIGINT);
if (sd > max_sd) init_failure:
max_sd = sd; cleanup();
exit(EXIT_FAILURE);
#if DEBUG > 1
printf ("created %d\n", sd);
#endif
++num_conn;
++current;
FD_SET(sd, &rdfds);
}
readable = rdfds;
n = select (max_sd + 1, &readable, NULL, NULL, NULL);
for (i = 0; i <= max_sd; ++i)
{
if (FD_ISSET (i, &readable))
{
#if DEBUG > 1
printf ("closed %d\n", i);
#endif
close (i);
--current;
--n;
++num_closed;
FD_CLR(i, &rdfds);
}
if (n <= 0)
break;
}
}
} }