clientmodel.cpp

Go to the documentation of this file.

// Copyright (c) 2011-2014 The Bitcoin developers
// Copyright (c) 2014-2015 The Dash developers
// Copyright (c) 2015-2022 The PIVX Core developers
// Distributed under the MIT/X11 software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
 
#include "clientmodel.h"
 
#include "bantablemodel.h"
#include "guiconstants.h"
#include "guiutil.h"
#include "peertablemodel.h"
 
#include "chainparams.h"
#include "checkpoints.h"
#include "clientversion.h"
#include "interfaces/handler.h"
#include "mapport.h"
#include "masternodeman.h"
#include "net.h"
#include "netbase.h"
#include "guiinterface.h"
#include "util/system.h"
#include "validation.h"
#include "warnings.h"
 
#include <stdint.h>
 
#include <QDebug>
#include <QTimer>
 
static const int64_t nClientStartupTime = GetTime();
// Last tip update notification
static int64_t nLastBlockTipUpdateNotification = 0;
 
ClientModel::ClientModel(OptionsModel* optionsModel, QObject* parent) : QObject(parent),
                                                                        optionsModel(optionsModel),
                                                                        peerTableModel(0),
                                                                        banTableModel(0),
                                                                        cacheTip(nullptr),
                                                                        cachedMasternodeCountString(""),
                                                                        cachedReindexing(0), cachedImporting(0),
                                                                        numBlocksAtStartup(-1), pollTimer(0)
{
    peerTableModel = new PeerTableModel(this);
    banTableModel = new BanTableModel(this);
    pollTimer = new QTimer(this);
    connect(pollTimer, &QTimer::timeout, this, &ClientModel::updateTimer);
    pollTimer->start(MODEL_UPDATE_DELAY);
 
    pollMnTimer = new QTimer(this);
    connect(pollMnTimer, &QTimer::timeout, this, &ClientModel::updateMnTimer);
    startMasternodesTimer();
 
    subscribeToCoreSignals();
}
 
ClientModel::~ClientModel()
{
    unsubscribeFromCoreSignals();
}
 
int ClientModel::getNumConnections(unsigned int flags) const
{
    CConnman::NumConnections connections = CConnman::CONNECTIONS_NONE;
 
    if(flags == CONNECTIONS_IN)
        connections = CConnman::CONNECTIONS_IN;
    else if (flags == CONNECTIONS_OUT)
        connections = CConnman::CONNECTIONS_OUT;
    else if (flags == CONNECTIONS_ALL)
        connections = CConnman::CONNECTIONS_ALL;
 
    if (g_connman)
         return g_connman->GetNodeCount(connections);
    return 0;
}
 
QString ClientModel::getMasternodeCountString()
{
    const auto& info = mnodeman.getMNsInfo();
    int unknown = std::max(0, info.total - info.ipv4 - info.ipv6 - info.onion);
    m_cached_masternodes_count = info.total;
    return tr("Total: %1 (IPv4: %2 / IPv6: %3 / Tor: %4 / Unknown: %5)").arg(QString::number(info.total))
                                                                        .arg(QString::number(info.ipv4))
                                                                        .arg(QString::number(info.ipv6))
                                                                        .arg(QString::number(info.onion))
                                                                        .arg(QString::number(unknown));
}
 
QString ClientModel::getMasternodesCountString()
{
    if (!cachedMasternodeCountString.isEmpty()) {
        return cachedMasternodeCountString;
    }
 
    // Force an update
    cachedMasternodeCountString = getMasternodeCountString();
    return cachedMasternodeCountString;
}
 
int ClientModel::getNumBlocks()
{
    if (!cacheTip) {
        cacheTip = WITH_LOCK(cs_main, return chainActive.Tip(););
    }
 
    return cacheTip ? cacheTip->nHeight : 0;
}
 
int ClientModel::getNumBlocksAtStartup()
{
    if (numBlocksAtStartup == -1) numBlocksAtStartup = getNumBlocks();
    return numBlocksAtStartup;
}
 
quint64 ClientModel::getTotalBytesRecv() const
{
    if(!g_connman)
        return 0;
    return g_connman->GetTotalBytesRecv();
}
 
quint64 ClientModel::getTotalBytesSent() const
{
    if(!g_connman)
        return 0;
    return g_connman->GetTotalBytesSent();
}
 
QDateTime ClientModel::getLastBlockDate() const
{
    const int nTime = (cacheTip == nullptr ? Params().GenesisBlock().GetBlockTime() : cacheTip->GetBlockTime());
    return QDateTime::fromTime_t(nTime);
}
 
QString ClientModel::getLastBlockHash() const
{
    const uint256& nHash = (cacheTip == nullptr ? Params().GenesisBlock().GetHash() : cacheTip->GetBlockHash());
    return QString::fromStdString(nHash.GetHex());
}
 
uint256 ClientModel::getLastBlockProcessed() const
{
    return cacheTip == nullptr ? Params().GenesisBlock().GetHash() : cacheTip->GetBlockHash();
}
 
int ClientModel::getLastBlockProcessedHeight() const
{
    return cacheTip == nullptr ? 0 : cacheTip->nHeight;
}
 
int64_t ClientModel::getLastBlockProcessedTime() const
{
    return cacheTip == nullptr ? Params().GenesisBlock().GetBlockTime() : cacheTip->GetBlockTime();
}
 
bool ClientModel::isTipCached() const
{
    return cacheTip;
}
 
double ClientModel::getVerificationProgress() const
{
    return Checkpoints::GuessVerificationProgress(cacheTip);
}
 
void ClientModel::updateTimer()
{
    // Get required lock upfront. This avoids the GUI from getting stuck on
    // periodical polls if the core is holding the locks for a longer time -
    // for example, during a wallet rescan.
    Q_EMIT bytesChanged(getTotalBytesRecv(), getTotalBytesSent());
}
 
void ClientModel::updateMnTimer()
{
    // Following method is locking the mnmanager mutex for now,
    // future: move to an event based update.
    QString newMasternodeCountString = getMasternodeCountString();
 
    if (cachedMasternodeCountString != newMasternodeCountString) {
        cachedMasternodeCountString = newMasternodeCountString;
 
        Q_EMIT strMasternodesChanged(cachedMasternodeCountString);
    }
}
 
void ClientModel::startMasternodesTimer()
{
    if (!pollMnTimer->isActive()) {
        // no need to update as frequent as data for balances/txes/blocks
        pollMnTimer->start(MODEL_UPDATE_DELAY * 40);
    }
}
 
void ClientModel::stopMasternodesTimer()
{
    if (pollMnTimer->isActive()) {
        pollMnTimer->stop();
    }
}
 
void ClientModel::updateNumConnections(int numConnections)
{
    Q_EMIT numConnectionsChanged(numConnections);
}
 
void ClientModel::updateNetworkActive(bool networkActive)
{
    Q_EMIT networkActiveChanged(networkActive);
}
 
void ClientModel::updateAlert()
{
    Q_EMIT alertsChanged(getStatusBarWarnings());
}
 
bool ClientModel::inInitialBlockDownload() const
{
    return cachedInitialSync;
}
 
enum BlockSource ClientModel::getBlockSource() const
{
    if (fReindex)
        return BLOCK_SOURCE_REINDEX;
    else if (fImporting)
        return BLOCK_SOURCE_DISK;
    else if (getNumConnections() > 0)
        return BLOCK_SOURCE_NETWORK;
 
    return BLOCK_SOURCE_NONE;
}
 
void ClientModel::setNetworkActive(bool active)
{
    if (g_connman) {
        g_connman->SetNetworkActive(active);
    }
}
 
bool ClientModel::getNetworkActive() const
{
    if (g_connman) {
        return g_connman->GetNetworkActive();
    }
    return false;
}
 
QString ClientModel::getStatusBarWarnings() const
{
    return QString::fromStdString(GetWarnings("statusbar"));
}
 
OptionsModel* ClientModel::getOptionsModel()
{
    return optionsModel;
}
 
PeerTableModel* ClientModel::getPeerTableModel()
{
    return peerTableModel;
}
 
BanTableModel *ClientModel::getBanTableModel()
{
    return banTableModel;
}
 
QString ClientModel::formatFullVersion() const
{
    return QString::fromStdString(FormatFullVersion());
}
 
bool ClientModel::isReleaseVersion() const
{
    return CLIENT_VERSION_IS_RELEASE;
}
 
QString ClientModel::clientName() const
{
    return QString::fromStdString(CLIENT_NAME);
}
 
QString ClientModel::formatClientStartupTime() const
{
    return QDateTime::fromTime_t(nClientStartupTime).toString();
}
 
QString ClientModel::dataDir() const
{
    return GUIUtil::boostPathToQString(GetDataDir());
}
 
void ClientModel::updateBanlist()
{
    banTableModel->refresh();
}
 
static void BlockTipChanged(ClientModel *clientmodel, bool initialSync, const CBlockIndex *pIndex)
{
    // lock free async UI updates in case we have a new block tip
    // during initial sync, only update the UI if the last update
    // was > 1000ms (MODEL_UPDATE_DELAY) ago
    int64_t now = 0;
    if (initialSync)
        now = GetTimeMillis();
 
    // if we are in-sync, update the UI regardless of last update time
    if (!initialSync || now - nLastBlockTipUpdateNotification > MODEL_UPDATE_DELAY) {
        //pass a async signal to the UI thread
        clientmodel->setCacheTip(pIndex);
        clientmodel->setCacheImporting(fImporting);
        clientmodel->setCacheReindexing(fReindex);
        clientmodel->setCacheInitialSync(initialSync);
        Q_EMIT clientmodel->numBlocksChanged(pIndex->nHeight);
        nLastBlockTipUpdateNotification = now;
    }
}
 
// Handlers for core signals
static void ShowProgress(ClientModel* clientmodel, const std::string& title, int nProgress)
{
    // emits signal "showProgress"
    QMetaObject::invokeMethod(clientmodel, "showProgress", Qt::QueuedConnection,
        Q_ARG(QString, QString::fromStdString(title)),
        Q_ARG(int, nProgress));
}
 
static void NotifyNumConnectionsChanged(ClientModel* clientmodel, int newNumConnections)
{
    // Too noisy: qDebug() << "NotifyNumConnectionsChanged : " + QString::number(newNumConnections);
    QMetaObject::invokeMethod(clientmodel, "updateNumConnections", Qt::QueuedConnection,
        Q_ARG(int, newNumConnections));
}
 
static void NotifyNetworkActiveChanged(ClientModel *clientmodel, bool networkActive)
{
    QMetaObject::invokeMethod(clientmodel, "updateNetworkActive", Qt::QueuedConnection,
                              Q_ARG(bool, networkActive));
}
 
static void NotifyAlertChanged(ClientModel* clientmodel)
{
    qDebug() << "NotifyAlertChanged";
    QMetaObject::invokeMethod(clientmodel, "updateAlert", Qt::QueuedConnection);
}
 
static void BannedListChanged(ClientModel *clientmodel)
{
    qDebug() << QString("%1: Requesting update for peer banlist").arg(__func__);
    QMetaObject::invokeMethod(clientmodel, "updateBanlist", Qt::QueuedConnection);
}
 
void ClientModel::subscribeToCoreSignals()
{
    // Connect signals to client
    m_handler_show_progress = interfaces::MakeHandler(uiInterface.ShowProgress.connect(std::bind(ShowProgress, this, std::placeholders::_1, std::placeholders::_2)));
    m_handler_notify_num_connections_changed = interfaces::MakeHandler(uiInterface.NotifyNumConnectionsChanged.connect(std::bind(NotifyNumConnectionsChanged, this, std::placeholders::_1)));
    m_handler_notify_net_activity_changed = interfaces::MakeHandler(uiInterface.NotifyNetworkActiveChanged.connect(std::bind(NotifyNetworkActiveChanged, this, std::placeholders::_1)));
    m_handler_notify_alert_changed = interfaces::MakeHandler(uiInterface.NotifyAlertChanged.connect(std::bind(NotifyAlertChanged, this)));
    m_handler_banned_list_changed = interfaces::MakeHandler(uiInterface.BannedListChanged.connect(std::bind(BannedListChanged, this)));
    m_handler_notify_block_tip = interfaces::MakeHandler(uiInterface.NotifyBlockTip.connect(std::bind(BlockTipChanged, this, std::placeholders::_1, std::placeholders::_2)));
}
 
void ClientModel::unsubscribeFromCoreSignals()
{
    // Disconnect signals from client
    m_handler_show_progress->disconnect();
    m_handler_notify_num_connections_changed->disconnect();
    m_handler_notify_net_activity_changed->disconnect();
    m_handler_notify_alert_changed->disconnect();
    m_handler_banned_list_changed->disconnect();
    m_handler_notify_block_tip->disconnect();
}
 
void ClientModel::mapPort(bool use_upnp, bool use_natpmp) {
    StartMapPort(use_upnp, use_natpmp);
}
 
bool ClientModel::getTorInfo(std::string& ip_port) const
{
    proxyType onion;
    if (GetProxy((Network) 3, onion) && IsReachable((Network) 3)) {
        {
            LOCK(cs_mapLocalHost);
            for (const std::pair<const CNetAddr, LocalServiceInfo>& item : mapLocalHost) {
                if (item.first.IsTor()) {
                    CService addrOnion(LookupNumeric(item.first.ToString(), item.second.nPort));
                    ip_port = addrOnion.ToStringIPPort();
                    return true;
                }
            }
        }
    }
    return false;
}