NATDetectClient.h

/* START_LICENSE_HEADER

Copyright (C) 2000 Martin Piper, original design and program code
Copyright (C) 2001 Replica Software

This program file is copyright (C) Replica Software and can only be used under license.
For more information visit: http://www.replicanet.com/
Or email: info@replicanet.com

END_LICENSE_HEADER */
#include "RNPlatform/Inc/MemoryTracking.h"
#ifndef _NATDETECTCLIENT_H_
#define _NATDETECTCLIENT_H_
#include "RNLobby/Inc/NATDetectServer.h"
#include "RNXPURL/Inc/Transport.h"

namespace RNReplicaNet
{

namespace RNLobby
{

class NATDetectClient : public BaseServer
{
public:
    enum NATType
    {
        kOpen = 0,
        kModerate,
        kStrict
    };

    NATDetectClient();
    virtual ~NATDetectClient();

    bool Start(const int channel = TransportAnyGameChannel);

    void SetServers(const XPAddress &address1,const XPAddress &address2);

    bool GetCompleted(void);

    NATDetectServer::Reply GetReply(void);

    NATType GetNATType(void) const;

    XPAddress GetPublicAddress(void) const;

    virtual void CallbackCompleted(void) {}

    void GetAddressReplies(XPAddress &addr1,XPAddress &addr2);

private:
    void Init(void);
    bool CallbackParseValidMessage(DynamicMessageHelper &message,const size_t length,const XPAddress &address);
    bool DoSuccessState(void);

    bool CallbackPoll(void);

    SysTimeType mLastSentTime;
    int mGot1Times;
    int mGot2Times;
    int mTries;
    volatile NATDetectServer::Reply mReply;
    XPAddress mAddress1;
    XPAddress mAddress2;

    XPAddress mAddress1Reply;
    XPAddress mAddress2Reply;

    bool mIsNiceNAT;

    bool mGotOther;
    bool mServerAddressSet;
};

} // namespace RNLobby

} // namespace RNReplicaNet

#endif