Print done when done.

1 files changed, 187 insertions, 0 deletions

diff --git a/indra/llmessage/llxfermanager.h b/indra/llmessage/llxfermanager.h
new file mode 100644
index 0000000000..eca3684df5
--- /dev/null
+++ b/indra/llmessage/llxfermanager.h
@@ -0,0 +1,187 @@
+/** 
+ * @file llxfermanager.h
+ * @brief definition of LLXferManager class for a keeping track of
+ * multiple xfers
+ *
+ * Copyright (c) 2001-$CurrentYear$, Linden Research, Inc.
+ * $License$
+ */
+
+#ifndef LL_LLXFERMANAGER_H
+#define LL_LLXFERMANAGER_H
+
+/**
+ * this manager keeps both a send list and a receive list; anything with a 
+ * LLXferManager can send and receive files via messages
+ */
+
+//Forward declaration to avoid circular dependencies
+class LLXfer;
+class LLVFS;
+
+#include "llxfer.h"
+#include "message.h"
+#include "llassetstorage.h"
+#include "linked_lists.h"
+#include "lldir.h"
+#include "lllinkedqueue.h"
+#include "llthrottle.h"
+
+class LLHostStatus
+{
+ public:
+	LLHost mHost;
+	S32    mNumActive;
+	S32    mNumPending;
+
+	LLHostStatus() {mNumActive = 0; mNumPending = 0;};
+	virtual ~LLHostStatus(){};
+};
+
+// Class stores ack information, to be put on list so we can throttle xfer rate.
+class LLXferAckInfo
+{
+public:
+	LLXferAckInfo(U32 dummy = 0)
+	{
+		mID = 0;
+		mPacketNum = -1;
+	}
+
+	U64 mID;
+	S32 mPacketNum;
+	LLHost mRemoteHost;
+};
+
+class LLXferManager
+{
+ private:
+	LLVFS *mVFS;
+
+ protected:
+	S32    mMaxOutgoingXfersPerCircuit;
+	S32    mMaxIncomingXfers;
+
+	BOOL	mUseAckThrottling; // Use ack throttling to cap file xfer bandwidth
+	LLLinkedQueue<LLXferAckInfo> mXferAckQueue;
+	LLThrottle mAckThrottle;
+ public:
+
+	// This enumeration is useful in the requestFile() to specify if
+	// an xfer must happen asap.
+	enum
+	{
+		LOW_PRIORITY = FALSE,
+		HIGH_PRIORITY = TRUE,
+	};
+
+	LLXfer *mSendList;
+	LLXfer *mReceiveList;
+
+	LLLinkedList <LLHostStatus> mOutgoingHosts;
+
+ private:
+ protected:
+	// implementation methods
+	virtual void startPendingDownloads();
+	virtual void addToList(LLXfer* xferp, LLXfer*& head, BOOL is_priority);
+
+ public:
+	LLXferManager(LLVFS *vfs);
+	virtual ~LLXferManager();
+
+	virtual void init(LLVFS *vfs);
+	virtual void free();
+
+	void setUseAckThrottling(const BOOL use);
+	void setAckThrottleBPS(const F32 bps);
+
+// list management routines
+	virtual LLXfer *findXfer(U64 id, LLXfer *list_head);
+	virtual void removeXfer (LLXfer *delp, LLXfer **list_head);
+	virtual U32 numActiveListEntries(LLXfer *list_head);
+	virtual S32 numActiveXfers(const LLHost &host);
+	virtual S32 numPendingXfers(const LLHost &host);
+	virtual void changeNumActiveXfers(const LLHost &host, S32 delta);
+
+	virtual void setMaxOutgoingXfersPerCircuit (S32 max_num);
+	virtual void setMaxIncomingXfers(S32 max_num);
+	virtual void updateHostStatus();
+	virtual void printHostStatus();
+
+// general utility routines
+	virtual void registerCallbacks(LLMessageSystem *mesgsys);
+	virtual U64 getNextID ();
+	virtual S32 encodePacketNum(S32 packet_num, BOOL is_eof);	
+	virtual S32 decodePacketNum(S32 packet_num);	
+	virtual BOOL isLastPacket(S32 packet_num);
+
+	virtual U64 registerXfer(const void *datap, const S32 length);
+
+// file requesting routines
+// .. to file
+	virtual void requestFile(const char* local_filename,
+							 const char* remote_filename,
+							 ELLPath remote_path,
+							 const LLHost& remote_host,
+ 							 BOOL delete_remote_on_completion,
+							 void (*callback)(void**,S32), void** user_data,
+							 BOOL is_priority = FALSE,
+							 BOOL use_big_packets = FALSE);
+
+// .. to memory
+	virtual void requestFile(const char* remote_filename, 
+							 ELLPath remote_path,
+							 const LLHost &remote_host,
+ 							 BOOL delete_remote_on_completion,
+							 void (*callback)(void*, S32, void**, S32),
+							 void** user_data,
+							 BOOL is_priority = FALSE);
+
+// vfile requesting
+// .. to vfile
+	virtual void requestVFile(const LLUUID &local_id, const LLUUID& remote_id,
+							  LLAssetType::EType type, LLVFS* vfs,
+							  const LLHost& remote_host,
+							  void (*callback)(void**, S32), void** user_data,
+							  BOOL is_priority = FALSE);
+
+/*
+// xfer request (may be memory or file)
+// .. to file
+	virtual void requestXfer(const char *local_filename, U64 xfer_id, 
+							 BOOL delete_remote_on_completion,
+							 const LLHost &remote_host, void (*callback)(void **,S32),void **user_data);
+// .. to memory
+	virtual void requestXfer(U64 xfer_id, 
+							 const LLHost &remote_host, 
+							 BOOL delete_remote_on_completion,
+							 void (*callback)(void *, S32, void **, S32),void **user_data);
+*/
+
+	virtual void processReceiveData (LLMessageSystem *mesgsys, void **user_data);
+	virtual void sendConfirmPacket (LLMessageSystem *mesgsys, U64 id, S32 packetnum, const LLHost &remote_host);
+
+// file sending routines
+	virtual void processFileRequest (LLMessageSystem *mesgsys, void **user_data);
+	virtual void processConfirmation (LLMessageSystem *mesgsys, void **user_data);
+	virtual void retransmitUnackedPackets ();
+
+// error handling
+	virtual void processAbort (LLMessageSystem *mesgsys, void **user_data);
+};
+
+extern LLXferManager*	gXferManager;
+
+// initialization and garbage collection
+void start_xfer_manager(LLVFS *vfs);
+void cleanup_xfer_manager();
+
+// message system callbacks
+void process_confirm_packet (LLMessageSystem *mesgsys, void **user_data);
+void process_request_xfer (LLMessageSystem *mesgsys, void **user_data);
+void continue_file_receive(LLMessageSystem *mesgsys, void **user_data);
+void process_abort_xfer (LLMessageSystem *mesgsys, void **user_data);
+#endif
+
+