ronin 1.0.0 → 1.1.0.rc1

data/lib/ronin/class_methods.rb

@@ -36,8 +36,14 @@ module Ronin
     #
     # @since 1.0.0
     #
+    # @api semipublic
+    #
     def const_missing(name)
-      Ronin.send(:const_missing,name) || super(name)
+      begin
+        Ronin.send(:const_missing,name)
+      rescue NameError
+        super(name)
+      end
     end
   end
 end

data/lib/ronin/config.rb

@@ -27,6 +27,7 @@ module Ronin
   #
   module Config
     include DataPaths
+    extend DataPaths::Finders
 
     # The users home directory
     HOME = Env.home
@@ -61,6 +62,8 @@ module Ronin
     #   Config.load :sql
     #   # => true
     #
+    # @api semipublic
+    #
     def Config.load(name=nil)
       path = if name
                CONFIG_DIR.join("#{name}.rb").expand_path
@@ -80,6 +83,8 @@ module Ronin
     # @return [Pathname]
     #   The full path within {TMP_DIR}.
     #
+    # @api semipublic
+    #
     def Config.tmp_dir(sub_path=nil)
       if sub_path
         sub_path = File.expand_path(File.join('',sub_path))

data/lib/ronin/credential.rb

@@ -18,11 +18,10 @@
 #
 
 require 'ronin/model'
+require 'ronin/user_name'
+require 'ronin/password'
 
 module Ronin
-  autoload :UserName, 'ronin/user_name'
-  autoload :Password, 'ronin/password'
-
   #
   # Represents Credentials used to access services or websites.
   #
@@ -50,6 +49,8 @@ module Ronin
     #
     # @since 1.0.0
     #
+    # @api public
+    #
     def self.for_user(name)
       all('user_name.name' => name)
     end
@@ -65,6 +66,8 @@ module Ronin
     #
     # @since 1.0.0
     #
+    # @api public
+    #
     def self.with_password(password)
       all('password.clear_text' => password)
     end
@@ -77,6 +80,8 @@ module Ronin
     #
     # @since 1.0.0
     #
+    # @api public
+    #
     def user
       self.user_name.name if self.user_name
     end
@@ -89,6 +94,8 @@ module Ronin
     #
     # @since 1.0.0
     #
+    # @api public
+    #
     def clear_text
       self.password.clear_text if self.password
     end
@@ -101,6 +108,8 @@ module Ronin
     #
     # @since 1.0.0
     #
+    # @api public
+    #
     def to_s
       "#{self.user_name}:#{self.password}"
     end
@@ -121,6 +130,8 @@ module Ronin
     #
     # @since 1.0.0
     #
+    # @api public
+    #
     def to_ary
       [self.user_name.name, self.password.clear_text]
     end

data/lib/ronin/database/database.rb

@@ -62,6 +62,8 @@ module Ronin
     #
     # @since 1.0.0
     #
+    # @api private
+    #
     def Database.repositories
       if @repositories.empty?
         @repositories[:default] = DEFAULT_REPOSITORY
@@ -93,8 +95,10 @@ module Ronin
     #
     # @since 1.0.0
     #
+    # @api semipublic
+    #
     def Database.repository?(name)
-      Database.repositories.has_key?(name.to_sym)
+      repositories.has_key?(name.to_sym)
     end
 
     #
@@ -108,13 +112,15 @@ module Ronin
     #
     # @since 1.0.0
     #
+    # @api private
+    #
     def Database.save
       yield if block_given?
 
       File.open(CONFIG_FILE,'w') do |file|
         hash = {}
         
-        Database.repositories.each do |name,value|
+        repositories.each do |name,value|
           hash[name.to_s] = value.to_s
         end
 
@@ -143,10 +149,12 @@ module Ronin
     # @return [true]
     #   Specifies that the log has been setup.
     #
+    # @api private
+    #
     def Database.setup_log(options={})
-      path = (options[:path] || DEFAULT_LOG_PATH)
-      stream = (options[:stream] || File.new(path,'w+'))
-      level = (options[:level] || DEFAULT_LOG_LEVEL)
+      path = options.fetch(:path,DEFAULT_LOG_PATH)
+      stream = options.fetch(:stream,File.new(path,'w+'))
+      level = options.fetch(:level,DEFAULT_LOG_LEVEL)
 
       @log = DataMapper::Logger.new(stream,level)
       return true
@@ -161,6 +169,8 @@ module Ronin
     # @return [Boolean]
     #   Specifies wether or not the Database is setup.
     #
+    # @api semipublic
+    #
     def Database.setup?(name=:default)
       repository = DataMapper.repository(name)
 
@@ -175,8 +185,10 @@ module Ronin
     #   Specifies whether the Database was migrated or is currently
     #   not setup.
     #
+    # @api semipublic
+    #
     def Database.upgrade!
-      if Database.setup?
+      if setup?
         Migrations.migrate_up!
       else
         false
@@ -188,17 +200,25 @@ module Ronin
     #
     # @see Database.upgrade!
     #
+    # @api semipublic
+    #
     def Database.setup
       # setup the database log
-      Database.setup_log unless @log
+      unless @log
+        if ENV['DEBUG']
+          setup_log(:stream => STDERR, :level => :debug)
+        else
+          setup_log
+        end
+      end
 
       # setup the database repositories
-      Database.repositories.each do |name,uri|
+      repositories.each do |name,uri|
         DataMapper.setup(name,uri)
       end
 
       # apply any new migrations to the database
-      Database.upgrade!
+      upgrade!
     end
 
     #
@@ -215,10 +235,12 @@ module Ronin
     #
     # @since 1.0.0
     #
+    # @api public
+    #
     def Database.repository(name,&block)
       name = name.to_sym
 
-      unless Database.repository?(name)
+      unless repository?(name)
         raise(UnknownRepository,"unknown database repository #{name}")
       end
 
@@ -242,10 +264,12 @@ module Ronin
     #
     # @since 1.0.0
     #
+    # @api private
+    #
     def Database.clear(name)
       name = name.to_sym
 
-      unless Database.repository?(name)
+      unless repository?(name)
         raise(UnknownRepository,"unknown database repository #{name}")
       end
 
@@ -268,10 +292,12 @@ module Ronin
     #
     # @since 1.0.0
     #
+    # @api public
+    #
     def Database.map
       results = []
 
-      Database.repositories.each_key do |name|
+      repositories.each_key do |name|
         DataMapper.repository(name) do
           result = yield
           results << result unless result.nil?

data/lib/ronin/database/migrations.rb

@@ -43,5 +43,7 @@ require 'ronin/database/migrations/create_countries_table'
 require 'ronin/database/migrations/create_organizations_table'
 require 'ronin/database/migrations/create_campaigns_table'
 require 'ronin/database/migrations/create_targets_table'
-require 'ronin/database/migrations/create_cached_files_table'
+require 'ronin/database/migrations/create_script_paths_table'
 require 'ronin/database/migrations/create_repositories_table'
+require 'ronin/database/migrations/add_updated_at_column_to_campaigns_table'
+require 'ronin/database/migrations/add_created_at_column_to_targets_table'

data/lib/ronin/database/migrations/add_created_at_column_to_targets_table.rb

@@ -0,0 +1,48 @@
+#
+# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This file is part of Ronin.
+#
+# Ronin 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 3 of the License, or
+# (at your option) any later version.
+#
+# Ronin 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 Ronin.  If not, see <http://www.gnu.org/licenses/>.
+#
+
+require 'ronin/database/migrations/create_targets_table'
+require 'ronin/database/migrations/migrations'
+require 'ronin/campaign'
+require 'ronin/target'
+
+module Ronin
+  module Database
+    module Migrations
+      migration(
+        :add_created_at_column_to_targets_table,
+        :needs => :create_targets_table
+      ) do
+        up do
+          modify_table :ronin_targets do
+            add_column :created_at, Time
+          end
+
+          # set the updated_at column to the created_at of the Campaign
+          Target.each do |target|
+            target.update(:created_at => target.campaign.created_at)
+          end
+        end
+
+        down do
+        end
+      end
+    end
+  end
+end

data/lib/ronin/database/migrations/add_updated_at_column_to_campaigns_table.rb

@@ -0,0 +1,47 @@
+#
+# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This file is part of Ronin.
+#
+# Ronin 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 3 of the License, or
+# (at your option) any later version.
+#
+# Ronin 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 Ronin.  If not, see <http://www.gnu.org/licenses/>.
+#
+
+require 'ronin/database/migrations/create_campaigns_table'
+require 'ronin/database/migrations/migrations'
+require 'ronin/campaign'
+
+module Ronin
+  module Database
+    module Migrations
+      migration(
+        :add_updated_at_column_to_campaigns_table,
+        :needs => :create_campaigns_table
+      ) do
+        up do
+          modify_table :ronin_campaigns do
+            add_column :updated_at, Time
+          end
+
+          # set the updated_at column to created_at
+          Campaign.each do |campaign|
+            campaign.update(:updated_at => campaign.created_at)
+          end
+        end
+
+        down do
+        end
+      end
+    end
+  end
+end

data/lib/ronin/database/migrations/create_licenses_table.rb

@@ -28,7 +28,7 @@ module Ronin
             column :id, Integer, :serial => true
             column :name, String, :not_null => true
             column :description, Text, :not_null => true
-            column :url, String
+            column :url, String, :length => 256
           end
 
           create_index :ronin_licenses, :name, :unique => true

data/lib/ronin/database/migrations/create_passwords_table.rb

@@ -26,7 +26,7 @@ module Ronin
         up do
           create_table :ronin_passwords do
             column :id, Integer, :serial => true
-            column :clear_text, String, :not_null => true
+            column :clear_text, String, :length => 256, :not_null => true
           end
 
           create_index :ronin_passwords, :clear_text, :unique => true

data/lib/ronin/database/migrations/{create_cached_files_table.rb → create_script_paths_table.rb}

@@ -24,21 +24,21 @@ module Ronin
   module Database
     module Migrations
       migration(
-        :create_cached_files_table,
+        :create_script_paths_table,
         :needs => :create_repositories_table
       ) do
         up do
-          create_table :ronin_cached_files do
+          create_table :ronin_script_paths do
             column :id, Integer, :serial => true
             column :path, FilePath, :not_null => true
             column :timestamp, Time, :not_null => true
-            column :model_name, String, :not_null => true
+            column :class_name, String, :not_null => true
             column :repository_id, Integer, :not_null => true
           end
         end
 
         down do
-          drop_table :ronin_cached_files
+          drop_table :ronin_script_paths
         end
       end
     end

data/lib/ronin/database/migrations/migration.rb

@@ -50,15 +50,12 @@ module Ronin
         # @option options [Array, Symbol] :needs ([])
         #   Other migrations that are dependencies of the migration.
         #
-        # @api semipublic
+        # @api private
         #
         def initialize(name,options={},&block)
-          options = options.merge(
-            :verbose => UI::Output.verbose?,
-            :needs => []
-          )
+          options[:verbose] = UI::Output.verbose?
 
-          @needs = Set[*options[:needs]]
+          @needs = Array(options.delete(:needs)).uniq
 
           super(0,name,options,&block)
         end

data/lib/ronin/database/migrations/migrations.rb

@@ -34,7 +34,7 @@ module Ronin
       #
       # @since 1.0.1
       #
-      # @api public
+      # @api private
       #
       def self.migrations
         @migrations ||= Graph.new
@@ -146,7 +146,7 @@ module Ronin
       #
       # @since 1.0.1
       #
-      # @api public
+      # @api private
       #
       def self.migrate_up!(position_or_name=nil)
         self.migrations.upto(position_or_name) do |migration|
@@ -171,7 +171,7 @@ module Ronin
       #
       # @since 1.0.1
       #
-      # @api public
+      # @api private
       #
       def self.migrate_down!(position_or_name=nil)
         self.migrations.downto(position_or_name) do |migration|