ronin 0.1.4 → 0.2.0

data/lib/ronin/ui/command_line/commands/overlay.rb

@@ -22,7 +22,7 @@
 #
 
 require 'ronin/ui/command_line/command'
-require 'ronin/cache/overlay'
+require 'ronin/platform/overlay'
 require 'ronin/version'
 
 require 'fileutils'
@@ -38,7 +38,7 @@ module Ronin
         command :overlay
 
         def initialize
-          @name = nil
+          @title = nil
           @source = nil
           @source_view = nil
           @website = nil
@@ -53,8 +53,8 @@ module Ronin
           opts.usage = '[options] PATH'
 
           opts.options do
-            opts.on('-n','--name NAME','Name of the Overlay') do |name|
-              @name = name
+            opts.on('-t','--title NAME','Name of the Overlay') do |title|
+              @title = title
             end
 
             opts.on('-S','--source URL','The URL where the source of the Overlay will be hosted') do |url|
@@ -101,14 +101,15 @@ module Ronin
 
           path = File.expand_path(args.first)
 
-          @name ||= File.basename(path).gsub(/[_\s]+/,' ').capitalize
+          @title ||= File.basename(path).gsub(/[_\s]+/,' ').capitalize
           @source_view ||= @source
           @website ||= @source_view
 
           FileUtils.mkdir_p(path)
+          FileUtils.mkdir_p(File.join(path,'lib'))
           FileUtils.mkdir_p(File.join(path,'objects'))
 
-          File.open(File.join(path,Cache::Overlay::METADATA_FILE),'w') do |file|
+          File.open(File.join(path,Platform::Overlay::METADATA_FILE),'w') do |file|
             doc = Document.new
             doc.add(XMLDecl.new)
             doc.add(Instruction.new('xml-stylesheet','type="text/xsl" href="http://ronin.rubyforge.org/dist/overlay.xsl"'))
@@ -116,9 +117,9 @@ module Ronin
             root = Element.new('ronin-overlay')
             root.attributes['version'] = Ronin::VERSION
 
-            name_tag = Element.new('name')
-            name_tag.text = @name
-            root.add_element(name_tag)
+            title_tag = Element.new('title')
+            title_tag.text = @title
+            root.add_element(title_tag)
 
             if @source
               source_tag = Element.new('source')

data/lib/ronin/ui/command_line/commands/remove.rb

@@ -22,7 +22,7 @@
 #
 
 require 'ronin/ui/command_line/command'
-require 'ronin/cache/overlay'
+require 'ronin/platform/overlay'
 
 module Ronin
   module UI
@@ -31,10 +31,21 @@ module Ronin
 
         command :remove, :rm
 
+        def initialize
+          @cache = nil
+          @verbose = false
+
+          super
+        end
+
         def define_options(opts)
           opts.usage = 'NAME [...] [options]'
 
           opts.options do
+            opts.on('-C','--cache DIR','Specify an alternate overlay cache') do |dir|
+              @cache = dir
+            end
+
             opts.on('-v','--verbose','Enable verbose output') do
               @verbose = true
             end
@@ -48,11 +59,11 @@ module Ronin
         end
 
         def arguments(*args)
+          Platform.load_overlays(@cache) if @cache
+
           args.each do |name|
-            Cache::Overlay.save_cache do
-              Cache::Overlay.remove(name) do |overlay|
-                puts "Removing #{overlay.name.dump} ..."
-              end
+            Platform.remove(name) do |overlay|
+              puts "Removing #{overlay.name.dump} ..."
             end
           end
         end

data/lib/ronin/ui/command_line/commands/uninstall.rb

@@ -22,7 +22,7 @@
 #
 
 require 'ronin/ui/command_line/command'
-require 'ronin/cache/overlay'
+require 'ronin/platform/overlay'
 
 module Ronin
   module UI
@@ -31,10 +31,21 @@ module Ronin
 
         command :uninstall
 
+        def initialize
+          @cache = nil
+          @verbose = false
+
+          super
+        end
+
         def define_options(opts)
           opts.usage = 'NAME [NAME ...] [options]'
 
           opts.options do
+            opts.on('-C','--cache DIR','Specify an alternate overlay cache') do |dir|
+              @cache = dir
+            end
+
             opts.on('-v','--verbose','Enable verbose output') do
               @verbose = true
             end
@@ -48,11 +59,11 @@ module Ronin
         end
 
         def arguments(*args)
+          Platform.load_overlays(@cache) if @cache
+
           args.each do |name|
-            Cache::Overlay.save_cache do
-              Cache::Overlay.uninstall(name) do |overlay|
-                puts "Uninstalling #{overlay.name.dump} ..."
-              end
+            Platform.uninstall(name) do |overlay|
+              puts "Uninstalling #{overlay.name.dump} ..."
             end
           end
         end

data/lib/ronin/ui/command_line/commands/update.rb

@@ -22,7 +22,7 @@
 #
 
 require 'ronin/ui/command_line/command'
-require 'ronin/cache/overlay'
+require 'ronin/platform/overlay'
 
 module Ronin
   module UI
@@ -31,10 +31,21 @@ module Ronin
 
         command :update, :up
 
+        def initialize
+          @cache = nil
+          @verbose = false
+
+          super
+        end
+
         def define_options(opts)
           opts.usage = '[NAME ...] [options]'
 
           opts.options do
+            opts.on('-C','--cache DIR','Specify an alternate overlay cache') do |dir|
+              @cache = dir
+            end
+
             opts.on('-v','--verbose','Enable verbose output') do
               @verbose = true
             end
@@ -48,10 +59,12 @@ module Ronin
         end
 
         def arguments(*args)
+          Platform.load_overlays(@cache) if @cache
+
           if args.empty?
-            Cache::Overlay.each { |overlay| overlay.update }
+            Platform.overlays.each_overlay { |overlay| overlay.update }
           else
-            args.each { |name| Cache::Overlay.update(name) }
+            args.each { |name| Platform.overlays.update(name) }
           end
         end
 

data/lib/ronin/ui/console.rb

@@ -27,101 +27,105 @@ require 'irb'
 require 'irb/completion'
 
 module Ronin
-  module Console
-    #
-    # Returns the default Console prompt style
-    #
-    def Console.prompt
-      @@ronin_console_prompt ||= :SIMPLE
-    end
+  module UI
+    module Console
+      #
+      # Returns the default Console prompt style
+      #
+      def Console.prompt
+        @@ronin_console_prompt ||= :SIMPLE
+      end
 
-    #
-    # Sets the default Console prompt style to the specified _style_.
-    #
-    def Console.prompt=(style)
-      @@ronin_console_prompt = style
-    end
+      #
+      # Sets the default Console prompt style to the specified _style_.
+      #
+      def Console.prompt=(style)
+        @@ronin_console_prompt = style
+      end
 
-    #
-    # Returns the default Console indent setting.
-    #
-    def Console.indent
-      @@ronin_console_indent ||= true
-    end
+      #
+      # Returns the default Console indent setting.
+      #
+      def Console.indent
+        @@ronin_console_indent ||= true
+      end
 
-    #
-    # Sets the default Console indent setting.
-    #
-    def Console.indent=(value)
-      @@ronin_console_indent = value
-    end
+      #
+      # Sets the default Console indent setting.
+      #
+      def Console.indent=(value)
+        @@ronin_console_indent = value
+      end
 
-    #
-    # Returns the Array of files to require when the Console starts.
-    #
-    def Console.auto_load
-      @@ronin_console_auto_load ||= []
-    end
+      #
+      # Returns the Array of files to require when the Console starts.
+      #
+      def Console.auto_load
+        @@ronin_console_auto_load ||= []
+      end
 
-    #
-    # Calls the specified _block_ from within the Console after it is
-    # started.
-    #
-    def Console.setup(&block)
-      Console.setup_blocks << block if block
-    end
+      #
+      # Calls the specified _block_ from within the Console after it is
+      # started.
+      #
+      def Console.setup(&block)
+        Console.setup_blocks << block if block
+      end
 
-    #
-    # Starts a Console with the given _script_. If a _block_ is given, it
-    # will be called from within the Console.
-    #
-    def Console.start(script=nil,&block)
-      IRB.setup(script)
+      #
+      # Starts a Console with the given _script_. If a _block_ is given, it
+      # will be called from within the Console.
+      #
+      def Console.start(script=nil,&block)
+        IRB.setup(script)
 
-      IRB.conf[:IRB_NAME] = 'ronin'
-      IRB.conf[:PROMPT_MODE] = Console.prompt
-      IRB.conf[:AUTO_INDENT] = Console.indent
-      IRB.conf[:LOAD_MODULES] = Console.auto_load
+        IRB.conf[:IRB_NAME] = 'ronin'
+        IRB.conf[:PROMPT_MODE] = Console.prompt
+        IRB.conf[:AUTO_INDENT] = Console.indent
 
-      irb = IRB::Irb.new(nil,script)
+        irb = IRB::Irb.new(nil,script)
 
-      # configure the irb workspace
-      irb.context.main.instance_eval do
-        require 'ronin'
-        require 'pp'
+        # configure the irb workspace
+        irb.context.main.instance_eval do
+          require 'ronin/environment'
 
-        include Ronin
-      end
+          Ronin::UI::Console.auto_load.each do |path|
+            require path
+          end
 
-      Console.setup_blocks.each do |setup_block|
-        irb.context.main.instance_eval(&setup_block)
-      end
+          include Ronin
+        end
 
-      # Load console configuration block is given
-      irb.context.main.instance_eval(&block) if block
+        Console.setup_blocks.each do |setup_block|
+          irb.context.main.instance_eval(&setup_block)
+        end
 
-      IRB.conf[:MAIN_CONTEXT] = irb.context
+        # Load console configuration block is given
+        irb.context.main.instance_eval(&block) if block
 
-      trap('SIGINT') do
-        irb.signal_handle
-      end
+        IRB.conf[:MAIN_CONTEXT] = irb.context
 
-      catch(:IRB_EXIT) do
-        irb.eval_input
-      end
+        trap('SIGINT') do
+          irb.signal_handle
+        end
 
-      print "\n"
-      return nil
-    end
+        catch(:IRB_EXIT) do
+          irb.eval_input
+        end
 
-    protected
+        putc "\n"
+        return nil
+      end
 
-    #
-    # Returns the Array of setup_blocks to run within the Console after it
-    # is started.
-    #
-    def Console.setup_blocks
-      @@console_setup_blocks ||= []
+      protected
+
+      #
+      # Returns the Array of setup_blocks to run within the Console after it
+      # is started.
+      #
+      def Console.setup_blocks
+        @@console_setup_blocks ||= []
+      end
     end
   end
 end

data/lib/ronin/ui/diagnostics.rb

@@ -0,0 +1,73 @@
+#
+#--
+# Ronin - A Ruby platform designed for information security and data
+# exploration tasks.
+#
+# Copyright (c) 2006-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#++
+#
+
+module Ronin
+  module UI
+    module Diagnostics
+      def Diagnostics.enable!
+        @@ronin_diagnostics = true
+      end
+
+      def Diagnostics.enabled?
+        (@@ronin_diagnostics ||= false) == true
+      end
+
+      def Diagnostics.disable!
+        @@ronin_diagnostics = false
+      end
+
+      def Diagnostics.disabled?
+        (@@ronin_diagnostics ||= false) == false
+      end
+
+      protected
+
+      #
+      # Prints the given _messages_ as info diagnostics.
+      #
+      def print_info(*messages)
+        if Diagnostics.enabled?
+          STDERR.puts(*(messages.map { |mesg| "[-] #{mesg}" }))
+        end
+      end
+
+      #
+      # Prints the given _messages_ as warning diagnostics.
+      #
+      def print_warning(*messages)
+        if Diagnostics.enabled?
+          STDERR.puts(*(messages.map { |mesg| "[*] #{mesg}" }))
+        end
+      end
+
+      #
+      # Prints the given _messages_ as error diagnostics.
+      #
+      def print_error(*messages)
+        if Diagnostics.enabled?
+          STDERR.puts(*(messages.map { |mesg| "[!] #{mesg}" }))
+        end
+      end
+    end
+  end
+end