librex 0.0.13 → 0.0.15

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

@@ -1,141 +0,0 @@
-#!/usr/bin/env ruby
-
-require 'rex/post/meterpreter/client'
-require 'rex/post/meterpreter/extensions/stdapi/constants'
-require 'rex/post/meterpreter/extensions/stdapi/sys/thread'
-
-module Rex
-module Post
-module Meterpreter
-module Extensions
-module Stdapi
-module Sys
-module ProcessSubsystem
-
-###
-#
-# Interfaces with a process' executing threads by enumerating,
-# opening, and creating threads.
-#
-###
-class Thread
-
-	##
-	#
-	# Constructor
-	#
-	##
-
-	#
-	# Initializes a thread instance that operates in the context of the
-	# supplied process instance.
-	#
-	def initialize(process)
-		self.process = process
-	end
-
-	##
-	#
-	# Process thread interaction
-	#
-	##
-
-	#
-	# Opens an existing thread that is running within the context
-	# of the process and returns a Sys::Thread instance.
-	#
-	def open(tid, access = THREAD_ALL)
-		request = Packet.create_request('stdapi_sys_process_thread_open')
-		real    = 0
-
-		# Translate access
-		if (access & THREAD_READ)
-			real |= THREAD_GET_CONTEXT | THREAD_QUERY_INFORMATION | SYNCHRONIZE
-		end
-
-		if (access & THREAD_WRITE)
-			real |= THREAD_SET_CONTEXT | THREAD_SET_INFORMATION | THREAD_SET_THREAD_TOKEN | THREAD_IMPERSONATE | THREAD_DIRECT_IMPERSONATION
-		end
-
-		if (access & THREAD_EXECUTE)
-			real |= THREAD_TERMINATE | THREAD_SUSPEND_RESUME | SYNCHRONIZE
-		end
-
-		# Add the thread identifier and permissions
-		request.add_tlv(TLV_TYPE_THREAD_ID, tid)
-		request.add_tlv(TLV_TYPE_THREAD_PERMS, real)
-
-		# Transmit the request
-		response = process.client.send_request(request)
-
-		# Create a thread class instance
-		return Rex::Post::Meterpreter::Extensions::Stdapi::Sys::Thread.new(
-				process, response.get_tlv_value(TLV_TYPE_THREAD_HANDLE), tid)
-	end
-
-	#
-	# Creates a new thread in the context of the process and
-	# returns a Sys::Thread instance.
-	#
-	def create(entry, parameter = nil, suspended = false)
-		request = Packet.create_request('stdapi_sys_process_thread_create')
-		creation_flags = 0
-
-		request.add_tlv(TLV_TYPE_PROCESS_HANDLE, process.handle)
-		request.add_tlv(TLV_TYPE_ENTRY_POINT, entry)
-
-		# Are we passing a parameter to the entry point of the thread?
-		if (parameter != nil)
-			request.add_tlv(TLV_TYPE_ENTRY_PARAMETER, parameter)
-		end
-
-		# Should we create the thread suspended?
-		if (suspended)
-			creation_flags |= CREATE_SUSPENDED
-		end
-
-		request.add_tlv(TLV_TYPE_CREATION_FLAGS, creation_flags)
-
-		# Transmit the request
-		response = process.client.send_request(request)	
-
-
-		thread_id     = response.get_tlv_value(TLV_TYPE_THREAD_ID)
-		thread_handle = response.get_tlv_value(TLV_TYPE_THREAD_HANDLE)
-
-		# Create a thread class instance
-		return Rex::Post::Meterpreter::Extensions::Stdapi::Sys::Thread.new(
-				process, thread_handle, thread_id)
-	end
-
-	#
-	# Enumerate through each thread identifier.
-	#
-	def each_thread(&block)
-		get_threads.each(&block)
-	end
-
-	#
-	# Returns an array of thread identifiers.
-	#
-	def get_threads
-		request = Packet.create_request('stdapi_sys_process_thread_get_threads')
-		threads = []
-
-		request.add_tlv(TLV_TYPE_PID, process.pid)
-
-		response = process.client.send_request(request)
-
-		response.each(TLV_TYPE_THREAD_ID) { |thr|
-			threads << thr.value
-		}
-
-		return threads
-	end
-
-protected
-	attr_accessor :process # :nodoc:
-
-end
-
-end; end; end; end; end; end; end

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

@@ -1,279 +0,0 @@
-#!/usr/bin/env ruby
-
-require 'rex/post/process'
-require 'rex/post/meterpreter/packet'
-require 'rex/post/meterpreter/client'
-require 'rex/post/meterpreter/extensions/stdapi/constants'
-require 'rex/post/meterpreter/extensions/stdapi/stdapi'
-require 'rex/post/meterpreter/extensions/stdapi/sys/registry_subsystem/registry_key'
-require 'rex/post/meterpreter/extensions/stdapi/sys/registry_subsystem/registry_value'
-
-module Rex
-module Post
-module Meterpreter
-module Extensions
-module Stdapi
-module Sys
-
-###
-#
-# This class provides access to the Windows registry on the remote
-# machine.
-#
-###
-class Registry
-
-	class << self
-		attr_accessor :client
-	end
-
-	##
-	#
-	# Registry key interaction
-	#
-	##
-
-	#
-	# Opens the supplied registry key relative to the root key with
-	# the supplied permissions.  Right now this is merely a wrapper around
-	# create_key.
-	#
-	def Registry.open_key(root_key, base_key, perm = KEY_READ)
-		# If no base key was provided, just return the root_key.
-		if (base_key == nil or base_key.length == 0)
-			return RegistrySubsystem::RegistryKey.new(client, root_key, base_key, perm, root_key)
-		end
-
-		return self.create_key(root_key, base_key, perm)
-	end
-
-	#
-	# Creates the supplied registry key or opens it if it already exists.
-	#
-	def Registry.create_key(root_key, base_key, perm = KEY_READ)
-		request = Packet.create_request('stdapi_registry_create_key')
-
-		request.add_tlv(TLV_TYPE_ROOT_KEY, root_key)
-		request.add_tlv(TLV_TYPE_BASE_KEY, base_key)
-		request.add_tlv(TLV_TYPE_PERMISSION, perm)
-
-		response = client.send_request(request)
-
-		return Rex::Post::Meterpreter::Extensions::Stdapi::Sys::RegistrySubsystem::RegistryKey.new(
-				client, root_key, base_key, perm, response.get_tlv(TLV_TYPE_HKEY).value)
-	end
-
-	#
-	# Deletes the supplied registry key.
-	#
-	def Registry.delete_key(root_key, base_key, recursive = true)
-		request = Packet.create_request('stdapi_registry_delete_key')
-		flags   = 0
-
-		if (recursive)
-			flags |= DELETE_KEY_FLAG_RECURSIVE
-		end
-
-		request.add_tlv(TLV_TYPE_ROOT_KEY, root_key)
-		request.add_tlv(TLV_TYPE_BASE_KEY, base_key)
-		request.add_tlv(TLV_TYPE_FLAGS, flags)
-
-		if (client.send_request(request) != nil)
-			return true
-		end
-
-		return false
-	end
-
-	#
-	# Closes the supplied registry key.
-	#
-	def Registry.close_key(hkey)
-		request = Packet.create_request('stdapi_registry_close_key')
-
-		request.add_tlv(TLV_TYPE_HKEY, hkey)
-
-		client.send_packet(request)
-
-		return true
-	end
-
-	#
-	# Enumerates the supplied registry key returning an array of key names.
-	#
-	def Registry.enum_key(hkey)
-		keys    = []
-		request = Packet.create_request('stdapi_registry_enum_key')
-
-		request.add_tlv(TLV_TYPE_HKEY, hkey)
-
-		response = client.send_request(request)
-
-		# Enumerate through all of the registry keys
-		response.each(TLV_TYPE_KEY_NAME) { |key_name|
-			keys << key_name.value
-		}
-
-		return keys
-	end
-
-	##
-	#
-	# Registry value interaction
-	#
-	##
-
-	#
-	# Sets the registry value relative to the supplied hkey.
-	#
-	def Registry.set_value(hkey, name, type, data)
-		request = Packet.create_request('stdapi_registry_set_value')
-
-		request.add_tlv(TLV_TYPE_HKEY, hkey)
-		request.add_tlv(TLV_TYPE_VALUE_NAME, name)
-		request.add_tlv(TLV_TYPE_VALUE_TYPE, type)
-
-		if (type == REG_SZ)
-			data += "\x00"
-		elsif (type == REG_DWORD)
-			data = [ data.to_i ].pack("V")
-		end
-
-		request.add_tlv(TLV_TYPE_VALUE_DATA, data)
-
-		response = client.send_request(request)
-
-		return true
-	end
-
-	#
-	# Queries the registry value supplied in name and returns an
-	# initialized RegistryValue instance if a match is found.
-	#
-	def Registry.query_value(hkey, name)
-		request = Packet.create_request('stdapi_registry_query_value')
-
-		request.add_tlv(TLV_TYPE_HKEY, hkey)
-		request.add_tlv(TLV_TYPE_VALUE_NAME, name)
-
-		response = client.send_request(request)
-
-		data = response.get_tlv(TLV_TYPE_VALUE_DATA).value;
-		type = response.get_tlv(TLV_TYPE_VALUE_TYPE).value;
-
-		if (type == REG_SZ)
-			data = data[0..-2]
-		elsif (type == REG_DWORD)
-			data = data.unpack("N")[0]
-		end
-
-		return Rex::Post::Meterpreter::Extensions::Stdapi::Sys::RegistrySubsystem::RegistryValue.new(
-				client, hkey, name, type, data)
-	end
-
-	#
-	# Deletes the registry value supplied in name from the supplied
-	# registry key.
-	#
-	def Registry.delete_value(hkey, name)
-		request = Packet.create_request('stdapi_registry_delete_value')
-
-		request.add_tlv(TLV_TYPE_HKEY, hkey)
-		request.add_tlv(TLV_TYPE_VALUE_NAME, name)
-
-		if (client.send_request(request) != nil)
-			return true
-		end
-
-		return false
-	end
-
-	#
-	# Queries the registry class name and returns a string
-	#
-	def Registry.query_class(hkey)
-		request = Packet.create_request('stdapi_registry_query_class')
-
-		request.add_tlv(TLV_TYPE_HKEY, hkey)
-
-		response = client.send_request(request)
-		cls = response.get_tlv(TLV_TYPE_VALUE_DATA)
-		return nil if not cls
-		data = cls.value.gsub(/\x00.*/, '')
-		return data
-	end
-
-	#
-	# Enumerates all of the values at the supplied hkey including their
-	# names.  An array of RegistryValue's is returned.
-	#
-	def Registry.enum_value(hkey)
-		request = Packet.create_request('stdapi_registry_enum_value')
-		values  = []
-
-		request.add_tlv(TLV_TYPE_HKEY, hkey)
-
-		response = client.send_request(request)
-
-		response.each(TLV_TYPE_VALUE_NAME) { |value_name|
-			values << Rex::Post::Meterpreter::Extensions::Stdapi::Sys::RegistrySubsystem::RegistryValue.new(
-					client, hkey, value_name.value)
-		}
-
-		return values
-	end
-
-	#
-	# Return the key value associated with the supplied string.  This is useful
-	# for converting HKLM as a string into its actual integer representation.
-	#
-	def self.key2str(key)
-		if (key == 'HKLM' or key == 'HKEY_LOCAL_MACHINE')
-			return HKEY_LOCAL_MACHINE
-		elsif (key == 'HKCU' or key == 'HKEY_CURRENT_USER')
-			return HKEY_CURRENT_USER
-		elsif (key == 'HKU' or key == 'HKEY_USERS')
-			return HKEY_USERS
-		elsif (key == 'HKCR' or key == 'HKEY_CLASSES_ROOT')
-			return HKEY_CLASSES_ROOT
-		elsif (key == 'HKEY_CURRENT_CONFIG')
-			return HKEY_CURRENT_CONFIG
-		elsif (key == 'HKEY_PERFORMANCE_DATA')
-			return HKEY_PERFORMANCE_DATA
-		elsif (key == 'HKEY_DYN_DATA')
-			return HKEY_DYN_DATA
-		else
-			raise ArgumentError, "Unknown key: #{key}"
-		end
-	end
-
-	#
-	# Returns the integer value associated with the supplied registry value
-	# type (like REG_SZ).
-	#
-	def self.type2str(type)
-		return REG_SZ if (type == 'REG_SZ')
-		return REG_DWORD if (type == 'REG_DWORD')
-		return REG_BINARY if (type == 'REG_BINARY')
-		return REG_EXPAND_SZ if (type == 'REG_EXPAND_SZ')
-		return REG_NONE if (type == 'REG_NONE')
-		return nil
-	end
-
-	#
-	# Split the supplied full registry key into its root key and base key.  For
-	# instance, passing HKLM\Software\Dog will return [ HKEY_LOCAL_MACHINE,
-	# 'Software\Dog' ]
-	#
-	def self.splitkey(str)
-		if (str =~ /^(.+?)\\(.*)$/)
-			[ key2str($1), $2 ]
-		else
-			[ key2str(str), nil ]
-		end
-	end
-
-end
-
-end; end; end; end; end; end
-

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

@@ -1,193 +0,0 @@
-#!/usr/bin/env ruby
-
-require 'rex/post/meterpreter/extensions/stdapi/constants'
-require 'rex/post/meterpreter/extensions/stdapi/sys/registry'
-
-module Rex
-module Post
-module Meterpreter
-module Extensions
-module Stdapi
-module Sys
-module RegistrySubsystem
-
-###
-#
-# Class wrapper around a logical registry key on the remote side
-#
-###
-class RegistryKey
-
-
-	#
-	# Initializes an instance of a registry key using the supplied properties
-	# and HKEY handle from the server.
-	#
-	def initialize(client, root_key, base_key, perm, hkey)
-		self.client   = client
-		self.root_key = root_key
-		self.base_key = base_key
-		self.perm     = perm
-		self.hkey     = hkey
-		
-		ObjectSpace.define_finalizer( self, self.class.finalize(self.client, self.hkey) )
-	end
-
-	def self.finalize(client,hkey)
-		proc { self.close(client,hkey) }
-	end
-
-	##
-	#
-	# Enumerators
-	#
-	##
-
-	#
-	# Enumerates all of the child keys within this registry key.
-	#
-	def each_key(&block)
-		return enum_key.each(&block)
-	end
-
-	#
-	# Enumerates all of the child values within this registry key.
-	#
-	def each_value(&block)
-		return enum_value.each(&block)
-	end
-
-	#
-	# Retrieves all of the registry keys that are direct descendents of
-	# the class' registry key.
-	#
-	def enum_key()
-		return self.client.sys.registry.enum_key(self.hkey)
-	end
-
-	#
-	# Retrieves all of the registry values that exist within the opened
-	# registry key.
-	#
-	def enum_value()
-		return self.client.sys.registry.enum_value(self.hkey)
-	end
-
-
-	##
-	#
-	# Registry key interaction
-	#
-	##
-
-	#
-	# Opens a registry key that is relative to this registry key.
-	#
-	def open_key(base_key, perm = KEY_READ)
-		return self.client.sys.registry.open_key(self.hkey, base_key, perm)
-	end
-
-	#
-	# Creates a registry key that is relative to this registry key.
-	#
-	def create_key(base_key, perm = KEY_READ)
-		return self.client.sys.registry.create_key(self.hkey, base_key, perm)
-	end
-
-	#
-	# Deletes a registry key that is relative to this registry key.
-	#
-	def delete_key(base_key, recursive = true)
-		return self.client.sys.registry.delete_key(self.hkey, base_key, recursive)
-	end
-
-	#
-	# Closes the open key.  This must be called if the registry
-	# key was opened.
-	#
-	def self.close(client, hkey)
-		if hkey != nil
-			return client.sys.registry.close_key(hkey)
-		end
-
-		return false	
-	end
-	
-	# Instance method for the same
-	def close()
-		self.class.close(self.client, self.hkey)
-	end
-
-	##
-	#
-	# Registry value interaction
-	#
-	##
-
-	#
-	# Sets a value relative to the opened registry key.
-	#
-	def set_value(name, type, data)
-		return self.client.sys.registry.set_value(self.hkey, name, type, data)
-	end
-
-	#
-	# Queries the attributes of the supplied registry value relative to
-	# the opened registry key.
-	#
-	def query_value(name)
-		return self.client.sys.registry.query_value(self.hkey, name)
-	end
-
-	#
-	# Queries the class of the specified key
-	#
-	def query_class
-		return self.client.sys.registry.query_class(self.hkey)
-	end
-
-	#
-	# Delete the supplied registry value.
-	#
-	def delete_value(name)
-		return self.client.sys.registry.delete_value(self.hkey, name)
-	end
-
-	##
-	#
-	# Serializers
-	#
-	##
-
-	#
-	# Returns the path to the key.
-	#
-	def to_s
-		return self.root_key.to_s + "\\" + self.base_key
-	end
-
-	#
-	# The open handle to the key on the server.
-	#
-	attr_reader   :hkey
-	#
-	# The root key name, such as HKEY_LOCAL_MACHINE.
-	#
-	attr_reader   :root_key
-	#
-	# The base key name, such as Software\Foo.
-	#
-	attr_reader   :base_key
-	#
-	# The permissions that the key was opened with.
-	#
-	attr_reader   :perm
-
-protected
-
-	attr_accessor :client # :nodoc:
-	attr_writer   :hkey, :root_key, :base_key, :perm # :nodoc:
-end
-
-end; end; end; end; end; end; end
-