http3: keep datagram receive loop running on datagram for unknown stream (#5136)

http3/conn.go

@@ -308,11 +308,10 @@ func (c *connection) receiveDatagrams() error {
 		streamID := protocol.StreamID(4 * quarterStreamID)
 		c.streamMx.Lock()
 		dg, ok := c.streams[streamID]
+		c.streamMx.Unlock()
 		if !ok {
-			c.streamMx.Unlock()
+			continue
-			return nil
 		}
-		c.streamMx.Unlock()
 		dg.enqueue(b[n:])
 	}
 }

http3/conn_test.go

@@ -392,6 +392,7 @@ var _ = Describe("Connection", func() {
 				close(delivered)
 				return b, nil
 			})
+			qconn.EXPECT().ReceiveDatagram(gomock.Any()).Return(nil, errors.New("test done"))
 			go func() {
 				defer GinkgoRecover()
 				conn.handleUnidirectionalStreams(nil)