lltransfertargetfile.cpp
上传用户:king477883
上传日期:2021-03-01
资源大小:9553k
文件大小:4k
源码类别:

游戏引擎

开发平台:

C++ Builder

  1. /** 
  2.  * @file lltransfertargetfile.cpp
  3.  * @brief Transfer system for receiving a file.
  4.  *
  5.  * $LicenseInfo:firstyear=2006&license=viewergpl$
  6.  * 
  7.  * Copyright (c) 2006-2010, Linden Research, Inc.
  8.  * 
  9.  * Second Life Viewer Source Code
  10.  * The source code in this file ("Source Code") is provided by Linden Lab
  11.  * to you under the terms of the GNU General Public License, version 2.0
  12.  * ("GPL"), unless you have obtained a separate licensing agreement
  13.  * ("Other License"), formally executed by you and Linden Lab.  Terms of
  14.  * the GPL can be found in doc/GPL-license.txt in this distribution, or
  15.  * online at http://secondlifegrid.net/programs/open_source/licensing/gplv2
  16.  * 
  17.  * There are special exceptions to the terms and conditions of the GPL as
  18.  * it is applied to this Source Code. View the full text of the exception
  19.  * in the file doc/FLOSS-exception.txt in this software distribution, or
  20.  * online at
  21.  * http://secondlifegrid.net/programs/open_source/licensing/flossexception
  22.  * 
  23.  * By copying, modifying or distributing this software, you acknowledge
  24.  * that you have read and understood your obligations described above,
  25.  * and agree to abide by those obligations.
  26.  * 
  27.  * ALL LINDEN LAB SOURCE CODE IS PROVIDED "AS IS." LINDEN LAB MAKES NO
  28.  * WARRANTIES, EXPRESS, IMPLIED OR OTHERWISE, REGARDING ITS ACCURACY,
  29.  * COMPLETENESS OR PERFORMANCE.
  30.  * $/LicenseInfo$
  31.  */
  32. #include "linden_common.h"
  33. #include "lltransfertargetfile.h"
  34. #include "llerror.h"
  35. LLTransferTargetFile::LLTransferTargetFile(
  36. const LLUUID& uuid,
  37. LLTransferSourceType src_type) :
  38. LLTransferTarget(LLTTT_FILE, uuid, src_type),
  39. mFP(NULL)
  40. {
  41. }
  42. LLTransferTargetFile::~LLTransferTargetFile()
  43. {
  44. if (mFP)
  45. {
  46. llerrs << "LLTransferTargetFile::~LLTransferTargetFile - Should have been cleaned up in completion callback" << llendl;
  47. fclose(mFP);
  48. mFP = NULL;
  49. }
  50. }
  51. // virtual
  52. bool LLTransferTargetFile::unpackParams(LLDataPacker& dp)
  53. {
  54. // we can safely ignore this call
  55. return true;
  56. }
  57. void LLTransferTargetFile::applyParams(const LLTransferTargetParams &params)
  58. {
  59. if (params.getType() != mType)
  60. {
  61. llwarns << "Target parameter type doesn't match!" << llendl;
  62. return;
  63. }
  64. mParams = (LLTransferTargetParamsFile &)params;
  65. }
  66. LLTSCode LLTransferTargetFile::dataCallback(const S32 packet_id, U8 *in_datap, const S32 in_size)
  67. {
  68. //llinfos << "LLTransferTargetFile::dataCallback" << llendl;
  69. //llinfos << "Packet: " << packet_id << llendl;
  70. if (!mFP)
  71. {
  72. mFP = LLFile::fopen(mParams.mFilename, "wb"); /* Flawfinder: ignore */
  73. if (!mFP)
  74. {
  75. llwarns << "Failure opening " << mParams.mFilename << " for write by LLTransferTargetFile" << llendl;
  76. return LLTS_ERROR;
  77. }
  78. }
  79. if (!in_size)
  80. {
  81. return LLTS_OK;
  82. }
  83. S32 count = (S32)fwrite(in_datap, 1, in_size, mFP);
  84. if (count != in_size)
  85. {
  86. llwarns << "Failure in LLTransferTargetFile::dataCallback!" << llendl;
  87. return LLTS_ERROR;
  88. }
  89. return LLTS_OK;
  90. }
  91. void LLTransferTargetFile::completionCallback(const LLTSCode status)
  92. {
  93. llinfos << "LLTransferTargetFile::completionCallback" << llendl;
  94. if (mFP)
  95. {
  96. fclose(mFP);
  97. }
  98. // Still need to gracefully handle error conditions.
  99. switch (status)
  100. {
  101. case LLTS_DONE:
  102. break;
  103. case LLTS_ABORT:
  104. case LLTS_ERROR:
  105. // We're aborting this transfer, we don't want to keep this file.
  106. llwarns << "Aborting file transfer for " << mParams.mFilename << llendl;
  107. if (mFP)
  108. {
  109. // Only need to remove file if we successfully opened it.
  110. LLFile::remove(mParams.mFilename);
  111. }
  112. default:
  113. break;
  114. }
  115. mFP = NULL;
  116. if (mParams.mCompleteCallback)
  117. {
  118. mParams.mCompleteCallback(status, mParams.mUserData);
  119. }
  120. }