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 "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, uint dataLen);

        uint Receive(std::vector<uint8> &data, uint 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;
        int 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;

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