i2pd/Transports.h

105 lines
2.6 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>
2014-04-05 00:29:40 +04:00
#include <queue>
2013-10-27 19:26:39 +04:00
#include <string>
#include <boost/asio.hpp>
#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
{
2014-04-05 00:29:40 +04:00
class DHKeysPairSupplier
{
public:
DHKeysPairSupplier (int size): m_QueueSize (size), m_IsRunning (false), m_Thread (nullptr) {};
~DHKeysPairSupplier ();
void Start ();
void Stop ();
i2p::data::DHKeysPair * Acquire ();
void Return (i2p::data::DHKeysPair * pair);
2014-04-05 00:29:40 +04:00
private:
void Run ();
void CreateDHKeysPairs (int num);
private:
const int m_QueueSize;
2014-04-05 00:29:40 +04:00
std::queue<i2p::data::DHKeysPair *> m_Queue;
bool m_IsRunning;
std::thread * m_Thread;
std::condition_variable m_Acquired;
std::mutex m_AcquiredMutex;
};
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; };
2014-04-04 21:30:13 +04:00
i2p::data::DHKeysPair * GetNextDHKeysPair ();
void ReuseDHKeysPair (i2p::data::DHKeysPair * pair);
2013-10-27 19:26:39 +04:00
void AddNTCPSession (i2p::ntcp::NTCPSession * session);
void RemoveNTCPSession (i2p::ntcp::NTCPSession * session);
i2p::ntcp::NTCPSession * GetNextNTCPSession ();
2013-11-29 16:52:09 +04:00
i2p::ntcp::NTCPSession * FindNTCPSession (const i2p::data::IdentHash& ident);
2013-10-27 19:26:39 +04:00
2013-11-29 16:52:09 +04:00
void SendMessage (const i2p::data::IdentHash& ident, i2p::I2NPMessage * msg);
void CloseSession (const i2p::data::RouterInfo * router);
2013-10-27 19:26:39 +04:00
private:
2013-11-29 16:52:09 +04:00
void Run ();
void HandleAccept (i2p::ntcp::NTCPServerConnection * 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 19:48:57 +04:00
void PostMessage (const i2p::data::IdentHash& ident, i2p::I2NPMessage * msg);
void PostCloseSession (const i2p::data::RouterInfo * router);
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;
boost::asio::ip::tcp::acceptor * m_NTCPAcceptor;
2013-11-29 16:52:09 +04:00
std::map<i2p::data::IdentHash, i2p::ntcp::NTCPSession *> m_NTCPSessions;
2014-01-24 01:10:33 +04:00
i2p::ssu::SSUServer * m_SSUServer;
2013-12-10 17:10:49 +04:00
2014-04-05 00:29:40 +04:00
DHKeysPairSupplier m_DHKeysPairSupplier;
2013-12-10 17:10:49 +04:00
public:
// for HTTP only
const decltype(m_NTCPSessions)& GetNTCPSessions () const { return m_NTCPSessions; };
2014-02-25 07:28:28 +04:00
const i2p::ssu::SSUServer * GetSSUServer () const { return m_SSUServer; };
2013-10-27 19:26:39 +04:00
};
extern Transports transports;
}
#endif