Jacob Robles
118 lines
3.7 KiB
Ruby
118 lines
3.7 KiB
Ruby
# -*- coding: binary -*-
|
|
require 'msf/core/modules/external'
|
|
require 'msf/core/modules/external/bridge'
|
|
|
|
class Msf::Modules::External::Shim
|
|
def self.generate(module_path)
|
|
mod = Msf::Modules::External::Bridge.open(module_path)
|
|
return '' unless mod.meta
|
|
case mod.meta['type']
|
|
when 'remote_exploit_cmd_stager'
|
|
remote_exploit_cmd_stager(mod)
|
|
when 'capture_server'
|
|
capture_server(mod)
|
|
when 'dos'
|
|
dos(mod)
|
|
when 'single_scanner'
|
|
single_scanner(mod)
|
|
when 'multi_scanner'
|
|
multi_scanner(mod)
|
|
else
|
|
# TODO have a nice load error show up in the logs
|
|
''
|
|
end
|
|
end
|
|
|
|
def self.render_template(name, meta = {})
|
|
template = File.join(File.dirname(__FILE__), 'templates', name)
|
|
ERB.new(File.read(template)).result(binding)
|
|
end
|
|
|
|
def self.common_metadata(meta = {})
|
|
render_template('common_metadata.erb', meta)
|
|
end
|
|
|
|
def self.mod_meta_common(mod, meta = {}, drop_rhost: false)
|
|
meta[:path] = mod.path.dump
|
|
meta[:name] = mod.meta['name'].dump
|
|
meta[:description] = mod.meta['description'].dump
|
|
meta[:authors] = mod.meta['authors'].map(&:dump).join(",\n ")
|
|
meta[:license] = mod.meta['license'].nil? ? 'MSF_LICENSE' : mod.meta['license']
|
|
|
|
options = if drop_rhost
|
|
mod.meta['options'].reject {|n, o| n == 'rhost'}
|
|
else
|
|
mod.meta['options']
|
|
end
|
|
|
|
meta[:options] = options.map do |n, o|
|
|
if o['values']
|
|
"Opt#{o['type'].camelize}.new(#{n.dump},
|
|
[#{o['required']}, #{o['description'].dump}, #{o['default'].inspect}, #{o['values'].inspect}])"
|
|
else
|
|
"Opt#{o['type'].camelize}.new(#{n.dump},
|
|
[#{o['required']}, #{o['description'].dump}, #{o['default'].inspect}])"
|
|
end
|
|
end.join(",\n ")
|
|
meta
|
|
end
|
|
|
|
def self.mod_meta_exploit(mod, meta = {})
|
|
meta[:date] = mod.meta['date'].dump
|
|
meta[:wfsdelay] = mod.meta['wfsdelay'] || 5
|
|
meta[:privileged] = mod.meta['privileged'].inspect
|
|
meta[:platform] = mod.meta['targets'].map do |t|
|
|
t['platform'].dump
|
|
end.uniq.join(",\n ")
|
|
meta[:references] = mod.meta['references'].map do |r|
|
|
"[#{r['type'].upcase.dump}, #{r['ref'].dump}]"
|
|
end.join(",\n ")
|
|
meta[:targets] = mod.meta['targets'].map do |t|
|
|
"[#{t['platform'].dump} + ' ' + #{t['arch'].dump}, {'Arch' => ARCH_#{t['arch'].upcase}, 'Platform' => #{t['platform'].dump} }]"
|
|
end.join(",\n ")
|
|
meta
|
|
end
|
|
|
|
def self.remote_exploit_cmd_stager(mod)
|
|
meta = mod_meta_common(mod)
|
|
meta = mod_meta_exploit(mod, meta)
|
|
meta[:command_stager_flavor] = mod.meta['payload']['command_stager_flavor'].dump
|
|
render_template('remote_exploit_cmd_stager.erb', meta)
|
|
end
|
|
|
|
def self.capture_server(mod)
|
|
meta = mod_meta_common(mod)
|
|
render_template('capture_server.erb', meta)
|
|
end
|
|
|
|
def self.single_scanner(mod)
|
|
meta = mod_meta_common(mod, drop_rhost: true)
|
|
meta[:date] = mod.meta['date'].dump
|
|
meta[:references] = mod.meta['references'].map do |r|
|
|
"[#{r['type'].upcase.dump}, #{r['ref'].dump}]"
|
|
end.join(",\n ")
|
|
|
|
render_template('single_scanner.erb', meta)
|
|
end
|
|
|
|
def self.multi_scanner(mod)
|
|
meta = mod_meta_common(mod)
|
|
meta[:date] = mod.meta['date'].dump
|
|
meta[:references] = mod.meta['references'].map do |r|
|
|
"[#{r['type'].upcase.dump}, #{r['ref'].dump}]"
|
|
end.join(",\n ")
|
|
|
|
render_template('multi_scanner.erb', meta)
|
|
end
|
|
|
|
def self.dos(mod)
|
|
meta = mod_meta_common(mod)
|
|
meta[:date] = mod.meta['date'].dump
|
|
meta[:references] = mod.meta['references'].map do |r|
|
|
"[#{r['type'].upcase.dump}, #{r['ref'].dump}]"
|
|
end.join(",\n ")
|
|
|
|
render_template('dos.erb', meta)
|
|
end
|
|
end
|