forked from quic-go/quic-go
147 lines
3.6 KiB
Go
147 lines
3.6 KiB
Go
package qlog
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
|
|
"github.com/francoispqt/gojay"
|
|
|
|
"github.com/lucas-clemente/quic-go/internal/protocol"
|
|
"github.com/lucas-clemente/quic-go/internal/wire"
|
|
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("Packet Header", func() {
|
|
Context("determining the packet type", func() {
|
|
It("recognizes Initial packets", func() {
|
|
Expect(getPacketType(&wire.ExtendedHeader{
|
|
Header: wire.Header{
|
|
IsLongHeader: true,
|
|
Type: protocol.PacketTypeInitial,
|
|
Version: protocol.VersionTLS,
|
|
},
|
|
})).To(Equal(packetTypeInitial))
|
|
})
|
|
|
|
It("recognizes Handshake packets", func() {
|
|
Expect(getPacketType(&wire.ExtendedHeader{
|
|
Header: wire.Header{
|
|
IsLongHeader: true,
|
|
Type: protocol.PacketTypeHandshake,
|
|
Version: protocol.VersionTLS,
|
|
},
|
|
})).To(Equal(packetTypeHandshake))
|
|
})
|
|
|
|
It("recognizes Retry packets", func() {
|
|
Expect(getPacketType(&wire.ExtendedHeader{
|
|
Header: wire.Header{
|
|
IsLongHeader: true,
|
|
Type: protocol.PacketTypeRetry,
|
|
Version: protocol.VersionTLS,
|
|
},
|
|
})).To(Equal(packetTypeRetry))
|
|
})
|
|
|
|
It("recognizes 0-RTT packets", func() {
|
|
Expect(getPacketType(&wire.ExtendedHeader{
|
|
Header: wire.Header{
|
|
IsLongHeader: true,
|
|
Type: protocol.PacketType0RTT,
|
|
Version: protocol.VersionTLS,
|
|
},
|
|
})).To(Equal(packetType0RTT))
|
|
})
|
|
|
|
It("recognizes Version Negotiation packets", func() {
|
|
Expect(getPacketType(&wire.ExtendedHeader{
|
|
Header: wire.Header{IsLongHeader: true},
|
|
})).To(Equal(packetTypeVersionNegotiation))
|
|
})
|
|
|
|
It("recognizes 1-RTT packets", func() {
|
|
Expect(getPacketType(&wire.ExtendedHeader{
|
|
Header: wire.Header{},
|
|
})).To(Equal(packetType1RTT))
|
|
})
|
|
})
|
|
|
|
Context("marshalling", func() {
|
|
check := func(hdr *wire.ExtendedHeader, expected map[string]interface{}) {
|
|
buf := &bytes.Buffer{}
|
|
enc := gojay.NewEncoder(buf)
|
|
ExpectWithOffset(1, enc.Encode(transformExtendedHeader(hdr))).To(Succeed())
|
|
data := buf.Bytes()
|
|
ExpectWithOffset(1, json.Valid(data)).To(BeTrue())
|
|
checkEncoding(data, expected)
|
|
}
|
|
|
|
It("marshals a header", func() {
|
|
check(
|
|
&wire.ExtendedHeader{PacketNumber: 42},
|
|
map[string]interface{}{
|
|
"packet_number": "42",
|
|
},
|
|
)
|
|
})
|
|
|
|
It("marshals a header with a payload length", func() {
|
|
check(
|
|
&wire.ExtendedHeader{
|
|
PacketNumber: 42,
|
|
Header: wire.Header{Length: 123},
|
|
},
|
|
map[string]interface{}{
|
|
"packet_number": "42",
|
|
"payload_length": 123,
|
|
},
|
|
)
|
|
})
|
|
|
|
It("marshals a header with a source connection ID", func() {
|
|
check(
|
|
&wire.ExtendedHeader{
|
|
PacketNumber: 42,
|
|
Header: wire.Header{
|
|
SrcConnectionID: protocol.ConnectionID{0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff},
|
|
},
|
|
},
|
|
map[string]interface{}{
|
|
"packet_number": "42",
|
|
"scil": "16",
|
|
"scid": "00112233445566778899aabbccddeeff",
|
|
},
|
|
)
|
|
})
|
|
|
|
It("marshals a header with a destination connection ID", func() {
|
|
check(
|
|
&wire.ExtendedHeader{
|
|
PacketNumber: 42,
|
|
Header: wire.Header{DestConnectionID: protocol.ConnectionID{0xde, 0xad, 0xbe, 0xef}},
|
|
},
|
|
map[string]interface{}{
|
|
"packet_number": "42",
|
|
"dcil": "4",
|
|
"dcid": "deadbeef",
|
|
},
|
|
)
|
|
})
|
|
|
|
It("marshals a header with a version number", func() {
|
|
check(
|
|
&wire.ExtendedHeader{
|
|
PacketNumber: 42,
|
|
Header: wire.Header{Version: protocol.VersionNumber(0xdecafbad)},
|
|
},
|
|
map[string]interface{}{
|
|
"packet_number": "42",
|
|
"version": "decafbad",
|
|
},
|
|
)
|
|
})
|
|
})
|
|
})
|