Luke Imhoff
471ac6d15d
Msf::Modules::Loader::Archive#each_module_reference_name tried to check the enabled types for the module_manager by accessing the enabledment_by_type Hash, which is protected. Instead, it should use the public type_enabled? method. Add specs to test all of Msf::Modules::Loader::Archive while testing each_module_reference_name. In order to properly test that modules could be found in archives, I had to produce a fastlib archive, so there is now a spec for FastLib.dump and FastLib.load. Some specs are marked pending as I found a bug in FastLib, which has a work-around. The bug is filed in PivotalTracker as https://www.pivotaltracker.com/story/show/38730815 and the pending tests include the URL also in their tags.
77 lines
2.2 KiB
Ruby
77 lines
2.2 KiB
Ruby
require 'msf/core/modules/loader/base'
|
|
|
|
# Concerns loading modules form fastlib archives
|
|
class Msf::Modules::Loader::Archive < Msf::Modules::Loader::Base
|
|
#
|
|
# CONSTANTS
|
|
#
|
|
|
|
# The extension for Fastlib archives.
|
|
ARCHIVE_EXTENSION = '.fastlib'
|
|
|
|
# Returns true if the path is a Fastlib archive.
|
|
#
|
|
# @param (see Msf::Modules::Loader::Base#loadable?)
|
|
# @return [true] if path has the {ARCHIVE_EXTENSION} extname.
|
|
# @return [false] otherwise
|
|
def loadable?(path)
|
|
if File.extname(path) == ARCHIVE_EXTENSION
|
|
true
|
|
else
|
|
false
|
|
end
|
|
end
|
|
|
|
protected
|
|
|
|
# Yields the module_reference_name for each module file in the Fastlib archive at path.
|
|
#
|
|
# @param path [String] The path to the Fastlib archive file.
|
|
# @yield (see Msf::Modules::Loader::Base#each_module_reference_name)
|
|
# @yieldparam (see Msf::Modules::Loader::Base#each_module_reference_name)
|
|
# @return (see Msf::Modules::Loader::Base#each_module_reference_name)
|
|
def each_module_reference_name(path)
|
|
entries = ::FastLib.list(path)
|
|
|
|
entries.each do |entry|
|
|
if entry.include?('.svn/')
|
|
next
|
|
end
|
|
|
|
type = entry.split('/', 2)[0]
|
|
type = type.singularize
|
|
|
|
unless module_manager.type_enabled?(type)
|
|
next
|
|
end
|
|
|
|
if module_path?(entry)
|
|
# The module_reference_name doesn't have a file extension
|
|
module_reference_name = module_reference_name_from_path(entry)
|
|
|
|
yield path, type, module_reference_name
|
|
end
|
|
end
|
|
end
|
|
|
|
# Returns the path to the module inside the Fastlib archive. The path to the archive is separated from the path to
|
|
# the file inside the archive by '::'.
|
|
#
|
|
# @param (see Msf::Modules::Loader::Base#module_path)
|
|
# @return [String] Path to module file inside the Fastlib archive.
|
|
def module_path(parent_path, type, module_reference_name)
|
|
file_path = typed_path(type, module_reference_name)
|
|
module_path = "#{parent_path}::#{file_path}"
|
|
|
|
module_path
|
|
end
|
|
|
|
# Loads the module content from the Fastlib archive.
|
|
#
|
|
# @return (see Msf::Modules::Loader::Base#read_module_content)
|
|
def read_module_content(path, type, module_reference_name)
|
|
file_path = typed_path(type, module_reference_name)
|
|
|
|
::FastLib.load(path, file_path)
|
|
end
|
|
end |