Merge pull request #99 from libp2p/fix/timeout-incoming-stream-reads

Fix: Stream read timeout
libp2p · Feb 25, 2021 · 4b9ee56 · 4b9ee56
2 parents a107453 + e4c7f4c
commit 4b9ee56
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/svc.go b/svc.go
@@ -17,6 +17,8 @@ import (
 	ma "github.com/multiformats/go-multiaddr"
 )
 
+var streamReadTimeout = 60 * time.Second
+
 // AutoNATService provides NAT autodetection services to other peers
 type autoNATService struct {
 	ctx          context.Context
@@ -48,6 +50,8 @@ func newAutoNATService(ctx context.Context, c *config) (*autoNATService, error)
 }
 
 func (as *autoNATService) handleStream(s network.Stream) {
+	s.SetReadDeadline(time.Now().Add(streamReadTimeout))
+
 	defer s.Close()
 
 	pid := s.Conn().RemotePeer()