small bugfix for asset_inventory

blacklanternsecurity · Oct 13, 2023 · 6b4a321 · 6b4a321
1 parent e4f2617
commit 6b4a321
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bbot/modules/output/asset_inventory.py b/bbot/modules/output/asset_inventory.py
@@ -92,7 +92,7 @@ def sort_key(asset):
             ports = getattr(asset, "ports", set())
             ports = [str(p) for p in sorted([int(p) for p in asset.ports])]
             ips = sorted([str(i) for i in getattr(asset, "ip_addresses", [])])
-            host = getattr(asset, "host", "")
+            host = self.helpers.make_ip_type(getattr(asset, "host", ""))
             if host and isinstance(host, str):
                 _, domain = self.helpers.split_domain(host)
                 if domain:
@@ -197,7 +197,7 @@ def __init__(self, host):
 
     def absorb_csv_row(self, row):
         # host
-        host = row.get("Host", "").strip()
+        host = make_ip_type(row.get("Host", "").strip())
         if host and not is_ip(host):
             self.host = host
         # ips