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

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

7
aquatic_ws_load_test/src/network.rs
View file

@ -176,6 +176,7 @@ impl Connection {
config: &Config,
state: &LoadTestState,
rng: &mut impl Rng,
connection_id: usize
) -> bool { // bool = remove connection
if !self.can_send {
return false;
@ -185,7 +186,8 @@ impl Connection {
let request = create_random_request(
&config,
&state,
rng
rng,
connection_id
);
// If self.send_answer is set and request is announce request, make
@ -217,7 +219,7 @@ impl Connection {
false
}
Err(err) => {
// eprintln!("send request error: {:?}", err);
eprintln!("send request error: {:?}", err);
true
}
@ -284,6 +286,7 @@ pub fn run_socket_thread(
config,
&state,
&mut rng,
*k
);
if drop_connection {

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,