fluencelabs/rust-libp2p
1
0
Fork 0
mirror of https://github.com/fluencelabs/rust-libp2p synced 2025-06-29 17:51:35 +00:00
Code Issues Projects Releases Wiki Activity

protocols/kad: Implement NetworkBehaviour::inject_address_change (#1649)

Browse Source 
With 826f513 a `StreamMuxer` can notify that the address of a remote
peer changed. This is needed to support the transport protocol QUIC as
remotes can change their IP addresses within the lifetime of a single
connection.

This commit implements the `NetworkBehaviour::inject_address_change`
handler to update the Kademlia routing table accordingly.
This commit is contained in:
Max Inden
2020-08-06 15:35:24 +02:00
committed by GitHub
parent 7601514eb8
commit 5139ec3ace
6 changed files with 124 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
protocols/kad/src/behaviour/test.rs
View File

@ -32,6 +32,7 @@ use futures::{
};
use futures_timer::Delay;
use libp2p_core::{
connection::{ConnectedPoint, ConnectionId},
PeerId,
Transport,
identity,
@ -1099,3 +1100,41 @@ fn manual_bucket_inserts() {
Poll::Pending
}));
}
#[test]
fn network_behaviour_inject_address_change() {
let local_peer_id = PeerId::random();
let remote_peer_id = PeerId::random();
let connection_id = ConnectionId::new(1);
let old_address: Multiaddr = Protocol::Memory(1).into();
let new_address: Multiaddr = Protocol::Memory(2).into();
let mut kademlia = Kademlia::new(
local_peer_id.clone(),
MemoryStore::new(local_peer_id),
);
kademlia.inject_connection_established(
&remote_peer_id,
&connection_id,
&ConnectedPoint::Dialer { address: old_address.clone() },
);
assert_eq!(
vec![old_address.clone()],
kademlia.addresses_of_peer(&remote_peer_id),
);
kademlia.inject_address_change(
&remote_peer_id,
&connection_id,
&ConnectedPoint::Dialer { address: old_address.clone() },
&ConnectedPoint::Dialer { address: new_address.clone() },
);
assert_eq!(
vec![new_address.clone()],
kademlia.addresses_of_peer(&remote_peer_id),
);
}