mirror of
https://github.com/qbittorrent/qBittorrent.git
synced 2024-12-16 13:11:36 +03:00
400 lines
17 KiB
C++
400 lines
17 KiB
C++
/*
|
|
* Bittorrent Client using Qt4 and libtorrent.
|
|
* Copyright (C) 2006 Christophe Dumez
|
|
*
|
|
* 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.
|
|
*
|
|
* Contact : chris@qbittorrent.org
|
|
*/
|
|
|
|
#include <QStandardItemModel>
|
|
#include <QSortFilterProxyModel>
|
|
#include <QSet>
|
|
#include <QHeaderView>
|
|
#include <QMenu>
|
|
#include <QClipboard>
|
|
#include <QMessageBox>
|
|
#ifdef QBT_USES_QT5
|
|
#include <QTableView>
|
|
#endif
|
|
|
|
#include "base/net/reverseresolution.h"
|
|
#include "base/bittorrent/torrenthandle.h"
|
|
#include "base/bittorrent/peerinfo.h"
|
|
#include "base/preferences.h"
|
|
#include "base/logger.h"
|
|
#include "base/unicodestrings.h"
|
|
#include "propertieswidget.h"
|
|
#include "base/net/geoipmanager.h"
|
|
#include "peersadditiondlg.h"
|
|
#include "speedlimitdlg.h"
|
|
#include "guiiconprovider.h"
|
|
#include "peerlistdelegate.h"
|
|
#include "peerlistsortmodel.h"
|
|
#include "peerlistwidget.h"
|
|
|
|
PeerListWidget::PeerListWidget(PropertiesWidget *parent)
|
|
: QTreeView(parent)
|
|
, m_properties(parent)
|
|
, m_displayFlags(false)
|
|
{
|
|
// Load settings
|
|
loadSettings();
|
|
// Visual settings
|
|
setUniformRowHeights(true);
|
|
setRootIsDecorated(false);
|
|
setItemsExpandable(false);
|
|
setAllColumnsShowFocus(true);
|
|
setSelectionMode(QAbstractItemView::ExtendedSelection);
|
|
// List Model
|
|
m_listModel = new QStandardItemModel(0, PeerListDelegate::COL_COUNT);
|
|
m_listModel->setHeaderData(PeerListDelegate::COUNTRY, Qt::Horizontal, QVariant()); // Country flag column
|
|
m_listModel->setHeaderData(PeerListDelegate::IP, Qt::Horizontal, tr("IP"));
|
|
m_listModel->setHeaderData(PeerListDelegate::PORT, Qt::Horizontal, tr("Port"));
|
|
m_listModel->setHeaderData(PeerListDelegate::FLAGS, Qt::Horizontal, tr("Flags"));
|
|
m_listModel->setHeaderData(PeerListDelegate::CONNECTION, Qt::Horizontal, tr("Connection"));
|
|
m_listModel->setHeaderData(PeerListDelegate::CLIENT, Qt::Horizontal, tr("Client", "i.e.: Client application"));
|
|
m_listModel->setHeaderData(PeerListDelegate::PROGRESS, Qt::Horizontal, tr("Progress", "i.e: % downloaded"));
|
|
m_listModel->setHeaderData(PeerListDelegate::DOWN_SPEED, Qt::Horizontal, tr("Down Speed", "i.e: Download speed"));
|
|
m_listModel->setHeaderData(PeerListDelegate::UP_SPEED, Qt::Horizontal, tr("Up Speed", "i.e: Upload speed"));
|
|
m_listModel->setHeaderData(PeerListDelegate::TOT_DOWN, Qt::Horizontal, tr("Downloaded", "i.e: total data downloaded"));
|
|
m_listModel->setHeaderData(PeerListDelegate::TOT_UP, Qt::Horizontal, tr("Uploaded", "i.e: total data uploaded"));
|
|
m_listModel->setHeaderData(PeerListDelegate::RELEVANCE, Qt::Horizontal, tr("Relevance", "i.e: How relevant this peer is to us. How many pieces it has that we don't."));
|
|
// Proxy model to support sorting without actually altering the underlying model
|
|
m_proxyModel = new PeerListSortModel();
|
|
m_proxyModel->setDynamicSortFilter(true);
|
|
m_proxyModel->setSourceModel(m_listModel);
|
|
m_proxyModel->setSortCaseSensitivity(Qt::CaseInsensitive);
|
|
setModel(m_proxyModel);
|
|
//Explicitly set the column visibility. When columns are added/removed
|
|
//between versions this prevents some of them being hidden due to
|
|
//incorrect restoreState() being used.
|
|
for (unsigned int i = 0; i < PeerListDelegate::IP_HIDDEN; i++)
|
|
showColumn(i);
|
|
hideColumn(PeerListDelegate::IP_HIDDEN);
|
|
hideColumn(PeerListDelegate::COL_COUNT);
|
|
if (!Preferences::instance()->resolvePeerCountries())
|
|
hideColumn(PeerListDelegate::COUNTRY);
|
|
//To also mitigate the above issue, we have to resize each column when
|
|
//its size is 0, because explicitly 'showing' the column isn't enough
|
|
//in the above scenario.
|
|
for (unsigned int i = 0; i < PeerListDelegate::IP_HIDDEN; i++)
|
|
if (!columnWidth(i))
|
|
resizeColumnToContents(i);
|
|
// Context menu
|
|
setContextMenuPolicy(Qt::CustomContextMenu);
|
|
connect(this, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(showPeerListMenu(QPoint)));
|
|
// List delegate
|
|
m_listDelegate = new PeerListDelegate(this);
|
|
setItemDelegate(m_listDelegate);
|
|
// Enable sorting
|
|
setSortingEnabled(true);
|
|
// IP to Hostname resolver
|
|
updatePeerHostNameResolutionState();
|
|
// SIGNAL/SLOT
|
|
connect(header(), SIGNAL(sectionClicked(int)), SLOT(handleSortColumnChanged(int)));
|
|
handleSortColumnChanged(header()->sortIndicatorSection());
|
|
m_copyHotkey = new QShortcut(QKeySequence(Qt::ControlModifier + Qt::Key_C), this, SLOT(copySelectedPeers()), 0, Qt::WidgetShortcut);
|
|
|
|
#ifdef QBT_USES_QT5
|
|
// This hack fixes reordering of first column with Qt5.
|
|
// https://github.com/qtproject/qtbase/commit/e0fc088c0c8bc61dbcaf5928b24986cd61a22777
|
|
QTableView unused;
|
|
unused.setVerticalHeader(this->header());
|
|
this->header()->setParent(this);
|
|
unused.setVerticalHeader(new QHeaderView(Qt::Horizontal));
|
|
#endif
|
|
}
|
|
|
|
PeerListWidget::~PeerListWidget()
|
|
{
|
|
saveSettings();
|
|
delete m_proxyModel;
|
|
delete m_listModel;
|
|
delete m_listDelegate;
|
|
if (m_resolver)
|
|
delete m_resolver;
|
|
delete m_copyHotkey;
|
|
}
|
|
|
|
void PeerListWidget::updatePeerHostNameResolutionState()
|
|
{
|
|
if (Preferences::instance()->resolvePeerHostNames()) {
|
|
if (!m_resolver) {
|
|
m_resolver = new Net::ReverseResolution(this);
|
|
connect(m_resolver, SIGNAL(ipResolved(QString, QString)), SLOT(handleResolved(QString, QString)));
|
|
loadPeers(m_properties->getCurrentTorrent(), true);
|
|
}
|
|
}
|
|
else if (m_resolver) {
|
|
delete m_resolver;
|
|
}
|
|
}
|
|
|
|
void PeerListWidget::updatePeerCountryResolutionState()
|
|
{
|
|
if (Preferences::instance()->resolvePeerCountries() != m_displayFlags) {
|
|
m_displayFlags = !m_displayFlags;
|
|
if (m_displayFlags) {
|
|
loadPeers(m_properties->getCurrentTorrent());
|
|
showColumn(PeerListDelegate::COUNTRY);
|
|
resizeColumnToContents(PeerListDelegate::COUNTRY);
|
|
}
|
|
else {
|
|
hideColumn(PeerListDelegate::COUNTRY);
|
|
}
|
|
}
|
|
}
|
|
|
|
void PeerListWidget::showPeerListMenu(const QPoint&)
|
|
{
|
|
QMenu menu;
|
|
bool emptyMenu = true;
|
|
BitTorrent::TorrentHandle *const torrent = m_properties->getCurrentTorrent();
|
|
if (!torrent) return;
|
|
|
|
// Add Peer Action
|
|
QAction *addPeerAct = 0;
|
|
if (!torrent->isQueued() && !torrent->isChecking()) {
|
|
addPeerAct = menu.addAction(GuiIconProvider::instance()->getIcon("user-group-new"), tr("Add a new peer..."));
|
|
emptyMenu = false;
|
|
}
|
|
QAction *banAct = 0;
|
|
QAction *copyPeerAct = 0;
|
|
if (!selectionModel()->selectedRows().isEmpty()) {
|
|
copyPeerAct = menu.addAction(GuiIconProvider::instance()->getIcon("edit-copy"), tr("Copy selected"));
|
|
menu.addSeparator();
|
|
banAct = menu.addAction(GuiIconProvider::instance()->getIcon("user-group-delete"), tr("Ban peer permanently"));
|
|
emptyMenu = false;
|
|
}
|
|
if (emptyMenu) return;
|
|
QAction *act = menu.exec(QCursor::pos());
|
|
if (act == 0) return;
|
|
if (act == addPeerAct) {
|
|
QList<BitTorrent::PeerAddress> peersList = PeersAdditionDlg::askForPeers();
|
|
int peerCount = 0;
|
|
foreach (const BitTorrent::PeerAddress &addr, peersList) {
|
|
if (torrent->connectPeer(addr)) {
|
|
qDebug("Adding peer %s...", qPrintable(addr.ip.toString()));
|
|
Logger::instance()->addMessage(tr("Manually adding peer '%1'...").arg(addr.ip.toString()));
|
|
peerCount++;
|
|
}
|
|
else {
|
|
Logger::instance()->addMessage(tr("The peer '%1' could not be added to this torrent.").arg(addr.ip.toString()), Log::WARNING);
|
|
}
|
|
}
|
|
if (peerCount < peersList.length())
|
|
QMessageBox::information(0, tr("Peer addition"), tr("Some peers could not be added. Check the Log for details."));
|
|
else if (peerCount > 0)
|
|
QMessageBox::information(0, tr("Peer addition"), tr("The peers were added to this torrent."));
|
|
return;
|
|
}
|
|
if (act == banAct) {
|
|
banSelectedPeers();
|
|
return;
|
|
}
|
|
if (act == copyPeerAct) {
|
|
copySelectedPeers();
|
|
return;
|
|
}
|
|
}
|
|
|
|
void PeerListWidget::banSelectedPeers()
|
|
{
|
|
// Confirm first
|
|
int ret = QMessageBox::question(this, tr("Ban peer permanently"), tr("Are you sure you want to ban permanently the selected peers?"),
|
|
tr("&Yes"), tr("&No"),
|
|
QString(), 0, 1);
|
|
if (ret)
|
|
return;
|
|
|
|
QModelIndexList selectedIndexes = selectionModel()->selectedRows();
|
|
foreach (const QModelIndex &index, selectedIndexes) {
|
|
int row = m_proxyModel->mapToSource(index).row();
|
|
QString ip = m_listModel->data(m_listModel->index(row, PeerListDelegate::IP_HIDDEN)).toString();
|
|
qDebug("Banning peer %s...", ip.toLocal8Bit().data());
|
|
Logger::instance()->addMessage(tr("Manually banning peer '%1'...").arg(ip));
|
|
BitTorrent::Session::instance()->banIP(ip);
|
|
}
|
|
// Refresh list
|
|
loadPeers(m_properties->getCurrentTorrent());
|
|
}
|
|
|
|
void PeerListWidget::copySelectedPeers()
|
|
{
|
|
QModelIndexList selectedIndexes = selectionModel()->selectedRows();
|
|
QStringList selectedPeers;
|
|
foreach (const QModelIndex &index, selectedIndexes) {
|
|
int row = m_proxyModel->mapToSource(index).row();
|
|
QString ip = m_listModel->data(m_listModel->index(row, PeerListDelegate::IP_HIDDEN)).toString();
|
|
QString myport = m_listModel->data(m_listModel->index(row, PeerListDelegate::PORT)).toString();
|
|
if (ip.indexOf(".") == -1) // IPv6
|
|
selectedPeers << "[" + ip + "]:" + myport;
|
|
else // IPv4
|
|
selectedPeers << ip + ":" + myport;
|
|
}
|
|
QApplication::clipboard()->setText(selectedPeers.join("\n"));
|
|
}
|
|
|
|
void PeerListWidget::clear()
|
|
{
|
|
qDebug("clearing peer list");
|
|
m_peerItems.clear();
|
|
m_peerAddresses.clear();
|
|
m_missingFlags.clear();
|
|
int nbrows = m_listModel->rowCount();
|
|
if (nbrows > 0) {
|
|
qDebug("Cleared %d peers", nbrows);
|
|
m_listModel->removeRows(0, nbrows);
|
|
}
|
|
}
|
|
|
|
void PeerListWidget::loadSettings()
|
|
{
|
|
header()->restoreState(Preferences::instance()->getPeerListState());
|
|
}
|
|
|
|
void PeerListWidget::saveSettings() const
|
|
{
|
|
Preferences::instance()->setPeerListState(header()->saveState());
|
|
}
|
|
|
|
void PeerListWidget::loadPeers(BitTorrent::TorrentHandle *const torrent, bool forceHostnameResolution)
|
|
{
|
|
if (!torrent) return;
|
|
|
|
QList<BitTorrent::PeerInfo> peers = torrent->peers();
|
|
QSet<QString> oldeersSet = m_peerItems.keys().toSet();
|
|
|
|
foreach (const BitTorrent::PeerInfo &peer, peers) {
|
|
BitTorrent::PeerAddress addr = peer.address();
|
|
if (addr.ip.isNull()) continue;
|
|
|
|
QString peerIp = addr.ip.toString();
|
|
if (m_peerItems.contains(peerIp)) {
|
|
// Update existing peer
|
|
updatePeer(peerIp, peer);
|
|
oldeersSet.remove(peerIp);
|
|
if (forceHostnameResolution && m_resolver)
|
|
m_resolver->resolve(peerIp);
|
|
}
|
|
else {
|
|
// Add new peer
|
|
m_peerItems[peerIp] = addPeer(peerIp, peer);
|
|
m_peerAddresses[peerIp] = addr;
|
|
// Resolve peer host name is asked
|
|
if (m_resolver)
|
|
m_resolver->resolve(peerIp);
|
|
}
|
|
}
|
|
// Delete peers that are gone
|
|
QSetIterator<QString> it(oldeersSet);
|
|
while (it.hasNext()) {
|
|
const QString& ip = it.next();
|
|
m_missingFlags.remove(ip);
|
|
m_peerAddresses.remove(ip);
|
|
QStandardItem *item = m_peerItems.take(ip);
|
|
m_listModel->removeRow(item->row());
|
|
}
|
|
}
|
|
|
|
QStandardItem* PeerListWidget::addPeer(const QString &ip, const BitTorrent::PeerInfo &peer)
|
|
{
|
|
int row = m_listModel->rowCount();
|
|
// Adding Peer to peer list
|
|
m_listModel->insertRow(row);
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::IP), ip);
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::IP), ip, Qt::ToolTipRole);
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::PORT), peer.address().port);
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::IP_HIDDEN), ip);
|
|
if (m_displayFlags) {
|
|
const QIcon ico = GuiIconProvider::instance()->getFlagIcon(peer.country());
|
|
if (!ico.isNull()) {
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::COUNTRY), ico, Qt::DecorationRole);
|
|
const QString countryName = Net::GeoIPManager::CountryName(peer.country());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::COUNTRY), countryName, Qt::ToolTipRole);
|
|
}
|
|
else {
|
|
m_missingFlags.insert(ip);
|
|
}
|
|
}
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::CONNECTION), peer.connectionType());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::FLAGS), peer.flags());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::FLAGS), peer.flagsDescription(), Qt::ToolTipRole);
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::CLIENT), peer.client());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::PROGRESS), peer.progress());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::DOWN_SPEED), peer.payloadDownSpeed());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::UP_SPEED), peer.payloadUpSpeed());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::TOT_DOWN), peer.totalDownload());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::TOT_UP), peer.totalUpload());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::RELEVANCE), peer.relevance());
|
|
return m_listModel->item(row, PeerListDelegate::IP);
|
|
}
|
|
|
|
void PeerListWidget::updatePeer(const QString &ip, const BitTorrent::PeerInfo &peer)
|
|
{
|
|
QStandardItem *item = m_peerItems.value(ip);
|
|
int row = item->row();
|
|
if (m_displayFlags) {
|
|
const QIcon ico = GuiIconProvider::instance()->getFlagIcon(peer.country());
|
|
if (!ico.isNull()) {
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::COUNTRY), ico, Qt::DecorationRole);
|
|
const QString countryName = Net::GeoIPManager::CountryName(peer.country());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::COUNTRY), countryName, Qt::ToolTipRole);
|
|
m_missingFlags.remove(ip);
|
|
}
|
|
}
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::CONNECTION), peer.connectionType());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::PORT), peer.address().port);
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::FLAGS), peer.flags());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::FLAGS), peer.flagsDescription(), Qt::ToolTipRole);
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::CLIENT), peer.client());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::PROGRESS), peer.progress());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::DOWN_SPEED), peer.payloadDownSpeed());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::UP_SPEED), peer.payloadUpSpeed());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::TOT_DOWN), peer.totalDownload());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::TOT_UP), peer.totalUpload());
|
|
m_listModel->setData(m_listModel->index(row, PeerListDelegate::RELEVANCE), peer.relevance());
|
|
}
|
|
|
|
void PeerListWidget::handleResolved(const QString &ip, const QString &hostname)
|
|
{
|
|
QStandardItem *item = m_peerItems.value(ip, 0);
|
|
if (item) {
|
|
qDebug("Resolved %s -> %s", qPrintable(ip), qPrintable(hostname));
|
|
item->setData(hostname, Qt::DisplayRole);
|
|
}
|
|
}
|
|
|
|
void PeerListWidget::handleSortColumnChanged(int col)
|
|
{
|
|
if (col == PeerListDelegate::COUNTRY) {
|
|
qDebug("Sorting by decoration");
|
|
m_proxyModel->setSortRole(Qt::ToolTipRole);
|
|
}
|
|
else {
|
|
m_proxyModel->setSortRole(Qt::DisplayRole);
|
|
}
|
|
}
|
|
|