mirror of
https://github.com/PurpleI2P/i2pd
synced 2024-11-10 08:00:38 +03:00
111 lines
3.1 KiB
C++
111 lines
3.1 KiB
C++
#ifndef TRANSIT_TUNNEL_H__
|
|
#define TRANSIT_TUNNEL_H__
|
|
|
|
#include <inttypes.h>
|
|
#include <vector>
|
|
#include <mutex>
|
|
#include "aes.h"
|
|
#include "I2NPProtocol.h"
|
|
#include "TunnelEndpoint.h"
|
|
#include "TunnelGateway.h"
|
|
#include "TunnelBase.h"
|
|
|
|
namespace i2p
|
|
{
|
|
namespace tunnel
|
|
{
|
|
class TransitTunnel: public TunnelBase
|
|
{
|
|
public:
|
|
|
|
TransitTunnel (uint32_t receiveTunnelID,
|
|
const uint8_t * nextIdent, uint32_t nextTunnelID,
|
|
const uint8_t * layerKey,const uint8_t * ivKey);
|
|
|
|
virtual size_t GetNumTransmittedBytes () const { return 0; };
|
|
|
|
uint32_t GetTunnelID () const { return m_TunnelID; };
|
|
|
|
// implements TunnelBase
|
|
void SendTunnelDataMsg (i2p::I2NPMessage * msg);
|
|
void HandleTunnelDataMsg (i2p::I2NPMessage * tunnelMsg);
|
|
void EncryptTunnelMsg (I2NPMessage * tunnelMsg);
|
|
uint32_t GetNextTunnelID () const { return m_NextTunnelID; };
|
|
const i2p::data::IdentHash& GetNextIdentHash () const { return m_NextIdent; };
|
|
|
|
private:
|
|
|
|
uint32_t m_TunnelID, m_NextTunnelID;
|
|
i2p::data::IdentHash m_NextIdent;
|
|
|
|
i2p::crypto::TunnelEncryption m_Encryption;
|
|
};
|
|
|
|
class TransitTunnelParticipant: public TransitTunnel
|
|
{
|
|
public:
|
|
|
|
TransitTunnelParticipant (uint32_t receiveTunnelID,
|
|
const uint8_t * nextIdent, uint32_t nextTunnelID,
|
|
const uint8_t * layerKey,const uint8_t * ivKey):
|
|
TransitTunnel (receiveTunnelID, nextIdent, nextTunnelID,
|
|
layerKey, ivKey), m_NumTransmittedBytes (0) {};
|
|
~TransitTunnelParticipant ();
|
|
|
|
size_t GetNumTransmittedBytes () const { return m_NumTransmittedBytes; };
|
|
void HandleTunnelDataMsg (i2p::I2NPMessage * tunnelMsg);
|
|
void FlushTunnelDataMsgs ();
|
|
|
|
private:
|
|
|
|
size_t m_NumTransmittedBytes;
|
|
std::vector<i2p::I2NPMessage *> m_TunnelDataMsgs;
|
|
};
|
|
|
|
class TransitTunnelGateway: public TransitTunnel
|
|
{
|
|
public:
|
|
|
|
TransitTunnelGateway (uint32_t receiveTunnelID,
|
|
const uint8_t * nextIdent, uint32_t nextTunnelID,
|
|
const uint8_t * layerKey,const uint8_t * ivKey):
|
|
TransitTunnel (receiveTunnelID, nextIdent, nextTunnelID,
|
|
layerKey, ivKey), m_Gateway(this) {};
|
|
|
|
void SendTunnelDataMsg (i2p::I2NPMessage * msg);
|
|
void FlushTunnelDataMsgs ();
|
|
size_t GetNumTransmittedBytes () const { return m_Gateway.GetNumSentBytes (); };
|
|
|
|
private:
|
|
|
|
std::mutex m_SendMutex;
|
|
TunnelGateway m_Gateway;
|
|
};
|
|
|
|
class TransitTunnelEndpoint: public TransitTunnel
|
|
{
|
|
public:
|
|
|
|
TransitTunnelEndpoint (uint32_t receiveTunnelID,
|
|
const uint8_t * nextIdent, uint32_t nextTunnelID,
|
|
const uint8_t * layerKey,const uint8_t * ivKey):
|
|
TransitTunnel (receiveTunnelID, nextIdent, nextTunnelID, layerKey, ivKey),
|
|
m_Endpoint (false) {}; // transit endpoint is always outbound
|
|
|
|
void HandleTunnelDataMsg (i2p::I2NPMessage * tunnelMsg);
|
|
size_t GetNumTransmittedBytes () const { return m_Endpoint.GetNumReceivedBytes (); }
|
|
|
|
private:
|
|
|
|
TunnelEndpoint m_Endpoint;
|
|
};
|
|
|
|
TransitTunnel * CreateTransitTunnel (uint32_t receiveTunnelID,
|
|
const uint8_t * nextIdent, uint32_t nextTunnelID,
|
|
const uint8_t * layerKey,const uint8_t * ivKey,
|
|
bool isGateway, bool isEndpoint);
|
|
}
|
|
}
|
|
|
|
#endif
|