udp_client.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 "netlib/netlib.h"

#include "common/wave_ex.h"
#include "perf/perf.h"
#include "util/parsing.h"

#include <iostream>


////////////////////////////////////////////////////////////////////////////////
//
//      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();

        if (!netlib::enqueueMessage(conn_id, buffer)) {
                WAVE_EX(wex);
                wex << "Failed to enqueue 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::address_t addressLocal;
        addressLocal.setlocal(local_port);
        addressLocal.dump("local address");
        ASSERT(addressLocal.isValid(), "Invalid local address");

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

        netlib::address_t address(server, remote_port);
        address.dump("server address");
        ASSERT(address.isValid(), "Invalid server address");

        netlib::conn_id_t remote = netlib::createUdpRemote(local, address);
        ASSERT(remote, "null");
        DPRINTF("Remote UDP connection: 0x%04lx", (long) 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::envelope_t envelope;
                        smart_ptr<netlib::MessageBuffer> msgbuf;
                        if (netlib::getNextMessage(100, envelope, msgbuf)) {
                                netlib::dumpMessage(std::cerr,
                                    "Received message from UDP listener",
                                    envelope, msgbuf);
                        }
                }
        }

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

        return 0;
}