webtransport: return error before wrapping opened / accepted streams

libp2p · Sep 22, 2022 · d46dc5c · d46dc5c
1 parent 17a342c
commit d46dc5c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/p2p/transport/webtransport/conn.go b/p2p/transport/webtransport/conn.go
@@ -46,11 +46,17 @@ func newConn(tr tpt.Transport, sess *webtransport.Session, sconn *connSecurityMu
 
 func (c *conn) OpenStream(ctx context.Context) (network.MuxedStream, error) {
 	str, err := c.session.OpenStreamSync(ctx)
+	if err != nil {
+		return nil, err
+	}
 	return &stream{str}, err
 }
 
 func (c *conn) AcceptStream() (network.MuxedStream, error) {
 	str, err := c.session.AcceptStream(context.Background())
+	if err != nil {
+		return nil, err
+	}
 	return &stream{str}, err
 }