publish NTCP2 addreses by index

This commit is contained in:
orignal 2023-01-24 14:07:22 -05:00
parent 1f23584c24
commit 5479ddd03c
2 changed files with 46 additions and 40 deletions

View file

@ -296,34 +296,50 @@ namespace i2p
UpdateRouterInfo (); UpdateRouterInfo ();
} }
void RouterContext::PublishNTCP2Address (std::shared_ptr<i2p::data::RouterInfo::Address> address,
int port, bool publish) const
{
if (!address) return;
if (!port && !address->port) port = SelectRandomPort ();
if (port) address->port = port;
address->published = publish;
memcpy (address->i, m_NTCP2Keys->iv, 16);
}
void RouterContext::PublishNTCP2Address (int port, bool publish, bool v4, bool v6, bool ygg) void RouterContext::PublishNTCP2Address (int port, bool publish, bool v4, bool v6, bool ygg)
{ {
if (!m_NTCP2Keys) return; if (!m_NTCP2Keys) return;
auto addresses = m_RouterInfo.GetAddresses (); auto addresses = m_RouterInfo.GetAddresses ();
if (!addresses) return; if (!addresses) return;
bool updated = false; bool updated = false;
for (auto& address : *addresses) if (v4)
{ {
if (address && address->IsNTCP2 () && (address->port != port || address->published != publish)) auto addr = (*addresses)[i2p::data::RouterInfo::eNTCP2V4Idx];
if (addr)
{ {
bool isAddr = v4 && address->IsV4 (); PublishNTCP2Address (addr, port, publish);
if (!isAddr && (v6 || ygg))
{
if (i2p::util::net::IsYggdrasilAddress (address->host))
isAddr = ygg;
else
isAddr = v6 && address->IsV6 ();
}
if (isAddr)
{
if (!port && !address->port) port = SelectRandomPort ();
if (port) address->port = port;
address->published = publish;
memcpy (address->i, m_NTCP2Keys->iv, 16);
updated = true; updated = true;
} }
} }
if (v6)
{
auto addr = (*addresses)[i2p::data::RouterInfo::eNTCP2V6Idx];
if (addr)
{
PublishNTCP2Address (addr, port, publish);
updated = true;
} }
}
if (ygg)
{
auto addr = (*addresses)[i2p::data::RouterInfo::eNTCP2V6MeshIdx];
if (addr)
{
PublishNTCP2Address (addr, port, publish);
updated = true;
}
}
if (updated) if (updated)
UpdateRouterInfo (); UpdateRouterInfo ();
} }
@ -916,30 +932,19 @@ namespace i2p
void RouterContext::UpdateNTCP2V6Address (const boost::asio::ip::address& host) void RouterContext::UpdateNTCP2V6Address (const boost::asio::ip::address& host)
{ {
bool isYgg = i2p::util::net::IsYggdrasilAddress (host);
bool updated = false;
auto addresses = m_RouterInfo.GetAddresses (); auto addresses = m_RouterInfo.GetAddresses ();
if (!addresses) return; if (!addresses) return;
for (auto& addr: *addresses) std::shared_ptr<i2p::data::RouterInfo::Address> addr;
{ if (i2p::util::net::IsYggdrasilAddress (host)) // yggdrasil
if (addr && addr->IsPublishedNTCP2 ()) addr = (*addresses)[i2p::data::RouterInfo::eNTCP2V6MeshIdx];
{ else if (host.is_v6 ())
bool isYgg1 = i2p::util::net::IsYggdrasilAddress (addr->host); addr = (*addresses)[i2p::data::RouterInfo::eNTCP2V6Idx];
if (addr->IsV6 () && ((isYgg && isYgg1) || (!isYgg && !isYgg1))) if (addr && addr->IsPublishedNTCP2 () && addr->host != host)
{
if (addr->host != host)
{ {
addr->host = host; addr->host = host;
updated = true;
}
break;
}
}
}
if (updated)
UpdateRouterInfo (); UpdateRouterInfo ();
} }
}
void RouterContext::UpdateStats () void RouterContext::UpdateStats ()
{ {

View file

@ -1,5 +1,5 @@
/* /*
* Copyright (c) 2013-2022, The PurpleI2P Project * Copyright (c) 2013-2023, The PurpleI2P Project
* *
* This file is part of Purple i2pd project and licensed under BSD3 * This file is part of Purple i2pd project and licensed under BSD3
* *
@ -181,6 +181,7 @@ namespace garlic
bool Load (); bool Load ();
void SaveKeys (); void SaveKeys ();
uint16_t SelectRandomPort () const; uint16_t SelectRandomPort () const;
void PublishNTCP2Address (std::shared_ptr<i2p::data::RouterInfo::Address> address, int port, bool publish) const;
bool DecryptECIESTunnelBuildRecord (const uint8_t * encrypted, uint8_t * data, size_t clearTextSize); bool DecryptECIESTunnelBuildRecord (const uint8_t * encrypted, uint8_t * data, size_t clearTextSize);