ip_to_geo : reset connection after 50 requests

plugins/post_analysis/iptogeo.py

@@ -45,12 +45,15 @@ class IPToGeo(object):
               4 : 'Bad IP version',
               5 : 'Unsupported IP version',
               6 : 'IP not found'}
-    
+
+    MAX_REQUESTS = 50
+
     def __init__(self, remote_addr='127.0.0.1', remote_port=53333, timeout=None, family=socket.AF_INET):
         self._remote_addr = remote_addr
         self._remote_port = remote_port
         self._timeout = timeout
         self._family = family
+        self._nb_requests_sent = 0
 
         self._create_socket()
 
@@ -108,7 +111,20 @@ class IPToGeo(object):
         (cc0, cc1, cc2, cc3) = struct.unpack_from('BBBB', packet, 7*4)
 
         return (ip_res, '%c%c%c%c' % (cc0, cc1, cc2, cc3))
-        
+
+    def _send_request(self, packet):
+        self._nb_requests_sent += 1
+        if self._nb_requests_sent == self.MAX_REQUESTS:
+            self.close()
+            self._create_socket()
+            self._nb_requests_sent = 0
+        try:
+            self._socket.send(packet)
+        except IOError, e:
+            # Give another chance (we may have been disconnected due to timeout)
+            self._create_socket()
+            self._socket.send(packet)
+    
     def ip_to_geo(self, ip):
         ip_type = IPToGeo.IPV4
         if ip.find('.') >= 0: