TCPServer.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_SERVER_H__INCL__
#define __NETWORK__TCP_SERVER_H__INCL__

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

extern const uint32 TCPSRV_ERRBUF_SIZE;
extern const uint32 TCPSRV_LOOP_GRANULARITY;

class BaseTCPServer
{
public:
    BaseTCPServer();
    virtual ~BaseTCPServer();

    uint16 GetPort() const { return mPort; }
    bool IsOpen() const;

    bool Open( uint16 port, char* errbuf = 0 );
    void Close();

protected:
    void StartLoop();
    void WaitLoop();

    virtual bool Process();
    void ListenNewConnections();

    virtual void CreateNewConnection( Socket* sock, uint32 rIP, uint16 rPort ) = 0;

    static void* TCPServerLoop( void* arg );
    void TCPServerLoop();

    mutable Mutex mMSock;
    Socket* mSock;
    uint16 mPort;

    mutable Mutex mMLoopRunning;
};

template<typename X>
class TCPServer : public BaseTCPServer
{
public:
    ~TCPServer()
    {
        MutexLock lock( mMQueue );

        X* conn(nullptr);
        while( ( conn = PopConnection() ) )
            SafeDelete( conn );
    }

    X* PopConnection()
    {
        MutexLock lock( mMQueue );

        X* ret(nullptr);
        if( !mQueue.empty() )
        {
            ret = mQueue.front();
            mQueue.pop();
        }

        return ret;
    }

protected:
    void AddConnection( X* con )
    {
        MutexLock lock( mMQueue );

        mQueue.push( con );
    }

    Mutex mMQueue;
    std::queue<X*> mQueue;
};

#endif /* !__NETWORK__TCP_SERVER_H__INCL__ */