Connect net requests manager load, dump and maintenance job.

PIVX-Project · Feb 10, 2022 · 42f76c9 · 42f76c9
1 parent ad60ad7
commit 42f76c9
Show file tree

Hide file tree

Showing 6 changed files with 48 additions and 51 deletions.
diff --git a/src/init.cpp b/src/init.cpp
@@ -1741,7 +1741,16 @@ bool AppInitMain()
 
     // ********************************************************* Step 10: setup layer 2 data
 
-    LoadTierTwo(chain_active_height, fReindexChainState);
+    bool load_cache_files = !(fReindex || fReindexChainState);
+    {
+        LOCK(cs_main);
+        // was blocks/chainstate deleted?
+        if (chainActive.Tip() == nullptr) {
+            load_cache_files = false;
+        }
+    }
+
+    LoadTierTwo(chain_active_height, load_cache_files);
     if (!InitActiveMN()) return false;
     RegisterTierTwoValidationInterface();
 

diff --git a/src/masternode-payments.cpp b/src/masternode-payments.cpp
@@ -11,6 +11,7 @@
 #include "budget/budgetmanager.h"
 #include "masternodeman.h"
 #include "netmessagemaker.h"
+#include "tiertwo/netfulfilledman.h"
 #include "spork.h"
 #include "sync.h"
 #include "tiertwo/tiertwo_sync_state.h"
@@ -411,13 +412,13 @@ bool CMasternodePayments::ProcessMessageMasternodePayments(CNode* pfrom, std::st
         vRecv >> nCountNeeded;
 
         if (Params().NetworkIDString() == CBaseChainParams::MAIN) {
-            if (pfrom->HasFulfilledRequest(NetMsgType::GETMNWINNERS)) {
+            if (g_netfulfilledman.HasFulfilledRequest(pfrom->addr, NetMsgType::GETMNWINNERS)) {
                 LogPrint(BCLog::MASTERNODE, "%s: mnget - peer already asked me for the list\n", __func__);
                 return state.DoS(20, false, REJECT_INVALID, "getmnwinners-request-already-fulfilled");
             }
         }
 
-        pfrom->FulfilledRequest(NetMsgType::GETMNWINNERS);
+        g_netfulfilledman.AddFulfilledRequest(pfrom->addr, NetMsgType::GETMNWINNERS);
         Sync(pfrom, nCountNeeded);
         LogPrint(BCLog::MASTERNODE, "mnget - Sent Masternode winners to peer %i\n", pfrom->GetId());
     } else if (strCommand == NetMsgType::MNWINNER) {

diff --git a/src/masternode-sync.cpp b/src/masternode-sync.cpp
@@ -12,6 +12,7 @@
 #include "masternode.h"
 #include "masternodeman.h"
 #include "netmessagemaker.h"
+#include "tiertwo/netfulfilledman.h"
 #include "spork.h"
 #include "tiertwo/tiertwo_sync_state.h"
 #include "util/system.h"
@@ -177,12 +178,7 @@ void CMasternodeSync::ProcessSyncStatusMsg(int nItemID, int nCount)
 
 void CMasternodeSync::ClearFulfilledRequest()
 {
-    g_connman->ForEachNode([](CNode* pnode) {
-        pnode->ClearFulfilledRequest("getspork");
-        pnode->ClearFulfilledRequest("mnsync");
-        pnode->ClearFulfilledRequest("mnwsync");
-        pnode->ClearFulfilledRequest("busync");
-    });
+    g_netfulfilledman.Clear();
 }
 
 void CMasternodeSync::Process()
@@ -276,8 +272,8 @@ bool CMasternodeSync::SyncWithNode(CNode* pnode, bool fLegacyMnObsolete)
         }
 
         // Request sporks sync if we haven't requested it yet.
-        if (pnode->HasFulfilledRequest("getspork")) return true;
-        pnode->FulfilledRequest("getspork");
+        if (g_netfulfilledman.HasFulfilledRequest(pnode->addr, "getspork")) return true;
+        g_netfulfilledman.AddFulfilledRequest(pnode->addr, "getspork");
 
         g_connman->PushMessage(pnode, msgMaker.Make(NetMsgType::GETSPORKS));
         RequestedMasternodeAttempt++;
@@ -318,15 +314,15 @@ bool CMasternodeSync::SyncWithNode(CNode* pnode, bool fLegacyMnObsolete)
         if (RequestedMasternodeAttempt >= MASTERNODE_SYNC_THRESHOLD * 4) return false;
 
         // Request mnb sync if we haven't requested it yet.
-        if (pnode->HasFulfilledRequest("mnsync")) return true;
+        if (g_netfulfilledman.HasFulfilledRequest(pnode->addr, "mnsync")) return true;
 
         // Try to request MN list sync.
         if (!mnodeman.RequestMnList(pnode)) {
             return true; // Failed, try next peer.
         }
 
         // Mark sync requested.
-        pnode->FulfilledRequest("mnsync");
+        g_netfulfilledman.AddFulfilledRequest(pnode->addr, "mnsync");
         // Increase the sync attempt count
         RequestedMasternodeAttempt++;
 
@@ -369,10 +365,10 @@ bool CMasternodeSync::SyncWithNode(CNode* pnode, bool fLegacyMnObsolete)
         if (RequestedMasternodeAttempt >= MASTERNODE_SYNC_THRESHOLD * 2) return false;
 
         // Request mnw sync if we haven't requested it yet.
-        if (pnode->HasFulfilledRequest("mnwsync")) return true;
+        if (g_netfulfilledman.HasFulfilledRequest(pnode->addr, "mnwsync")) return true;
 
         // Mark sync requested.
-        pnode->FulfilledRequest("mnwsync");
+        g_netfulfilledman.AddFulfilledRequest(pnode->addr, "mnwsync");
 
         // Sync mn winners
         int nMnCount = mnodeman.CountEnabled(true /* only_legacy */);
@@ -408,10 +404,10 @@ bool CMasternodeSync::SyncWithNode(CNode* pnode, bool fLegacyMnObsolete)
         if (RequestedMasternodeAttempt >= MASTERNODE_SYNC_THRESHOLD * 3) return false;
 
         // Request bud sync if we haven't requested it yet.
-        if (pnode->HasFulfilledRequest("busync")) return true;
+        if (g_netfulfilledman.HasFulfilledRequest(pnode->addr, "busync")) return true;
 
         // Mark sync requested.
-        pnode->FulfilledRequest("busync");
+        g_netfulfilledman.AddFulfilledRequest(pnode->addr, "busync");
 
         // Sync proposals, finalizations and votes
         uint256 n;

diff --git a/src/net.h b/src/net.h
@@ -710,8 +710,6 @@ class CNode
     mapMsgCmdSize mapSendBytesPerMsgCmd;
     mapMsgCmdSize mapRecvBytesPerMsgCmd;
 
-    std::vector<std::string> vecRequestsFulfilled; //keep track of what client has asked for
-
 public:
     uint256 hashContinue;
     std::atomic<int> nStartingHeight;
@@ -892,32 +890,6 @@ class CNode
     // inv response received, clear it from the waiting inv set.
     void AskForInvReceived(const uint256& invHash);
 
-    bool HasFulfilledRequest(std::string strRequest)
-    {
-        for (std::string& type : vecRequestsFulfilled) {
-            if (type == strRequest) return true;
-        }
-        return false;
-    }
-
-    void ClearFulfilledRequest(std::string strRequest)
-    {
-        std::vector<std::string>::iterator it = vecRequestsFulfilled.begin();
-        while (it != vecRequestsFulfilled.end()) {
-            if ((*it) == strRequest) {
-                vecRequestsFulfilled.erase(it);
-                return;
-            }
-            ++it;
-        }
-    }
-
-    void FulfilledRequest(std::string strRequest)
-    {
-        if (HasFulfilledRequest(strRequest)) return;
-        vecRequestsFulfilled.push_back(strRequest);
-    }
-
     void CloseSocketDisconnect();
     bool DisconnectOldProtocol(int nVersionIn, int nVersionRequired);
 

diff --git a/src/tiertwo/init.cpp b/src/tiertwo/init.cpp
@@ -13,7 +13,9 @@
 #include "masternode-payments.h"
 #include "masternodeconfig.h"
 #include "llmq/quorums_init.h"
+#include "scheduler.h"
 #include "tiertwo/masternode_meta_manager.h"
+#include "tiertwo/netfulfilledman.h"
 #include "validation.h"
 #include "wallet/wallet.h"
 
@@ -65,7 +67,7 @@ static void LoadBlockHashesCache(CMasternodeMan& man)
     }
 }
 
-bool LoadTierTwo(int chain_active_height, bool fReindexChainState)
+bool LoadTierTwo(int chain_active_height, bool load_cache_files)
 {
     // ################################# //
     // ## Legacy Masternodes Manager ### //
@@ -118,11 +120,9 @@ bool LoadTierTwo(int chain_active_height, bool fReindexChainState)
     // ############################## //
     // ## Net MNs Metadata Manager ## //
     // ############################## //
-    bool fLoadCacheFiles = !(fReindex || fReindexChainState);
-    fs::path pathDB = GetDataDir();
     uiInterface.InitMessage(_("Loading masternode cache..."));
     CFlatDB<CMasternodeMetaMan> metadb(MN_META_CACHE_FILENAME, MN_META_CACHE_FILE_ID);
-    if (fLoadCacheFiles) {
+    if (load_cache_files) {
         if (!metadb.Load(g_mmetaman)) {
             return UIError(strprintf(_("Failed to load masternode metadata cache from: %s"), metadb.GetDbPath().string()));
         }
@@ -133,6 +133,22 @@ bool LoadTierTwo(int chain_active_height, bool fReindexChainState)
         }
     }
 
+    // ############################## //
+    // ## Network Requests Manager ## //
+    // ############################## //
+    uiInterface.InitMessage(_("Loading network requests cache..."));
+    CFlatDB<CNetFulfilledRequestManager> netRequestsDb(NET_REQUESTS_CACHE_FILENAME, NET_REQUESTS_CACHE_FILE_ID);
+    if (load_cache_files) {
+        if (!netRequestsDb.Load(g_netfulfilledman)) {
+            LogPrintf("Failed to load network requests cache from %s", netRequestsDb.GetDbPath().string());
+        }
+    } else {
+        CNetFulfilledRequestManager netfulfilledmanTmp;
+        if (!netRequestsDb.Dump(netfulfilledmanTmp)) {
+            LogPrintf("Failed to clear network requests cache at %s", netRequestsDb.GetDbPath().string());
+        }
+    }
+
     return true;
 }
 
@@ -148,7 +164,8 @@ void DumpTierTwo()
     DumpMasternodes();
     DumpBudgets(g_budgetman);
     DumpMasternodePayments();
-    CFlatDB<CMasternodeMetaMan>("mnmetacache.dat", "magicMasternodeMetaCache").Dump(g_mmetaman);
+    CFlatDB<CMasternodeMetaMan>(MN_META_CACHE_FILENAME, MN_META_CACHE_FILE_ID).Dump(g_mmetaman);
+    CFlatDB<CNetFulfilledRequestManager>(NET_REQUESTS_CACHE_FILENAME, NET_REQUESTS_CACHE_FILE_ID).Dump(g_netfulfilledman);
 }
 
 void SetBudgetFinMode(const std::string& mode)
@@ -240,6 +257,7 @@ bool InitActiveMN()
 void StartTierTwoThreadsAndScheduleJobs(boost::thread_group& threadGroup, CScheduler& scheduler)
 {
     threadGroup.create_thread(std::bind(&ThreadCheckMasternodes));
+    scheduler.scheduleEvery(std::bind(&CNetFulfilledRequestManager::DoMaintenance, std::ref(g_netfulfilledman)), 60 * 1000);
 
     // Start LLMQ system
     if (gArgs.GetBoolArg("-disabledkg", false)) {

diff --git a/src/tiertwo/init.h b/src/tiertwo/init.h
@@ -6,6 +6,7 @@
 #define PIVX_TIERTWO_INIT_H
 
 #include <string>
+#include "fs.h"
 
 static const bool DEFAULT_MASTERNODE  = false;
 static const bool DEFAULT_MNCONFLOCK = true;
@@ -24,7 +25,7 @@ void InitTierTwoPreChainLoad(bool fReindex);
 void InitTierTwoPostCoinsCacheLoad();
 
 /** Loads from disk all the tier two related objects */
-bool LoadTierTwo(int chain_active_height, bool fReindexChainState);
+bool LoadTierTwo(int chain_active_height, bool load_cache_files);
 
 /** Register all tier two objects */
 void RegisterTierTwoValidationInterface();