fluencelabs/rust-libp2p
1
0
Fork 0
mirror of https://github.com/fluencelabs/rust-libp2p synced 2025-04-30 21:22:16 +00:00
Code Issues Projects Releases Wiki Activity

Add public key to buckets WIP

Browse Source
This commit is contained in:
folex 2020-03-19 12:11:04 +03:00
parent 783ff2575e
commit 619ed94e64
3 changed files with 19 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
protocols/kad/src/addresses.rs
View File

@ -23,7 +23,7 @@ use smallvec::SmallVec;
use std::fmt;
/// A non-empty list of (unique) addresses of a peer in the routing table.
#[derive(Clone)]
#[derive(Clone, PartialEq, Eq)]
pub struct Addresses {
addrs: SmallVec<[Multiaddr; 6]>,
}

31
protocols/kad/src/behaviour.rs
View File

@ -620,8 +620,9 @@ where
let key = kbucket::Key::new(peer.clone());
match self.kbuckets.entry(&key) {
kbucket::Entry::Present(mut entry, old_status) => {
if let Some(address) = address {
if entry.value().insert(address) {
if let Some(contact) = contact {
if *entry.value() != contact { // TODO: what about public key change?
*entry.value() = contact; // TODO: is there a better way to do that?
self.queued_events.push_back(NetworkBehaviourAction::GenerateEvent(
KademliaEvent::RoutingUpdated {
peer,
@ -637,8 +638,8 @@ where
},
kbucket::Entry::Pending(mut entry, old_status) => {
if let Some(address) = address {
entry.value().insert(address);
if let Some(contact) = contact {
*entry.value() = contact;
}
if old_status != new_status {
entry.update(new_status);
@ -649,11 +650,11 @@ where
// Only connected nodes with a known address are newly inserted.
if new_status == NodeStatus::Connected {
if let Some(contact) = contact {
match entry.insert(contact, new_status) {
match entry.insert(contact.clone(), new_status) {
kbucket::InsertResult::Inserted => {
let event = KademliaEvent::RoutingUpdated {
peer: peer.clone(),
addresses: contact.addresses.clone(),
addresses: contact.addresses,
old_peer: None,
};
self.queued_events.push_back(
@ -1103,15 +1104,17 @@ where
ConnectedPoint::Listener { .. } => None,
};
let contact = self.queries.iter().find_map(|q| q.inner.contacts.get(&peer));
let contact = match (address, public_key) {
(Some(addr), Some(pk)) => Contact::new(Addresses::new(addr), pk.clone()),
_ => None
};
let contact = self.queries
.iter()
.find_map(|q| q.inner.contacts.get(&peer)) // Option<&Contact>
.as_mut() // &mut Option<&Contact>
.and_then(|c|
new_address.map(|addr| { c.insert(addr); c }) // insert(&mut self)
)
.cloned(); // Option<&Contact>
self.connection_updated(peer.clone(), address, NodeStatus::Connected);
// Need Option<Contact> here
self.connection_updated(peer.clone(), contact, NodeStatus::Connected);
self.connected_peers.insert(peer);
}

2
protocols/kad/src/contact.rs
View File

@ -23,7 +23,7 @@ use libp2p_core::identity::ed25519::PublicKey;
use crate::protocol::KadPeer;
use smallvec::SmallVec;
#[derive(Clone)]
#[derive(Clone, PartialEq, Eq)]
pub struct Contact {
pub addresses: Addresses,
pub public_key: PublicKey