raw-broadcast.cpp

Go to the documentation of this file.

/*
 * raw-broadcast.cpp
 *
 * Basic code to test raw UDP broadcasting
 */

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

#include "common/wave_ex.h"
#include "netlib/wavesock.h"
#include "perf/perf.h"



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

static void
throwError
(
IN const char * msg
)
{
        ASSERT(msg, "null");

        const int bufsize = 1024;
        char buffer[bufsize];

        netlib::wsGetErrorMessage(buffer, bufsize);

        WAVE_EX(wex);
        wex << msg << "\n";
        wex << buffer;
}



static void
doBroadcast
(
IN const char * server,
IN int port
)
{
        ASSERT2(server, "null");
        ASSERT2(port > 0, "Bad port: " << port);

        // look up our own address!
        DPRINTF("Looking up server: '%s'", server);
        netlib::address_t a;
        if (!a.set(server, port)) {
                throwError("Failed to resolve server name");
        }
        a.dump("broadcast address");

        int s = netlib::wsCreateUdpSocket(true);
        if (s < 0) {
                throwError("Failed to create UDP broadcast socket");
        }

        const char * msg = "Hello world!";
        int len = strlen(msg);
        DPRINTF("Message (%d bytes): '%s'", len, msg);

        int nMessages = 100;
        for (int i = 0; i < nMessages; ++i) {
                int nBytes = netlib::wsSendTo(s, msg, len, a);
                DPRINTF("Sent %d bytes", nBytes);
                if (nBytes != len) {
                        throwError("Sent bad number of bytes!");
                }
                sleep(3);
        }
}



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

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

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

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

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

        perf::dumpTimingSummary(std::cerr);

        return 0;
}