limit remote_tx_hashes set to 10000 items

cyberaa · Nov 5, 2012 · 62c10bd · 62c10bd
1 parent d1c47c2
commit 62c10bd
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/p2pool/p2p.py b/p2pool/p2p.py
@@ -325,13 +325,15 @@ def handle_bestblock(self, header):
         ('tx_hashes', pack.ListType(pack.IntType(256))),
     ])
     def handle_have_tx(self, tx_hashes):
-        assert self.remote_tx_hashes.isdisjoint(tx_hashes)
+        #assert self.remote_tx_hashes.isdisjoint(tx_hashes)
         self.remote_tx_hashes.update(tx_hashes)
+        while len(self.remote_tx_hashes) > 10000:
+            self.remote_tx_hashes.pop()
     message_losing_tx = pack.ComposedType([
         ('tx_hashes', pack.ListType(pack.IntType(256))),
     ])
     def handle_losing_tx(self, tx_hashes):
-        assert self.remote_tx_hashes.issuperset(tx_hashes)
+        #assert self.remote_tx_hashes.issuperset(tx_hashes)
         self.remote_tx_hashes.difference_update(tx_hashes)