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

peer has full torrent debug message

Browse source
This commit is contained in:
Igor Katson 2024-08-31 12:39:18 +01:00
parent b60850946e
commit b60ec6d919
No known key found for this signature in database
GPG key ID: B4EC22B66D61A3F5
2 changed files with 19 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1222,6 +1222,13 @@ impl PeerHandler {
} }
}; };
trace!("updated bitfield with have={}", have); trace!("updated bitfield with have={}", have);
if let Some(true) = live
.bitfield
.get(..self.state.lengths.total_pieces() as usize)
.map(|s| s.all())
{
debug!("peer has full torrent");
}
}); });
self.on_bitfield_notify.notify_waiters(); self.on_bitfield_notify.notify_waiters();
} }
@ -1234,9 +1241,14 @@ impl PeerHandler {
self.state.lengths.piece_bitfield_bytes(), self.state.lengths.piece_bitfield_bytes(),
); );
} }
self.state let bf = BF::from_boxed_slice(bitfield.0.to_vec().into_boxed_slice());
.peers if let Some(true) = bf
.update_bitfield_from_vec(self.addr, bitfield.0.to_vec().into_boxed_slice()); .get(..self.state.lengths.total_pieces() as usize)
.map(|s| s.all())
{
debug!("peer has full torrent");
}
self.state.peers.update_bitfield(self.addr, bf);
self.on_bitfield_notify.notify_waiters(); self.on_bitfield_notify.notify_waiters();
Ok(()) Ok(())
} }

7
crates/librqbit/src/torrent_state/live/peers/mod.rs
View file

@ -105,11 +105,12 @@ impl PeerStates {
}) })
} }
pub fn update_bitfield_from_vec(&self, handle: PeerHandle, bitfield: Box<[u8]>) -> Option<()> { pub fn update_bitfield(&self, handle: PeerHandle, bitfield: BF) -> Option<()> {
self.with_live_mut(handle, "update_bitfield_from_vec", |live| { self.with_live_mut(handle, "update_bitfield", |live| {
live.bitfield = BF::from_boxed_slice(bitfield); live.bitfield = bitfield;
}) })
} }
pub fn mark_peer_connecting(&self, h: PeerHandle) -> anyhow::Result<(PeerRx, PeerTx)> { pub fn mark_peer_connecting(&self, h: PeerHandle) -> anyhow::Result<(PeerRx, PeerTx)> {
let rx = self let rx = self
.with_peer_mut(h, "mark_peer_connecting", |peer| { .with_peer_mut(h, "mark_peer_connecting", |peer| {