mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-03 20:27:35 +03:00
95575f5fe7
uTLS is not yet bumped to the new version, so this commit breaks the dependencies relationship by getting rid of the local replace.
73 lines
2.4 KiB
Go
73 lines
2.4 KiB
Go
package wire
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/refraction-networking/uquic/internal/protocol"
|
|
"github.com/refraction-networking/uquic/internal/utils"
|
|
)
|
|
|
|
// ParseShortHeader parses a short header packet.
|
|
// It must be called after header protection was removed.
|
|
// Otherwise, the check for the reserved bits will (most likely) fail.
|
|
func ParseShortHeader(data []byte, connIDLen int) (length int, _ protocol.PacketNumber, _ protocol.PacketNumberLen, _ protocol.KeyPhaseBit, _ error) {
|
|
if len(data) == 0 {
|
|
return 0, 0, 0, 0, io.EOF
|
|
}
|
|
if data[0]&0x80 > 0 {
|
|
return 0, 0, 0, 0, errors.New("not a short header packet")
|
|
}
|
|
if data[0]&0x40 == 0 {
|
|
return 0, 0, 0, 0, errors.New("not a QUIC packet")
|
|
}
|
|
pnLen := protocol.PacketNumberLen(data[0]&0b11) + 1
|
|
if len(data) < 1+int(pnLen)+connIDLen {
|
|
return 0, 0, 0, 0, io.EOF
|
|
}
|
|
|
|
pos := 1 + connIDLen
|
|
var pn protocol.PacketNumber
|
|
switch pnLen {
|
|
case protocol.PacketNumberLen1:
|
|
pn = protocol.PacketNumber(data[pos])
|
|
case protocol.PacketNumberLen2:
|
|
pn = protocol.PacketNumber(utils.BigEndian.Uint16(data[pos : pos+2]))
|
|
case protocol.PacketNumberLen3:
|
|
pn = protocol.PacketNumber(utils.BigEndian.Uint24(data[pos : pos+3]))
|
|
case protocol.PacketNumberLen4:
|
|
pn = protocol.PacketNumber(utils.BigEndian.Uint32(data[pos : pos+4]))
|
|
default:
|
|
return 0, 0, 0, 0, fmt.Errorf("invalid packet number length: %d", pnLen)
|
|
}
|
|
kp := protocol.KeyPhaseZero
|
|
if data[0]&0b100 > 0 {
|
|
kp = protocol.KeyPhaseOne
|
|
}
|
|
|
|
var err error
|
|
if data[0]&0x18 != 0 {
|
|
err = ErrInvalidReservedBits
|
|
}
|
|
return 1 + connIDLen + int(pnLen), pn, pnLen, kp, err
|
|
}
|
|
|
|
// AppendShortHeader writes a short header.
|
|
func AppendShortHeader(b []byte, connID protocol.ConnectionID, pn protocol.PacketNumber, pnLen protocol.PacketNumberLen, kp protocol.KeyPhaseBit) ([]byte, error) {
|
|
typeByte := 0x40 | uint8(pnLen-1)
|
|
if kp == protocol.KeyPhaseOne {
|
|
typeByte |= byte(1 << 2)
|
|
}
|
|
b = append(b, typeByte)
|
|
b = append(b, connID.Bytes()...)
|
|
return appendPacketNumber(b, pn, pnLen)
|
|
}
|
|
|
|
func ShortHeaderLen(dest protocol.ConnectionID, pnLen protocol.PacketNumberLen) protocol.ByteCount {
|
|
return 1 + protocol.ByteCount(dest.Len()) + protocol.ByteCount(pnLen)
|
|
}
|
|
|
|
func LogShortHeader(logger utils.Logger, dest protocol.ConnectionID, pn protocol.PacketNumber, pnLen protocol.PacketNumberLen, kp protocol.KeyPhaseBit) {
|
|
logger.Debugf("\tShort Header{DestConnectionID: %s, PacketNumber: %d, PacketNumberLen: %d, KeyPhase: %s}", dest, pn, pnLen, kp)
|
|
}
|