Node listens on default port and connects to bootnodes.

This commit is contained in:
Age Manning 2019-03-08 11:07:30 +11:00
parent 9f13731d6d
commit 3c51769428
No known key found for this signature in database
GPG Key ID: 05EED64B79E06A93
6 changed files with 61 additions and 38 deletions

View File

@ -7,6 +7,7 @@ edition = "2018"
[dependencies] [dependencies]
beacon_chain = { path = "../beacon_chain" } beacon_chain = { path = "../beacon_chain" }
network = { path = "../network" } network = { path = "../network" }
libp2p = { path = "../libp2p" }
sync = { path = "../sync" } sync = { path = "../sync" }
db = { path = "../db" } db = { path = "../db" }
fork_choice = { path = "../../eth2/fork_choice" } fork_choice = { path = "../../eth2/fork_choice" }

View File

@ -1,10 +1,12 @@
use clap::ArgMatches; use clap::ArgMatches;
use db::DBType; use db::DBType;
use fork_choice::ForkChoiceAlgorithm; use fork_choice::ForkChoiceAlgorithm;
use libp2p::multiaddr::ToMultiaddr;
use network::NetworkConfig; use network::NetworkConfig;
use slog::error; use slog::error;
use std::fs; use std::fs;
use std::net::IpAddr; use std::net::IpAddr;
use std::net::SocketAddr;
use std::path::PathBuf; use std::path::PathBuf;
use types::ChainSpec; use types::ChainSpec;
@ -52,16 +54,6 @@ impl ClientConfig {
// Network related args // Network related args
// Custom listening address ipv4/ipv6
// TODO: Handle list of addresses
if let Some(listen_address_str) = args.value_of("listen_address") {
if let Ok(listen_address) = listen_address_str.parse::<IpAddr>() {
config.net_conf.listen_addresses = vec![listen_address];
} else {
error!(log, "Invalid IP Address"; "Address" => listen_address_str);
return Err("Invalid IP Address");
}
}
// Custom p2p listen port // Custom p2p listen port
if let Some(port_str) = args.value_of("port") { if let Some(port_str) = args.value_of("port") {
if let Ok(port) = port_str.parse::<u16>() { if let Ok(port) = port_str.parse::<u16>() {
@ -71,6 +63,19 @@ impl ClientConfig {
return Err("Invalid port"); return Err("Invalid port");
} }
} }
// Custom listening address ipv4/ipv6
// TODO: Handle list of addresses
if let Some(listen_address_str) = args.value_of("listen_address") {
if let Ok(listen_address) = listen_address_str.parse::<IpAddr>() {
let multiaddr = SocketAddr::new(listen_address, config.net_conf.listen_port)
.to_multiaddr()
.expect("Invalid listen address format");
config.net_conf.listen_addresses = vec![multiaddr];
} else {
error!(log, "Invalid IP Address"; "Address" => listen_address_str);
return Err("Invalid IP Address");
}
}
// filesystem args // filesystem args

View File

@ -11,3 +11,4 @@ slog = "2.4.1"
version = { path = "../version" } version = { path = "../version" }
tokio = "0.1.16" tokio = "0.1.16"
futures = "0.1.25" futures = "0.1.25"
parity-multiaddr = "0.2.0"

View File

@ -6,6 +6,8 @@ mod behaviour;
mod network_config; mod network_config;
mod service; mod service;
pub use libp2p::multiaddr;
pub use libp2p::Multiaddr;
pub use libp2p::{ pub use libp2p::{
gossipsub::{GossipsubConfig, GossipsubConfigBuilder}, gossipsub::{GossipsubConfig, GossipsubConfigBuilder},
PeerId, PeerId,

View File

@ -1,20 +1,20 @@
use libp2p::gossipsub::{GossipsubConfig, GossipsubConfigBuilder}; use libp2p::gossipsub::{GossipsubConfig, GossipsubConfigBuilder};
use libp2p::secio; use libp2p::secio;
use libp2p::Multiaddr;
use std::fmt; use std::fmt;
use std::net::IpAddr;
#[derive(Clone)] #[derive(Clone)]
/// Network configuration for lighthouse. /// Network configuration for lighthouse.
pub struct NetworkConfig { pub struct NetworkConfig {
//TODO: stubbing networking initial params, change in the future //TODO: stubbing networking initial params, change in the future
/// IP address to listen on. /// IP address to listen on.
pub listen_addresses: Vec<IpAddr>, pub listen_addresses: Vec<Multiaddr>,
/// Listen port UDP/TCP. /// Listen port UDP/TCP.
pub listen_port: u16, pub listen_port: u16,
/// Gossipsub configuration parameters. /// Gossipsub configuration parameters.
pub gs_config: GossipsubConfig, pub gs_config: GossipsubConfig,
/// List of nodes to initially connect to. /// List of nodes to initially connect to.
pub boot_nodes: Vec<IpAddr>, pub boot_nodes: Vec<Multiaddr>,
/// Peer key related to this nodes PeerId. /// Peer key related to this nodes PeerId.
pub local_private_key: secio::SecioKeyPair, pub local_private_key: secio::SecioKeyPair,
/// Client version /// Client version
@ -24,21 +24,15 @@ pub struct NetworkConfig {
impl Default for NetworkConfig { impl Default for NetworkConfig {
/// Generate a default network configuration. /// Generate a default network configuration.
fn default() -> Self { fn default() -> Self {
// hard-coded defaults
let bootnodes = ["127.0.0.1"];
let default_port = 9000;
// TODO: Currently using ed25519 key pairs. Wire protocol specifies RSA. Waiting for this // TODO: Currently using ed25519 key pairs. Wire protocol specifies RSA. Waiting for this
// PR to be merged to generate RSA keys: https://github.com/briansmith/ring/pull/733 // PR to be merged to generate RSA keys: https://github.com/briansmith/ring/pull/733
NetworkConfig { NetworkConfig {
listen_addresses: vec!["127.0.0.1".parse().expect("correct IP address")], listen_addresses: vec!["/ip4/127.0.0.1/tcp/9000"
listen_port: default_port, .parse()
.expect("is a correct multi-address")],
listen_port: 9000,
gs_config: GossipsubConfigBuilder::new().build(), gs_config: GossipsubConfigBuilder::new().build(),
boot_nodes: bootnodes boot_nodes: Vec::new(),
.iter()
.map(|s| s.parse().expect("Bootnodes must be IP addresses"))
.collect(),
local_private_key: secio::SecioKeyPair::ed25519_generated().unwrap(), local_private_key: secio::SecioKeyPair::ed25519_generated().unwrap(),
client_version: version::version(), client_version: version::version(),
} }

View File

@ -5,12 +5,12 @@ use libp2p::core::{
muxing::StreamMuxerBox, muxing::StreamMuxerBox,
nodes::Substream, nodes::Substream,
transport::boxed::Boxed, transport::boxed::Boxed,
upgrade::{InboundUpgrade, InboundUpgradeExt, OutboundUpgrade, OutboundUpgradeExt}, upgrade::{InboundUpgradeExt, OutboundUpgradeExt},
}; };
use libp2p::{build_tcp_ws_secio_mplex_yamux, core, secio, Transport}; use libp2p::multiaddr::Protocol;
use libp2p::{core, secio, Transport};
use libp2p::{PeerId, Swarm}; use libp2p::{PeerId, Swarm};
use slog::debug; use slog::{debug, info, warn};
use std::error;
use std::io::{Error, ErrorKind}; use std::io::{Error, ErrorKind};
use std::time::Duration; use std::time::Duration;
@ -21,9 +21,6 @@ pub struct Service {
/// This node's PeerId. /// This node's PeerId.
local_peer_id: PeerId, local_peer_id: PeerId,
} }
//Swarm<impl std::clone::Clone+libp2p_core::transport::Transport, behaviour::Behaviour<libp2p_core::muxing::SubstreamRef<std::sync::Arc<impl std::marker::Send+std::marker::Sync+libp2p_core::muxing::StreamMuxer>>>>
//swarm: Swarm<Boxed<(PeerId, StreamMuxerBox), IoError>, Behaviour<TMessage, Substream<StreamMuxerBox>>>,
impl Service { impl Service {
pub fn new(config: NetworkConfig, log: slog::Logger) -> Self { pub fn new(config: NetworkConfig, log: slog::Logger) -> Self {
@ -33,14 +30,37 @@ impl Service {
let local_peer_id = local_private_key.to_peer_id(); let local_peer_id = local_private_key.to_peer_id();
debug!(log, "Local peer id: {:?}", local_peer_id); debug!(log, "Local peer id: {:?}", local_peer_id);
let mut swarm = {
// Set up the transport // Set up the transport
let transport = build_transport(local_private_key); let transport = build_transport(local_private_key);
// Set up gossipsub routing // Set up gossipsub routing
let behaviour = Behaviour::new(local_peer_id.clone(), config.gs_config); let behaviour = Behaviour::new(local_peer_id.clone(), config.gs_config);
// Set up Topology // Set up Topology
let topology = local_peer_id.clone(); let topology = local_peer_id.clone();
Swarm::new(transport, behaviour, topology)
};
let swarm = Swarm::new(transport, behaviour, topology); // listen on all addresses
for address in &config.listen_addresses {
match Swarm::listen_on(&mut swarm, address.clone()) {
Ok(mut listen_addr) => {
listen_addr.append(Protocol::P2p(local_peer_id.clone().into()));
info!(log, "Listening on: {}", listen_addr);
}
Err(err) => warn!(log, "Cannot listen on: {} : {:?}", address, err),
};
}
// connect to boot nodes - these are currently stored as multiadders
// Once we have discovery, can set to peerId
for bootnode in config.boot_nodes {
match Swarm::dial_addr(&mut swarm, bootnode.clone()) {
Ok(()) => debug!(log, "Dialing bootnode: {}", bootnode),
Err(err) => debug!(
log,
"Could not connect to bootnode: {} error: {:?}", bootnode, err
),
};
}
Service { Service {
local_peer_id, local_peer_id,