mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-02-26 15:47:21 +00:00
0cba803fba
This changes the p2p protocol handlers to delay message decoding. It's the first part of a larger change that will delay decoding all the way through message processing. For responses, we delay the decoding until it is confirmed that the response matches an active request and does not exceed its limits. In order to make this work, all messages have been changed to use rlp.RawList instead of a slice of the decoded item type. For block bodies specifically, the decoding has been delayed all the way until after verification of the response hash. The role of p2p/tracker.Tracker changes significantly in this PR. The Tracker's original purpose was to maintain metrics about requests and responses in the peer-to-peer protocols. Each protocol maintained a single global Tracker instance. As of this change, the Tracker is now always active (regardless of metrics collection), and there is a separate instance of it for each peer. Whenever a response arrives, it is first verified that a request exists for it in the tracker. The tracker is also the place where limits are kept.
88 lines
2.2 KiB
Go
88 lines
2.2 KiB
Go
// Copyright 2023 The go-ethereum Authors
|
|
// This file is part of go-ethereum.
|
|
//
|
|
// go-ethereum is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// go-ethereum is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package ethtest
|
|
|
|
import (
|
|
"github.com/ethereum/go-ethereum/p2p"
|
|
"github.com/ethereum/go-ethereum/rlp"
|
|
)
|
|
|
|
// Unexported devp2p message codes from p2p/peer.go.
|
|
const (
|
|
handshakeMsg = 0x00
|
|
discMsg = 0x01
|
|
pingMsg = 0x02
|
|
pongMsg = 0x03
|
|
)
|
|
|
|
// Unexported devp2p protocol lengths from p2p package.
|
|
const (
|
|
baseProtoLen = 16
|
|
ethProtoLen = 18
|
|
snapProtoLen = 8
|
|
)
|
|
|
|
// Unexported handshake structure from p2p/peer.go.
|
|
type protoHandshake struct {
|
|
Version uint64
|
|
Name string
|
|
Caps []p2p.Cap
|
|
ListenPort uint64
|
|
ID []byte
|
|
Rest []rlp.RawValue `rlp:"tail"`
|
|
}
|
|
|
|
type Hello = protoHandshake
|
|
|
|
// Proto is an enum representing devp2p protocol types.
|
|
type Proto int
|
|
|
|
const (
|
|
baseProto Proto = iota
|
|
ethProto
|
|
snapProto
|
|
)
|
|
|
|
// getProto returns the protocol a certain message code is associated with
|
|
// (assuming the negotiated capabilities are exactly {eth,snap})
|
|
func getProto(code uint64) Proto {
|
|
switch {
|
|
case code < baseProtoLen:
|
|
return baseProto
|
|
case code < baseProtoLen+ethProtoLen:
|
|
return ethProto
|
|
case code < baseProtoLen+ethProtoLen+snapProtoLen:
|
|
return snapProto
|
|
default:
|
|
panic("unhandled msg code beyond last protocol")
|
|
}
|
|
}
|
|
|
|
// protoOffset will return the offset at which the specified protocol's messages
|
|
// begin.
|
|
func protoOffset(proto Proto) uint64 {
|
|
switch proto {
|
|
case baseProto:
|
|
return 0
|
|
case ethProto:
|
|
return baseProtoLen
|
|
case snapProto:
|
|
return baseProtoLen + ethProtoLen
|
|
default:
|
|
panic("unhandled protocol")
|
|
}
|
|
}
|