Skip to content

Commit 86c4952

Browse files
committed
net: add CNode::IsInboundOnion() public getter and unit tests
1 parent 6609eb8 commit 86c4952

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

src/net.h

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1234,6 +1234,9 @@ class CNode
12341234
void MaybeSetAddrName(const std::string& addrNameIn);
12351235

12361236
std::string ConnectionTypeAsString() const;
1237+
1238+
/** Whether this peer is an inbound onion, e.g. connected via our Tor onion service. */
1239+
bool IsInboundOnion() const { return m_inbound_onion; }
12371240
};
12381241

12391242
/** Return a timestamp in the future (in microseconds) for exponentially distributed events. */

src/test/net_tests.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -200,6 +200,7 @@ BOOST_AUTO_TEST_CASE(cnode_simple_test)
200200
BOOST_CHECK(pnode1->IsFeelerConn() == false);
201201
BOOST_CHECK(pnode1->IsAddrFetchConn() == false);
202202
BOOST_CHECK(pnode1->IsInboundConn() == false);
203+
BOOST_CHECK(pnode1->IsInboundOnion() == false);
203204
BOOST_CHECK_EQUAL(pnode1->ConnectedThroughNetwork(), Network::NET_IPV4);
204205

205206
std::unique_ptr<CNode> pnode2 = MakeUnique<CNode>(
@@ -214,6 +215,7 @@ BOOST_AUTO_TEST_CASE(cnode_simple_test)
214215
BOOST_CHECK(pnode2->IsFeelerConn() == false);
215216
BOOST_CHECK(pnode2->IsAddrFetchConn() == false);
216217
BOOST_CHECK(pnode2->IsInboundConn() == true);
218+
BOOST_CHECK(pnode2->IsInboundOnion() == false);
217219
BOOST_CHECK_EQUAL(pnode2->ConnectedThroughNetwork(), Network::NET_IPV4);
218220

219221
std::unique_ptr<CNode> pnode3 = MakeUnique<CNode>(
@@ -228,6 +230,7 @@ BOOST_AUTO_TEST_CASE(cnode_simple_test)
228230
BOOST_CHECK(pnode3->IsFeelerConn() == false);
229231
BOOST_CHECK(pnode3->IsAddrFetchConn() == false);
230232
BOOST_CHECK(pnode3->IsInboundConn() == false);
233+
BOOST_CHECK(pnode3->IsInboundOnion() == false);
231234
BOOST_CHECK_EQUAL(pnode3->ConnectedThroughNetwork(), Network::NET_IPV4);
232235

233236
std::unique_ptr<CNode> pnode4 = MakeUnique<CNode>(
@@ -242,6 +245,7 @@ BOOST_AUTO_TEST_CASE(cnode_simple_test)
242245
BOOST_CHECK(pnode4->IsFeelerConn() == false);
243246
BOOST_CHECK(pnode4->IsAddrFetchConn() == false);
244247
BOOST_CHECK(pnode4->IsInboundConn() == true);
248+
BOOST_CHECK(pnode4->IsInboundOnion() == true);
245249
BOOST_CHECK_EQUAL(pnode4->ConnectedThroughNetwork(), Network::NET_ONION);
246250
}
247251

0 commit comments

Comments
 (0)