i2pd/libi2pd/Transports.h

202 lines
6.7 KiB
C
Raw Normal View History

/*
2022-03-12 00:17:44 +03:00
* Copyright (c) 2013-2022, 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-27 19:26:39 +04:00
#ifndef TRANSPORTS_H__
#define TRANSPORTS_H__
#include <thread>
2014-04-05 00:29:40 +04:00
#include <mutex>
#include <condition_variable>
2013-10-27 19:26:39 +04:00
#include <functional>
2020-06-30 01:19:31 +03:00
#include <unordered_map>
2015-01-21 05:05:57 +03:00
#include <vector>
2014-04-05 00:29:40 +04:00
#include <queue>
2013-10-27 19:26:39 +04:00
#include <string>
2014-11-26 00:30:15 +03:00
#include <memory>
2015-03-17 02:33:59 +03:00
#include <atomic>
2013-10-27 19:26:39 +04:00
#include <boost/asio.hpp>
2014-10-21 00:09:59 +04:00
#include "TransportSession.h"
2014-01-24 01:10:33 +04:00
#include "SSU.h"
2022-03-12 00:17:44 +03:00
#include "SSU2.h"
2018-07-13 22:59:28 +03:00
#include "NTCP2.h"
2013-10-27 19:26:39 +04:00
#include "RouterInfo.h"
#include "I2NPProtocol.h"
2014-04-04 21:30:13 +04:00
#include "Identity.h"
2013-10-27 19:26:39 +04:00
namespace i2p
{
namespace transport
2013-10-27 19:26:39 +04:00
{
template<typename Keys>
class EphemeralKeysSupplier
2014-04-05 00:29:40 +04:00
{
// called from this file only, so implementation is in Transports.cpp
2014-04-05 00:29:40 +04:00
public:
EphemeralKeysSupplier (int size);
~EphemeralKeysSupplier ();
2014-04-05 00:29:40 +04:00
void Start ();
void Stop ();
std::shared_ptr<Keys> Acquire ();
void Return (std::shared_ptr<Keys> pair);
2014-04-05 00:29:40 +04:00
private:
void Run ();
void CreateEphemeralKeys (int num);
2014-04-05 00:29:40 +04:00
private:
const int m_QueueSize;
std::queue<std::shared_ptr<Keys> > m_Queue;
2014-04-05 00:29:40 +04:00
bool m_IsRunning;
2017-05-29 08:28:16 +03:00
std::thread * m_Thread;
2014-04-05 00:29:40 +04:00
std::condition_variable m_Acquired;
std::mutex m_AcquiredMutex;
};
typedef EphemeralKeysSupplier<i2p::crypto::X25519Keys> X25519KeysPairSupplier;
const int PEER_ROUTER_INFO_UPDATE_INTERVAL = 31*60; // in seconds
const int PEER_ROUTER_INFO_UPDATE_INTERVAL_VARIANCE = 7*60; // in seconds
2015-01-13 06:53:35 +03:00
struct Peer
{
2015-01-14 05:31:39 +03:00
int numAttempts;
2015-01-13 06:53:35 +03:00
std::shared_ptr<const i2p::data::RouterInfo> router;
std::list<std::shared_ptr<TransportSession> > sessions;
uint64_t creationTime, nextRouterInfoUpdateTime;
std::vector<std::shared_ptr<i2p::I2NPMessage> > delayedMessages;
2015-01-13 06:53:35 +03:00
void Done ()
{
2016-08-09 01:53:37 +03:00
for (auto& it: sessions)
it->Done ();
2017-05-29 08:28:16 +03:00
}
};
const uint64_t SESSION_CREATION_TIMEOUT = 15; // in seconds
2016-11-13 17:14:05 +03:00
const int PEER_TEST_INTERVAL = 71; // in minutes
const int MAX_NUM_DELAYED_MESSAGES = 150;
2013-10-27 19:26:39 +04:00
class Transports
{
public:
Transports ();
~Transports ();
2022-03-14 04:34:11 +03:00
void Start (bool enableNTCP2=true, bool enableSSU=true, bool enableSSU2=false);
2013-10-27 19:26:39 +04:00
void Stop ();
2016-06-13 18:34:44 +03:00
bool IsBoundSSU() const { return m_SSUServer != nullptr; }
bool IsBoundSSU2() const { return m_SSU2Server != nullptr; }
2018-08-10 22:35:43 +03:00
bool IsBoundNTCP2() const { return m_NTCP2Server != nullptr; }
2017-05-29 08:28:16 +03:00
2016-07-13 17:09:22 +03:00
bool IsOnline() const { return m_IsOnline; };
2020-10-12 00:51:40 +03:00
void SetOnline (bool online);
2016-07-13 17:09:22 +03:00
boost::asio::io_service& GetService () { return *m_Service; };
std::shared_ptr<i2p::crypto::X25519Keys> GetNextX25519KeysPair ();
void ReuseX25519KeysPair (std::shared_ptr<i2p::crypto::X25519Keys> pair);
2013-10-27 19:26:39 +04:00
2015-06-17 18:41:07 +03:00
void SendMessage (const i2p::data::IdentHash& ident, std::shared_ptr<i2p::I2NPMessage> msg);
void SendMessages (const i2p::data::IdentHash& ident, const std::vector<std::shared_ptr<i2p::I2NPMessage> >& msgs);
2015-01-13 06:53:35 +03:00
void PeerConnected (std::shared_ptr<TransportSession> session);
void PeerDisconnected (std::shared_ptr<TransportSession> session);
bool IsConnected (const i2p::data::IdentHash& ident) const;
2017-05-29 08:28:16 +03:00
2015-03-17 02:33:59 +03:00
void UpdateSentBytes (uint64_t numBytes) { m_TotalSentBytes += numBytes; };
void UpdateReceivedBytes (uint64_t numBytes) { m_TotalReceivedBytes += numBytes; };
uint64_t GetTotalSentBytes () const { return m_TotalSentBytes; };
2017-05-02 21:20:00 +03:00
uint64_t GetTotalReceivedBytes () const { return m_TotalReceivedBytes; };
uint64_t GetTotalTransitTransmittedBytes () const { return m_TotalTransitTransmittedBytes; }
void UpdateTotalTransitTransmittedBytes (uint32_t add) { m_TotalTransitTransmittedBytes += add; };
uint32_t GetInBandwidth () const { return m_InBandwidth; };
uint32_t GetOutBandwidth () const { return m_OutBandwidth; };
2017-05-02 21:20:00 +03:00
uint32_t GetTransitBandwidth () const { return m_TransitBandwidth; };
bool IsBandwidthExceeded () const;
bool IsTransitBandwidthExceeded () const;
size_t GetNumPeers () const { return m_Peers.size (); };
std::shared_ptr<const i2p::data::RouterInfo> GetRandomPeer () const;
2015-03-17 22:19:38 +03:00
/** get a trusted first hop for restricted routes */
std::shared_ptr<const i2p::data::RouterInfo> GetRestrictedPeer() const;
/** do we want to use restricted routes? */
bool RoutesRestricted() const;
/** restrict routes to use only these router families for first hops */
2022-03-24 22:50:20 +03:00
void RestrictRoutesToFamilies(const std::set<std::string>& families);
/** restrict routes to use only these routers for first hops */
void RestrictRoutesToRouters(std::set<i2p::data::IdentHash> routers);
bool IsRestrictedPeer(const i2p::data::IdentHash & ident) const;
2017-05-29 08:28:16 +03:00
void PeerTest (bool ipv4 = true, bool ipv6 = true);
2017-05-29 08:28:16 +03:00
void SetCheckReserved (bool check) { m_CheckReserved = check; };
bool IsCheckReserved () { return m_CheckReserved; };
2013-10-27 19:26:39 +04:00
private:
2013-11-29 16:52:09 +04:00
void Run ();
void RequestComplete (std::shared_ptr<const i2p::data::RouterInfo> r, const i2p::data::IdentHash& ident);
void HandleRequestComplete (std::shared_ptr<const i2p::data::RouterInfo> r, i2p::data::IdentHash ident);
2015-06-17 18:41:07 +03:00
void PostMessages (i2p::data::IdentHash ident, std::vector<std::shared_ptr<i2p::I2NPMessage> > msgs);
2015-01-14 05:31:39 +03:00
bool ConnectToPeer (const i2p::data::IdentHash& ident, Peer& peer);
2017-05-29 08:28:16 +03:00
void HandlePeerCleanupTimer (const boost::system::error_code& ecode);
2016-11-13 17:14:05 +03:00
void HandlePeerTestTimer (const boost::system::error_code& ecode);
2017-05-29 08:28:16 +03:00
2015-03-17 22:19:38 +03:00
void UpdateBandwidth ();
2014-02-09 06:06:40 +04:00
void DetectExternalIP ();
2017-05-29 08:28:16 +03:00
2013-10-27 19:26:39 +04:00
private:
2020-10-12 00:51:40 +03:00
volatile bool m_IsOnline;
bool m_IsRunning, m_IsNAT, m_CheckReserved;
2017-05-29 08:28:16 +03:00
std::thread * m_Thread;
boost::asio::io_service * m_Service;
boost::asio::io_service::work * m_Work;
boost::asio::deadline_timer * m_PeerCleanupTimer, * m_PeerTestTimer;
2013-10-27 19:26:39 +04:00
SSUServer * m_SSUServer;
2022-03-12 00:17:44 +03:00
SSU2Server * m_SSU2Server;
2018-07-13 22:59:28 +03:00
NTCP2Server * m_NTCP2Server;
2016-01-16 00:23:03 +03:00
mutable std::mutex m_PeersMutex;
2020-06-30 01:19:31 +03:00
std::unordered_map<i2p::data::IdentHash, Peer> m_Peers;
2017-05-29 08:28:16 +03:00
X25519KeysPairSupplier m_X25519KeysPairSupplier;
2015-03-17 22:19:38 +03:00
2017-05-02 21:20:00 +03:00
std::atomic<uint64_t> m_TotalSentBytes, m_TotalReceivedBytes, m_TotalTransitTransmittedBytes;
uint32_t m_InBandwidth, m_OutBandwidth, m_TransitBandwidth; // bytes per second
2017-05-29 08:28:16 +03:00
uint64_t m_LastInBandwidthUpdateBytes, m_LastOutBandwidthUpdateBytes, m_LastTransitBandwidthUpdateBytes;
uint64_t m_LastBandwidthUpdateTime;
2015-03-17 22:19:38 +03:00
2016-11-15 22:11:55 +03:00
/** which router families to trust for first hops */
2022-03-24 22:50:20 +03:00
std::vector<i2p::data::FamilyID> m_TrustedFamilies;
2016-11-15 22:11:55 +03:00
mutable std::mutex m_FamilyMutex;
2016-11-15 22:11:55 +03:00
/** which routers for first hop to trust */
std::vector<i2p::data::IdentHash> m_TrustedRouters;
mutable std::mutex m_TrustedRoutersMutex;
2017-05-29 08:28:16 +03:00
i2p::I2NPMessagesHandler m_LoopbackHandler;
2013-12-10 17:10:49 +04:00
public:
// for HTTP only
const SSUServer * GetSSUServer () const { return m_SSUServer; };
2018-07-13 22:59:28 +03:00
const NTCP2Server * GetNTCP2Server () const { return m_NTCP2Server; };
2022-03-28 02:29:50 +03:00
const SSU2Server * GetSSU2Server () const { return m_SSU2Server; };
2015-01-14 21:21:41 +03:00
const decltype(m_Peers)& GetPeers () const { return m_Peers; };
2017-05-29 08:28:16 +03:00
};
2013-10-27 19:26:39 +04:00
extern Transports transports;
2017-05-29 08:28:16 +03:00
}
}
2013-10-27 19:26:39 +04:00
#endif