diff --git a/flask_soc_site/src/host_lookup.py b/flask_soc_site/src/host_lookup.py index 29bf81e..a7f940c 100644 --- a/flask_soc_site/src/host_lookup.py +++ b/flask_soc_site/src/host_lookup.py @@ -39,13 +39,13 @@ class Lookedup(object): self.ip_address = socket.gethostbyname(self.domain) self.metadata = domain_info(self.domain) self.email_security = spf_dmarc(self.domain) - self.vt, self.vt_dict = virustotal_api.analyse2(self.host, self.host_type) + self.vt = virustotal_api.analyse2(self.host, self.host_type) self.abuseipdb = abuseipdb_api.analyse(self.ip_address) return self def ip_lookup(self): - self.ip_address = self.host self.domain = socket.gethostbyaddr(self.host)[0] + self.ip_address = socket.gethostbyname(self.domain) self.metadata = domain_info(self.host) self.domain = self.metadata["domain_name"] self.email_security = spf_dmarc(self.domain) @@ -69,7 +69,7 @@ class Lookedup(object): self.metadata = domain_info(self.domain) self.ip_address = socket.gethostbyname(self.domain) self.email_security = spf_dmarc(self.domain) - self.vt, self.vt_dict = virustotal_api.analyse2(self.domain, self.host_type) + self.vt = virustotal_api.analyse2(self.domain, self.host_type) self.abuseipdb = abuseipdb_api.analyse(self.ip_address) return self diff --git a/flask_soc_site/templates/results-overview-new.html b/flask_soc_site/templates/results-overview-new.html index d3c4196..758cf47 100644 --- a/flask_soc_site/templates/results-overview-new.html +++ b/flask_soc_site/templates/results-overview-new.html @@ -28,7 +28,7 @@ {% for host in hosts %} - {{host.host}} + {{host.ip_address}} Score: {{host.vt['score']}} / {{host.vt['total']}}
Last updated: {{host.vt['last_update']}}