Socket.h

// $Id: Socket.h 21 2010-09-05 04:18:17Z cschwarz1 $

#ifndef BASE_SOCKET_H
#define BASE_SOCKET_H

#ifdef _WIN32
    #define WIN32_LEAN_AND_MEAN
    #include <winsock2.h>
#else
    #include <sys/socket.h>
    #include <sys/un.h>
    #include <netinet/in.h>
    #include <arpa/inet.h>
#endif
#include "String.h"
#include "IOException.h"

namespace base {

    class PPBASE_EXPORT Socket {
    public:
    #ifdef _WIN32
        typedef SOCKET socket_type;                               
        static const socket_type invalid_socket = INVALID_SOCKET; 
        typedef int socklen_type;                                 
    #else
        typedef int socket_type;                                  
        static const socket_type invalid_socket = -1;             
        typedef socklen_t socklen_type;                           
    #endif


        Socket(socket_type sd = invalid_socket, int tout_acc = 60, int tout_conn = 60, int tout_recv = 60, int tout_send = 60);

        ~Socket();


        Socket *accept();

        void close();


        void connect(struct sockaddr *addr, socklen_type len);


        void connect(const String &host, int port = 0);


        bool connected() const;


        void discard();


        bool eof() const;


        static void finit();

        void flush();


        socket_type getSD() const;


        String getLocalAddr(bool full = false);


        String getLocalIP();


        String getLocalIPPort();


        int getLocalPort();


        String getRemoteAddr(bool full = false);


        String getRemoteIP();


        String getRemoteIPPort();


        int getRemotePort();


        static void init();


        static socket_type openInet(int type = SOCK_STREAM, int port = -1, int listenq = -1, const char *ip = NULL);

    #ifndef _WIN32


        static socket_type openUnix(int type, int port = -1, int listenq = -1);


        static socket_type openUnix(int type = SOCK_STREAM, const char *path = NULL, int listenq = -1);
    #endif


        void open(int type = SOCK_STREAM, int port = -1, int listenq = -1, const char *ip = NULL);


        void open(int type, const char *addr, int listenq = -1);


        size_t read(char *buf, size_t len);


        size_t read(String &buf, size_t len);


        static void readEOF(int fd, String &ret, int timeout = 0);


        void readEOF(String &ret);


        bool readln(String &line, bool strip = false);


        static size_t readto(socket_type sd, char *buf, size_t len, int timeout);


        void relay(socket_type sd, int timeout = 0);


        void relay(Socket &sock, int timeout = 0);


        static void setBlocking(socket_type sd, bool blocking);
    

        void setBlocking(bool blocking);


        void setSD(socket_type sd);


        void setReadBuffer(size_t len);


        void setWriteBuffer(size_t len);


        void setTimeout(int tout_acc, int tout_conn, int tout_recv, int tout_send);


        void write(const char *buf, size_t len);


        void write(const String &s);


        static size_t writeto(socket_type sd, const char *buf, size_t len, int timeout);


        size_t writef(const char *fmt, ...)
        #if __GNUC__
            __attribute__((format(printf, 2, 3)))
        #endif
        ;


        size_t writevf(const char *fmt, va_list ap);

    private:
        bool checkLocalAddr();
        bool checkRemoteAddr();

        bool                  _connected;   
        bool                  _eof;         
        union {
          struct sockaddr      sa;          
          struct sockaddr_in   sin;         
        #ifndef _WIN32
          struct sockaddr_un   sun;         
        #endif
        }                     _localaddr;   
        char                 *_rdbuf;       
        size_t                _rdlen;       
        char                 *_rdptr;       
        size_t                _rdsize;      
        union {
          struct sockaddr      sa;          
          struct sockaddr_in   sin;         
        #ifndef _WIN32
          struct sockaddr_un   sun;         
        #endif
        }                     _remoteaddr;  
        socket_type           _sd;          
        bool                  _server;      
        int                   _tout_acc;    
        int                   _tout_conn;   
        int                   _tout_recv;   
        int                   _tout_send;   
        char                 *_wrbuf;       
        size_t                _wrlen;       
        size_t                _wrsize;      

        DISALLOW_COPY_CONSTRUCTOR_AND_ASSIGNMENT(Socket);
    };
}

#endif