diff --git a/client.go b/client.go index 1906abdf9..58316b30a 100644 --- a/client.go +++ b/client.go @@ -44,7 +44,7 @@ type client struct { var ( // make it possible to mock connection ID generation in the tests - generateConnectionID = utils.GenerateConnectionID + generateConnectionID = protocol.GenerateConnectionID errCloseSessionForNewVersion = errors.New("closing session in order to recreate it with a new version") ) @@ -384,7 +384,7 @@ func (c *client) handleVersionNegotiationPacket(hdr *wire.Header) error { c.initialVersion = c.version c.version = newVersion var err error - c.connectionID, err = utils.GenerateConnectionID() + c.connectionID, err = protocol.GenerateConnectionID() if err != nil { return err } diff --git a/internal/utils/connection_id.go b/internal/protocol/connection_id.go similarity index 51% rename from internal/utils/connection_id.go rename to internal/protocol/connection_id.go index b4af4e780..afb35d161 100644 --- a/internal/utils/connection_id.go +++ b/internal/protocol/connection_id.go @@ -1,18 +1,19 @@ -package utils +package protocol import ( "crypto/rand" "encoding/binary" - - "github.com/lucas-clemente/quic-go/internal/protocol" ) +// A ConnectionID in QUIC +type ConnectionID uint64 + // GenerateConnectionID generates a connection ID using cryptographic random -func GenerateConnectionID() (protocol.ConnectionID, error) { +func GenerateConnectionID() (ConnectionID, error) { b := make([]byte, 8) _, err := rand.Read(b) if err != nil { return 0, err } - return protocol.ConnectionID(binary.LittleEndian.Uint64(b)), nil + return ConnectionID(binary.LittleEndian.Uint64(b)), nil } diff --git a/internal/utils/connection_id_test.go b/internal/protocol/connection_id_test.go similarity index 95% rename from internal/utils/connection_id_test.go rename to internal/protocol/connection_id_test.go index cd3f9af1f..8f8f8daeb 100644 --- a/internal/utils/connection_id_test.go +++ b/internal/protocol/connection_id_test.go @@ -1,4 +1,4 @@ -package utils +package protocol import ( . "github.com/onsi/ginkgo" diff --git a/internal/protocol/protocol.go b/internal/protocol/protocol.go index 2821d2cd9..e89b22279 100644 --- a/internal/protocol/protocol.go +++ b/internal/protocol/protocol.go @@ -52,9 +52,6 @@ func (t PacketType) String() string { } } -// A ConnectionID in QUIC -type ConnectionID uint64 - // A ByteCount in QUIC type ByteCount uint64