fix(tcpreuse): handle connection that failed to be sampled

libp2p · Nov 12, 2024 · 279be8d · 279be8d
1 parent 10045d1
commit 279be8d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/p2p/transport/tcpreuse/demultiplex.go b/p2p/transport/tcpreuse/demultiplex.go
@@ -50,6 +50,9 @@ func identifyConnType(c manet.Conn) (DemultiplexedConnType, manet.Conn, error) {
 
 	s, c, err := sampledconn.PeekBytes(c)
 	if err != nil {
+		if c == nil {
+			return 0, nil, err
+		}
 		closeErr := c.Close()
 		return 0, nil, errors.Join(err, closeErr)
 	}