84 lines
1.4 KiB
Ruby
84 lines
1.4 KiB
Ruby
|
# -*- coding: binary -*-
|
||
|
|
|
||
|
|
module Rex
|
||
|
|
module Post
|
||
|
|
module HWBridge
|
||
|
|
|
||
|
|
###
|
||
|
|
#
|
||
|
|
# 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::HWBridge::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::HWBridge::ObjectAliasesContainer
|
||
|
|
|
||
|
|
##
|
||
|
|
#
|
||
|
|
# Constructor
|
||
|
|
#
|
||
|
|
##
|
||
|
|
|
||
|
|
# An instance
|
||
|
|
def initialize(aliases = {})
|
||
|
|
initialize_aliases(aliases)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
|
||
|
|
end; end; end
|