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

/*

 * socketstream.h

 *

 *  Created on: Dec 14, 2009

 *      Author: sam

 */

#ifndef SOCKETSTREAM_H_

#define SOCKETSTREAM_H_

#include <iostream>

#include <unistd.h>

#include <sys/types.h>

#include <sys/un.h>

#include <stdlib.h>

#include <sys/socket.h>

#include <arpa/inet.h>

#include <netdb.h>

#include <netinet/tcp.h>

#include <cstring>

#define NS_DEBUG

// #define NS_DEBUG_EXTRA

#define NS_MAX_PENDING 1

#define NS_HOST "local"

#define NS_PORT 0

using namespace std;

class SocketStream

{

public:

        int so; // the socket

        struct sockaddr_un addr;

        bool bIsAlive;

public:

        SocketStream() : so(-1), bIsAlive(false)

        {

        }

        // create from an existing socket

        SocketStream(int sock) : so(sock), bIsAlive(true)

        {

        }

        ~SocketStream()

        {

                if (so != -1)

                {

                        close(so);

                }

        }

        SocketStream(int sock, struct sockaddr *in_addr, socklen_t in_addrlen)

        {

                so = sock;

                bIsAlive = true;

                memcpy(&addr, in_addr, in_addrlen);

        }

        int listen(const char *host, int port);

        SocketStream* accept();

        int connect(const char *host, int port);

    // read from the socket

    int get(void *data, int number);

    // write to socket

    int put(const void *data, int number);

        bool isAlive()

        {

                return bIsAlive;

        }

};

#endif /* SOCKETSTREAM_H_ */