forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-07-24 05:36:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

eth: use salted hash function for tx broadcast

Browse source
This commit is contained in:
Felix Lange 2025-08-28 12:17:24 +02:00
parent 3da045cccf
commit b9bd4509b6
2 changed files with 20 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
eth/handler.go
View file

@ -18,8 +18,8 @@ package eth
import (
"cmp"
crand "crypto/rand"
"errors"
"hash/fnv"
"maps"
"math"
"slices"
@ -27,6 +27,7 @@ import (
"sync/atomic"
"time"
"github.com/dchest/siphash"
"github.com/ethereum/go-ethereum/common"
"github.com/ethereum/go-ethereum/core"
"github.com/ethereum/go-ethereum/core/rawdb"
@ -119,9 +120,10 @@ type handler struct {
chain *core.BlockChain
maxPeers int
downloader *downloader.Downloader
txFetcher *fetcher.TxFetcher
peers *peerSet
downloader *downloader.Downloader
txFetcher *fetcher.TxFetcher
peers *peerSet
txBroadcastKey [16]byte
eventMux *event.TypeMux
txsCh chan core.NewTxsEvent
@ -203,6 +205,7 @@ func newHandler(config *handlerConfig) (*handler, error) {
addTxs := func(txs []*types.Transaction) []error {
return h.txpool.Add(txs, false)
}
h.txBroadcastKey = newBroadcastChoiceKey()
h.txFetcher = fetcher.NewTxFetcher(h.txpool.Has, addTxs, fetchTx, h.removePeer)
return h, nil
}
@ -482,7 +485,7 @@ func (h *handler) BroadcastTransactions(txs types.Transactions) {
annos = make(map[*ethPeer][]common.Hash) // Set peer->hash to announce
signer = types.LatestSigner(h.chain.Config())
choice = newBroadcastChoice(h.nodeID)
choice = newBroadcastChoice(h.nodeID, h.txBroadcastKey)
peers = h.peers.all()
)
@ -689,6 +692,7 @@ func (st *blockRangeState) currentRange() eth.BlockRangeUpdatePacket {
type broadcastChoice struct {
self enode.ID
key [16]byte
buffer map[*ethPeer]struct{}
tmp []broadcastPeer
}
@ -698,9 +702,15 @@ type broadcastPeer struct {
score uint64
}
func newBroadcastChoice(self enode.ID) *broadcastChoice {
func newBroadcastChoiceKey() (k [16]byte) {
crand.Read(k[:])
return k
}
func newBroadcastChoice(self enode.ID, key [16]byte) *broadcastChoice {
return &broadcastChoice{
self: self,
key: key,
buffer: make(map[*ethPeer]struct{}),
}
}
@ -710,7 +720,7 @@ func newBroadcastChoice(self enode.ID) *broadcastChoice {
func (bc *broadcastChoice) choosePeers(peers []*ethPeer, txSender common.Address) map[*ethPeer]struct{} {
// Compute scores.
bc.tmp = slices.Grow(bc.tmp[:0], len(peers))[:len(peers)]
hash := fnv.New64()
hash := siphash.New(bc.key[:])
for i, peer := range peers {
hash.Reset()
hash.Write(bc.self[:])

6
eth/handler_test.go
View file

@ -221,8 +221,8 @@ func (b *testHandler) close() {
func TestBroadcastChoice(t *testing.T) {
self := enode.HexID("1111111111111111111111111111111111111111111111111111111111111111")
choice49 := newBroadcastChoice(self)
choice50 := newBroadcastChoice(self)
choice49 := newBroadcastChoice(self, [16]byte{1})
choice50 := newBroadcastChoice(self, [16]byte{1})
// Create test peers and random tx sender addresses.
rand := rand.New(rand.NewSource(33))
@ -289,7 +289,7 @@ func benchmarkBroadcastChoice(b *testing.B, npeers int) {
}
self := enode.HexID("1111111111111111111111111111111111111111111111111111111111111111")
choice := newBroadcastChoice(self)
choice := newBroadcastChoice(self, [16]byte{1})
b.ResetTimer()
for i := range b.N {