YiTCPSocket.h

Go to the documentation of this file.

#ifndef _YI_TCPSOCKET_H_
#define _YI_TCPSOCKET_H_

#include "framework/YiPredef.h"
#include "network/YiSocket.h"
#include "utility/YiString.h"

class CYITCPSocketPriv;

class CYITCPSocket : public IYISocket
{
public:
    static std::unique_ptr<CYITCPSocket> Create(const CYIString &host, uint16_t uPort);

    static std::unique_ptr<CYITCPSocket> Create(int32_t nSocketDescriptor, const CYIString &host, uint16_t uPort);

    virtual ~CYITCPSocket();
    
    virtual bool Connect() override;
    virtual bool Disconnect() override;
    virtual bool IsConnected() const override;

    virtual IYISocket::SOCKET_RESULT Send(const char *pBuffer, size_t uLength) override;
    virtual IYISocket::SOCKET_RESULT Send(const char *pBuffer, size_t uLength, size_t &uSent) override;

    virtual IYISocket::SOCKET_RESULT Receive(char *pBuffer, size_t uLength, size_t &uReceived) override;

    virtual IYISocket::SOCKET_RESULT TryReceive(char *pBuffer, size_t uLength, size_t &uReceived);
    virtual IYISocket::SOCKET_RESULT TryReceive(char *pBuffer, size_t uLength, size_t &uReceived, int16_t uTimeout);

    virtual uint64_t GetTotalBytesSent() const override;
    virtual uint64_t GetTotalBytesReceived() const override;

    virtual void SetConnectionTimeoutMs(uint32_t uTimeoutMs) override;
    virtual uint32_t GetConnectionTimeoutMs() const override;

    bool SetSocketOptions(int32_t nLevel, int32_t nName, void *pOption, uint32_t uSize);

    bool SetSocketBlockingEnabled(bool bBlocking) override;
    
    bool Bind();
    
    bool IsBound() const;

    bool Listen(int32_t bBacklog) const;

    std::unique_ptr<CYITCPSocket> Accept() const;
    std::unique_ptr<CYITCPSocket> TryAccept() const;

protected:
    CYITCPSocket(const CYIString &host, uint16_t uPort);
    CYITCPSocket(int32_t nSocketDescriptor, const CYIString &host, uint16_t uPort);

private:
    YI_DISALLOW_COPY_AND_ASSIGN(CYITCPSocket);
    
    CYITCPSocketPriv *m_pPriv;
};

#endif /* _YI_TCPSOCKET_H_ */