144 lines
2.3 KiB
Ruby
144 lines
2.3 KiB
Ruby
require 'socket'
|
|
require 'resolv'
|
|
|
|
module Rex
|
|
|
|
###
|
|
#
|
|
# Socket
|
|
# ------
|
|
#
|
|
# Base class for all sockets.
|
|
#
|
|
###
|
|
class Socket
|
|
|
|
module Comm
|
|
end
|
|
|
|
require 'rex/socket/parameters'
|
|
|
|
##
|
|
#
|
|
# Factory methods
|
|
#
|
|
##
|
|
|
|
def self.create(opts = {})
|
|
return create_param(Rex::Socket::Parameters.from_hash(opts))
|
|
end
|
|
|
|
def self.create_param(param)
|
|
return param.comm.create(param)
|
|
end
|
|
|
|
def self.create_tcp(opts = {})
|
|
return create_param(Rex::Socket::Parameters.from_hash(opts.merge('Proto' => 'tcp')))
|
|
end
|
|
|
|
def self.create_tcp_server(opts)
|
|
return create_tcp(opts.merge('Server' => true))
|
|
end
|
|
|
|
def self.create_udp(opts = {})
|
|
return create_param(Rex::Socket::Parameters.from_hash(opts.merge('Proto' => 'udp')))
|
|
end
|
|
|
|
##
|
|
#
|
|
# Serialization
|
|
#
|
|
##
|
|
|
|
#
|
|
# Create a sockaddr structure using the supplied IP address, port, and
|
|
# address family
|
|
#
|
|
def self.to_sockaddr(ip, port, af = ::Socket::AF_INET)
|
|
ip = "0.0.0.0" unless ip
|
|
ip = Resolv.getaddress(ip)
|
|
data = [ af, port.to_i ] + ip.split('.').collect { |o| o.to_i } + [ "" ]
|
|
|
|
return data.pack('snCCCCa8')
|
|
end
|
|
|
|
#
|
|
# Returns the address family, host, and port of the supplied sockaddr as
|
|
# [ af, host, port ]
|
|
#
|
|
def self.from_sockaddr(saddr)
|
|
up = saddr.unpack('snCCCC')
|
|
|
|
af = up.shift
|
|
port = up.shift
|
|
|
|
return [ af, up.join('.'), port ]
|
|
end
|
|
|
|
#
|
|
# Resolves a host to raw network-byte order
|
|
#
|
|
def self.resolv_nbo(host)
|
|
return to_sockaddr(host, 0)[4,4]
|
|
end
|
|
|
|
##
|
|
#
|
|
# Class initialization
|
|
#
|
|
##
|
|
|
|
def initialize(sock, params = nil)
|
|
self.sock = sock
|
|
|
|
if (params)
|
|
self.peerhost = params.peerhost
|
|
self.peerport = params.peerport
|
|
self.localhost = params.localhost
|
|
self.localport = params.localport
|
|
end
|
|
end
|
|
|
|
#
|
|
# Closes the associated socket
|
|
#
|
|
def close
|
|
self.sock.close if (self.sock)
|
|
end
|
|
|
|
#
|
|
# Returns the sock context that was supplied to the constructor as the
|
|
# default poll_fd
|
|
#
|
|
def poll_fd
|
|
return self.sock
|
|
end
|
|
|
|
#
|
|
# Returns local connection information.
|
|
#
|
|
def getlocalname
|
|
return Socket.from_sockaddr(self.sock.getsockname)
|
|
end
|
|
|
|
alias getsockname getlocalname
|
|
|
|
#
|
|
# Return peer connection information.
|
|
#
|
|
def getpeername
|
|
return Socket.from_sockaddr(self.sock.getpeername)
|
|
end
|
|
|
|
attr_reader :sock
|
|
attr_reader :peerhost, :peerport, :localhost, :localport
|
|
|
|
protected
|
|
|
|
attr_writer :sock
|
|
attr_writer :peerhost, :peerport, :localhost, :localport
|
|
|
|
end
|
|
|
|
end
|