mirror of https://github.com/PurpleI2P/i2pd
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.8 KiB
C++
62 lines
1.8 KiB
C++
#ifndef DATAGRAM_H__
|
|
#define DATAGRAM_H__
|
|
|
|
#include <inttypes.h>
|
|
#include <memory>
|
|
#include <functional>
|
|
#include <map>
|
|
#include "Base.h"
|
|
#include "Identity.h"
|
|
#include "LeaseSet.h"
|
|
#include "I2NPProtocol.h"
|
|
|
|
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;
|
|
|
|
public:
|
|
|
|
DatagramDestination (std::shared_ptr<i2p::client::ClientDestination> owner);
|
|
~DatagramDestination ();
|
|
|
|
void SendDatagramTo (const uint8_t * payload, size_t len, const i2p::data::IdentHash& ident, uint16_t fromPort = 0, uint16_t toPort = 0);
|
|
void HandleDataMessagePayload (uint16_t fromPort, uint16_t toPort, const uint8_t * buf, size_t len);
|
|
|
|
void SetReceiver (const Receiver& receiver) { m_Receiver = receiver; };
|
|
void ResetReceiver () { m_Receiver = nullptr; };
|
|
|
|
void SetReceiver (const Receiver& receiver, uint16_t port) { m_ReceiversByPorts[port] = receiver; };
|
|
void ResetReceiver (uint16_t port) { m_ReceiversByPorts.erase (port); };
|
|
|
|
private:
|
|
|
|
void HandleLeaseSetRequestComplete (std::shared_ptr<i2p::data::LeaseSet> leaseSet, I2NPMessage * msg);
|
|
|
|
I2NPMessage * CreateDataMessage (const uint8_t * payload, size_t len, uint16_t fromPort, uint16_t toPort);
|
|
void SendMsg (I2NPMessage * msg, std::shared_ptr<const i2p::data::LeaseSet> remote);
|
|
void HandleDatagram (uint16_t fromPort, uint16_t toPort, const uint8_t * buf, size_t len);
|
|
|
|
private:
|
|
|
|
std::shared_ptr<i2p::client::ClientDestination> m_Owner;
|
|
Receiver m_Receiver; // default
|
|
std::map<uint16_t, Receiver> m_ReceiversByPorts;
|
|
|
|
i2p::data::GzipInflator m_Inflator;
|
|
i2p::data::GzipDeflator m_Deflator;
|
|
};
|
|
}
|
|
}
|
|
|
|
#endif
|
|
|