i2pd/Transports.h

156 lines
4.9 KiB
C
Raw Normal View History

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>
#include <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"
2013-10-27 19:26:39 +04:00
#include "NTCPSession.h"
2014-01-24 01:10:33 +04:00
#include "SSU.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
{
2014-04-05 00:29:40 +04:00
class DHKeysPairSupplier
{
public:
2014-10-20 23:19:56 +04:00
DHKeysPairSupplier (int size);
2014-04-05 00:29:40 +04:00
~DHKeysPairSupplier ();
void Start ();
void Stop ();
2015-11-03 17:15:49 +03:00
std::shared_ptr<i2p::crypto::DHKeys> Acquire ();
void Return (std::shared_ptr<i2p::crypto::DHKeys> pair);
2014-04-05 00:29:40 +04:00
private:
void Run ();
void CreateDHKeysPairs (int num);
private:
const int m_QueueSize;
2015-11-03 17:15:49 +03:00
std::queue<std::shared_ptr<i2p::crypto::DHKeys> > m_Queue;
2014-04-05 00:29:40 +04:00
bool m_IsRunning;
std::thread * m_Thread;
std::condition_variable m_Acquired;
std::mutex m_AcquiredMutex;
};
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;
2015-02-11 22:45:25 +03:00
uint64_t creationTime;
std::vector<std::shared_ptr<i2p::I2NPMessage> > delayedMessages;
2015-01-13 06:53:35 +03:00
void Done ()
{
for (auto it: sessions)
it->Done ();
}
2015-01-13 06:53:35 +03:00
};
2015-02-11 22:45:25 +03:00
const size_t SESSION_CREATION_TIMEOUT = 10; // in seconds
const uint32_t LOW_BANDWIDTH_LIMIT = 32*1024; // 32KBs
2013-10-27 19:26:39 +04:00
class Transports
{
public:
Transports ();
~Transports ();
void Start ();
void Stop ();
boost::asio::io_service& GetService () { return m_Service; };
2015-11-03 17:15:49 +03:00
std::shared_ptr<i2p::crypto::DHKeys> GetNextDHKeysPair ();
void ReuseDHKeysPair (std::shared_ptr<i2p::crypto::DHKeys> 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);
2014-11-24 20:26:11 +03:00
void CloseSession (std::shared_ptr<const i2p::data::RouterInfo> router);
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;
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; };
2015-03-17 22:19:38 +03:00
uint64_t GetTotalReceivedBytes () const { return m_TotalReceivedBytes; };
uint32_t GetInBandwidth () const { return m_InBandwidth; }; // bytes per second
uint32_t GetOutBandwidth () const { return m_OutBandwidth; }; // bytes per second
bool IsBandwidthExceeded () 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
2015-11-03 17:15:49 +03:00
void PeerTest ();
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, const 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);
2014-11-24 20:26:11 +03:00
void PostCloseSession (std::shared_ptr<const i2p::data::RouterInfo> router);
2015-01-14 05:31:39 +03:00
bool ConnectToPeer (const i2p::data::IdentHash& ident, Peer& peer);
2015-02-11 22:45:25 +03:00
void HandlePeerCleanupTimer (const boost::system::error_code& ecode);
2015-01-17 07:01:40 +03:00
void NTCPResolve (const std::string& addr, const i2p::data::IdentHash& ident);
2015-01-16 23:25:44 +03:00
void HandleNTCPResolve (const boost::system::error_code& ecode, boost::asio::ip::tcp::resolver::iterator it,
2015-12-03 23:45:01 +03:00
i2p::data::IdentHash ident, std::shared_ptr<boost::asio::ip::tcp::resolver> resolver);
void SSUResolve (const std::string& addr, const i2p::data::IdentHash& ident);
void HandleSSUResolve (const boost::system::error_code& ecode, boost::asio::ip::tcp::resolver::iterator it,
2015-02-15 18:23:06 +03:00
i2p::data::IdentHash ident, std::shared_ptr<boost::asio::ip::tcp::resolver> resolver);
2015-01-16 23:25:44 +03:00
2015-03-17 22:19:38 +03:00
void UpdateBandwidth ();
2014-02-09 06:06:40 +04:00
void DetectExternalIP ();
2013-10-27 19:26:39 +04:00
private:
2013-12-29 19:48:57 +04:00
bool m_IsRunning;
2013-10-27 19:26:39 +04:00
std::thread * m_Thread;
boost::asio::io_service m_Service;
boost::asio::io_service::work m_Work;
2015-02-11 22:45:25 +03:00
boost::asio::deadline_timer m_PeerCleanupTimer;
2013-10-27 19:26:39 +04:00
2015-01-12 01:41:56 +03:00
NTCPServer * m_NTCPServer;
SSUServer * m_SSUServer;
2015-01-13 06:53:35 +03:00
std::map<i2p::data::IdentHash, Peer> m_Peers;
2014-04-05 00:29:40 +04:00
DHKeysPairSupplier m_DHKeysPairSupplier;
2015-03-17 22:19:38 +03:00
2015-03-17 02:33:59 +03:00
std::atomic<uint64_t> m_TotalSentBytes, m_TotalReceivedBytes;
2015-03-17 22:19:38 +03:00
uint32_t m_InBandwidth, m_OutBandwidth;
uint64_t m_LastInBandwidthUpdateBytes, m_LastOutBandwidthUpdateBytes;
uint64_t m_LastBandwidthUpdateTime;
2013-12-10 17:10:49 +04:00
public:
// for HTTP only
2015-01-12 01:41:56 +03:00
const NTCPServer * GetNTCPServer () const { return m_NTCPServer; };
const SSUServer * GetSSUServer () const { return m_SSUServer; };
2015-01-14 21:21:41 +03:00
const decltype(m_Peers)& GetPeers () const { return m_Peers; };
2013-10-27 19:26:39 +04:00
};
extern Transports transports;
}
}
2013-10-27 19:26:39 +04:00
#endif