Socket.h

Go to the documentation of this file.

#ifndef SOCKET_H_
#define SOCKET_H_

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>
#include <unistd.h>
#include <arpa/inet.h>

#include "SocketException.h"

const int MAXHOSTNAME = 200;
const int MAXCONNECTIONS = 5;
const int MAXRECV = 500;
const int SIMPORT = 54327;

class Socket
{
public:

    virtual ~Socket();
    // Data Transimission
    int send( const char* buffer, size_t bufferLength ) const throw( SocketException& );

    int sendall( const char* buffer, size_t bufferLength );

    int recv( char* buffer, size_t bufferLength ) const throw( SocketException& );

    int recv( char* buffer, size_t bufferLength, int timeoutMillis ) throw( SocketException& );

    int setTimeout( const int milliseconds );
    int close();
    int getPort()
    {
        return addr_.sin_port;
    }

    int getDescriptor()
    {
        return sock_;
    }

protected:

    Socket();

    // Server initialization
    void create() throw( SocketException& );
    void bind( const int port ) throw( SocketException& );
    void listen() const throw( SocketException& );
    void accept( Socket& newSocket ) throw( SocketException& );

    void accept( Socket& newSocket, int timeoutMillis ) throw( SocketException& );

    // Client initialization
    void connect( const char* host, const int port )  throw( SocketException& );

    void connect( const char* host, const int port, int timeoutMillis )  throw( SocketException& );

    void setNonBlocking( const bool nonBlocking ) throw( SocketException& );

    bool isValid() const
    {
        return sock_ != -1;
    }

    void setTimeval( const int milliseconds, struct timeval& tv );

private:

    int sock_;
    sockaddr_in addr_;
    bool connected_;
};

#endif /*SOCKET_H_*/