mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 20:07:17 +00:00
1b430e0c17
* deadline warn * add proper godoc information * clarify * Merge refs/heads/master into corrupted-clock * fix broken build * Merge refs/heads/master into corrupted-clock * Merge refs/heads/master into corrupted-clock
55 lines
2.2 KiB
Go
55 lines
2.2 KiB
Go
package sync
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/libp2p/go-libp2p-core/network"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var defaultReadDuration = ttfbTimeout
|
|
var defaultWriteDuration = params.BeaconNetworkConfig().RespTimeout // RESP_TIMEOUT
|
|
|
|
// SetRPCStreamDeadlines sets read and write deadlines for libp2p-based connection streams.
|
|
func SetRPCStreamDeadlines(stream network.Stream) {
|
|
SetStreamReadDeadline(stream, defaultReadDuration)
|
|
SetStreamWriteDeadline(stream, defaultWriteDuration)
|
|
}
|
|
|
|
// SetStreamReadDeadline for reading from libp2p connection streams, deciding when to close
|
|
// a connection based on a particular duration.
|
|
//
|
|
// NOTE: libp2p uses the system clock time for determining the deadline so we use
|
|
// time.Now() instead of the synchronized roughtime.Now(). If the system
|
|
// time is corrupted (i.e. time does not advance), the node will experience
|
|
// issues being able to properly close streams, leading to unexpected failures and possible
|
|
// memory leaks.
|
|
func SetStreamReadDeadline(stream network.Stream, duration time.Duration) {
|
|
if err := stream.SetReadDeadline(time.Now().Add(duration)); err != nil {
|
|
log.WithError(err).WithFields(logrus.Fields{
|
|
"peer": stream.Conn().RemotePeer(),
|
|
"protocol": stream.Protocol(),
|
|
"direction": stream.Stat().Direction,
|
|
}).Debug("Failed to set stream deadline")
|
|
}
|
|
}
|
|
|
|
// SetStreamWriteDeadline for writing to libp2p connection streams, deciding when to close
|
|
// a connection based on a particular duration.
|
|
//
|
|
// NOTE: libp2p uses the system clock time for determining the deadline so we use
|
|
// time.Now() instead of the synchronized roughtime.Now(). If the system
|
|
// time is corrupted (i.e. time does not advance), the node will experience
|
|
// issues being able to properly close streams, leading to unexpected failures and possible
|
|
// memory leaks.
|
|
func SetStreamWriteDeadline(stream network.Stream, duration time.Duration) {
|
|
if err := stream.SetWriteDeadline(time.Now().Add(duration)); err != nil {
|
|
log.WithError(err).WithFields(logrus.Fields{
|
|
"peer": stream.Conn().RemotePeer(),
|
|
"protocol": stream.Protocol(),
|
|
"direction": stream.Stat().Direction,
|
|
}).Debug("Failed to set stream deadline")
|
|
}
|
|
}
|