diff options
author | Jimmy Hu <jimmy.hu@dexon.org> | 2019-01-24 10:39:36 +0800 |
---|---|---|
committer | Wei-Ning Huang <w@dexon.org> | 2019-03-12 12:19:09 +0800 |
commit | 2fd6ae06ffe660d27766b40bf130514150efc4cc (patch) | |
tree | fd3fe59de949681879d185fb70a5b985c142919e | |
parent | d711fef2d53b6dd8f55060c8463f7b38716c44c7 (diff) | |
download | dexon-2fd6ae06ffe660d27766b40bf130514150efc4cc.tar dexon-2fd6ae06ffe660d27766b40bf130514150efc4cc.tar.gz dexon-2fd6ae06ffe660d27766b40bf130514150efc4cc.tar.bz2 dexon-2fd6ae06ffe660d27766b40bf130514150efc4cc.tar.lz dexon-2fd6ae06ffe660d27766b40bf130514150efc4cc.tar.xz dexon-2fd6ae06ffe660d27766b40bf130514150efc4cc.tar.zst dexon-2fd6ae06ffe660d27766b40bf130514150efc4cc.zip |
dex: Add rate limit for pullVote (#169)
-rw-r--r-- | dex/handler.go | 28 |
1 files changed, 21 insertions, 7 deletions
diff --git a/dex/handler.go b/dex/handler.go index a745bbfc7..b8b39c6f4 100644 --- a/dex/handler.go +++ b/dex/handler.go @@ -80,6 +80,8 @@ const ( recordChanSize = 10240 maxPullPeers = 3 + + pullVoteRateLimit = 10 * time.Second ) // errIncompatibleConfig is returned if the requested protocols and configs are @@ -97,13 +99,14 @@ type ProtocolManager struct { fastSync uint32 // Flag whether fast sync is enabled (gets disabled if we already have blocks) acceptTxs uint32 // Flag whether we're considered synchronised (enables transaction processing) - txpool txPool - nodeTable *nodeTable - gov governance - blockchain *core.BlockChain - chainconfig *params.ChainConfig - cache *cache - maxPeers int + txpool txPool + nodeTable *nodeTable + gov governance + blockchain *core.BlockChain + chainconfig *params.ChainConfig + cache *cache + nextPullVote *sync.Map + maxPeers int downloader *downloader.Downloader fetcher *fetcher.Fetcher @@ -166,6 +169,7 @@ func NewProtocolManager( gov: gov, blockchain: blockchain, cache: newCache(5120, dexDB.NewDatabase(chaindb)), + nextPullVote: &sync.Map{}, chainconfig: config, newPeerCh: make(chan *peer), noMorePeers: make(chan struct{}), @@ -250,6 +254,8 @@ func (pm *ProtocolManager) removePeer(id string) { } log.Debug("Removing Ethereum peer", "peer", id) + pm.nextPullVote.Delete(peer.ID()) + // Unregister the peer from the downloader and Ethereum peer set pm.downloader.UnregisterPeer(id) if err := pm.peers.Unregister(id); err != nil { @@ -863,6 +869,14 @@ func (pm *ProtocolManager) handleMsg(p *peer) error { if !pm.isBlockProposer { break } + next, ok := pm.nextPullVote.Load(p.ID()) + if ok { + nextTime := next.(time.Time) + if nextTime.After(time.Now()) { + break + } + } + pm.nextPullVote.Store(p.ID(), time.Now().Add(pullVoteRateLimit)) var pos coreTypes.Position if err := msg.Decode(&pos); err != nil { return errResp(ErrDecode, "msg %v: %v", msg, err) |