diff --git a/src/base/CMakeLists.txt b/src/base/CMakeLists.txt index ee304d4de..2bf4aa368 100644 --- a/src/base/CMakeLists.txt +++ b/src/base/CMakeLists.txt @@ -7,6 +7,7 @@ add_library(qbt_base STATIC bittorrent/addtorrentparams.h bittorrent/bandwidthscheduler.h bittorrent/cachestatus.h + bittorrent/customstorage.h bittorrent/downloadpriority.h bittorrent/filterparserthread.h bittorrent/infohash.h @@ -87,6 +88,7 @@ add_library(qbt_base STATIC # sources asyncfilestorage.cpp bittorrent/bandwidthscheduler.cpp + bittorrent/customstorage.cpp bittorrent/downloadpriority.cpp bittorrent/filterparserthread.cpp bittorrent/infohash.cpp diff --git a/src/base/base.pri b/src/base/base.pri index 65667a70a..90b07a537 100644 --- a/src/base/base.pri +++ b/src/base/base.pri @@ -4,6 +4,7 @@ HEADERS += \ $$PWD/bittorrent/addtorrentparams.h \ $$PWD/bittorrent/bandwidthscheduler.h \ $$PWD/bittorrent/cachestatus.h \ + $$PWD/bittorrent/customstorage.h \ $$PWD/bittorrent/downloadpriority.h \ $$PWD/bittorrent/filterparserthread.h \ $$PWD/bittorrent/infohash.h \ @@ -85,6 +86,7 @@ HEADERS += \ SOURCES += \ $$PWD/asyncfilestorage.cpp \ $$PWD/bittorrent/bandwidthscheduler.cpp \ + $$PWD/bittorrent/customstorage.cpp \ $$PWD/bittorrent/downloadpriority.cpp \ $$PWD/bittorrent/filterparserthread.cpp \ $$PWD/bittorrent/infohash.cpp \ diff --git a/src/base/bittorrent/customstorage.cpp b/src/base/bittorrent/customstorage.cpp new file mode 100644 index 000000000..60766e629 --- /dev/null +++ b/src/base/bittorrent/customstorage.cpp @@ -0,0 +1,94 @@ +/* + * Bittorrent Client using Qt and libtorrent. + * Copyright (C) 2020 Vladimir Golovnev + * + * 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 this program with the OpenSSL project's "OpenSSL" library (or with + * modified versions of it that use the same license as the "OpenSSL" library), + * and distribute the linked executables. 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), 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. + */ + +#include "customstorage.h" + +#if (LIBTORRENT_VERSION_NUM >= 10200) +#include + +#include + +#include "base/utils/fs.h" + +extern const QString QB_EXT {QStringLiteral(".!qB")}; + +lt::storage_interface *customStorageConstructor(const lt::storage_params ¶ms, lt::file_pool &pool) +{ + return new CustomStorage {params, pool}; +} + +CustomStorage::CustomStorage(const lt::storage_params ¶ms, lt::file_pool &filePool) + : lt::default_storage {params, filePool} +{ + m_savePath = Utils::Fs::expandPathAbs(QString::fromStdString(params.path)); +} + +bool CustomStorage::verify_resume_data(const lt::add_torrent_params &rd, const lt::aux::vector &links, lt::storage_error &ec) +{ + const QDir saveDir {m_savePath}; + + const lt::file_storage &fileStorage = files(); + for (const lt::file_index_t fileIndex : fileStorage.file_range()) { + // ignore files that have priority 0 + if ((m_filePriorities.end_index() > fileIndex) && (m_filePriorities[fileIndex] == lt::dont_download)) + continue; + + // ignore pad files + if (fileStorage.pad_file_at(fileIndex)) continue; + + const QString filePath = QString::fromStdString(fileStorage.file_path(fileIndex)); + if (filePath.endsWith(QB_EXT)) { + const QString completeFilePath = filePath.left(filePath.size() - QB_EXT.size()); + QFile completeFile {saveDir.absoluteFilePath(completeFilePath)}; + if (completeFile.exists()) { + QFile currentFile {saveDir.absoluteFilePath(filePath)}; + if (currentFile.exists()) + currentFile.remove(); + completeFile.rename(currentFile.fileName()); + } + } + } + + return lt::default_storage::verify_resume_data(rd, links, ec); +} + +void CustomStorage::set_file_priority(lt::aux::vector &priorities, lt::storage_error &ec) +{ + m_filePriorities = priorities; + lt::default_storage::set_file_priority(priorities, ec); +} + +lt::status_t CustomStorage::move_storage(const std::string &savePath, lt::move_flags_t flags, lt::storage_error &ec) +{ + const lt::status_t ret = lt::default_storage::move_storage(savePath, flags, ec); + if (ret != lt::status_t::fatal_disk_error) + m_savePath = Utils::Fs::expandPathAbs(QString::fromStdString(savePath)); + + return ret; +} +#endif diff --git a/src/base/bittorrent/customstorage.h b/src/base/bittorrent/customstorage.h new file mode 100644 index 000000000..fa8767348 --- /dev/null +++ b/src/base/bittorrent/customstorage.h @@ -0,0 +1,55 @@ +/* + * Bittorrent Client using Qt and libtorrent. + * Copyright (C) 2020 Vladimir Golovnev + * + * 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 this program with the OpenSSL project's "OpenSSL" library (or with + * modified versions of it that use the same license as the "OpenSSL" library), + * and distribute the linked executables. 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), 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. + */ + +#pragma once + +#include + +#if (LIBTORRENT_VERSION_NUM >= 10200) +#include +#include +#include + +#include + +lt::storage_interface *customStorageConstructor(const lt::storage_params ¶ms, lt::file_pool &pool); + +class CustomStorage final : public lt::default_storage +{ +public: + explicit CustomStorage(const lt::storage_params ¶ms, lt::file_pool &filePool); + + bool verify_resume_data(const lt::add_torrent_params &rd, const lt::aux::vector &links, lt::storage_error &ec) override; + void set_file_priority(lt::aux::vector &priorities, lt::storage_error &ec) override; + lt::status_t move_storage(const std::string &savePath, lt::move_flags_t flags, lt::storage_error &ec) override; + +private: + lt::aux::vector m_filePriorities; + QString m_savePath; +}; +#endif diff --git a/src/base/bittorrent/session.cpp b/src/base/bittorrent/session.cpp index d18bf7c51..8db43787c 100644 --- a/src/base/bittorrent/session.cpp +++ b/src/base/bittorrent/session.cpp @@ -94,6 +94,7 @@ #include "base/utils/net.h" #include "base/utils/random.h" #include "bandwidthscheduler.h" +#include "customstorage.h" #include "filterparserthread.h" #include "ltunderlyingtype.h" #include "magneturi.h" @@ -2456,6 +2457,8 @@ bool Session::addTorrent_impl(CreateTorrentParams params, const MagnetUri &magne p.max_connections = maxConnectionsPerTorrent(); p.max_uploads = maxUploadsPerTorrent(); + p.storage = customStorageConstructor; + m_addingTorrents.insert(hash, params); // Adding torrent to BitTorrent session m_nativeSession->async_add_torrent(p); @@ -2528,21 +2531,23 @@ bool Session::loadMetadata(const MagnetUri &magnetUri) const QString savePath = Utils::Fs::tempPath() + static_cast(hash); p.save_path = Utils::Fs::toNativePath(savePath).toStdString(); - // Forced start #if (LIBTORRENT_VERSION_NUM < 10200) + // Forced start p.flags &= ~lt::add_torrent_params::flag_paused; p.flags &= ~lt::add_torrent_params::flag_auto_managed; -#else - p.flags &= ~lt::torrent_flags::paused; - p.flags &= ~lt::torrent_flags::auto_managed; -#endif // Solution to avoid accidental file writes -#if (LIBTORRENT_VERSION_NUM < 10200) p.flags |= lt::add_torrent_params::flag_upload_mode; #else + // Forced start + p.flags &= ~lt::torrent_flags::paused; + p.flags &= ~lt::torrent_flags::auto_managed; + + // Solution to avoid accidental file writes p.flags |= lt::torrent_flags::upload_mode; -#endif + + p.storage = customStorageConstructor; +#endif // Adding torrent to BitTorrent session lt::error_code ec;