use crossbeam_channel::{unbounded, Receiver, RecvTimeoutError, Sender}; use eth2_libp2p::rpc::methods::*; use eth2_libp2p::rpc::{RPCMethod, RPCRequest, RPCResponse, RequestId}; use eth2_libp2p::{PeerId, RPCEvent}; use network::beacon_chain::BeaconChain as NetworkBeaconChain; use network::message_handler::{HandlerMessage, MessageHandler}; use network::service::{NetworkMessage, OutgoingMessage}; use sloggers::terminal::{Destination, TerminalLoggerBuilder}; use sloggers::types::Severity; use sloggers::Build; use std::time::Duration; use test_harness::BeaconChainHarness; use tokio::runtime::TaskExecutor; use types::{test_utils::TestingBeaconStateBuilder, *}; pub struct SyncNode { pub id: usize, sender: Sender, receiver: Receiver, peer_id: PeerId, harness: BeaconChainHarness, } impl SyncNode { fn from_beacon_state_builder( id: usize, executor: &TaskExecutor, state_builder: TestingBeaconStateBuilder, spec: &ChainSpec, logger: slog::Logger, ) -> Self { let harness = BeaconChainHarness::from_beacon_state_builder(state_builder, spec.clone()); let (network_sender, network_receiver) = unbounded(); let message_handler_sender = MessageHandler::spawn( harness.beacon_chain.clone(), network_sender, executor, logger, ) .unwrap(); Self { id, sender: message_handler_sender, receiver: network_receiver, peer_id: PeerId::random(), harness, } } fn increment_beacon_chain_slot(&mut self) { self.harness.increment_beacon_chain_slot(); } fn send(&self, message: HandlerMessage) { self.sender.send(message).unwrap(); } fn recv(&self) -> Result { self.receiver.recv_timeout(Duration::from_millis(500)) } fn hello_message(&self) -> HelloMessage { self.harness.beacon_chain.hello_message() } pub fn connect_to(&mut self, node: &SyncNode) { let message = HandlerMessage::PeerDialed(self.peer_id.clone()); node.send(message); } /// Reads the receive queue from one node and passes the message to the other. Also returns a /// copy of the message. /// /// self -----> node /// | /// us /// /// Named after the unix `tee` command. fn tee(&mut self, node: &SyncNode) -> NetworkMessage { let network_message = self.recv().expect("Timeout on tee"); let handler_message = match network_message.clone() { NetworkMessage::Send(_to_peer_id, OutgoingMessage::RPC(event)) => { HandlerMessage::RPC(self.peer_id.clone(), event) } _ => panic!("tee cannot parse {:?}", network_message), }; node.send(handler_message); network_message } fn tee_hello_request(&mut self, node: &SyncNode) -> HelloMessage { let request = self.tee_rpc_request(node); match request { RPCRequest::Hello(message) => message, _ => panic!("tee_hello_request got: {:?}", request), } } fn tee_hello_response(&mut self, node: &SyncNode) -> HelloMessage { let response = self.tee_rpc_response(node); match response { RPCResponse::Hello(message) => message, _ => panic!("tee_hello_response got: {:?}", response), } } fn tee_block_root_request(&mut self, node: &SyncNode) -> BeaconBlockRootsRequest { let msg = self.tee_rpc_request(node); match msg { RPCRequest::BeaconBlockRoots(data) => data, _ => panic!("tee_block_root_request got: {:?}", msg), } } fn tee_block_root_response(&mut self, node: &SyncNode) -> BeaconBlockRootsResponse { let msg = self.tee_rpc_response(node); match msg { RPCResponse::BeaconBlockRoots(data) => data, _ => panic!("tee_block_root_response got: {:?}", msg), } } fn tee_block_header_request(&mut self, node: &SyncNode) -> BeaconBlockHeadersRequest { let msg = self.tee_rpc_request(node); match msg { RPCRequest::BeaconBlockHeaders(data) => data, _ => panic!("tee_block_header_request got: {:?}", msg), } } fn tee_block_header_response(&mut self, node: &SyncNode) -> BeaconBlockHeadersResponse { let msg = self.tee_rpc_response(node); match msg { RPCResponse::BeaconBlockHeaders(data) => data, _ => panic!("tee_block_header_response got: {:?}", msg), } } fn tee_block_body_request(&mut self, node: &SyncNode) -> BeaconBlockBodiesRequest { let msg = self.tee_rpc_request(node); match msg { RPCRequest::BeaconBlockBodies(data) => data, _ => panic!("tee_block_body_request got: {:?}", msg), } } fn tee_block_body_response(&mut self, node: &SyncNode) -> BeaconBlockBodiesResponse { let msg = self.tee_rpc_response(node); match msg { RPCResponse::BeaconBlockBodies(data) => data, _ => panic!("tee_block_body_response got: {:?}", msg), } } fn tee_rpc_request(&mut self, node: &SyncNode) -> RPCRequest { let network_message = self.tee(node); match network_message { NetworkMessage::Send( _peer_id, OutgoingMessage::RPC(RPCEvent::Request { id: _, method_id: _, body, }), ) => body, _ => panic!("tee_rpc_request failed! got {:?}", network_message), } } fn tee_rpc_response(&mut self, node: &SyncNode) -> RPCResponse { let network_message = self.tee(node); match network_message { NetworkMessage::Send( _peer_id, OutgoingMessage::RPC(RPCEvent::Response { id: _, method_id: _, result, }), ) => result, _ => panic!("tee_rpc_response failed! got {:?}", network_message), } } pub fn get_block_root_request(&self) -> BeaconBlockRootsRequest { let request = self.recv_rpc_request().expect("No block root request"); match request { RPCRequest::BeaconBlockRoots(request) => request, _ => panic!("Did not get block root request"), } } pub fn get_block_headers_request(&self) -> BeaconBlockHeadersRequest { let request = self.recv_rpc_request().expect("No block headers request"); match request { RPCRequest::BeaconBlockHeaders(request) => request, _ => panic!("Did not get block headers request"), } } pub fn get_block_bodies_request(&self) -> BeaconBlockBodiesRequest { let request = self.recv_rpc_request().expect("No block bodies request"); match request { RPCRequest::BeaconBlockBodies(request) => request, _ => panic!("Did not get block bodies request"), } } fn _recv_rpc_response(&self) -> Result { let network_message = self.recv()?; Ok(match network_message { NetworkMessage::Send( _peer_id, OutgoingMessage::RPC(RPCEvent::Response { id: _, method_id: _, result, }), ) => result, _ => panic!("get_rpc_response failed! got {:?}", network_message), }) } fn recv_rpc_request(&self) -> Result { let network_message = self.recv()?; Ok(match network_message { NetworkMessage::Send( _peer_id, OutgoingMessage::RPC(RPCEvent::Request { id: _, method_id: _, body, }), ) => body, _ => panic!("get_rpc_request failed! got {:?}", network_message), }) } } fn get_logger() -> slog::Logger { let mut builder = TerminalLoggerBuilder::new(); builder.level(Severity::Debug); builder.destination(Destination::Stderr); builder.build().unwrap() } pub struct SyncMaster { harness: BeaconChainHarness, peer_id: PeerId, response_ids: Vec, } impl SyncMaster { fn from_beacon_state_builder( state_builder: TestingBeaconStateBuilder, node_count: usize, spec: &ChainSpec, ) -> Self { let harness = BeaconChainHarness::from_beacon_state_builder(state_builder, spec.clone()); let peer_id = PeerId::random(); let response_ids = vec![RequestId::from(0); node_count]; Self { harness, peer_id, response_ids, } } pub fn response_id(&mut self, node: &SyncNode) -> RequestId { let id = self.response_ids[node.id].clone(); self.response_ids[node.id].increment(); id } pub fn do_hello_with(&mut self, node: &SyncNode) { let message = HandlerMessage::PeerDialed(self.peer_id.clone()); node.send(message); let request = node.recv_rpc_request().expect("No hello response"); match request { RPCRequest::Hello(_hello) => { let hello = self.harness.beacon_chain.hello_message(); let response = self.rpc_response(node, RPCResponse::Hello(hello)); node.send(response); } _ => panic!("Got message other than hello from node."), } } pub fn respond_to_block_roots_request( &mut self, node: &SyncNode, request: BeaconBlockRootsRequest, ) { let roots = self .harness .beacon_chain .get_block_roots(request.start_slot, request.count as usize, 0) .expect("Beacon chain did not give block roots") .iter() .enumerate() .map(|(i, root)| BlockRootSlot { block_root: *root, slot: Slot::from(i) + request.start_slot, }) .collect(); let response = RPCResponse::BeaconBlockRoots(BeaconBlockRootsResponse { roots }); self.send_rpc_response(node, response) } pub fn respond_to_block_headers_request( &mut self, node: &SyncNode, request: BeaconBlockHeadersRequest, ) { let roots = self .harness .beacon_chain .get_block_roots( request.start_slot, request.max_headers as usize, request.skip_slots as usize, ) .expect("Beacon chain did not give blocks"); if roots.is_empty() { panic!("Roots was empty when trying to get headers.") } assert_eq!( roots[0], request.start_root, "Got the wrong start root when getting headers" ); let headers: Vec = roots .iter() .map(|root| { let block = self .harness .beacon_chain .get_block(root) .expect("Failed to load block") .expect("Block did not exist"); block.block_header() }) .collect(); let response = RPCResponse::BeaconBlockHeaders(BeaconBlockHeadersResponse { headers }); self.send_rpc_response(node, response) } pub fn respond_to_block_bodies_request( &mut self, node: &SyncNode, request: BeaconBlockBodiesRequest, ) { let block_bodies: Vec = request .block_roots .iter() .map(|root| { let block = self .harness .beacon_chain .get_block(root) .expect("Failed to load block") .expect("Block did not exist"); block.body }) .collect(); let response = RPCResponse::BeaconBlockBodies(BeaconBlockBodiesResponse { block_bodies }); self.send_rpc_response(node, response) } fn send_rpc_response(&mut self, node: &SyncNode, rpc_response: RPCResponse) { node.send(self.rpc_response(node, rpc_response)); } fn rpc_response(&mut self, node: &SyncNode, rpc_response: RPCResponse) -> HandlerMessage { HandlerMessage::RPC( self.peer_id.clone(), RPCEvent::Response { id: self.response_id(node), method_id: RPCMethod::Hello.into(), result: rpc_response, }, ) } } fn test_setup( state_builder: TestingBeaconStateBuilder, node_count: usize, spec: &ChainSpec, logger: slog::Logger, ) -> (tokio::runtime::Runtime, SyncMaster, Vec) { let runtime = tokio::runtime::Runtime::new().unwrap(); let mut nodes = Vec::with_capacity(node_count); for id in 0..node_count { let node = SyncNode::from_beacon_state_builder( id, &runtime.executor(), state_builder.clone(), &spec, logger.clone(), ); nodes.push(node); } let master = SyncMaster::from_beacon_state_builder(state_builder, node_count, &spec); (runtime, master, nodes) } pub fn build_blocks(blocks: usize, master: &mut SyncMaster, nodes: &mut Vec) { for _ in 0..blocks { master.harness.advance_chain_with_block(); for i in 0..nodes.len() { nodes[i].increment_beacon_chain_slot(); } } master.harness.run_fork_choice(); for i in 0..nodes.len() { nodes[i].harness.run_fork_choice(); } } #[test] #[ignore] fn sync_node_with_master() { let logger = get_logger(); let spec = ChainSpec::few_validators(); let validator_count = 8; let node_count = 1; let state_builder = TestingBeaconStateBuilder::from_default_keypairs_file_if_exists(validator_count, &spec); let (runtime, mut master, mut nodes) = test_setup(state_builder, node_count, &spec, logger.clone()); let original_node_slot = nodes[0].hello_message().best_slot; build_blocks(2, &mut master, &mut nodes); master.do_hello_with(&nodes[0]); let roots_request = nodes[0].get_block_root_request(); assert_eq!(roots_request.start_slot, original_node_slot + 1); assert_eq!(roots_request.count, 2); master.respond_to_block_roots_request(&nodes[0], roots_request); let headers_request = nodes[0].get_block_headers_request(); assert_eq!(headers_request.start_slot, original_node_slot + 1); assert_eq!(headers_request.max_headers, 2); assert_eq!(headers_request.skip_slots, 0); master.respond_to_block_headers_request(&nodes[0], headers_request); let bodies_request = nodes[0].get_block_bodies_request(); assert_eq!(bodies_request.block_roots.len(), 2); master.respond_to_block_bodies_request(&nodes[0], bodies_request); std::thread::sleep(Duration::from_millis(10000)); runtime.shutdown_now(); } #[test] #[ignore] fn sync_two_nodes() { let logger = get_logger(); let spec = ChainSpec::few_validators(); let validator_count = 8; let node_count = 2; let state_builder = TestingBeaconStateBuilder::from_default_keypairs_file_if_exists(validator_count, &spec); let (runtime, _master, mut nodes) = test_setup(state_builder, node_count, &spec, logger.clone()); // let original_node_slot = nodes[0].hello_message().best_slot; let mut node_a = nodes.remove(0); let mut node_b = nodes.remove(0); let blocks = 2; // Node A builds out a longer, better chain. for _ in 0..blocks { // Node A should build a block. node_a.harness.advance_chain_with_block(); // Node B should just increment it's slot without a block. node_b.harness.increment_beacon_chain_slot(); } node_a.harness.run_fork_choice(); // A connects to B. node_a.connect_to(&node_b); // B says hello to A. node_b.tee_hello_request(&node_a); // A says hello back. node_a.tee_hello_response(&node_b); // B requests block roots from A. node_b.tee_block_root_request(&node_a); // A provides block roots to A. node_a.tee_block_root_response(&node_b); // B requests block headers from A. node_b.tee_block_header_request(&node_a); // A provides block headers to B. node_a.tee_block_header_response(&node_b); // B requests block bodies from A. node_b.tee_block_body_request(&node_a); // A provides block bodies to B. node_a.tee_block_body_response(&node_b); std::thread::sleep(Duration::from_secs(10)); node_b.harness.run_fork_choice(); let node_a_chain = node_a .harness .beacon_chain .chain_dump() .expect("Can't dump node a chain"); let node_b_chain = node_b .harness .beacon_chain .chain_dump() .expect("Can't dump node b chain"); assert_eq!( node_a_chain.len(), node_b_chain.len(), "Chains should be equal length" ); assert_eq!(node_a_chain, node_b_chain, "Chains should be identical"); runtime.shutdown_now(); }