mirror of
https://github.com/adulau/DomainClassifier.git
synced 2024-11-07 11:56:25 +00:00
DomainClassifier Class extended
The class has been extended to add the localizedomain method to geolocalize DNS records associated for an existing domain. The localization rely on the Team Cymru ip2asn lookup via DNS.
This commit is contained in:
parent
e2d08d181f
commit
542c53dcca
1 changed files with 51 additions and 4 deletions
|
@ -1,9 +1,37 @@
|
||||||
|
#!/usr/bin/env python
|
||||||
|
"""DomainClassifier is a simple Python library to extract and classify Internet
|
||||||
|
domains from raw text files following their existence, localization or
|
||||||
|
attributes.
|
||||||
|
"""
|
||||||
|
|
||||||
import re
|
import re
|
||||||
import dns.resolver
|
import dns.resolver
|
||||||
|
import IPy
|
||||||
|
import socket
|
||||||
|
|
||||||
|
__author__ = "Alexandre Dulaunoy"
|
||||||
|
__copyright__ = "Copyright 2012, Alexandre Dulaunoy"
|
||||||
|
__license__ = "AGPL version 3"
|
||||||
|
__version__ = "0.0.1"
|
||||||
|
|
||||||
|
|
||||||
class Extract:
|
class Extract:
|
||||||
def __init__(self, rawtext = None):
|
def __init__(self, rawtext = None):
|
||||||
self.rawtext = rawtext
|
self.rawtext = rawtext
|
||||||
|
self.presolver = dns.resolver.Resolver()
|
||||||
|
self.presolver.nameservers = ['149.13.33.69']
|
||||||
|
|
||||||
|
def __origin(self, ipaddr=None):
|
||||||
|
|
||||||
|
if ipaddr:
|
||||||
|
clook = IPy.IP(str(ipaddr)).reverseName().replace('.in-addr.arpa.','.origin.asn.cymru.com')
|
||||||
|
a = self.presolver.query(clook, 'TXT')
|
||||||
|
if a:
|
||||||
|
x = str(a[0]).split("|")
|
||||||
|
return x[2].strip()
|
||||||
|
else:
|
||||||
|
return None
|
||||||
|
|
||||||
def domain(self):
|
def domain(self):
|
||||||
self.domain = []
|
self.domain = []
|
||||||
domain = re.compile(r'\b([a-zA-Z\d-]{,63}(\.[a-zA-Z\d-]{,63})+)\b')
|
domain = re.compile(r'\b([a-zA-Z\d-]{,63}(\.[a-zA-Z\d-]{,63})+)\b')
|
||||||
|
@ -13,21 +41,40 @@ class Extract:
|
||||||
|
|
||||||
return self.domain
|
return self.domain
|
||||||
|
|
||||||
def validdomain(self, rtype=['A','AAAA','SOA','MX','CNAME'], extended=None):
|
def validdomain(self, rtype=['A','AAAA','SOA','MX','CNAME'], extended=True):
|
||||||
if extended is None:
|
if extended is False:
|
||||||
self.validdomain = set()
|
self.validdomain = set()
|
||||||
else:
|
else:
|
||||||
self.validdomain = []
|
self.validdomain = []
|
||||||
for domain in self.domain:
|
for domain in self.domain:
|
||||||
for dnstype in rtype:
|
for dnstype in rtype:
|
||||||
try:
|
try:
|
||||||
answers = dns.resolver.query(domain, dnstype)
|
answers = self.presolver.query(domain, dnstype)
|
||||||
except:
|
except:
|
||||||
pass
|
pass
|
||||||
else:
|
else:
|
||||||
if extended is None:
|
if extended is False:
|
||||||
self.validdomain.add((domain))
|
self.validdomain.add((domain))
|
||||||
else:
|
else:
|
||||||
self.validdomain.append((domain,dnstype,answers[0]))
|
self.validdomain.append((domain,dnstype,answers[0]))
|
||||||
return self.validdomain
|
return self.validdomain
|
||||||
|
|
||||||
|
def localizedomain(self, cc=None):
|
||||||
|
self.localdom = []
|
||||||
|
|
||||||
|
for dom in self.validdomain:
|
||||||
|
if dom[1] == 'A':
|
||||||
|
ip = dom[2]
|
||||||
|
orig = self.__origin(ipaddr=dom[2])
|
||||||
|
if(orig == cc): self.localdom.append(dom)
|
||||||
|
elif dom[1] == 'CNAME':
|
||||||
|
cname = str(dom[2])
|
||||||
|
ip = socket.gethostbyname(cname)
|
||||||
|
orig = self.__origin(ipaddr=ip)
|
||||||
|
if(orig == cc): self.localdom.append(dom)
|
||||||
|
return self.localdom
|
||||||
|
|
||||||
|
def filterdomain(self,filter=None):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue