85 lines
1.4 KiB
Ruby
85 lines
1.4 KiB
Ruby
#!/usr/bin/env ruby
|
|
# -*- 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
|