i2pd/libi2pd/RouterContext.h

176 lines
6.5 KiB
C
Raw Normal View History

/*
* Copyright (c) 2013-2020, The PurpleI2P Project
*
* This file is part of Purple i2pd project and licensed under BSD3
*
* See full license text in LICENSE file at top of project tree
*/
2013-10-23 06:45:40 +04:00
#ifndef ROUTER_CONTEXT_H__
#define ROUTER_CONTEXT_H__
#include <inttypes.h>
2014-10-27 22:08:50 +03:00
#include <string>
2014-11-20 23:48:28 +03:00
#include <memory>
2015-06-10 05:14:31 +03:00
#include <mutex>
2019-06-19 18:43:04 +03:00
#include <chrono>
2014-10-29 20:49:21 +03:00
#include <boost/asio.hpp>
2014-04-01 21:42:04 +04:00
#include "Identity.h"
2013-10-23 06:45:40 +04:00
#include "RouterInfo.h"
2014-10-07 04:18:18 +04:00
#include "Garlic.h"
2013-10-23 06:45:40 +04:00
namespace i2p
{
const char ROUTER_INFO[] = "router.info";
2018-01-06 06:48:51 +03:00
const char ROUTER_KEYS[] = "router.keys";
const char NTCP2_KEYS[] = "ntcp2.keys";
const int ROUTER_INFO_UPDATE_INTERVAL = 1800; // 30 minutes
2015-02-26 21:44:18 +03:00
enum RouterStatus
{
eRouterStatusOK = 0,
eRouterStatusTesting = 1,
2016-09-19 01:42:21 +03:00
eRouterStatusFirewalled = 2,
eRouterStatusError = 3,
eRouterStatusUnknown = 4
2018-01-06 06:48:51 +03:00
};
2015-02-26 21:44:18 +03:00
2016-09-20 04:37:04 +03:00
enum RouterError
{
eRouterErrorNone = 0,
2020-10-12 00:51:40 +03:00
eRouterErrorClockSkew = 1,
eRouterErrorOffline = 2
2018-01-06 06:48:51 +03:00
};
class RouterContext: public i2p::garlic::GarlicDestination
2013-10-23 06:45:40 +04:00
{
2018-06-11 22:33:48 +03:00
private:
struct NTCP2PrivateKeys
2018-06-11 22:33:48 +03:00
{
2018-06-15 19:52:43 +03:00
uint8_t staticPublicKey[32];
uint8_t staticPrivateKey[32];
2018-06-11 22:33:48 +03:00
uint8_t iv[16];
};
2018-06-11 22:33:48 +03:00
2013-10-23 06:45:40 +04:00
public:
RouterContext ();
2014-09-04 17:31:42 +04:00
void Init ();
2013-10-23 06:45:40 +04:00
const i2p::data::PrivateKeys& GetPrivateKeys () const { return m_Keys; };
2013-10-23 06:45:40 +04:00
i2p::data::RouterInfo& GetRouterInfo () { return m_RouterInfo; };
2018-01-06 06:48:51 +03:00
std::shared_ptr<const i2p::data::RouterInfo> GetSharedRouterInfo () const
{
return std::shared_ptr<const i2p::data::RouterInfo> (&m_RouterInfo,
2014-11-20 23:48:28 +03:00
[](const i2p::data::RouterInfo *) {});
}
2018-01-06 06:48:51 +03:00
std::shared_ptr<i2p::garlic::GarlicDestination> GetSharedDestination ()
{
2018-01-06 06:48:51 +03:00
return std::shared_ptr<i2p::garlic::GarlicDestination> (this,
[](i2p::garlic::GarlicDestination *) {});
2018-01-06 06:48:51 +03:00
}
2018-06-15 19:52:43 +03:00
const uint8_t * GetNTCP2StaticPublicKey () const { return m_NTCP2Keys ? m_NTCP2Keys->staticPublicKey : nullptr; };
const uint8_t * GetNTCP2StaticPrivateKey () const { return m_NTCP2Keys ? m_NTCP2Keys->staticPrivateKey : nullptr; };
const uint8_t * GetNTCP2IV () const { return m_NTCP2Keys ? m_NTCP2Keys->iv : nullptr; };
i2p::crypto::X25519Keys& GetStaticKeys ();
2018-01-06 06:48:51 +03:00
2019-06-19 18:43:04 +03:00
uint32_t GetUptime () const; // in seconds
uint64_t GetLastUpdateTime () const { return m_LastUpdateTime; };
uint64_t GetBandwidthLimit () const { return m_BandwidthLimit; };
uint64_t GetTransitBandwidthLimit () const { return (m_BandwidthLimit*m_ShareRatio)/100LL; };
2015-02-26 21:44:18 +03:00
RouterStatus GetStatus () const { return m_Status; };
2015-11-03 17:15:49 +03:00
void SetStatus (RouterStatus status);
2016-09-20 04:37:04 +03:00
RouterError GetError () const { return m_Error; };
void SetError (RouterError error) { m_Status = eRouterStatusError; m_Error = error; };
2016-10-12 18:26:48 +03:00
int GetNetID () const { return m_NetID; };
2018-01-06 06:48:51 +03:00
void SetNetID (int netID) { m_NetID = netID; };
bool DecryptTunnelBuildRecord (const uint8_t * encrypted, uint8_t * data, BN_CTX * ctx);
2016-10-12 18:26:48 +03:00
void UpdatePort (int port); // called from Daemon
2014-10-29 20:49:21 +03:00
void UpdateAddress (const boost::asio::ip::address& host); // called from SSU or Daemon
void PublishNTCP2Address (int port, bool publish = true, bool v4only = false);
2018-08-03 21:49:28 +03:00
void UpdateNTCP2Address (bool enable);
2020-10-04 05:29:52 +03:00
void RemoveNTCPAddress (bool v4only = true); // delete NTCP address for older routers. TODO: remove later
2015-11-03 17:15:49 +03:00
bool AddIntroducer (const i2p::data::RouterInfo::Introducer& introducer);
2014-09-07 04:43:20 +04:00
void RemoveIntroducer (const boost::asio::ip::udp::endpoint& e);
bool IsUnreachable () const;
2018-01-06 06:48:51 +03:00
void SetUnreachable ();
void SetReachable ();
2018-01-06 06:48:51 +03:00
bool IsFloodfill () const { return m_IsFloodfill; };
void SetFloodfill (bool floodfill);
2016-02-21 04:20:19 +03:00
void SetFamily (const std::string& family);
std::string GetFamily () const;
void SetBandwidth (int limit); /* in kilobytes */
void SetBandwidth (char L); /* by letter */
void SetShareRatio (int percents); // 0 - 100
2014-09-30 21:34:29 +04:00
bool AcceptsTunnels () const { return m_AcceptsTunnels; };
void SetAcceptsTunnels (bool acceptsTunnels) { m_AcceptsTunnels = acceptsTunnels; };
bool SupportsV6 () const { return m_RouterInfo.IsV6 (); };
2016-03-25 01:44:41 +03:00
bool SupportsV4 () const { return m_RouterInfo.IsV4 (); };
2021-02-01 01:25:07 +03:00
bool SupportsMesh () const { return m_RouterInfo.IsMesh (); };
void SetSupportsV6 (bool supportsV6);
2016-03-25 01:44:41 +03:00
void SetSupportsV4 (bool supportsV4);
2021-02-01 02:30:53 +03:00
void SetSupportsMesh (bool supportsmesh, const boost::asio::ip::address_v6& host);
bool IsECIES () const { return GetIdentity ()->GetCryptoKeyType () == i2p::data::CRYPTO_KEY_TYPE_ECIES_X25519_AEAD; };
std::unique_ptr<i2p::crypto::NoiseSymmetricState>& GetCurrentNoiseState () { return m_CurrentNoiseState; };
void UpdateNTCP2V6Address (const boost::asio::ip::address& host); // called from Daemon. TODO: remove
2018-01-06 06:48:51 +03:00
void UpdateStats ();
2018-09-21 17:13:18 +03:00
void UpdateTimestamp (uint64_t ts); // in seconds, called from NetDb before publishing
2016-07-28 20:24:25 +03:00
void CleanupDestination (); // garlic destination
2014-10-27 22:08:50 +03:00
2014-04-01 21:42:04 +04:00
// implements LocalDestination
std::shared_ptr<const i2p::data::IdentityEx> GetIdentity () const { return m_Keys.GetPublic (); };
2020-04-01 00:35:51 +03:00
bool Decrypt (const uint8_t * encrypted, uint8_t * data, BN_CTX * ctx, i2p::data::CryptoKeyType preferredCrypto) const;
2018-01-06 06:48:51 +03:00
void Sign (const uint8_t * buf, int len, uint8_t * signature) const { m_Keys.Sign (buf, len, signature); };
2014-08-16 03:21:30 +04:00
void SetLeaseSetUpdated () {};
2014-10-08 05:08:00 +04:00
// implements GarlicDestination
2016-05-25 22:10:28 +03:00
std::shared_ptr<const i2p::data::LocalLeaseSet> GetLeaseSet () { return nullptr; };
std::shared_ptr<i2p::tunnel::TunnelPool> GetTunnelPool () const;
2015-06-10 05:14:31 +03:00
// override GarlicDestination
2015-06-16 17:14:14 +03:00
void ProcessGarlicMessage (std::shared_ptr<I2NPMessage> msg);
2018-01-06 06:48:51 +03:00
void ProcessDeliveryStatusMessage (std::shared_ptr<I2NPMessage> msg);
2020-01-07 00:14:41 +03:00
protected:
// implements GarlicDestination
2020-01-07 23:20:55 +03:00
void HandleI2NPMessage (const uint8_t * buf, size_t len);
2020-11-05 23:27:37 +03:00
bool HandleCloveI2NPMessage (I2NPMessageType typeID, const uint8_t * payload, size_t len);
2020-01-07 00:14:41 +03:00
2013-10-23 06:45:40 +04:00
private:
void CreateNewRouter ();
void NewRouterInfo ();
2014-02-23 20:48:09 +04:00
void UpdateRouterInfo ();
2018-06-11 22:33:48 +03:00
void NewNTCP2Keys ();
2013-10-23 06:45:40 +04:00
bool Load ();
void SaveKeys ();
2018-01-06 06:48:51 +03:00
2013-10-23 06:45:40 +04:00
private:
i2p::data::RouterInfo m_RouterInfo;
2018-01-06 06:48:51 +03:00
i2p::data::PrivateKeys m_Keys;
std::shared_ptr<i2p::crypto::CryptoKeyDecryptor> m_Decryptor;
2018-09-21 17:13:18 +03:00
uint64_t m_LastUpdateTime; // in seconds
bool m_AcceptsTunnels, m_IsFloodfill;
2019-06-19 18:43:04 +03:00
std::chrono::time_point<std::chrono::steady_clock> m_StartupTime;
uint64_t m_BandwidthLimit; // allowed bandwidth
int m_ShareRatio;
2015-02-26 21:44:18 +03:00
RouterStatus m_Status;
2016-09-20 04:37:04 +03:00
RouterError m_Error;
2016-10-12 18:26:48 +03:00
int m_NetID;
2015-06-10 05:14:31 +03:00
std::mutex m_GarlicMutex;
2018-06-11 22:33:48 +03:00
std::unique_ptr<NTCP2PrivateKeys> m_NTCP2Keys;
2018-09-09 05:08:08 +03:00
std::unique_ptr<i2p::crypto::X25519Keys> m_StaticKeys;
// for ECIESx25519
std::unique_ptr<i2p::crypto::NoiseSymmetricState> m_InitialNoiseState, m_CurrentNoiseState;
2013-10-23 06:45:40 +04:00
};
extern RouterContext context;
2018-01-06 06:48:51 +03:00
}
2013-10-23 06:45:40 +04:00
#endif