mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-11-07 10:47:48 -05:00
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package wire
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
"v2ray.com/core/external/github.com/lucas-clemente/quic-go/internal/protocol"
|
|
"v2ray.com/core/external/github.com/lucas-clemente/quic-go/internal/utils"
|
|
)
|
|
|
|
// A StreamsBlockedFrame is a STREAMS_BLOCKED frame
|
|
type StreamsBlockedFrame struct {
|
|
Type protocol.StreamType
|
|
StreamLimit uint64
|
|
}
|
|
|
|
func parseStreamsBlockedFrame(r *bytes.Reader, _ protocol.VersionNumber) (*StreamsBlockedFrame, error) {
|
|
typeByte, err := r.ReadByte()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
f := &StreamsBlockedFrame{}
|
|
switch typeByte {
|
|
case 0x16:
|
|
f.Type = protocol.StreamTypeBidi
|
|
case 0x17:
|
|
f.Type = protocol.StreamTypeUni
|
|
}
|
|
streamLimit, err := utils.ReadVarInt(r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
f.StreamLimit = streamLimit
|
|
|
|
return f, nil
|
|
}
|
|
|
|
func (f *StreamsBlockedFrame) Write(b *bytes.Buffer, _ protocol.VersionNumber) error {
|
|
switch f.Type {
|
|
case protocol.StreamTypeBidi:
|
|
b.WriteByte(0x16)
|
|
case protocol.StreamTypeUni:
|
|
b.WriteByte(0x17)
|
|
}
|
|
utils.WriteVarInt(b, f.StreamLimit)
|
|
return nil
|
|
}
|
|
|
|
// Length of a written frame
|
|
func (f *StreamsBlockedFrame) Length(_ protocol.VersionNumber) protocol.ByteCount {
|
|
return 1 + utils.VarIntLen(f.StreamLimit)
|
|
}
|