51 lines
851 B
Ruby
51 lines
851 B
Ruby
#!/usr/bin/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
|