fluencelabs/rust-libp2p
1
0
Fork 0
mirror of https://github.com/fluencelabs/rust-libp2p synced 2025-06-27 08:41:36 +00:00
Code Issues Projects Releases Wiki Activity

Allow changing the Kademlia protocol name (#1118)

Browse Source 
* Allow changing the Kademlia protocol name

* Expose the method to the behaviour

* Address review
This commit is contained in:
Pierre Krieger
2019-05-15 15:44:51 +02:00
committed by GitHub
parent 87a352c84f
commit e9448ec8ca
3 changed files with 54 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
protocols/kad/src/handler.rs
View File

@ -32,7 +32,7 @@ use libp2p_core::protocols_handler::{
};
use libp2p_core::{upgrade, either::EitherOutput, InboundUpgrade, OutboundUpgrade, PeerId, upgrade::Negotiated};
use multihash::Multihash;
use std::{error, fmt, io, time::Duration};
use std::{borrow::Cow, error, fmt, io, time::Duration};
use tokio_io::{AsyncRead, AsyncWrite};
use wasm_timer::Instant;
@ -303,7 +303,6 @@ where
{
/// Create a `KademliaHandler` that only allows sending messages to the remote but denying
/// incoming connections.
#[inline]
pub fn dial_only() -> Self {
KademliaHandler::with_allow_listening(false)
}
@ -312,7 +311,6 @@ where
/// requests.
///
/// The `Default` trait implementation wraps around this function.
#[inline]
pub fn dial_and_listen() -> Self {
KademliaHandler::with_allow_listening(true)
}
@ -326,6 +324,13 @@ where
keep_alive: KeepAlive::Yes,
}
}
/// Modifies the protocol name used on the wire. Can be used to create incompatibilities
/// between networks on purpose.
pub fn with_protocol_name(mut self, name: impl Into<Cow<'static, [u8]>>) -> Self {
self.config = self.config.with_protocol_name(name);
self
}
}
impl<TSubstream, TUserData> Default for KademliaHandler<TSubstream, TUserData>
@ -355,7 +360,7 @@ where
#[inline]
fn listen_protocol(&self) -> SubstreamProtocol<Self::InboundProtocol> {
if self.allow_listening {
SubstreamProtocol::new(self.config).map_upgrade(upgrade::EitherUpgrade::A)
SubstreamProtocol::new(self.config.clone()).map_upgrade(upgrade::EitherUpgrade::A)
} else {
SubstreamProtocol::new(upgrade::EitherUpgrade::B(upgrade::DeniedUpgrade))
}
@ -493,7 +498,7 @@ where
let mut substream = self.substreams.swap_remove(n);
loop {
match advance_substream(substream, self.config) {
match advance_substream(substream, self.config.clone()) {
(Some(new_state), Some(event), _) => {
self.substreams.push(new_state);
return Ok(Async::Ready(event));