Skip to content

Commit d968616

Browse files
committed
Merge bitcoin/bitcoin#22179: Torv2 removal followups
00b875b addrman: remove invalid addresses when unserializing (Vasil Dimov) bdb6209 fuzz: reduce possible networks check (Vasil Dimov) a164cd3 net: simplify CNetAddr::IsRoutable() (Vasil Dimov) Pull request description: * Simplify some code, now that we know `CNetAddr::IsRFC4193()` and `CNetAddr::IsTor()` cannot be `true` at the same time. * Drop Tor v2 addresses when loading addrman from `peers.dat` - they would have been loaded as dummy-all-zeros IPv6 addresses and linger in addrman, wasting space. ACKs for top commit: sipa: ACK 00b875b. Reviewed the code, and tested with -DDEBUG_ADDRMAN (unit tests + mainnet run with peers.dat that contained v2 onions). laanwj: Code review and lightly tested ACK 00b875b jonatack: ACK 00b875b reviewed, debug-built with -DEBUG_ADDRMAN rebased to current master, restarted node on mainnet/signet/testnet and verified that on each chain -addrinfo shows no change in address counts (as expected). Added some sanity check asserts, rebuilt/re-ran test. Checked that the new test fails on master with "test/addrman_tests.cpp(824): error: in "addrman_tests/remove_invalid": check addrman.size() == 2 has failed [4 != 2]" jarolrod: ACK 00b875b Tree-SHA512: 6ed8e6745134b1b94fffaba28482de909ea39483b46b7f57bda61cdbae7a51251d15cb674de3631772fbeabe153d77a19269f96e62a89102a2d5c01e48f0ba06
2 parents 4129134 + 00b875b commit d968616

File tree

5 files changed

+79
-2
lines changed

5 files changed

+79
-2
lines changed

src/addrman.cpp

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,38 @@ double CAddrInfo::GetChance(int64_t nNow) const
7777
return fChance;
7878
}
7979

80+
void CAddrMan::RemoveInvalid()
81+
{
82+
for (size_t bucket = 0; bucket < ADDRMAN_NEW_BUCKET_COUNT; ++bucket) {
83+
for (size_t i = 0; i < ADDRMAN_BUCKET_SIZE; ++i) {
84+
const auto id = vvNew[bucket][i];
85+
if (id != -1 && !mapInfo[id].IsValid()) {
86+
ClearNew(bucket, i);
87+
}
88+
}
89+
}
90+
91+
for (size_t bucket = 0; bucket < ADDRMAN_TRIED_BUCKET_COUNT; ++bucket) {
92+
for (size_t i = 0; i < ADDRMAN_BUCKET_SIZE; ++i) {
93+
const auto id = vvTried[bucket][i];
94+
if (id == -1) {
95+
continue;
96+
}
97+
const auto& addr_info = mapInfo[id];
98+
if (addr_info.IsValid()) {
99+
continue;
100+
}
101+
vvTried[bucket][i] = -1;
102+
--nTried;
103+
SwapRandom(addr_info.nRandomPos, vRandom.size() - 1);
104+
vRandom.pop_back();
105+
mapAddr.erase(addr_info);
106+
mapInfo.erase(id);
107+
m_tried_collisions.erase(id);
108+
}
109+
}
110+
}
111+
80112
CAddrInfo* CAddrMan::Find(const CNetAddr& addr, int* pnId)
81113
{
82114
AssertLockHeld(cs);

src/addrman.h

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -450,6 +450,8 @@ class CAddrMan
450450
LogPrint(BCLog::ADDRMAN, "addrman lost %i new and %i tried addresses due to collisions\n", nLostUnk, nLost);
451451
}
452452

453+
RemoveInvalid();
454+
453455
Check();
454456
}
455457

@@ -762,6 +764,9 @@ class CAddrMan
762764
//! Update an entry's service bits.
763765
void SetServices_(const CService &addr, ServiceFlags nServices) EXCLUSIVE_LOCKS_REQUIRED(cs);
764766

767+
//! Remove invalid addresses.
768+
void RemoveInvalid() EXCLUSIVE_LOCKS_REQUIRED(cs);
769+
765770
friend class CAddrManTest;
766771
};
767772

src/netaddress.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -489,7 +489,7 @@ bool CNetAddr::IsValid() const
489489
*/
490490
bool CNetAddr::IsRoutable() const
491491
{
492-
return IsValid() && !(IsRFC1918() || IsRFC2544() || IsRFC3927() || IsRFC4862() || IsRFC6598() || IsRFC5737() || (IsRFC4193() && !IsTor()) || IsRFC4843() || IsRFC7343() || IsLocal() || IsInternal());
492+
return IsValid() && !(IsRFC1918() || IsRFC2544() || IsRFC3927() || IsRFC4862() || IsRFC6598() || IsRFC5737() || IsRFC4193() || IsRFC4843() || IsRFC7343() || IsLocal() || IsInternal());
493493
}
494494

495495
/**

src/test/addrman_tests.cpp

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -783,6 +783,46 @@ BOOST_AUTO_TEST_CASE(addrman_serialization)
783783
BOOST_CHECK(bucketAndEntry_asmap1_deser_addr1.second != bucketAndEntry_asmap1_deser_addr2.second);
784784
}
785785

786+
BOOST_AUTO_TEST_CASE(remove_invalid)
787+
{
788+
// Confirm that invalid addresses are ignored in unserialization.
789+
790+
CAddrManTest addrman;
791+
CDataStream stream(SER_NETWORK, PROTOCOL_VERSION);
792+
793+
const CAddress new1{ResolveService("5.5.5.5"), NODE_NONE};
794+
const CAddress new2{ResolveService("6.6.6.6"), NODE_NONE};
795+
const CAddress tried1{ResolveService("7.7.7.7"), NODE_NONE};
796+
const CAddress tried2{ResolveService("8.8.8.8"), NODE_NONE};
797+
798+
addrman.Add({new1, tried1, new2, tried2}, CNetAddr{});
799+
addrman.Good(tried1);
800+
addrman.Good(tried2);
801+
BOOST_REQUIRE_EQUAL(addrman.size(), 4);
802+
803+
stream << addrman;
804+
805+
const std::string str{stream.str()};
806+
size_t pos;
807+
808+
const char new2_raw[]{6, 6, 6, 6};
809+
const uint8_t new2_raw_replacement[]{0, 0, 0, 0}; // 0.0.0.0 is !IsValid()
810+
pos = str.find(new2_raw, 0, sizeof(new2_raw));
811+
BOOST_REQUIRE(pos != std::string::npos);
812+
BOOST_REQUIRE(pos + sizeof(new2_raw_replacement) <= stream.size());
813+
memcpy(stream.data() + pos, new2_raw_replacement, sizeof(new2_raw_replacement));
814+
815+
const char tried2_raw[]{8, 8, 8, 8};
816+
const uint8_t tried2_raw_replacement[]{255, 255, 255, 255}; // 255.255.255.255 is !IsValid()
817+
pos = str.find(tried2_raw, 0, sizeof(tried2_raw));
818+
BOOST_REQUIRE(pos != std::string::npos);
819+
BOOST_REQUIRE(pos + sizeof(tried2_raw_replacement) <= stream.size());
820+
memcpy(stream.data() + pos, tried2_raw_replacement, sizeof(tried2_raw_replacement));
821+
822+
addrman.Clear();
823+
stream >> addrman;
824+
BOOST_CHECK_EQUAL(addrman.size(), 2);
825+
}
786826

787827
BOOST_AUTO_TEST_CASE(addrman_selecttriedcollision)
788828
{

src/test/fuzz/netaddress.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ FUZZ_TARGET(netaddress)
5454
(void)net_addr.IsRFC3927();
5555
(void)net_addr.IsRFC3964();
5656
if (net_addr.IsRFC4193()) {
57-
assert(net_addr.GetNetwork() == Network::NET_ONION || net_addr.GetNetwork() == Network::NET_INTERNAL || net_addr.GetNetwork() == Network::NET_UNROUTABLE);
57+
assert(net_addr.GetNetwork() == Network::NET_INTERNAL || net_addr.GetNetwork() == Network::NET_UNROUTABLE);
5858
}
5959
(void)net_addr.IsRFC4380();
6060
(void)net_addr.IsRFC4843();

0 commit comments

Comments
 (0)