forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-07-26 22:56:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

p2p/discover/v5wire: use Encoded field of Whoareyou

Browse source
This commit is contained in:
Felix Lange 2025-04-02 12:08:09 +02:00
parent c5e94b67db
commit 9b74d7912e
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
p2p/discover/v5wire/encoding.go
View file

@ -190,8 +190,9 @@ func (c *Codec) Encode(id enode.ID, addr string, packet Packet, challenge *Whoar
switch { switch {
case packet.Kind() == WhoareyouPacket: case packet.Kind() == WhoareyouPacket:
// just send the WHOAREYOU packet raw again, rather than the re-encoded challenge data // just send the WHOAREYOU packet raw again, rather than the re-encoded challenge data
if sentWhoareyou := c.sc.getHandshake(id, addr); sentWhoareyou != nil { w := packet.(*Whoareyou)
return sentWhoareyou.Encoded, sentWhoareyou.Nonce, nil if len(w.Encoded) > 0 {
return w.Encoded, w.Nonce, nil
} }
head, err = c.encodeWhoareyou(id, packet.(*Whoareyou)) head, err = c.encodeWhoareyou(id, packet.(*Whoareyou))
case challenge != nil: case challenge != nil:
@ -227,7 +228,7 @@ func (c *Codec) Encode(id enode.ID, addr string, packet Packet, challenge *Whoar
if err != nil { if err != nil {
return nil, Nonce{}, err return nil, Nonce{}, err
} }
challenge.Encoded = enc challenge.Encoded = bytes.Clone(enc)
c.sc.storeSentHandshake(id, addr, challenge) c.sc.storeSentHandshake(id, addr, challenge)
return enc, head.Nonce, err return enc, head.Nonce, err
} }