Add the spool command for redirecting output to both the console and a file

git-svn-id: file:///home/svn/framework3/trunk@13028 4d416f70-5f16-0410-b530-b9f4589650da

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

@@ -66,7 +66,6 @@ class Core
 	@@search_opts = Rex::Parser::Arguments.new(
 		"-h" => [ false, "Help banner."                                   ])
 
-
 	# The list of data store elements that cannot be set when in defanged
 	# mode.
 	DefangedProhibitedDataStoreElements = [ "MsfModulePaths" ]
@@ -106,6 +105,7 @@ class Core
 			"unsetg"   => "Unsets one or more global variables",
 			"use"      => "Selects a module by name",
 			"version"  => "Show the framework and console library version numbers",
+			"spool"    => "Write console output into a file as well the screen"
 		}
 	end
 
@@ -289,7 +289,7 @@ class Core
 		if(avdwarn)
 			avdwarn.map{|line| print_error(line) }
 		end
-		
+
 	end
 
 	#
@@ -1138,8 +1138,8 @@ class Core
 				match += val + " "
 			end
 		}
-		
-		tbl = generate_module_table("Matching Modules")	
+
+		tbl = generate_module_table("Matching Modules")
 		framework.modules.each do |m|
 			o = framework.modules.create(m[0])
 			if not o.search_filter(match)
@@ -1169,7 +1169,7 @@ class Core
 		print_line "Usage: search [keywords]"
 		print_line
 		print_line "Keywords:"
-		{ 
+		{
 			"name"     => "Modules with a matching descriptive name",
 			"path"     => "Modules with a matching path or reference name",
 			"platform" => "Modules affecting this platform",
@@ -1178,7 +1178,7 @@ class Core
 			"author"   => "Modules written by this author",
 			"cve"      => "Modules with a matching CVE ID",
 			"bid"      => "Modules with a matching Bugtraq ID",
-			"osvdb"    => "Modules with a matching OSVDB ID" 
+			"osvdb"    => "Modules with a matching OSVDB ID"
 		}.each_pair do |keyword, description|
 			print_line "  #{keyword.ljust 10}:  #{description}"
 		end
@@ -1190,6 +1190,30 @@ class Core
 		#print @@search_opts.usage
 	end
 
+	def cmd_spool(*args)
+		if args.include?('-h') or args.empty?
+			cmd_spool_help
+			return
+		end
+
+		if args[0] == "off"
+			driver.init_ui(driver.input, Rex::Ui::Text::Output::Stdio.new)
+			print_status("Spooling is now disabled")
+			return
+		end
+
+		driver.init_ui(driver.input, Rex::Ui::Text::Output::Tee.new(args[0]))
+		print_status("Spooling to file #{args[0]}...")
+	end
+
+	def cmd_spool_help
+		print_line "Usage: spool <off>|<filename>"
+		print_line
+		print_line "Example:"
+		print_line "  spool /tmp/console.log"
+		print_line
+	end
+
 	#
 	# Provides an interface to the sessions currently active in the framework.
 	#
@@ -1202,7 +1226,7 @@ class Core
 		cmds    = []
 		script  = nil
 		reset_ring = false
-		
+
 		# any arguments that don't correspond to an option or option arg will
 		# be put in here
 		extra   = []
@@ -1258,8 +1282,8 @@ class Core
 				when "-u"
 					method = 'upexec'
 					sid = val
-					
-				# Reset the ring buffer read pointer 
+
+				# Reset the ring buffer read pointer
 				when "-r"
 					reset_ring = true
 					method = 'reset_ring'
@@ -1695,7 +1719,7 @@ class Core
 					end
 				when 'sessions'
 					if (active_module and active_module.respond_to?(:compatible_sessions))
-						sessions = active_module.compatible_sessions 
+						sessions = active_module.compatible_sessions
 					else
 						sessions = framework.sessions.keys.sort
 					end

lib/rex/ui/text/output.rb

@@ -16,6 +16,7 @@ class Output < Rex::Ui::Output
 	require 'rex/ui/text/output/socket'
 	require 'rex/ui/text/output/buffer'
 	require 'rex/ui/text/output/file'
+	require 'rex/ui/text/output/tee'
 	require 'rex/ui/text/color'
 
 	include Rex::Ui::Text::Color

lib/rex/ui/text/output/tee.rb

@@ -0,0 +1,55 @@
+require 'rex/ui'
+
+module Rex
+module Ui
+module Text
+
+###
+#
+# This class implements output against a file and stdout
+#
+###
+class Output::Tee < Rex::Ui::Text::Output
+
+	attr_accessor :fd
+
+	def initialize(path)
+		self.fd = ::File.open(path, "ab")
+		super()
+	end
+
+	def supports_color?
+		case config[:color]
+		when true
+			return true
+		when false
+			return false
+		else # auto
+			term = Rex::Compat.getenv('TERM')
+			return (term and term.match(/(?:vt10[03]|xterm(?:-color)?|linux|screen|rxvt)/i) != nil)
+		end
+	end
+
+	#
+	# Prints the supplied message to file output.
+	#
+	def print_raw(msg = '')
+		$stdout.print(msg)
+		$stdout.flush
+
+		return if not self.fd
+		self.fd.write(msg)
+		self.fd.flush
+		msg
+	end
+
+	def close
+		self.fd.close if self.fd
+		self.fd = nil
+	end
+end
+
+end
+end
+end
+

lib/rex/ui/text/shell.rb

@@ -81,7 +81,7 @@ module Shell
 
 			self.input.output = self.output
 		end
-		update_prompt('')
+		update_prompt('') if not input.prompt
 	end
 
 	#