github/reth
1
1
Fork 0
mirror of https://github.com/paradigmxyz/reth.git synced 2026-04-30 03:01:58 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix: Returns an error if multiplex message is empty (#7314)

Browse Source
This commit is contained in:
Krishang
2024-03-25 16:59:41 +05:30
committed by GitHub
parent b58cca7f91
commit 61801b98b6
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
crates/net/eth-wire/src/multiplex.rs
View File

@@ -166,7 +166,10 @@ impl<St> RlpxProtocolMultiplexer<St> {
tokio::select! {
Some(Ok(msg)) = self.inner.conn.next() => {
// Ensure the message belongs to the primary protocol
let offset = msg[0];
let Some(offset) = msg.first().copied()
else {
return Err(P2PStreamError::EmptyProtocolMessage.into())
};
if let Some(cap) = self.shared_capabilities().find_by_relative_offset(offset).cloned() {
if cap == shared_cap {
// delegate to primary
@@ -518,7 +521,11 @@ where
match this.inner.conn.poll_next_unpin(cx) {
Poll::Ready(Some(Ok(msg))) => {
delegated = true;
let offset = msg[0];
let Some(offset) = msg.first().copied() else {
return Poll::Ready(Some(Err(
P2PStreamError::EmptyProtocolMessage.into()
)))
};
// delegate the multiplexed message to the correct protocol
if let Some(cap) =
this.inner.conn.shared_capabilities().find_by_relative_offset(offset)