fixed IP and URL VirusTotal lookup, corrected template IP address error

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
 

flask_soc_site/templates/results-overview-new.html

@@ -28,7 +28,7 @@
       <tbody>
         {% for host in hosts %}
         <tr>
-          <td>{{host.host}}</td>
+          <td>{{host.ip_address}}</td>
           <td>Score: {{host.vt['score']}} / {{host.vt['total']}}<br>
             Last updated: {{host.vt['last_update']}}<br>
           </td>