Expose option for setting autonat throttling

config/config.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"crypto/rand"
 	"fmt"
+	"time"
 
 	"github.com/libp2p/go-libp2p-core/connmgr"
 	"github.com/libp2p/go-libp2p-core/crypto"
@@ -83,10 +84,12 @@ type Config struct {
 
 	Routing RoutingC
 
-	EnableAutoRelay bool
-	Reachability    network.Reachability
-	AutoNATService  bool
-	StaticRelays    []peer.AddrInfo
+	EnableAutoRelay   bool
+	Reachability      network.Reachability
+	AutoNATService    bool
+	AutoNATThrottling bool
+	AutoNATThrottles  [2]int
+	StaticRelays      []peer.AddrInfo
 }
 
 func (cfg *Config) makeSwarm(ctx context.Context) (*swarm.Swarm, error) {
@@ -280,9 +283,16 @@ func (cfg *Config) NewNode(ctx context.Context) (host.Host, error) {
 		}
 	}
 
-	autonatOpts := []autonat.Option{autonat.UsingAddresses(func() []ma.Multiaddr {
-		return addrF(h.AllAddrs())
-	})}
+	autonatOpts := []autonat.Option{
+		autonat.UsingAddresses(func() []ma.Multiaddr {
+			return addrF(h.AllAddrs())
+		}),
+	}
+	if cfg.AutoNATThrottling {
+		autonatOpts = append(autonatOpts,
+			autonat.WithThrottling(cfg.AutoNATThrottles[0], 1*time.Minute),
+			autonat.WithPeerThrottling(cfg.AutoNATThrottles[1]))
+	}
 	if cfg.AutoNATService {
 		autonatPrivKey, _, err := crypto.GenerateEd25519Key(rand.Reader)
 		if err != nil {

options.go

@@ -307,6 +307,19 @@ func EnableNATService() Option {
 	}
 }
 
+// OverrideNATServiceThrottling changes the default rate limiting configured in helping
+// other peers determine their reachability status. When set, the host will limit
+// the number of requests it responds to in each 60 second period to the set
+// numbers. A value of '0' disables throttling.
+func OverrideNATServiceThrottling(globalThrottle, perPeerThrottle int) Option {
+	return func(cfg *Config) error {
+		cfg.AutoNATThrottling = true
+		cfg.AutoNATThrottles[0] = globalThrottle
+		cfg.AutoNATThrottles[1] = perPeerThrottle
+		return nil
+	}
+}
+
 // FilterAddresses configures libp2p to never dial nor accept connections from
 // the given addresses. FilterAddresses should be used for cases where the
 // addresses you want to deny are known ahead of time.