dstruct 0.0.1

data/lib/rex/ui/text/irb_shell.rb

@@ -0,0 +1,62 @@
+# -*- coding: binary -*-
+module Rex
+module Ui
+module Text
+
+###
+#
+# This class wraps the creation of an IRB shell.
+#
+###
+class IrbShell
+
+  @@IrbInitialized = false
+
+  def initialize(binding)
+    @binding_ctx = binding
+  end
+
+  #
+  # Runs the IRB shell until completion.  The binding parameter initializes
+  # IRB to the appropriate binding context.
+  #
+  def run
+    # Initialize IRB by setting up its internal configuration hash and
+    # stuff.
+    if (@@IrbInitialized == false)
+      load('irb.rb')
+
+      IRB.setup(nil)
+      IRB.conf[:PROMPT_MODE]  = :SIMPLE
+
+      @@IrbInitialized = true
+    end
+
+    # Create a new IRB instance
+    irb = IRB::Irb.new(IRB::WorkSpace.new(@binding_ctx))
+
+    # Set the primary irb context so that exit and other intrinsic
+    # commands will work.
+    IRB.conf[:MAIN_CONTEXT] = irb.context
+
+    # Trap interrupt
+    old_sigint = trap("SIGINT") do
+      begin
+        irb.signal_handle
+      rescue RubyLex::TerminateLineInput
+        irb.eval_input
+      end
+    end
+
+    # Keep processing input until the cows come home...
+    catch(:IRB_EXIT) do
+      irb.eval_input
+    end
+
+    trap("SIGINT", old_sigint)
+  end
+
+end
+end
+end
+end

data/lib/rex/ui/text/output.rb

@@ -0,0 +1,86 @@
+# -*- coding: binary -*-
+require 'rex/ui'
+
+module Rex
+module Ui
+module Text
+
+###
+#
+# This class implements text-based output but is not
+# tied to an output medium.
+#
+###
+class Output < Rex::Ui::Output
+
+  require 'rex/ui/text/output/stdio'
+  require 'rex/ui/text/output/socket'
+  require 'rex/ui/text/output/buffer'
+  require 'rex/ui/text/output/file'
+  require 'rex/ui/text/output/tee'
+  require 'rex/ui/text/color'
+
+  include Rex::Ui::Text::Color
+
+  def initialize
+    @config = {
+      :color => :auto, # true, false, :auto
+    }
+    super
+  end
+  attr_reader :config
+
+  def disable_color
+    @config[:color] = false
+  end
+
+  def enable_color
+    @config[:color] = true
+  end
+
+  def auto_color
+    @config[:color] = :auto
+  end
+
+  def update_prompt(prompt = nil)
+    return if prompt.nil?
+    substitute_colors(prompt, true)
+  end
+
+  def print_error(msg = '')
+    print_line("%bld%red[-]%clr #{msg}")
+  end
+
+  def print_good(msg = '')
+    print_line("%bld%grn[+]%clr #{msg}")
+  end
+
+  def print_debug(msg = '')
+    print_line("%bld%cya[!]%clr #{msg}")
+  end
+
+  def print_status(msg = '')
+    print_line("%bld%blu[*]%clr #{msg}")
+  end
+
+  def print_line(msg = '')
+    print(msg + "\n")
+  end
+
+  def print_warning(msg = '')
+    print_line("%bld%yel[!]%clr #{msg}")
+  end
+
+  def print(msg = '')
+    print_raw(substitute_colors(msg))
+  end
+
+  def reset
+  end
+
+end
+
+end
+end
+end
+

data/lib/rex/ui/text/output/buffer.rb

@@ -0,0 +1,62 @@
+# -*- coding: binary -*-
+require 'rex/ui'
+
+module Rex
+module Ui
+module Text
+
+###
+#
+# This class implements output against a buffer.
+#
+###
+class Output::Buffer < Rex::Ui::Text::Output
+
+  #
+  # Initializes an output buffer.
+  #
+  def initialize
+    self.buf = ''
+  end
+
+  def supports_color?
+    false
+  end
+
+  #
+  # Appends the supplied message to the output buffer.
+  #
+  def print_raw(msg = '')
+    self.buf += msg || ''
+
+    msg
+  end
+
+
+  #
+  # Read everything out of the buffer and reset it
+  #
+  def dump_buffer
+    self.buf ||= ''
+    buffer = self.buf.dup
+    reset()
+    buffer
+  end
+
+  #
+  # Reset the buffer to an empty string.
+  #
+  def reset
+    self.buf = ''
+  end
+
+  #
+  # The underlying buffer state.
+  #
+  attr_accessor :buf
+
+end
+
+end
+end
+end

data/lib/rex/ui/text/output/buffer/stdout.rb

@@ -0,0 +1,26 @@
+# -*- coding: binary -*-
+# make sure the classes are defined before opening it to define submodule
+require 'rex/ui/text/output'
+require 'rex/ui/text/output/buffer'
+
+module Rex
+  module Ui
+    module Text
+      class Output
+        class Buffer
+          # Adds {#write} method to {Rex::Ui::Text::Output::Buffer} so it can
+          # function as a stand-in for `$stdout`
+          module Stdout
+            # Prints raw message.
+            #
+            # @param (see Rex::Ui::Text::Output::Buffer#write)
+            # @return (see Rex::Ui::Text::Output::Buffer#write)
+            def write(msg = '')
+              print_raw(msg)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rex/ui/text/output/file.rb

@@ -0,0 +1,44 @@
+# -*- coding: binary -*-
+require 'rex/ui'
+
+module Rex
+module Ui
+module Text
+
+###
+#
+# This class implements output against a file
+#
+###
+class Output::File < Rex::Ui::Text::Output
+
+  attr_accessor :fd
+
+  def initialize(path)
+    self.fd = ::File.open(path, "wb")
+  end
+
+  def supports_color?
+    false
+  end
+
+  #
+  # Prints the supplied message to file output.
+  #
+  def print_raw(msg = '')
+    return if not self.fd
+    self.fd.write(msg)
+    self.fd.flush
+    msg
+  end
+
+  def close
+    self.fd.close if self.fd
+    self.fd = nil
+  end
+end
+
+end
+end
+end
+

data/lib/rex/ui/text/output/socket.rb

@@ -0,0 +1,44 @@
+# -*- coding: binary -*-
+require 'rex/ui'
+
+module Rex
+module Ui
+module Text
+
+###
+#
+# This class implements the output interface against a socket.
+#
+###
+class Output::Socket < Rex::Ui::Text::Output
+
+  def initialize(sock)
+    @sock = sock
+    super()
+  end
+
+  def supports_color?
+    case config[:color]
+    when true
+      # Allow color if the user forces it on
+      return true
+    else
+      false
+    end
+  end
+
+  #
+  # Prints the supplied message to the socket.
+  #
+  def print_raw(msg = '')
+    @sock.write(msg)
+    @sock.flush
+
+    msg
+  end
+end
+
+end
+end
+end
+

data/lib/rex/ui/text/output/stdio.rb

@@ -0,0 +1,53 @@
+# -*- coding: binary -*-
+require 'rex/ui'
+
+begin
+  require 'windows_console_color_support'
+rescue ::LoadError
+end
+
+module Rex
+module Ui
+module Text
+
+###
+#
+# This class implements output against standard out.
+#
+###
+class Output::Stdio < Rex::Ui::Text::Output
+
+  def supports_color?
+    case config[:color]
+    when true
+      return true
+    when false
+      return false
+    else # auto
+      if (Rex::Compat.is_windows)
+        return true
+      end
+      term = Rex::Compat.getenv('TERM')
+      return (term and term.match(/(?:vt10[03]|xterm(?:-color)?|linux|screen|rxvt)/i) != nil)
+    end
+  end
+
+  #
+  # Prints the supplied message to standard output.
+  #
+  def print_raw(msg = '')
+    if (Rex::Compat.is_windows and supports_color?)
+      WindowsConsoleColorSupport.new($stdout).write(msg)
+    else
+      $stdout.print(msg)
+    end
+    $stdout.flush
+
+    msg
+  end
+end
+
+end
+end
+end
+

data/lib/rex/ui/text/output/tee.rb

@@ -0,0 +1,56 @@
+# -*- coding: binary -*-
+require 'rex/ui'
+
+module Rex
+module Ui
+module Text
+
+###
+#
+# This class implements output against a file and stdout
+#
+###
+class Output::Tee < Rex::Ui::Text::Output
+
+  attr_accessor :fd
+
+  def initialize(path)
+    self.fd = ::File.open(path, "ab")
+    super()
+  end
+
+  def supports_color?
+    case config[:color]
+    when true
+      return true
+    when false
+      return false
+    else # auto
+      term = Rex::Compat.getenv('TERM')
+      return (term and term.match(/(?:vt10[03]|xterm(?:-color)?|linux|screen|rxvt)/i) != nil)
+    end
+  end
+
+  #
+  # Prints the supplied message to file output.
+  #
+  def print_raw(msg = '')
+    $stdout.print(msg)
+    $stdout.flush
+
+    return if not self.fd
+    self.fd.write(msg)
+    self.fd.flush
+    msg
+  end
+
+  def close
+    self.fd.close if self.fd
+    self.fd = nil
+  end
+end
+
+end
+end
+end
+

data/lib/rex/ui/text/progress_tracker.rb

@@ -0,0 +1,57 @@
+# -*- coding: binary -*-
+require 'rex/ui/progress_tracker'
+
+module Rex
+module Ui
+module Text
+
+###
+#
+# This module implements progress tracking against a text interface.
+#
+###
+class ProgressTracker < Rex::Ui::ProgressTracker
+
+  def initialize(output)
+    self.output = output
+    self.start  = 0
+    self.stop   = 0
+    self.pos    = 0
+  end
+
+  #
+  # Passes a generic status message that isn't necessarily associated
+  # with a step event.
+  #
+  def status(msg = '')
+    output.print_status(msg)
+  end
+
+  #
+  # Updates the status associated with the current step.
+  #
+  def step_status(msg = '')
+    output.print_status("#{pos}: #{msg}") if (msg and msg.length > 0)
+  end
+
+  #
+  # An error occurred that may result in aborting the progress.
+  #
+  def error(msg = '')
+    output.print_error(msg)
+  end
+
+  #
+  # Progress has been aborted, the reason is supplied in msg.
+  #
+  def abort(msg = '')
+    output.print_error("fatal: #{msg}")
+  end
+
+  attr_accessor :output
+
+end
+
+end
+end
+end