From 3502bf4d1652431c9bd674a98e4d15e0e9518232 Mon Sep 17 00:00:00 2001
From: Joost Agterhoek <joostagterhoek@protonmail.com>
Date: Mon, 14 Jul 2025 21:34:30 +0200
Subject: [PATCH] fixed IP and URL VirusTotal lookup, corrected template IP
 address error

---
 flask_soc_site/src/host_lookup.py                  | 6 +++---
 flask_soc_site/templates/results-overview-new.html | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

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 @@
       <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>