98 lines
2.9 KiB
Ruby
98 lines
2.9 KiB
Ruby
#
|
|
# Project
|
|
#
|
|
|
|
require 'metasploit/framework/command'
|
|
require 'metasploit/framework/command/base'
|
|
|
|
# Based on pattern used for lib/rails/commands in the railties gem.
|
|
class Metasploit::Framework::Command::Console < Metasploit::Framework::Command::Base
|
|
|
|
# Provides an animated spinner in a seperate thread.
|
|
#
|
|
# See GitHub issue #4147, as this may be blocking some
|
|
# Windows instances, which is why Windows platforms
|
|
# should simply return immediately.
|
|
|
|
def spinner
|
|
return if Rex::Compat.is_windows
|
|
return if Rex::Compat.is_cygwin
|
|
return if $msf_spinner_thread
|
|
$msf_spinner_thread = Thread.new do
|
|
base_line = "[*] Starting the Metasploit Framework console..."
|
|
cycle = 0
|
|
loop do
|
|
%q{/-\|}.each_char do |c|
|
|
status = "#{base_line}#{c}\r"
|
|
cycle += 1
|
|
off = cycle % base_line.length
|
|
case status[off, 1]
|
|
when /[a-z]/
|
|
status[off, 1] = status[off, 1].upcase
|
|
when /[A-Z]/
|
|
status[off, 1] = status[off, 1].downcase
|
|
end
|
|
$stderr.print status
|
|
::IO.select(nil, nil, nil, 0.10)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def start
|
|
case parsed_options.options.subcommand
|
|
when :version
|
|
$stderr.puts "Framework Version: #{Metasploit::Framework::VERSION}"
|
|
else
|
|
spinner unless parsed_options.options.console.quiet
|
|
driver.run
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
# The console UI driver.
|
|
#
|
|
# @return [Msf::Ui::Console::Driver]
|
|
def driver
|
|
unless @driver
|
|
# require here so minimum loading is done before {start} is called.
|
|
require 'msf/ui'
|
|
|
|
@driver = Msf::Ui::Console::Driver.new(
|
|
Msf::Ui::Console::Driver::DefaultPrompt,
|
|
Msf::Ui::Console::Driver::DefaultPromptChar,
|
|
driver_options
|
|
)
|
|
end
|
|
|
|
@driver
|
|
end
|
|
|
|
def driver_options
|
|
unless @driver_options
|
|
options = parsed_options.options
|
|
|
|
driver_options = {}
|
|
driver_options['Config'] = options.framework.config
|
|
driver_options['ConfirmExit'] = options.console.confirm_exit
|
|
driver_options['DatabaseEnv'] = options.environment
|
|
driver_options['DatabaseMigrationPaths'] = options.database.migrations_paths
|
|
driver_options['DatabaseYAML'] = options.database.config
|
|
driver_options['DeferModuleLoads'] = options.modules.defer_loads
|
|
driver_options['DisableBanner'] = options.console.quiet
|
|
driver_options['DisableDatabase'] = options.database.disable
|
|
driver_options['LocalOutput'] = options.console.local_output
|
|
driver_options['ModulePath'] = options.modules.path
|
|
driver_options['Plugins'] = options.console.plugins
|
|
driver_options['RealReadline'] = options.console.real_readline
|
|
driver_options['Resource'] = options.console.resources
|
|
driver_options['XCommands'] = options.console.commands
|
|
|
|
@driver_options = driver_options
|
|
end
|
|
|
|
@driver_options
|
|
end
|
|
end
|