177 lines
3.7 KiB
Ruby
177 lines
3.7 KiB
Ruby
module Msf
|
|
module Ui
|
|
module Console
|
|
module CommandDispatcher
|
|
|
|
###
|
|
#
|
|
# Recon module command dispatcher.
|
|
#
|
|
###
|
|
class Auxiliary
|
|
|
|
include Msf::Ui::Console::ModuleCommandDispatcher
|
|
|
|
|
|
@@auxiliary_opts = Rex::Parser::Arguments.new(
|
|
"-h" => [ false, "Help banner." ],
|
|
"-j" => [ false, "Run in the context of a job." ],
|
|
"-o" => [ true, "A comma separated list of options in VAR=VAL format." ],
|
|
"-a" => [ true, "The action to use. If none is specified, ACTION is used." ],
|
|
"-q" => [ false, "Run the module in quiet mode with no output" ]
|
|
)
|
|
|
|
#
|
|
# Returns the hash of commands specific to auxiliary modules.
|
|
#
|
|
def commands
|
|
{
|
|
"run" => "Launches the auxiliary module",
|
|
"rerun" => "Reloads and launches the auxiliary module",
|
|
"exploit" => "This is an alias for the run command",
|
|
"rexploit" => "This is an alias for the rerun command",
|
|
"reload" => "Reloads the auxiliary module"
|
|
}.merge( (mod ? mod.auxiliary_commands : {}) )
|
|
end
|
|
|
|
#
|
|
# Allow modules to define their own commands
|
|
#
|
|
def method_missing(meth, *args)
|
|
if (mod and mod.respond_to?(meth.to_s))
|
|
|
|
# Initialize user interaction
|
|
mod.init_ui(driver.input, driver.output)
|
|
|
|
return mod.send(meth.to_s, *args)
|
|
end
|
|
return
|
|
end
|
|
|
|
#
|
|
#
|
|
# Returns the command dispatcher name.
|
|
#
|
|
def name
|
|
"Auxiliary"
|
|
end
|
|
|
|
#
|
|
# This is an alias for 'rerun'
|
|
#
|
|
def cmd_rexploit(*args)
|
|
cmd_rerun(*args)
|
|
end
|
|
|
|
#
|
|
# Reloads an auxiliary module
|
|
#
|
|
def cmd_reload(*args)
|
|
begin
|
|
omod = self.mod
|
|
self.mod = framework.modules.reload_module(mod)
|
|
if(not self.mod)
|
|
print_status("Failed to reload module: #{framework.modules.failed[omod.file_path]}")
|
|
self.mod = omod
|
|
end
|
|
rescue
|
|
log_error("Failed to reload: #{$!}")
|
|
end
|
|
end
|
|
|
|
#
|
|
# Reloads an auxiliary module and executes it
|
|
#
|
|
def cmd_rerun(*args)
|
|
begin
|
|
omod = self.mod
|
|
self.mod = framework.modules.reload_module(mod)
|
|
if(not self.mod)
|
|
print_status("Failed to reload module: #{framework.modules.failed[omod.file_path]}")
|
|
self.mod = omod
|
|
return
|
|
end
|
|
cmd_run(*args)
|
|
rescue
|
|
log_error("Failed to rerun: #{$!}")
|
|
end
|
|
end
|
|
|
|
#
|
|
# This is an alias for 'run'
|
|
#
|
|
def cmd_exploit(*args)
|
|
cmd_run(*args)
|
|
end
|
|
|
|
#
|
|
# Executes an auxiliary module
|
|
#
|
|
def cmd_run(*args)
|
|
defanged?
|
|
|
|
opt_str = nil
|
|
action = mod.datastore['ACTION']
|
|
jobify = false
|
|
quiet = false
|
|
|
|
@@auxiliary_opts.parse(args) { |opt, idx, val|
|
|
case opt
|
|
when '-j'
|
|
jobify = true
|
|
when '-o'
|
|
opt_str = val
|
|
when '-a'
|
|
action = val
|
|
when '-q'
|
|
quiet = true
|
|
when '-h'
|
|
print(
|
|
"Usage: run [options]\n\n" +
|
|
"Launches an auxiliary module.\n" +
|
|
@@auxiliary_opts.usage)
|
|
return false
|
|
end
|
|
}
|
|
|
|
# Always run passive modules in the background
|
|
if (mod.passive or mod.passive_action?(action))
|
|
jobify = true
|
|
end
|
|
|
|
begin
|
|
mod.run_simple(
|
|
'Action' => action,
|
|
'OptionStr' => opt_str,
|
|
'LocalInput' => driver.input,
|
|
'LocalOutput' => driver.output,
|
|
'RunAsJob' => jobify,
|
|
'Quiet' => quiet
|
|
)
|
|
rescue ::Interrupt
|
|
print_error("Auxiliary interrupted by the console user")
|
|
rescue ::Exception => e
|
|
print_error("Auxiliary failed: #{e.class} #{e}")
|
|
if(e.class.to_s != 'Msf::OptionValidateError')
|
|
print_error("Call stack:")
|
|
e.backtrace.each do |line|
|
|
break if line =~ /lib.msf.base.simple/
|
|
print_error(" #{line}")
|
|
end
|
|
end
|
|
|
|
return false
|
|
end
|
|
|
|
if (jobify)
|
|
print_status("Auxiliary module running as background job")
|
|
else
|
|
print_status("Auxiliary module execution completed")
|
|
end
|
|
end
|
|
|
|
end
|
|
|
|
end end end end
|
|
|