i2pd/I2CP.h

69 lines
1.5 KiB
C
Raw Normal View History

2016-05-12 22:37:46 +03:00
#ifndef I2CP_H__
#define I2CP_H__
2016-05-12 23:17:10 +03:00
#include <inttypes.h>
2016-05-12 22:37:46 +03:00
#include <string>
2016-05-12 23:17:10 +03:00
#include <memory>
2016-05-12 22:37:46 +03:00
#include <boost/asio.hpp>
namespace i2p
{
namespace client
{
2016-05-12 23:17:10 +03:00
const uint8_t I2CP_PRTOCOL_BYTE = 0x2A;
2016-05-13 22:13:36 +03:00
const size_t I2CP_SESSION_BUFFER_SIZE = 4096;
2016-05-12 23:17:10 +03:00
2016-05-13 22:13:36 +03:00
const size_t I2CP_HEADER_LENGTH_OFFSET = 0;
const size_t I2CP_HEADER_TYPE_OFFSET = I2CP_HEADER_LENGTH_OFFSET + 4;
const size_t I2CP_HEADER_SIZE = I2CP_HEADER_TYPE_OFFSET + 1;
const uint8_t I2CP_GET_DATE_MESSAGE = 32;
class I2CPServer;
2016-05-12 23:17:10 +03:00
class I2CPSession: public std::enable_shared_from_this<I2CPSession>
{
public:
2016-05-13 22:13:36 +03:00
I2CPSession (I2CPServer& owner, std::shared_ptr<boost::asio::ip::tcp::socket> socket);
~I2CPSession ();
// message handlers
void GetDateMessageHandler (const uint8_t * buf, size_t len);
2016-05-12 23:17:10 +03:00
private:
void ReadProtocolByte ();
void Receive ();
void HandleReceived (const boost::system::error_code& ecode, std::size_t bytes_transferred);
2016-05-13 22:13:36 +03:00
void HandleNextMessage (const uint8_t * buf);
2016-05-12 23:17:10 +03:00
void Terminate ();
private:
2016-05-13 22:13:36 +03:00
I2CPServer& m_Owner;
2016-05-12 23:17:10 +03:00
std::shared_ptr<boost::asio::ip::tcp::socket> m_Socket;
2016-05-13 22:13:36 +03:00
uint8_t m_Buffer[I2CP_SESSION_BUFFER_SIZE], * m_NextMessage;
size_t m_NextMessageLen, m_NextMessageOffset;
2016-05-12 23:17:10 +03:00
};
2016-05-13 22:13:36 +03:00
typedef void (I2CPSession::*I2CPMessageHandler)(const uint8_t * buf, size_t len);
2016-05-12 22:37:46 +03:00
class I2CPServer
{
public:
I2CPServer (const std::string& interface, int port);
2016-05-13 22:13:36 +03:00
private:
I2CPMessageHandler m_MessagesHandlers[256];
public:
const decltype(m_MessagesHandlers)& GetMessagesHandlers () const { return m_MessagesHandlers; };
2016-05-12 22:37:46 +03:00
};
}
}
#endif