115 lines
1.6 KiB
Ruby
115 lines
1.6 KiB
Ruby
module Msf
|
|
|
|
###
|
|
#
|
|
# Tcp
|
|
# ---
|
|
#
|
|
# This module provides methods for establish a connection to a remote host and
|
|
# communicating with it.
|
|
#
|
|
###
|
|
module Exploit::Remote::Tcp
|
|
|
|
def initialize(info = {})
|
|
super
|
|
|
|
register_options(
|
|
[
|
|
Opt::RHOST,
|
|
Opt::RPORT,
|
|
], Msf::Exploit::Remote::Tcp)
|
|
end
|
|
|
|
#
|
|
# Establishes a TCP connection to the specified RHOST/RPORT
|
|
#
|
|
def connect(global = true)
|
|
nsock = Rex::Socket::Tcp.create(
|
|
'PeerHost' => datastore['RHOST'],
|
|
'PeerPort' => datastore['RPORT'].to_i,
|
|
'LocalHost' => datastore['CHOST'] || "0.0.0.0",
|
|
'LocalPort' => datastore['CPORT'] ? datastore['CPORT'].to_i : 0)
|
|
|
|
# Set this socket to the global socket as necessary
|
|
self.sock = nsock if (global)
|
|
|
|
return nsock
|
|
end
|
|
|
|
def handler(nsock = self.sock)
|
|
# If the handler claims the socket, then we don't want it to get closed
|
|
# during cleanup
|
|
if (((rv = super) == Handler::Claimed) and
|
|
(nsock == self.sock))
|
|
self.sock = nil
|
|
end
|
|
|
|
return rv
|
|
end
|
|
|
|
#
|
|
# Closes the TCP connection
|
|
#
|
|
def disconnect(nsock = self.sock)
|
|
if (nsock)
|
|
nsock.shutdown
|
|
nsock.close
|
|
end
|
|
|
|
if (nsock == sock)
|
|
self.sock = nil
|
|
end
|
|
end
|
|
|
|
#
|
|
# Performs cleanup, disconnects the socket if necessary
|
|
#
|
|
def cleanup
|
|
super
|
|
|
|
disconnect
|
|
end
|
|
|
|
##
|
|
#
|
|
# Wrappers for getters
|
|
#
|
|
##
|
|
|
|
#
|
|
# Returns the target host
|
|
#
|
|
def rhost
|
|
datastore['RHOST']
|
|
end
|
|
|
|
#
|
|
# Returns the remote port
|
|
#
|
|
def rport
|
|
datastore['RPORT']
|
|
end
|
|
|
|
#
|
|
# Returns the local host
|
|
#
|
|
def lhost
|
|
datastore['LHOST']
|
|
end
|
|
|
|
#
|
|
# Returns the local port
|
|
#
|
|
def lport
|
|
datastore['LPORT']
|
|
end
|
|
|
|
protected
|
|
|
|
attr_accessor :sock
|
|
|
|
end
|
|
|
|
end
|