i2pd/Transports.h

116 lines
3.0 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>
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>
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
};
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
2014-11-25 21:30:15 +00:00
void AddNTCPSession (std::shared_ptr<NTCPSession> session);
void RemoveNTCPSession (std::shared_ptr<NTCPSession> session);
2013-10-27 15:26:39 +00:00
2014-11-25 21:30:15 +00:00
std::shared_ptr<NTCPSession> GetNextNTCPSession ();
std::shared_ptr<NTCPSession> FindNTCPSession (const i2p::data::IdentHash& ident);
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);
2014-11-24 17:26:11 +00:00
void CloseSession (std::shared_ptr<const i2p::data::RouterInfo> router);
2013-10-27 15:26:39 +00:00
private:
2013-11-29 12:52:09 +00:00
void Run ();
2014-11-25 21:30:15 +00:00
void HandleAccept (std::shared_ptr<NTCPSession> conn, const boost::system::error_code& error);
void HandleAcceptV6 (std::shared_ptr<NTCPSession> conn, const boost::system::error_code& error);
void HandleResendTimer (const boost::system::error_code& ecode, boost::asio::deadline_timer * timer,
const i2p::data::IdentHash& ident, i2p::I2NPMessage * msg);
2013-12-29 15:48:57 +00:00
void PostMessage (const i2p::data::IdentHash& ident, i2p::I2NPMessage * msg);
2014-11-24 17:26:11 +00:00
void PostCloseSession (std::shared_ptr<const i2p::data::RouterInfo> router);
2014-11-25 21:30:15 +00:00
void Connect (const boost::asio::ip::address& address, int port, std::shared_ptr<NTCPSession> conn);
void HandleConnect (const boost::system::error_code& ecode, std::shared_ptr<NTCPSession> conn);
2014-11-25 20:16:03 +00:00
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;
2014-10-27 19:08:50 +00:00
boost::asio::ip::tcp::acceptor * m_NTCPAcceptor, * m_NTCPV6Acceptor;
2013-10-27 15:26:39 +00:00
2014-11-25 21:30:15 +00:00
std::map<i2p::data::IdentHash, std::shared_ptr<NTCPSession> > m_NTCPSessions;
SSUServer * m_SSUServer;
2013-12-10 13:10:49 +00:00
2014-04-04 20:29:40 +00:00
DHKeysPairSupplier m_DHKeysPairSupplier;
2013-12-10 13:10:49 +00:00
public:
// for HTTP only
const decltype(m_NTCPSessions)& GetNTCPSessions () const { return m_NTCPSessions; };
const SSUServer * GetSSUServer () const { return m_SSUServer; };
2013-10-27 15:26:39 +00:00
};
extern Transports transports;
}
}
2013-10-27 15:26:39 +00:00
#endif