Account manager, bootnodes, RPC display and sync fixes

This commit is contained in:
Age Manning 2019-09-03 07:50:44 +10:00
parent cd7b6da88e
commit 13b5df56b3
No known key found for this signature in database
GPG Key ID: 05EED64B79E06A93
10 changed files with 168 additions and 73 deletions

View File

@ -125,9 +125,13 @@ fn main() {
} }
} }
} }
_ => panic!( _ => {
crit!(
log,
"The account manager must be run with a subcommand. See help for more information." "The account manager must be run with a subcommand. See help for more information."
), );
return;
}
} }
} }

View File

@ -114,7 +114,7 @@ impl<TSubstream> Discovery<TSubstream> {
self.find_peers(); self.find_peers();
} }
/// Add an Enr to the routing table of the discovery mechanism. /// Add an ENR to the routing table of the discovery mechanism.
pub fn add_enr(&mut self, enr: Enr) { pub fn add_enr(&mut self, enr: Enr) {
self.discovery.add_enr(enr); self.discovery.add_enr(enr);
} }

View File

@ -157,3 +157,51 @@ impl ErrorMessage {
String::from_utf8(self.error_message.clone()).unwrap_or_else(|_| "".into()) String::from_utf8(self.error_message.clone()).unwrap_or_else(|_| "".into())
} }
} }
impl std::fmt::Display for HelloMessage {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "Hello Message: Fork Version: {:?}, Finalized Root: {}, Finalized Epoch: {}, Head Root: {}, Head Slot: {}", self.fork_version, self.finalized_root, self.finalized_epoch, self.head_root, self.head_slot)
}
}
impl std::fmt::Display for RPCResponse {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self {
RPCResponse::Hello(hello) => write!(f, "{}", hello),
RPCResponse::BeaconBlocks(_) => write!(f, "<BeaconBlocks>"),
RPCResponse::RecentBeaconBlocks(_) => write!(f, "<RecentBeaconBlocks>"),
}
}
}
impl std::fmt::Display for RPCErrorResponse {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self {
RPCErrorResponse::Success(res) => write!(f, "{}", res),
RPCErrorResponse::InvalidRequest(err) => write!(f, "Invalid Request: {:?}", err),
RPCErrorResponse::ServerError(err) => write!(f, "Server Error: {:?}", err),
RPCErrorResponse::Unknown(err) => write!(f, "Unknown Error: {:?}", err),
}
}
}
impl std::fmt::Display for GoodbyeReason {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self {
GoodbyeReason::ClientShutdown => write!(f, "Client Shutdown"),
GoodbyeReason::IrrelevantNetwork => write!(f, "Irrelevant Network"),
GoodbyeReason::Fault => write!(f, "Fault"),
GoodbyeReason::Unknown => write!(f, "Unknown Reason"),
}
}
}
impl std::fmt::Display for BeaconBlocksRequest {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(
f,
"Head Block Root: {}, Start Slot: {}, Count: {}, Step: {}",
self.head_block_root, self.start_slot, self.count, self.step
)
}
}

View File

@ -47,6 +47,16 @@ impl RPCEvent {
} }
} }
impl std::fmt::Display for RPCEvent {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self {
RPCEvent::Request(id, req) => write!(f, "RPC Request(Id: {}, {})", id, req),
RPCEvent::Response(id, res) => write!(f, "RPC Response(Id: {}, {})", id, res),
RPCEvent::Error(id, err) => write!(f, "RPC Request(Id: {}, Error: {:?})", id, err),
}
}
}
/// Implements the libp2p `NetworkBehaviour` trait and therefore manages network-level /// Implements the libp2p `NetworkBehaviour` trait and therefore manages network-level
/// logic. /// logic.
pub struct RPC<TSubstream> { pub struct RPC<TSubstream> {

View File

@ -288,3 +288,14 @@ impl std::error::Error for RPCError {
} }
} }
} }
impl std::fmt::Display for RPCRequest {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self {
RPCRequest::Hello(hello) => write!(f, "Hello Message: {}", hello),
RPCRequest::Goodbye(reason) => write!(f, "Goodbye: {}", reason),
RPCRequest::BeaconBlocks(req) => write!(f, "Beacon Blocks: {}", req),
RPCRequest::RecentBeaconBlocks(req) => write!(f, "Recent Beacon Blocks: {:?}", req),
}
}
}

View File

@ -79,8 +79,8 @@ impl Service {
} }
}; };
// attempt to connect to user-input libp2p nodes // helper closure for dialing peers
for multiaddr in config.libp2p_nodes { let mut dial_addr = |multiaddr: Multiaddr| {
match Swarm::dial_addr(&mut swarm, multiaddr.clone()) { match Swarm::dial_addr(&mut swarm, multiaddr.clone()) {
Ok(()) => debug!(log, "Dialing libp2p peer"; "address" => format!("{}", multiaddr)), Ok(()) => debug!(log, "Dialing libp2p peer"; "address" => format!("{}", multiaddr)),
Err(err) => debug!( Err(err) => debug!(
@ -88,6 +88,18 @@ impl Service {
"Could not connect to peer"; "address" => format!("{}", multiaddr), "error" => format!("{:?}", err) "Could not connect to peer"; "address" => format!("{}", multiaddr), "error" => format!("{:?}", err)
), ),
}; };
};
// attempt to connect to user-input libp2p nodes
for multiaddr in config.libp2p_nodes {
dial_addr(multiaddr);
}
// attempt to connect to any specified boot-nodes
for bootnode_enr in config.boot_nodes {
for multiaddr in bootnode_enr.multiaddr() {
dial_addr(multiaddr);
}
} }
// subscribe to default gossipsub topics // subscribe to default gossipsub topics

View File

@ -161,7 +161,7 @@ fn network_service(
Ok(Async::Ready(Some(message))) => match message { Ok(Async::Ready(Some(message))) => match message {
NetworkMessage::Send(peer_id, outgoing_message) => match outgoing_message { NetworkMessage::Send(peer_id, outgoing_message) => match outgoing_message {
OutgoingMessage::RPC(rpc_event) => { OutgoingMessage::RPC(rpc_event) => {
trace!(log, "Sending RPC Event: {:?}", rpc_event); trace!(log, "{}", rpc_event);
libp2p_service.lock().swarm.send_rpc(peer_id, rpc_event); libp2p_service.lock().swarm.send_rpc(peer_id, rpc_event);
} }
}, },
@ -185,7 +185,7 @@ fn network_service(
match libp2p_service.lock().poll() { match libp2p_service.lock().poll() {
Ok(Async::Ready(Some(event))) => match event { Ok(Async::Ready(Some(event))) => match event {
Libp2pEvent::RPC(peer_id, rpc_event) => { Libp2pEvent::RPC(peer_id, rpc_event) => {
trace!(log, "RPC Event: RPC message received: {:?}", rpc_event); trace!(log, "{}", rpc_event);
message_handler_send message_handler_send
.try_send(HandlerMessage::RPC(peer_id, rpc_event)) .try_send(HandlerMessage::RPC(peer_id, rpc_event))
.map_err(|_| "Failed to send RPC to handler")?; .map_err(|_| "Failed to send RPC to handler")?;

View File

@ -68,7 +68,7 @@ use types::{BeaconBlock, EthSpec, Hash256, Slot};
/// Blocks are downloaded in batches from peers. This constant specifies how many blocks per batch /// Blocks are downloaded in batches from peers. This constant specifies how many blocks per batch
/// is requested. Currently the value is small for testing. This will be incremented for /// is requested. Currently the value is small for testing. This will be incremented for
/// production. /// production.
const MAX_BLOCKS_PER_REQUEST: u64 = 10; const MAX_BLOCKS_PER_REQUEST: u64 = 100;
/// The number of slots ahead of us that is allowed before requesting a long-range (batch) Sync /// The number of slots ahead of us that is allowed before requesting a long-range (batch) Sync
/// from a peer. If a peer is within this tolerance (forwards or backwards), it is treated as a /// from a peer. If a peer is within this tolerance (forwards or backwards), it is treated as a
@ -120,6 +120,8 @@ struct BlockRequests<T: EthSpec> {
target_head_root: Hash256, target_head_root: Hash256,
/// The blocks that we have currently downloaded from the peer that are yet to be processed. /// The blocks that we have currently downloaded from the peer that are yet to be processed.
downloaded_blocks: Vec<BeaconBlock<T>>, downloaded_blocks: Vec<BeaconBlock<T>>,
/// The number of blocks successfully processed in this request.
blocks_processed: usize,
/// The number of empty batches we have consecutively received. If a peer returns more than /// The number of empty batches we have consecutively received. If a peer returns more than
/// EMPTY_BATCHES_TOLERANCE, they are dropped. /// EMPTY_BATCHES_TOLERANCE, they are dropped.
consecutive_empty_batches: usize, consecutive_empty_batches: usize,
@ -302,6 +304,7 @@ impl<T: BeaconChainTypes> ImportManager<T> {
target_head_root: remote.head_root, target_head_root: remote.head_root,
consecutive_empty_batches: 0, consecutive_empty_batches: 0,
downloaded_blocks: Vec::new(), downloaded_blocks: Vec::new(),
blocks_processed: 0,
state: BlockRequestsState::Queued, state: BlockRequestsState::Queued,
sync_direction: SyncDirection::Initial, sync_direction: SyncDirection::Initial,
current_start_slot: chain.best_slot(), current_start_slot: chain.best_slot(),
@ -356,6 +359,10 @@ impl<T: BeaconChainTypes> ImportManager<T> {
warn!(self.log, "Peer returned too many empty block batches"; warn!(self.log, "Peer returned too many empty block batches";
"peer" => format!("{:?}", peer_id)); "peer" => format!("{:?}", peer_id));
block_requests.state = BlockRequestsState::Failed; block_requests.state = BlockRequestsState::Failed;
} else if block_requests.current_start_slot >= block_requests.target_head_slot {
warn!(self.log, "Peer did not return blocks it claimed to possess";
"peer" => format!("{:?}", peer_id));
block_requests.state = BlockRequestsState::Failed;
} else { } else {
block_requests.update_start_slot(); block_requests.update_start_slot();
} }
@ -561,19 +568,19 @@ impl<T: BeaconChainTypes> ImportManager<T> {
// only process batch requests if there are any // only process batch requests if there are any
if !self.import_queue.is_empty() { if !self.import_queue.is_empty() {
// process potential block requests // process potential block requests
self.process_potential_block_requests(); re_run = re_run || self.process_potential_block_requests();
// process any complete long-range batches // process any complete long-range batches
re_run = self.process_complete_batches(); re_run = re_run || self.process_complete_batches();
} }
// only process parent objects if we are in regular sync // only process parent objects if we are in regular sync
if let ManagerState::Regular = self.state { if !self.parent_queue.is_empty() {
// process any parent block lookup-requests // process any parent block lookup-requests
self.process_parent_requests(); re_run = re_run || self.process_parent_requests();
// process any complete parent lookups // process any complete parent lookups
re_run = self.process_complete_parent_requests(); re_run = re_run || self.process_complete_parent_requests();
} }
// return any queued events // return any queued events
@ -613,20 +620,23 @@ impl<T: BeaconChainTypes> ImportManager<T> {
} }
} }
fn process_potential_block_requests(&mut self) { fn process_potential_block_requests(&mut self) -> bool {
// check if an outbound request is required // check if an outbound request is required
// Managing a fixed number of outbound requests is maintained at the RPC protocol libp2p // Managing a fixed number of outbound requests is maintained at the RPC protocol libp2p
// layer and not needed here. Therefore we create many outbound requests and let the RPC // layer and not needed here. Therefore we create many outbound requests and let the RPC
// handle the number of simultaneous requests. Request all queued objects. // handle the number of simultaneous requests. Request all queued objects.
let mut re_run = false;
// remove any failed batches // remove any failed batches
let debug_log = &self.log; let debug_log = &self.log;
let full_peer_ref = &mut self.full_peers;
self.import_queue.retain(|peer_id, block_request| { self.import_queue.retain(|peer_id, block_request| {
if let BlockRequestsState::Failed = block_request.state { if let BlockRequestsState::Failed = block_request.state {
debug!(debug_log, "Block import from peer failed"; debug!(debug_log, "Block import from peer failed";
"peer_id" => format!("{:?}", peer_id), "peer_id" => format!("{:?}", peer_id),
"downloaded_blocks" => block_request.downloaded_blocks.len() "downloaded_blocks" => block_request.blocks_processed
); );
full_peer_ref.remove(peer_id);
false false
} else { } else {
true true
@ -654,7 +664,10 @@ impl<T: BeaconChainTypes> ImportManager<T> {
request, request,
request_id, request_id,
}); });
re_run = true;
} }
re_run
} }
fn process_complete_batches(&mut self) -> bool { fn process_complete_batches(&mut self) -> bool {
@ -667,13 +680,14 @@ impl<T: BeaconChainTypes> ImportManager<T> {
let event_queue_ref = &mut self.event_queue; let event_queue_ref = &mut self.event_queue;
self.import_queue.retain(|peer_id, block_requests| { self.import_queue.retain(|peer_id, block_requests| {
if block_requests.state == BlockRequestsState::ReadyToProcess {
// check that the chain still exists // check that the chain still exists
if let Some(chain) = chain_ref.upgrade() { if let Some(chain) = chain_ref.upgrade() {
let downloaded_blocks = let downloaded_blocks =
std::mem::replace(&mut block_requests.downloaded_blocks, Vec::new()); std::mem::replace(&mut block_requests.downloaded_blocks, Vec::new());
let last_element = block_requests.downloaded_blocks.len() - 1; let last_element = downloaded_blocks.len() - 1;
let start_slot = block_requests.downloaded_blocks[0].slot; let start_slot = downloaded_blocks[0].slot;
let end_slot = block_requests.downloaded_blocks[last_element].slot; let end_slot = downloaded_blocks[last_element].slot;
match process_blocks(chain, downloaded_blocks, log_ref) { match process_blocks(chain, downloaded_blocks, log_ref) {
Ok(()) => { Ok(()) => {
@ -683,6 +697,7 @@ impl<T: BeaconChainTypes> ImportManager<T> {
"end_slot" => end_slot, "end_slot" => end_slot,
"no_blocks" => last_element + 1, "no_blocks" => last_element + 1,
); );
block_requests.blocks_processed += last_element + 1;
// check if the batch is complete, by verifying if we have reached the // check if the batch is complete, by verifying if we have reached the
// target head // target head
@ -707,7 +722,8 @@ impl<T: BeaconChainTypes> ImportManager<T> {
"no_blocks" => last_element + 1, "no_blocks" => last_element + 1,
"error" => format!("{:?}", e), "error" => format!("{:?}", e),
); );
event_queue_ref.push(ImportManagerOutcome::DownvotePeer(peer_id.clone())); event_queue_ref
.push(ImportManagerOutcome::DownvotePeer(peer_id.clone()));
false false
} }
} }
@ -716,17 +732,23 @@ impl<T: BeaconChainTypes> ImportManager<T> {
event_queue_ref.clear(); event_queue_ref.clear();
return false; return false;
} }
} else {
// not ready to process
true
}
}); });
re_run re_run
} }
fn process_parent_requests(&mut self) { fn process_parent_requests(&mut self) -> bool {
// check to make sure there are peers to search for the parent from // check to make sure there are peers to search for the parent from
if self.full_peers.is_empty() { if self.full_peers.is_empty() {
return; return false;
} }
let mut re_run = false;
// remove any failed requests // remove any failed requests
let debug_log = &self.log; let debug_log = &self.log;
self.parent_queue.retain(|parent_request| { self.parent_queue.retain(|parent_request| {
@ -766,8 +788,10 @@ impl<T: BeaconChainTypes> ImportManager<T> {
self.event_queue self.event_queue
.push(ImportManagerOutcome::RecentRequest(peer_id.clone(), req)); .push(ImportManagerOutcome::RecentRequest(peer_id.clone(), req));
re_run = true;
} }
} }
re_run
} }
fn process_complete_parent_requests(&mut self) -> bool { fn process_complete_parent_requests(&mut self) -> bool {

View File

@ -453,7 +453,7 @@ impl<T: BeaconChainTypes> SimpleSync<T> {
} }
BlockProcessingOutcome::ParentUnknown { parent: _ } => { BlockProcessingOutcome::ParentUnknown { parent: _ } => {
// Inform the sync manager to find parents for this block // Inform the sync manager to find parents for this block
trace!(self.log, "Unknown parent gossip"; trace!(self.log, "Block with unknown parent received";
"peer_id" => format!("{:?}",peer_id)); "peer_id" => format!("{:?}",peer_id));
self.manager.add_unknown_block(block.clone(), peer_id); self.manager.add_unknown_block(block.clone(), peer_id);
SHOULD_FORWARD_GOSSIP_BLOCK SHOULD_FORWARD_GOSSIP_BLOCK

View File

@ -187,13 +187,6 @@ fn main() {
.possible_values(&["info", "debug", "trace", "warn", "error", "crit"]) .possible_values(&["info", "debug", "trace", "warn", "error", "crit"])
.default_value("trace"), .default_value("trace"),
) )
.arg(
Arg::with_name("verbosity")
.short("v")
.multiple(true)
.help("Sets the verbosity level")
.takes_value(true),
)
/* /*
* The "testnet" sub-command. * The "testnet" sub-command.
* *
@ -332,13 +325,6 @@ fn main() {
_ => unreachable!("guarded by clap"), _ => unreachable!("guarded by clap"),
}; };
let drain = match matches.occurrences_of("verbosity") {
0 => drain.filter_level(Level::Info),
1 => drain.filter_level(Level::Debug),
2 => drain.filter_level(Level::Trace),
_ => drain.filter_level(Level::Trace),
};
let log = slog::Logger::root(drain.fuse(), o!()); let log = slog::Logger::root(drain.fuse(), o!());
warn!( warn!(