dandri/i2pd
1
0
Fork 0
mirror of https://github.com/PurpleI2P/i2pd.git synced 2026-08-28 13:54:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

incoming peer connections

Browse Source
This commit is contained in:
orignal
2026-07-11 10:14:19 -04:00
parent 76212eb9d7
commit ef5eea9286
2 changed files with 153 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libi2pd_client/Torrents.cpp
+131 -1
View File
@@ -288,10 +288,118 @@ namespace torrents
}
PeerConnection::PeerConnection (i2p::client::I2PService * owner, std::shared_ptr<i2p::stream::Stream> stream):
i2p::client::I2PServiceHandler (owner), m_Stream (stream)
i2p::client::I2PServiceHandler (owner), m_Stream (stream), m_ReceiveBufferOffset (0)
{
}
void PeerConnection::Terminate ()
{
if (Kill()) return;
if (m_Stream)
{
m_Stream->Close ();
m_Stream = nullptr;
}
Done(shared_from_this ());
}
void PeerConnection::ReceiveHandshake ()
{
StreamReceive ();
}
void PeerConnection::StreamReceive ()
{
if (m_Stream && m_ReceiveBufferOffset < PEER_CONNECTION_RECEIVE_BUFFER_SIZE)
{
if (m_Stream->GetStatus () == i2p::stream::eStreamStatusNew ||
m_Stream->GetStatus () == i2p::stream::eStreamStatusOpen) // regular
{
m_Stream->AsyncReceive (boost::asio::buffer (m_ReceiveBuffer + m_ReceiveBufferOffset,
PEER_CONNECTION_RECEIVE_BUFFER_SIZE - m_ReceiveBufferOffset),
std::bind (&PeerConnection::HandleStreamReceive, shared_from_this (),
std::placeholders::_1, std::placeholders::_2),
PEER_CONNECTION_MAX_IDLE);
}
else // closed by peer
{
// get remaining data
auto len = m_Stream->ReadSome (m_ReceiveBuffer + m_ReceiveBufferOffset,
PEER_CONNECTION_RECEIVE_BUFFER_SIZE - m_ReceiveBufferOffset);
if (len > 0) // still some data
{
m_ReceiveBufferOffset += len;
HandleReceived ();
}
else // no more data*/
Terminate ();
}
}
}
void PeerConnection::HandleStreamReceive (const boost::system::error_code& ecode, size_t bytes_transferred)
{
if (ecode)
{
if (ecode != boost::asio::error::operation_aborted)
{
LogPrint (eLogError, "Torrents: Stream read error: ", ecode.message ());
if (bytes_transferred > 0)
{
m_ReceiveBufferOffset += bytes_transferred;
HandleReceived ();
}
else if (ecode == boost::asio::error::timed_out && m_Stream && m_Stream->IsOpen ())
StreamReceive ();
else
Terminate ();
}
else
Terminate ();
}
else
{
m_ReceiveBufferOffset += bytes_transferred;
HandleReceived ();
StreamReceive ();
}
}
void PeerConnection::HandleReceived ()
{
size_t offset = 0;
while (size_t len = HandleNextMsg (offset) > 0)
offset += len;
if (offset && offset < m_ReceiveBufferOffset)
{
// move reamining data
m_ReceiveBufferOffset -= offset;
memmove (m_ReceiveBuffer, m_ReceiveBuffer + offset, m_ReceiveBufferOffset);
}
else
m_ReceiveBufferOffset = 0;
}
size_t PeerConnection::HandleNextMsg (size_t offset)
{
if (offset >= m_ReceiveBufferOffset) return 0;
return HandleHandshakeMsg ();
}
size_t PeerConnection::HandleHandshakeMsg ()
{
if (m_ReceiveBufferOffset < HANDSHAKE_MSG_LENGTH) return 0;
if (m_ReceiveBuffer[0] != 19 || std::string_view ((const char *)(m_ReceiveBuffer + 1), 19) != "BitTorrent protocol")
{
LogPrint (eLogError, "Torrents: Unexpected handshake protocol string");
Terminate ();
return 0;
}
m_RemotePeerID = std::string_view ((const char *)(m_ReceiveBuffer + 48), 20);
// TODO:: send reply
return HANDSHAKE_MSG_LENGTH;
}
TorrentsTunnel::TorrentsTunnel (std::shared_ptr<i2p::client::ClientDestination> localDestination, std::string_view torrentsDir):
i2p::client::I2PService (localDestination), m_TorrentsDir (torrentsDir),
m_PeerID ("-I2PD-")
@@ -304,6 +412,8 @@ namespace torrents
void TorrentsTunnel::Start ()
{
i2p::client::I2PService::Start ();
Accept ();
if (!m_TorrentsDir.empty() && i2p::fs::Exists (m_TorrentsDir))
{
std::vector<std::string> files;
@@ -350,6 +460,26 @@ namespace torrents
LogPrint (eLogError, "Torrents: Can't open file ", path);
}
void TorrentsTunnel::Accept ()
{
auto localDestination = GetLocalDestination ();
if (localDestination)
{
if (!localDestination->IsAcceptingStreams ()) // set it as default if not set yet
localDestination->AcceptStreams ([this](std::shared_ptr<i2p::stream::Stream> stream)
{
if (stream)
{
auto conn = std::make_shared<PeerConnection> (this, stream);
AddHandler (conn);
conn->ReceiveHandshake ();
}
});
}
else
LogPrint (eLogError, "Torrents: Local destination not set");
}
void TorrentsTunnel::RequestTracker (std::shared_ptr<Torrent> torrent)
{
if (!torrent) return;
libi2pd_client/Torrents.h
+22 -1
View File
@@ -30,6 +30,10 @@ namespace torrents
constexpr size_t REQUEST_BLOCK_SIZE = 16384;
constexpr int TRACKER_RESPONSE_TIMEOUT = 8; // in seconds
constexpr size_t TRACKER_RESPONSE_BUFFER_SIZE = 65535;
constexpr size_t PEER_CONNECTION_RECEIVE_BUFFER_SIZE = 16384;
constexpr int PEER_CONNECTION_MAX_IDLE = 3600; // in seconds
constexpr size_t HANDSHAKE_MSG_LENGTH = 68;
class Piece final
{
@@ -75,15 +79,30 @@ namespace torrents
std::list<std::pair<std::string, std::shared_ptr<const i2p::client::Address> > > m_Peers;
};
class PeerConnection: public i2p::client::I2PServiceHandler
class PeerConnection: public i2p::client::I2PServiceHandler, public std::enable_shared_from_this<PeerConnection>
{
public:
PeerConnection (i2p::client::I2PService * owner, std::shared_ptr<i2p::stream::Stream> stream);
void ReceiveHandshake ();
private:
void Terminate ();
void StreamReceive ();
void HandleStreamReceive (const boost::system::error_code& ecode, size_t bytes_transferred);
void HandleReceived ();
size_t HandleNextMsg (size_t offset);
size_t HandleHandshakeMsg ();
private:
std::shared_ptr<i2p::stream::Stream> m_Stream;
uint8_t m_ReceiveBuffer[PEER_CONNECTION_RECEIVE_BUFFER_SIZE];
size_t m_ReceiveBufferOffset;
std::string m_RemotePeerID;
};
class TorrentsTunnel: public i2p::client::I2PService
@@ -101,6 +120,8 @@ namespace torrents
private:
void Accept ();
void ReadTorrentFile (const std::string& path);
void RequestTracker (std::shared_ptr<Torrent> torrent);
void ReceiveFromTracker (std::shared_ptr<i2p::stream::Stream> stream,