metasploit-framework/lib/rex/socket.rb

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