ronin 0.2.2 → 0.2.3

data/lib/ronin/database/database.rb

@@ -22,6 +22,7 @@
 #
 
 require 'ronin/database/exceptions/invalid_config'
+require 'ronin/model'
 require 'ronin/arch'
 require 'ronin/os'
 require 'ronin/author'
@@ -103,6 +104,25 @@ module Ronin
       return @@ronin_database_log = DataMapper::Logger.new(stream,level)
     end
 
+    #
+    # Returns +true+ if the Database is setup, returns +false+ otherwise.
+    #
+    def Database.setup?
+      repository = DataMapper.repository(Model::REPOSITORY_NAME)
+
+      return repository.class.adapters.has_key?(repository.name)
+    end
+
+    #
+    # Call the given _block_ then run auto-upgrades on the Database.
+    #
+    def Database.update!(&block)
+      block.call if block
+
+      DataMapper.auto_upgrade!(Model::REPOSITORY_NAME) if Database.setup?
+      return nil
+    end
+
     #
     # Sets up the Database with the given _configuration_. If
     # _configuration is not given, +DEFAULT_CONFIG+ will be used to setup
@@ -115,9 +135,7 @@ module Ronin
       # setup the database repository
       DataMapper.setup(Model::REPOSITORY_NAME, configuration)
 
-      block.call if block
-
-      DataMapper.auto_upgrade!(Model::REPOSITORY_NAME)
+      Database.update!(&block)
       return nil
     end
   end

data/lib/ronin/database/exceptions/invalid_config.rb

@@ -23,7 +23,7 @@
 
 module Ronin
   module Database
-    class InvalidConfig < RuntimeError
+    class InvalidConfig < StandardError
     end
   end
 end

data/lib/ronin/environment.rb

@@ -26,11 +26,14 @@ require 'ronin/extensions'
 require 'ronin/formatting'
 require 'ronin/path'
 require 'ronin/network'
-require 'ronin/models'
+require 'ronin/database'
 require 'ronin/ui'
 require 'ronin/ronin'
 
 require 'chars'
 require 'pp'
 
-Ronin::Config.load
+module Ronin
+  Config.load
+  Database.setup
+end

data/lib/ronin/extensions.rb

@@ -22,7 +22,9 @@
 #
 
 require 'ronin/extensions/meta'
+require 'ronin/extensions/kernel'
 require 'ronin/extensions/hash'
 require 'ronin/extensions/uri'
 require 'ronin/extensions/string'
 require 'ronin/extensions/file'
+require 'ronin/extensions/ip_addr'

data/lib/ronin/{models.rb → extensions/array.rb}

@@ -21,13 +21,26 @@
 #++
 #
 
-require 'ronin/database'
-require 'ronin/model'
+class Array
 
-require 'reverse_require'
+  #
+  # Returns the power set of the array. Shamelessly borrowed from
+  # http://johncarrino.net/blog/2006/08/11/powerset-in-ruby/.
+  #
+  #   [1,2,3].power_set
+  #   # => [[], [3], [2], [2, 3], [1], [1, 3], [1, 2], [1, 2, 3]]
+  #
+  def power_set
+    inject([[]]) do |power_set,element|
+      sub_set = []
 
-module Ronin
-  Database.setup do
-    require_for 'ronin', 'ronin/models'
+      power_set.each do |i|
+        sub_set << i
+        sub_set << i + [element]
+      end
+
+      sub_set
+    end
   end
+
 end

data/lib/ronin/extensions/ip_addr.rb

@@ -0,0 +1,127 @@
+#
+#--
+# 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 'ipaddr'
+
+class IPAddr
+
+  include Enumerable
+
+  #
+  # Iterates over the CIDR or globbed address range, passing each
+  # address in the range to the specified _block_. Supports both
+  # IPv4 and IPv6 address ranges.
+  #
+  #   IPAddr.each('10.1.1.1/24') do |ip|
+  #     ...
+  #   end
+  #
+  #   IPAddr.each('10.1.1-5.*') do |ip|
+  #     ...
+  #   end
+  #
+  #   IPAddr.each('::ff::02-0a::c3') do |ip|
+  #     ...
+  #   end
+  #
+  def IPAddr.each(cidr_or_glob,&block)
+    unless (cidr_or_glob.include?('*') || cidr_or_glob.include?('-'))
+      IPAddr.new(cidr_or_glob).each(&block)
+      return nil
+    end
+
+    if cidr_or_glob.include?('::')
+      prefix = if cidr_or_glob =~ /^::/
+                 '::'
+               else
+                 ''
+               end
+
+      separator = '::'
+      base = 16
+
+      format = lambda { |address|
+        prefix + address.map { |i| '%.2x' % i }.join('::')
+      }
+    else
+      separator = '.'
+      base = 10
+
+      format = lambda { |address| address.join('.') }
+    end
+
+    ranges = cidr_or_glob.split(separator).map { |segment|
+      if segment == '*'
+        (1..254)
+      elsif segment.include?('-')
+        start, stop = segment.split('-',2).map { |i| i.to_i(base) }
+
+        (start..stop)
+      elsif !(segment.empty?)
+        segment.to_i(base)
+      end
+    }.compact
+
+    expand_range = lambda { |address,remaining|
+      if remaining.empty?
+        block.call(format.call(address))
+      else
+        n = remaining.first
+        remaining = remaining[1..-1]
+
+        if n.kind_of?(Range)
+          n.each { |i| expand_range.call(address + [i], remaining) }
+        else
+          expand_range.call(address + [n], remaining)
+        end
+      end
+    }
+
+    expand_range.call([], ranges)
+    return nil
+  end
+
+  #
+  # Iterates over each IP address that is included in the addresses mask,
+  # passing each address to the specified _block_.
+  #
+  #   IPAddr.new('10.1.1.1/24').each do |ip|
+  #     puts ip
+  #   end
+  #
+  def each(&block)
+    case @family
+    when Socket::AF_INET
+      family_mask = IN4MASK
+    when Socket::AF_INET6
+      family_mask = IN6MASK
+    end
+
+    (0..((~@mask_addr) & family_mask)).each do |i|
+      block.call(_to_string(@addr | i))
+    end
+
+    return self
+  end
+
+end

data/lib/ronin/{objectify/exceptions/object_context_not_found.rb → extensions/kernel.rb}

@@ -21,9 +21,20 @@
 #++
 #
 
-module Ronin
-  module Objectify
-    class ObjectContextNotFound < RuntimeError
+module Kernel
+  #
+  # Calls the given _block_ and ignores any raised exceptions.
+  # If an exception is raised, +nil+ will be returned.
+  #
+  #   try do
+  #     Resolv.getaddress('might.not.exist.com')
+  #   end
+  #
+  def try(&block)
+    begin
+      block.call if block
+    rescue
+      return nil
     end
   end
 end

data/lib/ronin/extensions/uri/query_params.rb

@@ -21,6 +21,8 @@
 #++
 #
 
+require 'cgi'
+
 module URI
   module QueryParams
     # Query parameters
@@ -89,9 +91,9 @@ module URI
             "#{name}=active"
           elsif value
             if value.kind_of?(Array)
-              "#{name}=#{URI.encode(value.join(' '))}"
+              "#{name}=#{CGI.escape(value.join(' '))}"
             else
-              "#{name}=#{URI.encode(value.to_s)}"
+              "#{name}=#{CGI.escape(value.to_s)}"
             end
           else
             "#{name}="

data/lib/ronin/formatting/extensions/binary/integer.rb

@@ -32,6 +32,12 @@ class Integer
   # _endian_ must be either <tt>:little</tt>, <tt>:big</tt> or
   # <tt>:net</tt>.
   #
+  #   0xff41.bytes(2)
+  #   # => [65, 255]
+  #
+  #   0xff41.bytes(4, :big)
+  #   # => [0, 0, 255, 65]
+  #
   def bytes(address_length,endian=:little)
     endian = endian.to_s
     buffer = []

data/lib/ronin/formatting/extensions/binary/string.rb

@@ -40,19 +40,21 @@ class String
   #
   def depack(arch,address_length=arch.address_length)
     integer = 0x0
+    byte_index = 0
 
-    if arch.endian=='little'
-      address_length.times do |i|
-        if self[i]
-          integer = (integer | (self[i] << (i*8)))
-        end
-      end
-    elsif arch.endian=='big'
-      address_length.times do |i|
-        if self[i]
-          integer = (integer | (self[i] << ((address_length-i-1)*8)))
-        end
-      end
+    if arch.endian == 'little'
+      mask = lambda { |b| b << (byte_index * 8) }
+    elsif arch.endian == 'big'
+      mask = lambda { |b|
+        b << ((address_length - byte_index - 1) * 8)
+      }
+    end
+
+    self.each_byte do |b|
+      break if byte_index >= address_length
+
+      integer |= mask.call(b)
+      byte_index += 1
     end
 
     return integer

data/lib/ronin/formatting/extensions/text/string.rb

@@ -26,7 +26,7 @@ require 'chars'
 class String
 
   #
-  # Creates a new String by passing each character to the specified _block_
+  # Creates a new String by passing each byte to the specified _block_
   # using the given _options_.
   #
   # _options_ may include the following keys:
@@ -34,7 +34,7 @@ class String
   #                      defaults to <tt>Chars.all</tt>.
   # <tt>:excluded</tt>:: The characters not to format.
   #
-  def format_chars(options={},&block)
+  def format_bytes(options={},&block)
     included = (options[:included] || Chars.all)
     excluded = (options[:excluded] || [])
 
@@ -42,12 +42,10 @@ class String
     formatted = ''
 
     self.each_byte do |b|
-      c = b.chr
-
       if targeted.include_byte?(b)
-        formatted << block.call(c)
+        formatted << block.call(b)
       else
-        formatted << c
+        formatted << b
       end
     end
 
@@ -55,7 +53,7 @@ class String
   end
 
   #
-  # Creates a new String by passing each byte to the specified _block_
+  # Creates a new String by passing each character to the specified _block_
   # using the given _options_.
   #
   # _options_ may include the following keys:
@@ -63,15 +61,9 @@ class String
   #                      defaults to <tt>Chars.all</tt>.
   # <tt>:excluded</tt>:: The characters not to format.
   #
-  def format_bytes(options={},&block)
-    format_chars(options) do |c|
-      i = block.call(c[0])
-
-      if i.kind_of?(Integer)
-        i.chr
-      else
-        i.to_s
-      end
+  def format_chars(options={},&block)
+    format_bytes(options) do |b|
+      block.call(b.chr)
     end
   end
 
@@ -83,14 +75,14 @@ class String
   # <tt>:probability</tt>:: The probability that a character will have it's
   #                         case changed; defaults to 0.5.
   #
-  # "get out your checkbook".rand_case
-  # # => "gEt Out YOur CHEckbook"
+  #   "get out your checkbook".random_case
+  #   # => "gEt Out YOur CHEckbook"
   #
   def random_case(options={})
     prob = (options[:probability] || 0.5)
 
     format_chars(options) do |c|
-      if rand < prob
+      if rand <= prob
         c.swapcase 
       else
         c
@@ -98,4 +90,30 @@ class String
     end
   end
 
+  #
+  # Pads the string using the specified _padding_ out to the given
+  # _max_length_. _max_length_ will default to the strings own length,
+  # if not given.
+  #
+  #   "hello".pad('A',50)
+  #   # => "helloAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA"
+  #
+  def pad(padding,max_length=self.length)
+    padding = padding.to_s
+
+    if max_length >= self.length
+      max_length -= self.length
+    else
+      max_length = 0
+    end
+
+    padded = self + (padding * (max_length / padding.length))
+
+    unless (remaining = max_length % padding.length) == 0
+      padded += padding[0...remaining]
+    end
+
+    return padded
+  end
+
 end