/indra/llmessage/lltransfertargetfile.cpp
C++ | 131 lines | 82 code | 17 blank | 32 comment | 12 complexity | 6b4605c4e1ba91f6b613ae16c1bcb569 MD5 | raw file
Possible License(s): LGPL-2.1
- /**
- * @file lltransfertargetfile.cpp
- * @brief Transfer system for receiving a file.
- *
- * $LicenseInfo:firstyear=2006&license=viewerlgpl$
- * Second Life Viewer Source Code
- * Copyright (C) 2010, Linden Research, Inc.
- *
- * This library is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License as published by the Free Software Foundation;
- * version 2.1 of the License only.
- *
- * This library is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public
- * License along with this library; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
- *
- * Linden Research, Inc., 945 Battery Street, San Francisco, CA 94111 USA
- * $/LicenseInfo$
- */
- #include "linden_common.h"
- #include "lltransfertargetfile.h"
- #include "llerror.h"
- LLTransferTargetFile::LLTransferTargetFile(
- const LLUUID& uuid,
- LLTransferSourceType src_type) :
- LLTransferTarget(LLTTT_FILE, uuid, src_type),
- mFP(NULL)
- {
- }
- LLTransferTargetFile::~LLTransferTargetFile()
- {
- if (mFP)
- {
- llerrs << "LLTransferTargetFile::~LLTransferTargetFile - Should have been cleaned up in completion callback" << llendl;
- fclose(mFP);
- mFP = NULL;
- }
- }
- // virtual
- bool LLTransferTargetFile::unpackParams(LLDataPacker& dp)
- {
- // we can safely ignore this call
- return true;
- }
- void LLTransferTargetFile::applyParams(const LLTransferTargetParams ¶ms)
- {
- if (params.getType() != mType)
- {
- llwarns << "Target parameter type doesn't match!" << llendl;
- return;
- }
-
- mParams = (LLTransferTargetParamsFile &)params;
- }
- LLTSCode LLTransferTargetFile::dataCallback(const S32 packet_id, U8 *in_datap, const S32 in_size)
- {
- //llinfos << "LLTransferTargetFile::dataCallback" << llendl;
- //llinfos << "Packet: " << packet_id << llendl;
- if (!mFP)
- {
- mFP = LLFile::fopen(mParams.mFilename, "wb"); /* Flawfinder: ignore */
- if (!mFP)
- {
- llwarns << "Failure opening " << mParams.mFilename << " for write by LLTransferTargetFile" << llendl;
- return LLTS_ERROR;
- }
- }
- if (!in_size)
- {
- return LLTS_OK;
- }
- S32 count = (S32)fwrite(in_datap, 1, in_size, mFP);
- if (count != in_size)
- {
- llwarns << "Failure in LLTransferTargetFile::dataCallback!" << llendl;
- return LLTS_ERROR;
- }
- return LLTS_OK;
- }
- void LLTransferTargetFile::completionCallback(const LLTSCode status)
- {
- llinfos << "LLTransferTargetFile::completionCallback" << llendl;
- if (mFP)
- {
- fclose(mFP);
- }
- // Still need to gracefully handle error conditions.
- switch (status)
- {
- case LLTS_DONE:
- break;
- case LLTS_ABORT:
- case LLTS_ERROR:
- // We're aborting this transfer, we don't want to keep this file.
- llwarns << "Aborting file transfer for " << mParams.mFilename << llendl;
- if (mFP)
- {
- // Only need to remove file if we successfully opened it.
- LLFile::remove(mParams.mFilename);
- }
- default:
- break;
- }
- mFP = NULL;
- if (mParams.mCompleteCallback)
- {
- mParams.mCompleteCallback(status, mParams.mUserData);
- }
- }