tcp_server.cpp

Go to the documentation of this file.

/*
 * test_tcp_server.cpp
 *
 * Copyright (C) 2007  Thomas A. Vaughan
 * All rights reserved.
 *
 * Quick program to test a TCP server.
 */

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

#include <iostream>


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


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

int
main
(
IN int argc,
IN const char * argv[]
)
{
        ASSERT(2 == argc,
            "Usage: test_tcp_server <port>");

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

        netlib::address_t address;
        address.setlocal(port);
        address.dump("Using local address");

        netlib::conn_id_t conn_id = netlib::createTcpListener(address, 2);
        DPRINTF("Created listener socket with ID = 0x%lx", (long) conn_id);

        try {
                perf::Timer timer("server-loop");
                while (1) {
                        netlib::envelope_t envelope;
                        smart_ptr<netlib::MessageBuffer> msgbuf;
                        if (!netlib::getNextMessage(1000, envelope, msgbuf))
                                continue;

                        //DPRINTF("Received a message!");
                        netlib::dumpMessage(std::cerr, 
                                        "Received message from client",
                                        envelope, msgbuf);
        
                        char buffer[128];
                        sprintf(buffer, "I received %ld bytes from you!",
                            msgbuf->getBytes());
                        int len = strlen(buffer);
                        DPRINTF("Sending: %s", buffer);

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

                        if (!netlib::enqueueMessage(envelope.fromConnId, mb)) {
                                WAVE_EX(wex);
                                wex << "Failed to enqueue message";
                        }
                }
        } catch (std::exception& e) {
                DPRINTF("Exception! %s", e.what());
        } catch (...) {
                DPRINTF("Unknown exception!");
        }

        netlib::closeConnection(conn_id);

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

        netlib::dumpStats();

        return 0;
}