Print done when done.

1 files changed, 350 insertions, 0 deletions

diff --git a/indra/llmessage/llxfer.cpp b/indra/llmessage/llxfer.cpp
new file mode 100644
index 0000000000..154696eb4e
--- /dev/null
+++ b/indra/llmessage/llxfer.cpp
@@ -0,0 +1,350 @@
+/** 
+ * @file llxfer.cpp
+ * @brief implementation of LLXfer class for a single xfer.
+ *
+ * Copyright (c) 2001-$CurrentYear$, Linden Research, Inc.
+ * $License$
+ */
+
+#include "linden_common.h"
+
+#include "llxfer.h"
+#include "lluuid.h"
+#include "llerror.h"
+#include "llmath.h"
+#include "u64.h"
+
+//number of bytes sent in each message
+const U32 LL_XFER_CHUNK_SIZE = 1000;
+
+const U32 LLXfer::XFER_FILE = 1;
+const U32 LLXfer::XFER_VFILE = 2;
+const U32 LLXfer::XFER_MEM = 3;
+
+///////////////////////////////////////////////////////////
+
+LLXfer::LLXfer (S32 chunk_size)
+{
+	init(chunk_size);
+}
+
+///////////////////////////////////////////////////////////
+
+LLXfer::~LLXfer ()
+{
+	free();
+}
+
+///////////////////////////////////////////////////////////
+
+void LLXfer::init (S32 chunk_size)
+{
+	mID = 0;
+
+	mPacketNum = -1; // there's a preincrement before sending the zeroth packet
+	mXferSize = 0;
+
+	mStatus = e_LL_XFER_UNINITIALIZED;
+	mNext = NULL;
+	mWaitingForACK = FALSE;
+	
+	mCallback = NULL;
+	mCallbackDataHandle = NULL;
+
+	mBufferContainsEOF = FALSE;
+	mBuffer = NULL;
+	mBufferLength = 0;
+	mBufferStartOffset = 0;
+
+	mRetries = 0;
+
+	if (chunk_size < 1)
+	{
+		chunk_size = LL_XFER_CHUNK_SIZE;
+	}
+	mChunkSize = chunk_size;
+}
+	
+///////////////////////////////////////////////////////////
+
+void LLXfer::free ()
+{
+	if (mBuffer)
+	{
+		delete[] mBuffer;
+		mBuffer = NULL;
+	}
+}
+
+///////////////////////////////////////////////////////////
+
+S32 LLXfer::startSend (U64 xfer_id, const LLHost &remote_host)
+{
+	llwarns << "undifferentiated LLXfer::startSend for " << getName() << llendl;
+	return (-1);
+}
+
+///////////////////////////////////////////////////////////
+
+void LLXfer::setXferSize (S32 xfer_size)
+{	
+	mXferSize = xfer_size;
+//	cout << "starting transfer of size: " << xfer_size << endl;
+}
+
+///////////////////////////////////////////////////////////
+
+S32 LLXfer::startDownload()
+{
+	llwarns << "undifferentiated LLXfer::startDownload for " << getName()
+			<< llendl;
+	return (-1);
+}
+
+///////////////////////////////////////////////////////////
+
+S32 LLXfer::receiveData (char *datap, S32 data_size)
+{
+	S32 retval = 0;
+
+	if (((S32) mBufferLength + data_size) > getMaxBufferSize())
+	{
+		retval = flush();
+	}
+
+	if (!retval)
+	{
+		if (datap != NULL)
+		{
+			memcpy(&mBuffer[mBufferLength],datap,data_size);
+			mBufferLength += data_size;
+		}
+		else
+		{
+			llerrs << "NULL data passed in receiveData" << llendl;
+		}
+	}
+
+	return (retval);
+}
+
+///////////////////////////////////////////////////////////
+
+S32 LLXfer::flush()
+{
+	// only files have somewhere to flush to
+	// if we get called with a flush it means we've blown past our
+	// allocated buffer size
+
+	return (-1);
+}
+
+
+///////////////////////////////////////////////////////////
+
+S32 LLXfer::suck(S32 start_position)
+{
+	llwarns << "Attempted to send a packet outside the buffer bounds in LLXfer::suck()" << llendl;
+	return (-1);
+}
+
+///////////////////////////////////////////////////////////
+
+void LLXfer::sendPacket(S32 packet_num)
+{
+	char fdata_buf[LL_XFER_LARGE_PAYLOAD+4];		/* Flawfinder: ignore */
+	S32 fdata_size = mChunkSize;
+	BOOL last_packet = FALSE;
+	S32 num_copy = 0;
+
+	// if the desired packet is not in our current buffered excerpt from the file. . . 
+	if (((U32)packet_num*fdata_size < mBufferStartOffset) 
+		|| ((U32)llmin((U32)mXferSize,(U32)((U32)(packet_num+1)*fdata_size)) > mBufferStartOffset + mBufferLength))
+	
+	{
+		if (suck(packet_num*fdata_size))  // returns non-zero on failure
+		{
+			abort(LL_ERR_EOF);
+			return;
+		}	
+	}
+		
+	S32 desired_read_position = 0;
+		
+	desired_read_position = packet_num * fdata_size - mBufferStartOffset;
+	
+	fdata_size = llmin((S32)mBufferLength-desired_read_position, mChunkSize);
+
+	if (fdata_size < 0)
+	{
+		llwarns << "negative data size in xfer send, aborting" << llendl;
+		abort(LL_ERR_EOF);
+		return;
+	}
+
+	if (((U32)(desired_read_position + fdata_size) >= (U32)mBufferLength) && (mBufferContainsEOF))
+	{
+		last_packet = TRUE;
+	}
+		
+	if (packet_num)
+	{ 
+		num_copy = llmin(fdata_size, (S32)sizeof(fdata_buf));
+		num_copy = llmin(num_copy, (S32)(sizeof(mBuffer) - desired_read_position));
+		if (num_copy > 0)
+		{
+			memcpy(fdata_buf,&mBuffer[desired_read_position],num_copy);
+		}
+	}
+	else  // if we're the first packet, encode size as an additional S32 at start of data
+	{	 
+		num_copy = llmin(fdata_size, (S32)(sizeof(fdata_buf)-sizeof(S32)));
+		num_copy = llmin(num_copy, (S32)(sizeof(mBuffer) - desired_read_position));
+		if (num_copy > 0)
+		{
+			memcpy(fdata_buf+sizeof(S32),&mBuffer[desired_read_position],fdata_size);
+		}
+		fdata_size += sizeof(S32);
+		htonmemcpy(fdata_buf,&mXferSize, MVT_S32, sizeof(S32));
+	}
+
+	S32 encoded_packetnum = encodePacketNum(packet_num,last_packet);
+		
+	if (fdata_size)
+	{
+		// send the packet 
+		gMessageSystem->newMessageFast(_PREHASH_SendXferPacket);
+		gMessageSystem->nextBlockFast(_PREHASH_XferID);
+		
+		gMessageSystem->addU64Fast(_PREHASH_ID, mID);
+		gMessageSystem->addU32Fast(_PREHASH_Packet, encoded_packetnum);
+		
+		gMessageSystem->nextBlockFast(_PREHASH_DataPacket);
+		gMessageSystem->addBinaryDataFast(_PREHASH_Data, &fdata_buf,fdata_size);
+			
+		gMessageSystem->sendMessage(mRemoteHost);
+
+		ACKTimer.reset();
+		mWaitingForACK = TRUE;
+	}
+	if (last_packet)
+	{
+		mStatus = e_LL_XFER_COMPLETE;	
+	}
+	else
+	{
+		mStatus = e_LL_XFER_IN_PROGRESS;
+	}
+}
+
+///////////////////////////////////////////////////////////
+
+void LLXfer::sendNextPacket()
+{
+	mRetries = 0;
+	sendPacket(++mPacketNum);
+}
+
+///////////////////////////////////////////////////////////
+
+void LLXfer::resendLastPacket()
+{
+	mRetries++;
+	sendPacket(mPacketNum);
+}
+
+///////////////////////////////////////////////////////////
+
+S32 LLXfer::processEOF()
+{
+	S32 retval = 0;
+
+	mStatus = e_LL_XFER_COMPLETE;
+
+	if (LL_ERR_NOERR == mCallbackResult)
+	{
+		llinfos << "xfer from " << mRemoteHost << " complete: " << getName()
+				<< llendl;
+	}
+	else
+	{
+		llinfos << "xfer from " << mRemoteHost << " failed, code "
+				<< mCallbackResult << ": " << getName() << llendl;
+	}
+
+	if (mCallback)
+	{
+		mCallback(mCallbackDataHandle,mCallbackResult);
+	}
+
+	return(retval);
+}
+
+///////////////////////////////////////////////////////////
+
+S32 LLXfer::encodePacketNum(S32 packet_num, BOOL is_EOF)
+{
+	if (is_EOF)
+	{
+		packet_num |= 0x80000000;
+	}
+	return packet_num;
+}
+
+///////////////////////////////////////////////////////////
+
+void LLXfer::abort (S32 result_code)
+{
+	mCallbackResult = result_code;
+
+	llinfos << "Aborting xfer from " << mRemoteHost << " named " << getName()
+			<< " - error: " << result_code << llendl;
+
+	gMessageSystem->newMessageFast(_PREHASH_AbortXfer);
+	gMessageSystem->nextBlockFast(_PREHASH_XferID);
+	gMessageSystem->addU64Fast(_PREHASH_ID, mID);
+	gMessageSystem->addS32Fast(_PREHASH_Result, result_code);
+	
+	gMessageSystem->sendMessage(mRemoteHost);
+
+	mStatus = e_LL_XFER_ABORTED;
+}
+
+
+///////////////////////////////////////////////////////////
+
+const char * LLXfer::getName() 
+{
+	static char tmp_str[256];		/* Flawfinder: ignore */
+
+	return (U64_to_str(mID, tmp_str, sizeof(tmp_str)));
+}
+
+///////////////////////////////////////////////////////////
+
+U32 LLXfer::getXferTypeTag()
+{
+	return 0;
+}
+
+///////////////////////////////////////////////////////////
+
+S32 LLXfer::getMaxBufferSize ()
+{
+	return(mXferSize);
+}
+
+
+std::ostream& operator<< (std::ostream& os, LLXfer &hh)
+{
+	os << hh.getName() ;
+	return os;
+}
+
+
+
+
+
+
+
+