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>
|
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
|
2014-10-21 16:25:53 +00:00
|
|
|
{
|
|
|
|
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 ();
|
2014-10-21 16:25:53 +00:00
|
|
|
DHKeysPair * Acquire ();
|
|
|
|
void Return (DHKeysPair * pair);
|
2014-04-04 20:29:40 +00:00
|
|
|
|
|
|
|
private:
|
|
|
|
|
|
|
|
void Run ();
|
|
|
|
void CreateDHKeysPairs (int num);
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
2014-09-17 15:13:25 +00:00
|
|
|
const int m_QueueSize;
|
2014-10-21 16:25:53 +00:00
|
|
|
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-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)
|
|
|
|
i2p::DeleteI2NPMessage (it);
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
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 ();
|
2014-10-21 16:25:53 +00:00
|
|
|
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);
|
2014-07-19 00:32:45 +00:00
|
|
|
|
2013-10-27 15:26:39 +00:00
|
|
|
private:
|
|
|
|
|
2013-11-29 12:52:09 +00:00
|
|
|
void Run ();
|
2015-01-14 21:37:03 +00:00
|
|
|
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);
|
2015-02-02 17:01:21 +00:00
|
|
|
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-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,
|
|
|
|
const i2p::data::IdentHash& ident, std::shared_ptr<boost::asio::ip::tcp::resolver> resolver);
|
|
|
|
|
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-01-11 22:41:56 +00:00
|
|
|
NTCPServer * m_NTCPServer;
|
2014-10-21 16:25:53 +00:00
|
|
|
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;
|
|
|
|
|
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; };
|
2014-10-21 16:25:53 +00:00
|
|
|
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;
|
|
|
|
}
|
2014-10-21 16:25:53 +00:00
|
|
|
}
|
2013-10-27 15:26:39 +00:00
|
|
|
|
|
|
|
#endif
|