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_ */