add methods needed for connection-level FlowController

ref #39

flow_controller.go

@@ -23,13 +23,21 @@ type flowController struct {
 	mutex sync.RWMutex
 }
 
-func newFlowController(connectionParametersManager *handshake.ConnectionParametersManager) *flowController {
+func newFlowController(streamID protocol.StreamID, connectionParametersManager *handshake.ConnectionParametersManager) *flowController {
-	return &flowController{
+	fc := flowController{
-		sendFlowControlWindow:             connectionParametersManager.GetSendStreamFlowControlWindow(),
-		receiveFlowControlWindow:          connectionParametersManager.GetReceiveStreamFlowControlWindow(),
 		receiveWindowUpdateThreshold:      protocol.WindowUpdateThreshold,
 		receiveFlowControlWindowIncrement: protocol.ReceiveStreamFlowControlWindowIncrement,
 	}
+
+	if streamID == 0 {
+		fc.sendFlowControlWindow = connectionParametersManager.GetSendConnectionFlowControlWindow()
+		fc.receiveFlowControlWindow = connectionParametersManager.GetReceiveConnectionFlowControlWindow()
+	} else {
+		fc.sendFlowControlWindow = connectionParametersManager.GetSendStreamFlowControlWindow()
+		fc.receiveFlowControlWindow = connectionParametersManager.GetReceiveStreamFlowControlWindow()
+	}
+
+	return &fc
 }
 
 func (c *flowController) AddBytesSent(n protocol.ByteCount) {
@@ -62,13 +70,27 @@ func (c *flowController) SendWindowSize() protocol.ByteCount {
 	return c.sendFlowControlWindow - c.bytesSent
 }
 
-func (c *flowController) UpdateHighestReceived(n protocol.ByteCount) {
+// UpdateHighestReceived updates the highestReceived value, if the byteOffset is higher
+// Should **only** be used for the stream-level FlowController
+func (c *flowController) UpdateHighestReceived(byteOffset protocol.ByteCount) protocol.ByteCount {
 	c.mutex.Lock()
 	defer c.mutex.Unlock()
 
-	if n > c.highestReceived {
+	if byteOffset > c.highestReceived {
-		c.highestReceived = n
+		increment := byteOffset - c.highestReceived
+		c.highestReceived = byteOffset
+		return increment
 	}
+	return 0
+}
+
+// IncrementHighestReceived adds an increment to the highestReceived value
+// Should **only** be used for the connection-level FlowController
+func (c *flowController) IncrementHighestReceived(increment protocol.ByteCount) {
+	c.mutex.Lock()
+	defer c.mutex.Unlock()
+
+	c.highestReceived += increment
 }
 
 func (c *flowController) AddBytesRead(n protocol.ByteCount) {

flow_controller_test.go

@@ -105,16 +105,24 @@ var _ = Describe("Flow controller", func() {
 
 		It("updates the highestReceived", func() {
 			controller.highestReceived = 1337
-			controller.UpdateHighestReceived(1338)
+			increment := controller.UpdateHighestReceived(1338)
+			Expect(increment).To(Equal(protocol.ByteCount(1338 - 1337)))
 			Expect(controller.highestReceived).To(Equal(protocol.ByteCount(1338)))
 		})
 
 		It("does not decrease the highestReceived", func() {
 			controller.highestReceived = 1337
-			controller.UpdateHighestReceived(1000)
+			increment := controller.UpdateHighestReceived(1000)
+			Expect(increment).To(Equal(protocol.ByteCount(0)))
 			Expect(controller.highestReceived).To(Equal(protocol.ByteCount(1337)))
 		})
 
+		It("increases the highestReceived by a given increment", func() {
+			controller.highestReceived = 1337
+			controller.IncrementHighestReceived(123)
+			Expect(controller.highestReceived).To(Equal(protocol.ByteCount(1337 + 123)))
+		})
+
 		It("detects a flow control violation", func() {
 			controller.UpdateHighestReceived(receiveFlowControlWindow + 1)
 			Expect(controller.CheckFlowControlViolation()).To(BeTrue())

stream.go

@@ -51,7 +51,7 @@ func newStream(session streamHandler, connectionParameterManager *handshake.Conn
 	s := &stream{
 		session:        session,
 		streamID:       StreamID,
-		flowController: newFlowController(connectionParameterManager),
+		flowController: newFlowController(StreamID, connectionParameterManager),
 	}
 
 	s.newFrameOrErrCond.L = &s.mutex