gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

also make ParsePublicHeader public

Browse Source
This commit is contained in:
Marten Seemann
2016-08-05 00:11:40 +07:00
parent df660bdbc5
commit 99d4e36a56
3 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
public_header.go
View File

@@ -98,8 +98,8 @@ func (h *PublicHeader) WritePublicHeader(b *bytes.Buffer, version protocol.Versi
return nil
}
// parsePublicHeader parses a QUIC packet's public header
func parsePublicHeader(b io.ByteReader) (*PublicHeader, error) {
// ParsePublicHeader parses a QUIC packet's public header
func ParsePublicHeader(b io.ByteReader) (*PublicHeader, error) {
header := &PublicHeader{}
// First byte

16
public_header_test.go
View File

@@ -12,7 +12,7 @@ var _ = Describe("Public Header", func() {
Context("when parsing", func() {
It("accepts a sample client header", func() {
b := bytes.NewReader([]byte{0x09, 0xf6, 0x19, 0x86, 0x66, 0x9b, 0x9f, 0xfa, 0x4c, 0x51, 0x30, 0x33, 0x34, 0x01})
hdr, err := parsePublicHeader(b)
hdr, err := ParsePublicHeader(b)
Expect(err).ToNot(HaveOccurred())
Expect(hdr.VersionFlag).To(BeTrue())
Expect(hdr.ResetFlag).To(BeFalse())
@@ -24,19 +24,19 @@ var _ = Describe("Public Header", func() {
It("does not accept 0-byte connection ID", func() {
b := bytes.NewReader([]byte{0x00, 0x01})
_, err := parsePublicHeader(b)
_, err := ParsePublicHeader(b)
Expect(err).To(MatchError(errReceivedTruncatedConnectionID))
})
It("rejects 0 as a connection ID", func() {
b := bytes.NewReader([]byte{0x09, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x51, 0x30, 0x33, 0x30, 0x01})
_, err := parsePublicHeader(b)
_, err := ParsePublicHeader(b)
Expect(err).To(MatchError(errInvalidConnectionID))
})
It("accepts 1-byte packet numbers", func() {
b := bytes.NewReader([]byte{0x08, 0xf6, 0x19, 0x86, 0x66, 0x9b, 0x9f, 0xfa, 0x4c, 0xde})
hdr, err := parsePublicHeader(b)
hdr, err := ParsePublicHeader(b)
Expect(err).ToNot(HaveOccurred())
Expect(hdr.PacketNumber).To(Equal(protocol.PacketNumber(0xde)))
Expect(b.Len()).To(BeZero())
@@ -44,7 +44,7 @@ var _ = Describe("Public Header", func() {
It("accepts 2-byte packet numbers", func() {
b := bytes.NewReader([]byte{0x18, 0xf6, 0x19, 0x86, 0x66, 0x9b, 0x9f, 0xfa, 0x4c, 0xde, 0xca})
hdr, err := parsePublicHeader(b)
hdr, err := ParsePublicHeader(b)
Expect(err).ToNot(HaveOccurred())
Expect(hdr.PacketNumber).To(Equal(protocol.PacketNumber(0xcade)))
Expect(b.Len()).To(BeZero())
@@ -52,7 +52,7 @@ var _ = Describe("Public Header", func() {
It("accepts 4-byte packet numbers", func() {
b := bytes.NewReader([]byte{0x28, 0xf6, 0x19, 0x86, 0x66, 0x9b, 0x9f, 0xfa, 0x4c, 0xad, 0xfb, 0xca, 0xde})
hdr, err := parsePublicHeader(b)
hdr, err := ParsePublicHeader(b)
Expect(err).ToNot(HaveOccurred())
Expect(hdr.PacketNumber).To(Equal(protocol.PacketNumber(0xdecafbad)))
Expect(b.Len()).To(BeZero())
@@ -60,7 +60,7 @@ var _ = Describe("Public Header", func() {
It("accepts 6-byte packet numbers", func() {
b := bytes.NewReader([]byte{0x38, 0xf6, 0x19, 0x86, 0x66, 0x9b, 0x9f, 0xfa, 0x4c, 0x23, 0x42, 0xad, 0xfb, 0xca, 0xde})
hdr, err := parsePublicHeader(b)
hdr, err := ParsePublicHeader(b)
Expect(err).ToNot(HaveOccurred())
Expect(hdr.PacketNumber).To(Equal(protocol.PacketNumber(0xdecafbad4223)))
Expect(b.Len()).To(BeZero())
@@ -71,7 +71,7 @@ var _ = Describe("Public Header", func() {
0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1,
0x01,
})
_, err := parsePublicHeader(b)
_, err := ParsePublicHeader(b)
Expect(err).To(MatchError("diversification nonces should only be sent by servers"))
})
})

2
server.go
View File

@@ -132,7 +132,7 @@ func (s *Server) handlePacket(conn *net.UDPConn, remoteAddr *net.UDPAddr, packet
r := bytes.NewReader(packet)
hdr, err := parsePublicHeader(r)
hdr, err := ParsePublicHeader(r)
if err != nil {
return qerr.Error(qerr.InvalidPacketHeader, err.Error())
}