i2pd/Transports.h
2013-10-27 11:26:39 -04:00

55 lines
1.2 KiB
C++

#ifndef TRANSPORTS_H__
#define TRANSPORTS_H__
#include <thread>
#include <functional>
#include <map>
#include <string>
#include <boost/asio.hpp>
#include "NTCPSession.h"
#include "RouterInfo.h"
#include "I2NPProtocol.h"
namespace i2p
{
class Transports
{
public:
Transports ();
~Transports ();
void Start ();
void Stop ();
boost::asio::io_service& GetService () { return m_Service; };
void AddNTCPSession (i2p::ntcp::NTCPSession * session);
void RemoveNTCPSession (i2p::ntcp::NTCPSession * session);
i2p::ntcp::NTCPSession * GetNextNTCPSession ();
i2p::ntcp::NTCPSession * FindNTCPSession (const uint8_t * ident);
void SendMessage (const uint8_t * ident, i2p::I2NPMessage * msg);
private:
void Run () { m_Service.run (); };
void HandleAccept (i2p::ntcp::NTCPServerConnection * conn,
const boost::system::error_code& error);
private:
std::thread * m_Thread;
boost::asio::io_service m_Service;
boost::asio::io_service::work m_Work;
boost::asio::ip::tcp::acceptor * m_NTCPAcceptor;
std::map<std::string, i2p::ntcp::NTCPSession *> m_NTCPSessions;
};
extern Transports transports;
}
#endif