From: Marten Seemann <martenseemann@gmail.com>
Date: Tue, 25 Nov 2025 04:36:51 +0000 (+0800)
Subject: handshake: add a benchmark for token decoding (#5460)
X-Git-Url: https://git.feebdaed.xyz/?a=commitdiff_plain;h=9cb01a2e95229d551e7f592fc14b876d897d7860;p=0xmirror%2Fquic-go.git

handshake: add a benchmark for token decoding (#5460)
---

diff --git a/internal/handshake/token_generator_test.go b/internal/handshake/token_generator_test.go
index c09917bd..bb58fd10 100644
--- a/internal/handshake/token_generator_test.go
+++ b/internal/handshake/token_generator_test.go
@@ -118,3 +118,23 @@ func TestTokenGeneratorNonUDPAddr(t *testing.T) {
 	require.False(t, token.ValidateRemoteAddr(&net.TCPAddr{IP: net.IPv4(192, 168, 13, 37), Port: 1338}))
 	require.WithinDuration(t, time.Now(), token.SentTime, 100*time.Millisecond)
 }
+
+func BenchmarkTokenGeneratorDecodeToken(b *testing.B) {
+	b.ReportAllocs()
+
+	var key TokenProtectorKey
+	_, err := rand.Read(key[:])
+	require.NoError(b, err)
+	tokenGen := NewTokenGenerator(key)
+	addr := &net.UDPAddr{IP: net.IPv4(192, 168, 0, 1), Port: 1337}
+	connID1 := protocol.ParseConnectionID([]byte{0xde, 0xad, 0xbe, 0xef})
+	connID2 := protocol.ParseConnectionID([]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10})
+	tokenEnc, err := tokenGen.NewRetryToken(addr, connID1, connID2)
+	require.NoError(b, err)
+
+	for b.Loop() {
+		if _, err := tokenGen.DecodeToken(tokenEnc); err != nil {
+			b.Fatal(err)
+		}
+	}
+}