leyoda/rqbit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Do not exchange peers or torrent metadata in private swarms

Browse source
This commit is contained in:
Alex Galvin 2025-04-17 02:34:56 -04:00
parent 77c0cc98d1
commit 8e929b3710
1 changed files with 20 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

24
crates/librqbit/src/torrent_state/live/mod.rs
View file

@ -990,11 +990,27 @@ impl PeerConnectionHandler for &PeerHandler {
Message::Extended(ExtendedMessage::UtMetadata(UtMetadata::Request(
metadata_piece_id,
))) => {
self.send_metadata_piece(metadata_piece_id)
.with_context(|| format!("error sending metadata piece {metadata_piece_id}"))?;
if self.state.metadata.info.private {
warn!(
"recieved noncompliant ut_metadata message from {}, ignoring",
self.addr
);
} else {
self.send_metadata_piece(metadata_piece_id)
.with_context(|| {
format!("error sending metadata piece {metadata_piece_id}")
})?;
}
}
Message::Extended(ExtendedMessage::UtPex(pex)) => {
self.on_pex_message(pex);
if self.state.metadata.info.private {
warn!(
"recieved noncompliant PEX message from {}, ignoring",
self.addr
);
} else {
self.on_pex_message(pex);
}
}
message => {
warn!("received unsupported message {:?}, ignoring", message);
@ -1032,7 +1048,7 @@ impl PeerConnectionHandler for &PeerHandler {
}
fn on_extended_handshake(&self, hs: &ExtendedHandshake<ByteBuf>) -> anyhow::Result<()> {
if let Some(_peer_pex_msg_id) = hs.ut_pex() {
if !self.state.metadata.info.private && hs.ut_pex().is_some() {
self.state.clone().spawn(
error_span!(
parent: self.state.shared.span.clone(),