diff --git a/eth/protocols/eth/handler.go b/eth/protocols/eth/handler.go index 154b75130c..325ffa0738 100644 --- a/eth/protocols/eth/handler.go +++ b/eth/protocols/eth/handler.go @@ -224,10 +224,10 @@ func handleMessage(backend Backend, peer *Peer) error { if err != nil { return err } + defer msg.Discard() if msg.Size > maxMessageSize { return fmt.Errorf("%w: %v > %v", errMsgTooLarge, msg.Size, maxMessageSize) } - defer msg.Discard() var handlers map[uint64]msgHandler switch peer.version { diff --git a/eth/protocols/snap/handler.go b/eth/protocols/snap/handler.go index ff5564ca65..6c2ff6149f 100644 --- a/eth/protocols/snap/handler.go +++ b/eth/protocols/snap/handler.go @@ -160,10 +160,10 @@ func HandleMessage(backend Backend, peer *Peer) error { if err != nil { return err } + defer msg.Discard() if msg.Size > maxMessageSize { return fmt.Errorf("%w: %v > %v", errMsgTooLarge, msg.Size, maxMessageSize) } - defer msg.Discard() var handlers map[uint64]msgHandler switch peer.version {