FileListTransfer.h

Go to the documentation of this file.

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

#ifndef __FILE_LIST_TRANFER_H
#define __FILE_LIST_TRANFER_H

#include "RakNetTypes.h"
#include "Export.h"
#include "PluginInterface2.h"
#include "DS_Map.h"
#include "RakNetTypes.h"
#include "PacketPriority.h"
#include "RakMemoryOverride.h"
#include "FileList.h"
#include "DS_Queue.h"
#include "SimpleMutex.h"

class IncrementalReadInterface;
class FileListTransferCBInterface;
class FileListProgress;
struct FileListReceiver;


class RAK_DLL_EXPORT FileListTransfer : public PluginInterface2
{
public:
        FileListTransfer();
        virtual ~FileListTransfer();
        
    unsigned short SetupReceive(FileListTransferCBInterface *handler, bool deleteHandler, SystemAddress allowedSender);

        void Send(FileList *fileList, RakPeerInterface *rakPeer, SystemAddress recipient, unsigned short setID, PacketPriority priority, char orderingChannel, bool compressData, IncrementalReadInterface *_incrementalReadInterface=0, unsigned int _chunkSize=262144*4*16);

        unsigned int GetPendingFilesToAddress(SystemAddress recipient);

        void CancelReceive(unsigned short setId);

        void RemoveReceiver(SystemAddress systemAddress);

        bool IsHandlerActive(unsigned short setId);

        void SetCallback(FileListProgress *cb);

        FileListProgress *GetCallback(void) const;

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

protected:
        bool DecodeSetHeader(Packet *packet);
        bool DecodeFile(Packet *packet, bool fullFile);

        void Clear(void);

        void OnReferencePush(Packet *packet, bool fullFile);
        void OnReferencePushAck(Packet *packet);
        void SendIRIToAddress(SystemAddress systemAddress);

        DataStructures::Map<unsigned short, FileListReceiver*> fileListReceivers;
        unsigned short setId;
        FileListProgress *callback;

        struct FileToPush
        {
                FileListNode fileListNode;
                PacketPriority packetPriority;
                char orderingChannel;
                unsigned int currentOffset;
                unsigned short setID;
                unsigned int setIndex;
                IncrementalReadInterface *incrementalReadInterface;
                unsigned int chunkSize;
        };
        struct FileToPushRecipient
        {
                SystemAddress systemAddress;
                DataStructures::Queue<FileToPush*> filesToPush;
        };
        DataStructures::List< FileToPushRecipient* > filesToPushAllSameAddress;
        // TODO - overagressive, only one read can happen at a time. See SendIRIToAddress
        SimpleMutex filesToPushAllSameAddressMutex;
};

#endif

#endif // _RAKNET_SUPPORT_*