mirror of
https://github.com/qbittorrent/qBittorrent.git
synced 2025-01-10 00:57:24 +03:00
324 lines
13 KiB
C++
324 lines
13 KiB
C++
/*
|
|
* Bittorrent Client using Qt and libtorrent.
|
|
* Copyright (C) 2015 Vladimir Golovnev <glassez@yandex.ru>
|
|
* Copyright (C) 2006 Christophe Dumez <chris@qbittorrent.org>
|
|
*
|
|
* 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 <functional>
|
|
|
|
#include <libtorrent/add_torrent_params.hpp>
|
|
#include <libtorrent/fwd.hpp>
|
|
#include <libtorrent/socket.hpp>
|
|
#include <libtorrent/torrent_handle.hpp>
|
|
#include <libtorrent/torrent_status.hpp>
|
|
|
|
#include <QDateTime>
|
|
#include <QHash>
|
|
#include <QMap>
|
|
#include <QObject>
|
|
#include <QQueue>
|
|
#include <QString>
|
|
#include <QVector>
|
|
|
|
#include "base/tagset.h"
|
|
#include "infohash.h"
|
|
#include "speedmonitor.h"
|
|
#include "torrent.h"
|
|
#include "torrentinfo.h"
|
|
|
|
namespace BitTorrent
|
|
{
|
|
class Session;
|
|
struct LoadTorrentParams;
|
|
|
|
enum class MoveStorageMode
|
|
{
|
|
KeepExistingFiles,
|
|
Overwrite
|
|
};
|
|
|
|
enum class MaintenanceJob
|
|
{
|
|
None,
|
|
HandleMetadata
|
|
};
|
|
|
|
struct FileErrorInfo
|
|
{
|
|
lt::error_code error;
|
|
lt::operation_t operation;
|
|
};
|
|
|
|
class TorrentImpl final : public QObject, public Torrent
|
|
{
|
|
Q_DISABLE_COPY_MOVE(TorrentImpl)
|
|
Q_DECLARE_TR_FUNCTIONS(BitTorrent::TorrentImpl)
|
|
|
|
public:
|
|
TorrentImpl(Session *session, lt::session *nativeSession
|
|
, const lt::torrent_handle &nativeHandle, const LoadTorrentParams ¶ms);
|
|
~TorrentImpl() override;
|
|
|
|
bool isValid() const;
|
|
|
|
InfoHash infoHash() const override;
|
|
QString name() const override;
|
|
QDateTime creationDate() const override;
|
|
QString creator() const override;
|
|
QString comment() const override;
|
|
bool isPrivate() const override;
|
|
qlonglong totalSize() const override;
|
|
qlonglong wantedSize() const override;
|
|
qlonglong completedSize() const override;
|
|
qlonglong pieceLength() const override;
|
|
qlonglong wastedSize() const override;
|
|
QString currentTracker() const override;
|
|
|
|
bool isAutoTMMEnabled() const override;
|
|
void setAutoTMMEnabled(bool enabled) override;
|
|
QString savePath() const override;
|
|
void setSavePath(const QString &path) override;
|
|
QString downloadPath() const override;
|
|
void setDownloadPath(const QString &path) override;
|
|
QString actualStorageLocation() const override;
|
|
QString rootPath() const override;
|
|
QString contentPath() const override;
|
|
QString category() const override;
|
|
bool belongsToCategory(const QString &category) const override;
|
|
bool setCategory(const QString &category) override;
|
|
|
|
TagSet tags() const override;
|
|
bool hasTag(const QString &tag) const override;
|
|
bool addTag(const QString &tag) override;
|
|
bool removeTag(const QString &tag) override;
|
|
void removeAllTags() override;
|
|
|
|
int filesCount() const override;
|
|
int piecesCount() const override;
|
|
int piecesHave() const override;
|
|
qreal progress() const override;
|
|
QDateTime addedTime() const override;
|
|
qreal ratioLimit() const override;
|
|
int seedingTimeLimit() const override;
|
|
|
|
QString filePath(int index) const override;
|
|
QString actualFilePath(int index) const override;
|
|
qlonglong fileSize(int index) const override;
|
|
QStringList filePaths() const override;
|
|
QVector<DownloadPriority> filePriorities() const override;
|
|
|
|
TorrentInfo info() const override;
|
|
bool isSeed() const override;
|
|
bool isPaused() const override;
|
|
bool isQueued() const override;
|
|
bool isForced() const override;
|
|
bool isChecking() const override;
|
|
bool isDownloading() const override;
|
|
bool isUploading() const override;
|
|
bool isCompleted() const override;
|
|
bool isActive() const override;
|
|
bool isInactive() const override;
|
|
bool isErrored() const override;
|
|
bool isSequentialDownload() const override;
|
|
bool hasFirstLastPiecePriority() const override;
|
|
TorrentState state() const override;
|
|
bool hasMetadata() const override;
|
|
bool hasMissingFiles() const override;
|
|
bool hasError() const override;
|
|
int queuePosition() const override;
|
|
QVector<TrackerEntry> trackers() const override;
|
|
QVector<QUrl> urlSeeds() const override;
|
|
QString error() const override;
|
|
qlonglong totalDownload() const override;
|
|
qlonglong totalUpload() const override;
|
|
qlonglong activeTime() const override;
|
|
qlonglong finishedTime() const override;
|
|
qlonglong eta() const override;
|
|
QVector<qreal> filesProgress() const override;
|
|
int seedsCount() const override;
|
|
int peersCount() const override;
|
|
int leechsCount() const override;
|
|
int totalSeedsCount() const override;
|
|
int totalPeersCount() const override;
|
|
int totalLeechersCount() const override;
|
|
int completeCount() const override;
|
|
int incompleteCount() const override;
|
|
QDateTime lastSeenComplete() const override;
|
|
QDateTime completedTime() const override;
|
|
qlonglong timeSinceUpload() const override;
|
|
qlonglong timeSinceDownload() const override;
|
|
qlonglong timeSinceActivity() const override;
|
|
int downloadLimit() const override;
|
|
int uploadLimit() const override;
|
|
bool superSeeding() const override;
|
|
bool isDHTDisabled() const override;
|
|
bool isPEXDisabled() const override;
|
|
bool isLSDDisabled() const override;
|
|
QVector<PeerInfo> peers() const override;
|
|
QBitArray pieces() const override;
|
|
QBitArray downloadingPieces() const override;
|
|
QVector<int> pieceAvailability() const override;
|
|
qreal distributedCopies() const override;
|
|
qreal maxRatio() const override;
|
|
int maxSeedingTime() const override;
|
|
qreal realRatio() const override;
|
|
int uploadPayloadRate() const override;
|
|
int downloadPayloadRate() const override;
|
|
qlonglong totalPayloadUpload() const override;
|
|
qlonglong totalPayloadDownload() const override;
|
|
int connectionsCount() const override;
|
|
int connectionsLimit() const override;
|
|
qlonglong nextAnnounce() const override;
|
|
QVector<qreal> availableFileFractions() const override;
|
|
|
|
void setName(const QString &name) override;
|
|
void setSequentialDownload(bool enable) override;
|
|
void setFirstLastPiecePriority(bool enabled) override;
|
|
void pause() override;
|
|
void resume(TorrentOperatingMode mode = TorrentOperatingMode::AutoManaged) override;
|
|
void forceReannounce(int index = -1) override;
|
|
void forceDHTAnnounce() override;
|
|
void forceRecheck() override;
|
|
void renameFile(int index, const QString &path) override;
|
|
void prioritizeFiles(const QVector<DownloadPriority> &priorities) override;
|
|
void setRatioLimit(qreal limit) override;
|
|
void setSeedingTimeLimit(int limit) override;
|
|
void setUploadLimit(int limit) override;
|
|
void setDownloadLimit(int limit) override;
|
|
void setSuperSeeding(bool enable) override;
|
|
void setDHTDisabled(bool disable) override;
|
|
void setPEXDisabled(bool disable) override;
|
|
void setLSDDisabled(bool disable) override;
|
|
void flushCache() const override;
|
|
void addTrackers(const QVector<TrackerEntry> &trackers) override;
|
|
void replaceTrackers(const QVector<TrackerEntry> &trackers) override;
|
|
void addUrlSeeds(const QVector<QUrl> &urlSeeds) override;
|
|
void removeUrlSeeds(const QVector<QUrl> &urlSeeds) override;
|
|
bool connectPeer(const PeerAddress &peerAddress) override;
|
|
void clearPeers() override;
|
|
|
|
QString createMagnetURI() const override;
|
|
|
|
bool needSaveResumeData() const;
|
|
|
|
// Session interface
|
|
lt::torrent_handle nativeHandle() const;
|
|
|
|
void handleAlert(const lt::alert *a);
|
|
void handleStateUpdate(const lt::torrent_status &nativeStatus);
|
|
void handleDownloadPathChanged();
|
|
void handleCategoryOptionsChanged();
|
|
void handleAppendExtensionToggled();
|
|
void saveResumeData();
|
|
void handleMoveStorageJobFinished(bool hasOutstandingJob);
|
|
void fileSearchFinished(const QString &savePath, const QStringList &fileNames);
|
|
|
|
private:
|
|
using EventTrigger = std::function<void ()>;
|
|
|
|
void updateStatus();
|
|
void updateStatus(const lt::torrent_status &nativeStatus);
|
|
void updateState();
|
|
|
|
void handleFastResumeRejectedAlert(const lt::fastresume_rejected_alert *p);
|
|
void handleFileCompletedAlert(const lt::file_completed_alert *p);
|
|
void handleFileErrorAlert(const lt::file_error_alert *p);
|
|
#ifdef QBT_USES_LIBTORRENT2
|
|
void handleFilePrioAlert(const lt::file_prio_alert *p);
|
|
#endif
|
|
void handleFileRenamedAlert(const lt::file_renamed_alert *p);
|
|
void handleFileRenameFailedAlert(const lt::file_rename_failed_alert *p);
|
|
void handleMetadataReceivedAlert(const lt::metadata_received_alert *p);
|
|
void handlePerformanceAlert(const lt::performance_alert *p) const;
|
|
void handleSaveResumeDataAlert(const lt::save_resume_data_alert *p);
|
|
void handleSaveResumeDataFailedAlert(const lt::save_resume_data_failed_alert *p);
|
|
void handleTorrentCheckedAlert(const lt::torrent_checked_alert *p);
|
|
void handleTorrentFinishedAlert(const lt::torrent_finished_alert *p);
|
|
void handleTorrentPausedAlert(const lt::torrent_paused_alert *p);
|
|
void handleTorrentResumedAlert(const lt::torrent_resumed_alert *p);
|
|
void handleTrackerErrorAlert(const lt::tracker_error_alert *p);
|
|
void handleTrackerReplyAlert(const lt::tracker_reply_alert *p);
|
|
void handleTrackerWarningAlert(const lt::tracker_warning_alert *p);
|
|
|
|
bool isMoveInProgress() const;
|
|
|
|
void setAutoManaged(bool enable);
|
|
|
|
void adjustStorageLocation();
|
|
void moveStorage(const QString &newPath, MoveStorageMode mode);
|
|
void manageIncompleteFiles();
|
|
void applyFirstLastPiecePriority(bool enabled, const QVector<DownloadPriority> &updatedFilePrio = {});
|
|
|
|
void prepareResumeData(const lt::add_torrent_params ¶ms);
|
|
void endReceivedMetadataHandling(const QString &savePath, const QStringList &fileNames);
|
|
void reload();
|
|
|
|
Session *const m_session;
|
|
lt::session *m_nativeSession;
|
|
lt::torrent_handle m_nativeHandle;
|
|
lt::torrent_status m_nativeStatus;
|
|
TorrentState m_state = TorrentState::Unknown;
|
|
TorrentInfo m_torrentInfo;
|
|
QStringList m_filePaths;
|
|
SpeedMonitor m_speedMonitor;
|
|
|
|
InfoHash m_infoHash;
|
|
|
|
// m_moveFinishedTriggers is activated only when the following conditions are met:
|
|
// all file rename jobs complete, all file move jobs complete
|
|
QQueue<EventTrigger> m_moveFinishedTriggers;
|
|
int m_renameCount = 0;
|
|
bool m_storageIsMoving = false;
|
|
|
|
MaintenanceJob m_maintenanceJob = MaintenanceJob::None;
|
|
|
|
QHash<QString, QMap<lt::tcp::endpoint, int>> m_trackerPeerCounts;
|
|
FileErrorInfo m_lastFileError;
|
|
|
|
// Persistent data
|
|
QString m_name;
|
|
QString m_savePath;
|
|
QString m_downloadPath;
|
|
QString m_category;
|
|
TagSet m_tags;
|
|
qreal m_ratioLimit;
|
|
int m_seedingTimeLimit;
|
|
TorrentOperatingMode m_operatingMode;
|
|
TorrentContentLayout m_contentLayout;
|
|
bool m_hasSeedStatus;
|
|
bool m_fastresumeDataRejected = false;
|
|
bool m_hasMissingFiles = false;
|
|
bool m_hasFirstLastPiecePriority = false;
|
|
bool m_useAutoTMM;
|
|
bool m_isStopped;
|
|
|
|
bool m_unchecked = false;
|
|
|
|
lt::add_torrent_params m_ltAddTorrentParams;
|
|
};
|
|
}
|