fix:make only sent whoareyou challenges first time

Signed-off-by: Chen Kai <281165273grape@gmail.com>
optimism-java · Oct 9, 2024 · 076e33c · 076e33c
1 parent 4605c46
commit 076e33c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/p2p/discover/v5wire/encoding.go b/p2p/discover/v5wire/encoding.go
@@ -217,6 +217,10 @@ func (c *Codec) Encode(id enode.ID, addr string, packet Packet, challenge *Whoar
 
 	// Store sent WHOAREYOU challenges.
 	if challenge, ok := packet.(*Whoareyou); ok {
+		// check if we already sent a challenge to this node
+		if c.sc.getHandshake(id, addr) != nil {
+			return nil, Nonce{}, nil
+		}
 		challenge.ChallengeData = bytesCopy(&c.buf)
 		c.sc.storeSentHandshake(id, addr, challenge)
 	} else if msgData == nil {