udp_broadcast.cpp

Go to the documentation of this file.

/*
 * udp_broadcast.cpp
 *
 * Copyright (C) 2009  Thomas A. Vaughan
 * All rights reserved.
 *
 * Testing of UDP broadcast (based on netlib, not raw)
 */

// 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 char * msg
)
{
        ASSERT(conn_id, "null");
        ASSERT(msg, "null");

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

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

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



static void
doBroadcast
(
IN int port,
IN const char * msg
)
{
        perf::Timer timer("doBroadcast");

        netlib::address_t address;
        address.setlocal(port);
        address.ip.setBroadcast();
        address.dump("Broadcast address");

        netlib::conn_id_t bcast = netlib::createUdpBroadcast(address);
        ASSERT2(bcast, "null");
        DPRINTF("broadcast UDP connection: 0x%04lx", (long) bcast);

        int nSends = 100;
        long nMicrosecondsPerIteration = 1000 * 1000;   // 1 second
        long nCallsPerIteration = 20;
        long nWait = nMicrosecondsPerIteration / nCallsPerIteration;
        for (int i = 0; i < nSends; ++i) {
                int percent = int((100.0 * i / nSends) + 0.5);
                DPRINTF("  Broadcasting (%2d%%)...", percent);

                sendMessage(bcast, msg);
                for (int i = 0; i < nCallsPerIteration; ++i) {
                        netlib::envelope_t envelope;
                        smart_ptr<netlib::MessageBuffer> msgbuf;
                        netlib::getNextMessage(nWait, envelope, msgbuf);
                }
        }
}



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

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

        int retval = 0;
        int remote_port = atoi(argv[1]);
        const char * msg = argv[2];

        try {
                perf::Timer timer("overall timer");

                doBroadcast(remote_port, msg);

        } catch (std::exception& e) {
                DPRINTF("EXCEPTION: %s", e.what());
                retval = 1;
        }

        perf::dumpTimingSummary(std::cerr);

        return retval;
}