fix: stream read timeout

libp2p · Dec 17, 2020 · e4c7f4c · e4c7f4c
1 parent 703926f
commit e4c7f4c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/svc.go b/svc.go
@@ -19,6 +19,8 @@ import (
 	manet "github.com/multiformats/go-multiaddr/net"
 )
 
+var streamReadTimeout = 60 * time.Second
+
 // AutoNATService provides NAT autodetection services to other peers
 type autoNATService struct {
 	ctx          context.Context
@@ -50,6 +52,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()