i2pd/TransitTunnel.cpp

114 lines
3.3 KiB
C++
Raw Normal View History

2013-11-10 23:19:49 +00:00
#include <string.h>
#include "I2PEndian.h"
2013-11-10 23:19:49 +00:00
#include "Log.h"
#include "RouterContext.h"
#include "I2NPProtocol.h"
#include "Tunnel.h"
#include "Transports.h"
#include "TransitTunnel.h"
namespace i2p
{
namespace tunnel
{
TransitTunnel::TransitTunnel (uint32_t receiveTunnelID,
const uint8_t * nextIdent, uint32_t nextTunnelID,
2013-11-29 12:52:09 +00:00
const uint8_t * layerKey,const uint8_t * ivKey):
2014-01-06 03:52:17 +00:00
m_TunnelID (receiveTunnelID), m_NextTunnelID (nextTunnelID),
2015-01-22 02:50:46 +00:00
m_NextIdent (nextIdent)
2013-11-10 23:19:49 +00:00
{
2014-05-15 21:55:09 +00:00
m_Encryption.SetKeys (layerKey, ivKey);
2013-11-10 23:19:49 +00:00
}
2013-11-29 12:52:09 +00:00
void TransitTunnel::EncryptTunnelMsg (I2NPMessage * tunnelMsg)
2014-05-15 21:55:09 +00:00
{
m_Encryption.Encrypt (tunnelMsg->GetPayload () + 4);
2013-11-10 23:19:49 +00:00
}
2015-01-22 02:50:46 +00:00
TransitTunnelParticipant::~TransitTunnelParticipant ()
{
for (auto it: m_TunnelDataMsgs)
i2p::DeleteI2NPMessage (it);
}
void TransitTunnelParticipant::HandleTunnelDataMsg (i2p::I2NPMessage * tunnelMsg)
2013-11-10 23:19:49 +00:00
{
2013-11-29 12:52:09 +00:00
EncryptTunnelMsg (tunnelMsg);
2013-11-10 23:19:49 +00:00
2014-08-02 14:06:38 +00:00
m_NumTransmittedBytes += tunnelMsg->GetLength ();
2015-01-22 02:50:46 +00:00
htobe32buf (tunnelMsg->GetPayload (), GetNextTunnelID ());
2013-11-29 12:52:09 +00:00
FillI2NPMessageHeader (tunnelMsg, eI2NPTunnelData);
2015-01-22 02:50:46 +00:00
m_TunnelDataMsgs.push_back (tunnelMsg);
2013-11-10 23:19:49 +00:00
}
2015-01-22 02:50:46 +00:00
void TransitTunnelParticipant::FlushTunnelDataMsgs ()
{
if (!m_TunnelDataMsgs.empty ())
{
2015-04-10 23:58:32 +00:00
auto num = m_TunnelDataMsgs.size ();
if (num > 1)
LogPrint (eLogDebug, "TransitTunnel: ",GetTunnelID (),"->", GetNextTunnelID (), " ", num);
2015-01-22 02:50:46 +00:00
i2p::transport::transports.SendMessages (GetNextIdentHash (), m_TunnelDataMsgs);
m_TunnelDataMsgs.clear ();
}
}
void TransitTunnel::SendTunnelDataMsg (i2p::I2NPMessage * msg)
2013-11-29 12:52:09 +00:00
{
2015-01-22 02:50:46 +00:00
LogPrint (eLogError, "We are not a gateway for transit tunnel ", m_TunnelID);
2013-11-29 12:52:09 +00:00
i2p::DeleteI2NPMessage (msg);
}
2015-01-22 02:50:46 +00:00
void TransitTunnel::HandleTunnelDataMsg (i2p::I2NPMessage * tunnelMsg)
{
LogPrint (eLogError, "Incoming tunnel message is not supported ", m_TunnelID);
DeleteI2NPMessage (tunnelMsg);
}
void TransitTunnelGateway::SendTunnelDataMsg (i2p::I2NPMessage * msg)
2013-11-29 12:52:09 +00:00
{
TunnelMessageBlock block;
block.deliveryType = eDeliveryTypeLocal;
block.data = msg;
2014-06-26 19:41:12 +00:00
std::unique_lock<std::mutex> l(m_SendMutex);
2015-01-24 03:05:33 +00:00
m_Gateway.PutTunnelDataMsg (block);
2013-11-29 12:52:09 +00:00
}
2015-01-24 03:05:33 +00:00
void TransitTunnelGateway::FlushTunnelDataMsgs ()
{
2015-02-11 23:47:20 +00:00
std::unique_lock<std::mutex> l(m_SendMutex);
2015-01-24 03:05:33 +00:00
m_Gateway.SendBuffer ();
}
2013-11-29 12:52:09 +00:00
void TransitTunnelEndpoint::HandleTunnelDataMsg (i2p::I2NPMessage * tunnelMsg)
{
EncryptTunnelMsg (tunnelMsg);
2015-01-22 02:50:46 +00:00
LogPrint (eLogDebug, "TransitTunnel endpoint for ", GetTunnelID ());
2013-11-29 12:52:09 +00:00
m_Endpoint.HandleDecryptedTunnelDataMsg (tunnelMsg);
}
TransitTunnel * CreateTransitTunnel (uint32_t receiveTunnelID,
const uint8_t * nextIdent, uint32_t nextTunnelID,
const uint8_t * layerKey,const uint8_t * ivKey,
bool isGateway, bool isEndpoint)
2013-11-10 23:19:49 +00:00
{
2013-11-29 12:52:09 +00:00
if (isEndpoint)
{
2015-01-22 02:50:46 +00:00
LogPrint (eLogInfo, "TransitTunnel endpoint: ", receiveTunnelID, " created");
2013-11-29 12:52:09 +00:00
return new TransitTunnelEndpoint (receiveTunnelID, nextIdent, nextTunnelID, layerKey, ivKey);
}
else if (isGateway)
{
2015-01-22 02:50:46 +00:00
LogPrint (eLogInfo, "TransitTunnel gateway: ", receiveTunnelID, " created");
2013-11-29 12:52:09 +00:00
return new TransitTunnelGateway (receiveTunnelID, nextIdent, nextTunnelID, layerKey, ivKey);
}
else
2013-11-10 23:19:49 +00:00
{
2015-01-22 02:50:46 +00:00
LogPrint (eLogInfo, "TransitTunnel: ", receiveTunnelID, "->", nextTunnelID, " created");
return new TransitTunnelParticipant (receiveTunnelID, nextIdent, nextTunnelID, layerKey, ivKey);
2013-11-10 23:19:49 +00:00
}
}
}
2014-06-26 19:41:12 +00:00
}