123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745 |
- /* <!-- copyright */
- /*
- * aria2 - The high speed download utility
- *
- * Copyright (C) 2006 Tatsuhiro Tsujikawa
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program 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 General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
- *
- * In addition, as a special exception, the copyright holders give
- * permission to link the code of portions of this program with the
- * OpenSSL library under certain conditions as described in each
- * individual source file, and distribute linked combinations
- * including the two.
- * You must obey the GNU General Public License in all respects
- * for all of the code used other than OpenSSL. If you modify
- * file(s) with this exception, you may extend this exception to your
- * version of the file(s), but you are not obligated to do so. If you
- * do not wish to do so, delete this exception statement from your
- * version. If you delete this exception statement from all source
- * files in the program, then also delete it here.
- */
- /* copyright --> */
- #include "RequestGroup.h"
- #include "DownloadEngine.h"
- #include "DefaultSegmentManFactory.h"
- #include "NullProgressInfoFile.h"
- #include "SegmentManFactory.h"
- #include "Dependency.h"
- #include "prefs.h"
- #include "InitiateConnectionCommandFactory.h"
- #include "CUIDCounter.h"
- #include "File.h"
- #include "message.h"
- #include "Util.h"
- #include "BtRegistry.h"
- #include "LogFactory.h"
- #include "DiskAdaptor.h"
- #include "DiskWriterFactory.h"
- #include "RecoverableException.h"
- #include "StreamCheckIntegrityEntry.h"
- #include "CheckIntegrityCommand.h"
- #include "UnknownLengthPieceStorage.h"
- #include "SingleFileDownloadContext.h"
- #include "DlAbortEx.h"
- #include "DownloadFailureException.h"
- #include "RequestGroupMan.h"
- #include "DefaultBtProgressInfoFile.h"
- #include "DefaultPieceStorage.h"
- #include "DownloadResult.h"
- #include "DownloadHandlerFactory.h"
- #include "MemoryBufferPreDownloadHandler.h"
- #include "DownloadHandlerConstants.h"
- #ifdef ENABLE_MESSAGE_DIGEST
- # include "CheckIntegrityCommand.h"
- #endif // ENABLE_MESSAGE_DIGEST
- #ifdef ENABLE_BITTORRENT
- # include "BtCheckIntegrityEntry.h"
- # include "DefaultPeerStorage.h"
- # include "DefaultBtAnnounce.h"
- # include "BtSetup.h"
- # include "BtFileAllocationEntry.h"
- # include "BtPostDownloadHandler.h"
- #endif // ENABLE_BITTORRENT
- #ifdef ENABLE_METALINK
- # include "MetalinkPostDownloadHandler.h"
- #endif // ENABLE_METALINK
- int32_t RequestGroup::_gidCounter = 0;
- RequestGroup::RequestGroup(const Option* option,
- const Strings& uris):
- _gid(++_gidCounter),
- _uris(uris),
- _numConcurrentCommand(0),
- _numStreamConnection(0),
- _numCommand(0),
- _segmentMan(0),
- _segmentManFactory(new DefaultSegmentManFactory(option)),
- _downloadContext(0),
- _pieceStorage(0),
- _progressInfoFile(new NullProgressInfoFile()),
- _diskWriterFactory(0),
- _dependency(0),
- _preLocalFileCheckEnabled(true),
- _haltRequested(false),
- _forceHaltRequested(false),
- _option(option),
- _logger(LogFactory::getInstance())
- {
- if(_option->get(PREF_FILE_ALLOCATION) == V_PREALLOC) {
- _fileAllocationEnabled = true;
- } else {
- _fileAllocationEnabled = false;
- }
- initializePreDownloadHandler();
- initializePostDownloadHandler();
- }
- RequestGroup::~RequestGroup() {}
- SegmentManHandle RequestGroup::initSegmentMan()
- {
- _segmentMan = _segmentManFactory->createNewInstance(_downloadContext,
- _pieceStorage);
- return _segmentMan;
- }
- bool RequestGroup::downloadFinished() const
- {
- if(_pieceStorage.isNull()) {
- return false;
- } else {
- return _pieceStorage->downloadFinished();
- }
- }
- bool RequestGroup::allDownloadFinished() const
- {
- if(_pieceStorage.isNull()) {
- return false;
- } else {
- return _pieceStorage->allDownloadFinished();
- }
- }
- void RequestGroup::closeFile()
- {
- if(!_pieceStorage.isNull()) {
- _pieceStorage->getDiskAdaptor()->closeFile();
- }
- }
- Commands RequestGroup::createInitialCommand(DownloadEngine* e)
- {
- #ifdef ENABLE_BITTORRENT
- {
- BtContextHandle btContext = _downloadContext;
- if(!btContext.isNull()) {
- if(e->_requestGroupMan->isSameFileBeingDownloaded(this)) {
- throw new DownloadFailureException(EX_DUPLICATE_FILE_DOWNLOAD,
- getFilePath().c_str());
- }
- initPieceStorage();
- if(btContext->getFileEntries().size() > 1) {
- // this is really multi file torrent.
- // clear http/ftp uris because the current implementation does not
- // allow integrating multi-file torrent and http/ftp.
- _logger->debug("Clearing http/ftp URIs because the current implementation does not allow integrating multi-file torrent and http/ftp.");
- _uris.clear();
- _pieceStorage->setFileFilter(Util::parseIntRange(_option->get(PREF_SELECT_FILE)));
- }
-
- BtProgressInfoFileHandle progressInfoFile =
- new DefaultBtProgressInfoFile(_downloadContext,
- _pieceStorage,
- _option);
-
- BtRegistry::registerBtContext(btContext->getInfoHashAsString(), btContext);
- BtRegistry::registerPieceStorage(btContext->getInfoHashAsString(),
- _pieceStorage);
- BtRegistry::registerBtProgressInfoFile(btContext->getInfoHashAsString(),
- progressInfoFile);
-
- BtRuntimeHandle btRuntime = new BtRuntime();
- btRuntime->setListenPort(_option->getAsInt(PREF_LISTEN_PORT));
- BtRegistry::registerBtRuntime(btContext->getInfoHashAsString(), btRuntime);
- PeerStorageHandle peerStorage = new DefaultPeerStorage(btContext, _option);
- BtRegistry::registerPeerStorage(btContext->getInfoHashAsString(), peerStorage);
- BtAnnounceHandle btAnnounce = new DefaultBtAnnounce(btContext, _option);
- BtRegistry::registerBtAnnounce(btContext->getInfoHashAsString(), btAnnounce);
- btAnnounce->shuffleAnnounce();
-
- BtRegistry::registerPeerObjectCluster(btContext->getInfoHashAsString(),
- new PeerObjectCluster());
- // Call Load, Save and file allocation command here
- if(progressInfoFile->exists()) {
- // load .aria2 file if it exists.
- progressInfoFile->load();
- _pieceStorage->getDiskAdaptor()->openFile();
- } else {
- if(_pieceStorage->getDiskAdaptor()->fileExists()) {
- if(_option->get(PREF_CHECK_INTEGRITY) != V_TRUE &&
- _option->get(PREF_ALLOW_OVERWRITE) != V_TRUE) {
- // TODO we need this->haltRequested = true?
- throw new DownloadFailureException(MSG_FILE_ALREADY_EXISTS,
- _pieceStorage->getDiskAdaptor()->getFilePath().c_str());
- } else {
- _pieceStorage->getDiskAdaptor()->openFile();
- }
- } else {
- _pieceStorage->getDiskAdaptor()->openFile();
- }
- }
- _progressInfoFile = progressInfoFile;
- CheckIntegrityEntryHandle entry = new BtCheckIntegrityEntry(this);
-
- return processCheckIntegrityEntry(entry, e);
- }
- }
- #endif // ENABLE_BITTORRENT
- // TODO I assume here when totallength is set to DownloadContext and it is
- // not 0, then filepath is also set DownloadContext correctly....
- if(_downloadContext->getTotalLength() == 0) {
- return createNextCommand(e, 1);
- }else {
- if(e->_requestGroupMan->isSameFileBeingDownloaded(this)) {
- throw new DownloadFailureException(EX_DUPLICATE_FILE_DOWNLOAD,
- getFilePath().c_str());
- }
- initPieceStorage();
- BtProgressInfoFileHandle infoFile =
- new DefaultBtProgressInfoFile(_downloadContext, _pieceStorage, _option);
- if(!infoFile->exists() && downloadFinishedByFileLength()) {
- return Commands();
- }
- loadAndOpenFile(infoFile);
- return processCheckIntegrityEntry(new StreamCheckIntegrityEntry(0, this), e);
- }
- }
- Commands RequestGroup::processCheckIntegrityEntry(const CheckIntegrityEntryHandle& entry, DownloadEngine* e)
- {
- #ifdef ENABLE_MESSAGE_DIGEST
- if(e->option->get(PREF_CHECK_INTEGRITY) == V_TRUE &&
- entry->isValidationReady()) {
- entry->initValidator();
- CheckIntegrityCommand* command =
- new CheckIntegrityCommand(CUIDCounterSingletonHolder::instance()->newID(), this, e, entry);
- Commands commands;
- commands.push_back(command);
- return commands;
- } else
- #endif // ENABLE_MESSAGE_DIGEST
- {
- return entry->onDownloadIncomplete(e);
- }
- }
- void RequestGroup::initPieceStorage()
- {
- if(_downloadContext->getTotalLength() == 0) {
- UnknownLengthPieceStorageHandle ps = new UnknownLengthPieceStorage(_downloadContext, _option);
- if(!_diskWriterFactory.isNull()) {
- ps->setDiskWriterFactory(_diskWriterFactory);
- }
- _pieceStorage = ps;
- } else {
- DefaultPieceStorageHandle ps = new DefaultPieceStorage(_downloadContext, _option);
- if(!_diskWriterFactory.isNull()) {
- ps->setDiskWriterFactory(_diskWriterFactory);
- }
- _pieceStorage = ps;
- }
- _pieceStorage->initStorage();
- initSegmentMan();
- }
- bool RequestGroup::downloadFinishedByFileLength()
- {
- // assuming that a control file doesn't exist.
- if(!isPreLocalFileCheckEnabled() ||
- _option->get(PREF_ALLOW_OVERWRITE) == V_TRUE ||
- _option->get(PREF_CHECK_INTEGRITY) == V_TRUE &&
- !_downloadContext->getPieceHashes().empty()) {
- return false;
- }
- // TODO consider the case when the getFilePath() returns dir path.
- File outfile(getFilePath());
- if(outfile.exists() && getTotalLength() == outfile.size()) {
- _pieceStorage->markAllPiecesDone();
- _logger->notice(MSG_DOWNLOAD_ALREADY_COMPLETED, _gid, getFilePath().c_str());
- return true;
- } else {
- return false;
- }
- }
- void RequestGroup::loadAndOpenFile(const BtProgressInfoFileHandle& progressInfoFile)
- {
- try {
- if(!isPreLocalFileCheckEnabled()) {
- _pieceStorage->getDiskAdaptor()->initAndOpenFile();
- return;
- }
- if(progressInfoFile->exists()) {
- progressInfoFile->load();
- _pieceStorage->getDiskAdaptor()->openExistingFile();
- } else {
- File outfile(getFilePath());
- if(outfile.exists() && _option->get(PREF_CONTINUE) == V_TRUE) {
- if(getTotalLength() < outfile.size()) {
- throw new DlAbortEx(EX_FILE_LENGTH_MISMATCH_BETWEEN_LOCAL_AND_REMOTE,
- getFilePath().c_str(),
- Util::llitos(outfile.size()).c_str(),
- Util::llitos(getTotalLength()).c_str());
- }
- _pieceStorage->getDiskAdaptor()->openExistingFile();
- _pieceStorage->markPiecesDone(outfile.size());
- } else {
- #ifdef ENABLE_MESSAGE_DIGEST
- if(outfile.exists() && _option->get(PREF_CHECK_INTEGRITY) == V_TRUE) {
- _pieceStorage->getDiskAdaptor()->openExistingFile();
- } else {
- shouldCancelDownloadForSafety();
- _pieceStorage->getDiskAdaptor()->initAndOpenFile();
- }
- #else // ENABLE_MESSAGE_DIGEST
- shouldCancelDownloadForSafety();
- _pieceStorage->getDiskAdaptor()->initAndOpenFile();
- #endif // ENABLE_MESSAGE_DIGEST
- }
- }
- setProgressInfoFile(progressInfoFile);
- } catch(RecoverableException* e) {
- throw new DownloadFailureException(e, EX_DOWNLOAD_ABORTED);
- }
- }
- // assuming that a control file does not exist
- void RequestGroup::shouldCancelDownloadForSafety()
- {
- if(_option->get(PREF_ALLOW_OVERWRITE) == V_TRUE) {
- return;
- }
- File outfile(getFilePath());
- if(outfile.exists()) {
- if(_option->get(PREF_AUTO_FILE_RENAMING) == V_TRUE) {
- if(tryAutoFileRenaming()) {
- _logger->notice(MSG_FILE_RENAMED, getFilePath().c_str());
- } else {
- throw new DownloadFailureException("File renaming failed: %s",
- getFilePath().c_str());
- }
- } else {
- throw new DownloadFailureException(MSG_FILE_ALREADY_EXISTS,
- getFilePath().c_str());
- }
- }
- }
- bool RequestGroup::tryAutoFileRenaming()
- {
- string filepath = getFilePath();
- if(filepath.empty()) {
- return false;
- }
- for(int32_t i = 1; i < 10000; ++i) {
- File newfile(filepath+"."+Util::itos(i));
- if(!newfile.exists()) {
- SingleFileDownloadContextHandle(_downloadContext)->setUFilename(newfile.getBasename());
- return true;
- }
- }
- return false;
- }
- Commands RequestGroup::createNextCommandWithAdj(DownloadEngine* e, int32_t numAdj)
- {
- int32_t numCommand = _numConcurrentCommand == 0 ? _uris.size() : _numConcurrentCommand+numAdj;
- return createNextCommand(e, numCommand, "GET");
- }
- Commands RequestGroup::createNextCommand(DownloadEngine* e, int32_t numCommand, const string& method)
- {
- Commands commands;
- for(;!_uris.empty() && numCommand--; _uris.pop_front()) {
- string uri = _uris.front();
- _spentUris.push_back(uri);
- RequestHandle req = new Request();
- req->setReferer(_option->get(PREF_REFERER));
- req->setMethod(method);
- if(req->setUrl(uri)) {
- commands.push_back(InitiateConnectionCommandFactory::createInitiateConnectionCommand(CUIDCounterSingletonHolder::instance()->newID(), req, this, e));
- } else {
- _logger->error(MSG_UNRECOGNIZED_URI, req->getUrl().c_str());
- }
- }
- return commands;
- }
- string RequestGroup::getFilePath() const
- {
- assert(!_downloadContext.isNull());
- if(_downloadContext.isNull()) {
- return "";
- } else {
- return _downloadContext->getActualBasePath();
- }
- }
- int64_t RequestGroup::getTotalLength() const
- {
- if(_pieceStorage.isNull()) {
- return 0;
- } else {
- if(_pieceStorage->isSelectiveDownloadingMode()) {
- return _pieceStorage->getFilteredTotalLength();
- } else {
- return _pieceStorage->getTotalLength();
- }
- }
- }
- int64_t RequestGroup::getCompletedLength() const
- {
- if(_pieceStorage.isNull()) {
- return 0;
- } else {
- if(_pieceStorage->isSelectiveDownloadingMode()) {
- return _pieceStorage->getFilteredCompletedLength();
- } else {
- return _pieceStorage->getCompletedLength();
- }
- }
- }
- void RequestGroup::validateFilename(const string& expectedFilename,
- const string& actualFilename) const
- {
- if(expectedFilename.empty()) {
- return;
- }
- if(expectedFilename != actualFilename) {
- throw new DlAbortEx(EX_FILENAME_MISMATCH,
- expectedFilename.c_str(),
- actualFilename.c_str());
- }
- }
- void RequestGroup::validateTotalLength(int64_t expectedTotalLength,
- int64_t actualTotalLength) const
- {
- if(expectedTotalLength <= 0) {
- return;
- }
- if(expectedTotalLength != actualTotalLength) {
- throw new DlAbortEx(EX_SIZE_MISMATCH,
- Util::llitos(expectedTotalLength, true).c_str(),
- Util::llitos(actualTotalLength, true).c_str());
- }
- }
- void RequestGroup::validateFilename(const string& actualFilename) const
- {
- validateFilename(_downloadContext->getFileEntries().front()->getBasename(), actualFilename);
- }
- void RequestGroup::validateTotalLength(int64_t actualTotalLength) const
- {
- validateTotalLength(getTotalLength(), actualTotalLength);
- }
- void RequestGroup::increaseStreamConnection()
- {
- ++_numStreamConnection;
- }
- void RequestGroup::decreaseStreamConnection()
- {
- --_numStreamConnection;
- }
- int32_t RequestGroup::getNumConnection() const
- {
- int32_t numConnection = _numStreamConnection;
- #ifdef ENABLE_BITTORRENT
- {
- BtContextHandle btContext = _downloadContext;
- if(!btContext.isNull()) {
- BtRuntimeHandle btRuntime = BT_RUNTIME(btContext);
- if(!btRuntime.isNull()) {
- numConnection += btRuntime->getConnections();
- }
- }
- }
- #endif // ENABLE_BITTORRENT
- return numConnection;
- }
- void RequestGroup::increaseNumCommand()
- {
- ++_numCommand;
- }
- void RequestGroup::decreaseNumCommand()
- {
- --_numCommand;
- }
- TransferStat RequestGroup::calculateStat()
- {
- TransferStat stat;
- #ifdef ENABLE_BITTORRENT
- {
- BtContextHandle btContext = _downloadContext;
- if(!btContext.isNull()) {
- PeerStorageHandle peerStorage = PEER_STORAGE(btContext);
- if(!peerStorage.isNull()) {
- stat = peerStorage->calculateStat();
- }
- }
- }
- #endif // ENABLE_BITTORRENT
- if(!_segmentMan.isNull()) {
- stat.setDownloadSpeed(stat.getDownloadSpeed()+_segmentMan->calculateDownloadSpeed());
- }
- return stat;
- }
- void RequestGroup::setHaltRequested(bool f)
- {
- _haltRequested = f;
- #ifdef ENABLE_BITTORRENT
- {
- BtContextHandle btContext = _downloadContext;
- if(!btContext.isNull()) {
- BtRuntimeHandle btRuntime = BT_RUNTIME(btContext);
- if(!btRuntime.isNull()) {
- btRuntime->setHalt(f);
- }
- }
- }
- #endif // ENABLE_BITTORRENT
- }
- void RequestGroup::setForceHaltRequested(bool f)
- {
- setHaltRequested(f);
- _forceHaltRequested = f;
- }
- void RequestGroup::releaseRuntimeResource()
- {
- #ifdef ENABLE_BITTORRENT
- BtContextHandle btContext = _downloadContext;
- if(!btContext.isNull()) {
- BtContextHandle btContextInReg = BtRegistry::getBtContext(btContext->getInfoHashAsString());
- if(!btContextInReg.isNull() &&
- btContextInReg->getOwnerRequestGroup()->getGID() ==
- btContext->getOwnerRequestGroup()->getGID()) {
- BtRegistry::unregister(btContext->getInfoHashAsString());
- }
- }
- #endif // ENABLE_BITTORRENT
- if(!_pieceStorage.isNull()) {
- _pieceStorage->removeAdvertisedPiece(0);
- }
- }
- void RequestGroup::preDownloadProcessing()
- {
- _logger->debug("Finding PreDownloadHandler for path %s.", getFilePath().c_str());
- try {
- for(PreDownloadHandlers::const_iterator itr = _preDownloadHandlers.begin();
- itr != _preDownloadHandlers.end(); ++itr) {
- if((*itr)->canHandle(this)) {
- (*itr)->execute(this);
- return;
- }
- }
- } catch(RecoverableException* ex) {
- _logger->error(EX_EXCEPTION_CAUGHT, ex);
- delete ex;
- return;
- }
- _logger->debug("No PreDownloadHandler found.");
- return;
- }
- RequestGroups RequestGroup::postDownloadProcessing()
- {
- _logger->debug("Finding PostDownloadHandler for path %s.", getFilePath().c_str());
- try {
- for(PostDownloadHandlers::const_iterator itr = _postDownloadHandlers.begin();
- itr != _postDownloadHandlers.end(); ++itr) {
- if((*itr)->canHandle(this)) {
- return (*itr)->getNextRequestGroups(this);
- }
- }
- } catch(RecoverableException* ex) {
- _logger->error(EX_EXCEPTION_CAUGHT, ex);
- delete ex;
- return RequestGroups();
- }
- _logger->debug("No PostDownloadHandler found.");
- return RequestGroups();
- }
- void RequestGroup::initializePreDownloadHandler()
- {
- #ifdef ENABLE_BITTORRENT
- if(_option->get(PREF_FOLLOW_TORRENT) == V_MEM) {
- _preDownloadHandlers.push_back(DownloadHandlerFactory::getBtPreDownloadHandler());
- }
- #endif // ENABLE_BITTORRENT
- #ifdef ENABLE_METALINK
- if(_option->get(PREF_FOLLOW_METALINK) == V_MEM) {
- _preDownloadHandlers.push_back(DownloadHandlerFactory::getMetalinkPreDownloadHandler());
- }
- #endif // ENABLE_METALINK
- }
- void RequestGroup::initializePostDownloadHandler()
- {
- #ifdef ENABLE_BITTORRENT
- if(_option->get(PREF_FOLLOW_TORRENT) == V_TRUE ||
- _option->get(PREF_FOLLOW_TORRENT) == V_MEM) {
- _postDownloadHandlers.push_back(DownloadHandlerFactory::getBtPostDownloadHandler());
- }
- #endif // ENABLE_BITTORRENT
- #ifdef ENABLE_METALINK
- if(_option->get(PREF_FOLLOW_METALINK) == V_TRUE ||
- _option->get(PREF_FOLLOW_METALINK) == V_MEM) {
- _postDownloadHandlers.push_back(DownloadHandlerFactory::getMetalinkPostDownloadHandler());
- }
- #endif // ENABLE_METALINK
- }
- Strings RequestGroup::getUris() const
- {
- Strings temp(_spentUris.begin(), _spentUris.end());
- temp.insert(temp.end(), _uris.begin(), _uris.end());
- return temp;
- }
- bool RequestGroup::isDependencyResolved()
- {
- if(_dependency.isNull()) {
- return true;
- }
- return _dependency->resolve();
- }
- void RequestGroup::setSegmentManFactory(const SegmentManFactoryHandle& segmentManFactory)
- {
- _segmentManFactory = segmentManFactory;
- }
- void RequestGroup::dependsOn(const DependencyHandle& dep)
- {
- _dependency = dep;
- }
- void RequestGroup::setDiskWriterFactory(const DiskWriterFactoryHandle& diskWriterFactory)
- {
- _diskWriterFactory = diskWriterFactory;
- }
- DiskWriterFactoryHandle RequestGroup::getDiskWriterFactory() const
- {
- return _diskWriterFactory;
- }
- void RequestGroup::addPostDownloadHandler(const PostDownloadHandlerHandle& handler)
- {
- _postDownloadHandlers.push_back(handler);
- }
- void RequestGroup::addPreDownloadHandler(const PreDownloadHandlerHandle& handler)
- {
- _preDownloadHandlers.push_back(handler);
- }
- void RequestGroup::clearPostDowloadHandler()
- {
- _postDownloadHandlers.clear();
- }
- void RequestGroup::clearPreDowloadHandler()
- {
- _preDownloadHandlers.clear();
- }
- SegmentManHandle RequestGroup::getSegmentMan() const
- {
- return _segmentMan;
- }
- DownloadContextHandle RequestGroup::getDownloadContext() const
- {
- return _downloadContext;
- }
- void RequestGroup::setDownloadContext(const DownloadContextHandle& downloadContext)
- {
- _downloadContext = downloadContext;
- }
- PieceStorageHandle RequestGroup::getPieceStorage() const
- {
- return _pieceStorage;
- }
- void RequestGroup::setPieceStorage(const PieceStorageHandle& pieceStorage)
- {
- _pieceStorage = pieceStorage;
- }
- BtProgressInfoFileHandle RequestGroup::getProgressInfoFile() const
- {
- return _progressInfoFile;
- }
- void RequestGroup::setProgressInfoFile(const BtProgressInfoFileHandle& progressInfoFile)
- {
- _progressInfoFile = progressInfoFile;
- }
- bool RequestGroup::needsFileAllocation() const
- {
- return isFileAllocationEnabled() &&
- _option->getAsLLInt(PREF_NO_FILE_ALLOCATION_LIMIT) <= getTotalLength() &&
- !_pieceStorage->getDiskAdaptor()->fileAllocationIterator()->finished();
- }
- DownloadResultHandle RequestGroup::createDownloadResult() const
- {
- Strings uris = getUris();
- return new DownloadResult(_gid,
- getFilePath(),
- getTotalLength(),
- uris.empty() ? "":uris.front(),
- uris.size(),
- downloadFinished()?
- DownloadResult::FINISHED :
- DownloadResult::NOT_YET);
- }
|