b03d6b52e7
git-svn-id: https://beef.googlecode.com/svn/trunk@945 b87d56ec-f9c0-11de-8c8a-61c5e9addfc9
47 lines
1.7 KiB
Ruby
47 lines
1.7 KiB
Ruby
module BeEF
|
|
module Module
|
|
|
|
# Checks to see if module is in configuration
|
|
def self.is_present(mod)
|
|
return BeEF::Core::Configuration.instance.get('beef.module').has_key?(mod.to_s)
|
|
end
|
|
|
|
# Checks to see if module is enabled in configuration
|
|
def self.is_enabled(mod)
|
|
return (self.is_present(mod) and BeEF::Core::Configuration.instance.get('beef.module.'+mod.to_s+'.enable') == true)
|
|
end
|
|
|
|
# Checks to see if the module reports that it has loaded through the configuration
|
|
def self.is_loaded(mod)
|
|
return (self.is_enabled(mod) and BeEF::Core::Configuration.instance.get('beef.module.'+mod.to_s+'.loaded') == true)
|
|
end
|
|
|
|
# Loads module
|
|
def self.load(mod, cat)
|
|
cat = BeEF::Module.safe_category(cat)
|
|
if File.exists?('modules/'+cat+'/'+mod+'/module.rb')
|
|
require 'modules/'+cat+'/'+mod+'/module.rb'
|
|
BeEF::Core::Configuration.instance.set('beef.module.'+mod+'.loaded', true)
|
|
BeEF::Core::Configuration.instance.set('beef.module.'+mod+'.class', mod.capitalize)
|
|
print_debug "Loaded module: '#{mod}'"
|
|
return true
|
|
end
|
|
print_error "Unable to load module '#{mod}'"
|
|
return false
|
|
end
|
|
|
|
# Return module key by database id
|
|
def self.get_key_by_database_id(id)
|
|
ret = BeEF::Core::Configuration.instance.get('beef.module').select {|k, v| v.has_key?('db') and v['db']['id'].to_i == id.to_i }
|
|
return (ret.kind_of?(Array)) ? ret.first.first : ret.keys.first
|
|
end
|
|
|
|
# Returns category name in a system folder format
|
|
def self.safe_category(cat)
|
|
return cat.to_s.strip.downcase.sub(/\s/, '_')
|
|
end
|
|
end
|
|
end
|
|
|
|
|