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

add client functionality to the streamsMap

Browse source
This commit is contained in:
Marten Seemann 2016-12-13 11:44:40 +07:00
parent 6cb48aad71
commit 16da08a440
No known key found for this signature in database
GPG key ID: 3603F40B121FCDEA
6 changed files with 70 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

45
streams_map_test.go
View file

@ -59,7 +59,7 @@ var _ = Describe("Streams Map", func() {
maxIncomingStreams: 75,
maxOutgoingStreams: 60,
}
m = newStreamsMap(nil, cpm)
m = newStreamsMap(nil, protocol.PerspectiveServer, cpm)
})
Context("getting and creating streams", func() {
@ -77,7 +77,7 @@ var _ = Describe("Streams Map", func() {
Expect(m.numOutgoingStreams).To(BeZero())
})
Context("client-side streams", func() {
Context("client-side streams, as a server", func() {
It("rejects streams with even IDs", func() {
_, err := m.GetOrOpenStream(6)
Expect(err).To(MatchError("InvalidStreamID: attempted to open stream 6 from client-side"))
@ -129,7 +129,26 @@ var _ = Describe("Streams Map", func() {
})
})
Context("server-side streams", func() {
Context("client-side streams, as a client", func() {
BeforeEach(func() {
m.perspective = protocol.PerspectiveClient
})
It("rejects streams with odd IDs", func() {
_, err := m.GetOrOpenStream(5)
Expect(err).To(MatchError("InvalidStreamID: attempted to open stream 5 from server-side"))
})
It("gets new streams", func() {
s, err := m.GetOrOpenStream(6)
Expect(err).NotTo(HaveOccurred())
Expect(s.StreamID()).To(Equal(protocol.StreamID(6)))
Expect(m.numOutgoingStreams).To(Equal(uint32(1)))
Expect(m.numIncomingStreams).To(BeZero())
})
})
Context("server-side streams, as a server", func() {
It("rejects streams with odd IDs", func() {
_, err := m.OpenStream(5)
Expect(err).To(MatchError("InvalidStreamID: attempted to open stream 5 from server-side"))
@ -188,6 +207,26 @@ var _ = Describe("Streams Map", func() {
})
})
Context("server-side streams, as a client", func() {
BeforeEach(func() {
m.perspective = protocol.PerspectiveClient
})
It("rejects streams with even IDs", func() {
_, err := m.OpenStream(6)
Expect(err).To(MatchError("InvalidStreamID: attempted to open stream 6 from client-side"))
})
It("opens a new stream", func() {
s, err := m.OpenStream(7)
Expect(err).ToNot(HaveOccurred())
Expect(s).ToNot(BeNil())
Expect(s.StreamID()).To(Equal(protocol.StreamID(7)))
Expect(m.numOutgoingStreams).To(BeZero())
Expect(m.numIncomingStreams).To(Equal(uint32(1)))
})
})
Context("DoS mitigation", func() {
It("opens and closes a lot of streams", func() {
for i := 1; i < 2*protocol.MaxNewStreamIDDelta; i += 2 {