mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-05 05:07:36 +03:00
759b0d87b1
This replaces version.UsesMaxDataFrame by versoin.UsesIETFFrameFormat. That way, we can have two separate code paths in the unpacker to unpack either gQUIC frames or IETF frames.
51 lines
1.4 KiB
Go
51 lines
1.4 KiB
Go
package wire
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
"github.com/lucas-clemente/quic-go/internal/protocol"
|
|
"github.com/lucas-clemente/quic-go/internal/utils"
|
|
)
|
|
|
|
// A MaxDataFrame carries flow control information for the connection
|
|
type MaxDataFrame struct {
|
|
ByteOffset protocol.ByteCount
|
|
}
|
|
|
|
// ParseMaxDataFrame parses a MAX_DATA frame
|
|
func ParseMaxDataFrame(r *bytes.Reader, version protocol.VersionNumber) (*MaxDataFrame, error) {
|
|
// read the TypeByte
|
|
if _, err := r.ReadByte(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
frame := &MaxDataFrame{}
|
|
byteOffset, err := utils.GetByteOrder(version).ReadUint64(r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
frame.ByteOffset = protocol.ByteCount(byteOffset)
|
|
return frame, nil
|
|
}
|
|
|
|
//Write writes a MAX_STREAM_DATA frame
|
|
func (f *MaxDataFrame) Write(b *bytes.Buffer, version protocol.VersionNumber) error {
|
|
if !version.UsesIETFFrameFormat() {
|
|
// write a gQUIC WINDOW_UPDATE frame (with stream ID 0, which means connection-level there)
|
|
return (&windowUpdateFrame{
|
|
StreamID: 0,
|
|
ByteOffset: f.ByteOffset,
|
|
}).Write(b, version)
|
|
}
|
|
b.WriteByte(0x4)
|
|
utils.GetByteOrder(version).WriteUint64(b, uint64(f.ByteOffset))
|
|
return nil
|
|
}
|
|
|
|
// MinLength of a written frame
|
|
func (f *MaxDataFrame) MinLength(version protocol.VersionNumber) (protocol.ByteCount, error) {
|
|
if !version.UsesIETFFrameFormat() { // writing this frame would result in a gQUIC WINDOW_UPDATE being written, which is longer
|
|
return 1 + 4 + 8, nil
|
|
}
|
|
return 1 + 8, nil
|
|
}
|