187 lines
3.0 KiB
Ruby
187 lines
3.0 KiB
Ruby
#!/usr/bin/env ruby
|
|
|
|
require 'socket'
|
|
require 'fcntl'
|
|
|
|
module Rex
|
|
module IO
|
|
|
|
###
|
|
#
|
|
# This class provides an abstraction to a stream based
|
|
# connection through the use of a streaming socketpair.
|
|
#
|
|
###
|
|
module StreamAbstraction
|
|
|
|
###
|
|
#
|
|
# Extension information for required Stream interface.
|
|
#
|
|
###
|
|
module Ext
|
|
|
|
#
|
|
# Initializes peer information.
|
|
#
|
|
def initinfo(peer,local)
|
|
@peer = peer
|
|
@local = local
|
|
end
|
|
|
|
#
|
|
# Symbolic peer information.
|
|
#
|
|
def peerinfo
|
|
(@peer || "Remote Pipe")
|
|
end
|
|
|
|
#
|
|
# Symbolic local information.
|
|
#
|
|
def localinfo
|
|
(@local || "Local Pipe")
|
|
end
|
|
end
|
|
|
|
#
|
|
# This method creates a streaming socket pair and initializes it.
|
|
#
|
|
def initialize_abstraction
|
|
self.lsock, self.rsock = Rex::Socket.tcp_socket_pair()
|
|
self.lsock.extend(Rex::IO::Stream)
|
|
self.lsock.extend(Ext)
|
|
self.rsock.extend(Rex::IO::Stream)
|
|
|
|
self.monitor_rsock
|
|
|
|
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
|
|
|
|
#
|
|
# Writes to the local side.
|
|
#
|
|
def syswrite(buffer)
|
|
lsock.syswrite(buffer)
|
|
end
|
|
|
|
#
|
|
# Reads from the local side.
|
|
#
|
|
def sysread(length)
|
|
lsock.sysread(length)
|
|
end
|
|
|
|
#
|
|
# Shuts down the local side of the stream abstraction.
|
|
#
|
|
def shutdown(how)
|
|
lsock.shutdown(how)
|
|
end
|
|
|
|
#
|
|
# Closes both sides of the stream abstraction.
|
|
#
|
|
def close
|
|
cleanup_abstraction
|
|
end
|
|
|
|
#
|
|
# Symbolic peer information.
|
|
#
|
|
def peerinfo
|
|
"Remote-side of Pipe"
|
|
end
|
|
|
|
#
|
|
# Symbolic local information.
|
|
#
|
|
def localinfo
|
|
"Local-side of Pipe"
|
|
end
|
|
|
|
#
|
|
# The left side of the stream.
|
|
#
|
|
attr_reader :lsock
|
|
#
|
|
# The right side of the stream.
|
|
#
|
|
attr_reader :rsock
|
|
|
|
protected
|
|
|
|
def monitor_rsock
|
|
self.monitor_thread = ::Thread.new {
|
|
loop do
|
|
closed = false
|
|
buf = nil
|
|
|
|
begin
|
|
s = Rex::ThreadSafe.select( [ self.rsock ], nil, nil, 0.2 )
|
|
if( s == nil || s[0] == nil )
|
|
next
|
|
end
|
|
rescue Exception => e
|
|
closed = true
|
|
end
|
|
|
|
if( closed == false )
|
|
begin
|
|
buf = self.rsock.sysread( 32768 )
|
|
closed = true if( buf == nil )
|
|
rescue
|
|
closed = true
|
|
end
|
|
end
|
|
|
|
if( closed == false )
|
|
total_sent = 0
|
|
total_length = buf.length
|
|
while( total_sent < total_length )
|
|
begin
|
|
data = buf[0, buf.length]
|
|
sent = self.write( data )
|
|
# sf: Only remove the data off the queue is syswrite was successfull.
|
|
# This way we naturally perform a resend if a failure occured.
|
|
# Catches an edge case with meterpreter TCP channels where remote send
|
|
# failes gracefully and a resend is required.
|
|
if( sent > 0 )
|
|
total_sent += sent
|
|
buf[0, sent] = ""
|
|
end
|
|
rescue ::IOError => e
|
|
closed = true
|
|
break
|
|
end
|
|
end
|
|
end
|
|
|
|
if( closed )
|
|
self.close_write
|
|
::Thread.exit
|
|
end
|
|
end
|
|
}
|
|
end
|
|
|
|
protected
|
|
attr_accessor :monitor_thread
|
|
attr_writer :lsock
|
|
attr_writer :rsock
|
|
|
|
end
|
|
|
|
end; end
|
|
|