NatPunchthroughServer.h

Go to the documentation of this file.

#include "NativeFeatureIncludes.h"
#if _RAKNET_SUPPORT_NatPunchthroughServer==1

#ifndef __NAT_PUNCHTHROUGH_SERVER_H
#define __NAT_PUNCHTHROUGH_SERVER_H

#include "RakNetTypes.h"
#include "Export.h"
#include "PluginInterface2.h"
#include "PacketPriority.h"
#include "SocketIncludes.h"
#include "DS_OrderedList.h"
#include "RakString.h"

class RakPeerInterface;
struct Packet;
#if _RAKNET_SUPPORT_PacketLogger==1
class PacketLogger;
#endif


struct NatPunchthroughServerDebugInterface
{
        NatPunchthroughServerDebugInterface() {}
        virtual ~NatPunchthroughServerDebugInterface() {}
        virtual void OnServerMessage(const char *msg)=0;
};

struct NatPunchthroughServerDebugInterface_Printf : public NatPunchthroughServerDebugInterface
{
        virtual void OnServerMessage(const char *msg);
};

#if _RAKNET_SUPPORT_PacketLogger==1

struct NatPunchthroughServerDebugInterface_PacketLogger : public NatPunchthroughServerDebugInterface
{
        // Set to non-zero to write to the packetlogger!
        PacketLogger *pl;

        NatPunchthroughServerDebugInterface_PacketLogger() {pl=0;}
        ~NatPunchthroughServerDebugInterface_PacketLogger() {}
        virtual void OnServerMessage(const char *msg);
};
#endif

class RAK_DLL_EXPORT NatPunchthroughServer : public PluginInterface2
{
public:
        // Constructor
        NatPunchthroughServer();

        // Destructor
        virtual ~NatPunchthroughServer();

        void SetDebugInterface(NatPunchthroughServerDebugInterface *i);

        virtual void Update(void);

        virtual PluginReceiveResult OnReceive(Packet *packet);

        virtual void OnClosedConnection(SystemAddress systemAddress, RakNetGUID rakNetGUID, PI2_LostConnectionReason lostConnectionReason );
        virtual void OnNewConnection(SystemAddress systemAddress, RakNetGUID rakNetGUID, bool isIncoming);

        // Each connected user has a ready state. Ready means ready for nat punchthrough.
        struct User;
        struct ConnectionAttempt
        {
                ConnectionAttempt() {sender=0; recipient=0; startTime=0; attemptPhase=NAT_ATTEMPT_PHASE_NOT_STARTED;}
                User *sender, *recipient;
                uint16_t sessionId;
                RakNetTime startTime;
                enum
                {
                        NAT_ATTEMPT_PHASE_NOT_STARTED,
                        NAT_ATTEMPT_PHASE_GETTING_RECENT_PORTS,
                } attemptPhase;
        };
        struct User
        {
                RakNetGUID guid;
                SystemAddress systemAddress;
                unsigned short mostRecentPort;
                bool isReady;

                DataStructures::List<ConnectionAttempt *> connectionAttempts;
                bool HasConnectionAttemptToUser(User *user);
                void DerefConnectionAttempt(ConnectionAttempt *ca);
                void DeleteConnectionAttempt(ConnectionAttempt *ca);
                void LogConnectionAttempts(RakNet::RakString &rs);
        };
        RakNetTime lastUpdate;
        static int NatPunchthroughUserComp( const RakNetGUID &key, User * const &data );
protected:
        void OnNATPunchthroughRequest(Packet *packet);
        DataStructures::OrderedList<RakNetGUID, User*, NatPunchthroughServer::NatPunchthroughUserComp> users;

        void OnGetMostRecentPort(Packet *packet);
        void OnClientReady(Packet *packet);

        void SendTimestamps(void);
        void StartPendingPunchthrough(void);
        void StartPunchthroughForUser(User*user);
        uint16_t sessionId;
        NatPunchthroughServerDebugInterface *natPunchthroughServerDebugInterface;

};

#endif

#endif // _RAKNET_SUPPORT_*