Resolve rapid7#3870, reinstance creds -R

Author: Meatballs1

lib/msf/ui/console/command_dispatcher/db.rb

@@ -677,6 +677,7 @@ def cmd_creds_help
     print_line "  -s <svc names>        List creds matching comma-separated service names"
     print_line "  -u,--user <regex>     List users that match this regex"
     print_line "  -t,--type <type>      List creds that match the following types: (password,ntlm,hash)"
+    print_line "  -R,--rhosts           Set RHOSTS from the results of the search"
 
     print_line
     print_line "Examples, listing:"
@@ -762,6 +763,9 @@ def creds_search(*args)
     host_ranges = []
     port_ranges = []
     svcs        = []
+    rhosts      = []
+
+    set_rhosts = false
 
     #cred_table_columns = [ 'host', 'port', 'user', 'pass', 'type', 'proof', 'active?' ]
     cred_table_columns = [ 'host', 'service', 'public', 'private', 'realm', 'private_type' ]
@@ -808,6 +812,8 @@ def creds_search(*args)
         end
       when "-d"
         mode = :delete
+      when '-R', '--rhosts'
+        set_rhosts = true
       else
         # Anything that wasn't an option is a host to search for
         unless (arg_host_range(arg, host_ranges))
@@ -879,6 +885,7 @@ def creds_search(*args)
             core.private ? core.private.class.model_name.human : "",
           ]
         else
+          rhosts = []
           core.logins.each do |login|
             if svcs.present? && !svcs.include?(login.service.name)
               next
@@ -896,6 +903,7 @@ def creds_search(*args)
               next
             end
             row = [ login.service.host.address ]
+            rhosts << login.service.host.address
             if login.service.name.present?
               row << "#{login.service.port}/#{login.service.proto} (#{login.service.name})"
             else
@@ -924,7 +932,8 @@ def creds_search(*args)
         ::File.open(output_file, "wb") { |f| f.write(tbl.to_csv) }
         print_status("Wrote creds to #{output_file}")
       end
-      
+
+      set_rhosts_from_addrs(rhosts.uniq) if set_rhosts
       print_status("Deleted #{delete_count} creds") if delete_count > 0
     }
   end