TCPConnection.h

Go to the documentation of this file.

/*
    ------------------------------------------------------------------------------------
    LICENSE:
    ------------------------------------------------------------------------------------
    This file is part of EVEmu: EVE Online Server Emulator
    Copyright 2006 - 2021 The EVEmu Team
    For the latest information visit https://evemu.dev
    ------------------------------------------------------------------------------------
    This program is free software; you can redistribute it and/or modify it under
    the terms of the GNU Lesser General Public License as published by the Free Software
    Foundation; either version 2 of the License, or (at your option) any later
    version.

    This program is distributed in the hope that it will be useful, but WITHOUT
    ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
    FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License along with
    this program; if not, write to the Free Software Foundation, Inc., 59 Temple
    Place - Suite 330, Boston, MA 02111-1307, USA, or go to
    http://www.gnu.org/copyleft/lesser.txt.
    ------------------------------------------------------------------------------------
    Author:     Zhur, Bloody.Rabbit
*/

#ifndef __NETWORK__TCP_CONNECTION_H__INCL__
#define __NETWORK__TCP_CONNECTION_H__INCL__

#include "network/Socket.h"
#include "threading/Mutex.h"
#include "utils/Buffer.h"

static const uint32 TCPCONN_ERRBUF_SIZE = 1024;
extern const uint32 TCPCONN_RECVBUF_SIZE;
extern const uint32 TCPCONN_LOOP_GRANULARITY;

class TCPConnection
{
public:
    enum state_t
    {
        STATE_DISCONNECTED, 
        STATE_CONNECTING,   
        STATE_CONNECTED,    
        STATE_DISCONNECTING 
    };

    TCPConnection();
    virtual ~TCPConnection();

    uint32 GetrIP() const     { return mrIP; }
    uint16 GetrPort() const { return mrPort; }
    std::string GetAddress();
    state_t GetState() const { return mSockState; }

    bool Connect( uint32 rIP, uint16 rPort, char* errbuf = 0 );
    void AsyncConnect( uint32 rIP, uint16 rPort );
    void Disconnect();

    bool Send( Buffer** data );

protected:
    TCPConnection( Socket* sock, uint32 rIP, uint16 rPort );

    void StartLoop();
    void WaitLoop();

    virtual bool Process();
    virtual bool ProcessReceivedData( char* errbuf = 0 ) = 0;

    virtual bool SendData( char* errbuf = 0 );
    virtual bool RecvData( char* errbuf = 0 );
    void DoDisconnect();

    virtual void ClearBuffers();

    static void* TCPConnectionLoop( void* arg );
    void TCPConnectionLoop();

    mutable Mutex mMSock;
    Socket* mSock;
    state_t mSockState;
    uint32 mrIP;
    uint16 mrPort;

    mutable Mutex mMLoopRunning;

    mutable Mutex mMSendQueue;
    std::deque<Buffer*> mSendQueue;

    Buffer* mRecvBuf;
};

#endif /* !__NETWORK__TCP_CONNECTION_H__INCL__ */