diff --git a/substrate/core/network/src/protocol.rs b/substrate/core/network/src/protocol.rs index 9521627536..b736b4b67a 100644 --- a/substrate/core/network/src/protocol.rs +++ b/substrate/core/network/src/protocol.rs @@ -576,6 +576,7 @@ impl, H: ExHashT> Protocol { self.on_finality_proof_request(who, request), GenericMessage::FinalityProofResponse(response) => return self.on_finality_proof_response(who, response), + GenericMessage::RemoteReadChildRequest(_) => {} GenericMessage::Consensus(msg) => { if self.context_data.peers.get(&who).map_or(false, |peer| peer.info.protocol_version > 2) { self.consensus_gossip.on_incoming( @@ -585,10 +586,10 @@ impl, H: ExHashT> Protocol { ); } } - other => self.specialization.on_message( + GenericMessage::ChainSpecific(msg) => self.specialization.on_message( &mut ProtocolContext::new(&mut self.context_data, &mut self.behaviour, &self.peerset_handle), who, - &mut Some(other), + msg, ), } diff --git a/substrate/core/network/src/protocol/specialization.rs b/substrate/core/network/src/protocol/specialization.rs index 085f201a45..7f6b7dc44f 100644 --- a/substrate/core/network/src/protocol/specialization.rs +++ b/substrate/core/network/src/protocol/specialization.rs @@ -38,7 +38,7 @@ pub trait NetworkSpecialization: Send + Sync + 'static { &mut self, ctx: &mut dyn Context, who: PeerId, - message: &mut Option> + message: Vec ); /// Called when a network-specific event arrives. @@ -130,7 +130,7 @@ macro_rules! construct_simple_protocol { &mut self, _ctx: &mut $crate::Context<$block>, _who: $crate::PeerId, - _message: &mut Option<$crate::message::Message<$block>> + _message: Vec, ) { $( self.$sub_protocol_name.on_message(_ctx, _who, _message); )* } diff --git a/substrate/core/network/src/test/mod.rs b/substrate/core/network/src/test/mod.rs index 7fc626f593..3f92696d27 100644 --- a/substrate/core/network/src/test/mod.rs +++ b/substrate/core/network/src/test/mod.rs @@ -114,7 +114,7 @@ impl NetworkSpecialization for DummySpecialization { &mut self, _ctx: &mut dyn Context, _peer_id: PeerId, - _message: &mut Option>, + _message: Vec, ) {} fn on_event(