librex 0.0.63 → 0.0.65

data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_netapi32.rb

@@ -24,7 +24,7 @@ class Def_netapi32
 
 		return dll
 	end
-	
+
 end
 
 end; end; end; end; end; end; end

data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_ntdll.rb

@@ -145,7 +145,7 @@ class Def_ntdll
 
 		return dll
 	end
-	
+
 end
 
 end; end; end; end; end; end; end

data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_shell32.rb

@@ -16,7 +16,7 @@ class Def_shell32
 
 		return dll
 	end
-	
+
 end
 
 end; end; end; end; end; end; end

data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_user32.rb

@@ -10,7 +10,7 @@ class Def_user32
 
 	def self.create_dll(dll_path = 'user32')
 		dll = DLL.new(dll_path, ApiConstants.manager)
-		
+
 		dll.add_function('ActivateKeyboardLayout', 'DWORD',[
 			["DWORD","hkl","in"],
 			["DWORD","Flags","in"],
@@ -3163,7 +3163,7 @@ class Def_user32
 
 		return dll
 	end
-	
+
 end
 
 end; end; end; end; end; end; end

data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_wlanapi.rb

@@ -11,18 +11,18 @@ class Def_wlanapi
 	def self.create_dll(dll_path = 'wlanapi')
 		dll = DLL.new(dll_path, ApiConstants.manager)
 
-		
+
 		dll.add_function( 'WlanOpenHandle', 'DWORD',[
 				['DWORD', 'dwClientVersion', 'in'],
 				['LPVOID', 'pReserved', 'in'],
 				['PDWORD', 'pdwNegotiatedVersion', 'out'],
 				['PDWORD', 'phClientHandle', 'out']])
-				
+
 		dll.add_function( 'WlanEnumInterfaces', 'DWORD',[
 				['DWORD', 'hClientHandle', 'in'],
 				['LPVOID', 'pReserved', 'in'],
 				['PDWORD', 'ppInterfaceList', 'out']])
-				
+
 		dll.add_function( 'WlanGetProfileList', 'DWORD',[
 				['DWORD', 'hClientHandle', 'in'],
 				['PBLOB', 'pInterfaceGuid', 'in'],
@@ -37,14 +37,14 @@ class Def_wlanapi
 				['PDWORD', 'pstrProfileXML', 'out'],
 				['PDWORD', 'pdwFlags', 'inout'],
 				['PDWORD', 'pdwGrantedAccess', 'out']])
-				
+
 		dll.add_function( 'WlanFreeMemory', 'DWORD',[
 				['LPVOID', 'pMemory', 'in']])
-				
+
 		dll.add_function( 'WlanCloseHandle', 'DWORD',[
 				['DWORD', 'hClientHandle', 'in'],
 				['LPVOID', 'pReserved', 'in']])
-				
+
 		dll.add_function( 'WlanQueryInterface', 'DWORD',[
 				['DWORD', 'hClientHandle', 'in'],
 				['PBLOB', 'pInterfaceGuid', 'in'],
@@ -53,14 +53,14 @@ class Def_wlanapi
 				['PDWORD', 'pdwDataSize', 'out'],
 				['PDWORD', 'ppData', 'out'],
 				['PDWORD', 'pWlanOpcodeValueType', 'out']])
-				
+
 		dll.add_function( 'WlanScan', 'DWORD',[
 				['DWORD', 'hClientHandle', 'in'],
 				['PBLOB', 'pInterfaceGuid', 'in'],
 				['PBLOB', 'pDot11Ssid', 'in'],
 				['PBLOB', 'pIeData', 'in'],
 				['LPVOID', 'pReserved', 'in']])
-				
+
 		dll.add_function( 'WlanGetNetworkBssList', 'DWORD',[
 				['DWORD', 'hClientHandle', 'in'],
 				['PBLOB', 'pInterfaceGuid', 'in'],
@@ -69,16 +69,16 @@ class Def_wlanapi
 				['BOOL', 'bSecurityEnabled', 'in'],
 				['LPVOID', 'pReserved', 'in'],
 				['PDWORD', 'ppWlanBssList', 'out']])
-		
+
 		dll.add_function( 'WlanDisconnect', 'DWORD',[
 				['DWORD', 'hClientHandle', 'in'],
 				['PBLOB', 'pInterfaceGuid', 'in'],
 				['LPVOID', 'pReserved', 'in']])
-		
-		
+
+
 		return dll
 	end
-	
+
 end
 
 end; end; end; end; end; end; end

data/lib/rex/post/meterpreter/extensions/stdapi/railgun/def/def_ws2_32.rb

@@ -598,7 +598,7 @@ class Def_ws2_32
 
 		return dll
 	end
-	
+
 end
 
 end; end; end; end; end; end; end

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

@@ -54,10 +54,10 @@ class DLL
 		self.functions = {}
 	end
 
-	def known_function_names 
+	def known_function_names
 		return functions.keys
 	end
-	
+
 	def get_function(name)
 		return functions[name]
 	end
@@ -74,7 +74,7 @@ class DLL
 	def call_function(func_symbol, args, client)
 		func_name = func_symbol.to_s
 
-		unless known_function_names.include? func_name 
+		unless known_function_names.include? func_name
 			raise "DLL-function #{func_name} not found. Known functions: #{PP.pp(known_function_names, '')}"
 		end
 

data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll.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/dll'
 require 'rex/post/meterpreter/extensions/stdapi/railgun/mock_magic'

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

@@ -50,7 +50,7 @@ class DLLFunction
 	@@directions = ["in", "out", "inout", "return"].freeze
 
 	attr_reader :return_type,  :params, :windows_name
-	
+
 	def initialize(return_type, params, windows_name)
 		check_return_type(return_type) # we do error checking as early as possible so the library is easier to use
 		check_params(params)
@@ -58,9 +58,9 @@ class DLLFunction
 		@params = params
 		@windows_name = windows_name
 	end
-	
+
 	private
-	
+
 	def check_type_exists (type)
 		if not @@allowed_datatypes.has_key?(type)
 			raise ArgumentError, "Type unknown: #{type}. Allowed types: #{PP.pp(@@allowed_datatypes.keys, "")}"
@@ -85,16 +85,16 @@ class DLLFunction
 
 			# Only our set of predefined directions are valid
 			unless @@directions.include?(direction)
-				raise ArgumentError, "invalid direction: #{direction}" 
+				raise ArgumentError, "invalid direction: #{direction}"
 			end
 
 			# 'return' is not a valid direction in this context
 			unless direction != "return"
-				raise "direction 'return' is only for the return value of the function." 
+				raise "direction 'return' is only for the return value of the function."
 			end
 		end
 	end
-	
+
 end
 
 end; end; end; end; end; end

data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll_function.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/dll_function'
 require 'test/unit'
@@ -31,7 +31,7 @@ class DLLFunction::UnitTest < Test::Unit::TestCase
 #
 #		assert_raised(ArgumentError, "check_type_exists should raise ArgumentError on unknown datatypes") do
 #		end
-	
+
 	end
 end
 end

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

@@ -32,7 +32,7 @@ module Railgun
 #
 # shared functions
 #
-# 
+#
 module DLLHelper
 
 	# converts ruby string to zero-terminated ASCII string
@@ -133,7 +133,7 @@ module DLLHelper
 					blob += buffer
 					# sf: force 8 byte alignment to satisfy x64, wont matter on x86.
 					while( blob.length % 8 != 0 )
-						blob += "\x00" 
+						blob += "\x00"
 					end
 					#puts "   heap blob size now #{blob.length}"
 				end
@@ -142,7 +142,7 @@ module DLLHelper
 		#puts "  built buffer: #{direction}"
 		return [layout, blob]
 	end
-	
+
 end
 
-end; end; end; end; end; end; 
+end; end; end; end; end; end;

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

@@ -16,8 +16,8 @@ module Railgun
 class DLLHelper::UnitTest < Test::Unit::TestCase
 
 	###
-	# We will test against this instance of DLLHelper (a module) 
-	# 
+	# We will test against this instance of DLLHelper (a module)
+	#
 	# We freeze the instance and make the reference constant to ensure consistency
 	##
 	TEST_DLL_HELPER = Object.new.extend(DLLHelper).freeze
@@ -28,13 +28,13 @@ class DLLHelper::UnitTest < Test::Unit::TestCase
 		# converts ruby string to zero-terminated ASCII string
 		zero_terminated_ascii_attempt = TEST_DLL_HELPER.str_to_ascii_z(original_string)
 
-		assert(zero_terminated_ascii_attempt  =~ /\x00$/, 
+		assert(zero_terminated_ascii_attempt  =~ /\x00$/,
 			"str_to_ascii_z should result in a 0 terminated string")
 
-		assert(zero_terminated_ascii_attempt =~ /^#{original_string}/, 
+		assert(zero_terminated_ascii_attempt =~ /^#{original_string}/,
 			"str_to_ascii_z should still start with original string")
 
-		assert_equal(original_string.length + 1, zero_terminated_ascii_attempt.length, 
+		assert_equal(original_string.length + 1, zero_terminated_ascii_attempt.length,
 			"str_to_ascii_z should have length of original pluss room for a terminal 0")
 	end
 
@@ -83,7 +83,7 @@ class DLLHelper::UnitTest < Test::Unit::TestCase
 	end
 
 	def test_assemble_buffer
-		# TODO: provide test coverage 
+		# TODO: provide test coverage
 		#skip("Currently DLLHelper.assemble_buffer does not have coverage")
 	end
 
@@ -95,7 +95,7 @@ class DLLHelper::UnitTest < Test::Unit::TestCase
 
 		y_key = 'Y'
 		y_value = 5
-		
+
 		logical_or = x_key +  '|' +  y_key
 		target_result_of_logical_or = x_value | y_value
 

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

@@ -6,7 +6,7 @@ module Stdapi
 module Railgun
 class DLLWrapper
 	attr_reader :_client, :_dll
-	
+
 	def initialize(dll, client)
 		@_dll    = dll
 		@_client = client
@@ -15,7 +15,7 @@ class DLLWrapper
 	# For backwards compatability. People check if functions are added this way
 	# XXX: Depricate this
 	def functions
-		# warn 'Depricated.' 
+		# warn 'Depricated.'
 		_dll.functions
 	end
 

data/lib/rex/post/meterpreter/extensions/stdapi/railgun/dll_wrapper.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/dll'
 require 'rex/post/meterpreter/extensions/stdapi/railgun/dll_wrapper'

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

@@ -10,13 +10,13 @@ require 'rex/post/meterpreter/extensions/stdapi/railgun/tlv'
 #
 # This mixin serves as a means of providing common mock objects and utilities
 # relevant to railgun until a better home is decided upon
-# 
+#
 module MockMagic
-	
+
 	TLV_TYPE_NAMES = {
 		TLV_TYPE_RAILGUN_SIZE_OUT => "TLV_TYPE_RAILGUN_SIZE_OUT",
 		TLV_TYPE_RAILGUN_STACKBLOB => "TLV_TYPE_RAILGUN_STACKBLOB",
-		TLV_TYPE_RAILGUN_BUFFERBLOB_IN => "TLV_TYPE_RAILGUN_BUFFERBLOB_IN", 
+		TLV_TYPE_RAILGUN_BUFFERBLOB_IN => "TLV_TYPE_RAILGUN_BUFFERBLOB_IN",
 		TLV_TYPE_RAILGUN_BUFFERBLOB_INOUT => "TLV_TYPE_RAILGUN_BUFFERBLOB_INOUT",
 		TLV_TYPE_RAILGUN_DLLNAME => "TLV_TYPE_RAILGUN_DLLNAME",
 		TLV_TYPE_RAILGUN_FUNCNAME => "TLV_TYPE_RAILGUN_FUNCNAME",
@@ -28,7 +28,7 @@ module MockMagic
 		def initialize(platform, response_tlvs, check_request)
 			@check_request = check_request
 			@response_tlvs = response_tlvs
-			@platform = platform 
+			@platform = platform
 		end
 
 		def send_request(request)
@@ -511,4 +511,4 @@ module MockMagic
 
 end
 
-end; end; end; end; end; end; 
+end; end; end; end; end; end;

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

@@ -40,7 +40,7 @@ module Railgun
 class MultiCaller
 
 		include DLLHelper
-		
+
 		def initialize( client, parent )
 			@parent = parent
 			@client = client
@@ -102,7 +102,7 @@ class MultiCaller
 							args[param_idx] = 8
 							buffer_size = args[param_idx]
 						end
-						
+
 						if( @native == 'Q' )
 							raise "Please pass 8 for 'out' PDWORDS, since they require a buffer of size 8" unless buffer_size == 8
 						elsif( @native == 'V' )
@@ -299,11 +299,11 @@ class MultiCaller
 			function_results
 		end
 		# process_multi_function_call
-	
+
 	protected
-	
+
 	attr_accessor :win_consts
-	
+
 end # MultiCall
-	
+
 end; end; end; end; end; end

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

@@ -81,7 +81,7 @@ class Railgun
 	##
 	# Returns a Hash containing DLLs added to this instance with #add_dll
 	# as well as references to any frozen cached dlls added directly in #get_dll
-	# and copies of any frozen dlls (added directly with #add_function) 
+	# and copies of any frozen dlls (added directly with #add_function)
 	# that the user attempted to modify with #add_function.
 	#
 	# Keys are friendly DLL names and values are the corresponding DLL instance
@@ -98,7 +98,7 @@ class Railgun
 
 	# if you are going to touch @@cached_dlls, wear protection
 	@@cache_semaphore = Mutex.new
-	
+
 	def initialize(client)
 		self.client = client
 		self.dlls = {}
@@ -129,11 +129,11 @@ class Railgun
 	# LPVOID parameters)
 	#
 	def memread(address, length)
-	
+
 		raise "Invalid parameters." if(not address or not length)
-		
+
 		request = Packet.create_request('stdapi_railgun_memread')
-		
+
 		request.add_tlv(TLV_TYPE_RAILGUN_MEM_ADDRESS, address)
 		request.add_tlv(TLV_TYPE_RAILGUN_MEM_LENGTH, length)
 
@@ -141,20 +141,20 @@ class Railgun
 		if(response.result == 0)
 			return response.get_tlv_value(TLV_TYPE_RAILGUN_MEM_DATA)
 		end
-		
+
 		return nil
 	end
-	
+
 	#
 	# Write data to a memory address on the host (useful for working with
 	# LPVOID parameters)
 	#
 	def memwrite(address, data, length)
-	
+
 		raise "Invalid parameters." if(not address or not data or not length)
-		
+
 		request = Packet.create_request('stdapi_railgun_memwrite')
-		
+
 		request.add_tlv(TLV_TYPE_RAILGUN_MEM_ADDRESS, address)
 		request.add_tlv(TLV_TYPE_RAILGUN_MEM_DATA, data)
 		request.add_tlv(TLV_TYPE_RAILGUN_MEM_LENGTH, length)
@@ -163,16 +163,16 @@ class Railgun
 		if(response.result == 0)
 			return true
 		end
-		
+
 		return false
 	end
-	
+
 	#
 	# Adds a function to an existing DLL definition.
 	#
 	# If the DLL definition is frozen (ideally this should be the case for all
 	# cached dlls) an unfrozen copy is created and used henceforth for this
-	# instance. 
+	# instance.
 	#
 	def add_function(dll_name, function_name, return_type, params, windows_name=nil)
 
@@ -182,7 +182,7 @@ class Railgun
 
 		dll = get_dll(dll_name)
 
-		# For backwards compatibility, we ensure the dll is thawed 
+		# For backwards compatibility, we ensure the dll is thawed
 		if dll.frozen?
 			# dup will copy values, but not the frozen status
 			dll = dll.dup
@@ -213,7 +213,7 @@ class Railgun
 		dlls[dll_name] = DLL.new(windows_name, constant_manager)
 	end
 
-	
+
 	def known_dll_names
 		return BUILTIN_DLLS | dlls.keys
 	end
@@ -238,9 +238,9 @@ class Railgun
 					if dll_name !~ /^\w+$/
 						raise "DLL name #{dll_name} is bad. Correct Railgun::BUILTIN_DLLS"
 					end
- 
-					require 'rex/post/meterpreter/extensions/stdapi/railgun/def/def_' << dll_name 
-                                        dll = Def.const_get('Def_' << dll_name).create_dll.freeze
+
+					require 'rex/post/meterpreter/extensions/stdapi/railgun/def/def_' << dll_name
+					dll = Def.const_get('Def_' << dll_name).create_dll.freeze
 
 					@@cached_dlls[dll_name] = dll
 					dlls[dll_name] = dll
@@ -251,7 +251,7 @@ class Railgun
 
 		return dlls[dll_name]
 	end
-	
+
 	#
 	# Fake having members like user32 and kernel32.
 	# reason is that
@@ -265,7 +265,7 @@ class Railgun
 		unless known_dll_names.include? dll_name
 			raise "DLL #{dll_name} not found. Known DLLs: #{PP.pp(known_dll_names, '')}"
 		end
-		
+
 		dll = get_dll(dll_name)
 
 		return DLLWrapper.new(dll, client)