fixed possible DoS - there was no limit on the number of addr entries

p2pool · Jan 8, 2013 · e04f198 · e04f198
1 parent 5d5f6f9
commit e04f198
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/p2pool/p2p.py b/p2pool/p2p.py
@@ -634,7 +634,8 @@ def got_addr(self, (host, port), services, timestamp):
             old_services, old_first_seen, old_last_seen = self.addr_store[host, port]
             self.addr_store[host, port] = services, old_first_seen, max(old_last_seen, timestamp)
         else:
-            self.addr_store[host, port] = services, timestamp, timestamp
+            if len(self.addr_store) < 10000:
+                self.addr_store[host, port] = services, timestamp, timestamp
 
     def handle_shares(self, shares, peer):
         print 'handle_shares', (shares, peer)