Merge pull request #336 from mozilla/add_ptr_lookup

Add PTR lookup on IP-only target
mozilla · Mar 16, 2017 · dc77e3d · dc77e3d
2 parents aff90a3 + 6d086a1
commit dc77e3d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/lib/ssh_scan/scan_engine.rb b/lib/ssh_scan/scan_engine.rb
@@ -47,7 +47,7 @@ def scan_target(socket, opts)
         client = SSHScan::Client.new(target, port, timeout)
         client.connect()
         result = client.get_kex_result()
-        result[:hostname] = ""
+        result[:hostname] = target.resolve_ptr
         return result if result.include?(:error)
       end
 

diff --git a/lib/string_ext.rb b/lib/string_ext.rb
@@ -55,7 +55,18 @@ def resolve_fqdn_as_ipv4(timeout = 3)
   end
 
   def resolve_fqdn
-    @fqdn ||= TCPSocket.gethostbyname(self)[3]
+    TCPSocket.gethostbyname(self)[3]
+  end
+
+  def resolve_ptr(timeout = 3)
+    begin
+      Timeout::timeout(timeout) {
+        reversed_dns = Resolv.new.getname(self)
+        return reversed_dns
+      }
+    rescue Timeout::Error
+      return ""
+    end
   end
 
   def fqdn?