mirror of
https://github.com/fluencelabs/rust-libp2p
synced 2025-05-29 02:31:20 +00:00
d5ea93dd71
Previously, the executor for connection tasks silently defaulted to a `futures::executor::ThreadPool`. This causes issues such as https://github.com/libp2p/rust-libp2p/issues/2230. With this patch, we force the user to choose, which executor they want to run the connection tasks on which results in overall simpler API with less footguns. Closes #3068.
103 lines
3.8 KiB
Rust
103 lines
3.8 KiB
Rust
// Copyright 2020 Parity Technologies (UK) Ltd.
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a
|
|
// copy of this software and associated documentation files (the "Software"),
|
|
// to deal in the Software without restriction, including without limitation
|
|
// the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
// and/or sell copies of the Software, and to permit persons to whom the
|
|
// Software is furnished to do so, subject to the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included in
|
|
// all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
// DEALINGS IN THE SOFTWARE.
|
|
|
|
use futures::{channel::oneshot, prelude::*, ready};
|
|
use libp2p::core::{
|
|
identity,
|
|
multiaddr::Protocol,
|
|
muxing::StreamMuxerBox,
|
|
transport::{self, MemoryTransport},
|
|
upgrade, Multiaddr, PeerId, Transport,
|
|
};
|
|
use libp2p::mplex::MplexConfig;
|
|
use libp2p::plaintext::PlainText2Config;
|
|
use libp2p::swarm::{dummy, Swarm, SwarmEvent};
|
|
use rand::random;
|
|
use std::task::Poll;
|
|
|
|
type TestTransport = transport::Boxed<(PeerId, StreamMuxerBox)>;
|
|
|
|
fn mk_transport(up: upgrade::Version) -> (PeerId, TestTransport) {
|
|
let keys = identity::Keypair::generate_ed25519();
|
|
let id = keys.public().to_peer_id();
|
|
(
|
|
id,
|
|
MemoryTransport::default()
|
|
.upgrade(up)
|
|
.authenticate(PlainText2Config {
|
|
local_public_key: keys.public(),
|
|
})
|
|
.multiplex(MplexConfig::default())
|
|
.boxed(),
|
|
)
|
|
}
|
|
|
|
/// Tests the transport upgrade process with all supported
|
|
/// upgrade protocol versions.
|
|
#[test]
|
|
fn transport_upgrade() {
|
|
let _ = env_logger::try_init();
|
|
|
|
fn run(up: upgrade::Version) {
|
|
let (dialer_id, dialer_transport) = mk_transport(up);
|
|
let (listener_id, listener_transport) = mk_transport(up);
|
|
|
|
let listen_addr = Multiaddr::from(Protocol::Memory(random::<u64>()));
|
|
|
|
let mut dialer =
|
|
Swarm::with_async_std_executor(dialer_transport, dummy::Behaviour, dialer_id);
|
|
let mut listener =
|
|
Swarm::with_async_std_executor(listener_transport, dummy::Behaviour, listener_id);
|
|
|
|
listener.listen_on(listen_addr).unwrap();
|
|
let (addr_sender, addr_receiver) = oneshot::channel();
|
|
|
|
let client = async move {
|
|
let addr = addr_receiver.await.unwrap();
|
|
dialer.dial(addr).unwrap();
|
|
futures::future::poll_fn(move |cx| loop {
|
|
if let SwarmEvent::ConnectionEstablished { .. } =
|
|
ready!(dialer.poll_next_unpin(cx)).unwrap()
|
|
{
|
|
return Poll::Ready(());
|
|
}
|
|
})
|
|
.await
|
|
};
|
|
|
|
let mut addr_sender = Some(addr_sender);
|
|
let server = futures::future::poll_fn(move |cx| loop {
|
|
match ready!(listener.poll_next_unpin(cx)).unwrap() {
|
|
SwarmEvent::NewListenAddr { address, .. } => {
|
|
addr_sender.take().unwrap().send(address).unwrap();
|
|
}
|
|
SwarmEvent::IncomingConnection { .. } => {}
|
|
SwarmEvent::ConnectionEstablished { .. } => return Poll::Ready(()),
|
|
_ => {}
|
|
}
|
|
});
|
|
|
|
async_std::task::block_on(future::select(Box::pin(server), Box::pin(client)));
|
|
}
|
|
|
|
run(upgrade::Version::V1);
|
|
run(upgrade::Version::V1Lazy);
|
|
}
|