Print done when done.

1 files changed, 151 insertions, 0 deletions

diff --git a/indra/llmessage/lltransfersourcefile.cpp b/indra/llmessage/lltransfersourcefile.cpp
new file mode 100644
index 0000000000..45b03d7653
--- /dev/null
+++ b/indra/llmessage/lltransfersourcefile.cpp
@@ -0,0 +1,151 @@
+/** 
+ * @file lltransfersourcefile.cpp
+ * @brief Transfer system for sending a file.
+ *
+ * Copyright (c) 2006-$CurrentYear$, Linden Research, Inc.
+ * $License$
+ */
+
+#include "linden_common.h"
+
+#include "lltransfersourcefile.h"
+
+#include "llerror.h"
+#include "message.h"
+#include "lldatapacker.h"
+#include "lldir.h"
+
+LLTransferSourceFile::LLTransferSourceFile(const LLUUID &request_id, const F32 priority) :
+	LLTransferSource(LLTST_FILE, request_id, priority),
+	mFP(NULL)
+{
+}
+
+LLTransferSourceFile::~LLTransferSourceFile()
+{
+	if (mFP)
+	{
+		llerrs << "Destructor called without the completion callback being called!" << llendl;
+	}
+}
+
+void LLTransferSourceFile::initTransfer()
+{
+	std::string filename = mParams.getFilename();
+	std::string delimiter = gDirUtilp->getDirDelimiter();
+
+	if((filename == ".")
+	   || (filename == "..")
+	   || (filename.find(delimiter[0]) != std::string::npos))
+	{
+		llwarns << "Attempting to transfer file " << filename << " with path delimiter, aborting!" << llendl;
+
+		sendTransferStatus(LLTS_ERROR);
+		return;
+	}
+	// Look for the file.
+	mFP = LLFile::fopen(mParams.getFilename().c_str(), "rb");		/* Flawfinder: ignore */
+	if (!mFP)
+	{
+		sendTransferStatus(LLTS_ERROR);
+		return;
+	}
+
+	// Get the size of the file using the hack from
+	fseek(mFP,0,SEEK_END);
+	mSize = ftell(mFP);
+	fseek(mFP,0,SEEK_SET);
+
+	sendTransferStatus(LLTS_OK);
+}
+
+F32 LLTransferSourceFile::updatePriority()
+{
+	return 0.f;
+}
+
+LLTSCode LLTransferSourceFile::dataCallback(const S32 packet_id,
+											const S32 max_bytes,
+											U8 **data_handle,
+											S32 &returned_bytes,
+											BOOL &delete_returned)
+{
+	//llinfos << "LLTransferSourceFile::dataCallback" << llendl;
+
+	if (!mFP)
+	{
+		llerrs << "Data callback without file set!" << llendl;
+		return LLTS_ERROR;
+	}
+
+	if (packet_id != mLastPacketID + 1)
+	{
+		llerrs << "Can't handle out of order file transfer yet!" << llendl;
+	}
+
+	// Grab up until the max number of bytes from the file.
+	delete_returned = TRUE;
+	U8 *tmpp = new U8[max_bytes];
+	*data_handle = tmpp;
+	returned_bytes = (S32)fread(tmpp, 1, max_bytes, mFP);
+	if (!returned_bytes)
+	{
+		delete[] tmpp;
+		*data_handle = NULL;
+		returned_bytes = 0;
+		delete_returned = FALSE;
+		return LLTS_DONE;
+	}
+
+	return LLTS_OK;
+}
+
+void LLTransferSourceFile::completionCallback(const LLTSCode status)
+{
+	// No matter what happens, all we want to do is close the file pointer if
+	// we've got it open.
+	if (mFP)
+	{
+		fclose(mFP);
+		mFP = NULL;
+
+	}
+	// Delete the file iff the filename begins with "TEMP"
+	if (mParams.getDeleteOnCompletion() && memcmp(mParams.getFilename().c_str(), "TEMP", 4) == 0)
+	{
+		LLFile::remove(mParams.getFilename().c_str());
+	}
+}
+
+BOOL LLTransferSourceFile::unpackParams(LLDataPacker &dp)
+{
+	//llinfos << "LLTransferSourceFile::unpackParams" << llendl;
+
+	return mParams.unpackParams(dp);
+}
+
+
+LLTransferSourceParamsFile::LLTransferSourceParamsFile() : LLTransferSourceParams(LLTST_FILE)
+{
+}
+
+
+void LLTransferSourceParamsFile::packParams(LLDataPacker &dp) const
+{
+	dp.packString(mFilename.c_str(), "Filename");
+	dp.packU8((U8)mDeleteOnCompletion, "Delete");
+}
+
+
+BOOL LLTransferSourceParamsFile::unpackParams(LLDataPacker &dp)
+{
+	char tmp_str[512];		/* Flawfinder: ignore */
+	dp.unpackString(tmp_str, "Filename");
+	mFilename = tmp_str;
+	U8 delete_flag;
+	dp.unpackU8(delete_flag, "Delete");
+	mDeleteOnCompletion = delete_flag;
+
+	llinfos << "Unpacked filename: " << mFilename << llendl;
+	return TRUE;
+}