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

Writing still broken

Browse source
This commit is contained in:
Igor Katson 2021-06-27 12:51:19 +01:00
parent 3f068883c2
commit 0ac1116aba
1 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

20
crates/librqbit/src/torrent_manager.rs
View file

@ -812,15 +812,15 @@ impl TorrentManager {
})
}
fn try_steal_piece(&self) -> Option<ValidPieceIndex> {
fn try_steal_piece(&self, handle: PeerHandle) -> Option<ValidPieceIndex> {
let mut rng = rand::thread_rng();
use rand::seq::IteratorRandom;
self.inner
.locked
.read()
.peers
let g = self.inner.locked.read();
let pl = g.peers.get_live(handle)?;
g.peers
.inflight_pieces
.iter()
.filter(|p| !pl.inflight_requests.iter().any(|req| req.piece == **p))
.choose(&mut rng)
.copied()
}
@ -859,9 +859,7 @@ impl TorrentManager {
return Ok(());
}
if let Some(piece) = self.try_steal_piece() {
// TODO: ok, so there's a problem here. It can keep looping here requesting the same
// pieces over and over again.
if let Some(piece) = self.try_steal_piece(handle) {
info!("{}: stole a piece {}", handle, piece);
(piece, true)
} else {
@ -895,8 +893,10 @@ impl TorrentManager {
.inflight_requests
.insert(InflightRequest::from(&chunk))
{
// we already requested this chunk.
// this should not happen, but due do a bug above it actually does.
warn!(
"{}: probably a bug, we already requested {:?}",
handle, chunk
);
continue;
}