Include :workspace in db_import opts

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

@@ -17,6 +17,7 @@ module DataProxyAutoLoader
   autoload :CredentialDataProxy, 'metasploit/framework/data_service/proxy/credential_data_proxy'
   autoload :NmapDataProxy, 'metasploit/framework/data_service/proxy/nmap_data_proxy'
   autoload :DbExportDataProxy, 'metasploit/framework/data_service/proxy/db_export_data_proxy'
+  autoload :DbImportDataProxy, 'metasploit/framework/data_service/proxy/db_import_data_proxy'
   autoload :VulnAttemptDataProxy, 'metasploit/framework/data_service/proxy/vuln_attempt_data_proxy'
 
   include ServiceDataProxy
@@ -33,5 +34,6 @@ module DataProxyAutoLoader
   include CredentialDataProxy
   include NmapDataProxy
   include DbExportDataProxy
+  include DbImportDataProxy
   include VulnAttemptDataProxy
 end

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

@@ -0,0 +1,21 @@
+module DbImportDataProxy
+  def import(opts, &block)
+    begin
+      data_service = self.get_data_service
+      add_opts_workspace(opts)
+      data_service.import(opts, &block)
+    rescue Exception => e
+      self.log_error(e, "Problem generating DB Export")
+    end
+  end
+
+  def import_file(opts, &block)
+    begin
+      data_service = self.get_data_service
+      add_opts_workspace(opts)
+      data_service.import_file(opts, &block)
+    rescue Exception => e
+      self.log_error(e, "Problem generating DB Export")
+    end
+  end
+end

lib/metasploit/framework/data_service/stubs/db_export_service.rb

@@ -0,0 +1,5 @@
+module DbExportDataService
+  def run_db_export(path, format)
+    raise 'DbExportDataService#run_db_export is not implemented'
+  end
+end

lib/metasploit/framework/data_service/stubs/db_import_service.rb

@@ -0,0 +1,9 @@
+module DbImportDataService
+  def import(opts, &block)
+    raise 'DbImportDataService#import is not implemented'
+  end
+
+  def import_file(opts, &block)
+    raise 'DbImportDataService#import_file is not implemented'
+  end
+end

lib/msf/core/db_manager/import.rb

@@ -85,14 +85,14 @@ module Msf::DBManager::Import
   # import_file_detect will raise an error if the filetype
   # is unknown.
   def import(args={}, &block)
-    wspace = args[:wspace] || args['wspace'] || workspace
+    wspace = Msf::Util::DBManager.process_opts_workspace(args, framework)
     preserve_hosts = args[:task].options["DS_PRESERVE_HOSTS"] if args[:task].present? && args[:task].options.present?
     wspace.update_attribute(:import_fingerprint, true)
     existing_host_ids = wspace.hosts.map(&:id)
     data = args[:data] || args['data']
     ftype = import_filetype_detect(data)
     yield(:filetype, @import_filedata[:type]) if block
-    self.send "import_#{ftype}".to_sym, args, &block
+    self.send "import_#{ftype}".to_sym, args.merge(workspace: wspace.name), &block
     if preserve_hosts
       new_host_ids = Mdm::Host.where(workspace: wspace).map(&:id)
       (new_host_ids - existing_host_ids).each do |id|
@@ -111,7 +111,7 @@ module Msf::DBManager::Import
   #
   def import_file(args={}, &block)
     filename = args[:filename] || args['filename']
-    wspace = args[:wspace] || args['wspace'] || workspace
+    wspace = Msf::Util::DBManager.process_opts_workspace(args, framework)
     @import_filedata            = {}
     @import_filedata[:filename] = filename
 
@@ -148,9 +148,9 @@ module Msf::DBManager::Import
     REXML::Security.entity_expansion_text_limit = 51200
 
     if block
-      import(args.merge(:data => data)) { |type,data| yield type,data }
+      import(args.merge(data: data, workspace: wspace.name)) { |type,data| yield type,data }
     else
-      import(args.merge(:data => data))
+      import(args.merge(data: data, workspace: wspace.name))
     end
   end