limit ping duration to 30s

so that ping streams cannot be squatted
libp2p · Jun 3, 2024 · a46972f · a46972f
1 parent f52cec1
commit a46972f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/p2p/protocol/ping/ping.go b/p2p/protocol/ping/ping.go
@@ -20,8 +20,9 @@ import (
 var log = logging.Logger("ping")
 
 const (
-	PingSize    = 32
-	pingTimeout = time.Second * 60
+	PingSize     = 32
+	pingTimeout  = 10 * time.Second
+	pingDuration = 30 * time.Second
 
 	ID = "/ipfs/ping/1.0.0"
 
@@ -52,6 +53,8 @@ func (p *PingService) PingHandler(s network.Stream) {
 	}
 	defer s.Scope().ReleaseMemory(PingSize)
 
+	s.SetDeadline(time.Now().Add(pingDuration))
+
 	buf := pool.Get(PingSize)
 	defer pool.Put(buf)