spec/support/acceptance/target.rb - master - githubmirror/metasploit-framework

spec/support/acceptance/target.rb
68 строк2 KB
adfoster-r7
Fix ssh login info details
01 июл 2026, 11:59
01 июл 2026, 11:59575e4f8
require_relative 'datastore_formatting'

module Acceptance
  ###
  # Stores the data for a target. These credentials can be used to create a sesion, or run a module against
  ###
  class Target
    include DatastoreFormatting

    attr_reader :session_module, :type, :datastore

    def initialize(options)
      @type = options.fetch(:type)
      @session_module = options.fetch(:session_module)
      @datastore = options.fetch(:datastore)
      @session_info_pattern = options[:session_info_pattern]
    end

    # Returns the session info pattern to verify after opening a session, or nil if not set.
    # @return [Regexp, nil]
    def session_info_pattern
      @session_info_pattern
    end

    # @param [Hash] default_global_datastore
    # @return [String] The setg commands for setting the global datastore
    def setg_commands(default_global_datastore: {})
      commands = []
      # Ensure the global framework datastore is always clear
      commands << "irb -e '(self.respond_to?(:framework) ? framework : self).datastore.user_defined.clear'"
      # Call setg
      global_datastore = default_global_datastore.merge(@datastore[:global])
      global_datastore.each do |key, value|
        commands << "setg #{key} #{value}"
      end
      commands.join("\n")
    end

    # @param [Hash] default_module_datastore
    # @return [String] The datastore options string
    def datastore_options(default_module_datastore: {})
      module_datastore = default_module_datastore.merge(@datastore[:module])
      format_datastore_options(module_datastore)
    end

    # @param [Hash] default_module_datastore
    # @return [String] The command which can be used on msfconsole to generate the payload
    def run_command(default_module_datastore: {})
      "run #{datastore_options(default_module_datastore: default_module_datastore)}"
    end

    # @param [Hash] default_global_datastore
    # @param [Hash] default_module_datastore
    # @return [String] A human readable representation of the payload configuration object
    def as_readable_text(default_global_datastore: {}, default_module_datastore: {})
      <<~EOF
         ## Session module 
         use #{session_module}

         ## Set global datastore
         #{setg_commands(default_global_datastore: default_global_datastore)}

         ## Run command
         #{run_command(default_module_datastore: default_module_datastore)}
      EOF
    end
  end
end