metasploit-framework/lib/rex/io/stream_abstraction.rb

#!/usr/bin/env ruby

require 'socket'

module Rex
module IO

###
#
# This class provides an abstraction to a stream based
# connection through the use of a streaming socketpair.
#
###
module StreamAbstraction

	#
	# This method creates a streaming socket pair and initializes it.
	#
	def initialize_abstraction
		self.lsock, self.rsock = ::Socket.pair(::Socket::AF_UNIX,
				::Socket::SOCK_STREAM, 0)

		self.lsock.extend(Rex::IO::Stream)
		self.rsock.extend(Rex::IO::Stream)
	end

	#
	# This method cleans up the abstraction layer.
	#
	def cleanup_abstraction
		self.lsock.close if (self.lsock)
		self.rsock.close if (self.rsock)

		self.lsock = nil
		self.rsock = nil
	end

	#
	# The left side of the stream.
	#
	attr_reader :lsock
	#
	# The right side of the stream.
	#
	attr_reader :rsock
protected
	attr_writer :lsock, :rsock # :nodoc:
end

end; end