ronin 0.1.2 → 0.1.3

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

@@ -3,7 +3,7 @@
 # Ronin - A Ruby platform designed for information security and data
 # exploration tasks.
 #
-# Copyright (c) 2006-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+# 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
@@ -101,7 +101,12 @@ module Ronin
             argv = argv[1..-1]
 
             if CommandLine.has_command?(cmd)
-              CommandLine.commands_by_name[cmd].run(*argv)
+              begin
+                CommandLine.commands_by_name[cmd].run(*argv)
+              rescue => excp
+                STDERR.puts excp
+                exit -1
+              end
             else
               CommandLine.fail("unknown command #{cmd.dump}")
             end

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

@@ -3,7 +3,7 @@
 # Ronin - A Ruby platform designed for information security and data
 # exploration tasks.
 #
-# Copyright (c) 2006-2007 Hal Brodigan (postmodern.mod3 at gmail.com)
+# 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
@@ -33,4 +33,4 @@ require 'ronin/ui/command_line/commands/help'
 
 require 'reverse_require'
 
-require_all 'ronin/ui/command_line/commands'
+require_for 'ronin', 'ronin/ui/command_line/commands'

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

@@ -3,7 +3,7 @@
 # Ronin - A Ruby platform designed for information security and data
 # exploration tasks.
 #
-# Copyright (c) 2006-2007 Hal Brodigan (postmodern.mod3 at gmail.com)
+# 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
@@ -35,11 +35,11 @@ module Ronin
           opts.usage = 'PATH [options]'
 
           opts.options do
-            opts.on('-m','--media MEDIA','Spedify the media-type of the repository') do |media|
+            opts.on('-m','--media MEDIA','Spedify the media-type of the overlay') do |media|
               @media = media
             end
 
-            opts.on('-U','--uri URI','Specify the source URI of the repository') do |uri|
+            opts.on('-U','--uri URI','Specify the source URI of the overlay') do |uri|
               @uri = uri
             end
 
@@ -49,22 +49,22 @@ module Ronin
           end
 
           opts.arguments(
-            'PATH' => 'Add the repository located at the specified PATH'
+            'PATH' => 'Add the overlay located at the specified PATH'
           )
 
-          opts.summary('Add a local repository located at the specified PATH to the repository cache')
+          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 repository path maybe specified')
+            fail('only one overlay path maybe specified')
           end
 
           path = args.first
 
           Cache::Overlay.save_cache do
-            Cache::Overlay.add(path,@media,@uri) do |repo|
-              puts "Overlay #{repo} added."
+            Cache::Overlay.add(path,@media,@uri) do |overlay|
+              puts "Overlay #{overlay.name.dump} added."
             end
           end
         end

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

@@ -3,7 +3,7 @@
 # Ronin - A Ruby platform designed for information security and data
 # exploration tasks.
 #
-# Copyright (c) 2006-2007 Hal Brodigan (postmodern.mod3 at gmail.com)
+# 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

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

@@ -3,7 +3,7 @@
 # Ronin - A Ruby platform designed for information security and data
 # exploration tasks.
 #
-# Copyright (c) 2006-2007 Hal Brodigan (postmodern.mod3 at gmail.com)
+# 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
@@ -42,13 +42,7 @@ module Ronin
         end
 
         def define_options(opts)
-          opts.usage = '[options] PATH'
-
-          opts.options do
-            opts.on('-u','--uses NAME','Name of an other extension to use') do |name|
-              @uses << name
-            end
-          end
+          opts.usage = 'PATH [...]'
 
           opts.arguments(
             'PATH' => 'The PATH of the Extension to be created'
@@ -58,24 +52,17 @@ module Ronin
         end
 
         def arguments(*args)
-          unless args.length == 1
-            fail('only one Extension path maybe specified')
-          end
-
-          path = File.expand_path(args.first)
-          extension_path = File.join(path,Cache::Extension::EXTENSION_FILE)
-          lib_dir = File.join(path,Cache::Extension::LIB_DIR)
-
-          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)))
-
-          File.open(extension_path,'w') do |file|
-            template_path = File.join(Config::STATIC_DIR,'extension.rb.erb')
-            template = ERB.new(File.read(template_path))
-
-            file.write(template.result(binding))
+          args.each do |path|
+            path = File.expand_path(path)
+            extension_path = File.join(path,Cache::Extension::EXTENSION_FILE)
+            lib_dir = File.join(path,Cache::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)
           end
         end
 

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

@@ -3,7 +3,7 @@
 # Ronin - A Ruby platform designed for information security and data
 # exploration tasks.
 #
-# Copyright (c) 2006-2007 Hal Brodigan (postmodern.mod3 at gmail.com)
+# 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

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

@@ -3,7 +3,7 @@
 # Ronin - A Ruby platform designed for information security and data
 # exploration tasks.
 #
-# Copyright (c) 2006-2007 Hal Brodigan (postmodern.mod3 at gmail.com)
+# 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
@@ -35,28 +35,28 @@ module Ronin
           opts.usage = 'URI [options]'
 
           opts.options do
-            opts.on('-m','--media [MEDIA]','Spedify the media-type of the repository') do |media|
+            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 repository to install'
+            'URI' => 'The URI of the overlay to install'
           )
 
-          opts.summary('Installs the repository located at the specified URI')
+          opts.summary('Installs the overlay located at the specified URI')
         end
 
         def arguments(args)
           unless args.length == 1
-            fail('only one repository URI maybe specified')
+            fail('only one overlay URI maybe specified')
           end
 
           uri = args.first
 
           Cache::Overlay.save_cache do
-            Cache::Overlay.install(:uri => uri, :media => @media) do |repo|
-              puts "Overlay #{repo} has been installed."
+            Cache::Overlay.install(:uri => uri, :media => @media) do |overlay|
+              puts "Overlay #{overlay.name.dump} has been installed."
             end
           end
         end

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

@@ -3,7 +3,7 @@
 # Ronin - A Ruby platform designed for information security and data
 # exploration tasks.
 #
-# Copyright (c) 2006-2007 Hal Brodigan (postmodern.mod3 at gmail.com)
+# 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
@@ -44,13 +44,13 @@ module Ronin
             'NAME' => 'Overlay to display'
           )
 
-          opts.summary('Display all or the specified repositories within the repository cache')
+          opts.summary('Display all or the specified repositories within the Overlay cache')
         end
 
         def arguments(*args)
           if args.empty?
             # list all repositories by name
-            Cache::Overlay.each { |repo| puts "  #{repo}" }
+            Cache::Overlay.each { |overlay| puts "  #{overlay}" }
             return
           end
 
@@ -77,12 +77,18 @@ module Ronin
                 puts "  Website: #{overlay.website}"
               end
 
-              if overlay.description
-                puts "  Description:\n\n    #{overlay.description}"
+              unless overlay.extensions.empty?
+                puts "  Extensions:\n\n"
+                overlay.each_extension { |ext| puts "    #{ext}" }
               end
 
-              puts "  Extensions:\n\n"
-              overlay.each_extension { |ext| puts "    #{ext.name}" }
+              if overlay.description
+                puts "  Description:\n\n    #{overlay.description}\n\n"
+              else
+                puts "\n"
+              end
+            else
+              puts "\n"
             end
           end
         end

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

@@ -3,7 +3,7 @@
 # Ronin - A Ruby platform designed for information security and data
 # exploration tasks.
 #
-# Copyright (c) 2006-2007 Hal Brodigan (postmodern.mod3 at gmail.com)
+# 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

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

@@ -3,7 +3,7 @@
 # Ronin - A Ruby platform designed for information security and data
 # exploration tasks.
 #
-# Copyright (c) 2006-2007 Hal Brodigan (postmodern.mod3 at gmail.com)
+# 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
@@ -41,7 +41,7 @@ module Ronin
           end
 
           opts.arguments(
-            'NAME' => 'The repository to remove'
+            'NAME' => 'The overlay to remove'
           )
 
           opts.summary('Remove the specified repositories')
@@ -50,8 +50,8 @@ module Ronin
         def arguments(*args)
           args.each do |name|
             Cache::Overlay.save_cache do
-              Cache::Overlay.remove(name) do |repo|
-                puts "Removing #{repo}..."
+              Cache::Overlay.remove(name) do |overlay|
+                puts "Removing #{overlay.name.dump} ..."
               end
             end
           end

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

@@ -3,7 +3,7 @@
 # Ronin - A Ruby platform designed for information security and data
 # exploration tasks.
 #
-# Copyright (c) 2006-2007 Hal Brodigan (postmodern.mod3 at gmail.com)
+# 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
@@ -41,7 +41,7 @@ module Ronin
           end
 
           opts.arguments(
-            'NAME' => 'The repository to uninstall'
+            'NAME' => 'The overlay to uninstall'
           )
 
           opts.summary('Uninstall the specified repositories')
@@ -50,8 +50,8 @@ module Ronin
         def arguments(*args)
           args.each do |name|
             Cache::Overlay.save_cache do
-              Cache::Overlay.uninstall(name) do |repo|
-                puts "Uninstalling #{repo}..."
+              Cache::Overlay.uninstall(name) do |overlay|
+                puts "Uninstalling #{overlay.name.dump} ..."
               end
             end
           end

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

@@ -3,7 +3,7 @@
 # Ronin - A Ruby platform designed for information security and data
 # exploration tasks.
 #
-# Copyright (c) 2006-2007 Hal Brodigan (postmodern.mod3 at gmail.com)
+# 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
@@ -41,7 +41,7 @@ module Ronin
           end
 
           opts.arguments(
-            'NAME' => 'The repository to update'
+            'NAME' => 'The overlay to update'
           )
 
           opts.summary('Updates all or the specified repositories')
@@ -49,7 +49,7 @@ module Ronin
 
         def arguments(*args)
           if args.empty?
-            Cache::Overlay.each { |repo| repo.update }
+            Cache::Overlay.each { |overlay| overlay.update }
           else
             args.each { |name| Cache::Overlay.update(name) }
           end

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

@@ -3,7 +3,7 @@
 # Ronin - A Ruby platform designed for information security and data
 # exploration tasks.
 #
-# Copyright (c) 2006-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+# 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

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

@@ -3,7 +3,7 @@
 # Ronin - A Ruby platform designed for information security and data
 # exploration tasks.
 #
-# Copyright (c) 2006-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+# 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

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

@@ -3,7 +3,7 @@
 # Ronin - A Ruby platform designed for information security and data
 # exploration tasks.
 #
-# Copyright (c) 2006-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+# 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

data/lib/ronin/ui/command_line/param_parser.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
+#++
+#
+
+require 'uri'
+
+module Ronin
+  module UI
+    module CommandLine
+      module ParamParser
+        # Hash of format patterns and their parsers
+        FORMATS = {
+          /^[0-9]+$/ => lambda { |value| value.to_i },
+          /^0x[0-9a-fA-F]+$/ => lambda { |value| value.hex },
+          /^[a-zA-Z][a-zA-Z0-9]*:\/\// => lambda { |value| URI(value) },
+          'true' => lambda { |value| true },
+          'false' => lambda { |value| false }
+        }
+
+        # The params Hash
+        attr_reader :params
+
+        #
+        # Creates an empty +params+ Hash.
+        #
+        def initialize
+          @params = {}
+        end
+
+        #
+        # Parses the specified _name_and_value_ string of the form
+        # "name=value" and extracts both the _name_ and the _value_, saving
+        # both the _name_ and _value_ within the +params+ Hash. If the
+        # extracted _value_ matches one of the patterns within +FORMATS+,
+        # then the associated parser will first parse the _value_.
+        #
+        def parse_param(name_and_value)
+          name, value = name_and_value.split('=',2)
+
+          if value
+            FORMATS.each do |pattern,parser|
+              if value.match(pattern)
+                value = parser.call(value)
+                break
+              end
+            end
+          end
+
+          @params[name.to_sym] = value
+        end
+      end
+    end
+  end
+end