cerc-io/lighthouse
16
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

Basic node handshake

Browse Source
This commit is contained in:
Age Manning 2019-03-19 13:25:25 +11:00
parent 67c09021f0
commit 5ae8079b44
No known key found for this signature in database
GPG Key ID: 05EED64B79E06A93
1 changed files with 21 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
beacon_node/network/src/message_handler.rs
View File

@ -19,6 +19,8 @@ use types::Hash256;
/// Timeout for RPC requests. /// Timeout for RPC requests.
const REQUEST_TIMEOUT: Duration = Duration::from_secs(30); const REQUEST_TIMEOUT: Duration = Duration::from_secs(30);
/// Timeout before banning a peer for non-identification.
const HELLO_TIMEOUT: Duration = Duration::from_secs(30);
/// Handles messages received from the network and client and organises syncing. /// Handles messages received from the network and client and organises syncing.
pub struct MessageHandler { pub struct MessageHandler {
@ -28,22 +30,17 @@ pub struct MessageHandler {
sync: SimpleSync, sync: SimpleSync,
/// The network channel to relay messages to the Network service. /// The network channel to relay messages to the Network service.
network_send: crossbeam_channel::Sender<NetworkMessage>, network_send: crossbeam_channel::Sender<NetworkMessage>,
/// A mapping of peers we have sent an RPC request to. /// A mapping of peers and the RPC id we have sent an RPC request to.
requests: HashMap<PeerId, Vec<RPCRequestInfo>>, requests: HashMap<(PeerId, u64), Instant>,
/// A mapping of HELLO requests we have sent. We drop/ban peers if they do not response
/// within the timeout
hello_requests: HashMap<PeerId, Instant>,
/// A counter of request id for each peer. /// A counter of request id for each peer.
request_ids: HashMap<PeerId, u64>, request_ids: HashMap<PeerId, u64>,
/// The `MessageHandler` logger. /// The `MessageHandler` logger.
log: slog::Logger, log: slog::Logger,
} }
/// RPC request information
pub struct RPCRequestInfo {
/// The id of the request
id: u64,
/// The time the request was sent, to check ttl.
request_time: Instant,
}
/// Types of messages the handler can receive. /// Types of messages the handler can receive.
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
pub enum HandlerMessage { pub enum HandlerMessage {
@ -79,7 +76,9 @@ impl MessageHandler {
sync, sync,
network_send, network_send,
requests: HashMap::new(), requests: HashMap::new(),
hello_requests: HashMap::new(),
request_ids: HashMap::new(), request_ids: HashMap::new(),
log: log.clone(), log: log.clone(),
}; };
@ -140,15 +139,13 @@ impl MessageHandler {
fn handle_rpc_response(&mut self, peer_id: PeerId, id: u64, response: RPCResponse) {} fn handle_rpc_response(&mut self, peer_id: PeerId, id: u64, response: RPCResponse) {}
fn handle_hello_response(&mut self, peer_id: PeerId, id: u64, response: HelloMessage) { fn handle_hello_response(&mut self, peer_id: PeerId, id: u64, response: HelloMessage) {
/* if self.hello_requests.remove(&peer_id).is_none() {
// if response id is not in our list, ignore (likely RPC timeout) // if response id is not in our list, ignore (likely RPC timeout)
match self.requests.get(peer_id) { return;
None => return; }
Some(rpc_info) => {
if rpc_info.con
*/
debug!(self.log, "Hello response received from peer: {:?}", peer_id);
// validate peer - decide whether to drop/ban or add to sync
} }
/// Sends a HELLO RPC request to a newly connected peer. /// Sends a HELLO RPC request to a newly connected peer.
@ -161,17 +158,12 @@ impl MessageHandler {
*borrowed_id += 1; *borrowed_id += 1;
id id
}; };
// register RPC request // register RPC Hello request
{ self.requests.insert((peer_id.clone(), id), Instant::now());
let requests = self debug!(
.requests self.log,
.entry(peer_id.clone()) "Hello request registered with peer: {:?}", peer_id
.or_insert_with(|| vec![]); );
requests.push(RPCRequestInfo {
id: id.clone(),
request_time: Instant::now(),
});
}
// build the rpc request // build the rpc request
let rpc_event = RPCEvent::Request { let rpc_event = RPCEvent::Request {