udp.cpp

Go to the documentation of this file.

/*
 * test_udp.cpp
 *
 * Copyright (C) 2008  Thomas A. Vaughan
 * All rights reserved.
 *
 * Testing of UDP send/receive
 */

// includes --------------------------------------------------------------------
#include <iostream>

#include "netlib/netlib.h"
#include "perf/perf.h"
#include "util/parsing.h"


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

static void
sendMessage
(
IN netlib::conn_id_t conn_id,
IN const std::string& msg
)
{
        ASSERT(conn_id, "null");
        const char * data = msg.c_str();
        if (!*data) {
                return;         // empty message...
        }

        smart_ptr<netlib::MessageBuffer> buffer =
            netlib::MessageBuffer::create();
        ASSERT(buffer, "null");

        buffer->appendToken(data);
        buffer->close();

        netlib::msg_t nmsg;
        nmsg.conn_id = conn_id;
        nmsg.msgbuf = buffer;

        ASSERT(netlib::sendMessage(nmsg),
            "Failed to send message!");
}



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

int
main
(
IN int argc,
IN const char  * argv[]
)
{
        ASSERT(4 == argc,
            "Usage: test_udp <local_port> <remote_server> <remote_port>");

        int local_port = atoi(argv[1]);
        const char * server = argv[2];
        int remote_port = atoi(argv[3]);

        netlib::conn_id_t local = netlib::createUdpLocal(local_port);
        ASSERT(local, "null");
        DPRINTF("Local UDP connection: 0x%04lx", local);

        netlib::conn_id_t remote =
            netlib::createUdpRemote(local, server, remote_port);
        ASSERT(remote, "null");
        DPRINTF("Remote UDP connection: 0x%04lx", remote);

        std::istream& stream = std::cin;

        while (stream.good()) {
                std::string line = getNextLineFromStream(stream, eParse_None);

                DPRINTF("Line = %s", line.c_str());

                sendMessage(remote, line);
                for (int i = 0; i < 5; ++i) {
                        netlib::msg_t msg = netlib::getNextMessage(5000);
                        if (!msg.is_empty()) {
                                DPRINTF("Got a message!");
                        }
                }
        }

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

        return 0;
}