diff --git a/internal/handshake/token_generator.go b/internal/handshake/token_generator.go index 215432c3..95776768 100644 --- a/internal/handshake/token_generator.go +++ b/internal/handshake/token_generator.go @@ -45,7 +45,7 @@ func NewTokenGenerator() (*TokenGenerator, error) { } // NewToken generates a new token for a given source address -func (g *TokenGenerator) NewToken(raddr net.Addr, origConnID protocol.ConnectionID) ([]byte, error) { +func (g *TokenGenerator) NewRetryToken(raddr net.Addr, origConnID protocol.ConnectionID) ([]byte, error) { data, err := asn1.Marshal(token{ RemoteAddr: encodeRemoteAddr(raddr), OriginalDestConnectionID: origConnID, diff --git a/internal/handshake/token_generator_test.go b/internal/handshake/token_generator_test.go index ddf2124b..4a3cdae7 100644 --- a/internal/handshake/token_generator_test.go +++ b/internal/handshake/token_generator_test.go @@ -22,7 +22,7 @@ var _ = Describe("Token Generator", func() { It("generates a token", func() { ip := net.IPv4(127, 0, 0, 1) - token, err := tokenGen.NewToken(&net.UDPAddr{IP: ip, Port: 1337}, nil) + token, err := tokenGen.NewRetryToken(&net.UDPAddr{IP: ip, Port: 1337}, nil) Expect(err).ToNot(HaveOccurred()) Expect(token).ToNot(BeEmpty()) }) @@ -35,7 +35,7 @@ var _ = Describe("Token Generator", func() { It("accepts a valid token", func() { ip := net.IPv4(192, 168, 0, 1) - tokenEnc, err := tokenGen.NewToken( + tokenEnc, err := tokenGen.NewRetryToken( &net.UDPAddr{IP: ip, Port: 1337}, nil, ) @@ -48,7 +48,7 @@ var _ = Describe("Token Generator", func() { }) It("saves the connection ID", func() { - tokenEnc, err := tokenGen.NewToken( + tokenEnc, err := tokenGen.NewRetryToken( &net.UDPAddr{}, protocol.ConnectionID{0xde, 0xad, 0xbe, 0xef}, ) @@ -101,7 +101,7 @@ var _ = Describe("Token Generator", func() { ip := net.ParseIP(addr) Expect(ip).ToNot(BeNil()) raddr := &net.UDPAddr{IP: ip, Port: 1337} - tokenEnc, err := tokenGen.NewToken(raddr, nil) + tokenEnc, err := tokenGen.NewRetryToken(raddr, nil) Expect(err).ToNot(HaveOccurred()) token, err := tokenGen.DecodeToken(tokenEnc) Expect(err).ToNot(HaveOccurred()) @@ -112,7 +112,7 @@ var _ = Describe("Token Generator", func() { It("uses the string representation an address that is not a UDP address", func() { raddr := &net.TCPAddr{IP: net.IPv4(192, 168, 13, 37), Port: 1337} - tokenEnc, err := tokenGen.NewToken(raddr, nil) + tokenEnc, err := tokenGen.NewRetryToken(raddr, nil) Expect(err).ToNot(HaveOccurred()) token, err := tokenGen.DecodeToken(tokenEnc) Expect(err).ToNot(HaveOccurred()) diff --git a/server.go b/server.go index a6c377e8..4878d155 100644 --- a/server.go +++ b/server.go @@ -468,7 +468,7 @@ func (s *server) createNewSession( } func (s *server) sendRetry(remoteAddr net.Addr, hdr *wire.Header) error { - token, err := s.tokenGenerator.NewToken(remoteAddr, hdr.DestConnectionID) + token, err := s.tokenGenerator.NewRetryToken(remoteAddr, hdr.DestConnectionID) if err != nil { return err } diff --git a/server_test.go b/server_test.go index b1099641..6d491a86 100644 --- a/server_test.go +++ b/server_test.go @@ -191,7 +191,7 @@ var _ = Describe("Server", func() { close(done) return false } - token, err := serv.tokenGenerator.NewToken(raddr, nil) + token, err := serv.tokenGenerator.NewRetryToken(raddr, nil) Expect(err).ToNot(HaveOccurred()) packet := getPacket(&wire.Header{ IsLongHeader: true,