WIP remote host update

Author: mkienow-r7

lib/metasploit/framework/data_service/proxy/host_data_proxy.rb

@@ -6,7 +6,7 @@ def hosts(wspace = workspace, non_dead = false, addresses = nil, search_term = n
       opts = {}
       opts[:wspace] = wspace
       opts[:non_dead] = non_dead
-      opts[:addresses] = addresses
+      opts[:address] = addresses
       opts[:search_term] = search_term
       data_service.hosts(opts)
     rescue Exception => e
@@ -40,6 +40,16 @@ def report_hosts(hosts)
     end
   end
 
+  def update_host(opts)
+    begin
+      $stderr.puts "HostDataProxy.update_host(): opts = #{opts}"  # TODO: remove
+      data_service = self.get_data_service()
+      data_service.update_host(opts)
+    rescue Exception => e
+      elog "Problem updating host: #{e.message}"
+    end
+  end
+
   def delete_host(opts)
     begin
       data_service = self.get_data_service()

lib/metasploit/framework/data_service/remote/http/remote_host_data_service.rb

@@ -23,6 +23,16 @@ def report_hosts(hosts)
     self.post_data(HOST_API_PATH, hosts)
   end
 
+  def update_host(opts)
+    $stderr.puts "RemoteHostDataService.update_host(): opts = #{opts}"  # TODO: remove
+    path = HOST_API_PATH
+    if opts && opts[:id]
+      id = opts.delete(:id)
+      path = "#{HOST_API_PATH}/#{id}"
+    end
+    json_to_mdm_object(self.put_data(path, opts), HOST_MDM_CLASS, [])
+  end
+
   def delete_host(opts)
     json_to_mdm_object(self.delete_data(HOST_API_PATH, opts), HOST_MDM_CLASS, [])
   end

lib/msf/core/db_manager/host.rb

@@ -143,6 +143,7 @@ def has_host?(wspace,addr)
 
   # Returns a list of all hosts in the database
   def hosts(opts)
+    $stderr.puts "DBManager::Host.hosts(): opts = #{opts}"  # TODO: remove
     wspace = opts[:workspace] || opts[:wspace] || workspace
     if wspace.kind_of? String
       wspace = find_workspace(wspace)
@@ -152,7 +153,9 @@ def hosts(opts)
 
       conditions = {}
       conditions[:state] = [Msf::HostState::Alive, Msf::HostState::Unknown] if opts[:non_dead]
-      conditions[:address] = opts[:addresses] if opts[:addresses] && !opts[:addresses].empty?
+      conditions[:address] = opts[:address] if opts[:address] && !opts[:address].empty?
+      conditions[:id] = opts[:id] if opts[:id] && !opts[:id].empty?
+      $stderr.puts "DBManager::Host.hosts(): conditions = #{conditions}"  # TODO: remove
 
       if opts[:search_term] && !opts[:search_term].empty?
         column_search_conditions = Msf::Util::DBManager.create_all_column_search_conditions(Mdm::Host, opts[:search_term])
@@ -268,6 +271,19 @@ def report_host(opts)
   }
   end
 
+  def update_host(opts)
+    $stderr.puts "DBManager::Host.update_host(): opts = #{opts}"  # TODO: remove
+    # TODO: remove unneeded workspace from opts until it's not automatically added to remote requests
+    wspace = opts.delete(:workspace)
+    $stderr.puts "DBManager::Host.update_host(): delete workspace from opts; wspace = #{wspace}"  # TODO: remove
+
+    ::ActiveRecord::Base.connection_pool.with_connection {
+      id = opts.delete(:id)
+      $stderr.puts "DBManager::Host.update_host(): id = #{id}, opts = #{opts}"  # TODO: remove
+      Mdm::Host.update(id, opts)
+    }
+  end
+
   def split_windows_os_name(os_name)
     return [] if os_name.nil?
     flavor_match = os_name.match(/Windows\s+(.*)/)

lib/msf/core/db_manager/http/servlet/host_servlet.rb

@@ -4,9 +4,14 @@ def self.api_path
     '/api/v1/hosts'
   end
 
+  def self.api_path_with_id
+    "#{HostServlet.api_path}/?:id?"
+  end
+
   def self.registered(app)
-    app.get HostServlet.api_path, &get_host
+    app.get HostServlet.api_path_with_id, &get_host
     app.post HostServlet.api_path, &report_host
+    app.put HostServlet.api_path_with_id, &update_host
     app.delete HostServlet.api_path, &delete_host
   end
 
@@ -17,8 +22,9 @@ def self.registered(app)
   def self.get_host
     lambda {
       begin
+        $stderr.puts "HostServlet.get_host(): params[:id] = #{params[:id]}" if params[:id]  # TODO: remove
         opts = parse_json_request(request, false)
-        data = get_db().hosts(params)
+        data = get_db().hosts(params.symbolize_keys)
         includes = [:loots]
         set_json_response(data, includes)
       rescue Exception => e
@@ -40,6 +46,24 @@ def self.report_host
     }
   end
 
+  def self.update_host
+    lambda {
+      begin
+        opts = parse_json_request(request, false)
+        $stderr.puts "HostServlet.update_host(): opts = #{opts}"  # TODO: remove
+        $stderr.puts "HostServlet.update_host(): params = #{params}"  # TODO: remove
+        # opts_and_params = params ? opts.merge(params.symbolize_keys) : opts
+        tmp_params = params.symbolize_keys
+        opts[:id] = tmp_params[:id] if tmp_params[:id]
+        $stderr.puts "HostServlet.update_host(): opts = #{opts}"  # TODO: remove
+        data = get_db().update_host(opts)
+        set_json_response(data)
+      rescue Exception => e
+        set_error_on_response(e)
+      end
+    }
+  end
+
   def self.delete_host
     lambda {
       begin

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

@@ -231,17 +231,22 @@ def cmd_hosts_help
             cmd_hosts("-h")
           end
 
-          def change_host_info(rws, data)
-            if rws == [nil]
+          def change_host_info(host_ranges, data)
+            if host_ranges == [nil]
               print_error("In order to change the host info, you must provide a range of hosts")
               return
             end
 
-            rws.each do |rw|
-              rw.each do |ip|
-                id = framework.db.get_host(:address => ip).id
-                framework.db.hosts.update(id, :info => data)
-                framework.db.report_note(:host => ip, :type => 'host.info', :data => data)
+            $stderr.puts "Msf::Ui::Console::CommandDispatcher::Db.change_host_info(): host_ranges = #{host_ranges}"
+            each_host_range_chunk(host_ranges) do |host_search|
+              $stderr.puts "Msf::Ui::Console::CommandDispatcher::Db.change_host_info(): host_search = #{host_search}\n"
+              break if !host_search.nil? && host_search.empty?
+
+              framework.db.hosts(framework.db.workspace, false, host_search).each do |host|
+                $stderr.puts "Msf::Ui::Console::CommandDispatcher::Db.change_host_info(): host = #{host}, host.id = #{host.id}, host.address = #{host.address}, host.workspace_id = #{host.workspace_id}"
+                framework.db.update_host(id: host.id, info: data)
+                framework.db.report_note(:host => host.address, :type => 'host.info', :data => data)
+                $stderr.puts "************************************\n"
               end
             end
           end