mirror of
https://github.com/dashpay/dash.git
synced 2024-12-25 12:02:48 +01:00
Merge #20210: net: assert CNode::m_inbound_onion is inbound in ctor, add getter, unit tests
86c495223f048e5ca2cf0d8730af7db3b76f7aba net: add CNode::IsInboundOnion() public getter and unit tests (Jon Atack) 6609eb8cb50fe92c7317b5db9e72d4333b3aab1b net: assert CNode::m_inbound_onion is inbound in ctor (Jon Atack) 993d1ecd191a7d9161082d4026f020cbf00835bb test, fuzz: fix constructing CNode with invalid inbound_onion (Jon Atack) Pull request description: The goal of this PR is to be able to depend on `m_inbound_onion` in AttemptToEvictConnection in #20197: - asserts `CNode::m_inbound_onion` is inbound in the CNode ctor to have a validity check at the class boundary - fixes a unit test and a fuzz utility that were passing invalid inbound onion values to the CNode ctor - drops an unneeded check in `CNode::ConnectedThroughNetwork()` for its inbound status - adds a public getter `IsInboundOnion()` that also allows unit testing it - adds unit test coverage ACKs for top commit: sipa: utACK 86c495223f048e5ca2cf0d8730af7db3b76f7aba LarryRuane: ACK 86c495223f048e5ca2cf0d8730af7db3b76f7aba vasild: ACK 86c495223f048e5ca2cf0d8730af7db3b76f7aba MarcoFalke: review ACK 86c495223f048e5ca2cf0d8730af7db3b76f7aba 🐍 Tree-SHA512: 21109105bc4e5e03076fadd489204be00eac710c9de0127708ca2d0a10a048ff81f640f589a7429967ac3eb51d35fe24bb2b12e53e7aa3efbc47aaff6396d204
This commit is contained in:
parent
0b109bed58
commit
5107598bf9
@ -638,7 +638,7 @@ std::string CNode::GetLogString() const
|
|||||||
|
|
||||||
Network CNode::ConnectedThroughNetwork() const
|
Network CNode::ConnectedThroughNetwork() const
|
||||||
{
|
{
|
||||||
return IsInboundConn() && m_inbound_onion ? NET_ONION : addr.GetNetClass();
|
return m_inbound_onion ? NET_ONION : addr.GetNetClass();
|
||||||
}
|
}
|
||||||
|
|
||||||
#undef X
|
#undef X
|
||||||
@ -4024,6 +4024,7 @@ CNode::CNode(NodeId idIn, ServiceFlags nLocalServicesIn, SOCKET hSocketIn, const
|
|||||||
nLocalServices(nLocalServicesIn),
|
nLocalServices(nLocalServicesIn),
|
||||||
m_inbound_onion(inbound_onion)
|
m_inbound_onion(inbound_onion)
|
||||||
{
|
{
|
||||||
|
if (inbound_onion) assert(conn_type_in == ConnectionType::INBOUND);
|
||||||
hSocket = hSocketIn;
|
hSocket = hSocketIn;
|
||||||
addrName = addrNameIn == "" ? addr.ToStringIPPort() : addrNameIn;
|
addrName = addrNameIn == "" ? addr.ToStringIPPort() : addrNameIn;
|
||||||
hashContinue = uint256();
|
hashContinue = uint256();
|
||||||
|
@ -1394,7 +1394,7 @@ private:
|
|||||||
CService addrLocal GUARDED_BY(cs_addrLocal);
|
CService addrLocal GUARDED_BY(cs_addrLocal);
|
||||||
mutable RecursiveMutex cs_addrLocal;
|
mutable RecursiveMutex cs_addrLocal;
|
||||||
|
|
||||||
//! Whether this peer connected via our Tor onion service.
|
//! Whether this peer is an inbound onion, e.g. connected via our Tor onion service.
|
||||||
const bool m_inbound_onion{false};
|
const bool m_inbound_onion{false};
|
||||||
|
|
||||||
// Challenge sent in VERSION to be answered with MNAUTH (only happens between MNs)
|
// Challenge sent in VERSION to be answered with MNAUTH (only happens between MNs)
|
||||||
@ -1532,6 +1532,9 @@ public:
|
|||||||
|
|
||||||
|
|
||||||
std::string ConnectionTypeAsString() const;
|
std::string ConnectionTypeAsString() const;
|
||||||
|
|
||||||
|
/** Whether this peer is an inbound onion, e.g. connected via our Tor onion service. */
|
||||||
|
bool IsInboundOnion() const { return m_inbound_onion; }
|
||||||
std::string GetLogString() const;
|
std::string GetLogString() const;
|
||||||
|
|
||||||
bool CanRelay() const { return !m_masternode_connection || m_masternode_iqr_connection; }
|
bool CanRelay() const { return !m_masternode_connection || m_masternode_iqr_connection; }
|
||||||
|
@ -317,7 +317,7 @@ auto ConsumeNode(FuzzedDataProvider& fuzzed_data_provider, const std::optional<N
|
|||||||
const std::string addr_name = fuzzed_data_provider.ConsumeRandomLengthString(64);
|
const std::string addr_name = fuzzed_data_provider.ConsumeRandomLengthString(64);
|
||||||
|
|
||||||
const ConnectionType conn_type = fuzzed_data_provider.PickValueInArray({ConnectionType::INBOUND, ConnectionType::OUTBOUND_FULL_RELAY, ConnectionType::MANUAL, ConnectionType::FEELER, ConnectionType::BLOCK_RELAY, ConnectionType::ADDR_FETCH});
|
const ConnectionType conn_type = fuzzed_data_provider.PickValueInArray({ConnectionType::INBOUND, ConnectionType::OUTBOUND_FULL_RELAY, ConnectionType::MANUAL, ConnectionType::FEELER, ConnectionType::BLOCK_RELAY, ConnectionType::ADDR_FETCH});
|
||||||
const bool inbound_onion = fuzzed_data_provider.ConsumeBool();
|
const bool inbound_onion{conn_type == ConnectionType::INBOUND ? fuzzed_data_provider.ConsumeBool() : false};
|
||||||
if constexpr (ReturnUniquePtr) {
|
if constexpr (ReturnUniquePtr) {
|
||||||
return std::make_unique<CNode>(node_id, local_services, socket, address, keyed_net_group, local_host_nonce, addr_bind, addr_name, conn_type, inbound_onion);
|
return std::make_unique<CNode>(node_id, local_services, socket, address, keyed_net_group, local_host_nonce, addr_bind, addr_name, conn_type, inbound_onion);
|
||||||
} else {
|
} else {
|
||||||
|
@ -199,6 +199,7 @@ BOOST_AUTO_TEST_CASE(cnode_simple_test)
|
|||||||
BOOST_CHECK(pnode1->IsFeelerConn() == false);
|
BOOST_CHECK(pnode1->IsFeelerConn() == false);
|
||||||
BOOST_CHECK(pnode1->IsAddrFetchConn() == false);
|
BOOST_CHECK(pnode1->IsAddrFetchConn() == false);
|
||||||
BOOST_CHECK(pnode1->IsInboundConn() == false);
|
BOOST_CHECK(pnode1->IsInboundConn() == false);
|
||||||
|
BOOST_CHECK(pnode1->IsInboundOnion() == false);
|
||||||
BOOST_CHECK_EQUAL(pnode1->ConnectedThroughNetwork(), Network::NET_IPV4);
|
BOOST_CHECK_EQUAL(pnode1->ConnectedThroughNetwork(), Network::NET_IPV4);
|
||||||
|
|
||||||
std::unique_ptr<CNode> pnode2 = std::make_unique<CNode>(
|
std::unique_ptr<CNode> pnode2 = std::make_unique<CNode>(
|
||||||
@ -213,6 +214,7 @@ BOOST_AUTO_TEST_CASE(cnode_simple_test)
|
|||||||
BOOST_CHECK(pnode2->IsFeelerConn() == false);
|
BOOST_CHECK(pnode2->IsFeelerConn() == false);
|
||||||
BOOST_CHECK(pnode2->IsAddrFetchConn() == false);
|
BOOST_CHECK(pnode2->IsAddrFetchConn() == false);
|
||||||
BOOST_CHECK(pnode2->IsInboundConn() == true);
|
BOOST_CHECK(pnode2->IsInboundConn() == true);
|
||||||
|
BOOST_CHECK(pnode2->IsInboundOnion() == false);
|
||||||
BOOST_CHECK_EQUAL(pnode2->ConnectedThroughNetwork(), Network::NET_IPV4);
|
BOOST_CHECK_EQUAL(pnode2->ConnectedThroughNetwork(), Network::NET_IPV4);
|
||||||
|
|
||||||
std::unique_ptr<CNode> pnode3 = std::make_unique<CNode>(
|
std::unique_ptr<CNode> pnode3 = std::make_unique<CNode>(
|
||||||
@ -220,13 +222,14 @@ BOOST_AUTO_TEST_CASE(cnode_simple_test)
|
|||||||
/* nKeyedNetGroupIn = */ 0,
|
/* nKeyedNetGroupIn = */ 0,
|
||||||
/* nLocalHostNonceIn = */ 0,
|
/* nLocalHostNonceIn = */ 0,
|
||||||
CAddress(), pszDest, ConnectionType::OUTBOUND_FULL_RELAY,
|
CAddress(), pszDest, ConnectionType::OUTBOUND_FULL_RELAY,
|
||||||
/* inbound_onion = */ true);
|
/* inbound_onion = */ false);
|
||||||
BOOST_CHECK(pnode3->IsFullOutboundConn() == true);
|
BOOST_CHECK(pnode3->IsFullOutboundConn() == true);
|
||||||
BOOST_CHECK(pnode3->IsManualConn() == false);
|
BOOST_CHECK(pnode3->IsManualConn() == false);
|
||||||
BOOST_CHECK(pnode3->IsBlockOnlyConn() == false);
|
BOOST_CHECK(pnode3->IsBlockOnlyConn() == false);
|
||||||
BOOST_CHECK(pnode3->IsFeelerConn() == false);
|
BOOST_CHECK(pnode3->IsFeelerConn() == false);
|
||||||
BOOST_CHECK(pnode3->IsAddrFetchConn() == false);
|
BOOST_CHECK(pnode3->IsAddrFetchConn() == false);
|
||||||
BOOST_CHECK(pnode3->IsInboundConn() == false);
|
BOOST_CHECK(pnode3->IsInboundConn() == false);
|
||||||
|
BOOST_CHECK(pnode3->IsInboundOnion() == false);
|
||||||
BOOST_CHECK_EQUAL(pnode3->ConnectedThroughNetwork(), Network::NET_IPV4);
|
BOOST_CHECK_EQUAL(pnode3->ConnectedThroughNetwork(), Network::NET_IPV4);
|
||||||
|
|
||||||
std::unique_ptr<CNode> pnode4 = std::make_unique<CNode>(
|
std::unique_ptr<CNode> pnode4 = std::make_unique<CNode>(
|
||||||
@ -241,6 +244,7 @@ BOOST_AUTO_TEST_CASE(cnode_simple_test)
|
|||||||
BOOST_CHECK(pnode4->IsFeelerConn() == false);
|
BOOST_CHECK(pnode4->IsFeelerConn() == false);
|
||||||
BOOST_CHECK(pnode4->IsAddrFetchConn() == false);
|
BOOST_CHECK(pnode4->IsAddrFetchConn() == false);
|
||||||
BOOST_CHECK(pnode4->IsInboundConn() == true);
|
BOOST_CHECK(pnode4->IsInboundConn() == true);
|
||||||
|
BOOST_CHECK(pnode4->IsInboundOnion() == true);
|
||||||
BOOST_CHECK_EQUAL(pnode4->ConnectedThroughNetwork(), Network::NET_ONION);
|
BOOST_CHECK_EQUAL(pnode4->ConnectedThroughNetwork(), Network::NET_ONION);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user