rapid7/metasploit-framework

# -*- coding: binary -*-

module Rex
module Post
module Meterpreter

###
#
# Mixin for classes that wish to have object aliases but do not
# really need to inherit from the ObjectAliases class.
#
###
module ObjectAliasesContainer

  #
  # Initialize the instance's aliases.
  #
  def initialize_aliases(aliases = {})
    self.aliases = aliases
  end

  #
  # Pass-thru aliases.
  #
  def method_missing(symbol, *args)
    self.aliases[symbol.to_s]
  end

  #
  # Recursively dumps all of the aliases registered with a class that
  # is kind_of? ObjectAliases.
  #
  def dump_alias_tree(parent_path, current = nil)
    items = []

    if (current == nil)
      current = self
    end

    # If the current object may have object aliases...
    if (current.kind_of?(Rex::Post::Meterpreter::ObjectAliases))
      current.aliases.each_key { |x|
        current_path = parent_path + '.' + x

        items << current_path

        items.concat(dump_alias_tree(current_path,
          current.aliases[x]))
      }
    end

    return items
  end

  #
  # The hash of aliases.
  #
  attr_accessor :aliases
end

###
#
# Generic object aliases from a class instance referenced symbol to an
# associated object of an arbitrary type
#
###
class ObjectAliases
  include Rex::Post::Meterpreter::ObjectAliasesContainer

  ##
  #
  # Constructor
  #
  ##

  # An instance
  def initialize(aliases = {})
    initialize_aliases(aliases)
  end
end


end; end; end