ronin 0.2.0 → 0.2.1

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

@@ -21,10 +21,11 @@
 #++
 #
 
-require 'ronin/ui/command_line/commands/default'
 require 'ronin/ui/command_line/exceptions/unknown_command'
+require 'ronin/ui/command_line/commands/console'
 require 'ronin/ui/console'
-require 'ronin/version'
+
+require 'reverse_require'
 
 module Ronin
   module UI
@@ -33,15 +34,20 @@ module Ronin
       # Returns the commands registered with the command-line utility.
       #
       def CommandLine.commands
-        @@ronin_commands ||= []
-      end
+        unless class_variable_defined?('@@ronin_commands')
+          paths = Gem.find_resources('bin/ronin-*')
+          
+          @@ronin_commands = {}
+            
+          paths.each do |path|
+            next unless File.executable?(path)
+            name = File.basename(path).gsub(/^ronin-/,'')
 
-      #
-      # Returns the Hash of the Command names and their Command objects
-      # registered with the command-line utility.
-      #
-      def CommandLine.commands_by_name
-        @@ronin_commands_by_name ||= {}
+            @@ronin_commands[name] ||= path
+          end
+        end
+
+        return @@ronin_commands
       end
 
       #
@@ -49,7 +55,7 @@ module Ronin
       # registered with the command-line utility.
       #
       def CommandLine.has_command?(name)
-        CommandLine.commands_by_name.has_key?(name.to_s)
+        CommandLine.commands.has_key?(name.to_s)
       end
 
       #
@@ -63,27 +69,7 @@ module Ronin
           raise(UnknownCommand,"unknown command #{name.dump}",caller)
         end
 
-        return CommandLine.commands_by_name[name]
-      end
-
-      #
-      # Prints the specified error _message_.
-      #
-      def CommandLine.error(message)
-        STDERR.puts "ronin: #{message}"
-        return false
-      end
-
-      #
-      # Prints the given error _message_ and exits unseccessfully from the
-      # command-line utility. If a _block_ is given, it will be called before
-      # any error _message_ are printed.
-      #
-      def CommandLine.fail(message,&block)
-        block.call(self) if block
-
-        CommandLine.error(message)
-        exit -1
+        return CommandLine.commands[name]
       end
 
       #
@@ -92,26 +78,18 @@ module Ronin
       # attempt to find and execute the Command with the same name.
       #
       def CommandLine.run(*argv)
-        begin
-          if (argv.empty? || argv[0][0..0]=='-')
-            DefaultCommand.run(*argv)
-          else
-            cmd = argv.first
-            argv = argv[1..-1]
+        if (argv.empty? || argv[0][0..0]=='-')
+          Commands::Console.run(*argv)
+        else
+          cmd = argv.first
+          argv = argv[1..-1]
 
-            if CommandLine.has_command?(cmd)
-              begin
-                CommandLine.commands_by_name[cmd].run(*argv)
-              rescue => excp
-                STDERR.puts excp
-                exit -1
-              end
-            else
-              CommandLine.fail("unknown command #{cmd.dump}")
-            end
+          begin
+            exec(CommandLine.get_command(cmd),*argv)
+          rescue UnknownCommand => e
+            STDERR.puts e
+            exit -1
           end
-        rescue OptionParser::MissingArgument, OptionParser::InvalidOption => e
-          CommandLine.fail(e)
         end
 
         return true

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

@@ -27,63 +27,64 @@ require 'ronin/platform/overlay'
 module Ronin
   module UI
     module CommandLine
-      class AddCommand < Command
+      module Commands
+        class Add < Command
 
-        command :add
+          def defaults
+            @cache = nil
+            @media = nil
+            @uri = nil
+          end
 
-        def initialize
-          @cache = nil
-          @media = nil
-          @uri = nil
+          def define_options(opts)
+            opts.usage = 'PATH [options]'
 
-          super
-        end
+            opts.options do
+              opts.on('-C','--cache DIR','Specify an alternate overlay cache') do |dir|
+                @cache = dir
+              end
 
-        def define_options(opts)
-          opts.usage = 'PATH [options]'
+              opts.on('-m','--media MEDIA','Spedify the media-type of the overlay') do |media|
+                @media = media
+              end
 
-          opts.options do
-            opts.on('-C','--cache DIR','Specify an alternate overlay cache') do |dir|
-              @cache = dir
-            end
+              opts.on('-U','--uri URI','Specify the source URI of the overlay') do |uri|
+                @uri = uri
+              end
 
-            opts.on('-m','--media MEDIA','Spedify the media-type of the overlay') do |media|
-              @media = media
+              opts.on('-L','--local','Similiar to: --media local') do
+                @media = :local
+              end
             end
 
-            opts.on('-U','--uri URI','Specify the source URI of the overlay') do |uri|
-              @uri = uri
-            end
+            opts.arguments(
+              'PATH' => 'Add the overlay located at the specified PATH'
+            )
 
-            opts.on('-L','--local','Similiar to: --media local') do
-              @media = :local
-            end
+            opts.summary %{
+              Add a local overlay located at the specified PATH to the
+              Overlay cache
+            }
           end
 
-          opts.arguments(
-            'PATH' => 'Add the overlay located at the specified PATH'
-          )
-
-          opts.summary('Add a local overlay located at the specified PATH to the Overlay cache')
-        end
-
-        def arguments(*args)
-          unless args.length == 1
-            fail('only one overlay path maybe specified')
-          end
+          def arguments(*args)
+            unless args.length == 1
+              fail('only one overlay path maybe specified')
+            end
 
-          Platform.load_overlays(@cache) if @cache
+            Platform.load_overlays(@cache) if @cache
 
-          overlay_options = {:path => path.first}
+            overlay_options = {:path => args.first}
 
-          overlay_options[:media] = @media if @media
-          overlay_options[:uri] = @uri if @uri
+            overlay_options[:media] = @media if @media
+            overlay_options[:uri] = @uri if @uri
 
-          Platform.add(overlay_options) do |overlay|
-            puts "Overlay #{overlay.name.dump} added."
+            Platform.add(overlay_options) do |overlay|
+              puts "Overlay #{overlay.name.dump} added."
+            end
           end
-        end
 
+        end
       end
     end
   end

data/lib/ronin/ui/command_line/commands/{default.rb → console.rb}

@@ -22,42 +22,49 @@
 #
 
 require 'ronin/ui/command_line/command'
+require 'ronin/ui/verbose'
 require 'ronin/ui/console'
 require 'ronin/database'
 
 module Ronin
   module UI
     module CommandLine
-      class DefaultCommand < Command
+      module Commands
+        class Console < Command
 
-        def define_options(opts)
-          opts.usage = '<command> [options]'
-          opts.options do
-            opts.on('-d','--database URI','The URI for the Database') do |uri|
-              Database.config = uri.to_s
-            end
+          def define_options(opts)
+            opts.usage = '<command> [options]'
+            opts.options do
+              opts.on('-D','--database URI','The URI for the Database') do |uri|
+                Database.config = uri.to_s
+              end
 
-            opts.on('-r','--require LIB','require the specified library or path') do |lib|
-              Console.auto_load << lib.to_s
-            end
+              opts.on('-r','--require LIB','Require the specified library or path') do |lib|
+                UI::Console.auto_load << lib.to_s
+              end
+
+              opts.on('-v','--verbose','Enables verbose output') do
+                UI::Verbose.enable!
+              end
 
-            opts.on('-V','--version','print version information and exit') do
-              CommandLine.success do
-                puts "Ronin #{Ronin::VERSION}"
+              opts.on('-V','--version','Print version information and exit') do
+                success do
+                  puts "Ronin #{Ronin::VERSION}"
+                end
               end
             end
+
+            opts.summary %{
+              Ronin is a Ruby development platform designed for information
+              security and data exploration tasks.
+            }
           end
 
-          opts.summary %{
-            Ronin is a Ruby development platform designed for information security
-            and data exploration tasks.
-          }
-        end
+          def arguments(*args)
+            UI::Console.start
+          end
 
-        def arguments(*args)
-          Console.start
         end
-
       end
     end
   end

data/lib/ronin/ui/command_line/commands/{extension.rb → ext.rb}

@@ -31,41 +31,39 @@ require 'erb'
 module Ronin
   module UI
     module CommandLine
-      class ExtensionCommand < Command
+      module Commands
+        class Ext < Command
 
-        command :extension, :ext
-
-        def initialize
-          @uses = []
-
-          super
-        end
+          def defaults
+            @uses = []
+          end
 
-        def define_options(opts)
-          opts.usage = 'PATH [...]'
+          def define_options(opts)
+            opts.usage = 'PATH [...]'
 
-          opts.arguments(
-            'PATH' => 'The PATH of the Extension to be created'
-          )
+            opts.arguments(
+              'PATH' => 'The PATH of the Extension to be created'
+            )
 
-          opts.summary('Create an empty Extension at the specified PATH')
-        end
+            opts.summary('Create an empty Extension at the specified PATH')
+          end
 
-        def arguments(*args)
-          args.each do |path|
-            path = File.expand_path(path)
-            extension_path = File.join(path,Platform::Extension::EXTENSION_FILE)
-            lib_dir = File.join(path,Platform::Extension::LIB_DIR)
-            template_path = File.join(Config::STATIC_DIR,'extension.rb')
+          def arguments(*args)
+            args.each do |path|
+              path = File.expand_path(path)
+              extension_path = File.join(path,Platform::Extension::EXTENSION_FILE)
+              lib_dir = File.join(path,Platform::Extension::LIB_DIR)
+              template_path = File.join(Config::STATIC_DIR,'extension.rb')
 
-            FileUtils.mkdir_p(path)
-            FileUtils.mkdir_p(lib_dir)
-            FileUtils.touch(File.join(lib_dir,File.basename(path) + '.rb'))
-            FileUtils.mkdir_p(File.join(lib_dir,File.basename(path)))
-            FileUtils.cp(template_path,extension_path)
+              FileUtils.mkdir_p(path)
+              FileUtils.mkdir_p(lib_dir)
+              FileUtils.touch(File.join(lib_dir,File.basename(path) + '.rb'))
+              FileUtils.mkdir_p(File.join(lib_dir,File.basename(path)))
+              FileUtils.cp(template_path,extension_path)
+            end
           end
-        end
 
+        end
       end
     end
   end

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

@@ -22,48 +22,46 @@
 #
 
 require 'ronin/ui/command_line/command'
+require 'ronin/ui/command_line/command_line'
 
 module Ronin
   module UI
     module CommandLine
-      class HelpCommand < Command
+      module Commands
+        class Help < Command
 
-        command :help
+          def define_options(opts)
+            opts.usage = '[COMMAND]'
 
-        def define_options(opts)
-          opts.usage = '[COMMAND]'
+            opts.arguments(
+              'COMMAND' => 'The command to view'
+            )
 
-          opts.arguments(
-            'COMMAND' => 'The command to view'
-          )
+            opts.summary %{
+              View a list of supported commands or information on a
+              specific command
+            }
+          end
 
-          opts.summary('View a list of supported commands or information on a specific command')
-        end
+          def arguments(*args)
+            if args.length > 1
+              fail('only one command maybe specified')
+            end
 
-        def arguments(*args)
-          if args.length > 1
-            fail('only one command maybe specified')
-          end
+            topic = args.first
 
-          topic = args.first
+            if topic
+              CommandLine.run(topic,'--help')
+            else
+              puts 'Available commands:'
 
-          if topic
-            begin
-              CommandLine.get_command(topic).help
-            rescue UnknownCommand => exp
-              CommandLine.fail(exp)
+              CommandLine.commands.keys.sort.each do |name|
+                puts "  #{name}"
+              end
             end
-          else
-            puts 'Available commands:'
-
-            CommandLine.commands.sort_by { |cmd|
-              cmd.command_names.first
-            }.each { |cmd|
-              puts "  #{cmd.command_names.join(', ')}"
-            }
           end
-        end
 
+        end
       end
     end
   end

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

@@ -27,51 +27,51 @@ require 'ronin/platform/overlay'
 module Ronin
   module UI
     module CommandLine
-      class InstallCommand < Command
+      module Commands
+        class Install < Command
 
-        command :install
-
-        def initialize
-          @cache = nil
-          @media = nil
+          def defaults
+            @cache = nil
+            @media = nil
+          end
 
-          super
-        end
+          def define_options(opts)
+            opts.usage = 'URI [options]'
 
-        def define_options(opts)
-          opts.usage = 'URI [options]'
+            opts.options do
+              opts.on('-C','--cache DIR','Specify an alternate overlay cache') do |dir|
+                @cache = dir
+              end
 
-          opts.options do
-            opts.on('-C','--cache DIR','Specify an alternate overlay cache') do |dir|
-              @cache = dir
+              opts.on('-m','--media [MEDIA]','Spedify the media-type of the overlay') do |media|
+                @media = media
+              end
             end
 
-            opts.on('-m','--media [MEDIA]','Spedify the media-type of the overlay') do |media|
-              @media = media
-            end
-          end
-
-          opts.arguments(
-            'URI' => 'The URI of the overlay to install'
-          )
+            opts.arguments(
+              'URI' => 'The URI of the overlay to install'
+            )
 
-          opts.summary('Installs the overlay located at the specified URI')
-        end
-
-        def arguments(args)
-          unless args.length == 1
-            fail('only one overlay URI maybe specified')
+            opts.summary %{
+              Installs the overlay located at the specified URI
+            }
           end
 
-          uri = args.first
+          def arguments(args)
+            unless args.length == 1
+              fail('only one overlay URI maybe specified')
+            end
+
+            uri = args.first
 
-          Platform.load_overlays(@cache) if @cache
+            Platform.load_overlays(@cache) if @cache
 
-          Platform.install(:uri => uri, :media => @media) do |overlay|
-            puts "Overlay #{overlay.name.dump} has been installed."
+            Platform.install(:uri => uri, :media => @media) do |overlay|
+              puts "Overlay #{overlay.name.dump} has been installed."
+            end
           end
-        end
 
+        end
       end
     end
   end