summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorTatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>2007-05-20 13:57:56 +0000
committerTatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>2007-05-20 13:57:56 +0000
commit079d2844ecde061d45c6884aa0ffee7fdc05de63 (patch)
tree9b2c47e8f2bb091daa442f4b97c8597ac03e2583 /src
parenteaf2d53a91d4019683bc6caf25c5a143f88a84b4 (diff)
New files
Diffstat (limited to 'src')
-rw-r--r--src/AbstractSegmentManFactory.h51
-rw-r--r--src/CUIDCounter.h59
-rw-r--r--src/CheckIntegrityCommand.cc90
-rw-r--r--src/CheckIntegrityCommand.h71
-rw-r--r--src/ChunkChecksum.h96
-rw-r--r--src/DefaultSegmentManFactory.cc48
-rw-r--r--src/DefaultSegmentManFactory.h51
-rw-r--r--src/FileAllocationCommand.cc73
-rw-r--r--src/FileAllocationCommand.h63
-rw-r--r--src/FileAllocationDispatcherCommand.cc62
-rw-r--r--src/FileAllocationDispatcherCommand.h54
-rw-r--r--src/FileAllocationEntry.cc54
-rw-r--r--src/FileAllocationEntry.h102
-rw-r--r--src/FileAllocationMan.h89
-rw-r--r--src/FillRequestGroupCommand.cc45
-rw-r--r--src/FillRequestGroupCommand.h66
-rw-r--r--src/IteratableChunkChecksumValidator.cc81
-rw-r--r--src/IteratableChunkChecksumValidator.h91
-rw-r--r--src/MultiUrlRequestInfo.cc119
-rw-r--r--src/MultiUrlRequestInfo.h61
-rw-r--r--src/RequestGroup.cc258
-rw-r--r--src/RequestGroup.h245
-rw-r--r--src/RequestGroupMan.cc120
-rw-r--r--src/RequestGroupMan.h164
-rw-r--r--src/SegmentManFactory.h50
-rw-r--r--src/TrackerSegmentManFactory.cc45
-rw-r--r--src/TrackerSegmentManFactory.h51
-rw-r--r--src/UriFileListParser.cc54
-rw-r--r--src/UriFileListParser.h53
29 files changed, 2466 insertions, 0 deletions
diff --git a/src/AbstractSegmentManFactory.h b/src/AbstractSegmentManFactory.h
new file mode 100644
index 00000000..cfcd8ea8
--- /dev/null
+++ b/src/AbstractSegmentManFactory.h
@@ -0,0 +1,51 @@
+/* <!-- 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 --> */
+#ifndef _D_ABSTRACT_SEGMENT_MAN_FACTORY_H_
+#define _D_ABSTRACT_SEGMENT_MAN_FACTORY_H_
+
+#include "SegmentManFactory.h"
+
+class AbstractSegmentManFactory : public SegmentManFactory {
+protected:
+ const Option* _option;
+public:
+ AbstractSegmentManFactory(const Option* option):_option(option) {}
+
+ virtual ~AbstractSegmentManFactory() {}
+};
+
+typedef SharedHandle<AbstractSegmentManFactory> AbstractSegmentManFactoryHandle;
+
+#endif // _D_ABSTRACT_SEGMENT_MAN_FACTORY_H_
diff --git a/src/CUIDCounter.h b/src/CUIDCounter.h
new file mode 100644
index 00000000..d251cceb
--- /dev/null
+++ b/src/CUIDCounter.h
@@ -0,0 +1,59 @@
+/* <!-- 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 --> */
+#ifndef _D_CUID_COUNTER_H_
+#define _D_CUID_COUNTER_H_
+
+#include "common.h"
+
+typedef int32_t CUID;
+
+class CUIDCounter {
+private:
+ int32_t count;
+public:
+ CUIDCounter():count(0) {}
+
+ ~CUIDCounter() {}
+
+ CUID newID()
+ {
+ return ++count;
+ }
+};
+
+typedef SharedHandle<CUIDCounter> CUIDCounterHandle;
+typedef SingletonHolder<CUIDCounterHandle> CUIDCounterSingletonHolder;
+
+#endif // _D_CUID_COUNTER_H_
diff --git a/src/CheckIntegrityCommand.cc b/src/CheckIntegrityCommand.cc
new file mode 100644
index 00000000..7a4bf430
--- /dev/null
+++ b/src/CheckIntegrityCommand.cc
@@ -0,0 +1,90 @@
+/* <!-- 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 "CheckIntegrityCommand.h"
+#include "FileAllocationEntry.h"
+#include "InitiateConnectionCommandFactory.h"
+#include "DlAbortEx.h"
+#include "message.h"
+
+void CheckIntegrityCommand::initValidator()
+{
+ _validator = new IteratableChunkChecksumValidator();
+ _validator->setChunkChecksum(_requestGroup->getChunkChecksum());
+ _validator->setDiskWriter(_requestGroup->getSegmentMan()->diskWriter);
+ _validator->setBitfield(_requestGroup->getSegmentMan()->getBitfield());
+ if(!_validator->canValidate()) {
+ // insufficient checksums.
+ throw new DlAbortEx("Insufficient checksums.");
+ }
+ _validator->init();
+}
+
+bool CheckIntegrityCommand::execute()
+{
+ setStatusRealtime();
+ _e->noWait = true;
+
+ try {
+ _validator->validateChunk();
+ if(_validator->finished()) {
+ if(_requestGroup->downloadFinished()) {
+ logger->notice(MSG_DOWNLOAD_ALREADY_COMPLETED, cuid, _requestGroup->getFilePath().c_str());
+ return true;
+ }
+ if(_requestGroup->needsFileAllocation()) {
+ _e->_fileAllocationMan->pushFileAllocationEntry(new FileAllocationEntry(cuid, _req, _requestGroup));
+ } else {
+ int32_t numCommandsToGenerate = 15;
+ Commands commands = _requestGroup->getNextCommand(_e, numCommandsToGenerate);
+ Command* command = InitiateConnectionCommandFactory::createInitiateConnectionCommand(cuid, _req, _requestGroup, _e);
+ commands.push_front(command);
+ _e->addCommand(commands);
+ }
+ return true;
+ } else {
+ _e->commands.push_back(this);
+ return false;
+ }
+ } catch(Exception* e) {
+ _requestGroup->getSegmentMan()->errors++;
+ logger->error("CUID#%d - Exception caught while validating file integrity.", e, cuid);
+ delete e;
+ logger->error(MSG_DOWNLOAD_NOT_COMPLETE, cuid, _requestGroup->getFilePath().c_str());
+ // TODO this is wrong. There may exist invalid chunk data before catching
+ // exception. Fix this.
+ _requestGroup->markPieceDone(_validator->getCurrentOffset());
+ return true;
+ }
+}
diff --git a/src/CheckIntegrityCommand.h b/src/CheckIntegrityCommand.h
new file mode 100644
index 00000000..88f19eb6
--- /dev/null
+++ b/src/CheckIntegrityCommand.h
@@ -0,0 +1,71 @@
+/* <!-- 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 --> */
+#ifndef _D_CHECK_INTEGRITY_COMMAND_H_
+#define _D_CHECK_INTEGRITY_COMMAND_H_
+
+#include "Command.h"
+#include "Request.h"
+#include "RequestGroup.h"
+#include "DownloadEngine.h"
+#include "IteratableChunkChecksumValidator.h"
+
+class CheckIntegrityCommand : public Command {
+private:
+ RequestHandle _req;
+ RequestGroup* _requestGroup;
+ DownloadEngine* _e;
+ IteratableChunkChecksumValidatorHandle _validator;
+public:
+ CheckIntegrityCommand(int cuid, const RequestHandle& req, RequestGroup* requestGroup, DownloadEngine* e):
+ Command(cuid),
+ _req(req),
+ _requestGroup(requestGroup),
+ _e(e),
+ _validator(0)
+ {
+ ++_requestGroup->numConnection;
+ }
+
+ virtual ~CheckIntegrityCommand()
+ {
+ --_requestGroup->numConnection;
+ }
+
+ void initValidator();
+
+ virtual bool execute();
+};
+
+#endif // _D_CHECK_INTEGRITY_COMMAND_H_
diff --git a/src/ChunkChecksum.h b/src/ChunkChecksum.h
new file mode 100644
index 00000000..0df27efa
--- /dev/null
+++ b/src/ChunkChecksum.h
@@ -0,0 +1,96 @@
+/* <!-- 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 --> */
+#ifndef _D_CHUNK_CHECKSUM_H_
+#define _D_CHUNK_CHECKSUM_H_
+
+#include "common.h"
+#include "messageDigest.h"
+
+class ChunkChecksum {
+private:
+ MessageDigestContext::DigestAlgo _algo;
+ Strings _checksums;
+ int32_t _checksumLength;
+public:
+ ChunkChecksum(const MessageDigestContext::DigestAlgo& algo,
+ const Strings& checksums,
+ int32_t checksumLength):
+ _algo(algo),
+ _checksums(checksums),
+ _checksumLength(checksumLength) {}
+
+ bool validateChunk(const string& actualChecksum,
+ int32_t checksumIndex) const
+ {
+ if(checksumIndex < (int32_t)_checksums.size()) {
+ return actualChecksum == getChecksum(checksumIndex);
+ } else {
+ return false;
+ }
+ }
+
+ int64_t getEstimatedDataLength() const
+ {
+ return ((int64_t)_checksumLength)*_checksums.size();
+ }
+
+ int32_t countChecksum() const
+ {
+ return _checksums.size();
+ }
+
+ string getChecksum(int32_t index) const
+ {
+ if(index < (int32_t)_checksums.size()) {
+ return _checksums[index];
+ } else {
+ return "";
+ }
+ }
+
+ int32_t getChecksumLength() const
+ {
+ return _checksumLength;
+ }
+
+ MessageDigestContext::DigestAlgo getAlgo() const
+ {
+ return _algo;
+ }
+};
+
+typedef SharedHandle<ChunkChecksum> ChunkChecksumHandle;
+
+#endif // _D_CHUNK_CHECKSUM_H_
diff --git a/src/DefaultSegmentManFactory.cc b/src/DefaultSegmentManFactory.cc
new file mode 100644
index 00000000..b6d9dee2
--- /dev/null
+++ b/src/DefaultSegmentManFactory.cc
@@ -0,0 +1,48 @@
+/* <!-- 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 "DefaultSegmentManFactory.h"
+#include "prefs.h"
+#include "DefaultDiskWriter.h"
+
+SegmentManHandle DefaultSegmentManFactory::createNewInstance()
+{
+ SegmentManHandle segmentMan = new SegmentMan();
+ segmentMan->diskWriter = new DefaultDiskWriter();
+ 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;
+}
diff --git a/src/DefaultSegmentManFactory.h b/src/DefaultSegmentManFactory.h
new file mode 100644
index 00000000..0a8dad1d
--- /dev/null
+++ b/src/DefaultSegmentManFactory.h
@@ -0,0 +1,51 @@
+/* <!-- 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 --> */
+#ifndef _D_DEFAULT_SEGMENT_MAN_FACTORY_H_
+#define _D_DEFAULT_SEGMENT_MAN_FACTORY_H_
+
+#include "AbstractSegmentManFactory.h"
+
+class DefaultSegmentManFactory : public AbstractSegmentManFactory {
+public:
+ DefaultSegmentManFactory(const Option* option):AbstractSegmentManFactory(option) {}
+
+ virtual ~DefaultSegmentManFactory() {}
+
+ virtual SegmentManHandle createNewInstance();
+};
+
+typedef SharedHandle<DefaultSegmentManFactory> DefaultSegmentManFactoryHandle;
+
+#endif // _D_DEFAULT_SEGMENT_MAN_FACTORY_H_
diff --git a/src/FileAllocationCommand.cc b/src/FileAllocationCommand.cc
new file mode 100644
index 00000000..92f26b6f
--- /dev/null
+++ b/src/FileAllocationCommand.cc
@@ -0,0 +1,73 @@
+/* <!-- 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 "FileAllocationCommand.h"
+#include "InitiateConnectionCommandFactory.h"
+#include "message.h"
+
+bool FileAllocationCommand::execute()
+{
+ setStatusRealtime();
+ _e->noWait = true;
+ try {
+ _fileAllocationEntry->allocateChunk();
+
+ if(_fileAllocationEntry->allocationFinished()) {
+ int64_t totalLength = _requestGroup->getSegmentMan()->totalSize;
+ logger->debug("%d seconds to allocate %lld byte(s)",
+ _timer.difference(), totalLength);
+
+ _e->_fileAllocationMan->markCurrentFileAllocationEntryDone();
+
+ int32_t numCommandsToGenerate = 15;
+ Commands commands = _requestGroup->getNextCommand(_e, numCommandsToGenerate);
+
+ Command* command = InitiateConnectionCommandFactory::createInitiateConnectionCommand(cuid, _req, _requestGroup, _e);
+
+ commands.push_front(command);
+
+ _e->addCommand(commands);
+ return true;
+ } else {
+ _e->commands.push_back(this);
+ return false;
+ }
+ } catch(Exception* e) {
+ _requestGroup->getSegmentMan()->errors++;
+ logger->error("CUID#%d - Exception caught while allocating file space.", e, cuid);
+ delete e;
+ logger->notice(MSG_DOWNLOAD_NOT_COMPLETE, cuid, _requestGroup->getFilePath().c_str());
+ return true;
+ }
+}
diff --git a/src/FileAllocationCommand.h b/src/FileAllocationCommand.h
new file mode 100644
index 00000000..0ca78937
--- /dev/null
+++ b/src/FileAllocationCommand.h
@@ -0,0 +1,63 @@
+/* <!-- 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 --> */
+#ifndef _D_FILE_ALLOCATION_COMMAND_H_
+#define _D_FILE_ALLOCATION_COMMAND_H_
+
+#include "Command.h"
+#include "Request.h"
+#include "RequestGroup.h"
+#include "DownloadEngine.h"
+#include "TimeA2.h"
+#include "FileAllocationEntry.h"
+
+class FileAllocationCommand : public Command {
+private:
+ RequestHandle _req;
+ RequestGroup* _requestGroup;
+ DownloadEngine* _e;
+ FileAllocationEntryHandle _fileAllocationEntry;
+ Time _timer;
+public:
+ FileAllocationCommand(int cuid, const RequestHandle& req, RequestGroup* requestGroup, DownloadEngine* e, const FileAllocationEntryHandle& fileAllocationEntry):
+ Command(cuid),
+ _req(req),
+ _requestGroup(requestGroup),
+ _e(e),
+ _fileAllocationEntry(fileAllocationEntry) {}
+
+ virtual bool execute();
+};
+
+#endif // _D_FILE_ALLOCATION_COMMAND_H_
diff --git a/src/FileAllocationDispatcherCommand.cc b/src/FileAllocationDispatcherCommand.cc
new file mode 100644
index 00000000..cbad7dcd
--- /dev/null
+++ b/src/FileAllocationDispatcherCommand.cc
@@ -0,0 +1,62 @@
+/* <!-- 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 "FileAllocationDispatcherCommand.h"
+#include "FileAllocationCommand.h"
+
+bool FileAllocationDispatcherCommand::execute()
+{
+ if(_e->_requestGroupMan->downloadFinished()) {
+ return true;
+ }
+
+ if(!_e->_fileAllocationMan->isFileAllocationBeingExecuted() &&
+ _e->_fileAllocationMan->nextFileAllocationEntryExists()) {
+ FileAllocationEntryHandle entry = _e->_fileAllocationMan->popNextFileAllocationEntry();
+ logger->info("Dispatching FileAllocationCommand for CUID#%d.",
+ entry->getCUID());
+ FileAllocationCommand* command =
+ new FileAllocationCommand(entry->getCUID(),
+ entry->getCurrentRequest(),
+ entry->getRequestGroup(),
+ _e,
+ _e->_fileAllocationMan->getCurrentFileAllocationEntry());
+ _e->commands.push_back(command);
+ _e->noWait = true;
+ }
+
+ _e->commands.push_back(this);
+
+ return false;
+}
diff --git a/src/FileAllocationDispatcherCommand.h b/src/FileAllocationDispatcherCommand.h
new file mode 100644
index 00000000..e52681d6
--- /dev/null
+++ b/src/FileAllocationDispatcherCommand.h
@@ -0,0 +1,54 @@
+/* <!-- 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 --> */
+#ifndef _D_FILE_ALLOCATION_DISPATCHER_COMMAND_H_
+#define _D_FILE_ALLOCATION_DISPATCHER_COMMAND_H_
+
+#include "Command.h"
+#include "DownloadEngine.h"
+
+class FileAllocationDispatcherCommand : public Command {
+private:
+ DownloadEngine* _e;
+public:
+ FileAllocationDispatcherCommand(int cuid, DownloadEngine* e):
+ Command(cuid), _e(e)
+ {
+ setStatusRealtime();
+ }
+
+ virtual bool execute();
+};
+
+#endif // _D_FILE_ALLOCATION_DISPATCHER_COMMAND_H_
diff --git a/src/FileAllocationEntry.cc b/src/FileAllocationEntry.cc
new file mode 100644
index 00000000..77690475
--- /dev/null
+++ b/src/FileAllocationEntry.cc
@@ -0,0 +1,54 @@
+/* <!-- 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 "FileAllocationEntry.h"
+
+#define BUFSIZE 16*1024
+
+void FileAllocationEntry::allocateChunk()
+{
+ int32_t bufSize = BUFSIZE;
+ char buf[BUFSIZE];
+ memset(buf, 0, bufSize);
+
+ _requestGroup->getSegmentMan()->diskWriter->writeData(buf, bufSize, _offset);
+ _offset += bufSize;
+
+ int64_t totalLength = _requestGroup->getSegmentMan()->totalSize;
+ if(totalLength < _offset) {
+ _requestGroup->getSegmentMan()->diskWriter->truncate(totalLength);
+ _offset = totalLength;
+ }
+}
+
diff --git a/src/FileAllocationEntry.h b/src/FileAllocationEntry.h
new file mode 100644
index 00000000..f3f9446c
--- /dev/null
+++ b/src/FileAllocationEntry.h
@@ -0,0 +1,102 @@
+/* <!-- 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 --> */
+#ifndef _D_FILE_ALLOCATION_ENTRY_H_
+#define _D_FILE_ALLOCATION_ENTRY_H_
+
+#include "common.h"
+#include "Request.h"
+#include "RequestGroup.h"
+
+class FileAllocationEntry {
+private:
+ int _cuid;
+ RequestHandle _currentRequest;
+ RequestGroup* _requestGroup;
+ int64_t _offset;
+public:
+ FileAllocationEntry(int cuid,
+ const RequestHandle& currentRequest,
+ RequestGroup* requestGroup,
+ int64_t offset = 0):
+ _cuid(cuid),
+ _currentRequest(currentRequest),
+ _requestGroup(requestGroup),
+ _offset(offset)
+ {
+ ++_requestGroup->numConnection;
+ }
+
+ ~FileAllocationEntry()
+ {
+ --_requestGroup->numConnection;
+ }
+
+ int getCUID() const
+ {
+ return _cuid;
+ }
+
+ RequestHandle getCurrentRequest() const
+ {
+ return _currentRequest;
+ }
+
+ RequestGroup* getRequestGroup() const
+ {
+ return _requestGroup;
+ }
+
+ void setOffset(int64_t offset)
+ {
+ _offset = offset;
+ }
+
+ int64_t getOffset() const
+ {
+ return _offset;
+ }
+
+ void allocateChunk();
+
+ bool allocationFinished() const
+ {
+ return _requestGroup->getSegmentMan()->totalSize <= _offset;
+ }
+};
+
+typedef SharedHandle<FileAllocationEntry> FileAllocationEntryHandle;
+typedef deque<FileAllocationEntryHandle> FileAllocationEntries;
+
+#endif // _D_FILE_ALLOCATION_ENTRY_H_
diff --git a/src/FileAllocationMan.h b/src/FileAllocationMan.h
new file mode 100644
index 00000000..527575a5
--- /dev/null
+++ b/src/FileAllocationMan.h
@@ -0,0 +1,89 @@
+/* <!-- 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 --> */
+#ifndef _D_FILE_ALLOCATION_MAN_H_
+#define _D_FILE_ALLOCATION_MAN_H_
+
+#include "common.h"
+#include "Request.h"
+#include "RequestGroup.h"
+#include "FileAllocationEntry.h"
+
+class FileAllocationMan {
+private:
+ FileAllocationEntries _fileAllocationEntries;
+ FileAllocationEntryHandle _currentFileAllocationEntry;
+public:
+ FileAllocationMan():_currentFileAllocationEntry(0) {}
+
+ bool isFileAllocationBeingExecuted() const
+ {
+ return _currentFileAllocationEntry.get() != 0;
+ }
+
+ FileAllocationEntryHandle getCurrentFileAllocationEntry()
+ {
+ return _currentFileAllocationEntry;
+ }
+
+ void markCurrentFileAllocationEntryDone()
+ {
+ _currentFileAllocationEntry = 0;
+ }
+
+ bool nextFileAllocationEntryExists() const
+ {
+ return !_fileAllocationEntries.empty();
+ }
+
+ FileAllocationEntryHandle popNextFileAllocationEntry()
+ {
+ if(!nextFileAllocationEntryExists()) {
+ return 0;
+ }
+ FileAllocationEntryHandle entry = _fileAllocationEntries.front();
+ _fileAllocationEntries.pop_front();
+ _currentFileAllocationEntry = entry;
+ return entry;
+ }
+
+ void pushFileAllocationEntry(const FileAllocationEntryHandle& entry)
+ {
+ _fileAllocationEntries.push_back(entry);
+ }
+};
+
+typedef SharedHandle<FileAllocationMan> FileAllocationManHandle;
+
+#endif // _D_FILE_ALLOCATION_MAN_H_
diff --git a/src/FillRequestGroupCommand.cc b/src/FillRequestGroupCommand.cc
new file mode 100644
index 00000000..5c4b6399
--- /dev/null
+++ b/src/FillRequestGroupCommand.cc
@@ -0,0 +1,45 @@
+/* <!-- 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 "FillRequestGroupCommand.h"
+
+bool FillRequestGroupCommand::execute()
+{
+ _e->_requestGroupMan->fillRequestGroupFromReserver(_e);
+ if(_e->_requestGroupMan->downloadFinished()) {
+ return true;
+ }
+ _e->commands.push_back(this);
+ return false;
+}
diff --git a/src/FillRequestGroupCommand.h b/src/FillRequestGroupCommand.h
new file mode 100644
index 00000000..68ffa9d2
--- /dev/null
+++ b/src/FillRequestGroupCommand.h
@@ -0,0 +1,66 @@
+/* <!-- 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 --> */
+#ifndef _D_FILL_REQUEST_GROUP_COMMAND_H_
+#define _D_FILL_REQUEST_GROUP_COMMAND_H_
+
+#include "Command.h"
+#include "TimeA2.h"
+#include "RequestGroup.h"
+#include "DownloadEngine.h"
+
+class FillRequestGroupCommand : public Command {
+private:
+ RequestGroups _reservedRequestGroups;
+ DownloadEngine* _e;
+ int32_t _interval;
+ Time _checkPoint;
+public:
+ FillRequestGroupCommand(int cuid, DownloadEngine* e, int32_t interval):
+ Command(cuid),
+ _e(e),
+ _interval(interval)
+ {
+ setStatusRealtime();
+ }
+
+ virtual bool execute();
+
+ void setInterval(int32_t interval)
+ {
+ _interval = interval;
+ }
+};
+
+#endif // _D_FILL_REQUEST_GROUP_COMMAND_H_
diff --git a/src/IteratableChunkChecksumValidator.cc b/src/IteratableChunkChecksumValidator.cc
new file mode 100644
index 00000000..0fa2fa10
--- /dev/null
+++ b/src/IteratableChunkChecksumValidator.cc
@@ -0,0 +1,81 @@
+/* <!-- 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 "IteratableChunkChecksumValidator.h"
+#include "Util.h"
+#include "message.h"
+
+void IteratableChunkChecksumValidator::validateChunk()
+{
+ if(!finished()) {
+ string actualChecksum = calculateActualChecksum();
+
+
+ if(!_chunkChecksum->validateChunk(actualChecksum, _currentIndex)) {
+ int64_t offset = ((int64_t)_currentIndex)*_chunkChecksum->getChecksumLength();
+ // wrong checksum
+ logger->info(EX_INVALID_CHUNK_CHECKSUM,
+ _currentIndex,
+ offset,
+ _chunkChecksum->getChecksum(_currentIndex).c_str(),
+ actualChecksum.c_str());
+ int32_t startIndex;
+ int32_t endIndex;
+ Util::indexRange(startIndex, endIndex, offset,
+ _chunkChecksum->getChecksumLength(),
+ _bitfield->getBlockLength());
+ _bitfield->unsetBitRange(startIndex, endIndex);
+ }
+ _currentIndex++;
+ }
+}
+
+string IteratableChunkChecksumValidator::calculateActualChecksum()
+{
+ int64_t offset = ((int64_t)_currentIndex)*_chunkChecksum->getChecksumLength();
+ int32_t length = _diskWriter->size() < offset+_chunkChecksum->getChecksumLength() ? _diskWriter->size()-offset : _chunkChecksum->getChecksumLength();
+ return _diskWriter->messageDigest(offset, length, _chunkChecksum->getAlgo());
+}
+
+bool IteratableChunkChecksumValidator::canValidate() const
+{
+ // We assume file is already opened using DiskWriter::open or openExistingFile.
+ return _chunkChecksum->getEstimatedDataLength() >= _diskWriter->size();
+}
+
+void IteratableChunkChecksumValidator::init()
+{
+ _bitfield->setAllBit();
+ _currentIndex = 0;
+}
diff --git a/src/IteratableChunkChecksumValidator.h b/src/IteratableChunkChecksumValidator.h
new file mode 100644
index 00000000..5291f94e
--- /dev/null
+++ b/src/IteratableChunkChecksumValidator.h
@@ -0,0 +1,91 @@
+/* <!-- 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 --> */
+#ifndef _D_ITERATABLE_CHUNK_CHECKSUM_VALIDATOR_H_
+#define _D_ITERATABLE_CHUNK_CHECKSUM_VALIDATOR_H_
+
+#include "common.h"
+#include "BitfieldMan.h"
+#include "ChunkChecksum.h"
+#include "DiskWriter.h"
+#include "LogFactory.h"
+
+class IteratableChunkChecksumValidator
+{
+private:
+ DiskWriterHandle _diskWriter;
+ BitfieldMan* _bitfield;
+ int32_t _currentIndex;
+ ChunkChecksumHandle _chunkChecksum;
+ const Logger* logger;
+
+ string calculateActualChecksum();
+public:
+ IteratableChunkChecksumValidator():_diskWriter(0), _bitfield(0), _currentIndex(0), _chunkChecksum(0), logger(LogFactory::getInstance()) {}
+
+ bool canValidate() const;
+
+ void init();
+
+ void validateChunk();
+
+ bool finished() const
+ {
+ return _currentIndex >= _chunkChecksum->countChecksum();
+ }
+
+ void setDiskWriter(const DiskWriterHandle& diskWriter)
+ {
+ _diskWriter = diskWriter;
+ }
+
+ void setBitfield(BitfieldMan* bitfield)
+ {
+ _bitfield = bitfield;
+ }
+
+ void setChunkChecksum(const ChunkChecksumHandle& chunkChecksum)
+ {
+ _chunkChecksum = chunkChecksum;
+ }
+
+ int64_t getCurrentOffset() const
+ {
+ return ((int64_t)_currentIndex)*_chunkChecksum->getChecksumLength();
+ }
+};
+
+typedef SharedHandle<IteratableChunkChecksumValidator> IteratableChunkChecksumValidatorHandle;
+
+#endif // _D_ITERATABLE_CHUNK_CHECKSUM_VALIDATOR_H_
diff --git a/src/MultiUrlRequestInfo.cc b/src/MultiUrlRequestInfo.cc
new file mode 100644
index 00000000..cf298138
--- /dev/null
+++ b/src/MultiUrlRequestInfo.cc
@@ -0,0 +1,119 @@
+/* <!-- 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 "MultiUrlRequestInfo.h"
+#include "prefs.h"
+#include "DownloadEngineFactory.h"
+#include "RecoverableException.h"
+#include "message.h"
+#include "DNSCache.h"
+#include "TorrentRequestInfo.h"
+#include "MetalinkRequestInfo.h"
+
+extern volatile sig_atomic_t haltRequested;
+
+RequestInfoHandle MultiUrlRequestInfo::createNextRequestInfo(const string& filename) const
+{
+#ifdef ENABLE_BITTORRENT
+ if(op->getAsBool(PREF_FOLLOW_TORRENT) &&
+ Util::endsWith(fileInfo.filename, ".torrent")) {
+ return new TorrentRequestInfo(filename, op);
+ } else
+#endif // ENABLE_BITTORRENT
+#ifdef ENABLE_METALINK
+ if(op->getAsBool(PREF_FOLLOW_METALINK) &&
+ Util::endsWith(fileInfo.filename, ".metalink")) {
+ return new MetalinkRequestInfo(filename, op);
+ } else
+#endif // ENABLE_METALINK
+ {
+ return 0;
+ }
+}
+
+static void handler(int signal) {
+ haltRequested = true;
+}
+
+RequestInfos MultiUrlRequestInfo::execute() {
+ {
+ DNSCacheHandle dnsCache = new SimpleDNSCache();
+ DNSCacheSingletonHolder::instance(dnsCache);
+ }
+
+ RequestInfos nextReqInfos;
+ try {
+ SharedHandle<ConsoleDownloadEngine> e(DownloadEngineFactory::newConsoleEngine(op, _requestGroups));
+
+ e->fillCommand();
+
+ // The number of simultaneous download is specified by PREF_MAX_SIMULTANEOUS_DOWNLOADS.
+ // The remaining urls are queued into FillRequestGroupCommand.
+ // It observes the number of simultaneous downloads and if it is under
+ // the limit, it adds RequestGroup object from its queue to DownloadEngine.
+ // This is done every 1 second. At the same time, it removes finished/error
+ // RequestGroup from DownloadEngine.
+
+ Util::setGlobalSignalHandler(SIGINT, handler, 0);
+ Util::setGlobalSignalHandler(SIGTERM, handler, 0);
+
+ e->run();
+
+ for(RequestGroups::iterator itr = _requestGroups.begin();
+ itr != _requestGroups.end(); ++itr) {
+ if((*itr)->downloadFinished()) {
+ RequestInfoHandle reqInfo = createNextRequestInfo((*itr)->getFilePath());
+ if(!reqInfo.isNull()) {
+ nextReqInfos.push_back(reqInfo);
+ }
+ }
+ }
+ RequestGroupMan rgman(_requestGroups);
+ // TODO print summary of the download result
+ rgman.showDownloadResults(cout);
+ cout << flush;
+ // TODO Do we have to print a message when some of the downloads are failed?
+ if(!rgman.downloadFinished()) {
+ printDownloadAbortMessage();
+ }
+ } catch(RecoverableException *ex) {
+ logger->error("Exception caught", ex);
+ delete ex;
+ fail = true;
+ }
+ Util::setGlobalSignalHandler(SIGINT, SIG_DFL, 0);
+ Util::setGlobalSignalHandler(SIGTERM, SIG_DFL, 0);
+
+ return nextReqInfos;
+}
diff --git a/src/MultiUrlRequestInfo.h b/src/MultiUrlRequestInfo.h
new file mode 100644
index 00000000..17227136
--- /dev/null
+++ b/src/MultiUrlRequestInfo.h
@@ -0,0 +1,61 @@
+/* <!-- 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 --> */
+#ifndef _D_MULTI_URL_REQUEST_INFO_H_
+#define _D_MULTI_URL_REQUEST_INFO_H_
+
+#include "RequestInfo.h"
+
+class MultiUrlRequestInfo : public RequestInfo {
+private:
+ RequestGroups _requestGroups;
+
+ RequestInfoHandle createNextRequestInfo(const string& filename) const;
+public:
+ MultiUrlRequestInfo(const RequestGroups& requestGroups, Option* op):
+ RequestInfo(op),
+ _requestGroups(requestGroups) {}
+
+ MultiUrlRequestInfo(const Strings& uris, Option* op):RequestInfo(op)
+ {
+ _requestGroups.push_back(new RequestGroup(uris, op));
+ }
+
+ virtual ~MultiUrlRequestInfo() {}
+
+ virtual RequestInfos execute();
+};
+
+typedef SharedHandle<MultiUrlRequestInfo> MultiUrlRequestInfoHandle;
+#endif // _D_MULTI_URL_REQUEST_INFO_H_
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);
+}
diff --git a/src/RequestGroup.h b/src/RequestGroup.h
new file mode 100644
index 00000000..6ae4d4fd
--- /dev/null
+++ b/src/RequestGroup.h
@@ -0,0 +1,245 @@
+/* <!-- 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 --> */
+#ifndef _D_REQUEST_GROUP_H_
+#define _D_REQUEST_GROUP_H_
+
+#include "common.h"
+#include "SegmentMan.h"
+#include "LogFactory.h"
+#include "Command.h"
+#include "ChunkChecksum.h"
+#include "Checksum.h"
+#include "SegmentManFactory.h"
+#include "DefaultSegmentManFactory.h"
+
+class DownloadEngine;
+
+class RequestGroup {
+private:
+ int64_t _hintTotalLength;
+ string _hintFilename;
+ Strings _uris;
+ Strings _spentUris;
+ SegmentManHandle _segmentMan;
+ SegmentManFactoryHandle _segmentManFactory;
+ const Option* _option;
+ const Logger* logger;
+ ChunkChecksumHandle _chunkChecksum;
+ Checksum _checksum;
+
+ void validateFilename(const string& expectedFilename,
+ const string& actualFilename) const;
+
+ void validateTotalLength(int64_t expectedTotalLength,
+ int64_t actualTotalLength) const;
+
+public:
+
+ int32_t numConnection;
+
+ bool isTorrent;
+
+ RequestGroup(const Strings& uris, const Option* option):
+ _uris(uris), _segmentMan(0), _segmentManFactory(new DefaultSegmentManFactory(option)), _option(option), logger(LogFactory::getInstance()), _chunkChecksum(0), numConnection(0), isTorrent(false) {}
+
+ RequestGroup(const string& uri, const Option* option):
+ _segmentMan(0), _segmentManFactory(new DefaultSegmentManFactory(option)), _option(option), logger(LogFactory::getInstance()), _chunkChecksum(0), numConnection(0), isTorrent(false)
+ {
+ _uris.push_back(uri);
+ }
+
+ /**
+ * Reinitializes SegmentMan based on current property values and
+ * returns new one.
+ */
+ SegmentManHandle initSegmentMan();
+
+ SegmentManHandle getSegmentMan() const
+ {
+ return _segmentMan;
+ }
+
+ Commands getNextCommand(DownloadEngine* e, int32_t maxNum, const string& method = "GET");
+
+ void addURI(const string& uri)
+ {
+ _uris.push_back(uri);
+ }
+
+ void setChunkChecksum(const ChunkChecksumHandle& chunkChecksum)
+ {
+ _chunkChecksum = chunkChecksum;
+ }
+
+ ChunkChecksumHandle getChunkChecksum() const
+ {
+ return _chunkChecksum;
+ }
+
+ void initBitfield();
+
+ void openExistingFile();
+
+ void markAllPiecesDone();
+
+ void markExistingPiecesDone();
+
+ void markPieceDone(int64_t length);
+
+ void shouldCancelDownloadForSafety();
+
+ void initAndOpenFile();
+
+ bool needsFileAllocation() const;
+
+ bool downloadFinished() const
+ {
+ if(_segmentMan.isNull()) {
+ return false;
+ } else {
+ return _segmentMan->finished();
+ }
+ }
+
+ void load()
+ {
+ _segmentMan->load();
+ }
+
+ void save()
+ {
+ _segmentMan->save();
+ }
+
+ void remove()
+ {
+ _segmentMan->remove();
+ }
+
+ void closeFile()
+ {
+ _segmentMan->diskWriter->closeFile();
+ }
+
+ bool fileExists() const;
+
+ bool segmentFileExists() const;
+
+ string getFilePath() const;
+
+ int64_t getTotalLength() const
+ {
+ return _segmentMan->totalSize;
+ }
+
+ int64_t getDownloadLength() const
+ {
+ return _segmentMan->getDownloadLength();
+ }
+
+ void loadAndOpenFile();
+
+ void prepareForNextAction(int cuid, const RequestHandle& req, DownloadEngine* e);
+
+ bool downloadFinishedByFileLength();
+
+ void setChecksum(const Checksum& checksum)
+ {
+ _checksum = checksum;
+ }
+
+ const string& getHintFilename() const
+ {
+ return _hintFilename;
+ }
+
+ void setHintFilename(const string& filename)
+ {
+ _hintFilename = filename;
+ }
+
+ int64_t getHintTotalLength() const
+ {
+ return _hintTotalLength;
+ }
+
+ void setHintTotalLength(int64_t totalLength)
+ {
+ _hintTotalLength = totalLength;
+ }
+
+ const Strings& getRemainingUris() const
+ {
+ return _uris;
+ }
+
+ const Strings& getSpentUris() const
+ {
+ return _spentUris;
+ }
+
+ Strings getUris() const
+ {
+ Strings temp(_spentUris.begin(), _spentUris.end());
+ temp.insert(temp.end(), _uris.begin(), _uris.end());
+ return temp;
+ }
+
+ /**
+ * Compares expected filename with specified actualFilename.
+ * The expected filename refers to SegmentMan::filename.
+ */
+ void validateFilename(const string& actualFilename) const;
+
+ void validateTotalLength(int64_t actualTotalLength) const;
+
+ /**
+ * Compares expected filename with specified actualFilename.
+ * The expected filename refers to RequestGroup::hintFilename.
+ */
+ void validateFilenameByHint(const string& actualFilename) const;
+
+ void validateTotalLengthByHint(int64_t actualTotalLength) const;
+
+ void setSegmentManFactory(const SegmentManFactoryHandle& segmentManFactory)
+ {
+ _segmentManFactory = segmentManFactory;
+ }
+};
+
+typedef SharedHandle<RequestGroup> RequestGroupHandle;
+typedef deque<RequestGroupHandle> RequestGroups;
+
+#endif // _D_REQUEST_GROUP_H_
diff --git a/src/RequestGroupMan.cc b/src/RequestGroupMan.cc
new file mode 100644
index 00000000..3b48f82a
--- /dev/null
+++ b/src/RequestGroupMan.cc
@@ -0,0 +1,120 @@
+/* <!-- 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 "RequestGroupMan.h"
+#include "DownloadEngine.h"
+#include <iomanip>
+
+void RequestGroupMan::removeStoppedGroup()
+{
+ int32_t count = 0;
+ RequestGroups temp;
+ for(RequestGroups::iterator itr = _requestGroups.begin();
+ itr != _requestGroups.end(); ++itr) {
+ if((*itr)->numConnection > 0) {
+ temp.push_back(*itr);
+ } else {
+ (*itr)->closeFile();
+ if((*itr)->downloadFinished()) {
+ _logger->notice("Download complete: %s", (*itr)->getFilePath().c_str());
+ (*itr)->remove();
+ } else {
+ (*itr)->save();
+ }
+ ++count;
+ }
+ }
+ _requestGroups = temp;
+ if(count > 0) {
+ _logger->debug("%d RequestGroup(s) deleted.", count);
+ }
+}
+
+void RequestGroupMan::fillRequestGroupFromReserver(DownloadEngine* e)
+{
+ removeStoppedGroup();
+
+ int32_t count = 0;
+ for(int32_t num = _maxSimultaneousDownloads-_requestGroups.size();
+ num > 0 && _reservedGroups.size() > 0; --num) {
+
+ RequestGroupHandle groupToAdd = _reservedGroups.front();
+ _reservedGroups.pop_front();
+
+ _requestGroups.push_back(groupToAdd);
+ groupToAdd->initSegmentMan();
+ Commands commands = groupToAdd->getNextCommand(e, 1);
+ count += commands.size();
+ e->addCommand(commands);
+ }
+ if(count > 0) {
+ _logger->debug("%d RequestGroup(s) added.", count);
+ }
+}
+
+void RequestGroupMan::showDownloadResults(ostream& o) const
+{
+ // Download Results:
+ // idx|stat|path/length
+ // ===+====+=======================================================================
+ o << "\n"
+ <<_("Download Results:") << "\n"
+ << "idx|stat|path/URI" << "\n"
+ << "===+====+======================================================================" << "\n";
+ int32_t count = 0;
+ for(RequestGroups::const_iterator itr = _requestGroups.begin();
+ itr != _requestGroups.end(); ++itr) {
+ o << setw(3) << ++count << "|";
+ if((*itr)->downloadFinished()) {
+ o << "OK ";
+ } else {
+ o << "ERR ";
+ }
+ o << "|";
+ if((*itr)->downloadFinished()) {
+ o << (*itr)->getFilePath();
+ } else {
+ Strings uris = (*itr)->getUris();
+ if(uris.empty()) {
+ o << "n/a";
+ } else {
+ o << uris.front();
+ if(uris.size() > 1) {
+ o << " (" << uris.size()-1 << "more)";
+ }
+ }
+ }
+ o << "\n";
+ }
+}
diff --git a/src/RequestGroupMan.h b/src/RequestGroupMan.h
new file mode 100644
index 00000000..0df9bb49
--- /dev/null
+++ b/src/RequestGroupMan.h
@@ -0,0 +1,164 @@
+/* <!-- 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 --> */
+#ifndef _D_REQUEST_GROUP_MAN_H_
+#define _D_REQUEST_GROUP_MAN_H_
+
+#include "common.h"
+#include "RequestGroup.h"
+#include "LogFactory.h"
+
+class DownloadEngine;
+
+class RequestGroupMan {
+private:
+ RequestGroups _requestGroups;
+ RequestGroups _reservedGroups;
+ const Logger* _logger;
+ int32_t _maxSimultaneousDownloads;
+public:
+ RequestGroupMan(const RequestGroups& requestGroups = RequestGroups(), int32_t maxSimultaneousDownloads = 1):
+ _requestGroups(requestGroups),
+ _logger(LogFactory::getInstance()),
+ _maxSimultaneousDownloads(maxSimultaneousDownloads) {}
+
+ bool downloadFinished()
+ {
+ for(RequestGroups::iterator itr = _requestGroups.begin();
+ itr != _requestGroups.end(); ++itr) {
+ if((*itr)->numConnection > 0 || !(*itr)->getSegmentMan()->finished()) {
+ return false;
+ }
+ }
+ return true;
+ }
+
+ void save()
+ {
+ for(RequestGroups::iterator itr = _requestGroups.begin();
+ itr != _requestGroups.end(); ++itr) {
+ if(!(*itr)->getSegmentMan()->finished()) {
+ (*itr)->getSegmentMan()->save();
+ }
+ }
+ }
+
+ void closeFile()
+ {
+ for(RequestGroups::iterator itr = _requestGroups.begin();
+ itr != _requestGroups.end(); ++itr) {
+ (*itr)->getSegmentMan()->diskWriter->closeFile();
+ }
+ }
+
+ int64_t getDownloadLength() const
+ {
+ int64_t downloadLength = 0;
+ for(RequestGroups::const_iterator itr = _requestGroups.begin();
+ itr != _requestGroups.end(); ++itr) {
+ downloadLength += (*itr)->getSegmentMan()->getDownloadLength();
+ }
+ return downloadLength;
+ }
+
+ int64_t getTotalLength() const
+ {
+ int64_t totalLength = 0;
+ for(RequestGroups::const_iterator itr = _requestGroups.begin();
+ itr != _requestGroups.end(); ++itr) {
+ totalLength += (*itr)->getSegmentMan()->totalSize;
+ }
+ return totalLength;
+ }
+
+ Commands getInitialCommands(DownloadEngine* e) const
+ {
+ Commands commands;
+ for(RequestGroups::const_iterator itr = _requestGroups.begin();
+ itr != _requestGroups.end(); ++itr) {
+ (*itr)->initSegmentMan();
+ commands.push_back((*itr)->getNextCommand(e, 1).front());
+ }
+ return commands;
+ }
+
+ void removeStoppedGroup();
+
+ void fillRequestGroupFromReserver(DownloadEngine* e);
+
+ void addRequestGroup(const RequestGroupHandle& group)
+ {
+ _requestGroups.push_back(group);
+ }
+
+ void addReservedGroup(const RequestGroups& groups)
+ {
+ _reservedGroups.insert(_reservedGroups.end(), groups.begin(), groups.end());
+ }
+
+ void addReservedGroup(const RequestGroupHandle& group)
+ {
+ _reservedGroups.push_back(group);
+ }
+
+ int32_t countRequestGroup() const
+ {
+ return _requestGroups.size();
+ }
+
+ RequestGroupHandle getRequestGroup(int32_t index) const
+ {
+ if(index < (int32_t)_requestGroups.size()) {
+ return _requestGroups[index];
+ } else {
+ return 0;
+ }
+ }
+
+ void showDownloadResults(ostream& o) const;
+
+ int32_t getErrors() const
+ {
+ int32_t errors = 0;
+ for(RequestGroups::const_iterator itr = _requestGroups.begin();
+ itr != _requestGroups.end(); ++itr) {
+ errors += (*itr)->getSegmentMan()->errors;
+ }
+ return errors;
+ }
+};
+
+typedef SharedHandle<RequestGroupMan> RequestGroupManHandle;
+
+#endif // _D_REQUEST_GROUP_MAN_H_
diff --git a/src/SegmentManFactory.h b/src/SegmentManFactory.h
new file mode 100644
index 00000000..4e601926
--- /dev/null
+++ b/src/SegmentManFactory.h
@@ -0,0 +1,50 @@
+/* <!-- 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 --> */
+#ifndef _D_SEGMENT_MAN_FACTORY_H_
+#define _D_SEGMENT_MAN_FACTORY_H_
+
+#include "common.h"
+#include "SegmentMan.h"
+
+class SegmentManFactory {
+public:
+ virtual ~SegmentManFactory() {}
+
+ virtual SegmentManHandle createNewInstance() = 0;
+};
+
+typedef SharedHandle<SegmentManFactory> SegmentManFactoryHandle;
+
+#endif // _D_SEGMENT_MAN_FACTORY_H_
diff --git a/src/TrackerSegmentManFactory.cc b/src/TrackerSegmentManFactory.cc
new file mode 100644
index 00000000..63dafa31
--- /dev/null
+++ b/src/TrackerSegmentManFactory.cc
@@ -0,0 +1,45 @@
+/* <!-- 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 "TrackerSegmentManFactory.h"
+#include "ByteArrayDiskWriter.h"
+
+SegmentManHandle TrackerSegmentManFactory::createNewInstance()
+{
+ SegmentManHandle segmentMan = new SegmentMan();
+ segmentMan->diskWriter = new ByteArrayDiskWriter();
+ segmentMan->option = _option;
+ segmentMan->ufilename = "[tracker.announce]";
+ return segmentMan;
+}
diff --git a/src/TrackerSegmentManFactory.h b/src/TrackerSegmentManFactory.h
new file mode 100644
index 00000000..dcec0096
--- /dev/null
+++ b/src/TrackerSegmentManFactory.h
@@ -0,0 +1,51 @@
+/* <!-- 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 --> */
+#ifndef _D_TRACKER_SEGMENT_MAN_FACTORY_H_
+#define _D_TRACKER_SEGMENT_MAN_FACTORY_H_
+
+#include "AbstractSegmentManFactory.h"
+
+class TrackerSegmentManFactory : public AbstractSegmentManFactory {
+public:
+ TrackerSegmentManFactory(const Option* option):AbstractSegmentManFactory(option) {}
+
+ virtual ~TrackerSegmentManFactory() {}
+
+ virtual SegmentManHandle createNewInstance();
+};
+
+typedef SharedHandle<TrackerSegmentManFactory> TrackerSegmentManFactoryHandle;
+
+#endif // _D_TRACKER_SEGMENT_MAN_FACTORY_H_
diff --git a/src/UriFileListParser.cc b/src/UriFileListParser.cc
new file mode 100644
index 00000000..c951cc0a
--- /dev/null
+++ b/src/UriFileListParser.cc
@@ -0,0 +1,54 @@
+/* <!-- 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 "UriFileListParser.h"
+#include "Util.h"
+
+bool UriFileListParser::hasNext() const
+{
+ return _ifs;
+}
+
+Strings UriFileListParser::next()
+{
+ string line;
+ while(getline(_ifs, line)) {
+ if(Util::trim(line) != "") {
+ Strings uris;
+ Util::slice(uris, line, '\t', true);
+ return uris;
+ }
+ }
+ return Strings();
+}
diff --git a/src/UriFileListParser.h b/src/UriFileListParser.h
new file mode 100644
index 00000000..a25a2ae0
--- /dev/null
+++ b/src/UriFileListParser.h
@@ -0,0 +1,53 @@
+/* <!-- 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 --> */
+#ifndef _D_URI_FILE_LIST_PARSER_H_
+#define _D_URI_FILE_LIST_PARSER_H_
+
+#include "common.h"
+#include <fstream>
+
+class UriFileListParser {
+private:
+ string _filename;
+ ifstream _ifs;
+public:
+ UriFileListParser(const string& filename):_filename(filename), _ifs(filename.c_str()) {}
+
+ bool hasNext() const;
+
+ Strings next();
+};
+
+#endif // _D_URI_FILE_LIST_PARSER_H_