adulau/DomainClassifier
1
0
Fork 0
mirror of https://github.com/adulau/DomainClassifier.git synced 2024-09-18 10:36:18 +00:00
Code Issues Projects Releases Packages Wiki Activity

Exception handled for unknown origins

Browse source
This commit is contained in:
Alexandre Dulaunoy 2014-08-20 10:51:55 +02:00
parent 94482dcfbf
commit d116cfc37b
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
DomainClassifier/domainclassifier.py
View file

@ -146,12 +146,18 @@ class Extract:
for dom in self.validdomain:
if dom[1] == 'A':
ip = dom[2]
try:
orig = self.__origin(ipaddr=dom[2])[1]
except:
continue
if(orig == cc): self.localdom.append(dom)
elif dom[1] == 'CNAME':
cname = str(dom[2])
ip = socket.gethostbyname(cname)
try:
orig = self.__origin(ipaddr=ip)[1]
except:
continue
if(orig == cc): self.localdom.append(dom)
return self.localdom