[p2p] introduce account rate limit (#4545)

Author: envestcc

chainservice/builder.go

@@ -11,6 +11,7 @@ import (
 	"net/url"
 	"time"
 
+	"github.com/iotexproject/go-pkgs/cache"
 	"github.com/iotexproject/iotex-address/address"
 	"github.com/iotexproject/iotex-election/committee"
 	"github.com/iotexproject/iotex-proto/golang/iotextypes"
@@ -102,6 +103,13 @@ func (builder *Builder) SetP2PAgent(agent p2p.Agent) *Builder {
 	return builder
 }
 
+func (builder *Builder) SetAccountRateLimit(r int) *Builder {
+	builder.createInstance()
+	builder.cs.accRateLimitCfg = r
+	builder.cs.rateLimiters = cache.NewThreadSafeLruCache(10000)
+	return builder
+}
+
 // SetRPCStats sets the RPCStats instance
 func (builder *Builder) SetRPCStats(stats *nodestats.APILocalStats) *Builder {
 	builder.createInstance()

chainservice/chainservice.go

@@ -11,8 +11,11 @@ import (
 	"github.com/libp2p/go-libp2p/core/peer"
 	"github.com/pkg/errors"
 	"github.com/prometheus/client_golang/prometheus"
+	"go.uber.org/zap"
+	"golang.org/x/time/rate"
 	"google.golang.org/protobuf/proto"
 
+	"github.com/iotexproject/go-pkgs/cache"
 	"github.com/iotexproject/go-pkgs/hash"
 	"github.com/iotexproject/iotex-election/committee"
 	"github.com/iotexproject/iotex-proto/golang/iotexrpc"
@@ -79,6 +82,8 @@ type ChainService struct {
 	apiStats                 *nodestats.APILocalStats
 	blockTimeCalculator      *blockutil.BlockTimeCalculator
 	actionsync               *actsync.ActionSync
+	rateLimiters             cache.LRUCache
+	accRateLimitCfg          int
 }
 
 // Start starts the server
@@ -102,6 +107,17 @@ func (cs *ChainService) HandleAction(ctx context.Context, actPb *iotextypes.Acti
 	if err != nil {
 		return err
 	}
+	if cs.accRateLimitCfg > 0 {
+		sender := ""
+		if act.SenderAddress() != nil {
+			sender = act.SenderAddress().String()
+		}
+		limiter := cs.getRateLimiter(sender)
+		if !limiter.Allow() {
+			log.L().Debug("rate limit exceeded", zap.String("sender", act.SenderAddress().String()))
+			return nil
+		}
+	}
 	ctx = protocol.WithRegistry(ctx, cs.registry)
 	err = cs.actpool.Add(ctx, act)
 	if err != nil {
@@ -251,3 +267,12 @@ func (cs *ChainService) NewAPIServer(cfg api.Config, archive bool) (*api.ServerV
 
 	return svr, nil
 }
+
+func (cs *ChainService) getRateLimiter(sender string) *rate.Limiter {
+	limiter, exists := cs.rateLimiters.Get(sender)
+	if !exists {
+		limiter = rate.NewLimiter(rate.Limit(cs.accRateLimitCfg), 2*cs.accRateLimitCfg) // account limit request per second with a burst of *2
+		cs.rateLimiters.Add(sender, limiter)
+	}
+	return limiter.(*rate.Limiter)
+}

e2etest/local_actpool_test.go

@@ -183,6 +183,7 @@ func newActPoolConfig(t *testing.T) (config.Config, error) {
 	cfg.ActPool.MinGasPriceStr = "0"
 	cfg.Consensus.Scheme = config.NOOPScheme
 	cfg.Network.Port = testutil.RandomPort()
+	cfg.Network.AccountRateLimit = 0
 
 	sk, err := crypto.GenerateKey()
 	if err != nil {

p2p/agent.go

@@ -95,6 +95,8 @@ type (
 		PrivateNetworkPSK string              `yaml:"privateNetworkPSK"`
 		MaxPeers          int                 `yaml:"maxPeers"`
 		MaxMessageSize    int                 `yaml:"maxMessageSize"`
+		// AccountRateLimit is the maximum number of requests per second per account.
+		AccountRateLimit int `yaml:"accountRateLimit"`
 	}
 
 	// Agent is the agent to help the blockchain node connect into the P2P networks and send/receive messages
@@ -145,6 +147,7 @@ var DefaultConfig = Config{
 	PrivateNetworkPSK: "",
 	MaxPeers:          30,
 	MaxMessageSize:    p2p.DefaultConfig.MaxMessageSize,
+	AccountRateLimit:  100,
 }
 
 // NewDummyAgent creates a dummy p2p agent

server/itx/server.go

@@ -73,6 +73,7 @@ func newServer(cfg config.Config, testing bool) (*Server, error) {
 	var cs *chainservice.ChainService
 	builder := chainservice.NewBuilder(cfg)
 	builder.SetP2PAgent(p2pAgent)
+	builder.SetAccountRateLimit(cfg.Network.AccountRateLimit)
 	rpcStats := nodestats.NewAPILocalStats()
 	builder.SetRPCStats(rpcStats)
 	if testing {