librex 0.0.68 → 0.0.70

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

@@ -1,108 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..'))
-
-require 'test/unit'
-require 'rex/socket'
-require 'rex/socket/tcp'
-
-class Rex::Socket::UnitTest < Test::Unit::TestCase
-
-	def test_ip
-		assert_equal(true,Rex::Socket.dotted_ip?('0.0.0.0'), 'valid IP min')
-		assert_equal(true,Rex::Socket.dotted_ip?('255.255.255.255'), 'valid IP max')
-		assert_equal(false,Rex::Socket.dotted_ip?('0.0.0.0.0'), 'too many sections')
-		assert_equal(false,Rex::Socket.dotted_ip?('0..0.0.0'), 'too many dots')
-		assert_equal(false,Rex::Socket.dotted_ip?('00.0.0'), 'not enough dots')
-		assert_equal(false,Rex::Socket.dotted_ip?('256.256.256.256'), 'numbers too big')
-	end
-
-	def test_create
-		port = 64442
-		serv = TCPServer.new('127.0.0.1', port)
-
-		sock = nil
-		assert_nothing_raised {
-			sock = Rex::Socket.create(
-				'PeerHost' => '127.0.0.1',
-				'PeerPort' => port,
-				'Proto'    => 'tcp')
-		}
-		assert_kind_of(Rex::Socket::Tcp, sock, "socket factory creation")
-
-		sock = nil
-		assert_nothing_raised {
-			sock = Rex::Socket.create_tcp(
-				'PeerHost' => '127.0.0.1',
-				'PeerPort' => port)
-		}
-		assert_kind_of(Rex::Socket::Tcp, sock, "tcp socket factory creation")
-
-		serv.close
-	end
-
-	def test_to_sockaddr
-		assert_equal(([2] + [0]*14).pack("sC*"), Rex::Socket.to_sockaddr(0, 0), "null sockaddr")
-=begin
-# This is platform dependent, pain to test
-		if (Rex::Socket.support_ipv6?)
-			# Use the constant for AF_INET6 since it is different per platform
-			# (10 on linux and 28 on BSD)
-			inaddr_any_sockaddr = ([::Socket::AF_INET6, 22] + [0]*24).pack('sSC*')
-		else
-			inaddr_any_sockaddr = ([2, 22] + [0]*12).pack('snC*')
-		end
-=end
-		assert_equal(([2, 0x16, 1, 2, 3, 4] + [0]*8).pack('snC*'), Rex::Socket.to_sockaddr("1.2.3.4", 22), "1.2.3.4 addr, port 22 sockaddr")
-	end
-
-	def test_from_sockaddr
-		# 1.9.1 raises ArgumentError if we don't have an af == AF_INET or AF_INET6
-		af, host, port = Rex::Socket.from_sockaddr(([2, 0] + [0]*12).pack('snC*'))
-		assert_equal(2, af, "af = 2")
-		assert_equal('0.0.0.0', host, "zero host")
-		assert_equal(0, port, "zero port")
-
-		af, host, port = Rex::Socket.from_sockaddr(([2, 22]+[0]*12).pack('snC*'))
-		assert_equal(2, af, "af = 2")
-		assert_equal(22, port, "port = 22")
-		assert_equal('0.0.0.0', host, "zero host")
-
-		af, host, port = Rex::Socket.from_sockaddr(([2, 22, 1, 2, 3, 4] + [0]*8).pack('snC*') )
-		assert_equal(2, af, "af = 2")
-		assert_equal('1.2.3.4', host, "host = '1.2.3.4'")
-		assert_equal(22, port, "port = 22")
-	end
-
-	def test_resolv_nbo
-		assert_equal("\x04\x03\x02\x01", Rex::Socket.resolv_nbo("4.3.2.1"))
-	end
-
-	def test_net2bitmask
-		assert_equal(32, Rex::Socket.net2bitmask('255.255.255.255'))
-		assert_equal(28, Rex::Socket.net2bitmask('255.255.255.240'))
-		assert_equal(24, Rex::Socket.net2bitmask('255.255.255.0'))
-		assert_equal(16, Rex::Socket.net2bitmask('255.255.0.0'))
-	end
-
-	def test_bit2netmask
-		assert_equal("255.255.255.255", Rex::Socket.bit2netmask(32))
-		assert_equal("255.255.255.254", Rex::Socket.bit2netmask(31))
-		assert_equal("255.255.255.240", Rex::Socket.bit2netmask(28))
-		assert_equal("255.255.255.0", Rex::Socket.bit2netmask(24))
-		assert_equal("255.255.0.0", Rex::Socket.bit2netmask(16))
-	end
-
-	def test_is_internal
-		assert( ! Rex::Socket.is_internal?("1.2.3.4"))
-		assert( ! Rex::Socket.is_internal?("172.15.3.4"))
-		assert( ! Rex::Socket.is_internal?("172.32.3.4"))
-		assert(Rex::Socket.is_internal?("10.2.3.4"))
-		assert(Rex::Socket.is_internal?("192.168.3.4"))
-		16.upto(31) do |octet|
-			assert(Rex::Socket.is_internal?("172.#{octet}.3.4"))
-		end
-	end
-
-end

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

@@ -1,76 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.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/parameters.rb.ut.rb

@@ -1,52 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..'))
-
-require 'test/unit'
-require 'rex/socket/parameters'
-
-class Rex::Socket::Parameters::UnitTest < Test::Unit::TestCase
-
-	def test_parameters
-		h = { 
-				'PeerHost'  => 'phost',
-				'PeerPort'  => 12,
-				'LocalHost' => 'lhost',
-				'LocalPort' => 47,
-				'Bare'      => true,
-				'Server'    => true,
-				'Comm'      => 'nothing',
-				'Proto'     => 'tcp',
-				'SSL'       => true
-		    }
-
-		p = Rex::Socket::Parameters.from_hash(h)
-
-		assert_equal('phost', p.peerhost, "peerhost")
-		assert_equal('phost', p.peeraddr, "peeraddr")
-		assert_equal(12, p.peerport, "peerport")
-		assert_equal('lhost', p.localhost, "localhost")
-		assert_equal('lhost', p.localaddr, "localaddr")
-		assert_equal(47, p.localport, "localport")
-		assert_equal(true, p.bare?, "bare")
-		assert_equal(true, p.server?, "server")
-		assert_equal(false, p.client?, "client")
-		assert_equal('nothing', p.comm, "comm")
-		assert_equal(true, p.tcp?, "proto tcp")
-		assert_equal(false, p.udp?, "proto udp")
-		assert_equal(true, p.ssl, "ssl")
-
-		p = Rex::Socket::Parameters.from_hash({})
-
-		assert_equal(nil, p.peerhost, "null peerhost")
-		assert_equal('0.0.0.0', p.localhost, "default localhost")
-		assert_equal(0, p.peerport, "0 peerport")
-		assert_equal(0, p.localport, "0 localport")
-		assert_equal(false, p.bare, "default false bare")
-		assert_equal('tcp', p.proto, "default tcp proto")
-		assert_equal(false, p.server, "default false server")
-		assert_equal(false, p.ssl, "default false ssl")
-	end
-
-end

data/lib/rex/socket/range_walker.rb.ut.rb

@@ -1,56 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..'))
-
-require 'test/unit'
-require 'rex/socket/range_walker'
-
-class Rex::Socket::RangeWalker::UnitTest < Test::Unit::TestCase
-
-	Klass = Rex::Socket::RangeWalker
-
-	def test_walker
-	
-		#
-		# Single argument
-		#
-		s = Klass.new('10.0.0.0-10.0.0.255')
-
-		0.upto(255) { |x|
-			assert_equal('10.0.0.' + x.to_s, s.next_ip)
-		}
-		assert_nil(s.next_ip)
-
-		s.reset
-
-		0.upto(255) { |x|
-			assert_equal('10.0.0.' + x.to_s, s.next_ip)
-		}
-		assert_nil(s.next_ip)
-
-		#
-		
-
-		#
-		# Backwards
-		#
-		s = Klass.new('10.0.0.255-10.0.0.0')
-
-		0.upto(255) { |x|
-			assert_equal('10.0.0.' + x.to_s, s.next_ip)
-		}
-		assert_nil(s.next_ip)
-
-		#
-		# Same address
-		#
-		s = Klass.new('10.0.0.255-10.0.0.255')
-		assert_equal('10.0.0.255', s.next_ip)
-		assert_nil(s.next_ip)
-		
-
-		
-	end
-
-end

data/lib/rex/socket/ssl_tcp.rb.ut.rb

@@ -1,40 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..'))
-
-require 'test/unit'
-require 'rex/socket/ssl_tcp'
-
-class Rex::Socket::SslTcp::UnitTest < Test::Unit::TestCase
-
-	def test_ssltcp
-
-		# Create an SslTcp instance
-		t = nil
-		assert_nothing_raised {
-			t = Rex::Socket::SslTcp.create(
-				'PeerHost' => 'www.google.com',
-				'PeerPort' => 443)
-		}
-		assert_kind_of(Rex::Socket::SslTcp, t, "valid ssl tcp")
-
-		# Send a HEAD request and make sure we get some kind of response
-		head_request = "HEAD / HTTP/1.0\r\n\r\n"
-
-		assert_equal(19, t.put(head_request), "sending head request")
-
-		head_response = ""
-
-		assert_nothing_raised {
-			head_response = t.get(nil) || ""
-		}
-
-		assert_match(/^HTTP\/1./, head_response, "valid head response")
-
-		assert_nothing_raised {
-			t.close
-		}
-	end
-
-end

data/lib/rex/socket/ssl_tcp_server.rb.ut.rb

@@ -1,62 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..'))
-
-require 'test/unit'
-require 'rex/socket/ssl_tcp_server'
-require 'rex/socket/ssl_tcp'
-require 'rex/text'
-
-class Rex::Socket::SslTcpServer::UnitTest < Test::Unit::TestCase
-
-	# XXX.  The client data is sent & decrypted just fine.  The server data is not.  the client thread just spins.  BAH.
-	#
-	# As of 2011-03-04, works fine on 1.8.6-p399, 1.8.7-p330, 1.9.1-p378
-	#
-	def test_tcp_server
-		#return;
-
-		serv_port = 65433
-		c = nil
-
-		threads = []
-
-		# Server thread
-		threads << Thread.new() {
-			serv = Rex::Socket.create_tcp_server('LocalPort' => serv_port, 'SSL' => true)
-			assert_kind_of(Rex::Socket::SslTcpServer, serv, "type => ssl")
-			assert_kind_of(Rex::Socket::TcpServer, serv, "type => tcp")
-			assert_kind_of(Rex::IO::StreamServer, serv, "type => stream")
-			s = serv.accept
-			assert_equal("client_data\n", s.get_once(), "s: get_once")
-			assert_equal(3, s.write("Yo\n"), "s: put Yo")
-			# Make sure methods are Strings for 1.9 compat (which returns
-			# symbols)
-			meths = s.methods.map {|m| m.to_s}
-			assert(meths.include?("<<"), "Has <<")
-			assert(meths.include?(">>"), "Has >>")
-			assert(meths.include?("has_read_data?"), "Has has_read_data?")
-			serv.close
-		}
-
-		# Client thread
-		threads << Thread.new() {
-			sleep(2)
-			assert_nothing_raised {
-				c = Rex::Socket::SslTcp.create(
-				'PeerHost' => '127.0.0.1',
-				'PeerPort' => serv_port
-				)
-			}
-			assert_kind_of(Rex::Socket::Tcp, c, "TCP")
-			assert_kind_of(Rex::Socket::SslTcp, c, "SSL")
-			assert_equal(12, c.write("client_data\n"), "c: write")
-			assert_equal("Yo\n", c.get_once(), "c: get_once")
-			c.close if (c)
-		}
-
-		threads.each { |aThread|  aThread.join }
-	end
-
-end

data/lib/rex/socket/subnet_walker.rb.ut.rb

@@ -1,29 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..'))
-
-require 'test/unit'
-require 'rex/socket/subnet_walker'
-
-class Rex::Socket::SubnetWalker::UnitTest < Test::Unit::TestCase
-
-	Klass = Rex::Socket::SubnetWalker
-
-	def test_walker
-		s = Klass.new('10.0.0.0', '255.255.255.0')
-
-		0.upto(255) { |x|
-			assert_equal('10.0.0.' + x.to_s, s.next_ip)
-		}
-		assert_nil(s.next_ip)
-
-		s.reset
-
-		0.upto(255) { |x|
-			assert_equal('10.0.0.' + x.to_s, s.next_ip)
-		}
-		assert_nil(s.next_ip)
-	end
-
-end

data/lib/rex/socket/switch_board.rb.ut.rb

@@ -1,53 +0,0 @@
-#!/usr/bin/env ruby
-# -*- coding: binary -*-
-
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..'))
-
-require 'test/unit'
-require 'rex/socket/switch_board'
-
-class Rex::Socket::SwitchBoard::UnitTest < Test::Unit::TestCase
-
-	Klass = Rex::Socket::SwitchBoard
-
-	def test_add
-		Klass.flush_routes
-		assert_equal(true, Klass.add_route('0.0.0.0', 0, 'foo'))
-		assert_equal(false, Klass.add_route('0.0.0.0', 0, 'foo'))
-		assert_equal(1, Klass.routes.length)
-
-		assert_equal('0.0.0.0', Klass.routes[0].subnet)
-		assert_equal('0.0.0.0', Klass.routes[0].netmask)
-		assert_equal(0, Klass.routes[0].bitmask)
-		assert_equal('foo', Klass.routes[0].comm)
-	end
-
-	def test_remove
-		Klass.flush_routes
-		assert_equal(true, Klass.add_route('0.0.0.0', 0, 'foo'))
-		assert_equal(true, Klass.remove_route('0.0.0.0', 0, 'foo'))
-		assert_equal(false, Klass.remove_route('0.0.0.0', 0, 'foo'))
-		assert_equal(0, Klass.routes.length)
-	end
-
-	def test_best_comm
-		Klass.flush_routes
-		Klass.add_route('0.0.0.0', 0, 'default')
-		Klass.add_route('1.2.3.0', 24, 'spec')
-
-		assert_equal('default', Klass.best_comm('4.5.6.7'))
-		assert_equal('spec', Klass.best_comm('1.2.3.7'))
-	end
-
-	def test_remove_by_comm
-		Klass.flush_routes
-		Klass.add_route('1.2.3.0', 24, 'foo')
-		Klass.add_route('1.2.4.0', 24, 'dog')
-
-		Klass.remove_by_comm('foo')
-
-		assert_equal('dog', Klass.best_comm('1.2.4.7'))
-		assert_nil(Klass.best_comm('1.2.3.7'))
-	end
-
-end