From 264d94cd2d6c1a604fd668a862482d8604827877 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= <raphael@vinot.info>
Date: Mon, 6 Nov 2023 21:17:34 +0100
Subject: [PATCH] fix: Pass IPs to whois instead of v4/v6...

---
 lookyloo/modules/uwhois.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/lookyloo/modules/uwhois.py b/lookyloo/modules/uwhois.py
index f8c8d533..d35d1d3e 100644
--- a/lookyloo/modules/uwhois.py
+++ b/lookyloo/modules/uwhois.py
@@ -32,7 +32,12 @@ class UniversalWhois(AbstractModule):
     def query_whois_hostnode(self, hostnode: HostNode) -> None:
         if hasattr(hostnode, 'resolved_ips'):
             ip: str
-            for ip in hostnode.resolved_ips:
+            if 'v4' in hostnode.resolved_ips and 'v6' in hostnode.resolved_ips:
+                _all_ips = set(hostnode.resolved_ips['v4']) | set(hostnode.resolved_ips['v6'])
+            else:
+                # old format
+                _all_ips = hostnode.resolved_ips
+            for ip in _all_ips:
                 self.whois(ip)
         if hasattr(hostnode, 'cnames'):
             cname: str