Socket.h

Go to the documentation of this file.

#ifndef INCLUDED_Socket
#define INCLUDED_Socket

#if defined(_MSC_VER) && (_MSC_VER >= 1020)
# pragma once
#endif

#include <vector>

#ifdef _WIN32
# include <winsock2.h>
#else
# include <sys/socket.h>
# include <arpa/inet.h>
#endif

#ifdef CROSSBASE_API
# include <Core/clplatform.h>
# include <Trace/cltrace.h>
START_CL_NAMESPACE
#else
# include "Matilda.h"
#endif

class CROSSBASE_CLASS_API Socket
{
protected:
    const static int MAXBUF = 1024; 
#ifndef _WIN32
    typedef int SOCKET;
#endif

    ClTrace mTrace;             
    SOCKET  mSocket;            
    char    mBuf[MAXBUF];       
    int     mIdx;               
    int     mBufLen;            

public:
    int     mConnectTimeout;    
    int     mSendTimeout;       
    int     mRecvTimeout;       
    int     mBufferSize;        

private:
    Socket(const Socket &); 

    Socket & operator=(const Socket &); 

    void connectSocket(SOCKET aSocket, const sockaddr_in * aServer);
    void waitForConnect(SOCKET aSocket);
    void setSocketOptions(SOCKET aSocket);
    void setSocketBufferSize(SOCKET aSocket);
    void setSocketBlocking(SOCKET aSocket, bool aBlocking);
    int  setSocketTimeout(SOCKET aSocket, int aType, int aTimeout);

protected:
    int ReceiveBytes(char * a_pszBuf, int a_nBufSiz);

public:
    enum ErrorCategory {
        ERR_OTHER = 0,
        ERR_DNS,        // connection error
        ERR_CONNECT,    // connection error
        ERR_SEND,       // comms error
        ERR_RECV,       // comms error
    };

    class Exception : public std::exception { 
    public:
        ErrorCategory   mType;      
        int             mCode;      
        const char *    mDetail;    

        Exception(ErrorCategory aType = ERR_OTHER, int aCode = 0, const char * aDetail = "") 
            : mType(aType), mCode(aCode), mDetail(aDetail) 
        { }
    };

    static bool Resolve(const char * aHost, std::vector<std::string> & aAddress); // throw Exception

    static void GetLastError(
        int             aError,
        WideString &    aErrorMsg
        );

public:
    Socket(const ClTrace & aTrace);

    ~Socket(); 

    void Connect(const char * aIpAddress, int aPort); // throw Exception

    bool IsConnected() const;

    void Disconnect();

    void SendBytes(const char * a_pszBuf, size_t a_nBufSiz); // throw Exception

    int GetBytes(char * a_pszBuf, int a_nBufSiz); // throw Exception

    void DiscardBytes(int a_nBytes); // throw Exception

    char GetByte();

    void ReceiveLine(std::string & a_sLine, bool aTrim); // throw Exception
};

#ifdef CROSSBASE_API
END_CL_NAMESPACE
#endif

#endif // INCLUDED_Socket