librex 0.0.16 → 0.0.17

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

@@ -0,0 +1,129 @@
+#!/usr/bin/env ruby
+
+require 'rex/post/meterpreter/client'
+require 'rex/post/meterpreter/extensions/stdapi/constants'
+
+module Rex
+module Post
+module Meterpreter
+module Extensions
+module Stdapi
+module Sys
+module ProcessSubsystem
+
+###
+#
+# Interacts with loading, unloading, enumerating, and querying
+# image files in the context of a given process.
+#
+###
+class Image
+
+	##
+	#
+	# Constructor
+	#
+	##
+
+	#
+	# Initializes the image instance.
+	#
+	def initialize(process)
+		self.process = process
+	end
+
+	#
+	# Returns the image base address associated with the supplied image name.
+	#
+	def [](key)
+		each_image { |i|
+			if (i['name'].downcase == key.downcase)
+				return i['base']
+			end
+		}
+
+		return nil
+	end
+
+	#
+	# Loads an image file into the context of the process.
+	#
+	def load(image_path)
+		request = Packet.create_request('stdapi_sys_process_image_load')
+
+		request.add_tlv(TLV_TYPE_HANDLE, process.handle)
+		request.add_tlv(TLV_TYPE_IMAGE_FILE_PATH, image_path)
+
+		response = process.client.send_request(request)
+
+		return response.get_tlv_value(TLV_TYPE_IMAGE_BASE)
+	end
+
+	#
+	# Returns the address of the procedure that is found in the supplied
+	# library.
+	#
+	def get_procedure_address(image_file, procedure)
+		request = Packet.create_request('stdapi_sys_process_image_get_proc_address')
+
+		request.add_tlv(TLV_TYPE_HANDLE, process.handle)
+		request.add_tlv(TLV_TYPE_IMAGE_FILE, image_file)
+		request.add_tlv(TLV_TYPE_PROCEDURE_NAME, procedure)
+
+		response = process.client.send_request(request)
+
+		return response.get_tlv_value(TLV_TYPE_PROCEDURE_ADDRESS)
+	end
+
+	#
+	# Unloads an image file that is loaded into the address space of the
+	# process by its base address.
+	#
+	def unload(base)
+		request = Packet.create_request('stdapi_sys_process_image_unload')
+
+		request.add_tlv(TLV_TYPE_HANDLE, process.handle)
+		request.add_tlv(TLV_TYPE_IMAGE_BASE, base)
+
+		response = process.client.send_request(request)
+
+		return true
+	end
+
+	#
+	# Enumerates through each image in the process.
+	#
+	def each_image(&block)
+		get_images.each(&block)
+	end
+
+	#
+	# Returns an array of images in the process with hash objects that
+	# have keys for 'name', 'path', and 'base'.
+	#
+	def get_images
+		request = Packet.create_request('stdapi_sys_process_image_get_images')
+		images  = []
+
+		request.add_tlv(TLV_TYPE_HANDLE, process.handle)
+
+		response = process.client.send_request(request)
+
+		response.each(TLV_TYPE_IMAGE_GROUP) { |i|
+			images <<
+				{
+					'name' => i.get_tlv_value(TLV_TYPE_IMAGE_NAME),
+					'base' => i.get_tlv_value(TLV_TYPE_IMAGE_BASE),
+					'path' => i.get_tlv_value(TLV_TYPE_IMAGE_FILE_PATH)
+				}
+		}
+
+		return images
+	end
+
+protected
+	attr_accessor :process # :nodoc:
+
+end
+
+end; end; end; end; end; end; end

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

@@ -0,0 +1,55 @@
+#!/usr/bin/env ruby
+
+require 'rex/post/meterpreter/client'
+require 'rex/post/meterpreter/extensions/stdapi/constants'
+
+module Rex
+module Post
+module Meterpreter
+module Extensions
+module Stdapi
+module Sys
+module ProcessSubsystem
+
+###
+#
+# This class provides an input/output interface to an executed
+# process' standard input and output.
+#
+###
+class IO
+
+	##
+	#
+	# Constructor
+	#
+	##
+
+	#
+	# Initializes the IO instance.
+	#
+	def initialize(process)
+		self.process = process
+	end
+
+	#
+	# Writes the supplied buffer to the standard input handle of the
+	# executed process.
+	#
+	def write(buf)
+		return process.channel.write(buf)
+	end
+
+	#
+	# Reads data from the standard output handle of the executed process.
+	#
+	def read(length = nil)
+		return process.channel.read(length)
+	end
+
+protected
+	attr_accessor :process # :nodoc:
+
+end
+
+end; end; end; end; end; end; end

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

@@ -0,0 +1,336 @@
+#!/usr/bin/env ruby
+
+require 'rex/post/meterpreter/client'
+require 'rex/post/meterpreter/extensions/stdapi/constants'
+
+module Rex
+module Post
+module Meterpreter
+module Extensions
+module Stdapi
+module Sys
+module ProcessSubsystem
+
+###
+#
+# Provides an interface to allocate, free, read, write, query,
+# protect, lock, and unlock memory in the context of a given
+# process.
+#
+###
+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_WRITE            => PAGE_EXECUTE_READWRITE,
+		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_COW              => PAGE_WRITECOPY,
+		PROT_WRITE               => PAGE_READWRITE
+	}
+
+	##
+	#
+	# Constructor
+	#
+	##
+
+	#
+	# Initializes a memory modification instance with the supplied process
+	# instance.
+	#
+	def initialize(process)
+		self.process = process
+	end
+
+	#
+	# Allocate storage of the supplied length and returns the 
+	# address at which the memory was allocated.
+	#
+	def allocate(length, protection = nil, base = nil) 
+		allocation_type = MEM_COMMIT
+
+		# If no protection was supplied, default to the most flexible
+		if (protection == nil)
+			protection = PAGE_EXECUTE_READWRITE
+		else
+			protection = gen_prot_to_specific(protection)
+		end
+
+		# If the preferred base is non-nil, set the reserve flag
+		if (base != nil)
+			allocation_type |= MEM_RESERVE
+		end
+
+		return _allocate(base, length, allocation_type, protection)
+	end
+
+	#
+	# Low-level memory allocation.
+	#
+	def _allocate(base, length, allocation_type, protection)
+		request = Packet.create_request('stdapi_sys_process_memory_allocate')
+
+		# Populate the request
+		if (base != nil)
+			request.add_tlv(TLV_TYPE_BASE_ADDRESS, base)
+		end
+
+		request.add_tlv(TLV_TYPE_HANDLE, process.handle)
+		request.add_tlv(TLV_TYPE_LENGTH, length)
+		request.add_tlv(TLV_TYPE_ALLOCATION_TYPE, allocation_type)
+		request.add_tlv(TLV_TYPE_PROTECTION, protection)
+
+		# Transmit the request
+		response = process.client.send_request(request);
+
+		return response.get_tlv_value(TLV_TYPE_BASE_ADDRESS)
+	end
+
+	#
+	# Deallocate a region of memory in the context of a process.
+	#
+	def free(base, length = 0)
+		return _free(base, length)
+	end
+
+	#
+	# Low-level memory deallocation.
+	#
+	def _free(base, length)
+		request = Packet.create_request('stdapi_sys_process_memory_free')
+
+		request.add_tlv(TLV_TYPE_HANDLE, process.handle)
+		request.add_tlv(TLV_TYPE_BASE_ADDRESS, base)
+		request.add_tlv(TLV_TYPE_LENGTH, length)
+
+		response = process.client.send_request(request)
+
+		return true
+	end
+
+	#
+	# Read memory from the context of a process and return the buffer.
+	#
+	def read(base, length)
+		request = Packet.create_request('stdapi_sys_process_memory_read')
+
+		request.add_tlv(TLV_TYPE_HANDLE, process.handle)
+		request.add_tlv(TLV_TYPE_BASE_ADDRESS, base)
+		request.add_tlv(TLV_TYPE_LENGTH, length)
+
+		response = process.client.send_request(request)
+
+		return response.get_tlv_value(TLV_TYPE_PROCESS_MEMORY)
+	end
+
+	#
+	# Write memory to the context of a process and return the number of bytes
+	# actually written.
+	#
+	def write(base, data)
+		request = Packet.create_request('stdapi_sys_process_memory_write')
+
+		request.add_tlv(TLV_TYPE_HANDLE, process.handle)
+		request.add_tlv(TLV_TYPE_BASE_ADDRESS, base)
+		request.add_tlv(TLV_TYPE_PROCESS_MEMORY, data)
+
+		response = process.client.send_request(request)
+
+		return response.get_tlv_value(TLV_TYPE_LENGTH)
+	end
+
+	#
+	# Queries an address for information about its state.
+	#
+	def query(base)
+		request = Packet.create_request('stdapi_sys_process_memory_query')
+
+		request.add_tlv(TLV_TYPE_HANDLE, process.handle)
+		request.add_tlv(TLV_TYPE_BASE_ADDRESS, base)
+
+		response = process.client.send_request(request)
+
+		# Build out the hash from the response information
+		info = {}
+
+		info['BaseAddress']       = response.get_tlv_value(TLV_TYPE_BASE_ADDRESS)
+		info['AllocationBase']    = response.get_tlv_value(TLV_TYPE_ALLOC_BASE_ADDRESS)
+		info['AllocationProtect'] = specific_prot_to_gen(response.get_tlv_value(TLV_TYPE_ALLOC_PROTECTION))
+		info['RegionSize']        = response.get_tlv_value(TLV_TYPE_LENGTH)
+
+		# Translate the memory state
+		state = response.get_tlv_value(TLV_TYPE_MEMORY_STATE)
+
+		if (state == MEM_FREE)
+			info['Available'] = true
+		elsif (state == MEM_COMMIT)
+			info['Available'] = false
+		elsif (state == MEM_RESERVE)
+			info['Reserved'] = true
+		end
+
+		# Translate the region protections
+		info['Protect'] = specific_prot_to_gen(response.get_tlv_value(TLV_TYPE_PROTECTION))
+
+		# Translate the memory type
+		type = response.get_tlv_value(TLV_TYPE_MEMORY_TYPE)
+
+		if (type == MEM_IMAGE)
+			info['ImageMapping'] = true
+		elsif (type == MEM_MAPPED)
+			info['MemoryMapping'] = true
+		elsif (type == MEM_PRIVATE)
+			info['PrivateMapping'] = true
+		end
+
+		return info
+	end
+
+	#
+	# Change the protection masks on the region supplied in base.
+	#
+	def protect(base, length = nil, protection = nil)
+		request = Packet.create_request('stdapi_sys_process_memory_protect')
+
+		if (length == nil)
+			length = 4096
+		end
+
+		# If no protection was supplied, default to the most flexible
+		if (protection == nil)
+			protection = PAGE_EXECUTE_READWRITE
+		else
+			protection = gen_prot_to_specific(protection)
+		end
+
+		request.add_tlv(TLV_TYPE_HANDLE, process.handle)
+		request.add_tlv(TLV_TYPE_BASE_ADDRESS, base)
+		request.add_tlv(TLV_TYPE_LENGTH, length)
+		request.add_tlv(TLV_TYPE_PROTECTION, protection)
+
+		# Send the request
+		response = process.client.send_request(request)
+
+		# Return the old protection to the caller
+		return specific_prot_to_gen(response.get_tlv_value(TLV_TYPE_PROTECTION))
+	end
+
+	#
+	# 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.
+	#
+	def lock(base, length)
+		request = Packet.create_request('stdapi_sys_process_memory_lock')
+
+		request.add_tlv(TLV_TYPE_BASE_ADDRESS, base)
+		request.add_tlv(TLV_TYPE_LENGTH, length)
+
+		response = process.client.send_request(request)
+
+		return true
+	end
+
+	#
+	# 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.
+	#
+	def unlock(base, length)
+		request = Packet.create_request('stdapi_sys_process_memory_unlock')
+
+		request.add_tlv(TLV_TYPE_BASE_ADDRESS, base)
+		request.add_tlv(TLV_TYPE_LENGTH, length)
+
+		response = process.client.send_request(request)
+
+		return true
+	end
+
+
+	##
+	#
+	# Conditionals
+	#
+	##
+
+	#
+	# Check to see if an address is readable.
+	#
+	def readable?(base)
+		info = nil
+
+		begin
+			info = query(base)
+		rescue
+		end
+
+		if ((info != nil) &&
+		    (info['Available'] == false) &&
+		    (info['Protect'] & PROT_READ == PROT_READ))
+			return true
+		end
+
+		return false
+	end
+
+	#
+	# Check to see if an address is writable.
+	#
+	def writable?(base)
+		info = nil
+
+		begin
+			info = query(base)
+		rescue
+		end
+
+		if ((info != nil) &&
+		    (info['Available'] == false) &&
+		    (info['Protect'] & PROT_WRITE == PROT_WRITE))
+			return true
+		end
+
+		return false
+	end
+
+protected
+
+	#
+	# Translates general protection flags to specific protection flags.
+	#
+	def gen_prot_to_specific(prot)
+		if (prot == nil)
+			return PAGE_READ
+		end
+
+		return @@page_protection_map[prot]
+	end
+
+	#
+	# Translates specific protection flags to general protection flags.
+	#
+	def specific_prot_to_gen(prot)
+		
+		if (prot == nil)
+			return PAGE_READONLY
+		end
+
+		return @@page_protection_map.invert[prot]
+	end
+
+	attr_accessor :process # :nodoc:
+end
+
+end; end; end; end; end; end; end