119 lines
2.2 KiB
Ruby
119 lines
2.2 KiB
Ruby
# -*- coding: binary -*-
|
|
require 'rex/ui'
|
|
|
|
begin
|
|
require 'windows_console_color_support'
|
|
rescue ::LoadError
|
|
end
|
|
|
|
module Rex
|
|
module Ui
|
|
module Text
|
|
|
|
###
|
|
#
|
|
# This class implements output against standard out.
|
|
#
|
|
###
|
|
class Output::Stdio < Rex::Ui::Text::Output
|
|
#
|
|
# Attributes
|
|
#
|
|
|
|
# @!attribute io
|
|
# The raw `IO` backing this Text output. Defaults to `$stdout`
|
|
#
|
|
# @return [#flush, #puts, #write]
|
|
attr_writer :io
|
|
|
|
#
|
|
# Constructor
|
|
#
|
|
|
|
# @param options [Hash{Symbol => IO}]
|
|
# @option options [IO]
|
|
def initialize(options={})
|
|
options.assert_valid_keys(:io)
|
|
|
|
super()
|
|
|
|
self.io = options[:io]
|
|
end
|
|
|
|
#
|
|
# Methods
|
|
#
|
|
|
|
def flush
|
|
io.flush
|
|
end
|
|
|
|
# IO to write to.
|
|
#
|
|
# @return [IO] Default to `$stdout`
|
|
def io
|
|
@io ||= $stdout
|
|
end
|
|
|
|
# Use ANSI Control chars to reset prompt position for async output
|
|
# SEE https://github.com/rapid7/metasploit-framework/pull/7570
|
|
def print_line(msg = '')
|
|
# TODO: there are unhandled quirks in async output buffering that
|
|
# we have not solved yet, for instance when loading meterpreter
|
|
# extensions, supporting Windows, printing output from commands, etc.
|
|
# Remove this guard when issues are resolved.
|
|
=begin
|
|
if (/mingw/ =~ RUBY_PLATFORM)
|
|
print(msg + "\n")
|
|
return
|
|
end
|
|
print("\033[s") # Save cursor position
|
|
print("\r\033[K" + msg + "\n")
|
|
if input and input.prompt
|
|
print("\r\033[K")
|
|
print(input.prompt.tr("\001\002", ''))
|
|
print(input.line_buffer.tr("\001\002", ''))
|
|
print("\033[u\033[B") # Restore cursor, move down one line
|
|
end
|
|
=end
|
|
|
|
print(msg + "\n")
|
|
end
|
|
|
|
#
|
|
# Prints the supplied message to standard output.
|
|
#
|
|
def print_raw(msg = '')
|
|
if (Rex::Compat.is_windows and supports_color?)
|
|
WindowsConsoleColorSupport.new(io).write(msg)
|
|
else
|
|
io.print(msg)
|
|
end
|
|
|
|
io.flush
|
|
|
|
msg
|
|
end
|
|
alias_method :write, :print_raw
|
|
|
|
def supports_color?
|
|
case config[:color]
|
|
when true
|
|
return true
|
|
when false
|
|
return false
|
|
else # auto
|
|
if (Rex::Compat.is_windows)
|
|
return true
|
|
end
|
|
term = Rex::Compat.getenv('TERM')
|
|
return (term and term.match(/(?:vt10[03]|xterm(?:-color)?|linux|screen|rxvt)/i) != nil)
|
|
end
|
|
end
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|
|
|