mirror of
https://github.com/PurpleI2P/i2pd
synced 2024-11-10 00:00:29 +03:00
130 lines
3.2 KiB
C++
130 lines
3.2 KiB
C++
#include <boost/bind.hpp>
|
|
#include "Log.h"
|
|
#include "RouterContext.h"
|
|
#include "I2NPProtocol.h"
|
|
#include "NetDb.h"
|
|
#include "Transports.h"
|
|
|
|
using namespace i2p::data;
|
|
|
|
namespace i2p
|
|
{
|
|
Transports transports;
|
|
|
|
Transports::Transports ():
|
|
m_Thread (0), m_Work (m_Service),m_NTCPAcceptor (0)
|
|
{
|
|
}
|
|
|
|
Transports::~Transports ()
|
|
{
|
|
Stop ();
|
|
}
|
|
|
|
void Transports::Start ()
|
|
{
|
|
m_Thread = new std::thread (std::bind (&Transports::Run, this));
|
|
// create acceptors
|
|
auto addresses = context.GetRouterInfo ().GetAddresses ();
|
|
for (auto& address : addresses)
|
|
{
|
|
if (address.transportStyle == RouterInfo::eTransportNTCP)
|
|
{
|
|
m_NTCPAcceptor = new boost::asio::ip::tcp::acceptor (m_Service,
|
|
boost::asio::ip::tcp::endpoint(boost::asio::ip::tcp::v4(), address.port));
|
|
|
|
LogPrint ("Start listening port ", address.port);
|
|
auto conn = new i2p::ntcp::NTCPServerConnection (m_Service);
|
|
m_NTCPAcceptor->async_accept(conn->GetSocket (), boost::bind (&Transports::HandleAccept, this,
|
|
conn, boost::asio::placeholders::error));
|
|
}
|
|
}
|
|
}
|
|
|
|
void Transports::Stop ()
|
|
{
|
|
for (auto session: m_NTCPSessions)
|
|
delete session.second;
|
|
m_NTCPSessions.clear ();
|
|
delete m_NTCPAcceptor;
|
|
|
|
m_Service.stop ();
|
|
if (m_Thread)
|
|
{
|
|
m_Thread->join ();
|
|
delete m_Thread;
|
|
m_Thread = 0;
|
|
}
|
|
}
|
|
|
|
void Transports::AddNTCPSession (i2p::ntcp::NTCPSession * session)
|
|
{
|
|
if (session)
|
|
m_NTCPSessions[std::string ((const char *)session->GetRemoteRouterInfo ().GetIdentHash (), 32)] = session;
|
|
}
|
|
|
|
void Transports::RemoveNTCPSession (i2p::ntcp::NTCPSession * session)
|
|
{
|
|
if (session)
|
|
m_NTCPSessions.erase (std::string ((const char *)session->GetRemoteRouterInfo ().GetIdentHash (), 32));
|
|
}
|
|
|
|
void Transports::HandleAccept (i2p::ntcp::NTCPServerConnection * conn, const boost::system::error_code& error)
|
|
{
|
|
if (!error)
|
|
{
|
|
LogPrint ("Connected from ", conn->GetSocket ().remote_endpoint().address ().to_string ());
|
|
conn->ServerLogin ();
|
|
}
|
|
else
|
|
{
|
|
delete conn;
|
|
}
|
|
|
|
conn = new i2p::ntcp::NTCPServerConnection (m_Service);
|
|
m_NTCPAcceptor->async_accept(conn->GetSocket (), boost::bind (&Transports::HandleAccept, this,
|
|
conn, boost::asio::placeholders::error));
|
|
}
|
|
|
|
i2p::ntcp::NTCPSession * Transports::GetNextNTCPSession ()
|
|
{
|
|
for (auto session: m_NTCPSessions)
|
|
if (session.second->IsEstablished ())
|
|
return session.second;
|
|
return 0;
|
|
}
|
|
|
|
i2p::ntcp::NTCPSession * Transports::FindNTCPSession (const uint8_t * ident)
|
|
{
|
|
auto it = m_NTCPSessions.find (std::string ((const char *)ident,32));
|
|
if (it != m_NTCPSessions.end ())
|
|
return it->second;
|
|
return 0;
|
|
}
|
|
|
|
void Transports::SendMessage (const uint8_t * ident, i2p::I2NPMessage * msg)
|
|
{
|
|
if (!memcmp (ident, i2p::context.GetRouterInfo ().GetIdentHash (), 32))
|
|
// we send it to ourself
|
|
i2p::HandleI2NPMessage (msg);
|
|
else
|
|
{
|
|
auto session = FindNTCPSession (ident);
|
|
if (!session)
|
|
{
|
|
RouterInfo * r = netdb.FindRouter (ident);
|
|
if (r)
|
|
{
|
|
auto address = r->GetNTCPAddress ();
|
|
if (address)
|
|
session = new i2p::ntcp::NTCPClient (m_Service, address->host.c_str (), address->port, *r);
|
|
}
|
|
}
|
|
if (session)
|
|
session->SendI2NPMessage (msg);
|
|
else
|
|
LogPrint ("Session not found");
|
|
}
|
|
}
|
|
}
|