YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-04-02 10:45:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

aquatic: announce handler: improve safety by small margin

Browse source
This commit is contained in:
Joakim Frostegård 2020-04-08 20:26:39 +02:00
parent 789783cfe4
commit 766606cc08
1 changed files with 39 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

74
aquatic/src/lib/handlers.rs
View file

@ -84,43 +84,47 @@ pub fn handle_announce_requests(
let max_num_peers_to_take = (request.peers_wanted.0.max(0) as usize) let max_num_peers_to_take = (request.peers_wanted.0.max(0) as usize)
.min(config.max_response_peers); .min(config.max_response_peers);
let torrent_data = state.torrents.get(&request.info_hash).unwrap(); // Since there is a miniscule risk of the torrent having been removed
// by now, don't unwrap the result.
match peer_status { if let Some(torrent_data) = state.torrents.get(&request.info_hash){
PeerStatus::Leeching => { match peer_status {
torrent_data.num_leechers.fetch_add(1, Ordering::SeqCst); PeerStatus::Leeching => {
}, torrent_data.num_leechers.fetch_add(1, Ordering::SeqCst);
PeerStatus::Seeding => { },
torrent_data.num_seeders.fetch_add(1, Ordering::SeqCst); PeerStatus::Seeding => {
}, torrent_data.num_seeders.fetch_add(1, Ordering::SeqCst);
PeerStatus::Stopped => {} },
}; PeerStatus::Stopped => {}
};
match opt_removed_peer_status { match opt_removed_peer_status {
Some(PeerStatus::Leeching) => { Some(PeerStatus::Leeching) => {
torrent_data.num_leechers.fetch_sub(1, Ordering::SeqCst); torrent_data.num_leechers.fetch_sub(1, Ordering::SeqCst);
}, },
Some(PeerStatus::Seeding) => { Some(PeerStatus::Seeding) => {
torrent_data.num_seeders.fetch_sub(1, Ordering::SeqCst); torrent_data.num_seeders.fetch_sub(1, Ordering::SeqCst);
}, },
_ => {} _ => {}
}
let response_peers = extract_response_peers(
&mut rng,
&torrent_data.peers,
max_num_peers_to_take,
);
let response = Response::Announce(AnnounceResponse {
transaction_id: request.transaction_id,
announce_interval: AnnounceInterval(600), // FIXME
leechers: NumberOfPeers(torrent_data.num_leechers.load(Ordering::SeqCst) as i32),
seeders: NumberOfPeers(torrent_data.num_seeders.load(Ordering::SeqCst) as i32),
peers: response_peers
});
Some((response, src))
} else {
None
} }
let response_peers = extract_response_peers(
&mut rng,
&torrent_data.peers,
max_num_peers_to_take,
);
let response = Response::Announce(AnnounceResponse {
transaction_id: request.transaction_id,
announce_interval: AnnounceInterval(600), // FIXME
leechers: NumberOfPeers(torrent_data.num_leechers.load(Ordering::SeqCst) as i32),
seeders: NumberOfPeers(torrent_data.num_seeders.load(Ordering::SeqCst) as i32),
peers: response_peers
});
Some((response, src))
})); }));
} }