Skip to content

Instantly share code, notes, and snippets.

@zh

zh/c10k-test-client.c

Forked from rsms/c10k-test-client.c
Created January 6, 2011 04:54
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save zh/767515 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save zh/767515 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. rsms created this gist Oct 4, 2009.
    115 changes: 115 additions & 0 deletions c10k-test-client.c
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,115 @@
    // HTTP client for testing high connection concurrency
    // Authors: Richard Jones and Rasmus Andersson
    // Released in the public domain. No restrictions, no support.
    #include <sys/types.h>
    #include <sys/time.h>
    #include <sys/queue.h>
    #include <stdlib.h>
    #include <err.h>
    #include <event.h>
    #include <evhttp.h>
    #include <unistd.h>
    #include <stdio.h>
    #include <sys/socket.h>
    #include <netinet/in.h>
    #include <time.h>
    #include <pthread.h>

    #define BUFSIZE 4096
    #define SLEEP_MS 10

    // options
    int num_conns = 100;

    // state
    int bytes_recvd = 0;
    int chunks_recvd = 0;
    int closed = 0;
    int connected = 0;
    int max_concurrency = 0;
    int responses_completed_ok = 0;

    // misc
    char buf[BUFSIZE];

    // called per chunk received
    void chunkcb(struct evhttp_request * req, void * arg) {
    int s = evbuffer_remove( req->input_buffer, &buf, BUFSIZE );
    //printf("Read %d bytes: %s\n", s, &buf);
    bytes_recvd += s;
    chunks_recvd++;
    }

    // gets called when request completes
    void reqcb(struct evhttp_request * req, void * arg) {
    int s = evbuffer_remove( req->input_buffer, &buf, BUFSIZE );
    bytes_recvd += s;
    chunks_recvd++;

    if (connected-closed > max_concurrency)
    max_concurrency = connected-closed;

    responses_completed_ok++;
    closed++;
    }

    int main(int argc, char * const *argv) {
    event_init();
    struct evhttp_connection *evhttp_connection;
    struct evhttp_request *evhttp_request;
    //char addr[16];
    const char *remote_addr = "217.213.5.37";
    int remote_port = 8088;
    const char *uri = "/msgq/listen?channel=pb";

    if (argc > 1 && (strcmp(argv[1], "-h") == 0 || strcmp(argv[1], "--help") == 0)) {
    fprintf(stderr, "%s [connections[ address[ port[ uri]]]]\n", argv[0]);
    return 1;
    }

    if (argc > 1) num_conns = atoi(argv[1]);
    if (argc > 2) remote_addr = argv[2];
    if (argc > 3) remote_port = atoi(argv[3]);
    if (argc > 4) uri = argv[4];

    if (num_conns > 65536) {
    // see http://www.mail-archive.com/[email protected]/msg01302.html
    // evhttp_connection_set_local_address can be used to send from different
    // local addresses.
    fprintf(stderr, "%s: connection count too high (>65536)\n", argv[0]);
    exit(1);
    }

    printf("Making %d connections to http://%s:%d%s\n", num_conns, remote_addr, remote_port, uri);

    int i;

    for (i=1;i<=num_conns;i++) {
    evhttp_connection = evhttp_connection_new(remote_addr, remote_port);
    evhttp_set_timeout((struct evhttp *)evhttp_connection, 864000); // 10 day timeout
    evhttp_request = evhttp_request_new(reqcb, NULL);
    evhttp_request->chunk_cb = chunkcb;
    evhttp_add_header(evhttp_request->output_headers, "Host", "hunch.se");
    evhttp_add_header(evhttp_request->output_headers, "Connection", "close");
    evhttp_make_request(evhttp_connection, evhttp_request, EVHTTP_REQ_GET, uri);
    connected++;
    if ( i % 100 == 0)
    printf("%d requests sent (%d connected)\n", i, connected-closed);
    evhttp_connection_set_timeout(evhttp_request->evcon, 864000);
    event_loop( EVLOOP_NONBLOCK );
    usleep(SLEEP_MS*1000);
    }

    printf("All %d requests sent (%d connected).\n", num_conns, connected);

    event_dispatch();

    printf("All connections are closed.\n");
    printf("connections: %d\tBytes: %d\tChunks: %d\tClosed: %d\n", num_conns, bytes_recvd, chunks_recvd, closed);
    printf("Completed: %d\tFailed: %d\n", responses_completed_ok, num_conns-chunks_recvd);
    printf("Max concurrency: %d\n", max_concurrency);

    return 0;
    }

    // gcc -o floodtest -levent -I/opt/local/include -L/opt/local/lib floodtest.c