librex 0.0.18 → 0.0.19

data/lib/rex/encoding/xor/generic.rb.ut.rb

@@ -0,0 +1,120 @@
+#!/usr/bin/env ruby
+
+$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))
+
+require 'test/unit'
+require 'rex/encoding/xor/generic'
+
+module Rex::Encoding::Xor
+
+class Generic::UnitTest < ::Test::Unit::TestCase
+
+	def enc
+		Generic
+	end
+
+	def hook_static_encode(data, key, expected)
+		if enc.keysize != 0 && key.length != enc.keysize
+			assert_raise(::ArgumentError) { enc.encode(data,key) }
+		else
+			assert_equal(enc.encode(data, key), [ expected, key ])
+		end
+	end
+
+	def test_static_encode
+		# Test key of zero length
+		assert_raise(::ArgumentError) { enc.encode("\x00", "") }
+
+		# Test non-string key
+		assert_raise(::ArgumentError) { enc.encode("\x00\x01", 1) }
+
+		# some simple single byte tests with 0x00
+		30.times {
+			byte = rand(256).chr
+			hook_static_encode("\x00" * 3, byte, byte * 3)
+		}
+
+		# misc tests, see below
+		misc_tests.each { |test|
+			hook_static_encode(test[0], test[1], test[2])
+		}
+
+	end
+
+	#
+	# Misc (mostly) autogenerated tests, we need more with
+	# different keysizes!
+	#
+
+	def misc_tests
+		[
+
+		# a 3 byte key test
+		[
+		  "\x54\x9a\x04\x02\x8f",
+		  "\x6d\x4b\x3c",
+		  "9\3218o\304"
+		],
+
+		# a 4 byte key test
+		[
+		  "\312/\216e\265\301\323\026Y\315",
+		  "m2{b",
+		  "\247\035\365\a\330\363\250t4\377"
+		],
+
+		# randomly generated 2 byte key tests...
+		[
+		  "\x82\x3f\xb4\x77\x55\x16\x4a\x56\x87\xad\x5b\xf5",
+		  "\x33\xdb",
+		  "\xb1\xe4\x87\xac\x66\xcd\x79\x8d\xb4\x76\x68\x2e"
+		],
+		[
+		  "\x9c\xbd\xaa\x83\x8d\x7e\x76\xd9\x4b\xb2\x04\xd5\x2b\x58\x66",
+		  "\xda\x10",
+		  "\x46\xad\x70\x93\x57\x6e\xac\xc9\x91\xa2\xde\xc5\xf1\x48\xbc"
+		],
+		[
+		  "\x7f\x3b\xfb\x3b\xce\x8c\xe8\x3d\x65\x40\x2d\x5a\x19",
+		  "\x62\x28",
+		  "\x1d\x13\x99\x13\xac\xa4\x8a\x15\x07\x68\x4f\x72\x7b"
+		],
+		[
+		  "\xc8\xab\xa4\x56\xd5\xf0",
+		  "\x1a\xd0",
+		  "\xd2\x7b\xbe\x86\xcf\x20"
+		],
+		[
+		  "\xcc\x5a\x84\xe0\x6c\x00\x7a\x20\xa0\xc9",
+		  "\xe6\xb6",
+		  "\x2a\xec\x62\x56\x8a\xb6\x9c\x96\x46\x7f"
+		],
+		[
+		  "\x46\x96\x83\x1f\x6a\x79\xfe\xec\x24\xe0\xc3\x20\xe9\xa5\x3a\x76",
+		  "\x36\x5e",
+		  "\x70\xc8\xb5\x41\x5c\x27\xc8\xb2\x12\xbe\xf5\x7e\xdf\xfb\x0c\x28"
+		],
+		[
+		  "\x74\x7c\xe9\x21\x30\x33\xb3\xe6\x77\x9e\x07\xbc\x6c\xee\xc5\x06",
+		  "\x02\xa0",
+		  "\x76\xdc\xeb\x81\x32\x93\xb1\x46\x75\x3e\x05\x1c\x6e\x4e\xc7\xa6"
+		],
+		[
+		  "\x64\x8c\xc3\x41\x5d\xe5\x18\x36\xda\xc4\x86",
+		  "\xe3\xb9",
+		  "\x87\x35\x20\xf8\xbe\x5c\xfb\x8f\x39\x7d\x65"
+		],
+		[
+		  "\xdb\xbb\xb2\x7c\xda\x1f\xd6\xa5\x34\x00\xad",
+		  "\x20\xfc",
+		  "\xfb\x47\x92\x80\xfa\xe3\xf6\x59\x14\xfc\x8d"
+		],
+		[
+		  "\xc1\x2e\xfc\x7b\x98\x41\xec\xe3\x40\x98\x0b\xfd\x2c",
+		  "\x4a\xd7",
+		  "\x8b\xf9\xb6\xac\xd2\x96\xa6\x34\x0a\x4f\x41\x2a\x66"
+		]
+		]
+	end
+end
+end

data/lib/rex/encoding/xor/qword.rb

@@ -0,0 +1,15 @@
+#!/usr/bin/env ruby
+
+require 'rex/encoding/xor/generic'
+
+module Rex
+module Encoding
+module Xor
+
+class Qword < Generic
+
+	def Qword.keysize
+		8
+	end
+
+end end end end

data/lib/rex/encoding/xor/word.rb

@@ -0,0 +1,21 @@
+#!/usr/bin/env ruby
+
+require 'rex/encoding/xor/generic'
+
+#
+# Routine for xor encoding a buffer by a 2-byte (intel word) key.  The perl
+# version used to pad this buffer out to a 2-byte boundary, but I can't think
+# of a good reason to do that anymore, so this doesn't.
+#
+
+module Rex
+module Encoding
+module Xor
+
+class Word < Generic
+
+	def Word.keysize
+		2
+	end
+
+end end end end # Word/Xor/Encoding/Rex

data/lib/rex/encoding/xor/word.rb.ut.rb

@@ -0,0 +1,13 @@
+#!/usr/bin/env ruby
+
+$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))
+
+require 'rex/encoding/xor/word'
+require 'rex/encoding/xor/byte.rb.ut'
+
+class Rex::Encoding::Xor::Word::UnitTest < Rex::Encoding::Xor::Byte::UnitTest
+
+	def enc
+		Rex::Encoding::Xor::Word
+	end
+end

data/lib/rex/exceptions.rb

@@ -0,0 +1,275 @@
+#!/usr/bin/env ruby
+
+module Rex
+
+###
+#
+# Base mixin for all exceptions that can be thrown from inside Rex.
+#
+###
+module Exception
+end
+
+###
+#
+# This exception is raised when a timeout occurs.
+#
+###
+class TimeoutError < Interrupt
+	include Exception
+
+	def to_s
+		"Operation timed out."
+	end
+end
+
+###
+#
+# This exception is raised when a method is called or a feature is used that
+# is not implemented.
+#
+###
+class NotImplementedError < ::NotImplementedError
+	include Exception
+
+	def to_s
+		"The requested method is not implemented."
+	end
+end
+
+###
+#
+# This exception is raised when a generalized runtime error occurs.
+#
+###
+class RuntimeError < ::RuntimeError
+	include Exception
+end
+
+###
+#
+# This exception is raised when an invalid argument is supplied to a method.
+#
+###
+class ArgumentError < ::ArgumentError
+	include Exception
+
+	def initialize(message = nil)
+		@message = message
+	end
+
+	def to_s
+		str = 'An invalid argument was specified.'
+		if @message
+			str << " #{@message}"
+		end
+		str
+	end
+end
+
+###
+#
+# This exception is raised when an argument that was supplied to a method
+# could not be parsed correctly.
+#
+###
+class ArgumentParseError < ::ArgumentError
+	include Exception
+
+	def to_s
+		"The argument could not be parsed correctly."
+	end
+end
+
+###
+#
+# This exception is raised when an argument is ambiguous.
+#
+###
+class AmbiguousArgumentError < ::RuntimeError
+	include Exception
+
+	def initialize(name = nil)
+		@name = name
+	end
+
+	def to_s
+		"The name #{@name} is ambiguous."
+	end
+end
+
+###
+#
+# This error is thrown when a stream is detected as being closed.
+#
+###
+class StreamClosedError < ::IOError
+	include Exception
+
+	def initialize(stream)
+		@stream = stream
+	end
+
+	def stream
+		@stream
+	end
+
+	def to_s
+		"Stream #{@stream} is closed."
+	end
+end
+
+##
+#
+# Socket exceptions
+#
+##
+
+###
+#
+# This exception is raised when a general socket error occurs.
+#
+###
+module SocketError
+	include Exception
+
+	def to_s
+		"A socket error occurred."
+	end
+end
+
+###
+#
+# This exception is raised when there is some kind of error related to
+# communication with a host.
+#
+###
+module HostCommunicationError
+	def initialize(addr = nil, port = nil)
+		self.host = addr
+		self.port = port
+	end
+
+	#
+	# This method returns a printable address and optional port associated
+	# with the host that triggered the exception.
+	#
+	def addr_to_s
+		if host and port
+			"(#{host}:#{port})"
+		elsif host
+			"(#{host})"
+		else
+			""
+		end
+	end
+
+	attr_accessor :host, :port
+end
+
+
+###
+#
+# This exception is raised when a connection attempt fails because the remote
+# side refused the connection.
+#
+###
+
+class ConnectionError < ::IOError
+	include SocketError
+	include HostCommunicationError
+end
+
+###
+#
+# This exception is raised when a connection attempt fails because the remote
+# side refused the connection.
+#
+###
+class ConnectionRefused < ConnectionError
+	def to_s
+		"The connection was refused by the remote host #{addr_to_s}."
+	end
+end
+
+###
+#
+# This exception is raised when a connection attempt fails because the remote
+# side is unreachable.
+#
+###
+class HostUnreachable < ConnectionError
+	def to_s
+		"The host #{addr_to_s} was unreachable."
+	end
+end
+
+###
+#
+# This exception is raised when a connection attempt times out.
+#
+###
+class ConnectionTimeout < ConnectionError
+	def to_s
+		"The connection timed out #{addr_to_s}."
+	end
+end
+
+
+###
+#
+# This exception is raised when an attempt to use an address or port that is
+# already in use occurs, such as binding to a host on a given port that is
+# already in use.
+#
+###
+class AddressInUse < ::RuntimeError
+	include SocketError
+	include HostCommunicationError
+
+	def to_s
+		"The address is already in use #{addr_to_s}."
+	end
+end
+
+###
+#
+# This exception is raised when an unsupported internet protocol is specified.
+#
+###
+class UnsupportedProtocol < ::ArgumentError
+	include SocketError
+
+	def initialize(proto = nil)
+		self.proto = proto
+	end
+
+	def to_s
+		"The protocol #{proto} is not supported."
+	end
+
+	attr_accessor :proto
+end
+
+
+###
+#
+# This exception is raised when a proxy fails to pass a connection
+#
+###
+class ConnectionProxyError < ConnectionError
+	def initialize(host,port,ptype,reason)
+		super(host,port)
+		self.ptype = ptype
+		self.reason = reason
+	end
+
+	def to_s
+		self.ptype + ": " + self.reason
+	end
+
+	attr_accessor :ptype, :reason
+end
+
+end
+