UDPForwarder.h

Go to the documentation of this file.

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

#ifndef __UDP_FORWARDER_H
#define __UDP_FORWARDER_H

#include "Export.h"
#include "DS_Multilist.h"
#include "RakNetTypes.h"
#include "SocketIncludes.h"
#include "UDPProxyCommon.h"
#include "SimpleMutex.h"
#include "RakString.h"
#include "RakThread.h"
#include "DS_Queue.h"


#define UDP_FORWARDER_EXECUTE_THREADED

namespace RakNet
{

enum UDPForwarderResult
{
    UDPFORWARDER_FORWARDING_ALREADY_EXISTS,
    UDPFORWARDER_NO_SOCKETS,
    UDPFORWARDER_BIND_FAILED,
    UDPFORWARDER_INVALID_PARAMETERS,
    UDPFORWARDER_SUCCESS,

};

class RAK_DLL_EXPORT UDPForwarder
{
public:
    UDPForwarder();
    ~UDPForwarder();

    void Startup(void);

    void Shutdown(void);

    void Update(void);

    void SetMaxForwardEntries(unsigned short maxEntries);

    int GetMaxForwardEntries(void) const;

    int GetUsedForwardEntries(void) const;

    short socketFamily;
    UDPForwarderResult StartForwarding(SystemAddress source, SystemAddress destination, RakNet::TimeMS timeoutOnNoDataMS, const char *forceHostAddress, unsigned short socketFamily,
        unsigned short *forwardingPort, SOCKET *forwardingSocket);

    void StopForwarding(SystemAddress source, SystemAddress destination);

    struct SrcAndDest
    {
        SrcAndDest() {}
        SrcAndDest(SystemAddress sa1, SystemAddress sa2)
        {
            if (sa1 < sa2)
            {
                source=sa1;
                dest=sa2;
            }
            else
            {
                source=sa2;
                dest=sa1;
            }
        }
        SystemAddress source;
        SystemAddress dest;
    };

    struct ForwardEntry
    {
        ForwardEntry();
        ~ForwardEntry();
        SrcAndDest srcAndDest;
        RakNet::TimeMS timeLastDatagramForwarded;
        SOCKET socket;
        RakNet::TimeMS timeoutOnNoDataMS;
        bool updatedSourcePort, updatedDestPort;
        short socketFamily;
    };


    struct ThreadOperation
    {
        enum {
        TO_NONE,
        TO_START_FORWARDING,
        TO_STOP_FORWARDING,
        } operation;

        SystemAddress source;
        SystemAddress destination;
        RakNet::TimeMS timeoutOnNoDataMS;
        RakNet::RakString forceHostAddress;
        unsigned short forwardingPort;
        SOCKET forwardingSocket;
        UDPForwarderResult result;
        unsigned short socketFamily;
    };
    SimpleMutex threadOperationIncomingMutex,threadOperationOutgoingMutex;
    DataStructures::Queue<ThreadOperation> threadOperationIncomingQueue;
    DataStructures::Queue<ThreadOperation> threadOperationOutgoingQueue;

#if RAKNET_SUPPORT_IPV6==1
    void UpdateThreaded(void);
#endif
    void UpdateThreaded_Old(void);
    UDPForwarderResult StartForwardingThreaded(SystemAddress source, SystemAddress destination, RakNet::TimeMS timeoutOnNoDataMS, const char *forceHostAddress, unsigned short socketFamily,
        unsigned short *forwardingPort, SOCKET *forwardingSocket);
    void StopForwardingThreaded(SystemAddress source, SystemAddress destination);

    DataStructures::Multilist<ML_ORDERED_LIST, ForwardEntry*, SrcAndDest> forwardList;
    unsigned short maxForwardEntries;

    UDPForwarderResult AddForwardingEntry(SrcAndDest srcAndDest, RakNet::TimeMS timeoutOnNoDataMS, unsigned short *port, const char *forceHostAddress, short socketFamily);


    bool isRunning, threadRunning;



};

} // End namespace

#endif

#endif // #if _RAKNET_SUPPORT_UDPForwarder==1