I2P -> .NET

libdotnet/RouterContext.h

@@ -0,0 +1,155 @@
+#ifndef ROUTER_CONTEXT_H__
+#define ROUTER_CONTEXT_H__
+
+#include <inttypes.h>
+#include <string>
+#include <memory>
+#include <mutex>
+#include <boost/asio.hpp>
+#include "Identity.h"
+#include "RouterInfo.h"
+#include "Garlic.h"
+
+namespace dotnet
+{
+	const char ROUTER_INFO[] = "router.info";
+	const char ROUTER_KEYS[] = "router.keys";
+	const char NTCP2_KEYS[] = "ntcp2.keys";	
+	const int ROUTER_INFO_UPDATE_INTERVAL = 1800; // 30 minutes
+
+	enum RouterStatus
+	{
+		eRouterStatusOK = 0,
+		eRouterStatusTesting = 1,
+		eRouterStatusFirewalled = 2,
+		eRouterStatusError = 3
+	};
+
+	enum RouterError
+	{
+		eRouterErrorNone = 0,
+		eRouterErrorClockSkew = 1
+	};
+
+	class RouterContext: public dotnet::garlic::GarlicDestination
+	{
+		private:
+
+			struct NTCP2PrivateKeys 
+			{
+				uint8_t staticPublicKey[32];
+				uint8_t staticPrivateKey[32];
+				uint8_t iv[16];
+			};			
+
+		public:
+
+			RouterContext ();
+			void Init ();
+
+			const dotnet::data::PrivateKeys& GetPrivateKeys () const { return m_Keys; };
+			dotnet::data::RouterInfo& GetRouterInfo () { return m_RouterInfo; };
+			std::shared_ptr<const dotnet::data::RouterInfo> GetSharedRouterInfo () const
+			{
+				return std::shared_ptr<const dotnet::data::RouterInfo> (&m_RouterInfo,
+					[](const dotnet::data::RouterInfo *) {});
+			}
+			std::shared_ptr<dotnet::garlic::GarlicDestination> GetSharedDestination ()
+			{
+				return std::shared_ptr<dotnet::garlic::GarlicDestination> (this,
+					[](dotnet::garlic::GarlicDestination *) {});
+			}
+			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; };
+			dotnet::crypto::X25519Keys& GetStaticKeys (); 
+
+			uint32_t GetUptime () const;
+			uint32_t GetStartupTime () const { return m_StartupTime; };
+			uint64_t GetLastUpdateTime () const { return m_LastUpdateTime; };
+			uint64_t GetBandwidthLimit () const { return m_BandwidthLimit; };
+			uint64_t GetTransitBandwidthLimit () const { return (m_BandwidthLimit*m_ShareRatio)/100LL; };
+			RouterStatus GetStatus () const { return m_Status; };
+			void SetStatus (RouterStatus status);
+			RouterError GetError () const { return m_Error; };
+			void SetError (RouterError error) { m_Status = eRouterStatusError; m_Error = error; };
+			int GetNetID () const { return m_NetID; };
+			void SetNetID (int netID) { m_NetID = netID; };
+			bool DecryptTunnelBuildRecord (const uint8_t * encrypted, uint8_t * data, BN_CTX * ctx) const;
+
+			void UpdatePort (int port); // called from Daemon	
+			void UpdateAddress (const boost::asio::ip::address& host);	// called from SSU or Daemon
+			void PublishNTCP2Address (int port, bool publish = true);
+			void UpdateNTCP2Address (bool enable);
+			void PublishNTCPAddress (bool publish, bool v4only = true);
+			bool AddIntroducer (const dotnet::data::RouterInfo::Introducer& introducer);
+			void RemoveIntroducer (const boost::asio::ip::udp::endpoint& e);
+			bool IsUnreachable () const;
+			void SetUnreachable ();
+			void SetReachable ();
+			bool IsFloodfill () const { return m_IsFloodfill; };
+			void SetFloodfill (bool floodfill);
+			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
+			bool AcceptsTunnels () const { return m_AcceptsTunnels; };
+			void SetAcceptsTunnels (bool acceptsTunnels) { m_AcceptsTunnels = acceptsTunnels; };
+			bool SupportsV6 () const { return m_RouterInfo.IsV6 (); };
+			bool SupportsV4 () const { return m_RouterInfo.IsV4 (); };
+			void SetSupportsV6 (bool supportsV6);
+			void SetSupportsV4 (bool supportsV4);
+
+			void UpdateNTCPV6Address (const boost::asio::ip::address& host); // called from NTCP session
+			void UpdateNTCP2V6Address (const boost::asio::ip::address& host); // called from NTCP2 session
+			void UpdateStats ();
+			void UpdateTimestamp (uint64_t ts); // in seconds, called from NetDb before publishing
+			void CleanupDestination ();	// garlic destination
+
+			// implements LocalDestination
+			std::shared_ptr<const dotnet::data::IdentityEx> GetIdentity () const { return m_Keys.GetPublic (); };
+			bool Decrypt (const uint8_t * encrypted, uint8_t * data, BN_CTX * ctx) const;
+			void Sign (const uint8_t * buf, int len, uint8_t * signature) const { m_Keys.Sign (buf, len, signature); };
+			void SetLeaseSetUpdated () {};
+
+			// implements GarlicDestination
+			std::shared_ptr<const dotnet::data::LocalLeaseSet> GetLeaseSet () { return nullptr; };
+			std::shared_ptr<dotnet::tunnel::TunnelPool> GetTunnelPool () const;
+			void HandleDNNPMessage (const uint8_t * buf, size_t len, std::shared_ptr<dotnet::tunnel::InboundTunnel> from);
+
+			// override GarlicDestination
+			void ProcessGarlicMessage (std::shared_ptr<DNNPMessage> msg);
+			void ProcessDeliveryStatusMessage (std::shared_ptr<DNNPMessage> msg);
+
+		private:
+
+			void CreateNewRouter ();
+			void NewRouterInfo ();
+			void UpdateRouterInfo ();
+			void NewNTCP2Keys ();
+			bool Load ();
+			void SaveKeys ();
+
+		private:
+
+			dotnet::data::RouterInfo m_RouterInfo;
+			dotnet::data::PrivateKeys m_Keys;
+			std::shared_ptr<dotnet::crypto::CryptoKeyDecryptor> m_Decryptor;
+			uint64_t m_LastUpdateTime; // in seconds
+			bool m_AcceptsTunnels, m_IsFloodfill;
+			uint64_t m_StartupTime; // in seconds since epoch
+			uint64_t m_BandwidthLimit; // allowed bandwidth
+			int m_ShareRatio;
+			RouterStatus m_Status;
+			RouterError m_Error;
+			int m_NetID;
+			std::mutex m_GarlicMutex;
+			std::unique_ptr<NTCP2PrivateKeys> m_NTCP2Keys;
+			std::unique_ptr<dotnet::crypto::X25519Keys> m_StaticKeys;
+	};
+
+	extern RouterContext context;
+}
+
+#endif