Only accept one mempool message per remote node

ShawnYun · Apr 18, 2020 · 359a3bb · 359a3bb
1 parent 5839014
commit 359a3bb
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/neo/Network/P2P/RemoteNode.ProtocolHandler.cs b/src/neo/Network/P2P/RemoteNode.ProtocolHandler.cs
@@ -30,6 +30,7 @@ protected override UInt256 GetKeyForItem((UInt256, DateTime) item)
         private readonly HashSetCache<UInt256> knownHashes = new HashSetCache<UInt256>(Blockchain.Singleton.MemPool.Capacity * 2 / 5);
         private readonly HashSetCache<UInt256> sentHashes = new HashSetCache<UInt256>(Blockchain.Singleton.MemPool.Capacity * 2 / 5);
         private bool verack = false;
+        private bool memPoolWasSent = false;
         private BloomFilter bloom_filter;
 
         private static readonly TimeSpan TimerInterval = TimeSpan.FromSeconds(30);
@@ -314,6 +315,14 @@ private void OnInvMessageReceived(InvPayload payload)
 
         private void OnMemPoolMessageReceived()
         {
+            if (memPoolWasSent)
+            {
+                Disconnect(true);
+                return;
+            }
+
+            memPoolWasSent = true;
+
             foreach (InvPayload payload in InvPayload.CreateGroup(InventoryType.TX, Blockchain.Singleton.MemPool.GetVerifiedTransactions().Select(p => p.Hash).ToArray()))
                 EnqueueMessage(Message.Create(MessageCommand.Inv, payload));
         }