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.
65 lines
2.2 KiB
Go
65 lines
2.2 KiB
Go
package wire
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
|
|
"github.com/refraction-networking/uquic/internal/protocol"
|
|
"github.com/refraction-networking/uquic/quicvarint"
|
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("NEW_TOKEN frame", func() {
|
|
Context("parsing", func() {
|
|
It("accepts a sample frame", func() {
|
|
token := "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua."
|
|
data := encodeVarInt(uint64(len(token)))
|
|
data = append(data, token...)
|
|
b := bytes.NewReader(data)
|
|
f, err := parseNewTokenFrame(b, protocol.Version1)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(string(f.Token)).To(Equal(token))
|
|
Expect(b.Len()).To(BeZero())
|
|
})
|
|
|
|
It("rejects empty tokens", func() {
|
|
data := encodeVarInt(0)
|
|
b := bytes.NewReader(data)
|
|
_, err := parseNewTokenFrame(b, protocol.Version1)
|
|
Expect(err).To(MatchError("token must not be empty"))
|
|
})
|
|
|
|
It("errors on EOFs", func() {
|
|
token := "Lorem ipsum dolor sit amet, consectetur adipiscing elit"
|
|
data := encodeVarInt(uint64(len(token)))
|
|
data = append(data, token...)
|
|
r := bytes.NewReader(data)
|
|
_, err := parseNewTokenFrame(r, protocol.Version1)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
for i := range data {
|
|
_, err := parseNewTokenFrame(bytes.NewReader(data[:i]), protocol.Version1)
|
|
Expect(err).To(MatchError(io.EOF))
|
|
}
|
|
})
|
|
})
|
|
|
|
Context("writing", func() {
|
|
It("writes a sample frame", func() {
|
|
token := "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat."
|
|
f := &NewTokenFrame{Token: []byte(token)}
|
|
b, err := f.Append(nil, protocol.Version1)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
expected := []byte{newTokenFrameType}
|
|
expected = append(expected, encodeVarInt(uint64(len(token)))...)
|
|
expected = append(expected, token...)
|
|
Expect(b).To(Equal(expected))
|
|
})
|
|
|
|
It("has the correct min length", func() {
|
|
frame := &NewTokenFrame{Token: []byte("foobar")}
|
|
Expect(frame.Length(protocol.Version1)).To(Equal(1 + quicvarint.Len(6) + 6))
|
|
})
|
|
})
|
|
})
|