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"

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


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

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

#if _RAKNET_SUPPORT_PacketLogger==1

struct RAK_DLL_EXPORT 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:

    STATIC_FACTORY_DECLARATIONS(NatPunchthroughServer)

    // Constructor
    NatPunchthroughServer();

    // Destructor
    virtual ~NatPunchthroughServer();

    void SetDebugInterface(NatPunchthroughServerDebugInterface *i);

    virtual void Update(void);

    virtual PluginReceiveResult OnReceive(Packet *packet);

    virtual void OnClosedConnection(const SystemAddress &systemAddress, RakNetGUID rakNetGUID, PI2_LostConnectionReason lostConnectionReason );
    virtual void OnNewConnection(const 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;
        RakNet::Time 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::OrderedList<RakNetGUID,RakNetGUID> groupPunchthroughRequests;

        DataStructures::List<ConnectionAttempt *> connectionAttempts;
        bool HasConnectionAttemptToUser(User *user);
        void DerefConnectionAttempt(ConnectionAttempt *ca);
        void DeleteConnectionAttempt(ConnectionAttempt *ca);
        void LogConnectionAttempts(RakNet::RakString &rs);
    };
    RakNet::Time 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;

};

} // namespace RakNet

#endif

#endif // _RAKNET_SUPPORT_*