masternode: replace RecursiveMutex with Mutex in meta store/info and add thread-safety annotations (LOCKS_EXCLUDED); avoid re-entrant locking

Author: PastaPastaPasta

src/masternode/meta.cpp

@@ -66,7 +66,7 @@ void CMasternodeMetaInfo::RemoveGovernanceObject(const uint256& nGovernanceObjec
     mapGovernanceObjectsVotedOn.erase(nGovernanceObjectHash);
 }
 
-CMasternodeMetaInfoPtr CMasternodeMetaMan::GetMetaInfo(const uint256& proTxHash, bool fCreate)
+CMasternodeMetaInfoPtr CMasternodeMetaMan::GetMetaInfo(const uint256& proTxHash, bool fCreate) LOCKS_EXCLUDED(cs)
 {
     LOCK(cs);
     auto it = metaInfos.find(proTxHash);
@@ -115,28 +115,28 @@ bool CMasternodeMetaMan::AddGovernanceVote(const uint256& proTxHash, const uint2
     return true;
 }
 
-void CMasternodeMetaMan::RemoveGovernanceObject(const uint256& nGovernanceObjectHash)
+void CMasternodeMetaMan::RemoveGovernanceObject(const uint256& nGovernanceObjectHash) LOCKS_EXCLUDED(cs)
 {
     LOCK(cs);
     for(const auto& p : metaInfos) {
         p.second->RemoveGovernanceObject(nGovernanceObjectHash);
     }
 }
 
-std::vector<uint256> CMasternodeMetaMan::GetAndClearDirtyGovernanceObjectHashes()
+std::vector<uint256> CMasternodeMetaMan::GetAndClearDirtyGovernanceObjectHashes() LOCKS_EXCLUDED(cs)
 {
     std::vector<uint256> vecTmp;
     WITH_LOCK(cs, vecTmp.swap(vecDirtyGovernanceObjectHashes));
     return vecTmp;
 }
 
-bool CMasternodeMetaMan::AlreadyHavePlatformBan(const uint256& inv_hash) const
+bool CMasternodeMetaMan::AlreadyHavePlatformBan(const uint256& inv_hash) const LOCKS_EXCLUDED(cs)
 {
     LOCK(cs);
     return m_seen_platform_bans.exists(inv_hash);
 }
 
-std::optional<PlatformBanMessage> CMasternodeMetaMan::GetPlatformBan(const uint256& inv_hash) const
+std::optional<PlatformBanMessage> CMasternodeMetaMan::GetPlatformBan(const uint256& inv_hash) const LOCKS_EXCLUDED(cs)
 {
     LOCK(cs);
     PlatformBanMessage ret;
@@ -147,13 +147,13 @@ std::optional<PlatformBanMessage> CMasternodeMetaMan::GetPlatformBan(const uint2
     return ret;
 }
 
-void CMasternodeMetaMan::RememberPlatformBan(const uint256& inv_hash, PlatformBanMessage&& msg)
+void CMasternodeMetaMan::RememberPlatformBan(const uint256& inv_hash, PlatformBanMessage&& msg) LOCKS_EXCLUDED(cs)
 {
     LOCK(cs);
     m_seen_platform_bans.insert(inv_hash, std::move(msg));
 }
 
-std::string MasternodeMetaStore::ToString() const
+std::string MasternodeMetaStore::ToString() const LOCKS_EXCLUDED(cs)
 {
     LOCK(cs);
     return strprintf("Masternodes: meta infos object count: %d, nDsqCount: %d", metaInfos.size(), nDsqCount);

src/masternode/meta.h

@@ -35,7 +35,7 @@ class CMasternodeMetaInfo
     friend class CMasternodeMetaMan;
 
 private:
-    mutable RecursiveMutex cs;
+    mutable Mutex cs;
 
     uint256 proTxHash GUARDED_BY(cs);
 
@@ -127,14 +127,14 @@ class MasternodeMetaStore
 protected:
     static const std::string SERIALIZATION_VERSION_STRING;
 
-    mutable RecursiveMutex cs;
+    mutable Mutex cs;
     std::map<uint256, CMasternodeMetaInfoPtr> metaInfos GUARDED_BY(cs);
     // keep track of dsq count to prevent masternodes from gaming coinjoin queue
     std::atomic<int64_t> nDsqCount{0};
 
 public:
     template<typename Stream>
-    void Serialize(Stream &s) const
+    void Serialize(Stream &s) const LOCKS_EXCLUDED(cs)
     {
         LOCK(cs);
         std::vector<CMasternodeMetaInfo> tmpMetaInfo;
@@ -145,7 +145,7 @@ class MasternodeMetaStore
     }
 
     template<typename Stream>
-    void Unserialize(Stream &s)
+    void Unserialize(Stream &s) LOCKS_EXCLUDED(cs)
     {
         Clear();
 
@@ -163,14 +163,14 @@ class MasternodeMetaStore
         }
     }
 
-    void Clear()
+    void Clear() LOCKS_EXCLUDED(cs)
     {
         LOCK(cs);
 
         metaInfos.clear();
     }
 
-    std::string ToString() const;
+    std::string ToString() const LOCKS_EXCLUDED(cs);
 };
 
 /**
@@ -233,7 +233,7 @@ class CMasternodeMetaMan : public MasternodeMetaStore
 
     bool IsValid() const { return is_valid; }
 
-    CMasternodeMetaInfoPtr GetMetaInfo(const uint256& proTxHash, bool fCreate = true);
+    CMasternodeMetaInfoPtr GetMetaInfo(const uint256& proTxHash, bool fCreate = true) LOCKS_EXCLUDED(cs);
 
     int64_t GetDsqCount() const { return nDsqCount; }
     int64_t GetDsqThreshold(const uint256& proTxHash, int nMnCount);
@@ -242,13 +242,13 @@ class CMasternodeMetaMan : public MasternodeMetaStore
     void DisallowMixing(const uint256& proTxHash);
 
     bool AddGovernanceVote(const uint256& proTxHash, const uint256& nGovernanceObjectHash);
-    void RemoveGovernanceObject(const uint256& nGovernanceObjectHash);
+    void RemoveGovernanceObject(const uint256& nGovernanceObjectHash) LOCKS_EXCLUDED(cs);
 
-    std::vector<uint256> GetAndClearDirtyGovernanceObjectHashes();
+    std::vector<uint256> GetAndClearDirtyGovernanceObjectHashes() LOCKS_EXCLUDED(cs);
 
-    bool AlreadyHavePlatformBan(const uint256& inv_hash) const;
-    std::optional<PlatformBanMessage> GetPlatformBan(const uint256& inv_hash) const;
-    void RememberPlatformBan(const uint256& inv_hash, PlatformBanMessage&& msg);
+    bool AlreadyHavePlatformBan(const uint256& inv_hash) const LOCKS_EXCLUDED(cs);
+    std::optional<PlatformBanMessage> GetPlatformBan(const uint256& inv_hash) const LOCKS_EXCLUDED(cs);
+    void RememberPlatformBan(const uint256& inv_hash, PlatformBanMessage&& msg) LOCKS_EXCLUDED(cs);
 };
 
 #endif // BITCOIN_MASTERNODE_META_H