YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-03-31 17:55:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

udp: make handle_scrape_request take protocol-specific TorrentMap

Browse source
This commit is contained in:
Joakim Frostegård 2022-04-16 02:21:19 +02:00
parent 1025391e4f
commit b8a74f0724
1 changed files with 14 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

40
aquatic_udp/src/workers/request/mod.rs
View file

@ -1,6 +1,5 @@
mod storage;
use std::collections::BTreeMap;
use std::net::IpAddr;
use std::sync::atomic::Ordering;
use std::time::Duration;
@ -67,8 +66,11 @@ pub fn run_request_worker(
ConnectedResponse::AnnounceIpv6(response)
}
(ConnectedRequest::Scrape(request), ip) => {
ConnectedResponse::Scrape(handle_scrape_request(&mut torrents, ip, request))
(ConnectedRequest::Scrape(request), IpAddr::V4(_)) => {
ConnectedResponse::Scrape(handle_scrape_request(&mut torrents.ipv4, request))
}
(ConnectedRequest::Scrape(request), IpAddr::V6(_)) => {
ConnectedResponse::Scrape(handle_scrape_request(&mut torrents.ipv6, request))
}
};
@ -181,18 +183,17 @@ fn calc_max_num_peers_to_take(config: &Config, peers_wanted: i32) -> usize {
}
}
fn handle_scrape_request(
torrents: &mut TorrentMaps,
src: IpAddr,
fn handle_scrape_request<I: Ip>(
torrents: &mut TorrentMap<I>,
request: PendingScrapeRequest,
) -> PendingScrapeResponse {
const EMPTY_STATS: TorrentScrapeStatistics = create_torrent_scrape_statistics(0, 0);
let mut torrent_stats: BTreeMap<usize, TorrentScrapeStatistics> = BTreeMap::new();
if src.is_ipv4() {
torrent_stats.extend(request.info_hashes.into_iter().map(|(i, info_hash)| {
let s = if let Some(torrent_data) = torrents.ipv4.0.get(&info_hash) {
let torrent_stats = request
.info_hashes
.into_iter()
.map(|(i, info_hash)| {
let s = if let Some(torrent_data) = torrents.0.get(&info_hash) {
create_torrent_scrape_statistics(
torrent_data.num_seeders as i32,
torrent_data.num_leechers as i32,
@ -202,21 +203,8 @@ fn handle_scrape_request(
};
(i, s)
}));
} else {
torrent_stats.extend(request.info_hashes.into_iter().map(|(i, info_hash)| {
let s = if let Some(torrent_data) = torrents.ipv6.0.get(&info_hash) {
create_torrent_scrape_statistics(
torrent_data.num_seeders as i32,
torrent_data.num_leechers as i32,
)
} else {
EMPTY_STATS
};
(i, s)
}));
}
})
.collect();
PendingScrapeResponse {
slab_key: request.slab_key,