librex 0.0.68 → 0.0.70

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

@@ -1,4 +1,3 @@
-#!/usr/bin/env ruby
 # -*- coding: binary -*-
 
 require 'rex/post/meterpreter/extensions/stdapi/constants'
@@ -20,174 +19,174 @@ module RegistrySubsystem
 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
+  #
+  # 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:
+  attr_accessor :client # :nodoc:
+  attr_writer   :hkey, :root_key, :base_key, :perm # :nodoc:
 end
 
 end; end; end; end; end; end; end

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

@@ -1,4 +1,3 @@
-#!/usr/bin/env ruby
 # -*- coding: binary -*-
 
 require 'rex/post/meterpreter/extensions/stdapi/constants'
@@ -19,85 +18,85 @@ module RegistrySubsystem
 ###
 class RegistryValue
 
-	#
-	# Initializes a registry value instance that's associated with the supplied
-	# server key handle.
-	#
-	def initialize(client, hkey, name, type = nil, data = nil)
-		self.client = client
-		self.hkey   = hkey
-		self.name   = name
-		self.type   = type
-		self.data   = data
-	end
+  #
+  # Initializes a registry value instance that's associated with the supplied
+  # server key handle.
+  #
+  def initialize(client, hkey, name, type = nil, data = nil)
+    self.client = client
+    self.hkey   = hkey
+    self.name   = name
+    self.type   = type
+    self.data   = data
+  end
 
-	#
-	# Sets the value's data.
-	#
-	def set(data, type = nil)
-		if (type == nil)
-			type = self.type
-		end
-		if (self.client.sys.registry.set_value(self.hkey, self.name,
-				type, data))
-			self.data = data
-			self.type = type
+  #
+  # Sets the value's data.
+  #
+  def set(data, type = nil)
+    if (type == nil)
+      type = self.type
+    end
+    if (self.client.sys.registry.set_value(self.hkey, self.name,
+        type, data))
+      self.data = data
+      self.type = type
 
-			return true
-		end
+      return true
+    end
 
-		return false
-	end
+    return false
+  end
 
-	#
-	# Queries the value's data.
-	#
-	def query()
-		val =  self.client.sys.registry.query_value(self.hkey, self.name)
+  #
+  # Queries the value's data.
+  #
+  def query()
+    val =  self.client.sys.registry.query_value(self.hkey, self.name)
 
-		if (val != nil)
-			self.data = val.data
-			self.type = val.type
-		end
+    if (val != nil)
+      self.data = val.data
+      self.type = val.type
+    end
 
-		return self.data
-	end
+    return self.data
+  end
 
-	#
-	# Deletes the value.
-	#
-	def delete()
-		return self.client.sys.registry.delete_value(self.hkey, self.name)
-	end
+  #
+  # Deletes the value.
+  #
+  def delete()
+    return self.client.sys.registry.delete_value(self.hkey, self.name)
+  end
 
-	def type_to_s
-		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
+  def type_to_s
+    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
 
-	#
-	# The remote server key handle.
-	#
-	attr_reader   :hkey
-	#
-	# The name of the registry value.
-	#
-	attr_reader   :name
-	#
-	# The type of data represented by the registry value.
-	#
-	attr_reader   :type
-	#
-	# The arbitrary data stored within the value, if any.
-	#
-	attr_reader   :data
+  #
+  # The remote server key handle.
+  #
+  attr_reader   :hkey
+  #
+  # The name of the registry value.
+  #
+  attr_reader   :name
+  #
+  # The type of data represented by the registry value.
+  #
+  attr_reader   :type
+  #
+  # The arbitrary data stored within the value, if any.
+  #
+  attr_reader   :data
 protected
-	attr_accessor :client # :nodoc:
-	attr_writer   :hkey, :name, :type, :data # :nodoc:
+  attr_accessor :client # :nodoc:
+  attr_writer   :hkey, :name, :type, :data # :nodoc:
 end
 
 end; end; end; end; end; end; end