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

aquatic_ws_load_test: for single connection, use same PeerId

Browse source
This commit is contained in:
Joakim Frostegård 2020-08-03 05:27:40 +02:00
parent 7733ee98ca
commit e916780438
3 changed files with 16 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

10
aquatic_ws_load_test/src/utils.rs
View file

@ -12,6 +12,7 @@ pub fn create_random_request(
config: &Config,
state: &LoadTestState,
rng: &mut impl Rng,
connection_id: usize,
) -> InMessage {
let weights = [
config.torrents.weight_announce as u32,
@ -31,6 +32,7 @@ pub fn create_random_request(
config,
state,
rng,
connection_id
),
RequestType::Scrape => create_scrape_request(
config,
@ -46,6 +48,7 @@ fn create_announce_request(
config: &Config,
state: &LoadTestState,
rng: &mut impl Rng,
connection_id: usize
) -> InMessage {
let (event, bytes_left) = {
if rng.gen_bool(config.torrents.peer_seeder_probability) {
@ -66,9 +69,14 @@ fn create_announce_request(
})
}
let mut peer_id = PeerId([0u8; 20]);
(&mut peer_id.0[..8])
.copy_from_slice(&connection_id.to_ne_bytes());
InMessage::AnnounceRequest(AnnounceRequest {
info_hash: state.info_hashes[info_hash_index],
peer_id: PeerId(rng.gen()),
peer_id,
bytes_left: Some(bytes_left),
event,
numwant: None,