From e75f6f24ba0f58b9bd049416a9395d667441083f Mon Sep 17 00:00:00 2001 From: Delweng Date: Thu, 26 Feb 2026 11:37:47 +0800 Subject: [PATCH] cmd/devp2p: fix discv5 PingMultiIP handshake reply addr Signed-off-by: Delweng --- cmd/devp2p/internal/v5test/framework.go | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/cmd/devp2p/internal/v5test/framework.go b/cmd/devp2p/internal/v5test/framework.go index 92a5048150..57f33d8561 100644 --- a/cmd/devp2p/internal/v5test/framework.go +++ b/cmd/devp2p/internal/v5test/framework.go @@ -65,6 +65,7 @@ type conn struct { log logger codec *v5wire.Codec idCounter uint32 + lastFrom string } type logger interface { @@ -200,7 +201,11 @@ func (tc *conn) findnode(c net.PacketConn, dists []uint) ([]*enode.Node, error) // write sends a packet on the given connection. func (tc *conn) write(c net.PacketConn, p v5wire.Packet, challenge *v5wire.Whoareyou) v5wire.Nonce { - packet, nonce, err := tc.codec.Encode(tc.remote.ID(), tc.remoteAddr.String(), p, challenge) + addr := tc.remoteAddr.String() + if challenge != nil && tc.lastFrom != "" { + addr = tc.lastFrom + } + packet, nonce, err := tc.codec.Encode(tc.remote.ID(), addr, p, challenge) if err != nil { panic(fmt.Errorf("can't encode %v packet: %v", p.Name(), err)) } @@ -222,6 +227,7 @@ func (tc *conn) read(c net.PacketConn) v5wire.Packet { if err != nil { return &readError{err} } + tc.lastFrom = fromAddr.String() _, _, p, err := tc.codec.Decode(buf[:n], fromAddr.String()) if err != nil { return &readError{err}