Browse Source

Improve ReverseResolution class

* Use QHostAddress type for IP
* Avoid double lookup
* Use larger cache size
adaptive-webui-19844
Chocobo1 5 years ago
parent
commit
86e3d0d816
No known key found for this signature in database
GPG Key ID: 210D9C873253A68C
  1. 51
      src/base/net/reverseresolution.cpp
  2. 9
      src/base/net/reverseresolution.h
  3. 7
      src/gui/properties/peerlistwidget.cpp
  4. 3
      src/gui/properties/peerlistwidget.h

51
src/base/net/reverseresolution.cpp

@ -28,19 +28,19 @@
#include "reverseresolution.h" #include "reverseresolution.h"
#include <QDebug> #include <QHostAddress>
#include <QHostInfo> #include <QHostInfo>
#include <QString> #include <QString>
const int CACHE_SIZE = 500; const int CACHE_SIZE = 2048;
using namespace Net; using namespace Net;
namespace namespace
{ {
bool isUsefulHostName(const QString &hostname, const QString &ip) bool isUsefulHostName(const QString &hostname, const QHostAddress &ip)
{ {
return (!hostname.isEmpty() && (hostname != ip)); return (!hostname.isEmpty() && (hostname != ip.toString()));
} }
} }
@ -52,37 +52,40 @@ ReverseResolution::ReverseResolution(QObject *parent)
ReverseResolution::~ReverseResolution() ReverseResolution::~ReverseResolution()
{ {
qDebug("Deleting host name resolver...");
} }
void ReverseResolution::resolve(const QString &ip) void ReverseResolution::resolve(const QHostAddress &ip)
{ {
if (m_cache.contains(ip)) { if (ip.isNull())
const QString &hostname = *m_cache.object(ip); return;
qDebug() << "Resolved host name using cache: " << ip << " -> " << hostname;
if (isUsefulHostName(hostname, ip)) const QString *hostname = m_cache.object(ip);
emit ipResolved(ip, hostname); if (hostname) {
} if (hostname->isEmpty())
else { ; // resolution didn't get meaningful results, so do nothing
// Actually resolve the ip else
m_lookups.insert(QHostInfo::lookupHost(ip, this, &ReverseResolution::hostResolved), ip); emit ipResolved(ip, *hostname);
return;
} }
// do reverse lookup: IP -> hostname
const int lookupId = QHostInfo::lookupHost(ip.toString(), this, &ReverseResolution::hostResolved);
m_lookups.insert(lookupId, ip);
} }
void ReverseResolution::hostResolved(const QHostInfo &host) void ReverseResolution::hostResolved(const QHostInfo &host)
{ {
const QString ip = m_lookups.take(host.lookupId()); const QHostAddress ip = m_lookups.take(host.lookupId());
Q_ASSERT(!ip.isNull());
if (host.error() != QHostInfo::NoError) { if (host.error() != QHostInfo::NoError)
qDebug() << "DNS Reverse resolution error: " << host.errorString();
return; return;
}
const QString hostname = host.hostName(); const QString hostname = host.hostName();
if (isUsefulHostName(hostname, ip)) {
qDebug() << Q_FUNC_INFO << ip << QString("->") << hostname; m_cache.insert(ip, new QString(hostname));
m_cache.insert(ip, new QString(hostname));
if (isUsefulHostName(hostname, ip))
emit ipResolved(ip, hostname); emit ipResolved(ip, hostname);
}
else {
m_cache.insert(ip, new QString());
}
} }

9
src/base/net/reverseresolution.h

@ -32,6 +32,7 @@
#include <QCache> #include <QCache>
#include <QObject> #include <QObject>
class QHostAddress;
class QHostInfo; class QHostInfo;
class QString; class QString;
@ -46,17 +47,17 @@ namespace Net
explicit ReverseResolution(QObject *parent = nullptr); explicit ReverseResolution(QObject *parent = nullptr);
~ReverseResolution(); ~ReverseResolution();
void resolve(const QString &ip); void resolve(const QHostAddress &ip);
signals: signals:
void ipResolved(const QString &ip, const QString &hostname); void ipResolved(const QHostAddress &ip, const QString &hostname);
private slots: private slots:
void hostResolved(const QHostInfo &host); void hostResolved(const QHostInfo &host);
private: private:
QHash<int /* LookupID */, QString /* IP */> m_lookups; QHash<int, QHostAddress> m_lookups; // <LookupID, IP>
QCache<QString /* IP */, QString /* HostName */> m_cache; QCache<QHostAddress, QString> m_cache; // <IP, HostName>
}; };
} }

7
src/gui/properties/peerlistwidget.cpp

@ -33,6 +33,7 @@
#include <QApplication> #include <QApplication>
#include <QClipboard> #include <QClipboard>
#include <QHeaderView> #include <QHeaderView>
#include <QHostAddress>
#include <QMenu> #include <QMenu>
#include <QMessageBox> #include <QMessageBox>
#include <QSet> #include <QSet>
@ -405,7 +406,7 @@ void PeerListWidget::updatePeer(const BitTorrent::TorrentHandle *torrent, const
m_listModel->setData(m_listModel->index(row, PeerListDelegate::DOWNLOADING_PIECE), downloadingFiles.join('\n'), Qt::ToolTipRole); m_listModel->setData(m_listModel->index(row, PeerListDelegate::DOWNLOADING_PIECE), downloadingFiles.join('\n'), Qt::ToolTipRole);
if (m_resolver) if (m_resolver)
m_resolver->resolve(peerIp); m_resolver->resolve(peerEndpoint.address.ip);
if (m_resolveCountries) { if (m_resolveCountries) {
const QIcon icon = UIThemeManager::instance()->getFlagIcon(peer.country()); const QIcon icon = UIThemeManager::instance()->getFlagIcon(peer.country());
@ -417,10 +418,10 @@ void PeerListWidget::updatePeer(const BitTorrent::TorrentHandle *torrent, const
} }
} }
void PeerListWidget::handleResolved(const QString &ip, const QString &hostname) void PeerListWidget::handleResolved(const QHostAddress &ip, const QString &hostname) const
{ {
for (auto iter = m_peerItems.cbegin(); iter != m_peerItems.cend(); ++iter) { for (auto iter = m_peerItems.cbegin(); iter != m_peerItems.cend(); ++iter) {
if (iter.key().address.ip.toString() == ip) { if (iter.key().address.ip == ip) {
QStandardItem *item {iter.value()}; QStandardItem *item {iter.value()};
item->setData(hostname, Qt::DisplayRole); item->setData(hostname, Qt::DisplayRole);
break; break;

3
src/gui/properties/peerlistwidget.h

@ -32,6 +32,7 @@
#include <QHash> #include <QHash>
#include <QTreeView> #include <QTreeView>
class QHostAddress;
class QStandardItem; class QStandardItem;
class QStandardItemModel; class QStandardItemModel;
@ -72,7 +73,7 @@ private slots:
void banSelectedPeers(); void banSelectedPeers();
void copySelectedPeers(); void copySelectedPeers();
void handleSortColumnChanged(int col); void handleSortColumnChanged(int col);
void handleResolved(const QString &ip, const QString &hostname); void handleResolved(const QHostAddress &ip, const QString &hostname) const;
private: private:
void updatePeer(const BitTorrent::TorrentHandle *torrent, const BitTorrent::PeerInfo &peer, bool &isNewPeer); void updatePeer(const BitTorrent::TorrentHandle *torrent, const BitTorrent::PeerInfo &peer, bool &isNewPeer);

Loading…
Cancel
Save