UDPForwarder.h

Go to the documentation of this file.

#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_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;

        UDPForwarderResult StartForwarding(SystemAddress source, SystemAddress destination, RakNetTimeMS timeoutOnNoDataMS, const char *forceHostAddress,
                unsigned short *srcToDestPort, unsigned short *destToSourcePort, SOCKET *srcToDestSocket, SOCKET *destToSourceSocket);

        void StopForwarding(SystemAddress source, SystemAddress destination);

        struct SrcAndDest
        {
                SystemAddress source;
                SystemAddress destination;
        };

        struct ForwardEntry
        {
                ForwardEntry();
                ~ForwardEntry();
                SrcAndDest srcAndDest;
                RakNetTimeMS timeLastDatagramForwarded;
                SOCKET readSocket;
                SOCKET writeSocket;
                RakNetTimeMS timeoutOnNoDataMS;
                bool updatedSourceAddress;
        };

#ifdef UDP_FORWARDER_EXECUTE_THREADED
        struct ThreadOperation
        {
                enum {
                TO_NONE,
                TO_START_FORWARDING,
                TO_STOP_FORWARDING,
                } operation;

                SystemAddress source;
                SystemAddress destination;
                RakNetTimeMS timeoutOnNoDataMS;
                RakNet::RakString forceHostAddress;
                unsigned short srcToDestPort;
                unsigned short destToSourcePort;
                SOCKET srcToDestSocket;
                SOCKET destToSourceSocket;
                UDPForwarderResult result;
        };
        SimpleMutex threadOperationIncomingMutex,threadOperationOutgoingMutex;
        DataStructures::Queue<ThreadOperation> threadOperationIncomingQueue;
        DataStructures::Queue<ThreadOperation> threadOperationOutgoingQueue;
#endif
        void UpdateThreaded(void);
        UDPForwarderResult StartForwardingThreaded(SystemAddress source, SystemAddress destination, RakNetTimeMS timeoutOnNoDataMS, const char *forceHostAddress,
                unsigned short *srcToDestPort, unsigned short *destToSourcePort, SOCKET *srcToDestSocket, SOCKET *destToSourceSocket);
        void StopForwardingThreaded(SystemAddress source, SystemAddress destination);

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

        unsigned short AddForwardingEntry(SrcAndDest srcAndDest, RakNetTimeMS timeoutOnNoDataMS, const char *forceHostAddress);


        bool isRunning, threadRunning;

};

} // End namespace

#endif