i2pd/Transports.h

152 lines
4.4 KiB
C
Raw Normal View History

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