address review comments 2

p2p/net/swarm/swarm.go

@@ -850,11 +850,14 @@ func (c *connWithMetrics) Close() error {
 }
 
 func (c *connWithMetrics) CloseWithError(errCode network.ConnErrorCode) error {
-	c.metricsTracer.ClosedConnection(c.dir, time.Since(c.opened), c.ConnState(), c.LocalMultiaddr())
-	if ce, ok := c.CapableConn.(network.CloseWithErrorer); ok {
-		return ce.CloseWithError(errCode)
-	}
-	return c.CapableConn.Close()
+	c.once.Do(func() {
+		c.metricsTracer.ClosedConnection(c.dir, time.Since(c.opened), c.ConnState(), c.LocalMultiaddr())
+		if ce, ok := c.CapableConn.(network.CloseWithErrorer); ok {
+			c.closeErr = ce.CloseWithError(errCode)
+		}
+		c.closeErr = c.CapableConn.Close()
+	})
+	return c.closeErr
 }
 
 func (c *connWithMetrics) Stat() network.ConnStats {

p2p/protocol/circuitv2/relay/relay_test.go

@@ -309,7 +309,6 @@ func TestRelayLimitData(t *testing.T) {
 
 	rc := relay.DefaultResources()
 	rc.Limit.Duration = time.Second
-	// Due to yamux framing, 4 blocks of 1024 bytes will exceed the data limit
 	rc.Limit.Data = 4096
 
 	r, err := relay.New(hosts[1], relay.WithResources(rc))

p2p/test/transport/transport_test.go

@@ -869,7 +869,6 @@ func TestConnClosedWhenRemoteCloses(t *testing.T) {
 }
 
 func TestErrorCodes(t *testing.T) {
-
 	assertStreamErrors := func(s network.Stream, expectedError error) {
 		buf := make([]byte, 10)
 		_, err := s.Read(buf)
@@ -925,20 +924,46 @@ func TestErrorCodes(t *testing.T) {
 				require.NoError(t, err)
 				pingPong(s)
 
+				remoteStream := <-remoteStreamQ
+				defer remoteStream.Reset()
+
 				err = s.ResetWithError(42)
 				require.NoError(t, err)
 				assertStreamErrors(s, &network.StreamError{
 					ErrorCode: 42,
 					Remote:    false,
 				})
 
+				assertStreamErrors(remoteStream, &network.StreamError{
+					ErrorCode: 42,
+					Remote:    true,
+				})
+			})
+			t.Run("StreamResetWithErrorByRemote", func(t *testing.T) {
+				if tc.Name == "WebTransport" {
+					t.Skipf("skipping: %s, not implemented", tc.Name)
+					return
+				}
+				ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+				defer cancel()
+				s, err := client.NewStream(ctx, server.ID(), "/test")
+				require.NoError(t, err)
+				pingPong(s)
+
 				remoteStream := <-remoteStreamQ
-				defer remoteStream.Reset()
 
-				assertStreamErrors(remoteStream, &network.StreamError{
+				err = remoteStream.ResetWithError(42)
+				require.NoError(t, err)
+
+				assertStreamErrors(s, &network.StreamError{
 					ErrorCode: 42,
 					Remote:    true,
 				})
+
+				assertStreamErrors(remoteStream, &network.StreamError{
+					ErrorCode: 42,
+					Remote:    false,
+				})
 			})
 
 			t.Run("StreamResetByConnCloseWithError", func(t *testing.T) {
@@ -952,6 +977,9 @@ func TestErrorCodes(t *testing.T) {
 				require.NoError(t, err)
 				pingPong(s)
 
+				remoteStream := <-remoteStreamQ
+				defer remoteStream.Reset()
+
 				err = s.Conn().CloseWithError(42)
 				require.NoError(t, err)
 
@@ -960,16 +988,13 @@ func TestErrorCodes(t *testing.T) {
 					Remote:    false,
 				})
 
-				remoteStream := <-remoteStreamQ
-				defer remoteStream.Reset()
-
 				assertStreamErrors(remoteStream, &network.ConnError{
 					ErrorCode: 42,
 					Remote:    true,
 				})
 			})
 
-			t.Run("StreamResetByConnCloseWithError", func(t *testing.T) {
+			t.Run("NewStreamErrorByConnCloseWithError", func(t *testing.T) {
 				if tc.Name == "WebTransport" || tc.Name == "WebRTC" {
 					t.Skipf("skipping: %s, not implemented", tc.Name)
 					return

p2p/transport/quic/stream.go

@@ -25,22 +25,28 @@ func parseStreamError(err error) error {
 	}
 	se := &quic.StreamError{}
 	if errors.As(err, &se) {
-		code := se.ErrorCode
-		if code > math.MaxUint32 {
-			// TODO(sukunrt): do we need this?
-			code = reset
+		var code network.StreamErrorCode
+		if se.ErrorCode > math.MaxUint32 {
+			code = network.StreamCodeOutOfRange
+		} else {
+			code = network.StreamErrorCode(se.ErrorCode)
 		}
 		err = &network.StreamError{
-			ErrorCode:      network.StreamErrorCode(code),
+			ErrorCode:      code,
 			Remote:         se.Remote,
 			TransportError: se,
 		}
 	}
 	ae := &quic.ApplicationError{}
 	if errors.As(err, &ae) {
-		code := ae.ErrorCode
+		var code network.ConnErrorCode
+		if ae.ErrorCode > math.MaxUint32 {
+			code = network.ConnCodeOutOfRange
+		} else {
+			code = network.ConnErrorCode(ae.ErrorCode)
+		}
 		err = &network.ConnError{
-			ErrorCode:      network.ConnErrorCode(code),
+			ErrorCode:      code,
 			Remote:         ae.Remote,
 			TransportError: ae,
 		}

p2p/transport/webtransport/conn.go

@@ -78,7 +78,7 @@ func (c *conn) Close() error {
 	return err
 }
 
-func (c *conn) CloseWithError(errCode network.ConnErrorCode) error {
+func (c *conn) CloseWithError(_ network.ConnErrorCode) error {
 	return c.Close()
 }
 

p2p/transport/webtransport/stream.go

@@ -56,6 +56,11 @@ func (s *stream) Reset() error {
 	return nil
 }
 
+// ResetWithError resets the stream ignoring the error code. Error codes aren't
+// specified for WebTransport as the current implementation of WebTransport in
+// browsers(https://www.ietf.org/archive/id/draft-kinnear-webtransport-http2-02.html)
+// only supports 1 byte error codes. For more details, see
+// https://github.com/libp2p/specs/blob/4eca305185c7aef219e936bef76c48b1ab0a8b43/error-codes/README.md?plain=1#L84
 func (s *stream) ResetWithError(_ network.StreamErrorCode) error {
 	s.Stream.CancelRead(reset)
 	s.Stream.CancelWrite(reset)