mirrors/uquic
1
0
Fork 0
mirror of https://github.com/refraction-networking/uquic.git synced 2025-04-03 20:27:35 +03:00
Code Issues Projects Releases Packages Wiki Activity

exit framer loop early if no space is left in packet

Browse source
This commit is contained in:
Lucas Clemente 2016-09-16 21:32:57 +02:00
parent 2970ee3ac5
commit e8ef0de5b4
3 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

5
stream_framer.go
View file

@ -125,6 +125,11 @@ func (f *streamFramer) maybePopNormalFrames(maxBytes protocol.ByteCount) (res []
res = append(res, frame)
currentLen += frameHeaderBytes + frame.DataLen()
if currentLen == maxBytes {
return false, nil
}
frame = &frames.StreamFrame{DataLenPresent: true}
return true, nil
}

2
streams_map.go
View file

@ -140,10 +140,10 @@ func (m *streamsMap) RoundRobinIterate(fn streamLambda) error {
if err != nil {
return err
}
m.roundRobinIndex = (m.roundRobinIndex + 1) % numStreams
if !cont {
break
}
m.roundRobinIndex = (m.roundRobinIndex + 1) % numStreams
}
return nil
}

8
streams_map_test.go
View file

@ -321,7 +321,7 @@ var _ = Describe("Streams Map", func() {
Expect(m.roundRobinIndex).To(Equal(3))
})
It("picks up at the index where it last stopped", func() {
It("picks up at the index+1 where it last stopped", func() {
fn := func(str *stream) (bool, error) {
lambdaCalledForStream = append(lambdaCalledForStream, str.StreamID())
numIterations++
@ -334,7 +334,7 @@ var _ = Describe("Streams Map", func() {
Expect(err).ToNot(HaveOccurred())
Expect(numIterations).To(Equal(2))
Expect(lambdaCalledForStream).To(Equal([]protocol.StreamID{4, 5}))
Expect(m.roundRobinIndex).To(Equal(1))
Expect(m.roundRobinIndex).To(Equal(2))
numIterations = 0
lambdaCalledForStream = lambdaCalledForStream[:0]
fn2 := func(str *stream) (bool, error) {
@ -347,8 +347,8 @@ var _ = Describe("Streams Map", func() {
}
err = m.RoundRobinIterate(fn2)
Expect(err).ToNot(HaveOccurred())
Expect(numIterations).To(Equal(3))
Expect(lambdaCalledForStream).To(Equal([]protocol.StreamID{5, 6, 7}))
Expect(numIterations).To(Equal(2))
Expect(lambdaCalledForStream).To(Equal([]protocol.StreamID{6, 7}))
})
It("adjust the RoundRobinIndex when deleting an element in front", func() {