Subversion Repositories an-fpga-implementation-of-low-latency-noc-based-mpsoc

/*
 * socketstream.cpp
 *
 *  Created on: Dec 16, 2009
 *      Author: sam
 * 
 * Modified by Wenbo Dai, University of Toronto, Aug 12, 2013
 * use Unix Socket, instead of Internet Socket
 */

#include "socketstream.h"

#include <cstdio>
#include <cerrno>

int SocketStream::listen(const char *host, int port)
{
        char *socket_path = "./socket";
    // Create a socket
        if ( (so = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
                cout << "Error creating socket." << endl;
                return -1;
        }

        memset(&addr, 0, sizeof(addr));
        addr.sun_family = AF_UNIX;
        strncpy(addr.sun_path, socket_path, sizeof(addr.sun_path)-1);

        // Bind it to the listening port
        unlink(socket_path);
        if (bind(so, (struct sockaddr*)&addr, sizeof(addr)) != 0) {
                 cout << "Error binding socket." << endl;
                 return -1;
        }
        // Listen for connections
        if (::listen(so, NS_MAX_PENDING) != 0) {
                 cout << "Error listening on socket." << endl;
                 return -1;
        }
        bIsAlive = true;

#ifdef NS_DEBUG
        cout << "Listening on socket" << endl;
#endif

        return 0;
}

// accept a new connection
SocketStream* SocketStream::accept()
{
        struct sockaddr_un clientaddr;
        socklen_t clientaddrlen = sizeof clientaddr;
        int clientsock = ::accept(so, (struct sockaddr*)&clientaddr, &clientaddrlen);
        if ( clientsock < 0 ){
        cout << "Error accepting a connection";
        return NULL;
        }
        // prevent small packets from getting stuck in OS queues
        //int on = 1;
        //setsockopt (so, SOL_TCP, TCP_NODELAY, &on, sizeof (on));

        return new SocketStream(clientsock, (struct sockaddr*)&clientaddr, clientaddrlen);
}

int SocketStream::connect(const char *host, int port)
{
        char *socket_path = "./socket";
        // Create a socket.
        if ( (so = socket(AF_UNIX, SOCK_STREAM, 0)) < 0 ){
                cout << "Error creating socket." << endl;
                return -1;
        }
        memset(&addr, 0, sizeof(addr));
        addr.sun_family = AF_UNIX;
        strncpy(addr.sun_path, socket_path, sizeof(addr.sun_path)-1);

    // Connect to the server.
        if ( ::connect(so, (struct sockaddr*)&addr, sizeof(addr)) != 0) {
        cout << "Connection failed." << endl;
        return -1;
        }

    // prevent small packets from getting stuck in OS queues
    //int on = 1;
    //setsockopt (so, SOL_TCP, TCP_NODELAY, &on, sizeof (on));

        bIsAlive = true;

#ifdef NS_DEBUG
        cout << "Connected to host" << endl;
#endif

        return 0;
}

// read from the socket
int SocketStream::get(void *data, int number)
{

    int remaining = number;
    int received = 0;
    char *dataRemaining = (char*) data;

    errno = 0;
    while (remaining > 0 && (errno == 0 || errno == EINTR))
    {
        received = recv(so, dataRemaining, remaining, 0); // MSG_WAITALL
        if (received > 0)
        {
            dataRemaining += received;
            remaining -= received;
        }
    }

    return number - remaining;
}

// write to socket
int SocketStream::put(const void *data, int number)
{
    // MSG_NOSIGNAL prevents SIGPIPE signal from being generated on failed send
    return send(so, data, number, MSG_NOSIGNAL);
}