ronin 0.3.0 → 1.0.0.pre1

data/lib/ronin/ui/shell.rb

@@ -1,7 +1,7 @@
 #
 # Ronin - A Ruby platform for exploit development and security research.
 #
-# Copyright (c) 2006-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2010 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
@@ -18,106 +18,71 @@
 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 #
 
+require 'ronin/ui/output/helpers'
+
 module Ronin
-  module Shell
+  module UI
+    module Shell
+
+      extend Output::Helpers
 
-    # Default shell prompt
-    DEFAULT_PROMPT = '>'
+      # Default shell prompt
+      DEFAULT_PROMPT = '>'
 
-    #
-    # Creates and starts a new Shell object with the specified _options_.
-    # If a _block_ is given, it will be passed every command.
-    #
-    # @param [Hash] options
-    #   Additional options.
-    #
-    # @option options [String] :name ('')
-    #   The shell-name to use before the prompt.
-    #
-    # @option options [String] :prompt (DEFAULT_PROMPT)
-    #   The prompt to use for the shell.
-    #
-    # @yield [shell, line]
-    #   The block that will be passed every command entered.
-    #
-    # @yieldparam [Shell] shell
-    #   The shell to use for output.
-    #
-    # @yieldparam [String] line
-    #   The command entered into the shell.
-    #
-    # @example
-    #   Shell.start(:prompt => '$') { |shell,line| system(line) }
-    #
-    def Shell.start(options={},&block)
-      name = (options[:name] || '')
-      prompt = (options[:prompt] || DEFAULT_PROMPT)
+      #
+      # Creates a new Shell object and starts it.
+      #
+      # @param [Hash] options
+      #   Additional options.
+      #
+      # @option options [String] :name ('')
+      #   The shell-name to use before the prompt.
+      #
+      # @option options [String] :prompt (DEFAULT_PROMPT)
+      #   The prompt to use for the shell.
+      #
+      # @yield [shell, line]
+      #   The block that will be passed every command entered.
+      #
+      # @yieldparam [Shell] shell
+      #   The shell to use for output.
+      #
+      # @yieldparam [String] line
+      #   The command entered into the shell.
+      #
+      # @return [nil]
+      #
+      # @example
+      #   Shell.start(:prompt => '$') { |shell,line| system(line) }
+      #
+      def Shell.start(options={},&block)
+        name = (options[:name] || '')
+        prompt = (options[:prompt] || DEFAULT_PROMPT)
 
-      history_rollback = 0
+        history_rollback = 0
 
-      loop do
-        line = Readline.readline("#{name}#{prompt} ")
+        loop do
+          raw_line = Readline.readline("#{name}#{prompt} ")
+          line = raw_line.strip
 
-        if line =~ /^\s*exit\s*$/
-          break
-        else
-          Readline::HISTORY << line
-          history_rollback += 1
+          if line =~ /^(exit|quit)$/
+            break
+          elsif !(line.empty?)
+            Readline::HISTORY << raw_line
+            history_rollback += 1
 
-          begin
-            block.call(self,line)
-          rescue => e
-            puts "#{e.class.name}: #{e.message}"
+            begin
+              yield self, line
+            rescue => e
+              puts "#{e.class.name}: #{e.message}"
+            end
           end
         end
-      end
-
-      history_rollback.times { Readline::HISTORY.pop }
-      return nil
-    end
-
-    #
-    # Print a character to the shell.
-    #
-    # @param [String] char
-    #   The character to print.
-    #
-    def Shell.putc(char)
-      STDOUT.putc(char)
-    end
-
-    #
-    # Print a String to the shell.
-    #
-    # @param [String] string
-    #   The String to print.
-    #
-    def Shell.print(string)
-      STDOUT.print(string)
-    end
 
-    #
-    # Print a String and a new-line character to the shell.
-    #
-    # @param [String] string
-    #   The String to print.
-    #
-    def Shell.puts(string)
-      STDOUT.puts(string)
-    end
+        history_rollback.times { Readline::HISTORY.pop }
+        return nil
+      end
 
-    #
-    # Render the format-string and print the result to the shell.
-    #
-    # @param [String] string
-    #   The format-string to render.
-    #
-    # @param [Array] objects
-    #   Additional objects to use in the format-string.
-    #
-    def Shell.printf(string,*objects)
-      STDOUT.printf(string,*objects)
     end
-
   end
 end

data/lib/ronin/ui.rb

@@ -1,7 +1,7 @@
 #
 # Ronin - A Ruby platform for exploit development and security research.
 #
-# Copyright (c) 2006-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2010 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/url.rb

@@ -0,0 +1,343 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2009-2010 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/url_scheme'
+require 'ronin/url_query_param'
+require 'ronin/host_name'
+require 'ronin/tcp_port'
+require 'ronin/web_credential'
+require 'ronin/model'
+
+require 'dm-timestamps'
+require 'dm-tags'
+require 'uri'
+require 'uri/query_params'
+
+module Ronin
+  class URL
+
+    include Model
+    include DataMapper::Timestamps
+
+    # Mapping of URL Schemes and URI classes
+    SCHEMES = {
+      'https' => ::URI::HTTPS,
+      'http' => ::URI::HTTP,
+      'ftp' => ::URI::FTP
+    }
+
+    # Primary key of the URL
+    property :id, Serial
+
+    # The scheme of the URL
+    belongs_to :scheme, :model => 'URLScheme'
+
+    # The host name of the URL
+    belongs_to :host_name
+
+    # Port of the URL
+    belongs_to :port, :model => 'TCPPort'
+
+    # Path of the URL
+    property :path, String, :default => ''
+
+    # The fragment of the URL
+    property :fragment, String
+
+    # The query params of the URL
+    has 0..n, :query_params, :model => 'URLQueryParam'
+
+    # Any credentials used with the URL
+    has 0..n, :web_credentials
+
+    # When the URL was last scanned
+    property :last_scanned_at, Time
+
+    # Defines the created_at timestamp
+    timestamps :created_at
+
+    # Tags
+    has_tags_on :tags
+
+    #
+    # Searches for all URLs using HTTP.
+    #
+    # @return [Array<URL>]
+    #   The matching URLs.
+    #
+    # @since 1.0.0
+    #
+    def self.http
+      all('scheme.name' => 'http')
+    end
+
+    #
+    # Searches for all URLs using HTTPS.
+    #
+    # @return [Array<URL>]
+    #   The matching URLs.
+    #
+    # @since 1.0.0
+    #
+    def self.https
+      all('scheme.name' => 'https')
+    end
+
+    #
+    # Searches for URLs with specific host name(s).
+    #
+    # @param [Array<String>, String] names
+    #   The host name(s) to search for.
+    #
+    # @return [Array<URL>]
+    #   The matching URLs.
+    #
+    # @since 1.0.0
+    #
+    def self.hosts(names)
+      all('host.address' => names)
+    end
+
+    #
+    # Searches for URLs with the specific port number(s).
+    #
+    # @param [Array<Integer>, Integer] numbers
+    #   The port numbers to search for.
+    #
+    # @return [Array<URL>]
+    #   The matching URLs.
+    #
+    # @since 1.0.0
+    #
+    def self.ports(numbers)
+      all('port.number' => numbers)
+    end
+
+    #
+    # Searches for all URLs sharing a common sub-directory.
+    #
+    # @param [String] sub_dir
+    #   The sub-directory to search for.
+    #
+    # @return [Array<URL>]
+    #   The URL with the common sub-directory.
+    #
+    # @since 1.0.0
+    #
+    def self.directory(sub_dir)
+      all(:path => sub_dir) | all(:path.like => "#{sub_dir}/%")
+    end
+
+    #
+    # Searches for all URLs with a common file-extension.
+    #
+    # @param [String] ext
+    #   The file extension to search for.
+    #
+    # @return [Array<URL>]
+    #   The URLs with the common file-extension.
+    #
+    # @since 1.0.0
+    #
+    def self.extension(ext)
+      all(:path => "%.#{ext}")
+    end
+
+    #
+    # Searches for URLs with the given query-param.
+    #
+    # @param [Array<String>, String] name
+    #   The query-param name to search for.
+    #
+    # @return [Array<URL>]
+    #   The URLs with the given query-param.
+    #
+    # @since 1.0.0
+    #
+    def self.query_param(name)
+      all('query_params.name' => name)
+    end
+
+    #
+    # Search for all URLs with a given query-param value.
+    #
+    # @param [Array<String>, String] value
+    #   The query-param value to search for.
+    #
+    # @return [Array<URL>]
+    #   The URLs with the given query-param value.
+    #
+    # @since 1.0.0
+    #
+    def self.query_value(value)
+      all('query_params.value' => value)
+    end
+
+    #
+    # Creates a new URL.
+    #
+    # @param [URI::HTTP] uri
+    #   The URI to create the URL from.
+    #
+    # @return [URL]
+    #   The new URL.
+    #
+    # @since 1.0.0
+    #
+    def URL.from(uri)
+      new_url = URL.new(
+        :scheme => URLScheme.first_or_new(:name => uri.scheme),
+        :host_name => HostName.first_or_new(:address => uri.host),
+        :port => TCPPort.first_or_new(:number => uri.port),
+        :path => uri.path,
+        :fragment => uri.fragment
+      )
+
+      if uri.respond_to?(:query_params)
+        uri.query_params.each do |name,value|
+          new_url.query_params.new(:name => name, :value => value)
+        end
+      end
+
+      return new_url
+    end
+
+    #
+    # Parses the URL.
+    #
+    # @param [String] url
+    #   The raw URL to parse.
+    #
+    # @return [URL]
+    #   The parsed URL.
+    #
+    # @see URL.from
+    #
+    # @since 1.0.0
+    #
+    def URL.parse(url)
+      URL.from(::URI.parse(url))
+    end
+
+    #
+    # The host name of the url.
+    #
+    # @return [String]
+    #   The address of host name.
+    #
+    # @since 1.0.0
+    #
+    def host
+      self.host_name.address
+    end
+
+    #
+    # The port number used by the URL.
+    #
+    # @return [Integer]
+    #   The port number.
+    #
+    # @since 1.0.0
+    #
+    def port_number
+      self.port.number
+    end
+
+    #
+    # Dumps the URL query params into a URI query string.
+    #
+    # @return [String]
+    #   The URI query string.
+    #
+    # @since 1.0.0
+    #
+    def query_string
+      params = {}
+
+      self.query_params.each do |param|
+        params[param.name] = param.value
+      end
+
+      return URI::QueryParams.dump(params)
+    end
+
+    #
+    # Sets the query params of the URL.
+    #
+    # @param [String] query
+    #   The query string to parse.
+    #
+    # @return [String]
+    #   The given query string.
+    #
+    # @since 1.0.0
+    #
+    def query_string=(query)
+      self.query_params.clear
+
+      URI::QueryParams.parse(query).each do |name,value|
+        self.query_params.new(:name => name, :value => value)
+      end
+
+      return query
+    end
+
+    #
+    # Builds a URI object from the url.
+    #
+    # @return [URI::HTTP, URI::HTTPS]
+    #   The URI object created from the url attributes.
+    #
+    # @since 1.0.0
+    #
+    def to_uri
+      url_class = (SCHEMES[self.scheme.name] || ::URI::Generic)
+
+      host = if self.host_name
+               self.host_name.address
+             end
+      port = if self.port
+               self.port.number
+             end
+
+      return url_class.build(
+        :scheme => self.scheme.name,
+        :host => host,
+        :port => port,
+        :path => self.path,
+        :query => self.query_string,
+        :fragment => self.fragment
+      )
+    end
+
+    #
+    # Converts the url to a String.
+    #
+    # @return [String]
+    #   The string form of the url.
+    #
+    # @since 1.0.0
+    #
+    def to_s
+      self.to_uri.to_s
+    end
+
+  end
+end

data/lib/ronin/url_query_param.rb

@@ -0,0 +1,40 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 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/model/has_name'
+require 'ronin/model'
+
+module Ronin
+  class URLQueryParam
+
+    include Model
+    include Model::HasName
+
+    # The primary-key of the URL query param
+    property :id, Serial
+
+    # The value of the URL query param
+    property :value, Text
+
+    # The URL
+    belongs_to :url, :model => 'URL'
+
+  end
+end

data/lib/ronin/url_scheme.rb

@@ -0,0 +1,37 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 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/model'
+require 'ronin/model/has_unique_name'
+
+module Ronin
+  class URLScheme
+
+    include Model
+    include Model::HasUniqueName
+
+    # primary key of the URL Scheme
+    property :id, Serial
+
+    # The URLs that use the scheme
+    has 1..n, :urls, :model => 'URL'
+
+  end
+end

data/lib/ronin/user_name.rb

@@ -0,0 +1,45 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 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/model'
+require 'ronin/model/has_unique_name'
+
+require 'dm-timestamps'
+
+module Ronin
+  class UserName
+
+    include Model
+    include Model::HasUniqueName
+
+    # The primary key of the user-name
+    property :id, Serial
+
+    # Tracks when the user-name was created.
+    timestamps :created_at
+
+    # Any credentials belonging to the user
+    has 0..n, :credentials
+
+    # Email addresses of the user
+    has 0..n, :email_addresses
+
+  end
+end

data/lib/ronin/vendor.rb

@@ -0,0 +1,40 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 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/model'
+require 'ronin/model/has_unique_name'
+
+module Ronin
+  #
+  # Represents a Software Vendor.
+  #
+  class Vendor
+
+    include Model
+    include Model::HasUniqueName
+
+    # The primary-key of the vendor
+    property :id, Serial
+
+    # Products published by the vendor
+    has 0..n, :software, :model => 'Software'
+
+  end
+end

data/lib/ronin/version.rb

@@ -1,7 +1,7 @@
 #
 # Ronin - A Ruby platform for exploit development and security research.
 #
-# Copyright (c) 2006-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2010 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
@@ -20,5 +20,5 @@
 
 module Ronin
   # Ronin version 
-  VERSION = '0.3.0'
+  VERSION = '1.0.0.pre1'
 end