mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-06 21:57:36 +03:00
3a12a898a5
The new parsing function is vastly faster than the combination of header and extended header parsing: BenchmarkShortHeaderParsing-10 44192314 26.79 ns/op 48 B/op 1 allocs/op BenchmarkShortHeaderParsingOld-10 12627363 99.99 ns/op 228 B/op 3 allocs/op
110 lines
2.9 KiB
Go
110 lines
2.9 KiB
Go
package wire
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/lucas-clemente/quic-go/internal/protocol"
|
|
"github.com/lucas-clemente/quic-go/internal/utils"
|
|
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("Short Header", func() {
|
|
Context("Parsing", func() {
|
|
It("parses", func() {
|
|
data := []byte{
|
|
0b01000110,
|
|
0xde, 0xad, 0xbe, 0xef,
|
|
0x13, 0x37, 0x99,
|
|
}
|
|
hdr, err := ParseShortHeader(data, 4)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(hdr.DestConnectionID).To(Equal(protocol.ParseConnectionID([]byte{0xde, 0xad, 0xbe, 0xef})))
|
|
Expect(hdr.KeyPhase).To(Equal(protocol.KeyPhaseOne))
|
|
Expect(hdr.PacketNumber).To(Equal(protocol.PacketNumber(0x133799)))
|
|
Expect(hdr.PacketNumberLen).To(Equal(protocol.PacketNumberLen3))
|
|
})
|
|
|
|
It("errors when the QUIC bit is not set", func() {
|
|
data := []byte{
|
|
0b00000101,
|
|
0xde, 0xad, 0xbe, 0xef,
|
|
0x13, 0x37,
|
|
}
|
|
_, err := ParseShortHeader(data, 4)
|
|
Expect(err).To(MatchError("not a QUIC packet"))
|
|
})
|
|
|
|
It("errors, but returns the header, when the reserved bits are set", func() {
|
|
data := []byte{
|
|
0b01010101,
|
|
0xde, 0xad, 0xbe, 0xef,
|
|
0x13, 0x37,
|
|
}
|
|
hdr, err := ParseShortHeader(data, 4)
|
|
Expect(err).To(MatchError(ErrInvalidReservedBits))
|
|
Expect(hdr).ToNot(BeNil())
|
|
Expect(hdr.PacketNumber).To(Equal(protocol.PacketNumber(0x1337)))
|
|
})
|
|
|
|
It("errors when passed a long header packet", func() {
|
|
_, err := ParseShortHeader([]byte{0x80}, 4)
|
|
Expect(err).To(MatchError("not a short header packet"))
|
|
})
|
|
|
|
It("errors on EOF", func() {
|
|
data := []byte{
|
|
0b01000110,
|
|
0xde, 0xad, 0xbe, 0xef,
|
|
0x13, 0x37, 0x99,
|
|
}
|
|
_, err := ParseShortHeader(data, 4)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
for i := range data {
|
|
_, err := ParseShortHeader(data[:i], 4)
|
|
Expect(err).To(MatchError(io.EOF))
|
|
}
|
|
})
|
|
})
|
|
|
|
Context("logging", func() {
|
|
var (
|
|
buf *bytes.Buffer
|
|
logger utils.Logger
|
|
)
|
|
|
|
BeforeEach(func() {
|
|
buf = &bytes.Buffer{}
|
|
logger = utils.DefaultLogger
|
|
logger.SetLogLevel(utils.LogLevelDebug)
|
|
log.SetOutput(buf)
|
|
})
|
|
|
|
AfterEach(func() {
|
|
log.SetOutput(os.Stdout)
|
|
})
|
|
|
|
It("logs Short Headers containing a connection ID", func() {
|
|
(&ShortHeader{
|
|
DestConnectionID: protocol.ParseConnectionID([]byte{0xde, 0xad, 0xbe, 0xef, 0xca, 0xfe, 0x13, 0x37}),
|
|
KeyPhase: protocol.KeyPhaseOne,
|
|
PacketNumber: 1337,
|
|
PacketNumberLen: 4,
|
|
}).Log(logger)
|
|
Expect(buf.String()).To(ContainSubstring("Short Header{DestConnectionID: deadbeefcafe1337, PacketNumber: 1337, PacketNumberLen: 4, KeyPhase: 1}"))
|
|
})
|
|
})
|
|
|
|
It("determines the length", func() {
|
|
Expect((&ShortHeader{
|
|
DestConnectionID: protocol.ParseConnectionID([]byte{0xde, 0xaf}),
|
|
PacketNumber: 0x1337,
|
|
PacketNumberLen: protocol.PacketNumberLen3,
|
|
KeyPhase: protocol.KeyPhaseOne,
|
|
}).Len()).To(Equal(protocol.ByteCount(1 + 2 + 3)))
|
|
})
|
|
})
|