librex 0.0.17 → 0.0.18

data/lib/rex/socket/comm/local.rb.ut.rb

@@ -1,75 +0,0 @@
-#!/usr/bin/env ruby
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))
-
-require 'test/unit'
-require 'rex/exceptions'
-require 'rex/socket/parameters'
-require 'rex/socket/comm/local'
-
-class Rex::Socket::Comm::Local::UnitTest < Test::Unit::TestCase
-
-	def test_create_tcp
-		test_port   = 64432
-		test_server = TCPServer.new('127.0.0.1', test_port)
-
-		# Create a stream connection to the stub listener
-		stream = nil
-
-		assert_nothing_raised {
-			stream = Rex::Socket::Comm::Local.create(
-				Rex::Socket::Parameters.from_hash(
-					'PeerHost' => '127.0.0.1',
-					'PeerPort' => test_port,
-					'Proto'    => 'tcp'))
-		}
-
-		assert_kind_of(Rex::IO::Stream, stream, "valid Stream instance")
-		assert_kind_of(Rex::Socket::Tcp, stream, "valid Tcp instance")
-		stream.close
-
-		# Now create a bare connection to the listener
-		stream = nil
-
-		assert_nothing_raised {
-			stream = Rex::Socket::Comm::Local.create(
-				Rex::Socket::Parameters.from_hash(
-					'PeerHost' => '127.0.0.1',
-					'PeerPort' => test_port,
-					'Proto'    => 'tcp',
-					'Bare'     => true))
-		}
-
-		assert_kind_of(Socket, stream, "valid Socket instance")
-
-		assert_raise(Rex::ConnectionRefused, "connection refused failed") {
-			Rex::Socket::Comm::Local.create(
-				Rex::Socket::Parameters.from_hash(
-					'PeerHost' => '127.0.0.1',
-					'PeerPort' => 1,
-					'Proto'    => 'tcp',
-					'Bare'     => true))
-		}
-
-		stream.close
-
-		test_server.close
-	end
-
-	def test_create_tcp_server
-		# TODO
-	end
-
-	def test_create_udp
-		# TODO
-	end
-
-	def test_create_invalid
-		assert_raise(Rex::UnsupportedProtocol, "invalid protocol check failed") {
-			Rex::Socket::Comm::Local.create(
-				Rex::Socket::Parameters.from_hash(
-					'Proto' => 'invalid'))
-		}
-	end
-
-end

data/lib/rex/socket/ip.rb

@@ -1,130 +0,0 @@
-require 'rex/socket'
-
-###
-#
-# This class provides methods for interacting with a IP socket.
-#
-###
-module Rex::Socket::Ip
-
-	include Rex::Socket
-
-	##
-	#
-	# Factory
-	#
-	##
-
-	#
-	# Creates the client using the supplied hash.
-	#
-	def self.create(hash = {})
-		hash['Proto'] = 'ip'
-		self.create_param(Rex::Socket::Parameters.from_hash(hash))
-	end
-
-	#
-	# Wrapper around the base socket class' creation method that automatically
-	# sets the parameter's protocol to IP.
-	#
-	def self.create_param(param)
-		param.proto = 'ip'
-		Rex::Socket.create_param(param)
-	end
-
-	##
-	#
-	# IP connected state methods
-	#
-	##
-
-	#
-	# Write the supplied datagram to the connected IP socket.
-	#
-	def write(gram)
-		raise RuntimeError, "IP sockets must use sendto(), not write()"
-	end
-
-	alias put write
-
-	#
-	# Read a datagram from the IP socket.
-	#
-	def read(length = 65535)
-		raise RuntimeError, "IP sockets must use recvfrom(), not read()"
-	end
-
-	##
-	#
-	# IP non-connected state methods
-	#
-	##
-
-	#
-	# Sends a datagram to the supplied host:port with optional flags.
-	#
-	def sendto(gram, peerhost, flags = 0)
-		dest = ::Socket.pack_sockaddr_in(0, peerhost)
-
-		# Some BSDs require byteswap for len and offset
-		if(
-			Rex::Compat.is_freebsd or
-			Rex::Compat.is_netbsd or
-			Rex::Compat.is_bsdi or
-			Rex::Compat.is_macosx
-		  )
-			gram=gram.dup
-			gram[2,2]=gram[2,2].unpack("n").pack("s")
-			gram[6,2]=gram[6,2].unpack("n").pack("s")
-		end
-
-		begin
-			send(gram, flags, dest)
-		rescue  ::Errno::EHOSTUNREACH,::Errno::ENETDOWN,::Errno::ENETUNREACH,::Errno::ENETRESET,::Errno::EHOSTDOWN,::Errno::EACCES,::Errno::EINVAL,::Errno::EADDRNOTAVAIL
-			return nil
-		end
-
-	end
-
-	#
-	# Receives a datagram and returns the data and host of the requestor
-	# as [ data, host ].
-	#
-	def recvfrom(length = 65535, timeout=def_read_timeout)
-		begin
-			if ((rv = ::IO.select([ fd ], nil, nil, timeout)) and
-			    (rv[0]) and (rv[0][0] == fd)
-			   )
-					data, saddr    = super(length)
-					af, host       = Rex::Socket.from_sockaddr(saddr)
-
-					return [ data, host ]
-			else
-				return [ '', nil ]
-			end
-		rescue Exception
-			return [ '', nil ]
-		end
-	end
-
-	#
-	# Calls recvfrom and only returns the data
-	#
-	def get(timeout=nil)
-		data, saddr = recvfrom(65535, timeout)
-		return data
-	end
-
-	#
-	# The default number of seconds to wait for a read operation to timeout.
-	#
-	def def_read_timeout
-		10
-	end
-
-	def type?
-		return 'ip'
-	end
-
-end
-

data/lib/rex/socket/parameters.rb

@@ -1,345 +0,0 @@
-require 'rex/socket'
-
-###
-#
-# This class represents the set of parameters that are used to create
-# a socket, whether it be a server or client socket.
-#
-###
-class Rex::Socket::Parameters
-
-	##
-	#
-	# Factory
-	#
-	##
-
-	#
-	# Creates an instance of the Parameters class using the supplied hash.
-	#
-	def self.from_hash(hash)
-		return self.new(hash)
-	end
-
-	##
-	#
-	# Constructor
-	#
-	##
-
-	#
-	# Initializes the attributes from the supplied hash.  The following hash
-	# keys can be specified.
-	#
-	# PeerHost / PeerAddr
-	#
-	# 	The remote host to connect to.
-	#
-	# PeerPort
-	#
-	# 	The remote port to connect to.
-	#
-	# LocalHost / LocalAddr
-	#
-	# 	The local host to communicate from, if any.
-	#
-	# LocalPort
-	#
-	# 	The local port to communicate from, if any.
-	#
-	# Bare
-	#
-	# 	Create a bare socket.
-	#
-	# Server
-	#
-	# 	Whether or not this should be a server.
-	#
-	# SSL
-	#
-	# 	Whether or not SSL should be used.
-	#
-	# SSLVersion
-	#
-	# 	Specify SSL2, SSL3, or TLS1 (SSL3 is default)
-	#	
-	# Proxies
-	#
-	#	List of proxies to use.
-	#
-	# Proto
-	#
-	#	The underlying protocol to use.
-	#
-	# IPv6
-	#
-	#	Force the use of IPv6.
-	#
-	# Comm
-	#
-	# 	The underlying Comm class to use to create the socket for this parameter
-	# 	set.
-	#
-	# Context
-	#
-	# 	A context hash that can allow users of this parameter class instance to
-	# 	determine who is responsible for requesting that a socket be created.
-	#
-	# Retries
-	#
-	# 	The number of times a connection should be retried.
-	#
-	# Timeout
-	#
-	# 	The number of seconds before a connection should time out
-	#
-	
-	def initialize(hash)
-		if (hash['PeerHost'])
-			self.peerhost = hash['PeerHost']
-		elsif (hash['PeerAddr'])
-			self.peerhost = hash['PeerAddr']
-		else
-			self.peerhost = nil
-		end
-
-		if (hash['LocalHost'])
-			self.localhost = hash['LocalHost']
-		elsif (hash['LocalAddr'])
-			self.localhost = hash['LocalAddr']
-		else
-			self.localhost = '0.0.0.0'
-		end
-
-		if (hash['PeerPort'])
-			self.peerport = hash['PeerPort'].to_i
-		else
-			self.peerport = 0
-		end
-
-		if (hash['LocalPort'])
-			self.localport = hash['LocalPort'].to_i
-		else
-			self.localport = 0
-		end
-
-		if (hash['Bare'])
-			self.bare = hash['Bare']
-		else
-			self.bare = false
-		end
-
-		if (hash['SSL'] and hash['SSL'].to_s =~ /^(t|y|1)/i)
-			self.ssl = true
-		else
-			self.ssl = false
-		end
-		
-		if (hash['SSLVersion'] and hash['SSLVersion'].to_s =~ /^(SSL2|SSL3|TLS1)$/i)
-			self.ssl_version = hash['SSLVersion']
-		end
-		
-		if hash['Proxies']	
-			self.proxies = hash['Proxies'].split('-').map{|a| a.strip}.map{|a| a.split(':').map{|b| b.strip}}
-		end
-		
-		# The protocol this socket will be using
-		if (hash['Proto'])
-			self.proto = hash['Proto'].downcase
-		else
-			self.proto = 'tcp'
-		end
-
-		# Whether or not the socket should be a server
-		self.server    = hash['Server'] || false
-
-		# The communication subsystem to use to create the socket
-		self.comm      = hash['Comm']
-
-		# The context that was passed in, if any.
-		self.context   = hash['Context'] || {}
-		
-		# If no comm was supplied, try to use the comm that is best fit to
-		# handle the provided host based on the current routing table.
-		if( self.server )
-			if (self.comm == nil and self.localhost)
-				self.comm  = Rex::Socket::SwitchBoard.best_comm(self.localhost)
-			end
-		else
-			if (self.comm == nil and self.peerhost)
-				self.comm  = Rex::Socket::SwitchBoard.best_comm(self.peerhost)
-			end
-		end
-
-		# If we still haven't found a comm, we default to the local comm.
-		self.comm      = Rex::Socket::Comm::Local if (self.comm == nil)
-		
-		# If we are a UDP server, turn off the server flag as it was only set when 
-		# creating the UDP socket in order to avail of the switch board above.
-		if( self.server and self.proto == 'udp' )
-			self.server = false
-		end
-		
-		# The number of connection retries to make (client only)
-		if hash['Retries']
-			self.retries = hash['Retries'].to_i
-		else
-			self.retries = 0
-		end
-
-		# The number of seconds before a connect attempt times out (client only)
-		if hash['Timeout']
-			self.timeout = hash['Timeout'].to_i
-		else
-			self.timeout = 5
-		end
-	
-		# Whether to force IPv6 addressing
-		self.v6        = hash['IPv6'] || false
-	end
-
-	##
-	#
-	# Conditionals
-	#
-	##
-
-	#
-	# Returns true if this represents parameters for a server.
-	#
-	def server?
-		return (server == true)
-	end
-
-	#
-	# Returns true if this represents parameters for a client.
-	#
-	def client?
-		return (server == false)
-	end
-
-	#
-	# Returns true if the protocol for the parameters is TCP.
-	#
-	def tcp?
-		return (proto == 'tcp')
-	end
-
-	#
-	# Returns true if the protocol for the parameters is UDP.
-	#
-	def udp?
-		return (proto == 'udp')
-	end
-
-	#
-	# Returns true if the protocol for the parameters is IP.
-	#
-	def ip?
-		return (proto == 'ip')
-	end
-	
-	#
-	# Returns true if the socket is a bare socket that does not inherit from
-	# any extended Rex classes.
-	#
-	def bare?
-		return (bare == true)
-	end
-
-	#
-	# Returns true if SSL has been requested.
-	#
-	def ssl?
-		return ssl
-	end
-
-	#
-	# Returns true if IPv6 has been enabled
-	#
-	def v6?
-		return v6
-	end
-
-
-	##
-	#
-	# Attributes
-	#
-	##
-
-	#
-	# The remote host information, equivalent to the PeerHost parameter hash
-	# key.
-	#
-	attr_accessor :peerhost
-	#
-	# The remote port.  Equivalent to the PeerPort parameter hash key.
-	#
-	attr_accessor :peerport
-	#
-	# The local host.  Equivalent to the LocalHost parameter hash key.
-	#
-	attr_accessor :localhost
-	#
-	# The local port.  Equivalent to the LocalPort parameter hash key.
-	#
-	attr_accessor :localport
-	#
-	# The protocol to to use, such as TCP.  Equivalent to the Proto parameter
-	# hash key.
-	#
-	attr_accessor :proto
-	#
-	# Whether or not this is a server.  Equivalent to the Server parameter hash
-	# key.
-	#
-	attr_accessor :server
-	#
-	# The Comm class that should be used to create the underlying socket.
-	#
-	attr_accessor :comm
-	#
-	# The context hash that was passed in to the structure.
-	#
-	attr_accessor :context
-	#
-	# The number of attempts that should be made.
-	#
-	attr_accessor :retries
-	#
-	# The number of seconds before a connection attempt should time out.
-	#
-	attr_accessor :timeout
-	#
-	# Whether or not this is a bare (non-extended) socket instance that should
-	# be created.
-	#
-	attr_accessor :bare
-	#
-	# Whether or not SSL should be used to wrap the connection.
-	#
-	attr_accessor :ssl
-	#
-	# What version of SSL to use (SSL2, SSL3, TLS1)
-	#
-	attr_accessor :ssl_version
-	#
-	# Whether we should use IPv6
-	#
-	attr_accessor :v6
-
-
-	attr_accessor :proxies
-
-
-	##
-	#
-	# Synonyms
-	#
-	##
-	
-	alias peeraddr  peerhost
-	alias localaddr localhost
-
-end