#!/usr/bin/env python # -*- coding: utf-8 -*- # # Copyright 2016 Grégory Soutadé # # This file is part of iptogeo. # # iptogeo is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # iptogeo is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with iptogeo. If not, see . # import socket import struct class IPToGeoException(Exception): pass class IPToGeo(object): MAGIC = 0x179E08EF VERSION = 1 REQ = 1 RESP = 0 IPV4 = 4 IPV6 = 16 IP_NOT_FOUND = 6 PACKET_SIZE = 32 ERRORS = {1 : 'Bad magic', 2 : 'Bad version', 3 : 'Bad request field' , 4 : 'Bad IP version', 5 : 'Unsupported IP version', 6 : 'IP not found'} 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._create_socket() def _create_socket(self): self._socket = socket.socket(self._family, socket.SOCK_STREAM) if not self._timeout is None: self._socket.settimeout(self._timeout) self._socket.connect((self._remote_addr, self._remote_port)) def _create_request(self, ip): packet = '' packet += struct.pack('