Refactor msfcli
sinn3r
parent
56367ef69c
commit
a0decf502f
|
|
@ -263,7 +263,7 @@ class Msf::Modules::Loader::Base
|
|||
|
||||
each_module_reference_name(path, options) do |parent_path, type, module_reference_name|
|
||||
# In msfcli mode, if a module is already loaded, avoid loading it again
|
||||
next if loaded_items.include?(module_reference_name) and modules
|
||||
next if loaded_items.include?(module_reference_name) and options[:whitelist]
|
||||
|
||||
# Keep track of loaded modules in msfcli mode
|
||||
loaded_items << module_reference_name if options[:whitelist]
|
||||
|
|
|
|||
835
msfcli
835
msfcli
|
|
@ -1,14 +1,10 @@
|
|||
#!/usr/bin/env ruby
|
||||
# -*- coding: binary -*-
|
||||
#
|
||||
# $Id$
|
||||
#
|
||||
# This user interface allows users to interact with the framework through a
|
||||
# command line interface (CLI) rather than having to use a prompting console
|
||||
# or web-based interface.
|
||||
#
|
||||
# $Revision$
|
||||
#
|
||||
|
||||
msfbase = __FILE__
|
||||
while File.symlink?(msfbase)
|
||||
|
|
@ -18,337 +14,532 @@ end
|
|||
$:.unshift(File.expand_path(File.join(File.dirname(msfbase), 'lib')))
|
||||
require 'rex'
|
||||
|
||||
Indent = ' '
|
||||
|
||||
# Payload naming style is kind of inconsistent, so instead of
|
||||
# finding the exact path name, we provide the most educated guess based
|
||||
# on platform/stage type/session type/payload name suffix/etc.
|
||||
def guess_payload_name(p)
|
||||
matches = []
|
||||
payload = p.split('/')
|
||||
platform = payload[0]
|
||||
suffix = payload[-1]
|
||||
stage_types = ['singles', 'stagers', 'stages']
|
||||
session_types = ['meterpreter', 'shell']
|
||||
arch = ''
|
||||
class Msfcli
|
||||
def initialize(args)
|
||||
@args = {}
|
||||
@indent = ' '
|
||||
@framework = nil
|
||||
|
||||
# Rule out some possibilities
|
||||
if p =~ /meterpreter/
|
||||
session_types.delete('shell')
|
||||
stage_types.delete('singles')
|
||||
end
|
||||
if p =~ /shell\/.+$/
|
||||
session_types.delete('meterpreter')
|
||||
stage_types.delete('singles')
|
||||
@args[:module_name] = args.shift # First argument should be the module name
|
||||
@args[:mode] = args.pop || 'h' # Last argument should be the mode
|
||||
@args[:params] = args # Whatever is in the middle should be the params
|
||||
end
|
||||
|
||||
if p =~ /x64/
|
||||
arch = 'x64'
|
||||
elsif p =~ /x86/
|
||||
arch = 'x86'
|
||||
#
|
||||
# Returns a usage Rex table
|
||||
#
|
||||
def usage (str = nil, extra = nil)
|
||||
tbl = Rex::Ui::Text::Table.new(
|
||||
'Header' => "Usage: #{$0} <exploit_name> <option=value> [mode]",
|
||||
'Indent' => 4,
|
||||
'Columns' => ['Mode', 'Description']
|
||||
)
|
||||
|
||||
tbl << ['(H)elp', "You're looking at it baby!"]
|
||||
tbl << ['(S)ummary', 'Show information about this module']
|
||||
tbl << ['(O)ptions', 'Show available options for this module']
|
||||
tbl << ['(A)dvanced', 'Show available advanced options for this module']
|
||||
tbl << ['(I)DS Evasion', 'Show available ids evasion options for this module']
|
||||
tbl << ['(P)ayloads', 'Show available payloads for this module']
|
||||
tbl << ['(T)argets', 'Show available targets for this exploit module']
|
||||
tbl << ['(AC)tions', 'Show available actions for this auxiliary module']
|
||||
tbl << ['(C)heck', 'Run the check routine of the selected module']
|
||||
tbl << ['(E)xecute', 'Execute the selected module']
|
||||
|
||||
tbl.to_s
|
||||
|
||||
$stdout.puts "Error: #{str}\n\n" if str
|
||||
$stdout.puts tbl.to_s + "\n"
|
||||
$stdout.puts extra + "\n" if extra
|
||||
end
|
||||
|
||||
# Determine if the payload is staged. If it is, then
|
||||
# we need to load that staged module too.
|
||||
if session_types.include?('shell') and stage_types.include?('stages')
|
||||
if arch == 'x64'
|
||||
matches << /stages\/#{platform}\/x64\/shell/
|
||||
elsif arch == 'x86'
|
||||
matches << /stages\/#{platform}\/x86\/shell/
|
||||
else
|
||||
matches << /stages\/#{platform}\/shell/
|
||||
|
||||
#
|
||||
# Loads up everything in framework, and then returns the module list
|
||||
#
|
||||
def dump_module_list
|
||||
# This is what happens if the user doesn't specify a module name:
|
||||
# msfcli will end up loading EVERYTHING to memory to show you a help
|
||||
# menu plus a list of modules available. Really expensive if you ask me.
|
||||
$stderr.puts "[*] Please wait while we load the module tree..."
|
||||
framework = Msf::Simple::Framework.create
|
||||
ext = ''
|
||||
|
||||
tbl = Rex::Ui::Text::Table.new(
|
||||
'Header' => 'Exploits',
|
||||
'Indent' => 4,
|
||||
'Columns' => [ 'Name', 'Description' ])
|
||||
|
||||
framework.exploits.each_module { |name, mod|
|
||||
tbl << [ 'exploit/' + name, mod.new.name ]
|
||||
}
|
||||
ext << tbl.to_s + "\n"
|
||||
|
||||
tbl = Rex::Ui::Text::Table.new(
|
||||
'Header' => 'Auxiliary',
|
||||
'Indent' => 4,
|
||||
'Columns' => [ 'Name', 'Description' ])
|
||||
|
||||
framework.auxiliary.each_module { |name, mod|
|
||||
tbl << [ 'auxiliary/' + name, mod.new.name ]
|
||||
}
|
||||
|
||||
ext << tbl.to_s + "\n"
|
||||
ext
|
||||
end
|
||||
|
||||
|
||||
#
|
||||
# Payload naming style is kind of inconsistent, so instead of
|
||||
# finding the exact path name, we provide the most educated guess (whitelist)
|
||||
# based on platform/stage type/session type/payload name suffix/etc.
|
||||
#
|
||||
def guess_payload_name(p)
|
||||
matches = []
|
||||
payload = p.split('/')
|
||||
platform = payload[0]
|
||||
suffix = payload[-1]
|
||||
stage_types = ['singles', 'stagers', 'stages']
|
||||
session_types = ['meterpreter', 'shell']
|
||||
arch = ''
|
||||
|
||||
# Rule out some possibilities
|
||||
if p =~ /meterpreter/i
|
||||
session_types.delete('shell')
|
||||
stage_types.delete('singles')
|
||||
end
|
||||
elsif session_types.include?('meterpreter') and stage_types.include?('stages')
|
||||
if arch == 'x64'
|
||||
matches << /stages\/#{platform}\/x64\/meterpreter/
|
||||
elsif arch == 'x86'
|
||||
matches << /stages\/#{platform}\/x86\/meterpreter/
|
||||
else
|
||||
matches << /stages\/#{platform}\/meterpreter/
|
||||
if p =~ /shell\/.+$/i
|
||||
session_types.delete('meterpreter')
|
||||
stage_types.delete('singles')
|
||||
end
|
||||
end
|
||||
|
||||
# Guess the second possible match
|
||||
stage_types *= "|"
|
||||
session_types *= "|"
|
||||
|
||||
if arch == 'x64'
|
||||
matches << /payloads\/(#{stage_types})\/#{platform}\/x64\/.*(#{suffix})\.rb$/
|
||||
elsif arch == 'x86'
|
||||
matches << /payloads\/(#{stage_types})\/#{platform}\/x86\/.*(#{suffix})\.rb$/
|
||||
else
|
||||
matches << /payloads\/(#{stage_types})\/#{platform}\/.*(#{suffix})\.rb$/
|
||||
end
|
||||
|
||||
matches
|
||||
end
|
||||
|
||||
def guess_encoder_name(e)
|
||||
[/#{e}/]
|
||||
end
|
||||
|
||||
def guess_nop_name(n)
|
||||
[/#{n}/]
|
||||
end
|
||||
|
||||
def usage (str = nil, extra = nil)
|
||||
tbl = Rex::Ui::Text::Table.new(
|
||||
'Header' => "Usage: #{$0} <exploit_name> <option=value> [mode]",
|
||||
'Indent' => 4,
|
||||
'Columns' => ['Mode', 'Description']
|
||||
)
|
||||
|
||||
tbl << ['(H)elp', "You're looking at it baby!"]
|
||||
tbl << ['(S)ummary', 'Show information about this module']
|
||||
tbl << ['(O)ptions', 'Show available options for this module']
|
||||
tbl << ['(A)dvanced', 'Show available advanced options for this module']
|
||||
tbl << ['(I)DS Evasion', 'Show available ids evasion options for this module']
|
||||
tbl << ['(P)ayloads', 'Show available payloads for this module']
|
||||
tbl << ['(T)argets', 'Show available targets for this exploit module']
|
||||
tbl << ['(AC)tions', 'Show available actions for this auxiliary module']
|
||||
tbl << ['(C)heck', 'Run the check routine of the selected module']
|
||||
tbl << ['(E)xecute', 'Execute the selected module']
|
||||
|
||||
$stdout.puts "Error: #{str}\n\n" if str
|
||||
$stdout.puts tbl.to_s + "\n"
|
||||
$stdout.puts extra + "\n" if extra
|
||||
|
||||
exit
|
||||
end
|
||||
|
||||
# Handle the help option before loading modules
|
||||
exploit_name = ARGV.shift
|
||||
exploit = nil
|
||||
module_class = "exploit"
|
||||
|
||||
if(exploit_name == "-h")
|
||||
usage()
|
||||
else
|
||||
$:.unshift(ENV['MSF_LOCAL_LIB']) if ENV['MSF_LOCAL_LIB']
|
||||
require 'fastlib'
|
||||
require 'msfenv'
|
||||
require 'msf/ui'
|
||||
require 'msf/base'
|
||||
end
|
||||
|
||||
if (not exploit_name)
|
||||
# This is what happens if the user doesn't specify a module name:
|
||||
# msfcli will end up loading EVERYTHING to memory to show you a help
|
||||
# menu plus a list of modules available. Really expensive if you ask me.
|
||||
$stderr.puts "[*] Please wait while we load the module tree..."
|
||||
$framework = Msf::Simple::Framework.create
|
||||
ext = ''
|
||||
|
||||
tbl = Rex::Ui::Text::Table.new(
|
||||
'Header' => 'Exploits',
|
||||
'Indent' => 4,
|
||||
'Columns' => [ 'Name', 'Description' ])
|
||||
|
||||
$framework.exploits.each_module { |name, mod|
|
||||
tbl << [ 'exploit/' + name, mod.new.name ]
|
||||
}
|
||||
ext << tbl.to_s + "\n"
|
||||
|
||||
tbl = Rex::Ui::Text::Table.new(
|
||||
'Header' => 'Auxiliary',
|
||||
'Indent' => 4,
|
||||
'Columns' => [ 'Name', 'Description' ])
|
||||
|
||||
$framework.auxiliary.each_module { |name, mod|
|
||||
tbl << [ 'auxiliary/' + name, mod.new.name ]
|
||||
}
|
||||
|
||||
ext << tbl.to_s + "\n"
|
||||
|
||||
usage(nil, ext)
|
||||
end
|
||||
|
||||
|
||||
# Process special var/val pairs...
|
||||
Msf::Ui::Common.process_cli_arguments($framework, ARGV)
|
||||
|
||||
# Add modules like exploit/aux, encoders, nops we want to load to the list
|
||||
whitelist = []
|
||||
whitelist << /#{exploit_name}/ # Add exploit
|
||||
whitelist << /x86\/single_byte/ # Add default NOP module
|
||||
whitelist << /x86\/shikata_ga_nai/ # Add default encoder
|
||||
whitelist << /generic\/none/ # Add another default encoder
|
||||
ARGV.each { |args|
|
||||
var, val = args.split('=', 2)
|
||||
next if var.nil? or val.nil?
|
||||
whitelist.concat(guess_payload_name(val)) if var =~ /payload/
|
||||
whitelist.concat(guess_encoder_name(val)) if var =~ /encoder/
|
||||
whitelist.concat(guess_nops_name(val)) if var =~ /nops/
|
||||
}
|
||||
|
||||
$stderr.puts "[*] Initializing modules..."
|
||||
$framework = Msf::Simple::Framework.create({'DeferModuleLoads'=>true})
|
||||
$framework.init_module_paths({:whitelist=>whitelist})
|
||||
if ($framework.modules.module_load_error_by_path.length > 0)
|
||||
print("Warning: The following modules could not be loaded!\n\n")
|
||||
|
||||
$framework.modules.module_load_error_by_path.each do |path, error|
|
||||
print("\t#{path}: #{error}\n\n")
|
||||
end
|
||||
end
|
||||
|
||||
# Determine what type of module it is
|
||||
case exploit_name
|
||||
when /exploit\/(.*)/
|
||||
exploit = $framework.exploits.create($1)
|
||||
module_class = 'exploit'
|
||||
|
||||
when /auxiliary\/(.*)/
|
||||
exploit = $framework.auxiliary.create($1)
|
||||
module_class = 'auxiliary'
|
||||
|
||||
else
|
||||
exploit = $framework.exploits.create(exploit_name)
|
||||
if exploit == nil
|
||||
# Try falling back on aux modules
|
||||
exploit = $framework.auxiliary.create(exploit_name)
|
||||
module_class = 'auxiliary'
|
||||
end
|
||||
|
||||
end
|
||||
|
||||
|
||||
if (exploit == nil)
|
||||
usage("Invalid module: #{exploit_name}")
|
||||
end
|
||||
|
||||
exploit.init_ui(
|
||||
Rex::Ui::Text::Input::Stdio.new,
|
||||
Rex::Ui::Text::Output::Stdio.new
|
||||
)
|
||||
|
||||
# Evalulate the command (default to "help")
|
||||
mode = ARGV.pop || 'h'
|
||||
|
||||
# Import options
|
||||
begin
|
||||
exploit.datastore.import_options_from_s(ARGV.join('_|_'), '_|_')
|
||||
rescue Rex::ArgumentParseError => e
|
||||
puts "[!] Error: #{e.message}\n\n"
|
||||
exit
|
||||
end
|
||||
|
||||
# Initialize associated modules
|
||||
payload = nil
|
||||
encoder = nil
|
||||
nop = nil
|
||||
|
||||
if (exploit.datastore['PAYLOAD'])
|
||||
payload = $framework.payloads.create(exploit.datastore['PAYLOAD'])
|
||||
if (payload != nil)
|
||||
payload.datastore.import_options_from_s(ARGV.join('_|_'), '_|_')
|
||||
end
|
||||
end
|
||||
|
||||
if (exploit.datastore['ENCODER'])
|
||||
encoder = $framework.encoders.create(exploit.datastore['ENCODER'])
|
||||
if (encoder != nil)
|
||||
encoder.datastore.import_options_from_s(ARGV.join('_|_'), '_|_')
|
||||
end
|
||||
end
|
||||
|
||||
if (exploit.datastore['NOP'])
|
||||
nop = $framework.nops.create(exploit.datastore['NOP'])
|
||||
if (nop != nil)
|
||||
nop.datastore.import_options_from_s(ARGV.join('_|_'), '_|_')
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
|
||||
case mode.downcase
|
||||
when 'h'
|
||||
usage
|
||||
when "s"
|
||||
$stdout.puts("\n" + Msf::Serializer::ReadableText.dump_module(exploit, Indent))
|
||||
$stdout.puts("\n" + Msf::Serializer::ReadableText.dump_module(payload, Indent)) if payload
|
||||
$stdout.puts("\n" + Msf::Serializer::ReadableText.dump_module(encoder, Indent)) if encoder
|
||||
$stdout.puts("\n" + Msf::Serializer::ReadableText.dump_module(nop, Indent)) if nop
|
||||
when "o"
|
||||
$stdout.puts("\n" + Msf::Serializer::ReadableText.dump_options(exploit, Indent))
|
||||
$stdout.puts("\nPayload:\n\n" + Msf::Serializer::ReadableText.dump_options(payload, Indent)) if payload
|
||||
$stdout.puts("\nEncoder:\n\n" + Msf::Serializer::ReadableText.dump_options(encoder, Indent)) if encoder
|
||||
$stdout.puts("\nNOP\n\n" + Msf::Serializer::ReadableText.dump_options(nop, Indent)) if nop
|
||||
when "a"
|
||||
$stdout.puts("\n" + Msf::Serializer::ReadableText.dump_advanced_options(exploit, Indent))
|
||||
$stdout.puts("\nPayload:\n\n" + Msf::Serializer::ReadableText.dump_advanced_options(payload, Indent)) if payload
|
||||
$stdout.puts("\nEncoder:\n\n" + Msf::Serializer::ReadableText.dump_advanced_options(encoder, Indent)) if encoder
|
||||
$stdout.puts("\nNOP:\n\n" + Msf::Serializer::ReadableText.dump_advanced_options(nop, Indent)) if nop
|
||||
when "i"
|
||||
$stdout.puts("\n" + Msf::Serializer::ReadableText.dump_evasion_options(exploit, Indent))
|
||||
$stdout.puts("\nPayload:\n\n" + Msf::Serializer::ReadableText.dump_evasion_options(payload, Indent)) if payload
|
||||
$stdout.puts("\nEncoder:\n\n" + Msf::Serializer::ReadableText.dump_evasion_options(encoder, Indent)) if encoder
|
||||
$stdout.puts("\nNOP:\n\n" + Msf::Serializer::ReadableText.dump_evasion_options(nop, Indent)) if nop
|
||||
when "p"
|
||||
if (module_class == 'exploit')
|
||||
$stdout.puts("\n" + Msf::Serializer::ReadableText.dump_compatible_payloads(exploit, Indent, "Compatible payloads"))
|
||||
else
|
||||
$stdout.puts("\nError: This type of module does not support payloads")
|
||||
if p =~ /x64/i
|
||||
arch = 'x64'
|
||||
elsif p =~ /x86/i
|
||||
arch = 'x86'
|
||||
end
|
||||
when "t"
|
||||
if (module_class == 'exploit')
|
||||
$stdout.puts("\n" + Msf::Serializer::ReadableText.dump_exploit_targets(exploit, Indent))
|
||||
else
|
||||
$stdout.puts("\nError: This type of module does not support targets")
|
||||
end
|
||||
when "ac"
|
||||
if (module_class == 'auxiliary')
|
||||
$stdout.puts("\n" + Msf::Serializer::ReadableText.dump_auxiliary_actions(exploit, Indent))
|
||||
else
|
||||
$stdout.puts("\nError: This type of module does not support actions")
|
||||
end
|
||||
when "c"
|
||||
if (module_class == 'exploit')
|
||||
begin
|
||||
if (code = exploit.check_simple(
|
||||
'LocalInput' => Rex::Ui::Text::Input::Stdio.new,
|
||||
'LocalOutput' => Rex::Ui::Text::Output::Stdio.new))
|
||||
stat = (code == Msf::Exploit::CheckCode::Vulnerable) ? '[+]' : '[*]'
|
||||
|
||||
$stdout.puts("#{stat} #{code[1]}")
|
||||
else
|
||||
$stderr.puts("Check failed: The state could not be determined.")
|
||||
end
|
||||
rescue
|
||||
$stderr.puts("Check failed: #{$!}")
|
||||
end
|
||||
else
|
||||
$stdout.puts("\nError: This type of module does not support the check feature")
|
||||
end
|
||||
when "e"
|
||||
con = Msf::Ui::Console::Driver.new(
|
||||
Msf::Ui::Console::Driver::DefaultPrompt,
|
||||
Msf::Ui::Console::Driver::DefaultPromptChar,
|
||||
{
|
||||
'Framework' => $framework,
|
||||
# When I use msfcli, chances are I want speed, so ASCII art fanciness
|
||||
# probably isn't much of a big deal for me.
|
||||
'DisableBanner' => true
|
||||
}
|
||||
)
|
||||
con.run_single("use #{module_class}/#{exploit.refname}")
|
||||
|
||||
ARGV.each do |arg|
|
||||
k,v = arg.split("=", 2)
|
||||
con.run_single("set #{k} #{v}")
|
||||
end
|
||||
|
||||
con.run_single("exploit")
|
||||
|
||||
# If we have sessions or jobs, keep running
|
||||
if $framework.sessions.length > 0 or $framework.jobs.length > 0
|
||||
con.run
|
||||
# Determine if the payload is staged. If it is, then
|
||||
# we need to load that staged module too.
|
||||
if session_types.include?('shell') and stage_types.include?('stages')
|
||||
if arch == 'x64'
|
||||
matches << /stages\/#{platform}\/x64\/shell/
|
||||
elsif arch == 'x86'
|
||||
matches << /stages\/#{platform}\/x86\/shell/
|
||||
else
|
||||
con.run_single("quit")
|
||||
matches << /stages\/#{platform}\/shell/
|
||||
end
|
||||
elsif session_types.include?('meterpreter') and stage_types.include?('stages')
|
||||
if arch == 'x64'
|
||||
matches << /stages\/#{platform}\/x64\/meterpreter/
|
||||
elsif arch == 'x86'
|
||||
matches << /stages\/#{platform}\/x86\/meterpreter/
|
||||
else
|
||||
matches << /stages\/#{platform}\/meterpreter/
|
||||
end
|
||||
end
|
||||
|
||||
# Guess the second possible match
|
||||
stage_types *= "|"
|
||||
session_types *= "|"
|
||||
|
||||
if arch == 'x64'
|
||||
matches << /payloads\/(#{stage_types})\/#{platform}\/x64\/.*(#{suffix})\.rb$/
|
||||
elsif arch == 'x86'
|
||||
matches << /payloads\/(#{stage_types})\/#{platform}\/x86\/.*(#{suffix})\.rb$/
|
||||
else
|
||||
matches << /payloads\/(#{stage_types})\/#{platform}\/.*(#{suffix})\.rb$/
|
||||
end
|
||||
|
||||
matches
|
||||
end
|
||||
|
||||
|
||||
#
|
||||
# Returns a whitelist for encoder modules
|
||||
#
|
||||
def guess_encoder_name(e)
|
||||
[/encoders\/#{e}/]
|
||||
end
|
||||
|
||||
|
||||
#
|
||||
# Returns a whitelist for nop modules
|
||||
#
|
||||
def guess_nop_name(n)
|
||||
[/nops\/#{n}/]
|
||||
end
|
||||
|
||||
|
||||
#
|
||||
# Returns a whitelist for post modules
|
||||
#
|
||||
def guess_post_name(p)
|
||||
[/post\/#{p}/]
|
||||
end
|
||||
|
||||
|
||||
#
|
||||
# Returns possible patterns like exploit/aux, encoders, nops we want to
|
||||
# load to the whitelist.
|
||||
#
|
||||
def generate_whitelist
|
||||
whitelist = []
|
||||
whitelist << /#{@args[:module_name]}/ # Add exploit
|
||||
|
||||
# nil = not set, empty = manually set to load nothing
|
||||
encoder_val = nil
|
||||
nops_val = nil
|
||||
post_val = nil
|
||||
payload_param = ''
|
||||
junk_args = []
|
||||
|
||||
@args[:params].each { |args|
|
||||
var, val = args.split('=', 2)
|
||||
|
||||
case var.downcase
|
||||
when 'payload'
|
||||
payload_param = val
|
||||
if val.empty?
|
||||
junk_args << args
|
||||
else
|
||||
whitelist.concat(guess_payload_name(val))
|
||||
end
|
||||
|
||||
when 'encoder'
|
||||
encoder_val = val
|
||||
if val.empty?
|
||||
junk_args << args
|
||||
else
|
||||
whitelist.concat(guess_encoder_name(val))
|
||||
end
|
||||
|
||||
when 'nop'
|
||||
nops_val = val
|
||||
if val.empty?
|
||||
junk_args << args
|
||||
else
|
||||
whitelist.concat(guess_nop_name(val))
|
||||
end
|
||||
|
||||
when 'post'
|
||||
post_val = val
|
||||
if val.empty?
|
||||
junk_args << args
|
||||
else
|
||||
whitelist.concat(guess_post_name(val))
|
||||
end
|
||||
end
|
||||
}
|
||||
|
||||
# Cleanup empty args
|
||||
junk_args.each { |args| @args[:params].delete(args) }
|
||||
|
||||
# Add post modules list if not set
|
||||
if post_val.nil?
|
||||
whitelist << /post\/.+/
|
||||
end
|
||||
|
||||
# Add default encoders if not set
|
||||
# This one is needed no matter what
|
||||
whitelist << /encoders\/generic\/*/
|
||||
if encoder_val.nil?
|
||||
if payload_param =~ /^.+\.x64.+/
|
||||
whitelist << /encoders\/x64\/.+/
|
||||
elsif payload_param =~ /^.+\.x86.+/
|
||||
whitelist << /encoders\/x86\/.+/
|
||||
else
|
||||
whitelist << /encoders\/.+/
|
||||
end
|
||||
end
|
||||
|
||||
# Add default NOP modules if not set
|
||||
if nops_val.nil?
|
||||
whitelist << /nops\/.+/
|
||||
end
|
||||
|
||||
whitelist
|
||||
end
|
||||
|
||||
|
||||
#
|
||||
# Initializes exploit/payload/encoder/nop modules.
|
||||
#
|
||||
def init_modules
|
||||
module_name = @args[:module_name]
|
||||
modules = {
|
||||
:module => nil, # aux or exploit instance
|
||||
:payload => nil, # payload instance
|
||||
:encoder => nil, # encoder instance
|
||||
:nop => nil # nop instance
|
||||
}
|
||||
|
||||
whitelist = generate_whitelist
|
||||
|
||||
# Load up all the possible modules, this is where things get slow again
|
||||
@framework.init_module_paths({:whitelist=>whitelist})
|
||||
if (@framework.modules.module_load_error_by_path.length > 0)
|
||||
print("Warning: The following modules could not be loaded!\n\n")
|
||||
|
||||
@framework.modules.module_load_error_by_path.each do |path, error|
|
||||
print("\t#{path}: #{error}\n\n")
|
||||
end
|
||||
|
||||
else
|
||||
usage("Invalid mode #{mode}")
|
||||
return {}
|
||||
end
|
||||
|
||||
# Determine what type of module it is
|
||||
if module_name =~ /exploit\/(.*)/
|
||||
modules[:module] = @framework.exploits.create($1)
|
||||
elsif module_name =~ /auxiliary\/(.*)/
|
||||
modules[:module] = @framework.auxiliary.create($1)
|
||||
else
|
||||
modules[:module] = @framework.exploits.create(module_name)
|
||||
if modules[:module].nil?
|
||||
# Try falling back on aux modules
|
||||
modules[:module] = @framework.auxiliary.create(module_name)
|
||||
end
|
||||
end
|
||||
|
||||
if modules[:module].nil?
|
||||
# Still nil? Ok then, probably invalid
|
||||
return {}
|
||||
end
|
||||
|
||||
modules[:module].init_ui(
|
||||
Rex::Ui::Text::Input::Stdio.new,
|
||||
Rex::Ui::Text::Output::Stdio.new
|
||||
)
|
||||
|
||||
# Import options
|
||||
begin
|
||||
modules[:module].datastore.import_options_from_s(@args[:params].join('_|_'), '_|_')
|
||||
rescue Rex::ArgumentParseError => e
|
||||
raise e
|
||||
end
|
||||
|
||||
# Create the payload to use
|
||||
if (modules[:module].datastore['PAYLOAD'])
|
||||
modules[:payload] = @framework.payloads.create(modules[:module].datastore['PAYLOAD'])
|
||||
if modules[:payload]
|
||||
modules[:payload].datastore.import_options_from_s(@args[:params].join('_|_'), '_|_')
|
||||
end
|
||||
end
|
||||
|
||||
# Create the encoder to use
|
||||
if modules[:module].datastore['ENCODER']
|
||||
modules[:encoder] = @framework.encoders.create(modules[:module].datastore['ENCODER'])
|
||||
if modules[:encoder]
|
||||
modules[:encoder].datastore.import_options_from_s(@args[:params].join('_|_'), '_|_')
|
||||
end
|
||||
end
|
||||
|
||||
# Create the NOP to use
|
||||
if modules[:module].datastore['NOP']
|
||||
modules[:nop] = @framework.nops.create(modules[:module].datastore['NOP'])
|
||||
if modules[:nop]
|
||||
modules[:nop].datastore.import_options_from_s(@args[:params].join('_|_'), '_|_')
|
||||
end
|
||||
end
|
||||
|
||||
modules
|
||||
end
|
||||
|
||||
|
||||
def show_summary(m)
|
||||
readable = Msf::Serializer::ReadableText
|
||||
$stdout.puts("\n" + readable.dump_module(m[:module], @indent))
|
||||
$stdout.puts("\n" + readable.dump_module(m[:payload], @indent)) if m[:payload]
|
||||
$stdout.puts("\n" + readable.dump_module(m[:encoder], @indent)) if m[:encoder]
|
||||
$stdout.puts("\n" + readable.dump_module(m[:nop], @indent)) if m[:nop]
|
||||
end
|
||||
|
||||
|
||||
def show_options(m)
|
||||
readable = Msf::Serializer::ReadableText
|
||||
$stdout.puts("\n" + readable.dump_options(m[:module], @indent))
|
||||
$stdout.puts("\nPayload:\n\n" + readable.dump_options(m[:payload], @indent)) if m[:payload]
|
||||
$stdout.puts("\nEncoder:\n\n" + readable.dump_options(m[:encoder], @indent)) if m[:encoder]
|
||||
$stdout.puts("\nNOP\n\n" + readable.dump_options(m[:nop], @indent)) if m[:nop]
|
||||
end
|
||||
|
||||
|
||||
def show_advanced(m)
|
||||
readable = Msf::Serializer::ReadableText
|
||||
$stdout.puts("\n" + readable.dump_advanced_options(m[:module], @indent))
|
||||
$stdout.puts("\nPayload:\n\n" + readable.dump_advanced_options(m[:payload], @indent)) if m[:payload]
|
||||
$stdout.puts("\nEncoder:\n\n" + readable.dump_advanced_options(m[:encoder], @indent)) if m[:encoder]
|
||||
$stdout.puts("\nNOP:\n\n" + readable.dump_advanced_options(m[:nop], @indent)) if m[:nop]
|
||||
end
|
||||
|
||||
|
||||
def show_ids_evasion(m)
|
||||
readable = Msf::Serializer::ReadableText
|
||||
$stdout.puts("\n" + readable.dump_evasion_options(m[:module], Indent))
|
||||
$stdout.puts("\nPayload:\n\n" + readable.dump_evasion_options(m[:payload], @indent)) if m[:payload]
|
||||
$stdout.puts("\nEncoder:\n\n" + readable.dump_evasion_options(m[:encoder], @indent)) if m[:encoder]
|
||||
$stdout.puts("\nNOP:\n\n" + readable.dump_evasion_options(m[:nop], @indent)) if m[:nop]
|
||||
end
|
||||
|
||||
|
||||
def show_payloads(m)
|
||||
readable = Msf::Serializer::ReadableText
|
||||
txt = "Compatible payloads"
|
||||
$stdout.puts("\n" + readable.dump_compatible_payloads(m[:module], @indent, txt))
|
||||
end
|
||||
|
||||
|
||||
def show_targets(m)
|
||||
readable = Msf::Serializer::ReadableText
|
||||
$stdout.puts("\n" + readable.dump_exploit_targets(m[:module], @indent))
|
||||
end
|
||||
|
||||
|
||||
def show_actions(m)
|
||||
readable = Msf::Serializer::ReadableText
|
||||
$stdout.puts("\n" + readable.dump_auxiliary_actions(m[:module], @indent))
|
||||
end
|
||||
|
||||
|
||||
def show_check(m)
|
||||
begin
|
||||
if (code = m[:module].check_simple(
|
||||
'LocalInput' => Rex::Ui::Text::Input::Stdio.new,
|
||||
'LocalOutput' => Rex::Ui::Text::Output::Stdio.new))
|
||||
stat = (code == Msf::Exploit::CheckCode::Vulnerable) ? '[+]' : '[*]'
|
||||
|
||||
$stdout.puts("#{stat} #{code[1]}")
|
||||
else
|
||||
$stderr.puts("Check failed: The state could not be determined.")
|
||||
end
|
||||
rescue
|
||||
$stderr.puts("Check failed: #{$!}")
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
def execute_module(m)
|
||||
con = Msf::Ui::Console::Driver.new(
|
||||
Msf::Ui::Console::Driver::DefaultPrompt,
|
||||
Msf::Ui::Console::Driver::DefaultPromptChar,
|
||||
{
|
||||
'Framework' => @framework,
|
||||
# When I use msfcli, chances are I want speed, so ASCII art fanciness
|
||||
# probably isn't much of a big deal for me.
|
||||
'DisableBanner' => true
|
||||
})
|
||||
|
||||
module_class = (m[:module].fullname =~ /^auxiliary/ ? 'auxiliary' : 'exploit')
|
||||
|
||||
con.run_single("use #{module_class}/#{m[:module].refname}")
|
||||
|
||||
# Assign console parameters
|
||||
@args[:params].each do |arg|
|
||||
k,v = arg.split("=", 2)
|
||||
con.run_single("set #{k} #{v}")
|
||||
end
|
||||
|
||||
# Run the exploit
|
||||
con.run_single("exploit")
|
||||
|
||||
# If we have sessions or jobs, keep running
|
||||
if @framework.sessions.length > 0 or @framework.jobs.length > 0
|
||||
con.run
|
||||
else
|
||||
con.run_single("quit")
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
#
|
||||
# Selects a mode chosen by the user and run it
|
||||
#
|
||||
def engage_mode(modules)
|
||||
ref_name = modules[:module].refname
|
||||
|
||||
case @args[:mode].downcase
|
||||
when 'h'
|
||||
usage
|
||||
when "s"
|
||||
show_summary(modules)
|
||||
when "o"
|
||||
show_options(modules)
|
||||
when "a"
|
||||
show_advanced(modules)
|
||||
when "i"
|
||||
show_ids_evasion(modules)
|
||||
when "p"
|
||||
if ref_name =~ /^auxiliary/
|
||||
$stdout.puts("\nError: This type of module does not support payloads")
|
||||
else
|
||||
show_payloads(modules)
|
||||
end
|
||||
when "t"
|
||||
if ref_name =~ /^auxiliary/
|
||||
$stdout.puts("\nError: This type of module does not support targets")
|
||||
else
|
||||
show_targets(modules)
|
||||
end
|
||||
when "ac"
|
||||
if ref_name =~ /auxiliary/
|
||||
show_actions(modules)
|
||||
else
|
||||
$stdout.puts("\nError: This type of module does not support actions")
|
||||
end
|
||||
when "c"
|
||||
show_check(modules)
|
||||
when "e"
|
||||
execute_module(modules)
|
||||
else
|
||||
usage("Invalid mode #{@args[:mode]}")
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
def run!
|
||||
if @args[:module_name] == "-h"
|
||||
usage()
|
||||
exit
|
||||
end
|
||||
|
||||
$:.unshift(ENV['MSF_LOCAL_LIB']) if ENV['MSF_LOCAL_LIB']
|
||||
require 'fastlib'
|
||||
require 'msfenv'
|
||||
require 'msf/ui'
|
||||
require 'msf/base'
|
||||
|
||||
if @args[:module_name].nil?
|
||||
ext = dump_module_list
|
||||
usage(nil, ext)
|
||||
exit
|
||||
end
|
||||
|
||||
@framework = Msf::Simple::Framework.create({'DeferModuleLoads'=>true})
|
||||
$stderr.puts "[*] Initializing modules..."
|
||||
|
||||
begin
|
||||
modules = init_modules
|
||||
rescue Rex::ArgumentParseError => e
|
||||
puts "[!] Error: #{e.message}\n\n"
|
||||
exit
|
||||
end
|
||||
|
||||
if modules[:module].nil?
|
||||
usage("Invalid module: #{@args[:module_name]}")
|
||||
exit
|
||||
end
|
||||
|
||||
# Process special var/val pairs...
|
||||
Msf::Ui::Common.process_cli_arguments(@framework, @args[:params])
|
||||
|
||||
engage_mode(modules)
|
||||
$stdout.puts
|
||||
end
|
||||
end
|
||||
|
||||
$stdout.puts
|
||||
|
||||
if __FILE__ == $PROGRAM_NAME
|
||||
cli = Msfcli.new(ARGV)
|
||||
cli.run!
|
||||
end
|
||||
|
|
|
|||
Loading…
Reference in New Issue