tests/tcptest.cc

Go to the documentation of this file.

//
//
// ***** BEGIN LICENSE BLOCK *****
// Version: MPL 1.1/GPL 2.0/LGPL 2.1
// 
// The contents of this file are subject to the Mozilla Public License
// Version 1.1 (the "License"); you may not use this file except in
// compliance with the License. You may obtain a copy of the License at
// http://www.mozilla.org/MPL/
// 
// Software distributed under the License is distributed on an "AS IS"
// basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
// the License for the specific language governing rights and limitations
// under the License.
// 
// The Original Code is the consmgr network/serial-line monitoring package.
// 
// The Initial Developer of the Original Code is Chris P. Ross.
// Portions created by the Initial Developer are Copyright (C) 2000-2008
// the Initial Developer.  All Rights Reserved.
// 
// Contributor(s):
//   Geoff Adams
// 
// Alternatively, the contents of this file may be used under the terms of
// either the GNU General Public License Version 2 or later (the "GPL"), or
// the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
// in which case the provisions of the GPL or the LGPL are applicable instead
// of those above. If you wish to allow use of your version of this file only
// under the terms of either the GPL or the LGPL, and not to allow others to
// use your version of this file under the terms of the MPL, indicate your
// decision by deleting the provisions above and replace them with the notice
// and other provisions required by the GPL or the LGPL. If you do not delete
// the provisions above, a recipient may use your version of this file under
// the terms of any one of the MPL, the GPL or the LGPL.
// 
// ***** END LICENSE BLOCK *****

extern "C" {
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <arpa/inet.h>
}

#include <fstream>
#include <sstream>

#include "consmgr.h" // Doesn't compile without this, but I feel like it should.

#include "sockconn.h"
#include "ipsockaddr.h"
#include "socklistener.h"

/* Internal function prototypes */
static int tcp_test();
static void accept_and_send_on_connection(Listener *listener,
        SocketAddressP &where);
static void *socket_sender_thread(void *addr);
static void socket_sender(SocketAddressP &where);

int
main(int argc, char *argv[])
{
    int ret;

    // Run some tests
    ret = tcp_test();

    exit(0);
}

static int
tcp_test()
{
    struct sockaddr_storage     ss;
    struct sockaddr_in          *sin = reinterpret_cast<struct sockaddr_in*>(&ss);
    
    ss.ss_family = AF_INET;
#ifndef NO_SA_LEN
    ss.ss_len = sizeof(struct sockaddr_in);
#endif
    sin->sin_addr.s_addr = INADDR_ANY;
    // Pick a fairly random port. Can we have the OS choose an ephemeral one
    // for us?
    sin->sin_port = htons(12345);

    SocketAddressP      addr(new IPSocketAddress(reinterpret_cast<sockaddr*>(&ss)));

    cerr << "Opening a socket at INADDR_ANY for listening." << endl;

    Listener *listener = new SocketListener(addr);

    SocketAddressP      target_addr(new IPSocketAddress("127.0.0.1", ntohs(sin->sin_port)));

    // We test accepting connections twice in a row, to make sure the Listener
    // really works properly.
    accept_and_send_on_connection(listener, target_addr);
    accept_and_send_on_connection(listener, target_addr);

    listener->disconnect();
    delete listener;

    return(0);
}

void
accept_and_send_on_connection(Listener *listener, SocketAddressP &where)
{
    // Spawn a thread to make a connection to this socket.
    pthread_t tid_tcp_sender;
    int ret = pthread_create(&tid_tcp_sender, NULL, socket_sender_thread, &where);
    PTHREAD_CHECK_AND_THROW(ret, "pthread_create(socket_sender_thread)");

    // OK. All set. Now, listen for the incoming connection.
    ConnectorP incoming = listener->listen_for_incoming();

    cerr << "Got a connection on " << listener->get_id() << ": ";
    cerr << incoming->get_id() << endl;

    // Read from the Connector and send it to stdout, just so we can see it
    // working.
    try {
        while (1) {  // Wait for a closure exception to break us out.
            Data *b = incoming->read();
            std::cout << *b;
            delete b;
        }
    } catch (...) {
        std::cout << endl << "Connection closed." << endl;
    }

    ret = pthread_join(tid_tcp_sender, NULL);
}

// Probably senseless wrapper. But it seems nice to have a general-purpose
// TCP connect-and-send method, not tied to any threadiness.
void *
socket_sender_thread(void *addr)
{
    socket_sender(*(SocketAddressP*) addr);
    return NULL;
}

// Connect to a remote port, send some data on it, and close it.
// I should probably ammend this some time to take the data to write as
// a MessageP, or a Data*, or std::string, or something.
void
socket_sender(SocketAddressP &where)
{
    IPSocketAddress     &addr = dynamic_cast<IPSocketAddress&>(*where);

    cerr << "Building a SocketConnector to " << addr.get_addr() << " (port " << addr.port() << ")" << endl;
    Connector *conn = new SocketConnector(where);

    cerr << "The new SocketConnector is " << conn << endl;

    Data b("TEST DATA 45678");
    cerr << "Writing to the SocketConnector" << endl;
    conn->write(b);
    cerr << "Written: deleting it and returning" << endl;
    delete conn;
}

---