librex 0.0.63 → 0.0.65

data/lib/rex/post/meterpreter/extensions/stdapi/railgun/railgun.rb.ut.rb

@@ -1,6 +1,6 @@
 #!/usr/bin/env ruby
 
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..','..','..','..','..', 'lib')) 
+$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..','..','..','..','..', 'lib'))
 
 require 'rex/post/meterpreter/extensions/stdapi/railgun/railgun'
 require 'rex/post/meterpreter/extensions/stdapi/railgun/mock_magic'
@@ -15,7 +15,7 @@ module Stdapi
 module Railgun
 class Railgun::UnitTest < Test::Unit::TestCase
 
-	# DLLs we know should be available at the time of this writing, 
+	# DLLs we know should be available at the time of this writing,
 	# and DLLs that because of changes since then should be available
 	STOCK_DLLS = [
 		'kernel32',
@@ -38,7 +38,7 @@ class Railgun::UnitTest < Test::Unit::TestCase
 
 		assert_equal(dll_names.length, dll_names.uniq.length,
 			"known_dll_names should not have duplicates")
-		
+
 		STOCK_DLLS.each do |name|
 			assert(dll_names.include?(name),
 				"known_dll_names should include #{name}")
@@ -89,7 +89,7 @@ class Railgun::UnitTest < Test::Unit::TestCase
 
 	def test_method_missing
 		railgun = Railgun.new(make_mock_client())
-		
+
 		STOCK_DLLS.each do |dll_name|
 			assert_nothing_raised do
 				railgun.send(dll_name.to_sym)
@@ -129,7 +129,7 @@ class Railgun::UnitTest < Test::Unit::TestCase
 
 			dll_name = func[:dll_name]
 			function_name = func[:name]
-			
+
 			railgun.add_dll(dll_name)
 			railgun.add_function(dll_name, function_name, func[:return_type], func[:params])
 

data/lib/rex/post/meterpreter/extensions/stdapi/railgun/util.rb

@@ -29,7 +29,7 @@ class  Util
 		:long_long => 8,
 		:float => 4,
 		:double => 8,
-		:long_double => 8,	
+		:long_double => 8,
 		:wchar_t => 2,
 	}
 
@@ -323,7 +323,7 @@ class  Util
 	#
 	def unpack_pointer(packed_pointer)
 		if is_64bit
-			# XXX: Only works if attacker and victim are like-endianed 
+			# XXX: Only works if attacker and victim are like-endianed
 			packed_pointer.unpack('Q')[0]
 		else
 			packed_pointer.unpack('V')[0]
@@ -343,7 +343,7 @@ class  Util
 		if pointer.class == String
 			pointer = unpack_pointer(pointer)
 		end
-		
+
 		return pointer.nil? || pointer == 0
 	end
 
@@ -360,13 +360,13 @@ class  Util
 			return ''
 		end
 
-		# If length not provided, use lstrlenW 
+		# If length not provided, use lstrlenW
 		if length.nil?
 			length = railgun.kernel32.lstrlenW(pointer)['return']
 		end
 
 		# Retrieve the array of characters
-		chars = read_array(:WCHAR, length, pointer) 
+		chars = read_array(:WCHAR, length, pointer)
 
 		# Concatenate the characters and convert to a ruby string
 		str = uniz_to_str(chars.join(''))
@@ -457,7 +457,7 @@ class  Util
 		offset = 0
 
 		1.upto(length).map do |n|
-			data = read_data(type, offset, buffer) 
+			data = read_data(type, offset, buffer)
 
 			offset = offset + size
 
@@ -524,11 +524,11 @@ class  Util
 
 		if TYPE_DEFINITIONS.has_key?(type)
 			primitive = TYPE_DEFINITIONS[type]
-	
+
 			if primitive == :pointer
 				return pointer_size
-			end		
-	
+			end
+
 			if PRIMITIVE_TYPE_SIZES.has_key?(primitive)
 				return PRIMITIVE_TYPE_SIZES[primitive]
 			else
@@ -551,7 +551,7 @@ class  Util
 	end
 
 	#
-	# Given a description of a data structure, returns an Array containing 
+	# Given a description of a data structure, returns an Array containing
 	# the offset from the beginning for each subsequent element, taking into
 	# consideration alignment and padding.
 	#
@@ -565,7 +565,7 @@ class  Util
 			if sizeof_type(data_type) > padding
 				offset = offset + padding
 			end
-	
+
 			offsets.push(offset)
 
 			offset = offset + sizeof_type(data_type)
@@ -575,7 +575,7 @@ class  Util
 
 		offsets
 	end
-	
+
 	# http://en.wikipedia.org/wiki/Data_structure_alignment
 	def required_alignment
 		is_64bit ? 8 : 4
@@ -605,13 +605,13 @@ class  Util
 		if type =~ /^(\w+)\[(\w+)\]$/
 			element_type = $1
 			length = $2
-	
+
 			unless length =~ /^\d+$/
 				length = railgun.const(length)
 			end
-	
+
 			return element_type, length
-		else	
+		else
 			raise "Can not split non-array type #{type}"
 		end
 	end

data/lib/rex/post/meterpreter/extensions/stdapi/railgun/win_const_manager.rb

@@ -38,10 +38,10 @@ class WinConstManager
 		@consts = {}
 
 		initial_consts.each_pair do |name, value|
-			add_const(name, value)			
+			add_const(name, value)
 		end
 
-		# Load utility 
+		# Load utility
 	end
 
 	def add_const(name, value)

data/lib/rex/post/meterpreter/extensions/stdapi/railgun/win_const_manager.rb.ut.rb

@@ -1,6 +1,6 @@
 #!/usr/bin/env ruby
 
-$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..','..','..','..','..', 'lib')) 
+$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..','..','..','..','..', 'lib'))
 
 require 'rex/post/meterpreter/extensions/stdapi/railgun/win_const_manager'
 require 'test/unit'
@@ -40,14 +40,14 @@ class WinConstManager::UnitTest < Test::Unit::TestCase
 	def test_add_const
 		target_key = 'VALID_KEY'
 		target_value = 23
-		
+
 		const_manager = WinConstManager.new
 
 		const_manager.add_const(target_key, target_value)
 
 		assert_equal(target_value, const_manager.parse(target_key),
 			"add_const should add a constant/value pair that can be trieved with parse")
-		
+
 	end
 
 	def test_initialization
@@ -73,7 +73,7 @@ class WinConstManager::UnitTest < Test::Unit::TestCase
 			"parse should retrieve the corresponding value when a key is provided")
 
 		# From API: "should not throw an exception given an invalid key"
-		assert_nothing_thrown do 
+		assert_nothing_thrown do
 			const_manager.parse(invalid_key)
 		end
 
@@ -83,7 +83,7 @@ class WinConstManager::UnitTest < Test::Unit::TestCase
 		x_key = 'X'
 		x_value = 228
 		y_key = 'Y'
-		y_value = 15 
+		y_value = 15
 
 		boolean_logic = x_key + ' | ' + y_key
 		target_boolean_logic_result = x_value | y_value

data/lib/rex/post/meterpreter/extensions/stdapi/stdapi.rb

@@ -43,7 +43,7 @@ class Stdapi < Extension
 		# can be directly referenced
 		client.register_extension_aliases(
 			[
-				{ 
+				{
 					'name' => 'fs',
 					'ext'  => ObjectAliases.new(
 						{
@@ -70,7 +70,7 @@ class Stdapi < Extension
 							'config'   => Rex::Post::Meterpreter::Extensions::Stdapi::Net::Config.new(client),
 							'socket'   => Rex::Post::Meterpreter::Extensions::Stdapi::Net::Socket.new(client)
 						})
-				},	
+				},
 				{
 					'name' => 'railgun',
 					'ext'  => Rex::Post::Meterpreter::Extensions::Stdapi::Railgun::Railgun.new(client)
@@ -146,4 +146,4 @@ class Stdapi < Extension
 	end
 end
 
-end; end; end; end; end
+end; end; end; end; end

data/lib/rex/post/meterpreter/extensions/stdapi/sys/event_log.rb

@@ -16,7 +16,7 @@ module Sys
 
 ###
 #
-# This class provides access to the Windows event log on the remote 
+# This class provides access to the Windows event log on the remote
 # machine.
 #
 ###
@@ -52,7 +52,7 @@ class EventLog
 	attr_accessor :handle # :nodoc:
 	attr_accessor :client # :nodoc:
 
-	public 
+	public
 
 	#
 	# Initializes an instance of the eventlog manipulator.
@@ -182,7 +182,7 @@ class EventLog
 		response = client.send_request(request, nil)
 		return nil
 	end
-	
+
 	# Instance method
 	def close
 		self.class.close(self.client, self.handle)

data/lib/rex/post/meterpreter/extensions/stdapi/sys/event_log_subsystem/event_record.rb

@@ -15,15 +15,15 @@ module EventLogSubsystem
 ###
 class EventRecord
 
-	attr_reader :num, :generated, :written, :eventid,
-	            :type, :category, :strings, :data
+	attr_reader :num, :generated, :written, :eventid
+	attr_reader :type, :category, :strings, :data
 
 	protected
 
-	attr_writer :num, :generated, :written, :eventid,
-	            :type, :category, :strings, :data
+	attr_writer :num, :generated, :written, :eventid
+	attr_writer :type, :category, :strings, :data
 
-	public 
+	public
 
 	def initialize(recnum, timegen, timewri, id, type, cat, strs, data)
 		self.num       = recnum
@@ -38,4 +38,4 @@ class EventRecord
 
 end
 
-end end end end end end end
+end end end end end end end

data/lib/rex/post/meterpreter/extensions/stdapi/sys/process_subsystem/memory.rb

@@ -22,17 +22,17 @@ class Memory
 
 	# Page protection translation hash
 	@@page_protection_map =
-	{ 
+	{
 		PROT_NONE                => PAGE_NOACCESS,
 		PROT_EXEC                => PAGE_EXECUTE,
 		PROT_EXEC | PROT_READ    => PAGE_EXECUTE_READ,
-		PROT_EXEC | PROT_READ | 
+		PROT_EXEC | PROT_READ |
 			PROT_WRITE            => PAGE_EXECUTE_READWRITE,
-		PROT_EXEC | PROT_READ | 
+		PROT_EXEC | PROT_READ |
 			PROT_WRITE | PROT_COW => PAGE_EXECUTE_WRITECOPY,
 		PROT_READ                => PAGE_READONLY,
 		PROT_READ | PROT_WRITE   => PAGE_READWRITE,
-		PROT_READ | PROT_WRITE | 
+		PROT_READ | PROT_WRITE |
 			PROT_COW              => PAGE_WRITECOPY,
 		PROT_WRITE               => PAGE_READWRITE
 	}
@@ -52,10 +52,10 @@ class Memory
 	end
 
 	#
-	# Allocate storage of the supplied length and returns the 
+	# Allocate storage of the supplied length and returns the
 	# address at which the memory was allocated.
 	#
-	def allocate(length, protection = nil, base = nil) 
+	def allocate(length, protection = nil, base = nil)
 		allocation_type = MEM_COMMIT
 
 		# If no protection was supplied, default to the most flexible
@@ -225,7 +225,7 @@ class Memory
 	end
 
 	#
-	# Lock a region of memory into physical memory so that it can't be 
+	# Lock a region of memory into physical memory so that it can't be
 	# swapped to disk.  This can only be done in the context of the
 	# process that is running the meterpreter server.  The instance's
 	# handle is ignored.
@@ -242,7 +242,7 @@ class Memory
 	end
 
 	#
-	# Unloock a region of memory into physical memory so that it can be 
+	# Unloock a region of memory into physical memory so that it can be
 	# swapped to disk.  This can only be done in the context of the
 	# process that is running the meterpreter server.  The instance's
 	# handle is ignored.
@@ -322,7 +322,7 @@ protected
 	# Translates specific protection flags to general protection flags.
 	#
 	def specific_prot_to_gen(prot)
-		
+
 		if (prot == nil)
 			return PAGE_READONLY
 		end
@@ -333,4 +333,4 @@ protected
 	attr_accessor :process # :nodoc:
 end
 
-end; end; end; end; end; end; end
+end; end; end; end; end; end; end

data/lib/rex/post/meterpreter/extensions/stdapi/sys/process_subsystem/thread.rb

@@ -97,7 +97,7 @@ class Thread
 		request.add_tlv(TLV_TYPE_CREATION_FLAGS, creation_flags)
 
 		# Transmit the request
-		response = process.client.send_request(request)	
+		response = process.client.send_request(request)
 
 
 		thread_id     = response.get_tlv_value(TLV_TYPE_THREAD_ID)
@@ -138,4 +138,4 @@ protected
 
 end
 
-end; end; end; end; end; end; end
+end; end; end; end; end; end; end

data/lib/rex/post/meterpreter/extensions/stdapi/sys/registry_subsystem/registry_key.rb

@@ -29,7 +29,7 @@ class RegistryKey
 		self.base_key = base_key
 		self.perm     = perm
 		self.hkey     = hkey
-		
+
 		ObjectSpace.define_finalizer( self, self.class.finalize(self.client, self.hkey) )
 	end
 
@@ -110,9 +110,9 @@ class RegistryKey
 			return client.sys.registry.close_key(hkey)
 		end
 
-		return false	
+		return false
 	end
-	
+
 	# Instance method for the same
 	def close()
 		self.class.close(self.client, self.hkey)

data/lib/rex/post/meterpreter/extensions/stdapi/sys/registry_subsystem/registry_value.rb

@@ -28,7 +28,7 @@ class RegistryValue
 		self.name   = name
 		self.type   = type
 		self.data   = data
-	end	
+	end
 
 	#
 	# Sets the value's data.
@@ -99,4 +99,4 @@ protected
 	attr_writer   :hkey, :name, :type, :data # :nodoc:
 end
 
-end; end; end; end; end; end; end
+end; end; end; end; end; end; end

data/lib/rex/post/meterpreter/extensions/stdapi/sys/registry_subsystem/remote_registry_key.rb

@@ -28,7 +28,7 @@ class RemoteRegistryKey
 		self.root_key = root_key
 		self.target_host = target_host
 		self.hkey     = hkey
-		
+
 		ObjectSpace.define_finalizer( self, self.class.finalize(self.client, self.hkey) )
 	end
 
@@ -109,9 +109,9 @@ class RemoteRegistryKey
 			return client.sys.registry.close_key(hkey)
 		end
 
-		return false	
+		return false
 	end
-	
+
 	# Instance method for the same
 	def close()
 		self.class.close(self.client, self.hkey)

data/lib/rex/post/meterpreter/extensions/stdapi/sys/thread.rb

@@ -13,7 +13,7 @@ module Sys
 
 ##
 #
-# This class implements the Rex::Post::Thread interface which 
+# This class implements the Rex::Post::Thread interface which
 # wrappers a logical thread for a given process.
 #
 ##
@@ -123,7 +123,7 @@ class Thread < Rex::Post::Thread
 		# Add all of the register that we're setting
 		regs_hash.each_key { |name|
 			t = request.add_tlv(TLV_TYPE_REGISTER)
-		
+
 			t.add_tlv(TLV_TYPE_REGISTER_NAME, name)
 			t.add_tlv(TLV_TYPE_REGISTER_VALUE_32, regs_hash[name])
 		}
@@ -139,13 +139,13 @@ class Thread < Rex::Post::Thread
 	def pretty_regs
 		regs = query_regs
 
-		buf  = sprintf("eax=%.8x ebx=%.8x ecx=%.8x edx=%.8x esi=%.8x edi=%.8x\n", 
+		buf  = sprintf("eax=%.8x ebx=%.8x ecx=%.8x edx=%.8x esi=%.8x edi=%.8x\n",
 		               regs['eax'], regs['ebx'], regs['ecx'], regs['edx'], regs['esi'], regs['edi'])
 		buf += sprintf("eip=%.8x esp=%.8x ebp=%.8x\n",
 		               regs['eip'], regs['esp'], regs['ebp'])
 		buf += sprintf("cs=%.4x ss=%.4x ds=%.4x es=%.4x fs=%.4x gs=%.4x\n",
 		               regs['cs'], regs['ss'], regs['ds'], regs['es'], regs['fs'], regs['gs'])
-	
+
 		return buf
 	end
 
@@ -165,14 +165,14 @@ class Thread < Rex::Post::Thread
 		handle = nil
 		return true
 	end
-	
+
 	# Instance method
 	def close
 		self.class.close(self.process.client, self.handle)
 	end
 
 	attr_reader :process, :handle, :tid # :nodoc:
-protected	
+protected
 	attr_writer :process, :handle, :tid # :nodoc:
 
 end