From 445676b9322710cf18736a1e0299768dbc0207ca Mon Sep 17 00:00:00 2001
From: HD Moore <hd_moore@rapid7.com>
Date: Sat, 20 Feb 2010 05:50:38 +0000
Subject: [PATCH] Merge in the current workspace name

git-svn-id: file:///home/svn/framework3/trunk@8567 4d416f70-5f16-0410-b530-b9f4589650da
---
 lib/msf/core/auxiliary/report.rb | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/lib/msf/core/auxiliary/report.rb b/lib/msf/core/auxiliary/report.rb
index dc6959a88c..843294b05a 100644
--- a/lib/msf/core/auxiliary/report.rb
+++ b/lib/msf/core/auxiliary/report.rb
@@ -8,11 +8,25 @@ module Msf
 
 module Auxiliary::Report
 
+
+	def initialize(info = {})
+		super
+
+		register_options([
+				OptString.new('WORKSPACE', [ false, "The name of the workspace to report data into"])
+			], Auxiliary::Report)
+	end
+
 	# Shortcut method for detecting when the DB is active
 	def db
 		framework.db.active
 	end
 
+	def myworkspace
+		return @myworkspace if @myworkspace
+		@myworkspace = Msf::DBManager::Workspace.find_by_name(datastore['WORKSPACE']) || framework.db.workspace
+	end
+
 	#
 	# Report a host's liveness and attributes such as operating system and service pack
 	#
@@ -23,11 +37,13 @@ module Auxiliary::Report
 	#
 	def report_host(opts)
 		return if not db
+		opts = {:workspace => myworkspace}.merge(opts)
 		framework.db.report_host(opts)
 	end
 
 	def get_host(addr)
 		return if not db
+		opts = {:workspace => myworkspace}.merge(opts)
 		framework.db.get_host(addr)
 	end
 
@@ -43,11 +59,13 @@ module Auxiliary::Report
 	#
 	def report_client(opts={})
 		return if not db
+		opts = {:workspace => myworkspace}.merge(opts)
 		framework.db.report_client(opts)
 	end
 
 	def get_client(opts={})
 		return if not db
+		opts = {:workspace => myworkspace}.merge(opts)
 		framework.db.get_client(opts)
 	end
 
@@ -56,21 +74,25 @@ module Auxiliary::Report
 	#
 	def report_service(opts={})
 		return if not db
+		opts = {:workspace => myworkspace}.merge(opts)
 		framework.db.report_service(opts)
 	end
 
 	def report_note(opts={})
 		return if not db
+		opts = {:workspace => myworkspace}.merge(opts)
 		framework.db.report_note(opts)
 	end
 
 	def report_auth_info(opts={})
 		return if not db
+		opts = {:workspace => myworkspace}.merge(opts)
 		framework.db.report_auth_info(opts)
 	end
 
 	def report_vuln(opts={})
 		return if not db
+		opts = {:workspace => myworkspace}.merge(opts)
 		framework.db.report_vuln(opts)
 	end