i2pd/tunnel/TunnelPool.h

#ifndef TUNNEL_POOL__
#define TUNNEL_POOL__

#include <inttypes.h>
#include <set>
#include <vector>
#include <utility>
#include <mutex>
#include <memory>
#include "Identity.h"
#include "LeaseSet.h"
#include "RouterInfo.h"
#include "I2NPProtocol.h"
#include "TunnelBase.h"
#include "RouterContext.h"
#include "Garlic.h"

namespace i2p
{
namespace tunnel
{
    class Tunnel;
    class InboundTunnel;
    class OutboundTunnel;

    class TunnelPool: public std::enable_shared_from_this<TunnelPool> // per local destination
    {
        public:

            TunnelPool (i2p::garlic::GarlicDestination * localDestination, int numInboundHops, int numOutboundHops, int numInboundTunnels, int numOutboundTunnels);
            ~TunnelPool ();
        
            i2p::garlic::GarlicDestination * GetLocalDestination () const { return m_LocalDestination; };
            void SetLocalDestination (i2p::garlic::GarlicDestination * destination) { m_LocalDestination = destination; };
            void SetExplicitPeers (std::shared_ptr<std::vector<i2p::data::IdentHash> > explicitPeers);

            void CreateTunnels ();
            void TunnelCreated (std::shared_ptr<InboundTunnel> createdTunnel);
            void TunnelExpired (std::shared_ptr<InboundTunnel> expiredTunnel);
            void TunnelCreated (std::shared_ptr<OutboundTunnel> createdTunnel);
            void TunnelExpired (std::shared_ptr<OutboundTunnel> expiredTunnel);
            void RecreateInboundTunnel (std::shared_ptr<InboundTunnel> tunnel);
            void RecreateOutboundTunnel (std::shared_ptr<OutboundTunnel> tunnel);
            std::vector<std::shared_ptr<InboundTunnel> > GetInboundTunnels (int num) const;
            std::shared_ptr<OutboundTunnel> GetNextOutboundTunnel (std::shared_ptr<OutboundTunnel> excluded = nullptr) const;
            std::shared_ptr<InboundTunnel> GetNextInboundTunnel (std::shared_ptr<InboundTunnel> excluded = nullptr) const;      
            std::shared_ptr<OutboundTunnel> GetNewOutboundTunnel (std::shared_ptr<OutboundTunnel> old) const;

            void TestTunnels ();
            void ProcessGarlicMessage (std::shared_ptr<I2NPMessage> msg);
            void ProcessDeliveryStatus (std::shared_ptr<I2NPMessage> msg);

            bool IsActive () const { return m_IsActive; };
            void SetActive (bool isActive) { m_IsActive = isActive; };
            void DetachTunnels ();
            
        private:

            void CreateInboundTunnel ();    
            void CreateOutboundTunnel ();
            void CreatePairedInboundTunnel (std::shared_ptr<OutboundTunnel> outboundTunnel);
            template<class TTunnels>
            typename TTunnels::value_type GetNextTunnel (TTunnels& tunnels, typename TTunnels::value_type excluded) const;
            std::shared_ptr<const i2p::data::RouterInfo> SelectNextHop (std::shared_ptr<const i2p::data::RouterInfo> prevHop) const;
            bool SelectPeers (std::vector<std::shared_ptr<const i2p::data::RouterInfo> >& hops, bool isInbound);
            bool SelectExplicitPeers (std::vector<std::shared_ptr<const i2p::data::RouterInfo> >& hops, bool isInbound);            

        private:

            i2p::garlic::GarlicDestination * m_LocalDestination;
            int m_NumInboundHops, m_NumOutboundHops, m_NumInboundTunnels, m_NumOutboundTunnels;
            std::shared_ptr<std::vector<i2p::data::IdentHash> > m_ExplicitPeers;    
            mutable std::mutex m_InboundTunnelsMutex;
            std::set<std::shared_ptr<InboundTunnel>, TunnelCreationTimeCmp> m_InboundTunnels; // recent tunnel appears first
            mutable std::mutex m_OutboundTunnelsMutex;
            std::set<std::shared_ptr<OutboundTunnel>, TunnelCreationTimeCmp> m_OutboundTunnels;
            std::map<uint32_t, std::pair<std::shared_ptr<OutboundTunnel>, std::shared_ptr<InboundTunnel> > > m_Tests;
            bool m_IsActive;

        public:

            // for HTTP only
            const decltype(m_OutboundTunnels)& GetOutboundTunnels () const { return m_OutboundTunnels; };
            const decltype(m_InboundTunnels)& GetInboundTunnels () const { return m_InboundTunnels; };

    };  
}
}

#endif