65 lines
1.5 KiB
Ruby
65 lines
1.5 KiB
Ruby
|
module Msf
|
||
|
|
module RPC
|
||
|
|
class Plugin < Base
|
||
|
|
|
||
|
|
def load(token,path,xopts = {})
|
||
|
|
|
||
|
|
authenticate(token)
|
||
|
|
opts = {}
|
||
|
|
|
||
|
|
xopts.each do |k,v|
|
||
|
|
if k.class == String
|
||
|
|
opts[k.to_sym] = v
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
if (path !~ /#{File::SEPARATOR}/)
|
||
|
|
plugin_file_name = path
|
||
|
|
|
||
|
|
# If the plugin isn't in the user direcotry (~/.msf3/plugins/), use the base
|
||
|
|
path = Msf::Config.user_plugin_directory + File::SEPARATOR + plugin_file_name
|
||
|
|
if not File.exists?( path + ".rb" )
|
||
|
|
# If the following "path" doesn't exist it will be caught when we attempt to load
|
||
|
|
path = Msf::Config.plugin_directory + File::SEPARATOR + plugin_file_name
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
begin
|
||
|
|
if (inst = @framework.plugins.load(path, opts))
|
||
|
|
return { "result" => "success" }
|
||
|
|
end
|
||
|
|
rescue ::Exception => e
|
||
|
|
elog("Error loading plugin #{path}: #{e}\n\n#{e.backtrace.join("\n")}", src = 'core', level = 0, from = caller)
|
||
|
|
return { "result" => "failure" }
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
def unload(token,name)
|
||
|
|
authenticate(token)
|
||
|
|
@framework.plugins.each { |plugin|
|
||
|
|
# Unload the plugin if it matches the name we're searching for
|
||
|
|
if (plugin.name == name)
|
||
|
|
@framework.plugins.unload(plugin)
|
||
|
|
return { "result" => "success" }
|
||
|
|
end
|
||
|
|
}
|
||
|
|
return { "result" => "failure" }
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
def loaded(token)
|
||
|
|
authenticate(token)
|
||
|
|
ret = {}
|
||
|
|
ret[:plugins] = []
|
||
|
|
@framework.plugins.each do |plugin|
|
||
|
|
ret[:plugins] << plugin.name
|
||
|
|
end
|
||
|
|
ret
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|