forked from quic-go/quic-go
8e332c2e13
We can now impose a limit on the number of stream for IETF QUIC, and advertise that in the transport parameters during the handshake.
122 lines
3.2 KiB
Go
122 lines
3.2 KiB
Go
package quic
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
|
|
"github.com/lucas-clemente/quic-go/internal/protocol"
|
|
"github.com/lucas-clemente/quic-go/internal/wire"
|
|
)
|
|
|
|
//go:generate genny -in $GOFILE -out streams_map_incoming_bidi.go gen "item=streamI Item=BidiStream"
|
|
//go:generate genny -in $GOFILE -out streams_map_incoming_uni.go gen "item=receiveStreamI Item=UniStream"
|
|
type incomingItemsMap struct {
|
|
mutex sync.RWMutex
|
|
cond sync.Cond
|
|
|
|
streams map[protocol.StreamID]item
|
|
|
|
nextStream protocol.StreamID // the next stream that will be returned by AcceptStream()
|
|
highestStream protocol.StreamID // the highest stream that the peer openend
|
|
maxStream protocol.StreamID // the highest stream that the peer is allowed to open
|
|
maxNumStreams int // maximum number of streams
|
|
|
|
newStream func(protocol.StreamID) item
|
|
queueMaxStreamID func(*wire.MaxStreamIDFrame)
|
|
|
|
closeErr error
|
|
}
|
|
|
|
func newIncomingItemsMap(
|
|
nextStream protocol.StreamID,
|
|
initialMaxStreamID protocol.StreamID,
|
|
maxNumStreams int,
|
|
queueControlFrame func(wire.Frame),
|
|
newStream func(protocol.StreamID) item,
|
|
) *incomingItemsMap {
|
|
m := &incomingItemsMap{
|
|
streams: make(map[protocol.StreamID]item),
|
|
nextStream: nextStream,
|
|
maxStream: initialMaxStreamID,
|
|
maxNumStreams: maxNumStreams,
|
|
newStream: newStream,
|
|
queueMaxStreamID: func(f *wire.MaxStreamIDFrame) { queueControlFrame(f) },
|
|
}
|
|
m.cond.L = &m.mutex
|
|
return m
|
|
}
|
|
|
|
func (m *incomingItemsMap) AcceptStream() (item, error) {
|
|
m.mutex.Lock()
|
|
defer m.mutex.Unlock()
|
|
|
|
var str item
|
|
for {
|
|
var ok bool
|
|
if m.closeErr != nil {
|
|
return nil, m.closeErr
|
|
}
|
|
str, ok = m.streams[m.nextStream]
|
|
if ok {
|
|
break
|
|
}
|
|
m.cond.Wait()
|
|
}
|
|
m.nextStream += 4
|
|
return str, nil
|
|
}
|
|
|
|
func (m *incomingItemsMap) GetOrOpenStream(id protocol.StreamID) (item, error) {
|
|
if id > m.maxStream {
|
|
return nil, fmt.Errorf("peer tried to open stream %d (current limit: %d)", id, m.maxStream)
|
|
}
|
|
// if the id is smaller than the highest we accepted
|
|
// * this stream exists in the map, and we can return it, or
|
|
// * this stream was already closed, then we can return the nil
|
|
if id <= m.highestStream {
|
|
m.mutex.RLock()
|
|
s := m.streams[id]
|
|
m.mutex.RUnlock()
|
|
return s, nil
|
|
}
|
|
|
|
m.mutex.Lock()
|
|
var start protocol.StreamID
|
|
if m.highestStream == 0 {
|
|
start = m.nextStream
|
|
} else {
|
|
start = m.highestStream + 4
|
|
}
|
|
for newID := start; newID <= id; newID += 4 {
|
|
m.streams[newID] = m.newStream(newID)
|
|
m.cond.Signal()
|
|
}
|
|
m.highestStream = id
|
|
s := m.streams[id]
|
|
m.mutex.Unlock()
|
|
return s, nil
|
|
}
|
|
|
|
func (m *incomingItemsMap) DeleteStream(id protocol.StreamID) error {
|
|
m.mutex.Lock()
|
|
defer m.mutex.Unlock()
|
|
|
|
if _, ok := m.streams[id]; !ok {
|
|
return fmt.Errorf("Tried to delete unknown stream %d", id)
|
|
}
|
|
delete(m.streams, id)
|
|
// queue a MAX_STREAM_ID frame, giving the peer the option to open a new stream
|
|
if numNewStreams := m.maxNumStreams - len(m.streams); numNewStreams > 0 {
|
|
m.maxStream = m.highestStream + protocol.StreamID(numNewStreams*4)
|
|
m.queueMaxStreamID(&wire.MaxStreamIDFrame{StreamID: m.maxStream})
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *incomingItemsMap) CloseWithError(err error) {
|
|
m.mutex.Lock()
|
|
m.closeErr = err
|
|
m.mutex.Unlock()
|
|
m.cond.Broadcast()
|
|
}
|