mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-01 19:27:35 +03:00
56 lines
2.2 KiB
Go
56 lines
2.2 KiB
Go
package wire
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/rand"
|
|
"encoding/binary"
|
|
"errors"
|
|
|
|
"github.com/refraction-networking/uquic/internal/protocol"
|
|
"github.com/refraction-networking/uquic/internal/utils"
|
|
)
|
|
|
|
// ParseVersionNegotiationPacket parses a Version Negotiation packet.
|
|
func ParseVersionNegotiationPacket(b []byte) (dest, src protocol.ArbitraryLenConnectionID, _ []protocol.VersionNumber, _ error) {
|
|
n, dest, src, err := ParseArbitraryLenConnectionIDs(b)
|
|
if err != nil {
|
|
return nil, nil, nil, err
|
|
}
|
|
b = b[n:]
|
|
if len(b) == 0 {
|
|
//nolint:stylecheck
|
|
return nil, nil, nil, errors.New("Version Negotiation packet has empty version list")
|
|
}
|
|
if len(b)%4 != 0 {
|
|
//nolint:stylecheck
|
|
return nil, nil, nil, errors.New("Version Negotiation packet has a version list with an invalid length")
|
|
}
|
|
versions := make([]protocol.VersionNumber, len(b)/4)
|
|
for i := 0; len(b) > 0; i++ {
|
|
versions[i] = protocol.VersionNumber(binary.BigEndian.Uint32(b[:4]))
|
|
b = b[4:]
|
|
}
|
|
return dest, src, versions, nil
|
|
}
|
|
|
|
// ComposeVersionNegotiation composes a Version Negotiation
|
|
func ComposeVersionNegotiation(destConnID, srcConnID protocol.ArbitraryLenConnectionID, versions []protocol.VersionNumber) []byte {
|
|
greasedVersions := protocol.GetGreasedVersions(versions)
|
|
expectedLen := 1 /* type byte */ + 4 /* version field */ + 1 /* dest connection ID length field */ + destConnID.Len() + 1 /* src connection ID length field */ + srcConnID.Len() + len(greasedVersions)*4
|
|
buf := bytes.NewBuffer(make([]byte, 0, expectedLen))
|
|
r := make([]byte, 1)
|
|
_, _ = rand.Read(r) // ignore the error here. It is not critical to have perfect random here.
|
|
// Setting the "QUIC bit" (0x40) is not required by the RFC,
|
|
// but it allows clients to demultiplex QUIC with a long list of other protocols.
|
|
// See RFC 9443 and https://mailarchive.ietf.org/arch/msg/quic/oR4kxGKY6mjtPC1CZegY1ED4beg/ for details.
|
|
buf.WriteByte(r[0] | 0xc0)
|
|
utils.BigEndian.WriteUint32(buf, 0) // version 0
|
|
buf.WriteByte(uint8(destConnID.Len()))
|
|
buf.Write(destConnID.Bytes())
|
|
buf.WriteByte(uint8(srcConnID.Len()))
|
|
buf.Write(srcConnID.Bytes())
|
|
for _, v := range greasedVersions {
|
|
utils.BigEndian.WriteUint32(buf, uint32(v))
|
|
}
|
|
return buf.Bytes()
|
|
}
|