TCP.h

Go to the documentation of this file.

//written by Luke Lenhart, 2008
//See /docs/License.txt for details on how this code may be used.

#pragma once

#include "../base/Setup.h"

#ifdef MPMA_COMPILE_NET

#include "Common.h"
#include <vector>

class MPMAMemoryManager; //FD

namespace NET
{
    class TCPClient
    {
    public:
        static TCPClient* Connect(const Address &addr, uint16 localPort=0);

        static void FreeClient(TCPClient *client);

        bool IsConnected() const;

        bool Send(const void *data, nuint dataLen);

        nuint Receive(std::vector<uint8> &data, nuint exactBytesToRetrieve=0);

        inline uint16 GetLocalPort() const { return port; }

        inline const Address& GetRemoteAddress() const { return remoteAddr; }

        void EnableSendCoalescing(bool allow);

        //TODO: Some sort of callback mechanism for notifications.

    private:
        //we are the only allocator
        TCPClient();
        inline ~TCPClient() {}
        inline TCPClient(const TCPClient&) {}
        inline void operator=(const TCPClient&) {}
        friend class ::MPMAMemoryManager;
        friend class TCPServer;

        //
        Address remoteAddr;
        uint16 port;
        nsint sock;
        mutable bool connected;
        std::vector<uint8> bufferedBytes;
    };

    class TCPServer
    {
    public:
        static TCPServer* Listen(uint16 port, bool openNatAutomatically=true);

        static void FreeServer(TCPServer *server);

        TCPClient* GetNextConnection();

        inline uint16 GetListenPort() const { return port; }

        //TODO: Some sort of callback mechanism for notifications.

    private:
        //we are the only allocator
        TCPServer();
        inline ~TCPServer() {}
        inline TCPServer(const TCPServer&) {}
        inline void operator=(const TCPServer&) {}
        friend class ::MPMAMemoryManager;

        //
        nsint sock;
        uint16 port;
        bool autoUPNP;
    };
}

#endif //#ifdef MPMA_COMPILE_NET