consistently return ErrUnexpectedEOF on fallbackSampledConn

libp2p · Dec 4, 2024 · 322cc44 · 322cc44
1 parent 2174a1e
commit 322cc44
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/p2p/transport/tcpreuse/internal/sampledconn/sampledconn_common.go b/p2p/transport/tcpreuse/internal/sampledconn/sampledconn_common.go
@@ -71,8 +71,11 @@ type ManetTCPConnInterface interface {
 
 func newFallbackSampledConn(conn ManetTCPConnInterface) (PeekedBytes, *fallbackPeekingConn, error) {
 	s := &fallbackPeekingConn{ManetTCPConnInterface: conn}
-	_, err := io.ReadFull(conn, s.peekedBytes[:])
+	n, err := io.ReadFull(conn, s.peekedBytes[:])
 	if err != nil {
+		if n == 0 && err == io.EOF {
+			err = io.ErrUnexpectedEOF
+		}
 		return s.peekedBytes, nil, err
 	}
 	return s.peekedBytes, s, nil