i2pd/RouterInfo.cpp

788 lines
21 KiB
C++
Raw Normal View History

2013-10-27 15:28:23 +00:00
#include <stdio.h>
#include <string.h>
#include "I2PEndian.h"
2013-10-27 15:28:23 +00:00
#include <fstream>
#include <boost/lexical_cast.hpp>
2016-01-16 20:36:30 +00:00
#include "version.h"
2015-11-03 14:15:49 +00:00
#include "Crypto.h"
#include "Base.h"
2013-10-27 15:28:23 +00:00
#include "Timestamp.h"
#include "Log.h"
#include "NetDb.h"
2013-10-27 15:28:23 +00:00
#include "RouterInfo.h"
2014-01-22 06:46:58 +00:00
2013-10-27 15:28:23 +00:00
namespace i2p
{
namespace data
{
2014-07-23 14:56:41 +00:00
RouterInfo::RouterInfo (const std::string& fullPath):
m_FullPath (fullPath), m_IsUpdated (false), m_IsUnreachable (false),
m_SupportedTransports (0), m_Caps (0)
2013-10-27 15:28:23 +00:00
{
m_Buffer = new uint8_t[MAX_RI_BUFFER_SIZE];
2014-07-23 14:56:41 +00:00
ReadFromFile ();
2013-10-27 15:28:23 +00:00
}
2013-11-20 12:46:09 +00:00
RouterInfo::RouterInfo (const uint8_t * buf, int len):
2014-03-19 16:02:51 +00:00
m_IsUpdated (true), m_IsUnreachable (false), m_SupportedTransports (0), m_Caps (0)
2013-10-27 15:28:23 +00:00
{
m_Buffer = new uint8_t[MAX_RI_BUFFER_SIZE];
2013-10-27 15:28:23 +00:00
memcpy (m_Buffer, buf, len);
m_BufferLen = len;
ReadFromBuffer (true);
2013-10-27 15:28:23 +00:00
}
2014-07-22 00:14:11 +00:00
RouterInfo::~RouterInfo ()
{
2015-04-07 20:26:35 +00:00
delete[] m_Buffer;
}
2014-07-22 00:14:11 +00:00
void RouterInfo::Update (const uint8_t * buf, int len)
{
2015-11-03 14:15:49 +00:00
// verify signature since we have indentity already
int l = len - m_RouterIdentity->GetSignatureLen ();
if (m_RouterIdentity->Verify (buf, l, buf + l))
{
// clean up
m_IsUpdated = true;
m_IsUnreachable = false;
m_SupportedTransports = 0;
m_Caps = 0;
m_Addresses.clear ();
m_Properties.clear ();
// copy buffer
if (!m_Buffer)
m_Buffer = new uint8_t[MAX_RI_BUFFER_SIZE];
memcpy (m_Buffer, buf, len);
m_BufferLen = len;
// skip identity
size_t identityLen = m_RouterIdentity->GetFullLen ();
// read new RI
std::stringstream str (std::string ((char *)m_Buffer + identityLen, m_BufferLen - identityLen));
ReadFromStream (str);
// don't delete buffer until saved to the file
}
else
{
2015-12-18 14:07:50 +00:00
LogPrint (eLogError, "RouterInfo: signature verification failed");
2015-11-03 14:15:49 +00:00
m_IsUnreachable = true;
}
2014-07-22 00:14:11 +00:00
}
2015-11-03 14:15:49 +00:00
void RouterInfo::SetRouterIdentity (std::shared_ptr<const IdentityEx> identity)
2013-10-27 15:28:23 +00:00
{
m_RouterIdentity = identity;
m_Timestamp = i2p::util::GetMillisecondsSinceEpoch ();
}
2014-07-23 14:56:41 +00:00
bool RouterInfo::LoadFile ()
2013-10-27 15:28:23 +00:00
{
2016-03-31 00:00:00 +00:00
std::ifstream s(m_FullPath, std::ifstream::binary);
2013-10-27 15:28:23 +00:00
if (s.is_open ())
{
s.seekg (0,std::ios::end);
2014-02-02 22:22:00 +00:00
m_BufferLen = s.tellg ();
2016-02-02 16:55:38 +00:00
if (m_BufferLen < 40 || m_BufferLen > MAX_RI_BUFFER_SIZE)
2014-02-02 22:22:00 +00:00
{
2015-12-18 14:07:50 +00:00
LogPrint(eLogError, "RouterInfo: File", m_FullPath, " is malformed");
2014-07-23 14:56:41 +00:00
return false;
2014-02-02 22:22:00 +00:00
}
2013-10-27 15:28:23 +00:00
s.seekg(0, std::ios::beg);
2014-07-23 14:56:41 +00:00
if (!m_Buffer)
m_Buffer = new uint8_t[MAX_RI_BUFFER_SIZE];
2014-07-10 19:33:42 +00:00
s.read((char *)m_Buffer, m_BufferLen);
2013-10-27 15:28:23 +00:00
}
else
2014-07-23 14:56:41 +00:00
{
2015-12-18 14:07:50 +00:00
LogPrint (eLogError, "RouterInfo: Can't open file ", m_FullPath);
2014-07-23 14:56:41 +00:00
return false;
}
return true;
}
void RouterInfo::ReadFromFile ()
{
if (LoadFile ())
ReadFromBuffer (false);
else
m_IsUnreachable = true;
2013-10-27 15:28:23 +00:00
}
void RouterInfo::ReadFromBuffer (bool verifySignature)
2013-10-27 15:28:23 +00:00
{
2015-11-03 14:15:49 +00:00
m_RouterIdentity = std::make_shared<IdentityEx>(m_Buffer, m_BufferLen);
size_t identityLen = m_RouterIdentity->GetFullLen ();
2016-02-02 16:55:38 +00:00
if (identityLen >= m_BufferLen)
{
LogPrint (eLogError, "RouterInfo: identity length ", identityLen, " exceeds buffer size ", m_BufferLen);
m_IsUnreachable = true;
return;
}
2014-11-19 21:21:40 +00:00
std::stringstream str (std::string ((char *)m_Buffer + identityLen, m_BufferLen - identityLen));
2013-10-27 15:28:23 +00:00
ReadFromStream (str);
2016-02-02 16:55:38 +00:00
if (!str)
{
LogPrint (eLogError, "RouterInfo: malformed message");
m_IsUnreachable = true;
return;
}
if (verifySignature)
2013-10-27 15:28:23 +00:00
{
// verify signature
2016-02-02 16:55:38 +00:00
int l = m_BufferLen - m_RouterIdentity->GetSignatureLen ();
if (l < 0 || !m_RouterIdentity->Verify ((uint8_t *)m_Buffer, l, (uint8_t *)m_Buffer + l))
2014-11-26 15:28:06 +00:00
{
2015-12-18 14:07:50 +00:00
LogPrint (eLogError, "RouterInfo: signature verification failed");
2014-11-26 15:28:06 +00:00
m_IsUnreachable = true;
}
2015-11-03 14:15:49 +00:00
m_RouterIdentity->DropVerifier ();
2013-10-27 15:28:23 +00:00
}
}
void RouterInfo::ReadFromStream (std::istream& s)
{
s.read ((char *)&m_Timestamp, sizeof (m_Timestamp));
m_Timestamp = be64toh (m_Timestamp);
// read addresses
uint8_t numAddresses;
2016-02-02 16:55:38 +00:00
s.read ((char *)&numAddresses, sizeof (numAddresses)); if (!s) return;
2014-07-25 00:47:12 +00:00
bool introducers = false;
2013-10-27 15:28:23 +00:00
for (int i = 0; i < numAddresses; i++)
{
uint8_t supportedTransports = 0;
2014-07-25 00:47:12 +00:00
bool isValidAddress = true;
2013-10-27 15:28:23 +00:00
Address address;
s.read ((char *)&address.cost, sizeof (address.cost));
s.read ((char *)&address.date, sizeof (address.date));
char transportStyle[5];
ReadString (transportStyle, s);
if (!strcmp (transportStyle, "NTCP"))
address.transportStyle = eTransportNTCP;
else if (!strcmp (transportStyle, "SSU"))
address.transportStyle = eTransportSSU;
else
address.transportStyle = eTransportUnknown;
2014-09-14 21:57:47 +00:00
address.port = 0;
address.mtu = 0;
2013-10-27 15:28:23 +00:00
uint16_t size, r = 0;
2016-02-02 16:55:38 +00:00
s.read ((char *)&size, sizeof (size)); if (!s) return;
2013-10-27 15:28:23 +00:00
size = be16toh (size);
while (r < size)
{
2014-01-22 02:02:22 +00:00
char key[500], value[500];
2013-10-27 15:28:23 +00:00
r += ReadString (key, s);
s.seekg (1, std::ios_base::cur); r++; // =
r += ReadString (value, s);
s.seekg (1, std::ios_base::cur); r++; // ;
if (!strcmp (key, "host"))
2014-01-21 23:01:11 +00:00
{
boost::system::error_code ecode;
address.host = boost::asio::ip::address::from_string (value, ecode);
if (ecode)
{
2015-01-17 04:01:40 +00:00
if (address.transportStyle == eTransportNTCP)
{
supportedTransports |= eNTCPV4; // TODO:
2015-01-17 04:01:40 +00:00
address.addressString = value;
}
else
{
supportedTransports |= eSSUV4; // TODO:
2015-12-03 20:45:01 +00:00
address.addressString = value;
2015-01-17 04:01:40 +00:00
}
}
else
{
// add supported protocol
if (address.host.is_v4 ())
supportedTransports |= (address.transportStyle == eTransportNTCP) ? eNTCPV4 : eSSUV4;
else
supportedTransports |= (address.transportStyle == eTransportNTCP) ? eNTCPV6 : eSSUV6;
}
2014-01-21 23:01:11 +00:00
}
2013-10-27 15:28:23 +00:00
else if (!strcmp (key, "port"))
address.port = boost::lexical_cast<int>(value);
2014-09-14 21:57:47 +00:00
else if (!strcmp (key, "mtu"))
address.mtu = boost::lexical_cast<int>(value);
2014-01-28 00:48:46 +00:00
else if (!strcmp (key, "key"))
Base64ToByteStream (value, strlen (value), address.key, 32);
2014-04-09 01:56:34 +00:00
else if (!strcmp (key, "caps"))
ExtractCaps (value);
2014-02-20 21:15:12 +00:00
else if (key[0] == 'i')
{
// introducers
2014-07-25 00:47:12 +00:00
introducers = true;
2014-02-20 21:15:12 +00:00
size_t l = strlen(key);
2014-02-24 01:48:28 +00:00
unsigned char index = key[l-1] - '0'; // TODO:
2014-02-20 21:15:12 +00:00
key[l-1] = 0;
if (index >= address.introducers.size ())
address.introducers.resize (index + 1);
Introducer& introducer = address.introducers.at (index);
if (!strcmp (key, "ihost"))
{
boost::system::error_code ecode;
introducer.iHost = boost::asio::ip::address::from_string (value, ecode);
}
else if (!strcmp (key, "iport"))
introducer.iPort = boost::lexical_cast<int>(value);
else if (!strcmp (key, "itag"))
introducer.iTag = boost::lexical_cast<uint32_t>(value);
else if (!strcmp (key, "ikey"))
Base64ToByteStream (value, strlen (value), introducer.iKey, 32);
}
2016-02-02 16:55:38 +00:00
if (!s) return;
2013-10-27 15:28:23 +00:00
}
2014-07-25 00:47:12 +00:00
if (isValidAddress)
{
2016-03-21 17:02:51 +00:00
m_Addresses.push_back(std::make_shared<Address>(address));
m_SupportedTransports |= supportedTransports;
}
2013-10-27 15:28:23 +00:00
}
// read peers
uint8_t numPeers;
2016-02-02 16:55:38 +00:00
s.read ((char *)&numPeers, sizeof (numPeers)); if (!s) return;
2013-10-27 15:28:23 +00:00
s.seekg (numPeers*32, std::ios_base::cur); // TODO: read peers
// read properties
uint16_t size, r = 0;
2016-02-02 16:55:38 +00:00
s.read ((char *)&size, sizeof (size)); if (!s) return;
2013-10-27 15:28:23 +00:00
size = be16toh (size);
while (r < size)
{
2014-01-22 12:51:55 +00:00
#ifdef _WIN32
2014-01-22 02:02:22 +00:00
char key[500], value[500];
2014-01-22 12:51:55 +00:00
// TODO: investigate why properties get read as one long string under Windows
// length should not be more than 44
#else
char key[50], value[50];
#endif
2013-10-27 15:28:23 +00:00
r += ReadString (key, s);
s.seekg (1, std::ios_base::cur); r++; // =
r += ReadString (value, s);
s.seekg (1, std::ios_base::cur); r++; // ;
m_Properties[key] = value;
2014-03-19 16:02:51 +00:00
// extract caps
2014-03-19 21:27:37 +00:00
if (!strcmp (key, "caps"))
2014-03-19 16:02:51 +00:00
ExtractCaps (value);
2016-01-16 20:36:30 +00:00
// check netId
2016-02-21 01:20:19 +00:00
else if (!strcmp (key, ROUTER_INFO_PROPERTY_NETID) && atoi (value) != I2PD_NET_ID)
2016-01-16 20:36:30 +00:00
{
2016-02-21 01:20:19 +00:00
LogPrint (eLogError, "Unexpected ", ROUTER_INFO_PROPERTY_NETID, "=", value);
2016-01-16 20:36:30 +00:00
m_IsUnreachable = true;
}
// family
2016-02-21 01:20:19 +00:00
else if (!strcmp (key, ROUTER_INFO_PROPERTY_FAMILY))
{
m_Family = value;
boost::to_lower (m_Family);
}
2016-02-21 01:20:19 +00:00
else if (!strcmp (key, ROUTER_INFO_PROPERTY_FAMILY_SIG))
{
if (!netdb.GetFamilies ().VerifyFamily (m_Family, GetIdentHash (), value))
{
LogPrint (eLogWarning, "RouterInfo: family signature verification failed");
m_Family.clear ();
}
}
2016-02-02 16:55:38 +00:00
if (!s) return;
2013-10-27 15:28:23 +00:00
}
2014-07-25 00:47:12 +00:00
if (!m_SupportedTransports || !m_Addresses.size() || (UsesIntroducer () && !introducers))
SetUnreachable (true);
2013-12-29 15:48:57 +00:00
}
2014-03-19 16:02:51 +00:00
void RouterInfo::ExtractCaps (const char * value)
{
const char * cap = value;
while (*cap)
{
switch (*cap)
{
2014-09-02 20:11:31 +00:00
case CAPS_FLAG_FLOODFILL:
2014-03-19 16:02:51 +00:00
m_Caps |= Caps::eFloodfill;
break;
2014-09-02 20:11:31 +00:00
case CAPS_FLAG_HIGH_BANDWIDTH1:
case CAPS_FLAG_HIGH_BANDWIDTH2:
case CAPS_FLAG_HIGH_BANDWIDTH3:
2014-03-19 19:58:57 +00:00
m_Caps |= Caps::eHighBandwidth;
2014-03-19 16:02:51 +00:00
break;
2016-01-03 03:17:04 +00:00
case CAPS_FLAG_EXTRA_BANDWIDTH1:
case CAPS_FLAG_EXTRA_BANDWIDTH2:
m_Caps |= Caps::eExtraBandwidth;
break;
2014-09-02 20:11:31 +00:00
case CAPS_FLAG_HIDDEN:
m_Caps |= Caps::eHidden;
break;
case CAPS_FLAG_REACHABLE:
2014-03-19 16:02:51 +00:00
m_Caps |= Caps::eReachable;
break;
2014-09-02 20:11:31 +00:00
case CAPS_FLAG_UNREACHABLE:
m_Caps |= Caps::eUnreachable;
break;
case CAPS_FLAG_SSU_TESTING:
2014-04-09 01:56:34 +00:00
m_Caps |= Caps::eSSUTesting;
break;
2014-09-02 20:11:31 +00:00
case CAPS_FLAG_SSU_INTRODUCER:
2014-04-09 01:56:34 +00:00
m_Caps |= Caps::eSSUIntroducer;
2014-06-18 14:41:59 +00:00
break;
2014-03-19 16:02:51 +00:00
default: ;
}
cap++;
}
}
2014-09-02 20:11:31 +00:00
void RouterInfo::UpdateCapsProperty ()
{
std::string caps;
if (m_Caps & eFloodfill) {
2015-01-28 20:12:15 +00:00
caps += CAPS_FLAG_FLOODFILL; // floodfill
caps += (m_Caps & eExtraBandwidth)
? CAPS_FLAG_EXTRA_BANDWIDTH1 // 'P'
: CAPS_FLAG_HIGH_BANDWIDTH3; // 'O'
} else {
if (m_Caps & eExtraBandwidth) {
caps += CAPS_FLAG_EXTRA_BANDWIDTH1; // 'P'
} else if (m_Caps & eHighBandwidth) {
caps += CAPS_FLAG_HIGH_BANDWIDTH3; // 'O'
} else {
caps += CAPS_FLAG_LOW_BANDWIDTH2; // 'L'
}
2016-01-03 03:17:04 +00:00
}
2014-09-02 20:11:31 +00:00
if (m_Caps & eHidden) caps += CAPS_FLAG_HIDDEN; // hidden
if (m_Caps & eReachable) caps += CAPS_FLAG_REACHABLE; // reachable
if (m_Caps & eUnreachable) caps += CAPS_FLAG_UNREACHABLE; // unreachable
2016-06-01 00:00:00 +00:00
SetProperty ("caps", caps);
2014-09-02 20:11:31 +00:00
}
2013-12-29 15:48:57 +00:00
2013-10-27 15:28:23 +00:00
void RouterInfo::WriteToStream (std::ostream& s)
{
uint64_t ts = htobe64 (m_Timestamp);
s.write ((char *)&ts, sizeof (ts));
// addresses
uint8_t numAddresses = m_Addresses.size ();
s.write ((char *)&numAddresses, sizeof (numAddresses));
2016-03-21 17:02:51 +00:00
for (auto addr : m_Addresses)
2013-10-27 15:28:23 +00:00
{
2016-03-21 17:02:51 +00:00
Address& address = *addr;
2013-10-27 15:28:23 +00:00
s.write ((char *)&address.cost, sizeof (address.cost));
s.write ((char *)&address.date, sizeof (address.date));
2014-02-23 16:48:09 +00:00
std::stringstream properties;
2013-10-27 15:28:23 +00:00
if (address.transportStyle == eTransportNTCP)
WriteString ("NTCP", s);
else if (address.transportStyle == eTransportSSU)
2014-02-23 16:48:09 +00:00
{
2013-10-27 15:28:23 +00:00
WriteString ("SSU", s);
2014-02-24 01:48:28 +00:00
// caps
WriteString ("caps", properties);
2014-02-23 16:48:09 +00:00
properties << '=';
2014-04-09 01:56:34 +00:00
std::string caps;
2014-09-02 20:11:31 +00:00
if (IsPeerTesting ()) caps += CAPS_FLAG_SSU_TESTING;
if (IsIntroducer ()) caps += CAPS_FLAG_SSU_INTRODUCER;
2014-04-09 01:56:34 +00:00
WriteString (caps, properties);
2014-02-23 16:48:09 +00:00
properties << ';';
}
2013-10-27 15:28:23 +00:00
else
WriteString ("", s);
WriteString ("host", properties);
properties << '=';
2014-01-21 21:07:16 +00:00
WriteString (address.host.to_string (), properties);
2013-10-27 15:28:23 +00:00
properties << ';';
2014-02-24 01:48:28 +00:00
if (address.transportStyle == eTransportSSU)
{
2014-09-01 21:34:20 +00:00
// write introducers if any
if (address.introducers.size () > 0)
{
int i = 0;
for (auto introducer: address.introducers)
{
WriteString ("ihost" + boost::lexical_cast<std::string>(i), properties);
properties << '=';
WriteString (introducer.iHost.to_string (), properties);
properties << ';';
i++;
}
i = 0;
for (auto introducer: address.introducers)
{
WriteString ("ikey" + boost::lexical_cast<std::string>(i), properties);
properties << '=';
char value[64];
size_t l = ByteStreamToBase64 (introducer.iKey, 32, value, 64);
value[l] = 0;
WriteString (value, properties);
properties << ';';
i++;
}
i = 0;
for (auto introducer: address.introducers)
{
WriteString ("iport" + boost::lexical_cast<std::string>(i), properties);
properties << '=';
WriteString (boost::lexical_cast<std::string>(introducer.iPort), properties);
properties << ';';
i++;
}
i = 0;
for (auto introducer: address.introducers)
{
WriteString ("itag" + boost::lexical_cast<std::string>(i), properties);
properties << '=';
WriteString (boost::lexical_cast<std::string>(introducer.iTag), properties);
properties << ';';
i++;
}
}
// write intro key
2014-02-24 01:48:28 +00:00
WriteString ("key", properties);
properties << '=';
char value[64];
size_t l = ByteStreamToBase64 (address.key, 32, value, 64);
value[l] = 0;
WriteString (value, properties);
properties << ';';
2014-10-30 14:07:39 +00:00
// write mtu
if (address.mtu)
{
WriteString ("mtu", properties);
properties << '=';
WriteString (boost::lexical_cast<std::string>(address.mtu), properties);
properties << ';';
}
2014-02-24 01:48:28 +00:00
}
2013-10-27 15:28:23 +00:00
WriteString ("port", properties);
properties << '=';
WriteString (boost::lexical_cast<std::string>(address.port), properties);
properties << ';';
uint16_t size = htobe16 (properties.str ().size ());
s.write ((char *)&size, sizeof (size));
s.write (properties.str ().c_str (), properties.str ().size ());
}
// peers
uint8_t numPeers = 0;
s.write ((char *)&numPeers, sizeof (numPeers));
// properties
std::stringstream properties;
for (auto& p : m_Properties)
{
WriteString (p.first, properties);
properties << '=';
WriteString (p.second, properties);
properties << ';';
}
uint16_t size = htobe16 (properties.str ().size ());
s.write ((char *)&size, sizeof (size));
s.write (properties.str ().c_str (), properties.str ().size ());
}
2016-02-17 20:36:55 +00:00
bool RouterInfo::IsNewer (const uint8_t * buf, size_t len) const
{
if (!m_RouterIdentity) return false;
size_t size = m_RouterIdentity->GetFullLen ();
if (size + 8 > len) return false;
return bufbe64toh (buf + size) > m_Timestamp;
}
const uint8_t * RouterInfo::LoadBuffer ()
2014-07-23 14:56:41 +00:00
{
if (!m_Buffer)
{
if (LoadFile ())
2015-12-18 14:07:50 +00:00
LogPrint (eLogDebug, "RouterInfo: Buffer for ", GetIdentHashAbbreviation (GetIdentHash ()), " loaded from file");
2014-07-23 14:56:41 +00:00
}
return m_Buffer;
}
2014-08-26 02:47:12 +00:00
void RouterInfo::CreateBuffer (const PrivateKeys& privateKeys)
2013-10-27 15:28:23 +00:00
{
2013-11-20 12:46:09 +00:00
m_Timestamp = i2p::util::GetMillisecondsSinceEpoch (); // refresh timstamp
2013-10-27 15:28:23 +00:00
std::stringstream s;
uint8_t ident[1024];
2015-11-03 14:15:49 +00:00
auto identLen = privateKeys.GetPublic ()->ToBuffer (ident, 1024);
s.write ((char *)ident, identLen);
2013-10-27 15:28:23 +00:00
WriteToStream (s);
m_BufferLen = s.str ().size ();
2014-07-22 12:03:02 +00:00
if (!m_Buffer)
m_Buffer = new uint8_t[MAX_RI_BUFFER_SIZE];
2013-10-27 15:28:23 +00:00
memcpy (m_Buffer, s.str ().c_str (), m_BufferLen);
// signature
2014-08-26 02:47:12 +00:00
privateKeys.Sign ((uint8_t *)m_Buffer, m_BufferLen, (uint8_t *)m_Buffer + m_BufferLen);
2015-11-03 14:15:49 +00:00
m_BufferLen += privateKeys.GetPublic ()->GetSignatureLen ();
2013-10-27 15:28:23 +00:00
}
2014-07-10 19:33:42 +00:00
bool RouterInfo::SaveToFile (const std::string& fullPath)
2014-07-10 19:33:42 +00:00
{
2014-07-23 14:56:41 +00:00
m_FullPath = fullPath;
if (!m_Buffer) {
2015-12-18 14:07:50 +00:00
LogPrint (eLogError, "RouterInfo: Can't save, m_Buffer == NULL");
return false;
}
std::ofstream f (fullPath, std::ofstream::binary | std::ofstream::out);
if (!f.is_open ()) {
LogPrint(eLogError, "RouterInfo: Can't save to ", fullPath);
return false;
}
f.write ((char *)m_Buffer, m_BufferLen);
return true;
2014-07-10 19:33:42 +00:00
}
2013-10-27 15:28:23 +00:00
size_t RouterInfo::ReadString (char * str, std::istream& s)
{
uint8_t len;
s.read ((char *)&len, 1);
s.read (str, len);
str[len] = 0;
return len+1;
}
void RouterInfo::WriteString (const std::string& str, std::ostream& s)
{
uint8_t len = str.size ();
s.write ((char *)&len, 1);
s.write (str.c_str (), len);
}
void RouterInfo::AddNTCPAddress (const char * host, int port)
{
2016-03-21 17:02:51 +00:00
auto addr = std::make_shared<Address>();
addr->host = boost::asio::ip::address::from_string (host);
addr->port = port;
addr->transportStyle = eTransportNTCP;
addr->cost = 2;
addr->date = 0;
addr->mtu = 0;
2015-11-03 14:15:49 +00:00
for (auto it: m_Addresses) // don't insert same address twice
2016-03-22 11:30:16 +00:00
if (*it == *addr) return;
2013-10-27 15:28:23 +00:00
m_Addresses.push_back(addr);
2016-03-21 17:02:51 +00:00
m_SupportedTransports |= addr->host.is_v6 () ? eNTCPV6 : eNTCPV4;
2013-10-27 15:28:23 +00:00
}
2014-10-30 14:07:39 +00:00
void RouterInfo::AddSSUAddress (const char * host, int port, const uint8_t * key, int mtu)
2014-02-23 16:48:09 +00:00
{
2016-03-21 17:02:51 +00:00
auto addr = std::make_shared<Address>();
addr->host = boost::asio::ip::address::from_string (host);
addr->port = port;
addr->transportStyle = eTransportSSU;
addr->cost = 10; // NTCP should have priority over SSU
addr->date = 0;
addr->mtu = mtu;
memcpy (addr->key, key, 32);
2015-11-03 14:15:49 +00:00
for (auto it: m_Addresses) // don't insert same address twice
2016-03-22 11:30:16 +00:00
if (*it == *addr) return;
2014-02-23 16:48:09 +00:00
m_Addresses.push_back(addr);
2016-03-21 17:02:51 +00:00
m_SupportedTransports |= addr->host.is_v6 () ? eSSUV6 : eSSUV4;
2014-04-16 22:28:44 +00:00
m_Caps |= eSSUTesting;
m_Caps |= eSSUIntroducer;
2014-02-23 16:48:09 +00:00
}
2014-09-01 21:34:20 +00:00
2015-11-03 14:15:49 +00:00
bool RouterInfo::AddIntroducer (const Introducer& introducer)
2014-09-01 21:34:20 +00:00
{
2016-03-21 17:02:51 +00:00
for (auto addr : m_Addresses)
2014-09-01 21:34:20 +00:00
{
2016-03-21 17:02:51 +00:00
if (addr->transportStyle == eTransportSSU && addr->host.is_v4 ())
2014-09-01 21:34:20 +00:00
{
2016-03-21 17:02:51 +00:00
for (auto intro: addr->introducers)
2015-11-03 14:15:49 +00:00
if (intro.iTag == introducer.iTag) return false; // already presented
2016-03-21 17:02:51 +00:00
addr->introducers.push_back (introducer);
2014-09-01 21:34:20 +00:00
return true;
}
}
return false;
}
2014-09-07 00:43:20 +00:00
bool RouterInfo::RemoveIntroducer (const boost::asio::ip::udp::endpoint& e)
2014-09-01 21:34:20 +00:00
{
2016-03-21 17:02:51 +00:00
for (auto addr: m_Addresses)
2014-09-01 21:34:20 +00:00
{
2016-03-21 17:02:51 +00:00
if (addr->transportStyle == eTransportSSU && addr->host.is_v4 ())
2014-09-01 21:34:20 +00:00
{
2016-03-21 17:02:51 +00:00
for (std::vector<Introducer>::iterator it = addr->introducers.begin (); it != addr->introducers.end (); it++)
2014-09-07 00:43:20 +00:00
if ( boost::asio::ip::udp::endpoint (it->iHost, it->iPort) == e)
2014-09-01 21:34:20 +00:00
{
2016-03-21 17:02:51 +00:00
addr->introducers.erase (it);
2014-09-01 21:34:20 +00:00
return true;
}
}
}
return false;
}
2014-09-02 20:11:31 +00:00
void RouterInfo::SetCaps (uint8_t caps)
{
m_Caps = caps;
UpdateCapsProperty ();
}
2014-09-01 21:34:20 +00:00
void RouterInfo::SetCaps (const char * caps)
{
SetProperty ("caps", caps);
m_Caps = 0;
ExtractCaps (caps);
}
2014-02-23 16:48:09 +00:00
void RouterInfo::SetProperty (const std::string& key, const std::string& value)
2013-10-27 15:28:23 +00:00
{
m_Properties[key] = value;
}
void RouterInfo::DeleteProperty (const std::string& key)
{
m_Properties.erase (key);
}
std::string RouterInfo::GetProperty (const std::string& key) const
{
auto it = m_Properties.find (key);
if (it != m_Properties.end ())
return it->second;
return "";
}
2014-01-22 00:14:30 +00:00
bool RouterInfo::IsNTCP (bool v4only) const
2013-10-27 15:28:23 +00:00
{
if (v4only)
return m_SupportedTransports & eNTCPV4;
else
return m_SupportedTransports & (eNTCPV4 | eNTCPV6);
}
2014-02-09 02:06:40 +00:00
bool RouterInfo::IsSSU (bool v4only) const
{
if (v4only)
return m_SupportedTransports & eSSUV4;
else
return m_SupportedTransports & (eSSUV4 | eSSUV6);
}
2014-02-21 21:13:36 +00:00
bool RouterInfo::IsV6 () const
{
return m_SupportedTransports & (eNTCPV6 | eSSUV6);
}
2016-03-24 22:44:41 +00:00
bool RouterInfo::IsV4 () const
{
return m_SupportedTransports & (eNTCPV4 | eSSUV4);
}
void RouterInfo::EnableV6 ()
{
if (!IsV6 ())
2014-10-29 22:46:35 +00:00
m_SupportedTransports |= eNTCPV6 | eSSUV6;
}
2016-03-24 22:44:41 +00:00
void RouterInfo::EnableV4 ()
{
if (!IsV4 ())
m_SupportedTransports |= eNTCPV4 | eSSUV4;
}
void RouterInfo::DisableV6 ()
{
if (IsV6 ())
{
2014-10-29 20:08:01 +00:00
// NTCP
m_SupportedTransports &= ~eNTCPV6;
for (size_t i = 0; i < m_Addresses.size (); i++)
{
2016-03-21 17:02:51 +00:00
if (m_Addresses[i]->transportStyle == i2p::data::RouterInfo::eTransportNTCP &&
m_Addresses[i]->host.is_v6 ())
{
m_Addresses.erase (m_Addresses.begin () + i);
break;
}
}
2014-10-29 20:08:01 +00:00
// SSU
m_SupportedTransports &= ~eSSUV6;
for (size_t i = 0; i < m_Addresses.size (); i++)
{
2016-03-21 17:02:51 +00:00
if (m_Addresses[i]->transportStyle == i2p::data::RouterInfo::eTransportSSU &&
m_Addresses[i]->host.is_v6 ())
2014-10-29 20:08:01 +00:00
{
m_Addresses.erase (m_Addresses.begin () + i);
break;
}
}
}
}
2016-03-24 22:44:41 +00:00
void RouterInfo::DisableV4 ()
{
if (IsV4 ())
{
// NTCP
m_SupportedTransports &= ~eNTCPV4;
for (size_t i = 0; i < m_Addresses.size (); i++)
{
if (m_Addresses[i]->transportStyle == i2p::data::RouterInfo::eTransportNTCP &&
m_Addresses[i]->host.is_v4 ())
{
m_Addresses.erase (m_Addresses.begin () + i);
break;
}
}
// SSU
m_SupportedTransports &= ~eSSUV4;
for (size_t i = 0; i < m_Addresses.size (); i++)
{
if (m_Addresses[i]->transportStyle == i2p::data::RouterInfo::eTransportSSU &&
m_Addresses[i]->host.is_v4 ())
{
m_Addresses.erase (m_Addresses.begin () + i);
break;
}
}
}
}
2014-02-21 21:13:36 +00:00
bool RouterInfo::UsesIntroducer () const
{
2014-08-19 15:01:11 +00:00
return m_Caps & Caps::eUnreachable; // non-reachable
2014-02-21 21:13:36 +00:00
}
2016-03-21 17:02:51 +00:00
std::shared_ptr<const RouterInfo::Address> RouterInfo::GetNTCPAddress (bool v4only) const
2014-01-28 00:48:46 +00:00
{
return GetAddress (eTransportNTCP, v4only);
}
2016-03-21 17:02:51 +00:00
std::shared_ptr<const RouterInfo::Address> RouterInfo::GetSSUAddress (bool v4only) const
2014-01-28 00:48:46 +00:00
{
return GetAddress (eTransportSSU, v4only);
}
2016-03-21 17:02:51 +00:00
std::shared_ptr<const RouterInfo::Address> RouterInfo::GetSSUV6Address () const
{
return GetAddress (eTransportSSU, false, true);
}
2016-03-21 17:02:51 +00:00
std::shared_ptr<const RouterInfo::Address> RouterInfo::GetAddress (TransportStyle s, bool v4only, bool v6only) const
2013-10-27 15:28:23 +00:00
{
2016-03-21 17:02:51 +00:00
for (auto address : m_Addresses)
2013-10-27 15:28:23 +00:00
{
2016-03-21 17:02:51 +00:00
if (address->transportStyle == s)
2014-01-22 00:14:30 +00:00
{
2016-03-21 17:02:51 +00:00
if ((!v4only || address->host.is_v4 ()) && (!v6only || address->host.is_v6 ()))
return address;
2014-01-22 00:14:30 +00:00
}
2013-10-27 15:28:23 +00:00
}
return nullptr;
}
std::shared_ptr<RouterProfile> RouterInfo::GetProfile () const
{
if (!m_Profile)
m_Profile = GetRouterProfile (GetIdentHash ());
return m_Profile;
}
2013-10-27 15:28:23 +00:00
}
}