Exemplo n.º 1
0
Arquivo: cache.py Projeto: hile/seine
    def __init__(self, cache_path=None, tld_cache_path=None):
        self.cache_path = None

        if cache_path is not None:
            self.cache_path = os.path.expandvars(os.path.expanduser(cache_path))

        else:
            for f in CACHE_FILES:
                if os.path.isfile(f) and os.access(f, os.W_OK):
                    self.cache_path = f
                    break

                fdir = os.path.dirname(f)
                if not os.path.isdir(fdir):
                    try:
                        os.makedirs(os.path.dirname(f))
                    except IOError:
                        continue
                    except OSError:
                        continue

                if not os.path.isfile(f):
                    try:
                        open(f, 'w').write('\n')
                        os.unlink(f)
                        self.cache_path = f
                        break
                    except IOError:
                        continue
                    except OSError:
                        continue

        if self.cache_path is None:
            raise WhoisError('ERROR: No whois cache path defined')

        try:
            self.tlds = TLDCache(tld_cache_path)
            if not self.tlds.is_downloaded:
                self.tlds.download()
        except DNSCacheError as e:
            raise WhoisEror(e)

        if os.path.isfile(self.cache_path):
            self.load()
Exemplo n.º 2
0
Arquivo: cache.py Projeto: hile/seine
class WhoisServerCache(dict):
    def __init__(self, cache_path=None, tld_cache_path=None):
        self.cache_path = None

        if cache_path is not None:
            self.cache_path = os.path.expandvars(os.path.expanduser(cache_path))

        else:
            for f in CACHE_FILES:
                if os.path.isfile(f) and os.access(f, os.W_OK):
                    self.cache_path = f
                    break

                fdir = os.path.dirname(f)
                if not os.path.isdir(fdir):
                    try:
                        os.makedirs(os.path.dirname(f))
                    except IOError:
                        continue
                    except OSError:
                        continue

                if not os.path.isfile(f):
                    try:
                        open(f, 'w').write('\n')
                        os.unlink(f)
                        self.cache_path = f
                        break
                    except IOError:
                        continue
                    except OSError:
                        continue

        if self.cache_path is None:
            raise WhoisError('ERROR: No whois cache path defined')

        try:
            self.tlds = TLDCache(tld_cache_path)
            if not self.tlds.is_downloaded:
                self.tlds.download()
        except DNSCacheError as e:
            raise WhoisEror(e)

        if os.path.isfile(self.cache_path):
            self.load()

    def load(self):
        if not os.path.isfile(self.cache_path):
            raise WhoisError('No such file: %s' % self.cache_path)

        try:
            cache = ConfigParser.ConfigParser()
            cache.read(self.cache_path)
        except OSError as e:
            raise WhoisError(e)
        except IOError as e:
            raise WhoisError(e)

        for tld in cache.sections():
            try:
                ipv4_addresses = []
                ipv6_addresses = []
                if cache.has_option(tld, 'ipv4_addresses') and cache.get(tld, 'ipv4_addresses'):
                    ipv4_addresses.extend(cache.get(tld, 'ipv4_addresses').split(','))
                if cache.has_option(tld, 'ipv6_addresses') and cache.get(tld, 'ipv6_addresses'):
                    ipv6_addresses.extend(cache.get(tld, 'ipv6_addresses').split(','))
                entry = TLDWhoisServerList(tld, ipv4_addresses, ipv6_addresses)
                self[tld] = entry
            except ConfigParser.NoOptionError:
                logger.debug('Invalid cache entry for %s' % tld)

    def save(self):
        if len(self) == 0:
            return

        cache_dir = os.path.dirname(self.cache_path)

        if not os.path.isdir(cache_dir):
            try:
                os.makedirs(cache_dir)
            except IOError as e:
                raise WhoisError('Error creating cache directory %s: %s' % (cache_dir, e))
            except OSError as e:
                raise WhoisError('Error creating cache directory %s: %s' % (cache_dir, e))

        try:
            cache = ConfigParser.ConfigParser()
            for entry in self.values():
                section = cache.add_section(entry.tld)
                cache.set(entry.tld, 'ipv4_addresses', ','.join(x.ipaddress for x in entry.ipv4_addresses))
                cache.set(entry.tld, 'ipv6_addresses', ','.join(x.address for x in entry.ipv6_addresses))
            cache.write(open(self.cache_path, 'w'))

        except OSError as e:
            raise WhoisError('Error updating cache %s: %s' % (self.cache_path, e))
        except IOError as e:
            raise WhoisError('Error updating cache %s: %s' % (self.cache_path, e))

    def get(self, tld, timeout=WHOIS_SERVER_TIMEOUT):
        try:
            timeout = int(timeout)
        except ValueError:
            raise WhoisError('Invalid timeout value: %s' % timeout)

        if tld not in self.tlds:
            raise WhoisError('Invalid TLD name: %s' % tld)

        if tld not in self:
            details = {}
            name = '.'.join([tld, SEARCH_DOMAIN])
            try:
                socket.setdefaulttimeout(timeout)
                name, aliases, addresses = socket.gethostbyname_ex(name)
                entry = TLDWhoisServerList(tld, addresses)
                self[tld] = entry

            except socket.timeout:
                raise WhoisError('Timeout resolving whois server %s' % name)
            except socket.gaierror as e:
                raise WhoisError(e)
            except ValueError as e:
                raise WhoisError(e)

        return self[tld]

    def query(self, name):
        try:
            tld = name.rstrip('.').split('.')[-1]
        except ValueError:
            raise WhoisError('Error parsing TLD from %s' % name)

        tldservers = self.get(tld)
        return tldservers.query(name)