mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-04 04:37:36 +03:00
remove the LongHeader field from the wire.Header
This commit is contained in:
parent
aca052dc7c
commit
86edf7fd4b
23 changed files with 154 additions and 363 deletions
|
@ -100,11 +100,10 @@ var _ = Describe("MITM test", func() {
|
|||
defer ticker.Stop()
|
||||
|
||||
if wire.IsLongHeaderPacket(raw[0]) {
|
||||
hdr, _, _, err := wire.ParsePacket(raw, connIDLen)
|
||||
hdr, _, _, err := wire.ParsePacket(raw)
|
||||
Expect(err).ToNot(HaveOccurred())
|
||||
replyHdr := &wire.ExtendedHeader{
|
||||
Header: wire.Header{
|
||||
IsLongHeader: hdr.IsLongHeader,
|
||||
DestConnectionID: hdr.DestConnectionID,
|
||||
SrcConnectionID: hdr.SrcConnectionID,
|
||||
Type: hdr.Type,
|
||||
|
@ -340,7 +339,7 @@ var _ = Describe("MITM test", func() {
|
|||
if dir == quicproxy.DirectionIncoming {
|
||||
defer GinkgoRecover()
|
||||
|
||||
hdr, _, _, err := wire.ParsePacket(raw, connIDLen)
|
||||
hdr, _, _, err := wire.ParsePacket(raw)
|
||||
Expect(err).ToNot(HaveOccurred())
|
||||
|
||||
if hdr.Type != protocol.PacketTypeInitial {
|
||||
|
@ -382,7 +381,7 @@ var _ = Describe("MITM test", func() {
|
|||
defer GinkgoRecover()
|
||||
defer close(done)
|
||||
|
||||
hdr, _, _, err := wire.ParsePacket(raw, connIDLen)
|
||||
hdr, _, _, err := wire.ParsePacket(raw)
|
||||
Expect(err).ToNot(HaveOccurred())
|
||||
|
||||
if hdr.Type != protocol.PacketTypeInitial {
|
||||
|
@ -415,7 +414,7 @@ var _ = Describe("MITM test", func() {
|
|||
if dir == quicproxy.DirectionIncoming {
|
||||
defer GinkgoRecover()
|
||||
|
||||
hdr, _, _, err := wire.ParsePacket(raw, connIDLen)
|
||||
hdr, _, _, err := wire.ParsePacket(raw)
|
||||
Expect(err).ToNot(HaveOccurred())
|
||||
if hdr.Type != protocol.PacketTypeInitial || injected {
|
||||
return 0
|
||||
|
@ -443,7 +442,7 @@ var _ = Describe("MITM test", func() {
|
|||
if dir == quicproxy.DirectionIncoming {
|
||||
defer GinkgoRecover()
|
||||
|
||||
hdr, _, _, err := wire.ParsePacket(raw, connIDLen)
|
||||
hdr, _, _, err := wire.ParsePacket(raw)
|
||||
Expect(err).ToNot(HaveOccurred())
|
||||
if hdr.Type != protocol.PacketTypeInitial || injected {
|
||||
return 0
|
||||
|
|
|
@ -38,7 +38,7 @@ var _ = Describe("0-RTT", func() {
|
|||
if !wire.IsLongHeaderPacket(data[0]) {
|
||||
break
|
||||
}
|
||||
hdr, _, rest, err := wire.ParsePacket(data, 0)
|
||||
hdr, _, rest, err := wire.ParsePacket(data)
|
||||
Expect(err).ToNot(HaveOccurred())
|
||||
if hdr.Type == protocol.PacketType0RTT {
|
||||
atomic.AddUint32(&num0RTTPackets, 1)
|
||||
|
@ -351,7 +351,7 @@ var _ = Describe("0-RTT", func() {
|
|||
RemoteAddr: fmt.Sprintf("localhost:%d", ln.Addr().(*net.UDPAddr).Port),
|
||||
DelayPacket: func(_ quicproxy.Direction, data []byte) time.Duration {
|
||||
if wire.IsLongHeaderPacket(data[0]) {
|
||||
hdr, _, _, err := wire.ParsePacket(data, 0)
|
||||
hdr, _, _, err := wire.ParsePacket(data)
|
||||
Expect(err).ToNot(HaveOccurred())
|
||||
if hdr.Type == protocol.PacketType0RTT {
|
||||
atomic.AddUint32(&num0RTTPackets, 1)
|
||||
|
@ -363,7 +363,7 @@ var _ = Describe("0-RTT", func() {
|
|||
if !wire.IsLongHeaderPacket(data[0]) {
|
||||
return false
|
||||
}
|
||||
hdr, _, _, err := wire.ParsePacket(data, 0)
|
||||
hdr, _, _, err := wire.ParsePacket(data)
|
||||
Expect(err).ToNot(HaveOccurred())
|
||||
if hdr.Type == protocol.PacketType0RTT {
|
||||
// drop 25% of the 0-RTT packets
|
||||
|
@ -398,7 +398,7 @@ var _ = Describe("0-RTT", func() {
|
|||
|
||||
countZeroRTTBytes := func(data []byte) (n protocol.ByteCount) {
|
||||
for len(data) > 0 {
|
||||
hdr, _, rest, err := wire.ParsePacket(data, 0)
|
||||
hdr, _, rest, err := wire.ParsePacket(data)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
|
|
@ -31,7 +31,6 @@ var _ = Describe("QUIC Proxy", func() {
|
|||
b := &bytes.Buffer{}
|
||||
hdr := wire.ExtendedHeader{
|
||||
Header: wire.Header{
|
||||
IsLongHeader: true,
|
||||
Type: protocol.PacketTypeInitial,
|
||||
Version: protocol.VersionTLS,
|
||||
Length: 4 + protocol.ByteCount(len(payload)),
|
||||
|
@ -48,7 +47,7 @@ var _ = Describe("QUIC Proxy", func() {
|
|||
}
|
||||
|
||||
readPacketNumber := func(b []byte) protocol.PacketNumber {
|
||||
hdr, data, _, err := wire.ParsePacket(b, 0)
|
||||
hdr, data, _, err := wire.ParsePacket(b)
|
||||
ExpectWithOffset(1, err).ToNot(HaveOccurred())
|
||||
Expect(hdr.Type).To(Equal(protocol.PacketTypeInitial))
|
||||
extHdr, err := hdr.ParseExtended(bytes.NewReader(data), protocol.VersionTLS)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue