Luke Imhoff
40 lines
878 B
Ruby
40 lines
878 B
Ruby
# Methods for print messages with status indicators
|
|
module Msf::Module::UI::Message
|
|
autoload :Verbose, 'msf/core/module/ui/message/verbose'
|
|
|
|
include Msf::Module::UI::Message::Verbose
|
|
|
|
def print_error(msg='')
|
|
super(print_prefix + msg)
|
|
end
|
|
|
|
def print_good(msg='')
|
|
super(print_prefix + msg)
|
|
end
|
|
|
|
def print_prefix
|
|
ret = ''
|
|
if (datastore['TimestampOutput'] =~ /^(t|y|1)/i) || (
|
|
framework && framework.datastore['TimestampOutput'] =~ /^(t|y|1)/i
|
|
)
|
|
prefix = "[#{Time.now.strftime("%Y.%m.%d-%H:%M:%S")}] "
|
|
|
|
xn ||= datastore['ExploitNumber']
|
|
xn ||= framework.datastore['ExploitNumber']
|
|
if xn.is_a?(Fixnum)
|
|
prefix << "[%04d] " % xn
|
|
end
|
|
|
|
ret = prefix
|
|
end
|
|
ret
|
|
end
|
|
|
|
def print_status(msg='')
|
|
super(print_prefix + msg)
|
|
end
|
|
|
|
def print_warning(msg='')
|
|
super(print_prefix + msg)
|
|
end
|
|
end |