librex 0.0.68 → 0.0.70

data/lib/rex/post/meterpreter/channel_container.rb

@@ -1,4 +1,3 @@
-#!/usr/bin/env ruby
 # -*- coding: binary -*-
 
 module Rex
@@ -13,42 +12,42 @@ module Meterpreter
 ###
 module ChannelContainer
 
-	#
-	# Initializes the channel association hash
-	#
-	def initialize_channels
-		self.channels = {}
-	end
-
-	#
-	# Adds a channel to the container that is indexed by its channel identifier
-	#
-	def add_channel(channel)
-		self.channels[channel.cid] = channel
-	end
-
-	#
-	# Looks up a channel instance based on its channel identifier
-	#
-	def find_channel(cid)
-		return self.channels[cid]
-	end
-
-	#
-	# Removes a channel based on its channel identifier
-	#
-	def remove_channel(cid)
-		return self.channels.delete(cid)
-	end
-
-	#
-	# The hash of channels.
-	#
-	attr_reader :channels
+  #
+  # Initializes the channel association hash
+  #
+  def initialize_channels
+    self.channels = {}
+  end
+
+  #
+  # Adds a channel to the container that is indexed by its channel identifier
+  #
+  def add_channel(channel)
+    self.channels[channel.cid] = channel
+  end
+
+  #
+  # Looks up a channel instance based on its channel identifier
+  #
+  def find_channel(cid)
+    return self.channels[cid]
+  end
+
+  #
+  # Removes a channel based on its channel identifier
+  #
+  def remove_channel(cid)
+    return self.channels.delete(cid)
+  end
+
+  #
+  # The hash of channels.
+  #
+  attr_reader :channels
 
 protected
 
-	attr_writer :channels # :nodoc:
+  attr_writer :channels # :nodoc:
 
 end
 

data/lib/rex/post/meterpreter/channels/pool.rb

@@ -1,4 +1,3 @@
-#!/usr/bin/env ruby
 # -*- coding: binary -*-
 
 require 'rex/post/meterpreter/channel'
@@ -23,137 +22,137 @@ module Channels
 ###
 class Pool < Rex::Post::Meterpreter::Channel
 
-	class << self
-		def cls
-			return CHANNEL_CLASS_POOL
-		end
-	end
-
-	##
-	#
-	# Constructor
-	#
-	##
-
-	#
-	# Passes the initialization information up to the base class
-	#
-	def initialize(client, cid, type, flags)
-		super(client, cid, type, flags)
-	end
-
-	##
-	#
-	# Channel interaction
-	#
-	##
-
-	#
-	# Checks to see if the EOF flag has been set on the pool.
-	#
-	def eof
-		request = Packet.create_request('core_channel_eof')
-
-		request.add_tlv(TLV_TYPE_CHANNEL_ID, self.cid)
-
-		begin
-			response = self.client.send_request(request)
-		rescue
-			return true
-		end
-
-		if (response.has_tlv?(TLV_TYPE_BOOL))
-			return response.get_tlv_value(TLV_TYPE_BOOL)
-		end
-
-		return false
-	end
-
-	#
-	# Reads data from the remote side of the pool and raises EOFError if the
-	# pool has been reached EOF.
-	#
-	def read(length = nil)
-		begin
-			data = super(length)
-		rescue
-			data = nil
-		end
-
-		if (((data == nil) || (data.length == 0)) &&
-		    (self.eof))
-			raise EOFError
-		end
-
-		return data
-	end
-
-	#
-	# This method seeks to an offset within the remote side of the pool using
-	# the standard seek whence clauses.
-	#
-	def seek(offset, whence = SEEK_SET)
-		sane = 0
-
-		# Just in case...
-		case whence
-			when ::IO::SEEK_SET
-				sane = 0
-			when ::IO::SEEK_CUR
-				sane = 1
-			when ::IO::SEEK_END
-				sane = 2
-			else
-				raise RuntimeError, "Invalid seek whence #{whence}.", caller
-		end
-
-		request = Packet.create_request('core_channel_seek')
-
-		request.add_tlv(TLV_TYPE_CHANNEL_ID, self.cid)
-		request.add_tlv(TLV_TYPE_SEEK_OFFSET, offset)
-		request.add_tlv(TLV_TYPE_SEEK_WHENCE, sane)
-
-		begin
-			response = self.client.send_request(request)
-		rescue
-			return -1
-		end
-
-		return tell
-	end
-
-	#
-	# Synonym for tell.
-	#
-	def pos
-		return tell
-	end
-
-	#
-	# This method returns the current file pointer position to the caller.
-	#
-	def tell
-		request = Packet.create_request('core_channel_tell')
-		pos     = -1
-
-		request.add_tlv(TLV_TYPE_CHANNEL_ID, self.cid)
-
-		begin
-			response = self.client.send_request(request)
-		rescue
-			return pos
-		end
-
-		# Set the return value to the position that we're at
-		if (response.has_tlv?(TLV_TYPE_SEEK_POS))
-			pos = response.get_tlv_value(TLV_TYPE_SEEK_POS)
-		end
-
-		return pos
-	end
+  class << self
+    def cls
+      return CHANNEL_CLASS_POOL
+    end
+  end
+
+  ##
+  #
+  # Constructor
+  #
+  ##
+
+  #
+  # Passes the initialization information up to the base class
+  #
+  def initialize(client, cid, type, flags)
+    super(client, cid, type, flags)
+  end
+
+  ##
+  #
+  # Channel interaction
+  #
+  ##
+
+  #
+  # Checks to see if the EOF flag has been set on the pool.
+  #
+  def eof
+    request = Packet.create_request('core_channel_eof')
+
+    request.add_tlv(TLV_TYPE_CHANNEL_ID, self.cid)
+
+    begin
+      response = self.client.send_request(request)
+    rescue
+      return true
+    end
+
+    if (response.has_tlv?(TLV_TYPE_BOOL))
+      return response.get_tlv_value(TLV_TYPE_BOOL)
+    end
+
+    return false
+  end
+
+  #
+  # Reads data from the remote side of the pool and raises EOFError if the
+  # pool has been reached EOF.
+  #
+  def read(length = nil)
+    begin
+      data = super(length)
+    rescue
+      data = nil
+    end
+
+    if (((data == nil) || (data.length == 0)) &&
+        (self.eof))
+      raise EOFError
+    end
+
+    return data
+  end
+
+  #
+  # This method seeks to an offset within the remote side of the pool using
+  # the standard seek whence clauses.
+  #
+  def seek(offset, whence = SEEK_SET)
+    sane = 0
+
+    # Just in case...
+    case whence
+      when ::IO::SEEK_SET
+        sane = 0
+      when ::IO::SEEK_CUR
+        sane = 1
+      when ::IO::SEEK_END
+        sane = 2
+      else
+        raise RuntimeError, "Invalid seek whence #{whence}.", caller
+    end
+
+    request = Packet.create_request('core_channel_seek')
+
+    request.add_tlv(TLV_TYPE_CHANNEL_ID, self.cid)
+    request.add_tlv(TLV_TYPE_SEEK_OFFSET, offset)
+    request.add_tlv(TLV_TYPE_SEEK_WHENCE, sane)
+
+    begin
+      response = self.client.send_request(request)
+    rescue
+      return -1
+    end
+
+    return tell
+  end
+
+  #
+  # Synonym for tell.
+  #
+  def pos
+    return tell
+  end
+
+  #
+  # This method returns the current file pointer position to the caller.
+  #
+  def tell
+    request = Packet.create_request('core_channel_tell')
+    pos     = -1
+
+    request.add_tlv(TLV_TYPE_CHANNEL_ID, self.cid)
+
+    begin
+      response = self.client.send_request(request)
+    rescue
+      return pos
+    end
+
+    # Set the return value to the position that we're at
+    if (response.has_tlv?(TLV_TYPE_SEEK_POS))
+      pos = response.get_tlv_value(TLV_TYPE_SEEK_POS)
+    end
+
+    return pos
+  end
 
 protected
-	attr_accessor :_eof # :nodoc:
+  attr_accessor :_eof # :nodoc:
 
 end
 

data/lib/rex/post/meterpreter/channels/pools/file.rb

@@ -1,4 +1,3 @@
-#!/usr/bin/env ruby
 # -*- coding: binary -*-
 
 require 'rex/post/meterpreter/channels/pool'
@@ -21,41 +20,41 @@ module Pools
 ###
 class File < Rex::Post::Meterpreter::Channels::Pool
 
-	##
-	#
-	# Factory
-	#
-	##
-
-	#
-	# This method returns an instance of a file pool channel that can be read
-	# from, written to, seeked on, and other interacted with.
-	#
-	def File.open(client, name, mode = "r", perm = 0)
-		return Channel.create(client, 'stdapi_fs_file',
-				self, CHANNEL_FLAG_SYNCHRONOUS,
-				[
-					{
-						'type'  => Rex::Post::Meterpreter::Extensions::Stdapi::TLV_TYPE_FILE_PATH,
-						'value' => client.unicode_filter_decode( name )
-					},
-					{
-						'type'  => Rex::Post::Meterpreter::Extensions::Stdapi::TLV_TYPE_FILE_MODE,
-						'value' => mode + "b"
-					},
-				])
-	end
-
-	##
-	#
-	# Constructor
-	#
-	##
-
-	# Initializes the file channel instance
-	def initialize(client, cid, type, flags)
-		super(client, cid, type, flags)
-	end
+  ##
+  #
+  # Factory
+  #
+  ##
+
+  #
+  # This method returns an instance of a file pool channel that can be read
+  # from, written to, seeked on, and other interacted with.
+  #
+  def File.open(client, name, mode = "r", perm = 0)
+    return Channel.create(client, 'stdapi_fs_file',
+        self, CHANNEL_FLAG_SYNCHRONOUS,
+        [
+          {
+            'type'  => Rex::Post::Meterpreter::Extensions::Stdapi::TLV_TYPE_FILE_PATH,
+            'value' => client.unicode_filter_decode( name )
+          },
+          {
+            'type'  => Rex::Post::Meterpreter::Extensions::Stdapi::TLV_TYPE_FILE_MODE,
+            'value' => mode + "b"
+          },
+        ])
+  end
+
+  ##
+  #
+  # Constructor
+  #
+  ##
+
+  # Initializes the file channel instance
+  def initialize(client, cid, type, flags)
+    super(client, cid, type, flags)
+  end
 
 end
 

data/lib/rex/post/meterpreter/channels/pools/stream_pool.rb

@@ -1,4 +1,3 @@
-#!/usr/bin/env ruby
 # -*- coding: binary -*-
 
 require 'rex/post/meterpreter/channels/pool'
@@ -25,78 +24,78 @@ module Pools
 ###
 class StreamPool < Rex::Post::Meterpreter::Channels::Pool
 
-	include Rex::IO::StreamAbstraction
-
-	##
-	#
-	# Constructor
-	#
-	##
-
-	# Initializes the file channel instance
-	def initialize(client, cid, type, flags)
-		super(client, cid, type, flags)
-
-		initialize_abstraction
-	end
-
-	##
-	#
-	# Streaming pools don't support tell, seek, or eof.
-	#
-	##
-
-	#
-	# This method returns the current offset into the pool.
-	#
-	def tell
-		raise NotImplementedError
-	end
-
-	#
-	# This method seeks to an offset in the pool.
-	#
-	def seek
-		raise NotImplementedError
-	end
-
-	#
-	# This method returns whether or not eof has been returned.
-	#
-	def eof
-		return false
-	end
-
-	#
-	# Transfers data to the local half of the pool for reading.
-	#
-	def dio_write_handler(packet, data)
-		rv = Rex::ThreadSafe.select(nil, [rsock], nil, 0.01)
-		if(rv)
-			rsock.write(data)
-			return true
-		else
-			return false
-		end
-	end
-
-	#
-	# Closes the local half of the pool stream.
-	#
-	def dio_close_handler(packet)
-		rsock.close
-
-		return super(packet)
-	end
-
-	#
-	# Cleans up resources used by the channel.
-	#
-	def cleanup
-		super
-
-		cleanup_abstraction
-	end
+  include Rex::IO::StreamAbstraction
+
+  ##
+  #
+  # Constructor
+  #
+  ##
+
+  # Initializes the file channel instance
+  def initialize(client, cid, type, flags)
+    super(client, cid, type, flags)
+
+    initialize_abstraction
+  end
+
+  ##
+  #
+  # Streaming pools don't support tell, seek, or eof.
+  #
+  ##
+
+  #
+  # This method returns the current offset into the pool.
+  #
+  def tell
+    raise NotImplementedError
+  end
+
+  #
+  # This method seeks to an offset in the pool.
+  #
+  def seek
+    raise NotImplementedError
+  end
+
+  #
+  # This method returns whether or not eof has been returned.
+  #
+  def eof
+    return false
+  end
+
+  #
+  # Transfers data to the local half of the pool for reading.
+  #
+  def dio_write_handler(packet, data)
+    rv = Rex::ThreadSafe.select(nil, [rsock], nil, 0.01)
+    if(rv)
+      rsock.write(data)
+      return true
+    else
+      return false
+    end
+  end
+
+  #
+  # Closes the local half of the pool stream.
+  #
+  def dio_close_handler(packet)
+    rsock.close
+
+    return super(packet)
+  end
+
+  #
+  # Cleans up resources used by the channel.
+  #
+  def cleanup
+    super
+
+    cleanup_abstraction
+  end
 
 end