ronin-support 0.2.0 → 0.3.0

data/.gitignore

@@ -0,0 +1,11 @@
+doc
+pkg
+vendor/cache
+Gemfile.lock
+.bundle
+.DS_Store
+.yardoc
+*.db
+*.log
+*.swp
+*~

data/ChangeLog.md

@@ -1,4 +1,45 @@
-### 0.2.0 / 2011-05-14
+### 0.3.0 / 2011-10-16
+
+* Require combinatorics ~> 0.4.
+* Added {Enumerable#map_hash}.
+* Added {String.generate}.
+* Added {String#fuzz}.
+* Added {File.each_line}.
+* Added {File.each_row}.
+* Added {Resolv.resolver}.
+* Added {URI::HTTP#request}.
+* Added {URI::HTTP#status}.
+* Added {URI::HTTP#ok?}.
+* Added {URI::HTTP#server}.
+* Added {URI::HTTP#powered_by}.
+* Added {URI::HTTP#copy}.
+* Added {URI::HTTP#delete}.
+* Added {URI::HTTP#get}.
+* Added {URI::HTTP#get_headers}.
+* Added {URI::HTTP#get_body}.
+* Added {URI::HTTP#head}.
+* Added {URI::HTTP#lock}.
+* Added {URI::HTTP#mkcol}.
+* Added {URI::HTTP#move}.
+* Added {URI::HTTP#options}.
+* Added {URI::HTTP#post}.
+* Added {URI::HTTP#post_headers}.
+* Added {URI::HTTP#post_body}.
+* Added {URI::HTTP#prop_find}.
+* Added {URI::HTTP#prop_match}.
+* Added {URI::HTTP#trace}.
+* Added {URI::HTTP#unlock}.
+* Added {Regexp::MAC}.
+* Added {Regexp::IPv6}, {Regexp::IPv4} and {Regexp::IP}.
+* Added {Regexp::HOST_NAME}.
+* Added {Regexp::USER_NAME}.
+* Added {Regexp::EMAIL_ADDR}.
+* Moved {Ronin::UI::Output}, {Ronin::UI::Shell} and {Ronin::Network::Mixins}
+  from ronin into ronin-support.
+* Refactored {Ronin::UI::Shell} into a Class where commands are defined as
+  protected methods.
+
+### 0.2.0 / 2011-07-04
 
 * Require data_paths ~> 0.3.
 * Added {Ronin::Mixin}.

data/README.md

@@ -25,6 +25,7 @@ or payloads over many common Source-Code-Management (SCM) systems.
     * URIs
     * HTML
     * JavaScript
+    * SQL
   * Generating random text.
   * Networking:
     * TCP
@@ -53,11 +54,13 @@ please see [Everyday Ronin](http://ronin-ruby.github.com/guides/everyday_ronin.h
 * [hexdump](http://github.com/postmodern/hexdump#readme)
   ~> 0.1
 * [combinatorics](http://github.com/postmodern/combinatorics#readme)
-  ~> 0.3
+  ~> 0.4
 * [uri-query_params](http://github.com/postmodern/uri-query_params#readme)
   ~> 0.5, >= 0.5.2
 * [data_paths](http://github.com/postmodern/data_paths#readme)
   ~> 0.3
+* [parameters](http://github.com/postmodern/parameters#readme)
+  ~> 0.2, >= 0.2.3
 
 ## Install
 

data/gemspec.yml

@@ -15,9 +15,10 @@ required_ruby_version: ">= 1.8.7"
 dependencies:
   chars: ~> 0.2
   hexdump: ~> 0.1
-  combinatorics: ~> 0.3
+  combinatorics: ~> 0.4
   uri-query_params: ~> 0.5, >= 0.5.2
   data_paths: ~> 0.3
+  parameters: ~> 0.2, >= 0.2.3
 
 development_dependencies:
   bundler: ~> 1.0.10

data/lib/ronin/extensions.rb

@@ -19,8 +19,10 @@
 
 require 'ronin/extensions/meta'
 require 'ronin/extensions/string'
+require 'ronin/extensions/regexp'
 require 'ronin/extensions/file'
 require 'ronin/extensions/ip_addr'
+require 'ronin/extensions/resolv'
 require 'ronin/extensions/kernel'
 
 require 'hexdump/extensions'

data/lib/ronin/extensions/enumerable.rb

@@ -0,0 +1,54 @@
+#
+# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This file is part of Ronin Support.
+#
+# Ronin Support is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Ronin Support 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 Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with Ronin Support.  If not, see <http://www.gnu.org/licenses/>.
+#
+
+module Enumerable
+
+  #
+  # Maps the elements to a Hash.
+  #
+  # @yield [element]
+  #   The given block will be passed each element.
+  #   The return value from the block will be stored in the hash.
+  #
+  # @yieldparam [Object] element
+  #   An element.
+  #
+  # @return [Hash{Object => Object}]
+  #   The hash of elements and their mappings.
+  #
+  # @example
+  #   host_names = %w[www.wired.com www.google.com]
+  #   host_names.map_hash { |name| Resolv.getaddresses(name) }
+  #   # {"www.wired.com"=>["184.84.183.17", "184.84.183.91"],
+  #   #  "www.google.com"=>["173.194.33.18", "173.194.33.17", "173.194.33.19", "173.194.33.20", "173.194.33.16"]}
+  #
+  # @since 0.3.0
+  #
+  # @api public
+  #
+  def map_hash
+    new_hash = Hash.new do |hash,key|
+      hash[key] = yield(key)
+    end
+
+    each { |element| new_hash[element] }
+    return new_hash
+  end
+
+end

data/lib/ronin/extensions/file.rb

@@ -19,6 +19,71 @@
 
 class File
 
+  #
+  # Reads each line from the file.
+  #
+  # @param [String] path
+  #   The path of the file.
+  #
+  # @yield [line]
+  #   The given block will be passed each line.
+  #
+  # @yieldparam [String] line
+  #   A line from the file, with the trailing newline characters removed.
+  #
+  # @return [Enumerator]
+  #   If no block is given, an Enumerator will be returned.
+  #
+  # @example
+  #   File.each_line('passwords.txt') do |line|
+  #     # ...
+  #   end
+  #
+  # @since 0.3.0
+  #
+  # @api public
+  #
+  def File.each_line(path)
+    return enum_for(:each_line,path) unless block_given?
+
+    File.open(path) do |file|
+      file.each_line { |line| yield line.chomp }
+    end
+  end
+
+  #
+  # Reads each row from the file.
+  #
+  # @param [String] path
+  #   The path of the file.
+  #
+  # @param [Regexp, String] separator
+  #   The pattern to split the line by.
+  #
+  # @yield [row]
+  #   The given block will be passed each row.
+  #
+  # @yieldparam [Array<String>] row
+  #   A row from the file.
+  #
+  # @return [Enumerator]
+  #   If no block is given, an Enumerator will be returned.
+  #
+  # @example
+  #   File.each_row('db_dump.txt', '|') do |row|
+  #     # ...
+  #   end
+  #
+  # @since 0.3.0
+  #
+  # @api public
+  #
+  def File.each_row(path,separator=/\s+/)
+    return enum_for(:each_row,path,separator) unless block_given?
+
+    File.each_line(path) { |line| yield line.split(separator) }
+  end
+
   #
   # Writes the given data to a specified path.
   #
@@ -30,6 +95,9 @@ class File
   #
   # @return [nil]
   #
+  # @example
+  #   File.write('dump.txt',data)
+  #
   # @api public
   #
   def File.write(path,data)
@@ -51,10 +119,10 @@ class File
     path = path.to_s
 
     # remove any \0 characters first
-    path.gsub!("\0",'')
+    path.tr!("\0",'')
 
     # remove any home-dir expansions
-    path.gsub!("~","\\~")
+    path.gsub!('~',"\\~")
 
     path = File.expand_path(File.join('/',path))
 

data/lib/ronin/extensions/ip_addr.rb

@@ -17,8 +17,10 @@
 # along with Ronin Support.  If not, see <http://www.gnu.org/licenses/>.
 #
 
+require 'ronin/extensions/resolv'
+require 'ronin/extensions/regexp'
+
 require 'ipaddr'
-require 'resolv'
 require 'strscan'
 require 'combinatorics/list_comprehension'
 
@@ -26,14 +28,11 @@ class IPAddr
 
   include Enumerable
 
-  # A regular expression for matching IPv4 Addresses.
-  IPV4_REGEXP = /[\d]{1,3}\.[\d]{1,3}\.[\d]{1,3}\.[\d]{1,3}/
-
-  # A regular expression for matching IPv6 Addresses.
-  IPV6_REGEXP = /:(:[0-9a-f]{1,4}){1,7}|([0-9a-f]{1,4}::?){1,7}[0-9a-f]{1,4}(:#{IPV4_REGEXP})?/
-
-  # A regular expression for matching IP Addresses.
-  REGEXP = /#{IPV4_REGEXP}|#{IPV6_REGEXP}/
+  # Socket families and IP address masks
+  MASKS = {
+    Socket::AF_INET  => IN4MASK,
+    Socket::AF_INET6 => IN6MASK
+  }
 
   #
   # Extracts IP Addresses from text.
@@ -41,8 +40,8 @@ class IPAddr
   # @param [String] text
   #   The text to scan for IP Addresses.
   #
-  # @param [Symbol] version
-  #   The version of IP Address to scan for (`:ipv4` or `:ipv6`).
+  # @param [Integer, Symbol] version
+  #   The version of IP Address to scan for (`4`, `6`, `:v4` or `:v6`).
   #
   # @yield [ip]
   #   The given block will be passed each extracted IP Address.
@@ -53,29 +52,36 @@ class IPAddr
   # @return [Array<String>]
   #   The IP Addresses found in the text.
   #
+  # @example
+  #   IPAddr.extract("Host: 127.0.0.1\n\rHost: 10.1.1.1\n\r")
+  #   # => ["127.0.0.1", "10.1.1.1"]
+  #
+  # @example Extract only IPv4 addresses from a large amount of text.
+  #   IPAddr.extract(text,:v4) do |ip|
+  #     puts ip
+  #   end
+  #   
   # @api public
   #
   def IPAddr.extract(text,version=nil,&block)
+    return enum_for(:extract,text,version).to_a unless block_given?
+
     regexp = case version
-             when :ipv4
-               IPV4_REGEXP
-             when :ipv6
-               IPV6_REGEXP
+             when :ipv4, :v4, 4
+               Regexp::IPv4
+             when :ipv6, :v6, 6
+               Regexp::IPv6
              else
-               REGEXP
+               Regexp::IP
              end
 
-    parser = StringScanner.new(text)
+    scanner = StringScanner.new(text)
 
-    if block_given?
-      yield parser.matched while parser.skip_until(regexp)
-      return nil
-    else
-      ips = []
-
-      ips << parser.matched while parser.skip_until(regexp)
-      return ips
+    while scanner.skip_until(regexp)
+      yield scanner.matched
     end
+
+    return nil
   end
 
   #
@@ -120,28 +126,27 @@ class IPAddr
     return enum_for(:each,cidr_or_glob) unless block
 
     if cidr_or_glob.include?('::')
-      prefix = if cidr_or_glob[0,2] == '::'
+      separator = '::'
+      base      = 16
+
+      prefix = if cidr_or_glob.start_with?('::')
                  '::'
                else
                  ''
                end
 
-      separator = '::'
-      base = 16
-
       format = lambda { |address|
         prefix + address.map { |i| '%.2x' % i }.join('::')
       }
     else
       separator = '.'
-      base = 10
-
-      format = lambda { |address| address.join('.') }
+      base      = 10
+      format    = lambda { |address| address.join('.') }
     end
 
     # split the address
     segments = cidr_or_glob.split(separator)
-    ranges = []
+    ranges   = []
     
     # map the components of the address to numeric ranges
     segments.each do |segment|
@@ -159,23 +164,23 @@ class IPAddr
     end
 
     # cycle through the address ranges
-    ranges.comprehension do |address|
-      yield format[address]
-    end
-
+    ranges.comprehension { |address| yield format[address] }
     return nil
   end
 
   #
   # Resolves the host-names for the IP address.
   #
+  # @param [String] nameserver
+  #   The optional nameserver to query.
+  #
   # @return [Array<String>]
   #   The host-names for the IP address.
   #
   # @api public
   #
-  def lookup
-    Resolv.getnames(self.to_s)
+  def lookup(nameserver=nil)
+    Resolv.resolver(nameserver).getnames(self.to_s)
   end
 
   #
@@ -201,12 +206,7 @@ class IPAddr
   def each
     return enum_for(:each) unless block_given?
 
-    case @family
-    when Socket::AF_INET
-      family_mask = IN4MASK
-    when Socket::AF_INET6
-      family_mask = IN6MASK
-    end
+    family_mask = MASKS[@family]
 
     (0..((~@mask_addr) & family_mask)).each do |i|
       yield _to_string(@addr | i)

data/lib/ronin/extensions/regexp.rb

@@ -0,0 +1,45 @@
+#
+# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This file is part of Ronin Support.
+#
+# Ronin Support is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Lesser General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Ronin Support 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 Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with Ronin Support.  If not, see <http://www.gnu.org/licenses/>.
+#
+
+require 'ronin/extensions/resolv'
+
+class Regexp
+
+  # Regular expression for finding MAC addresses in text
+  MAC = /[0-9a-fA-F]{2}(?::[0-9a-fA-F]{2}){5}/
+
+  # A regular expression for matching IPv4 Addresses.
+  IPv4 = /[\d]{1,3}\.[\d]{1,3}\.[\d]{1,3}\.[\d]{1,3}/
+
+  # A regular expression for matching IPv6 Addresses.
+  IPv6 = /:(:[0-9a-f]{1,4}){1,7}|([0-9a-f]{1,4}::?){1,7}[0-9a-f]{1,4}(:#{IPv4})?/
+
+  # A regular expression for matching IP Addresses.
+  IP = /#{IPv4}|#{IPv6}/
+
+  # Regular expression used to find host-names in text
+  HOST_NAME = /(?:[a-zA-Z0-9]+(?:[_-][a-zA-Z0-9]+)*\.)+(?:#{union(Resolv::TLDS)})/
+
+  # Regular expression to match a word in the username of an email address
+  USER_NAME = /[A-Za-z](?:[A-Za-z0-9]+[\._-])*[A-Za-z0-9]+/
+
+  # Regular expression to find email addresses in text
+  EMAIL_ADDR = /#{USER_NAME}(?:\.#{USER_NAME})*\@#{HOST_NAME}/
+
+end