tests/udstest.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 <fstream>
#include <sstream>

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

#include "sockconn.h"
#include "udslistener.h"
#include "udssockaddr.h"

/* Internal function prototypes */
static int uds_test();
static void accept_and_send_on_connection(Listener *listener,
        SocketAddressP &path);
static void *socket_sender_thread(void *path);
static void socket_sender(SocketAddressP &path);

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

    // Run some tests
    ret = uds_test();

    exit(0);
}

static int
uds_test()
{
    char *ptmp = strdup("/tmp/test-sock.XXXXXX");
    std::string path(mktemp(ptmp));
    free(ptmp);

    if (path.length() == 0) {
        cerr << "mktemp() failed to return a path, unable to continue." << endl;
        return(1);
    }

    SocketAddressP      addr(new UDSSocketAddress(path));

    cerr << "Opening a socket on \"" << path << "\" for listening." << endl;

    Listener *listener = new UDSListener(addr);

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

    listener->disconnect();
    delete listener;

    return(0);
}

void
accept_and_send_on_connection(Listener *listener, SocketAddressP &path)
{
    // Spawn a thread to make a connection to this socket.
    pthread_t tid_uds_sender;
    int ret = pthread_create(&tid_uds_sender, NULL, socket_sender_thread, &path);
    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_uds_sender, NULL);
}

void *
socket_sender_thread(void *path)
{
    socket_sender(*(SocketAddressP*)path);
    return NULL;
}

void
socket_sender(SocketAddressP &path)
{
    Connector *conn = new SocketConnector(path);

    Data b("TEST DATA 12345");
    int ret = conn->write(b);
    std::cout << "socket_sender wrote " << ret << " bytes" << std::endl;
    delete conn;
}

---