tcp_client.cpp

Go to the documentation of this file.

/*
 * test_tcp_client.cpp
 * 
 * Copyright (C) 2007  Thomas A. Vaughan
 * All rights reserved.
 *
 * Simple TCP client.
 */

// includes --------------------------------------------------------------------
#include "netlib/netlib.h"
#include "common/wave_ex.h"

#include "perf/perf.h"

#include <iostream>


////////////////////////////////////////////////////////////////////////////////
//
//      static helper methods
//
////////////////////////////////////////////////////////////////////////////////

static void
sendData
(
IN netlib::conn_id_t conn_id,
IN const char * data,
IN long bytes
)
{
        perf::Timer timer("sendData");
        ASSERT(conn_id, "null");
        ASSERT(data, "null");
        ASSERT(bytes > 0, "bad byte count: %ld", bytes);

        DPRINTF("Sending %ld bytes...", bytes);

        smart_ptr<netlib::MessageBuffer> mb = netlib::MessageBuffer::create();
        ASSERT(mb, "failed to create message buffer?");

        mb->appendData(data, bytes);
        mb->close();

        if (!netlib::enqueueMessage(conn_id, mb)) {
                WAVE_EX(wex);
                wex << "Failed to enqueue message";
        }
}



static void
sendMessage
(
IN netlib::conn_id_t conn_id
)
{
        ASSERT(conn_id, "Bad connection ID");

        const int bufsize = 10000;
        char buffer[bufsize];
        for (int i = 0; i < 3; ++i) {
                sprintf(buffer, "This is test message number %d", i);
                int len = strlen(buffer);

                sendData(conn_id, buffer, len);
        }

        for (int j = 0; j < 3; ++j) {
                int bytes = rand() % (bufsize - 1);
                bytes++;
                for (int i = 0; i < bytes; ++i) {
                        buffer[i] = 'A' + (i % 26);
                        if (0 == ((i + 1) % 80)) {
                                buffer[i] = '\n';
                        }
                }
                sendData(conn_id, buffer, bytes);
        }
}



////////////////////////////////////////////////////////////////////////////////
//
//      entry point
//
////////////////////////////////////////////////////////////////////////////////

int
main
(
IN int argc,
IN const char * argv[]
)
{
        ASSERT(3 == argc, "Usage: netlib-tcp_client <server> <port>");

        const char * server = argv[1];
        int port = atoi(argv[2]);
        ASSERT(port > 0, "Bad port: %d", port);

        netlib::address_t address(server, port);
        ASSERT(address.isValid(), "invalid address?");

        netlib::conn_id_t conn_id = netlib::createTcpConnection(address);
        ASSERT(conn_id, "Failed to create connection to %s:%d", server, port);

        for (int i = 0; i < 3000; ++i) {
                netlib::envelope_t envelope;
                smart_ptr<netlib::MessageBuffer> msgbuf;
                if (netlib::getNextMessage(1000, envelope, msgbuf)) {
                        netlib::dumpMessage(std::cerr,
                                "Received message from server",
                                envelope, msgbuf);
                }

                if (i < 5) {
                        sendMessage(conn_id);
                }
        }

        std::string summary;
        perf::getTimingSummary(summary);
        DPRINTF("Timers:\n%s", summary.c_str());

        netlib::dumpStats();
}