fix nessus when no dns resolution its enabled

infobyte · Jun 2, 2022 · 5cc16a3 · 5cc16a3
1 parent 63bf561
commit 5cc16a3
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/faraday_plugins/plugins/repo/nessus/DTO.py b/faraday_plugins/plugins/repo/nessus/DTO.py
@@ -357,6 +357,11 @@ def host_fqdn(self) -> str:
         _dict = self.dict_tags
         return _dict.get("host-fqdn", None)
 
+    @property
+    def host_rdns(self) -> str:
+        _dict = self.dict_tags
+        return _dict.get("host-rdns", None)
+
     @property
     def dict_tags(self):
         host_tags = {}

diff --git a/faraday_plugins/plugins/repo/nessus/plugin.py b/faraday_plugins/plugins/repo/nessus/plugin.py
@@ -69,11 +69,17 @@ def parse_compliance_data(data: dict):
                 compliance_data[compliance_name] = value
         return compliance_data
 
-    @staticmethod
-    def map_properties(host: ReportHost):
+    def map_properties(self, host: ReportHost):
+        if self.hostname_resolution:
+            name = host.host_properties.host_ip if host.host_properties.host_ip else host.name
+        else:
+            name = host.name
+        hostnames = [host.host_properties.host_fqdn]
+        if host.host_properties.host_rdns and host.host_properties.host_rdns not in hostnames:
+            hostnames.append(host.host_properties.host_rdns)
         return {
-            "name": host.host_properties.host_ip if host.host_properties.host_ip else host.name,
-            "hostnames": host.host_properties.host_fqdn,
+            "name": name,
+            "hostnames": hostnames,
             "mac": host.host_properties.mac_address,
             "os": host.host_properties.operating_system
         }