FileListTransfer.h

Go to the documentation of this file.

#include "NativeFeatureIncludes.h"
#if _RAKNET_SUPPORT_FileListTransfer==1 && _RAKNET_SUPPORT_FileOperations==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"
#include "ThreadPool.h"

namespace RakNet
{
class IncrementalReadInterface;
class FileListTransferCBInterface;
class FileListProgress;
struct FileListReceiver;


class RAK_DLL_EXPORT FileListTransfer : public PluginInterface2
{
public:

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

    FileListTransfer();
    virtual ~FileListTransfer();

    void StartIncrementalReadThreads(int numThreads, int threadPriority=-99999);
    
    unsigned short SetupReceive(FileListTransferCBInterface *handler, bool deleteHandler, SystemAddress allowedSender);

    void Send(FileList *fileList, RakNet::RakPeerInterface *rakPeer, SystemAddress recipient, unsigned short setID, PacketPriority priority, char orderingChannel, 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 AddCallback(FileListProgress *cb);

    void RemoveCallback(FileListProgress *cb);

    void ClearCallbacks(void);

    void GetCallbacks(DataStructures::List<FileListProgress*> &callbacks);

    virtual PluginReceiveResult OnReceive(Packet *packet);
    virtual void OnRakPeerShutdown(void);
    virtual void OnClosedConnection(const 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;
    DataStructures::List<FileListProgress*> fileListProgressCallbacks;

    struct FileToPush
    {
        FileListNode fileListNode;
        PacketPriority packetPriority;
        char orderingChannel;
        unsigned int currentOffset;
        unsigned short setID;
        unsigned int setIndex;
        IncrementalReadInterface *incrementalReadInterface;
        unsigned int chunkSize;
    };
    struct FileToPushRecipient
    {
        unsigned int refCount;
        SimpleMutex refCountMutex;
        void DeleteThis(void);
        void AddRef(void);
        void Deref(void);

        SystemAddress systemAddress;
        DataStructures::Queue<FileToPush*> filesToPush;
    };
    DataStructures::List< FileToPushRecipient* > fileToPushRecipientList;
    SimpleMutex fileToPushRecipientListMutex;
    void RemoveFromList(FileToPushRecipient *ftpr);

    struct ThreadData
    {
        FileListTransfer *fileListTransfer;
        SystemAddress systemAddress;
    };

    ThreadPool<ThreadData, int> threadPool;

    friend int SendIRIToAddressCB(FileListTransfer::ThreadData threadData, bool *returnOutput, void* perThreadData);
};

} // namespace RakNet

#endif

#endif // _RAKNET_SUPPORT_*