diff --git a/plugins/post_analysis/iptogeo.py b/plugins/post_analysis/iptogeo.py index 76a8d1c..4af69c6 100644 --- a/plugins/post_analysis/iptogeo.py +++ b/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: