webtransport: return error before wrapping opened / accepted streams (#…

…1775)
libp2p · Sep 22, 2022 · 031ec50 · 031ec50
1 parent 17a342c
commit 031ec50
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/p2p/transport/webtransport/conn.go b/p2p/transport/webtransport/conn.go
@@ -46,12 +46,18 @@ 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)
-	return &stream{str}, err
+	if err != nil {
+		return nil, err
+	}
+	return &stream{str}, nil
 }
 
 func (c *conn) AcceptStream() (network.MuxedStream, error) {
 	str, err := c.session.AcceptStream(context.Background())
-	return &stream{str}, err
+	if err != nil {
+		return nil, err
+	}
+	return &stream{str}, nil
 }
 
 func (c *conn) Close() error             { return c.session.Close() }