i2pd/TunnelPool.h

89 lines
3.0 KiB
C
Raw Normal View History

2014-03-14 16:35:02 +00:00
#ifndef TUNNEL_POOL__
#define TUNNEL_POOL__
2014-03-17 20:50:03 +00:00
#include <inttypes.h>
#include <set>
#include <vector>
2014-03-17 20:50:03 +00:00
#include <utility>
2014-10-03 14:35:11 +00:00
#include <mutex>
#include "Identity.h"
2014-03-14 16:35:02 +00:00
#include "LeaseSet.h"
#include "RouterInfo.h"
#include "I2NPProtocol.h"
#include "TunnelBase.h"
2014-04-02 17:14:21 +00:00
#include "RouterContext.h"
2014-10-07 00:18:18 +00:00
#include "Garlic.h"
2014-03-14 16:35:02 +00:00
namespace i2p
{
namespace tunnel
{
class Tunnel;
class InboundTunnel;
class OutboundTunnel;
class TunnelPool // per local destination
{
public:
2014-10-07 00:18:18 +00:00
TunnelPool (i2p::garlic::GarlicDestination& localDestination, int numHops, int numTunnels = 5);
2014-03-14 16:35:02 +00:00
~TunnelPool ();
const uint8_t * GetEncryptionPrivateKey () const { return m_LocalDestination.GetEncryptionPrivateKey (); };
const uint8_t * GetEncryptionPublicKey () const { return m_LocalDestination.GetEncryptionPublicKey (); };
const i2p::data::LocalDestination& GetLocalDestination () const { return m_LocalDestination; };
2014-10-07 00:18:18 +00:00
i2p::garlic::GarlicDestination& GetGarlicDestination () const { return m_LocalDestination; };
2014-10-21 19:44:28 +00:00
bool IsExploratory () const { return GetIdentHash () == i2p::context.GetIdentHash (); };
2014-04-02 17:14:21 +00:00
void CreateTunnels ();
void TunnelCreated (InboundTunnel * createdTunnel);
2014-03-15 01:22:59 +00:00
void TunnelExpired (InboundTunnel * expiredTunnel);
2014-03-16 20:03:20 +00:00
void TunnelCreated (OutboundTunnel * createdTunnel);
void TunnelExpired (OutboundTunnel * expiredTunnel);
2014-03-15 00:24:12 +00:00
std::vector<InboundTunnel *> GetInboundTunnels (int num) const;
2014-10-03 14:35:11 +00:00
OutboundTunnel * GetNextOutboundTunnel (OutboundTunnel * suggested = nullptr) const;
InboundTunnel * GetNextInboundTunnel (InboundTunnel * suggested = nullptr) const;
2014-08-26 02:47:12 +00:00
const i2p::data::IdentHash& GetIdentHash () const { return m_LocalDestination.GetIdentHash (); };
2014-04-01 19:08:53 +00:00
2014-03-17 20:50:03 +00:00
void TestTunnels ();
void ProcessDeliveryStatus (I2NPMessage * msg);
bool IsActive () const { return m_IsActive; };
void SetActive (bool isActive) { m_IsActive = isActive; };
2014-10-11 13:47:24 +00:00
void DetachTunnels ();
2014-10-11 13:01:08 +00:00
private:
void CreateInboundTunnel ();
2014-03-16 20:03:20 +00:00
void CreateOutboundTunnel ();
2014-08-09 02:44:33 +00:00
void RecreateInboundTunnel (InboundTunnel * tunnel);
void RecreateOutboundTunnel (OutboundTunnel * tunnel);
template<class TTunnels>
typename TTunnels::value_type GetNextTunnel (TTunnels& tunnels,
2014-10-03 14:35:11 +00:00
typename TTunnels::value_type suggested = nullptr) const;
const i2p::data::RouterInfo * SelectNextHop (const i2p::data::RouterInfo * prevHop) const;
2014-03-16 20:03:20 +00:00
2014-03-14 16:35:02 +00:00
private:
2014-10-07 00:18:18 +00:00
i2p::garlic::GarlicDestination& m_LocalDestination;
int m_NumHops, m_NumTunnels;
2014-10-03 14:35:11 +00:00
mutable std::mutex m_InboundTunnelsMutex;
std::set<InboundTunnel *, TunnelCreationTimeCmp> m_InboundTunnels; // recent tunnel appears first
2014-10-03 14:35:11 +00:00
mutable std::mutex m_OutboundTunnelsMutex;
2014-03-16 20:03:20 +00:00
std::set<OutboundTunnel *, TunnelCreationTimeCmp> m_OutboundTunnels;
2014-03-17 20:50:03 +00:00
std::map<uint32_t, std::pair<OutboundTunnel *, InboundTunnel *> > m_Tests;
bool m_IsActive;
2014-09-30 02:18:32 +00:00
public:
// for HTTP only
const decltype(m_OutboundTunnels)& GetOutboundTunnels () const { return m_OutboundTunnels; };
const decltype(m_InboundTunnels)& GetInboundTunnels () const { return m_InboundTunnels; };
2014-03-14 16:35:02 +00:00
};
}
}
#endif