dwelch-r7
143 lines
4.5 KiB
Ruby
143 lines
4.5 KiB
Ruby
# -*- coding: binary -*-
|
|
|
|
#
|
|
# Rex
|
|
#
|
|
|
|
|
|
|
|
module Msf
|
|
module Ui
|
|
module Console
|
|
module CommandDispatcher
|
|
|
|
#
|
|
# {CommandDispatcher} for commands related to background jobs in Metasploit Framework.
|
|
#
|
|
class Resource
|
|
|
|
include Msf::Ui::Console::CommandDispatcher
|
|
|
|
|
|
def commands
|
|
{
|
|
"resource" => "Run the commands stored in a file",
|
|
"makerc" => "Save commands entered since start to a file",
|
|
}
|
|
end
|
|
|
|
#
|
|
# Returns the name of the command dispatcher.
|
|
#
|
|
def name
|
|
"Resource Script"
|
|
end
|
|
|
|
def cmd_resource_help
|
|
print_line "Usage: resource path1 [path2 ...]"
|
|
print_line
|
|
print_line "Run the commands stored in the supplied files (- for stdin)."
|
|
print_line "Resource files may also contain ERB or Ruby code between <ruby></ruby> tags."
|
|
print_line
|
|
print_line "See also: makerc"
|
|
print_line
|
|
end
|
|
|
|
def cmd_resource(*args)
|
|
if args.empty?
|
|
cmd_resource_help
|
|
return false
|
|
end
|
|
|
|
args.each do |res|
|
|
res_expand = ::File.expand_path(res)
|
|
good_res = nil
|
|
if res == '-'
|
|
good_res = res
|
|
elsif ::File.file?(res_expand) && File.readable?(res_expand)
|
|
good_res = res_expand
|
|
else
|
|
# let's check to see if it's in the scripts/resource dir (like when tab completed)
|
|
[
|
|
::Msf::Config.script_directory + ::File::SEPARATOR + 'resource',
|
|
::Msf::Config.user_script_directory + ::File::SEPARATOR + 'resource'
|
|
].each do |dir|
|
|
res_path = dir + ::File::SEPARATOR + res
|
|
if ::File.file?(res_path) && File.readable?(res_path)
|
|
good_res = res_path
|
|
break
|
|
end
|
|
end
|
|
end
|
|
if good_res
|
|
driver.load_resource(good_res)
|
|
else
|
|
print_error("#{res} is not a valid resource file")
|
|
next
|
|
end
|
|
end
|
|
end
|
|
|
|
#
|
|
# Tab completion for the resource command
|
|
#
|
|
# @param str [String] the string currently being typed before tab was hit
|
|
# @param words [Array<String>] the previously completed words on the command line. words is always
|
|
# at least 1 when tab completion has reached this stage since the command itself has been completed
|
|
|
|
def cmd_resource_tabs(str, words)
|
|
tabs = []
|
|
#return tabs if words.length > 1
|
|
if !str.nil? && (str.start_with?('~') || str =~ /^#{Regexp.escape(File::SEPARATOR)}/)
|
|
# then you are probably specifying a full path so let's just use normal file completion
|
|
return tab_complete_filenames(str, words)
|
|
elsif (not words[1] or not words[1].match(/^\//))
|
|
# then let's start tab completion in the scripts/resource directories
|
|
begin
|
|
[
|
|
::Msf::Config.script_directory + File::SEPARATOR + "resource",
|
|
::Msf::Config.user_script_directory + File::SEPARATOR + "resource",
|
|
'.'
|
|
].each do |dir|
|
|
next unless ::File.exist?(dir)
|
|
tabs += ::Dir.new(dir).find_all { |e|
|
|
path = dir + File::SEPARATOR + e
|
|
::File.file?(path) && File.readable?(path)
|
|
}
|
|
end
|
|
rescue
|
|
end
|
|
else
|
|
tabs += tab_complete_filenames(str,words)
|
|
end
|
|
return tabs
|
|
end
|
|
|
|
def cmd_makerc_help
|
|
print_line "Usage: makerc <output rc file>"
|
|
print_line
|
|
print_line "Save the commands executed since startup to the specified file."
|
|
print_line
|
|
end
|
|
|
|
def cmd_makerc_tabs(str, words)
|
|
tab_complete_filenames(str, words)
|
|
end
|
|
|
|
#
|
|
# Saves commands executed since the ui started to the specified msfrc file
|
|
#
|
|
def cmd_makerc(*args)
|
|
if args.empty? || args.include?('-h')
|
|
cmd_makerc_help
|
|
return false
|
|
end
|
|
driver.save_recent_history(args[0])
|
|
end
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|
|
end
|