expose cnode info for RPC

amitiuttarwar · Nov 9, 2023 · cd930ac · cd930ac
1 parent 38bb1cb
commit cd930ac
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/src/net.cpp b/src/net.cpp
@@ -3698,6 +3698,14 @@ uint32_t CConnman::GetMappedAS(const CNetAddr& addr) const
     return m_netgroupman.GetMappedAS(addr);
 }
 
+void CConnman::GetNodeMemory(std::map<NodeId, size_t>& info) const
+{
+    LOCK(m_nodes_mutex);
+    for (CNode* pnode : m_nodes) {
+        info[pnode->GetId()] = pnode->DynamicMemoryUsage();
+    }
+}
+
 void CConnman::GetNodeStats(std::vector<CNodeStats>& vstats) const
 {
     vstats.clear();
@@ -4003,6 +4011,24 @@ bool CConnman::ForNode(NodeId id, std::function<bool(CNode* pnode)> func)
     return found != nullptr && NodeFullyConnected(found) && func(found);
 }
 
+std::string CConnman::NodeToString(NodeId id) const
+{
+    CNode* found = nullptr;
+    LOCK(m_nodes_mutex);
+    for (auto&& pnode : m_nodes) {
+        if(pnode->GetId() == id) {
+            found = pnode;
+            break;
+        }
+    }
+
+    if (found != nullptr) {
+        return ConnectionTypeAsString(found->m_conn_type);
+    } else {
+        return "";
+    }
+}
+
 CSipHasher CConnman::GetDeterministicRandomizer(uint64_t id) const
 {
     return CSipHasher(nSeed0, nSeed1).Write(id);

diff --git a/src/net.h b/src/net.h
@@ -1235,6 +1235,8 @@ class CConnman
     size_t GetNodeCount(ConnectionDirection) const;
     uint32_t GetMappedAS(const CNetAddr& addr) const;
     void GetNodeStats(std::vector<CNodeStats>& vstats) const;
+    void GetNodeMemory(std::map<NodeId, size_t>& info) const;
+    std::string NodeToString(NodeId id) const;
     bool DisconnectNode(const std::string& node);
     bool DisconnectNode(const CSubNet& subnet);
     bool DisconnectNode(const CNetAddr& addr);