prevent nodes that refuse to roll back from spamming

bismuthfoundation · Sep 19, 2019 · f2bc8a5 · f2bc8a5
1 parent 087dd15
commit f2bc8a5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/node.py b/node.py
@@ -790,6 +790,10 @@ def handle(self):
                                     send(self.request, "blocknfhb")  # announce we are on hyperblocks
                                 send(self.request, data)
 
+                                if node.peers.warning(self.request, peer_ip, "Forked", 2):
+                                    node.logger.app_log.info(f"{peer_ip} banned")
+                                    break
+
                             else:
                                 node.logger.app_log.info(f"Inbound: Client is at block {client_block}")  # now check if we have any newer
 

diff --git a/worker.py b/worker.py
@@ -167,6 +167,9 @@ def worker(host, port, node):
                                 send(s, "blocknfhb")
                             send(s, data)
 
+                            if node.peers.warning(s, peer_ip, "Forked", 1):
+                                raise ValueError(f"{peer_ip} is banned")
+
                         else:
                             node.logger.app_log.warning(
                                 f"Outbound: Node is at block {client_block}")  # now check if we have any newer