ronin 0.1.2 → 0.1.3

data/lib/ronin/cache/extension_cache.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
@@ -37,7 +37,7 @@ module Ronin
           hash[name] = load_extension(key.to_s)
         end
 
-        catch('EXIT') do
+        at_exit do
           each_extension { |ext| ext.perform_teardown }
         end
 

data/lib/ronin/cache/maintainer.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/cache/overlay.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
@@ -262,7 +262,7 @@ module Ronin
       def cache_objects
         require 'ronin/models'
 
-        return ObjectContext.cache_objects_in(objects_dir)
+        return Objectify.cache_objects_in(objects_dir)
       end
 
       #
@@ -271,7 +271,7 @@ module Ronin
       def mirror_objects
         require 'ronin/models'
 
-        return ObjectContext.mirror_objects_in(objects_dir)
+        return Objectify.mirror_objects_in(objects_dir)
       end
 
       #
@@ -280,7 +280,7 @@ module Ronin
       def expunge_objects
         require 'ronin/models'
 
-        return ObjectContext.expunge_objects_from(objects_dir)
+        return Objectify.expunge_objects_from(objects_dir)
       end
 
       #
@@ -338,7 +338,7 @@ module Ronin
       # Returns the paths of all extensions within the overlay.
       #
       def extension_paths
-        directories
+        directories.reject { |dir| File.basename(dir) == 'objects' }
       end
 
       #
@@ -375,9 +375,10 @@ module Ronin
       # If a _block_ is given, it will be passed the newly created
       # extension.
       #
-      #   repo.extension('awesome') # => Extension
+      #   overlay.extension('awesome')
+      #   # => #<Ronin::Cache::Extension: ...>
       #
-      #   repo.extension('shellcode') do |ext|
+      #   overlay.extension('shellcode') do |ext|
       #     ...
       #   end
       #

data/lib/ronin/cache/overlay_cache.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/cache/ronin.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/chars.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/chars/char_set.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/chars/chars.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/code/emittable.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/code/reference.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/code/symbol_table.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/code/token.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/config.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
@@ -25,17 +25,20 @@ require 'fileutils'
 
 module Ronin
   module Config
+    # The users home directory
+    HOME = File.expand_path(ENV['HOME'] || ENV['HOMEPATH'])
+
     # Ronin home directory
-    PATH = FileUtils.mkdir_p(File.join(ENV['HOME'],'.ronin'))
+    PATH = FileUtils.mkdir_p(File.join(HOME,'.ronin'))
 
     # Path to static directory
     STATIC_DIR = File.expand_path(File.join(File.dirname(__FILE__),'..','..','static'))
 
     # Main configuration file
-    CONFIG_PATH = File.expand_path(File.join(PATH,'config.rb'))
+    CONFIG_PATH = File.join(PATH,'config.rb')
 
     # Configuration files directory
-    CONFIG_DIR = File.expand_path(File.join(PATH,'config'))
+    CONFIG_DIR = FileUtils.mkdir_p(File.join(PATH,'config'))
 
     # Temporary file directory
     TMP_DIR = FileUtils.mkdir_p(File.join(PATH,'tmp'))
@@ -45,15 +48,19 @@ module Ronin
     # Ronin configuration files directory. If _name_ is not given, than the
     # main Ronin configuration file will be loaded.
     #
-    #   # Load the main config file at <tt>~/.ronin/config.rb</tt>
-    #   Config.load # => true
+    # Load the main config file at <tt>~/.ronin/config.rb</tt>
+    #
+    #   Config.load
+    #   # => true
+    #
+    # Load a specific config file in <tt>~/.ronin/config/</tt>
     #
-    #   # Load a specific config file in <tt>~/.ronin/config/</tt>
-    #   Config.load 'sql' # => true
+    #   Config.load :sql
+    #   # => true
     #
     def Config.load(name=nil)
       if name
-        path = File.expand_path(File.join(CONFIG_DIR,name))
+        path = File.expand_path(File.join(CONFIG_DIR,name.to_s))
       else
         path = CONFIG_PATH
       end

data/lib/ronin/database.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
@@ -21,115 +21,4 @@
 #++
 #
 
-require 'ronin/exceptions/invalid_database_config'
-require 'ronin/arch'
-require 'ronin/platform'
-require 'ronin/author'
-require 'ronin/license'
-require 'ronin/product'
-require 'ronin/config'
-
-require 'yaml'
-require 'dm-core'
-
-module Ronin
-  module Database
-    # Database configuration file
-    CONFIG_FILE = File.join(Config::PATH,'database.yml')
-
-    # Database log file
-    DEFAULT_LOG_PATH = File.join(Config::PATH,'database.log')
-
-    # Database log level
-    DEFAULT_LOG_LEVEL = :info
-
-    # Default configuration of the database
-    DEFAULT_CONFIG = {
-      :adapter => :sqlite3,
-      :database => File.join(Config::PATH,'database.sqlite3')
-    }
-
-    #
-    # Returns the Database configuration that is stored in the
-    # +CONFIG_FILE+. Defaults to +DEFAULT_CONFIG+ if +CONFIG_FILE+ does not
-    # exist.
-    #
-    def Database.config
-      unless (class_variable_defined?('@@ronin_database_config'))
-        @@ronin_database_config = DEFAULT_CONFIG
-
-        if File.file?(CONFIG_FILE)
-          conf = YAML.load(CONFIG_FILE)
-
-          unless (conf.kind_of?(Hash) || conf.kind_of?(String))
-            raise(InvalidDatabaseConfig,"#{CONFIG_FILE} must contain either a Hash or a String",caller)
-          end
-
-          @@ronin_database_config = conf
-        end
-      end
-
-      return @@ronin_database_config ||= DEFAULT_CONFIG
-    end
-
-    #
-    # Sets the Database configuration to the specified _configuration_.
-    #
-    def Database.config=(configuration)
-      @@ronin_database_config = configuration
-    end
-
-    #
-    # Returns the current Database log.
-    #
-    def Database.log
-      @@ronin_database_log ||= nil
-    end
-
-    #
-    # Setup the Database log with the given _options_.
-    #
-    # _options_ may contain the following keys:
-    # <tt>:path</tt>:: The path of the log file. Defaults to
-    #                  +DEFAULT_LOG_PATH+.
-    # <tt>:stream</tt>:: The stream to use for the log.
-    # <tt>:level</tt>:: The level of messages to log.
-    #
-    def Database.setup_log(options={})
-      path = (options[:path] || DEFAULT_LOG_PATH)
-      stream = (options[:stream] || File.new(path,'w+'))
-      level = (options[:level] || DEFAULT_LOG_LEVEL)
-
-      return @@ronin_database_log = DataMapper::Logger.new(stream,level)
-    end
-
-    #
-    # Sets up the Database with the given _configuration_. If
-    # _configuration is not given, +DEFAULT_CONFIG+ will be used to setup
-    # the Database.
-    #
-    def Database.setup(configuration=Database.config,&block)
-      # setup the database log
-      Database.setup_log unless Database.log
-
-      # setup the database repository
-      DataMapper.setup(Model::REPOSITORY_NAME, configuration)
-
-      block.call if block
-
-      # sourced from http://gist.github.com/3010
-      # in order to fix a has-many lazy-loading bug
-      # in dm-core <= 0.9.4
-      descendants = DataMapper::Resource.descendants.dup
-      descendants.each do |model|
-        descendants.merge(model.descendants) if model.respond_to?(:descendants)
-      end
-      descendants.each do |model|
-        model.relationships.each_value { |r| r.child_key if r.child_model == model }
-      end
-
-      DataMapper.auto_upgrade!(Model::REPOSITORY_NAME)
-      return nil
-    end
-  end
-end
+require 'ronin/database/database'

data/lib/ronin/database/database.rb

@@ -0,0 +1,124 @@
+#
+#--
+# 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 'ronin/database/exceptions/invalid_config'
+require 'ronin/arch'
+require 'ronin/platform'
+require 'ronin/author'
+require 'ronin/license'
+require 'ronin/product'
+require 'ronin/config'
+
+require 'yaml'
+require 'dm-core'
+
+module Ronin
+  module Database
+    # Database configuration file
+    CONFIG_FILE = File.join(Config::PATH,'database.yml')
+
+    # Database log file
+    DEFAULT_LOG_PATH = File.join(Config::PATH,'database.log')
+
+    # Database log level
+    DEFAULT_LOG_LEVEL = :info
+
+    # Default configuration of the database
+    DEFAULT_CONFIG = {
+      :adapter => :sqlite3,
+      :database => File.join(Config::PATH,'database.sqlite3')
+    }
+
+    #
+    # Returns the Database configuration that is stored in the
+    # +CONFIG_FILE+. Defaults to +DEFAULT_CONFIG+ if +CONFIG_FILE+ does not
+    # exist.
+    #
+    def Database.config
+      unless (class_variable_defined?('@@ronin_database_config'))
+        @@ronin_database_config = DEFAULT_CONFIG
+
+        if File.file?(CONFIG_FILE)
+          conf = YAML.load(CONFIG_FILE)
+
+          unless (conf.kind_of?(Hash) || conf.kind_of?(String))
+            raise(InvalidConfig,"#{CONFIG_FILE} must contain either a Hash or a String",caller)
+          end
+
+          @@ronin_database_config = conf
+        end
+      end
+
+      return @@ronin_database_config ||= DEFAULT_CONFIG
+    end
+
+    #
+    # Sets the Database configuration to the specified _configuration_.
+    #
+    def Database.config=(configuration)
+      @@ronin_database_config = configuration
+    end
+
+    #
+    # Returns the current Database log.
+    #
+    def Database.log
+      @@ronin_database_log ||= nil
+    end
+
+    #
+    # Setup the Database log with the given _options_.
+    #
+    # _options_ may contain the following keys:
+    # <tt>:path</tt>:: The path of the log file. Defaults to
+    #                  +DEFAULT_LOG_PATH+.
+    # <tt>:stream</tt>:: The stream to use for the log.
+    # <tt>:level</tt>:: The level of messages to log.
+    #
+    def Database.setup_log(options={})
+      path = (options[:path] || DEFAULT_LOG_PATH)
+      stream = (options[:stream] || File.new(path,'w+'))
+      level = (options[:level] || DEFAULT_LOG_LEVEL)
+
+      return @@ronin_database_log = DataMapper::Logger.new(stream,level)
+    end
+
+    #
+    # Sets up the Database with the given _configuration_. If
+    # _configuration is not given, +DEFAULT_CONFIG+ will be used to setup
+    # the Database.
+    #
+    def Database.setup(configuration=Database.config,&block)
+      # setup the database log
+      Database.setup_log unless Database.log
+
+      # setup the database repository
+      DataMapper.setup(Model::REPOSITORY_NAME, configuration)
+
+      block.call if block
+
+      DataMapper.auto_upgrade!(Model::REPOSITORY_NAME)
+      return nil
+    end
+  end
+end