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

bittorrent_udp: response_to_bytes: improve scrape encoding

Browse source
This commit is contained in:
Joakim Frostegård 2020-04-09 20:27:50 +02:00
parent 9705d1822e
commit a37b44ec89
1 changed files with 13 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

36
bittorrent_udp/src/converters/responses.rs
View file

@ -175,33 +175,23 @@ pub fn response_from_bytes(
2 => {
let mut stats = Vec::new();
// TODO: transition to while let && when available
loop {
if let Ok(seeders) = bytes.read_i32::<NetworkEndian>() {
if let Ok(downloaded) = bytes.read_i32::<NetworkEndian>() {
if let Ok(leechers) = bytes.read_i32::<NetworkEndian>() {
stats.push(types::TorrentScrapeStatistics {
seeders: types::NumberOfPeers(seeders),
completed: types::NumberOfDownloads(downloaded),
leechers: types::NumberOfPeers(leechers)
});
}
else {
break;
}
}
else {
break;
}
}
else {
break;
}
let mut buf = [0u8; 12];
while let Ok(()) = bytes.read_exact(&mut buf){
let seeders = (&buf[0..4]).read_i32::<NetworkEndian>().unwrap();
let downloads = (&buf[4..8]).read_i32::<NetworkEndian>().unwrap();
let leechers = (&buf[8..12]).read_i32::<NetworkEndian>().unwrap();
stats.push(types::TorrentScrapeStatistics {
seeders: types::NumberOfPeers(seeders),
completed: types::NumberOfDownloads(downloads),
leechers:types::NumberOfPeers(leechers)
})
}
Ok(types::Response::Scrape(types::ScrapeResponse {
transaction_id: types::TransactionId(transaction_id),
torrent_stats: stats
torrent_stats: stats
}))
},