NatTypeDetectionServer.h

Go to the documentation of this file.

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

#ifndef __NAT_TYPE_DETECTION_SERVER_H
#define __NAT_TYPE_DETECTION_SERVER_H

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


namespace RakNet
{
class RakPeerInterface;
struct Packet;

class RAK_DLL_EXPORT NatTypeDetectionServer : public PluginInterface2
{
public:

    // GetInstance() and DestroyInstance(instance*)
    STATIC_FACTORY_DECLARATIONS(NatTypeDetectionServer)

    // Constructor
    NatTypeDetectionServer();

    // Destructor
    virtual ~NatTypeDetectionServer();

    void Startup(
        const char *nonRakNetIP2,
        const char *nonRakNetIP3,
        const char *nonRakNetIP4);

    // Releases the sockets created in Startup();
    void Shutdown(void);

    virtual void Update(void);

    virtual PluginReceiveResult OnReceive(Packet *packet);
    virtual void OnClosedConnection(const SystemAddress &systemAddress, RakNetGUID rakNetGUID, PI2_LostConnectionReason lostConnectionReason );

    enum NATDetectionState
    {
        STATE_NONE,
        STATE_TESTING_NONE_1,
        STATE_TESTING_NONE_2,
        STATE_TESTING_FULL_CONE_1,
        STATE_TESTING_FULL_CONE_2,
        STATE_TESTING_ADDRESS_RESTRICTED_1,
        STATE_TESTING_ADDRESS_RESTRICTED_2,
        STATE_TESTING_PORT_RESTRICTED_1,
        STATE_TESTING_PORT_RESTRICTED_2,
        STATE_DONE,
    };

    struct NATDetectionAttempt
    {
        SystemAddress systemAddress;
        NATDetectionState detectionState;
        RakNet::TimeMS nextStateTime;
        RakNet::TimeMS timeBetweenAttempts;
        unsigned short c2Port;
        RakNetGUID guid;
    };

protected:
    void OnDetectionRequest(Packet *packet);
    DataStructures::List<NATDetectionAttempt> natDetectionAttempts;
    unsigned int GetDetectionAttemptIndex(const SystemAddress &sa);
    unsigned int GetDetectionAttemptIndex(RakNetGUID guid);

    // s1p1 is rakpeer itself
    SOCKET s1p2,s2p3,s3p4,s4p5;
    unsigned short s1p2Port, s2p3Port, s3p4Port, s4p5Port;
    char s3p4Address[64];
};
}


#endif

#endif // _RAKNET_SUPPORT_*