i2pd/Datagram.h

62 lines
1.9 KiB
C
Raw Normal View History

2014-10-22 19:30:25 +00:00
#ifndef DATAGRAM_H__
#define DATAGRAM_H__
#include <inttypes.h>
2015-01-29 02:37:08 +00:00
#include <memory>
2014-10-31 20:44:44 +00:00
#include <functional>
2015-04-04 00:34:37 +00:00
#include <map>
2015-11-03 14:15:49 +00:00
#include "Base.h"
2014-10-31 20:44:44 +00:00
#include "Identity.h"
2014-10-23 20:56:50 +00:00
#include "LeaseSet.h"
#include "I2NPProtocol.h"
2014-10-22 19:30:25 +00:00
namespace i2p
{
namespace client
{
class ClientDestination;
}
namespace datagram
{
const size_t MAX_DATAGRAM_SIZE = 32768;
class DatagramDestination
{
typedef std::function<void (const i2p::data::IdentityEx& from, uint16_t fromPort, uint16_t toPort, const uint8_t * buf, size_t len)> Receiver;
2014-10-31 20:44:44 +00:00
2014-10-22 19:30:25 +00:00
public:
2015-11-03 14:15:49 +00:00
DatagramDestination (std::shared_ptr<i2p::client::ClientDestination> owner);
~DatagramDestination ();
2014-10-22 19:30:25 +00:00
2015-03-27 15:29:40 +00:00
void SendDatagramTo (const uint8_t * payload, size_t len, const i2p::data::IdentHash& ident, uint16_t fromPort = 0, uint16_t toPort = 0);
2015-03-02 02:08:34 +00:00
void HandleDataMessagePayload (uint16_t fromPort, uint16_t toPort, const uint8_t * buf, size_t len);
2014-10-22 19:30:25 +00:00
2014-10-31 20:44:44 +00:00
void SetReceiver (const Receiver& receiver) { m_Receiver = receiver; };
void ResetReceiver () { m_Receiver = nullptr; };
2015-04-04 00:34:37 +00:00
void SetReceiver (const Receiver& receiver, uint16_t port) { m_ReceiversByPorts[port] = receiver; };
void ResetReceiver (uint16_t port) { m_ReceiversByPorts.erase (port); };
2014-10-23 20:56:50 +00:00
private:
2015-11-24 18:09:12 +00:00
void HandleLeaseSetRequestComplete (std::shared_ptr<i2p::data::LeaseSet> leaseSet, std::shared_ptr<I2NPMessage> msg);
2015-03-27 01:23:59 +00:00
2015-11-24 18:09:12 +00:00
std::shared_ptr<I2NPMessage> CreateDataMessage (const uint8_t * payload, size_t len, uint16_t fromPort, uint16_t toPort);
void SendMsg (std::shared_ptr<I2NPMessage> msg, std::shared_ptr<const i2p::data::LeaseSet> remote);
2015-03-02 02:08:34 +00:00
void HandleDatagram (uint16_t fromPort, uint16_t toPort, const uint8_t * buf, size_t len);
2014-10-23 20:56:50 +00:00
2014-10-22 19:30:25 +00:00
private:
2015-11-03 14:15:49 +00:00
std::shared_ptr<i2p::client::ClientDestination> m_Owner;
2015-04-04 00:34:37 +00:00
Receiver m_Receiver; // default
std::map<uint16_t, Receiver> m_ReceiversByPorts;
2015-11-03 14:15:49 +00:00
i2p::data::GzipInflator m_Inflator;
i2p::data::GzipDeflator m_Deflator;
2014-10-22 19:30:25 +00:00
};
}
}
#endif