diff options
Diffstat (limited to 'src/RequestGroup.cc')
| -rw-r--r-- | src/RequestGroup.cc | 258 |
1 files changed, 258 insertions, 0 deletions
diff --git a/src/RequestGroup.cc b/src/RequestGroup.cc new file mode 100644 index 00000000..40f9df80 --- /dev/null +++ b/src/RequestGroup.cc @@ -0,0 +1,258 @@ +/* <!-- 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 "prefs.h" +#include "DefaultDiskWriter.h" +#include "RequestFactory.h" +#include "InitiateConnectionCommandFactory.h" +#include "CUIDCounter.h" +#include "DlAbortEx.h" +#include "File.h" +#include "message.h" +#include "DlAbortEx.h" +#include "Util.h" +#include "CheckIntegrityCommand.h" +#include "FatalException.h" + +SegmentManHandle RequestGroup::initSegmentMan() +{ + _segmentMan = _segmentManFactory->createNewInstance(); + /* + _segmentMan = new SegmentMan(); + _segmentMan->diskWriter = new DefaultDiskWriter();// DefaultDiskWriter::createNewDiskWriter(_option); + _segmentMan->dir = _option->get(PREF_DIR); + // TODO disable this in multi-simultaneous download mode. + _segmentMan->ufilename = _option->get(PREF_OUT); + _segmentMan->option = _option; + */ + return _segmentMan; +} + +Commands RequestGroup::getNextCommand(DownloadEngine* e, int32_t maxNum, const string& method) +{ + Commands commands; + for(;!_uris.empty() && commands.size() < (size_t)maxNum; _uris.pop_front()) { + string uri = _uris.front(); + _spentUris.push_back(uri); + RequestHandle req = RequestFactorySingletonHolder::instance()->createRequest(); + 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->info(_("Unrecognized URL or unsupported protocol: %s\n"), + req->getUrl().c_str()); + } + } + return commands; +} + +void RequestGroup::initBitfield() +{ + _segmentMan->initBitfield(_option->getAsInt(PREF_SEGMENT_SIZE), + _segmentMan->totalSize); +} + +void RequestGroup::openExistingFile() +{ + _segmentMan->diskWriter->openExistingFile(_segmentMan->getFilePath()); +} + +void RequestGroup::markAllPiecesDone() +{ + _segmentMan->markAllPiecesDone(); +} + +void RequestGroup::markExistingPiecesDone() +{ + _segmentMan->markPieceDone(File(_segmentMan->getFilePath()).size()); +} + +void RequestGroup::markPieceDone(int64_t length) +{ + _segmentMan->markPieceDone(length); +} + +void RequestGroup::shouldCancelDownloadForSafety() +{ + if(_segmentMan->shouldCancelDownloadForSafety()) { + logger->notice(MSG_FILE_ALREADY_EXISTS, + _segmentMan->getFilePath().c_str(), + _segmentMan->getSegmentFilePath().c_str()); + + throw new FatalException(EX_FILE_ALREADY_EXISTS, + _segmentMan->getFilePath().c_str(), + _segmentMan->getSegmentFilePath().c_str()); + } +} + +void RequestGroup::initAndOpenFile() +{ + _segmentMan->diskWriter->initAndOpenFile(_segmentMan->getFilePath()); +} + +bool RequestGroup::needsFileAllocation() const +{ + return _option->get(PREF_FILE_ALLOCATION) == V_PREALLOC + && File(_segmentMan->getFilePath()).size() < _segmentMan->totalSize; +} + +bool RequestGroup::fileExists() const +{ + return _segmentMan->fileExists(); +} + +bool RequestGroup::segmentFileExists() const +{ + return _segmentMan->segmentFileExists(); +} + +string RequestGroup::getFilePath() const +{ + if(_segmentMan.isNull()) { + return ""; + } else { + return _segmentMan->getFilePath(); + } +} + +void RequestGroup::loadAndOpenFile() +{ + bool segFileExists = segmentFileExists(); + if(segFileExists) { + load(); + openExistingFile(); + } else if(fileExists() && _option->get(PREF_CONTINUE) == V_TRUE) { + File existingFile(getFilePath()); + if(getTotalLength() < existingFile.size()) { + throw new DlAbortEx("Invalid file length. Cannot continue download %s: local %s, remote %s", + getFilePath().c_str(), + Util::llitos(existingFile.size()).c_str(), + Util::llitos(getTotalLength()).c_str()); + } + initBitfield(); + openExistingFile(); + } else { + shouldCancelDownloadForSafety(); + initBitfield(); + if(fileExists() && _option->get(PREF_CHECK_INTEGRITY) == V_TRUE) { + openExistingFile(); + } else { + initAndOpenFile(); + } + } +} + +bool RequestGroup::downloadFinishedByFileLength() +{ + if(_segmentMan->segmentFileExists()) { + return false; + } + File existingFile(getFilePath()); + if(existingFile.exists() && + getTotalLength() == existingFile.size()) { + _segmentMan->downloadStarted = true; + initBitfield(); + _segmentMan->markAllPiecesDone(); + return true; + } else { + return false; + } +} + +void RequestGroup::prepareForNextAction(int cuid, const RequestHandle& req, DownloadEngine* e) +{ + File existingFile(getFilePath()); + if(existingFile.size() > 0 && _option->get(PREF_CHECK_INTEGRITY) == V_TRUE) { + CheckIntegrityCommand* command = new CheckIntegrityCommand(cuid, req, this, e); + command->initValidator(); + e->commands.push_back(command); + } else if(needsFileAllocation()) { + e->_fileAllocationMan->pushFileAllocationEntry(new FileAllocationEntry(cuid, req, this)); + } else { + int32_t numCommandsToGenerate = 15; + Commands commands = getNextCommand(e, numCommandsToGenerate); + Command* command = InitiateConnectionCommandFactory::createInitiateConnectionCommand(cuid, req, this, e); + commands.push_front(command); + e->addCommand(commands); + } +} + +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, + expectedTotalLength, + actualTotalLength); + } +} + +void RequestGroup::validateFilename(const string& actualFilename) const +{ + validateFilename(_segmentMan->filename, actualFilename); +} + +void RequestGroup::validateTotalLength(int64_t actualTotalLength) const +{ + validateTotalLength(_segmentMan->totalSize, actualTotalLength); +} + +void RequestGroup::validateFilenameByHint(const string& actualFilename) const +{ + validateFilename(_hintFilename, actualFilename); +} + +void RequestGroup::validateTotalLengthByHint(int64_t actualTotalLength) const +{ + validateTotalLength(_hintTotalLength, actualTotalLength); +} |