diff --git a/src/base/bittorrent/torrent.h b/src/base/bittorrent/torrent.h index 41b070f4d..e1416c063 100644 --- a/src/base/bittorrent/torrent.h +++ b/src/base/bittorrent/torrent.h @@ -84,11 +84,6 @@ namespace BitTorrent Error }; - struct TrackerInfo - { - int numPeers = 0; - }; - uint qHash(TorrentState key, uint seed); class Torrent : public AbstractFileStorage @@ -210,7 +205,6 @@ namespace BitTorrent virtual bool hasFilteredPieces() const = 0; virtual int queuePosition() const = 0; virtual QVector trackers() const = 0; - virtual QHash trackerInfos() const = 0; virtual QVector urlSeeds() const = 0; virtual QString error() const = 0; virtual qlonglong totalDownload() const = 0; diff --git a/src/base/bittorrent/torrentimpl.cpp b/src/base/bittorrent/torrentimpl.cpp index 804aa23ed..2bc160745 100644 --- a/src/base/bittorrent/torrentimpl.cpp +++ b/src/base/bittorrent/torrentimpl.cpp @@ -97,9 +97,11 @@ namespace } #if (LIBTORRENT_VERSION_NUM >= 20000) - TrackerEntry fromNativeAnnouncerEntry(const lt::announce_entry &nativeEntry, const lt::info_hash_t &hashes) + TrackerEntry fromNativeAnnouncerEntry(const lt::announce_entry &nativeEntry + , const lt::info_hash_t &hashes, const QMap &trackerPeerCounts) #else - TrackerEntry fromNativeAnnouncerEntry(const lt::announce_entry &nativeEntry) + TrackerEntry fromNativeAnnouncerEntry(const lt::announce_entry &nativeEntry + , const QMap &trackerPeerCounts) #endif { TrackerEntry trackerEntry {QString::fromStdString(nativeEntry.url), nativeEntry.tier}; @@ -122,6 +124,7 @@ namespace TrackerEntry::EndpointStats trackerEndpoint; trackerEndpoint.protocolVersion = (protocolVersion == lt::protocol_version::V1) ? 1 : 2; + trackerEndpoint.numPeers = trackerPeerCounts.value(endpoint.local_endpoint, -1); trackerEndpoint.numSeeds = infoHash.scrape_complete; trackerEndpoint.numLeeches = infoHash.scrape_incomplete; trackerEndpoint.numDownloaded = infoHash.scrape_downloaded; @@ -151,6 +154,7 @@ namespace trackerEndpoint.message = (!trackerMessage.isEmpty() ? trackerMessage : errorMessage); trackerEntry.endpoints.append(trackerEndpoint); + trackerEntry.numPeers = std::max(trackerEntry.numPeers, trackerEndpoint.numPeers); trackerEntry.numSeeds = std::max(trackerEntry.numSeeds, trackerEndpoint.numSeeds); trackerEntry.numLeeches = std::max(trackerEntry.numLeeches, trackerEndpoint.numLeeches); trackerEntry.numDownloaded = std::max(trackerEntry.numDownloaded, trackerEndpoint.numDownloaded); @@ -168,6 +172,7 @@ namespace for (const lt::announce_endpoint &endpoint : nativeEntry.endpoints) { TrackerEntry::EndpointStats trackerEndpoint; + trackerEndpoint.numPeers = trackerPeerCounts.value(endpoint.local_endpoint, -1); trackerEndpoint.numSeeds = endpoint.scrape_complete; trackerEndpoint.numLeeches = endpoint.scrape_incomplete; trackerEndpoint.numDownloaded = endpoint.scrape_downloaded; @@ -197,6 +202,7 @@ namespace trackerEndpoint.message = (!trackerMessage.isEmpty() ? trackerMessage : errorMessage); trackerEntry.endpoints.append(trackerEndpoint); + trackerEntry.numPeers = std::max(trackerEntry.numPeers, trackerEndpoint.numPeers); trackerEntry.numSeeds = std::max(trackerEntry.numSeeds, trackerEndpoint.numSeeds); trackerEntry.numLeeches = std::max(trackerEntry.numLeeches, trackerEndpoint.numLeeches); trackerEntry.numDownloaded = std::max(trackerEntry.numDownloaded, trackerEndpoint.numDownloaded); @@ -471,21 +477,17 @@ QVector TorrentImpl::trackers() const for (const lt::announce_entry &tracker : nativeTrackers) { + const QString trackerURL = QString::fromStdString(tracker.url); #if (LIBTORRENT_VERSION_NUM >= 20000) - entries << fromNativeAnnouncerEntry(tracker, m_nativeHandle.info_hashes()); + entries << fromNativeAnnouncerEntry(tracker, m_nativeHandle.info_hashes(), m_trackerPeerCounts[trackerURL]); #else - entries << fromNativeAnnouncerEntry(tracker); + entries << fromNativeAnnouncerEntry(tracker, m_trackerPeerCounts[trackerURL]); #endif } return entries; } -QHash TorrentImpl::trackerInfos() const -{ - return m_trackerInfos; -} - void TorrentImpl::addTrackers(const QVector &trackers) { QSet currentTrackers; @@ -1646,10 +1648,8 @@ void TorrentImpl::handleMoveStorageJobFinished(const bool hasOutstandingJob) void TorrentImpl::handleTrackerReplyAlert(const lt::tracker_reply_alert *p) { - const QString trackerUrl(p->tracker_url()); - qDebug("Received a tracker reply from %s (Num_peers = %d)", qUtf8Printable(trackerUrl), p->num_peers); - // Connection was successful now. Remove possible old errors - m_trackerInfos[trackerUrl] = {p->num_peers}; + const QString trackerUrl = p->tracker_url(); + m_trackerPeerCounts[trackerUrl][p->local_endpoint] = p->num_peers; m_session->handleTorrentTrackerReply(this, trackerUrl); } diff --git a/src/base/bittorrent/torrentimpl.h b/src/base/bittorrent/torrentimpl.h index c97444e8f..8f59e722f 100644 --- a/src/base/bittorrent/torrentimpl.h +++ b/src/base/bittorrent/torrentimpl.h @@ -33,11 +33,13 @@ #include #include +#include #include #include #include #include +#include #include #include #include @@ -154,7 +156,6 @@ namespace BitTorrent bool hasFilteredPieces() const override; int queuePosition() const override; QVector trackers() const override; - QHash trackerInfos() const override; QVector urlSeeds() const override; QString error() const override; qlonglong totalDownload() const override; @@ -308,7 +309,7 @@ namespace BitTorrent // we will rely on this workaround to remove empty leftover folders QHash> m_oldPath; - QHash m_trackerInfos; + QHash> m_trackerPeerCounts; // Persistent data QString m_name; diff --git a/src/base/bittorrent/trackerentry.h b/src/base/bittorrent/trackerentry.h index fe9fe7090..63f7434db 100644 --- a/src/base/bittorrent/trackerentry.h +++ b/src/base/bittorrent/trackerentry.h @@ -49,6 +49,7 @@ namespace BitTorrent int protocolVersion = 1; Status status = NotContacted; + int numPeers = -1; int numSeeds = -1; int numLeeches = -1; int numDownloaded = -1; @@ -62,6 +63,7 @@ namespace BitTorrent // Deprecated fields Status status = NotContacted; + int numPeers = -1; int numSeeds = -1; int numLeeches = -1; int numDownloaded = -1; diff --git a/src/gui/properties/trackerlistwidget.cpp b/src/gui/properties/trackerlistwidget.cpp index 5e49b0d01..a1d00af09 100644 --- a/src/gui/properties/trackerlistwidget.cpp +++ b/src/gui/properties/trackerlistwidget.cpp @@ -362,7 +362,6 @@ void TrackerListWidget::loadTrackers() loadStickyItems(torrent); // Load actual trackers information - const QHash trackerData = torrent->trackerInfos(); QStringList oldTrackerURLs = m_trackerItems.keys(); for (const BitTorrent::TrackerEntry &entry : asConst(torrent->trackers())) @@ -384,8 +383,6 @@ void TrackerListWidget::loadTrackers() item->setText(COL_TIER, QString::number(entry.tier)); - const BitTorrent::TrackerInfo data = trackerData.value(trackerURL); - switch (entry.status) { case BitTorrent::TrackerEntry::Working: @@ -403,7 +400,9 @@ void TrackerListWidget::loadTrackers() } item->setText(COL_MSG, entry.message); - item->setText(COL_PEERS, QString::number(data.numPeers)); + item->setText(COL_PEERS, ((entry.numPeers > -1) + ? QString::number(entry.numPeers) + : tr("N/A"))); item->setText(COL_SEEDS, ((entry.numSeeds > -1) ? QString::number(entry.numSeeds) : tr("N/A"))); diff --git a/src/webui/api/torrentscontroller.cpp b/src/webui/api/torrentscontroller.cpp index 3f88854cc..8596b112f 100644 --- a/src/webui/api/torrentscontroller.cpp +++ b/src/webui/api/torrentscontroller.cpp @@ -453,18 +453,15 @@ void TorrentsController::trackersAction() QJsonArray trackerList = getStickyTrackers(torrent); - QHash trackersData = torrent->trackerInfos(); for (const BitTorrent::TrackerEntry &tracker : asConst(torrent->trackers())) { - const BitTorrent::TrackerInfo data = trackersData.value(tracker.url); - trackerList << QJsonObject { {KEY_TRACKER_URL, tracker.url}, {KEY_TRACKER_TIER, tracker.tier}, {KEY_TRACKER_STATUS, static_cast(tracker.status)}, - {KEY_TRACKER_PEERS_COUNT, data.numPeers}, {KEY_TRACKER_MSG, tracker.message}, + {KEY_TRACKER_PEERS_COUNT, tracker.numPeers}, {KEY_TRACKER_SEEDS_COUNT, tracker.numSeeds}, {KEY_TRACKER_LEECHES_COUNT, tracker.numLeeches}, {KEY_TRACKER_DOWNLOADED_COUNT, tracker.numDownloaded}