p2p: Add CConnman::GetCurrentBlockRelayOnlyConns()

hebasto · hebasto · commit bad16aff490d · 2020-10-09T14:29:04.000+03:00
diff --git a/src/net.cpp b/src/net.cpp
@@ -2019,6 +2019,19 @@ void CConnman::ThreadOpenConnections(const std::vector<std::string> connect)
     }
 }
 
+std::vector<CAddress> CConnman::GetCurrentBlockRelayOnlyConns() const
+{
+    std::vector<CAddress> ret;
+    LOCK(cs_vNodes);
+    for (const CNode* pnode : vNodes) {
+        if (pnode->IsBlockOnlyConn()) {
+            ret.push_back(pnode->addr);
+        }
+    }
+
+    return ret;
+}
+
 std::vector<AddedNodeInfo> CConnman::GetAddedNodeInfo()
 {
     std::vector<AddedNodeInfo> ret;
diff --git a/src/net.h b/src/net.h
@@ -460,6 +460,11 @@ class CConnman
     void RecordBytesRecv(uint64_t bytes);
     void RecordBytesSent(uint64_t bytes);
 
+    /**
+     * Return vector of current BLOCK_RELAY peers.
+     */
+    std::vector<CAddress> GetCurrentBlockRelayOnlyConns() const;
+
     // Whether the node should be passed out in ForEach* callbacks
     static bool NodeFullyConnected(const CNode* pnode);
 

Original file line number	Diff line number	Diff line change
`@@ -2019,6 +2019,19 @@ void CConnman::ThreadOpenConnections(const std::vector<std::string> connect)`
`2019`	`2019`	`}`
`2020`	`2020`	`}`
`2021`	`2021`
	`2022`	`+std::vector<CAddress> CConnman::GetCurrentBlockRelayOnlyConns() const`
	`2023`	`+{`
	`2024`	`+ std::vector<CAddress> ret;`
	`2025`	`+ LOCK(cs_vNodes);`
	`2026`	`+ for (const CNode* pnode : vNodes) {`
	`2027`	`+ if (pnode->IsBlockOnlyConn()) {`
	`2028`	`+ ret.push_back(pnode->addr);`
	`2029`	`+ }`
	`2030`	`+ }`
	`2031`	`+`
	`2032`	`+ return ret;`
	`2033`	`+}`
	`2034`	`+`
`2022`	`2035`	`std::vector<AddedNodeInfo> CConnman::GetAddedNodeInfo()`
`2023`	`2036`	`{`
`2024`	`2037`	`std::vector<AddedNodeInfo> ret;`