Merge branch 'wchen-r7-print_warning' into rapid7

[Closes rapid7#899]

Author: egypt

lib/msf/core/exploit/http/server.rb

@@ -80,6 +80,12 @@ def print_error(msg='')
 	def print_debug(msg='')
 		(cli) ? super("#{cli.peerhost.ljust(16)} #{self.shortname} - #{msg}") : super
 	end
+	#
+	# :category: print_* overrides
+	# Prepends client and module name if inside a thread with a #cli
+	def print_warning(msg='')
+		(cli) ? super("#{cli.peerhost.ljust(16)} #{self.shortname} - #{msg}") : super
+	end
 
 	# :category: print_* overrides
 	# Prepends client and module name if inside a thread with a #cli
@@ -101,6 +107,11 @@ def vprint_error(msg='')
 	def vprint_debug(msg='')
 		(cli) ? super("#{cli.peerhost.ljust(16)} #{self.shortname} - #{msg}") : super
 	end
+	# :category: print_* overrides
+	# Prepends client and module name if inside a thread with a #cli
+	def vprint_warning(msg='')
+		(cli) ? super("#{cli.peerhost.ljust(16)} #{self.shortname} - #{msg}") : super
+	end
 
 
 	#

lib/msf/core/module.rb

@@ -208,6 +208,10 @@ def print_good(msg='')
 		super(print_prefix + msg)
 	end
 
+	def print_warning(msg='')
+		super(print_prefix + msg)
+	end
+
 
 	#
 	# Overwrite the Subscriber print_line to do custom prefixes
@@ -241,6 +245,10 @@ def vprint_line(msg)
 	def vprint_debug(msg)
 		print_debug(msg) if datastore['VERBOSE'] || framework.datastore['VERBOSE']
 	end
+	# Verbose version of #print_warning
+	def vprint_warning(msg)
+		print_warning(msg) if datastore['VERBOSE'] || framework.datastore['VERBOSE']
+	end
 
 	#
 	# Returns the module's framework full reference name.  This is the

lib/msf/core/plugin.rb

@@ -140,6 +140,14 @@ def print_line(msg='')
 		output.print_line(msg) if (output)
 	end
 
+	#
+	# Prints a warning
+	#
+	def print_warning(msg='')
+		output.print_warning(msg) if (output)
+	end
+
+
 	#
 	# Prints a message with no decoration.
 	#

lib/rex/io/bidirectional_pipe.rb

@@ -98,6 +98,10 @@ def print_status(msg='')
 		print_line('[*] ' + msg)
 	end
 
+	def print_warning(msg='')
+		print_warning('[!] ' + msg)
+	end
+
 	#
 	# Wrappers for the pipe_input methods
 	#

lib/rex/script/base.rb

@@ -9,6 +9,7 @@ def print_line(msg); end
 		def print_status(msg); end
 		def print_good(msg); end
 		def print_error(msg); end
+		def print_warning(msg); end
 	end
 
 	attr_accessor :client, :framework, :path, :error, :args

lib/rex/ui/output.rb

@@ -45,6 +45,12 @@ def print_status(msg='')
 	def print_line(msg='')
 	end
 
+	#
+	# Prints a warning
+	#
+	def print_warning(msg='')
+	end
+
 	#
 	# Prints a message with no decoration.
 	#

lib/rex/ui/subscriber.rb

@@ -66,6 +66,16 @@ def print_debug(msg='')
 			end
 		end
 
+		#
+		# Wraps user_output.print_warning
+		#
+		def print_warning(msg='')
+			if (user_output)
+				print_blank_line if user_output.prompting?
+				user_output.print_warning(msg)
+			end
+		end
+
 		#
 		# Wraps user_output.print
 		#

lib/rex/ui/text/dispatcher_shell.rb

@@ -81,6 +81,13 @@ def print_good(msg = '')
 			shell.print_good(msg)
 		end
 
+		#
+		# Wraps shell.print_warning
+		#
+		def print_warning(msg = '')
+			shell.print_warning(msg)
+		end
+
 		#
 		# Wraps shell.print
 		#

lib/rex/ui/text/output.rb

@@ -67,6 +67,10 @@ def print_line(msg = '')
 		print(msg + "\n")
 	end
 
+	def print_warning(msg = '')
+		print_line("%bld%yel[!]%clr #{msg}")
+	end
+
 	def print(msg = '')
 		print_raw(substitute_colors(msg))
 	end

lib/rex/ui/text/shell.rb

@@ -303,6 +303,16 @@ def print_line(msg='')
 		log_output(output.print_line(msg))
 	end
 
+	#
+	# Prints a warning message to the output handle.
+	#
+	def print_warning(msg='')
+		return if (disable_output == true)
+
+		self.on_print_proc.call(msg) if self.on_print_proc
+		log_output(output.print_warning(msg))
+	end
+
 	#
 	# Prints a raw message to the output handle.
 	#