metasploit-framework/lib/rex/post/meterpreter/channel_container.rb

#!/usr/bin/ruby

module Rex
module Post
module Meterpreter

###
#
# This interface is meant to be included by things that are meant to contain
# zero or more channel instances in the form of a hash.
#
###
module ChannelContainer

	#
	# Initializes the channel association hash
	#
	def initialize_channels
		self.channels = {}
	end

	#
	# Adds a channel to the container that is indexed by its channel identifier
	#
	def add_channel(channel)
		self.channels[channel.cid] = channel
	end

	#
	# Looks up a channel instance based on its channel identifier
	#
	def find_channel(cid)
		return self.channels[cid]
	end

	#
	# Removes a channel based on its channel identifier
	#
	def remove_channel(cid)
		return self.channels.delete(cid)
	end

	#
	# The hash of channels.
	#
	attr_reader :channels

protected

	attr_writer :channels # :nodoc:

end

end; end; end