forked from quic-go/quic-go
bc1ef31531
The RSEQ tag is mandatory according to the gQUIC spec, however, Google doesn't include that tag in its Public Resets.
97 lines
3.2 KiB
Go
97 lines
3.2 KiB
Go
package wire
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
|
|
"github.com/lucas-clemente/quic-go/internal/handshake"
|
|
"github.com/lucas-clemente/quic-go/internal/protocol"
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("public reset", func() {
|
|
Context("writing", func() {
|
|
It("writes public reset packets", func() {
|
|
Expect(WritePublicReset(0xdeadbeef, 0x8badf00d, 0xdecafbad)).To(Equal([]byte{
|
|
0x0a,
|
|
0x0, 0x0, 0x0, 0x0, 0xde, 0xad, 0xbe, 0xef,
|
|
'P', 'R', 'S', 'T',
|
|
0x02, 0x00, 0x00, 0x00,
|
|
'R', 'N', 'O', 'N',
|
|
0x08, 0x00, 0x00, 0x00,
|
|
'R', 'S', 'E', 'Q',
|
|
0x10, 0x00, 0x00, 0x00,
|
|
0xad, 0xfb, 0xca, 0xde, 0x0, 0x0, 0x0, 0x0,
|
|
0x0d, 0xf0, 0xad, 0x8b, 0x0, 0x0, 0x0, 0x0,
|
|
}))
|
|
})
|
|
})
|
|
|
|
Context("parsing", func() {
|
|
var b *bytes.Buffer
|
|
|
|
BeforeEach(func() {
|
|
b = &bytes.Buffer{}
|
|
})
|
|
|
|
It("parses a public reset", func() {
|
|
packet := WritePublicReset(0xdeadbeef, 0x8badf00d, 0xdecafbad)
|
|
pr, err := ParsePublicReset(bytes.NewReader(packet[9:])) // 1 byte Public Flag, 8 bytes connection ID
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(pr.Nonce).To(Equal(uint64(0xdecafbad)))
|
|
Expect(pr.RejectedPacketNumber).To(Equal(protocol.PacketNumber(0x8badf00d)))
|
|
})
|
|
|
|
It("rejects packets that it can't parse", func() {
|
|
_, err := ParsePublicReset(bytes.NewReader([]byte{}))
|
|
Expect(err).To(MatchError(io.EOF))
|
|
})
|
|
|
|
It("rejects packets with the wrong tag", func() {
|
|
handshake.HandshakeMessage{Tag: handshake.TagREJ, Data: nil}.Write(b)
|
|
_, err := ParsePublicReset(bytes.NewReader(b.Bytes()))
|
|
Expect(err).To(MatchError("wrong public reset tag"))
|
|
})
|
|
|
|
It("rejects packets missing the nonce", func() {
|
|
data := map[handshake.Tag][]byte{
|
|
handshake.TagRSEQ: []byte{0xde, 0xad, 0xbe, 0xef, 0xca, 0xfe, 0x13, 0x37},
|
|
}
|
|
handshake.HandshakeMessage{Tag: handshake.TagPRST, Data: data}.Write(b)
|
|
_, err := ParsePublicReset(bytes.NewReader(b.Bytes()))
|
|
Expect(err).To(MatchError("RNON missing"))
|
|
})
|
|
|
|
It("rejects packets with a wrong length nonce", func() {
|
|
data := map[handshake.Tag][]byte{
|
|
handshake.TagRSEQ: []byte{0xde, 0xad, 0xbe, 0xef, 0xca, 0xfe, 0x13, 0x37},
|
|
handshake.TagRNON: []byte{0xde, 0xad, 0xbe, 0xef, 0xca, 0xfe, 0x13},
|
|
}
|
|
handshake.HandshakeMessage{Tag: handshake.TagPRST, Data: data}.Write(b)
|
|
_, err := ParsePublicReset(bytes.NewReader(b.Bytes()))
|
|
Expect(err).To(MatchError("invalid RNON tag"))
|
|
})
|
|
|
|
It("accepts packets missing the rejected packet number", func() {
|
|
data := map[handshake.Tag][]byte{
|
|
handshake.TagRNON: []byte{0xde, 0xad, 0xbe, 0xef, 0xca, 0xfe, 0x13, 0x37},
|
|
}
|
|
handshake.HandshakeMessage{Tag: handshake.TagPRST, Data: data}.Write(b)
|
|
pr, err := ParsePublicReset(bytes.NewReader(b.Bytes()))
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(pr.Nonce).To(Equal(uint64(0x3713fecaefbeadde)))
|
|
})
|
|
|
|
It("rejects packets with a wrong length rejected packet number", func() {
|
|
data := map[handshake.Tag][]byte{
|
|
handshake.TagRSEQ: []byte{0xde, 0xad, 0xbe, 0xef, 0xca, 0xfe, 0x13},
|
|
handshake.TagRNON: []byte{0xde, 0xad, 0xbe, 0xef, 0xca, 0xfe, 0x13, 0x37},
|
|
}
|
|
handshake.HandshakeMessage{Tag: handshake.TagPRST, Data: data}.Write(b)
|
|
_, err := ParsePublicReset(bytes.NewReader(b.Bytes()))
|
|
Expect(err).To(MatchError("invalid RSEQ tag"))
|
|
})
|
|
})
|
|
})
|