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

/*

 * socketstream.cpp

 *

 *  Created on: Dec 16, 2009

 *      Author: sam

 */

#include "socketstream.h"

#include 

#include 

int SocketStream::listen(const char *host, int port)

{

    // Create a socket.

    so = socket(PF_INET, SOCK_STREAM, 0);

    if (so < 0) {

                cout << "Error creating socket." << endl;

                return -1;

    }

    // Allow listening port to be reused if defunct.

    int yes = 1;

    if (setsockopt(so, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(yes)) != 0) {

                cout << "Error configuring socket." << endl;

                return -1;

    }

        // Bind it to the listening port.

        memset(&addr, 0, sizeof addr);

        addr.sin_family = AF_INET;

        addr.sin_port = htons(port);

        inet_pton(AF_INET, host, &(addr.sin_addr));

        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_in 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)

{

    // Create a socket.

    so = socket(PF_INET, SOCK_STREAM, 0);

    if (so < 0) {

                cout << "Error creating socket." << endl;

                return -1;

    }

    // Get info about the server.

    struct addrinfo serveraddr, *res;

    memset(&serveraddr, 0, sizeof serveraddr);

    serveraddr.ai_family = AF_UNSPEC;

    serveraddr.ai_socktype = SOCK_STREAM;

    char portstr[NS_MAX_PORT_LEN];

    snprintf(portstr, sizeof portstr, "%d", port);

    int status = getaddrinfo(host, portstr, &serveraddr, &res);

    if (status != 0)

    {

        cout << "Host address not found" << endl;

        return -1;

    }

    // Connect to the server.

    status = ::connect(so, res->ai_addr, res->ai_addrlen);

    if (status != 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)

{

    //Socket soObj = so;

    //return soObj.get(data, 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

        // cout << "Got: " << received << endl;

        if (received > 0)

        {

            dataRemaining += received;

            remaining -= received;

        }

    }

    return number - remaining;

}

// write to socket

int SocketStream::put(const void *data, int number)

{

    //Socket soObj = so;

    //return soObj.put(data, number);

    // MSG_NOSIGNAL prevents SIGPIPE signal from being generated on failed send

    return send(so, data, number, MSG_NOSIGNAL);

}