dwelch-r7
96 lines
2.4 KiB
Ruby
96 lines
2.4 KiB
Ruby
# -*- coding: binary -*-
|
|
# frozen_string_literal: true
|
|
|
|
|
|
module Msf
|
|
###
|
|
#
|
|
# The feature manager is responsible for managing feature flags that can change characteristics of framework.
|
|
# Each feature will have a default value. The user can choose to override this default value if they wish.
|
|
###
|
|
class FeatureManager
|
|
|
|
include Singleton
|
|
|
|
CONFIG_KEY = 'framework/features'
|
|
WRAPPED_TABLES = 'wrapped_tables'
|
|
DEFAULTS = [
|
|
{
|
|
name: 'wrapped_tables',
|
|
description: 'When enabled Metasploit will wordwrap all tables to fit into the available terminal width',
|
|
default_value: false
|
|
}.freeze,
|
|
{
|
|
name: 'RHOST_HTTP_URL',
|
|
description: 'When enabled in supported modules you can specify a URL as a target',
|
|
default_value: false
|
|
}.freeze
|
|
].freeze
|
|
|
|
#
|
|
# Initializes the feature manager.
|
|
#
|
|
def initialize
|
|
@flag_lookup = DEFAULTS.each_with_object({}) do |feature, acc|
|
|
key = feature[:name]
|
|
acc[key] = feature.dup
|
|
end
|
|
end
|
|
|
|
def all
|
|
@flag_lookup.values.map do |feature|
|
|
feature.slice(:name, :description).merge(enabled: enabled?(feature[:name]))
|
|
end
|
|
end
|
|
|
|
def enabled?(name)
|
|
return false unless @flag_lookup[name]
|
|
|
|
feature = @flag_lookup[name]
|
|
feature.key?(:user_preference) ? feature[:user_preference] : feature[:default_value]
|
|
end
|
|
|
|
def exists?(name)
|
|
@flag_lookup.key?(name)
|
|
end
|
|
|
|
def names
|
|
all.map { |feature| feature[:name] }
|
|
end
|
|
|
|
def set(name, value)
|
|
return false unless @flag_lookup[name]
|
|
|
|
@flag_lookup[name][:user_preference] = value
|
|
|
|
if name == WRAPPED_TABLES
|
|
if value
|
|
Rex::Text::Table.wrap_tables!
|
|
else
|
|
Rex::Text::Table.unwrap_tables!
|
|
end
|
|
end
|
|
end
|
|
|
|
def load_config
|
|
conf = Msf::Config.load
|
|
conf.fetch(CONFIG_KEY, {}).each do |name, value|
|
|
set(name, value == 'true')
|
|
end
|
|
end
|
|
|
|
def save_config
|
|
# Note, we intentionally omit features that have not explicitly been set by the user.
|
|
config = Msf::Config.load
|
|
old_config = config.fetch(CONFIG_KEY, {})
|
|
new_config = @flag_lookup.values.each_with_object(old_config) do |feature, config|
|
|
next unless feature.key?(:user_preference)
|
|
|
|
config.merge!(feature[:name] => feature[:user_preference].to_s)
|
|
end
|
|
|
|
Msf::Config.save(CONFIG_KEY => new_config)
|
|
end
|
|
end
|
|
end
|