ronin-support 0.5.0.rc1 → 0.5.0.rc2

data/lib/ronin/fuzzing/repeater.rb

@@ -0,0 +1,81 @@
+#
+# Copyright (c) 2006-2012 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 Ronin
+  module Fuzzing
+    #
+    # Fuzzing class that generates repeated data.
+    #
+    # @api semipublic
+    #
+    # @since 0.5.0
+    #
+    class Repeater
+
+      # The lengths to repeat the data by
+      attr_reader :lengths
+
+      #
+      # Initializes a new Repeater.
+      #
+      # @param [Enumerable, Integer] lengths
+      #   The lengths to repeat the data by.
+      #
+      # @raise [TypeError]
+      #   `lengths` must either be Enumerable or an Integer.
+      #
+      def initialize(lengths)
+        @lengths = case lengths
+                   when Integer
+                     [lengths]
+                   when Enumerable
+                     lengths
+                   else
+                     raise(TypeError,"argument must be Enumerable or an Integer")
+                   end
+      end
+
+      #
+      # Enumerates through each length of repeated data.
+      #
+      # @param [#*] repeatable
+      #   The repeatable data.
+      #
+      # @yield [repeated]
+      #   The given block will be passed every repeated String.
+      #
+      # @yieldparam [String] repeated
+      #   A repeated version of the String.
+      #
+      # @return [Enumerator]
+      #   If no block is given, an Enumerator will be returned.
+      #
+      def each(repeatable)
+        return enum_for(__method__,repeatable) unless block_given?
+
+        @lengths.each do |length|
+          yield(repeatable * length)
+        end
+
+        return nil
+      end
+
+    end
+  end
+end

data/lib/ronin/fuzzing/template.rb

@@ -0,0 +1,133 @@
+#
+# Copyright (c) 2006-2012 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 'chars'
+require 'combinatorics/generator'
+require 'combinatorics/list_comprehension'
+
+module Ronin
+  module Fuzzing
+    #
+    # Fuzzing class that generates Strings based on a template.
+    #
+    # @api semipublic
+    #
+    # @since 0.5.0
+    #
+    class Template
+
+      include Enumerable
+
+      #
+      # Initializes a new Fuzzing template.
+      #
+      # @param [Array(<String,Symbol,Enumerable>, <Integer,Array,Range>)] fields
+      #   The fields which defines the string or character sets which will
+      #   make up parts of the String.
+      #
+      # @raise [ArgumentError]
+      #   A given character set name was unknown.
+      #
+      # @raise [TypeError]
+      #   A given string set was not a String, Symbol or Enumerable.
+      #   A given string set length was not an Integer or Enumerable.
+      #
+      def initialize(fields)
+        @enumerators = []
+
+        fields.each do |(set,length)|
+          enum = case set
+                 when String
+                   [set].each
+                 when Enumerable
+                   set.each
+                 when Symbol
+                   name = set.to_s.upcase
+
+                   unless Chars.const_defined?(name)
+                     raise(ArgumentError,"unknown charset #{set.inspect}")
+                   end
+
+                   Chars.const_get(name).each_char
+                 else
+                   raise(TypeError,"set must be a String, Symbol or Enumerable")
+                 end
+
+          case length
+          when Integer
+            length.times { @enumerators << enum.dup }
+          when Array, Range
+            @enumerators << Combinatorics::Generator.new do |g|
+              length.each do |sublength|
+                superset = Array.new(sublength) { enum.dup }
+
+                superset.comprehension { |strings| g.yield strings.join }
+              end
+            end
+          when nil
+            @enumerators << enum
+          else
+            raise(TypeError,"length must be an Integer, Range or Array")
+          end
+        end
+      end
+
+      #
+      # @see #initialize
+      #
+      def self.[](*fields)
+        new(fields)
+      end
+
+      #
+      # Generate permutations of Strings from a format template.
+      #
+      # @yield [string]
+      #   The given block will be passed each unique String.
+      #
+      # @yieldparam [String] string
+      #   A newly generated String.
+      #
+      # @return [Enumerator]
+      #   If no block is given, an Enumerator will be returned.
+      #
+      def each
+        return enum_for(__method__) unless block_given?
+
+        @enumerators.comprehension do |fields|
+          string = ''
+
+          fields.each do |field|
+            string << case field
+                      when Integer
+                        field.chr
+                      else
+                        field.to_s
+                      end
+          end
+
+          yield string
+        end
+
+        return nil
+      end
+
+    end
+  end
+end

data/lib/ronin/network/mixins/http.rb

@@ -78,7 +78,7 @@ module Ronin
         end
 
         #
-        # Connects to the HTTP server.
+        # Starts a HTTP connection with the server.
         #
         # @param [Hash] options
         #   Additional options
@@ -86,51 +86,46 @@ module Ronin
         # @option options [String, URI::HTTP] :url
         #   The full URL to request.
         #
-        # @option options [String] :user
-        #   The user to authenticate with when connecting to the HTTP
-        #   server.
-        #
-        # @option options [String] :password
-        #   The password to authenticate with when connecting to the HTTP
-        #   server.
-        #
         # @option options [String] :host
         #   The host the HTTP server is running on.
         #
         # @option options [Integer] :port (Net::HTTP.default_port)
         #   The port the HTTP server is listening on.
         #
-        # @option options [String] :path
-        #   The path to request from the HTTP server.
+        # @option options [String, Hash] :proxy (HTTP.proxy)
+        #   A Hash of proxy settings to use when connecting to the HTTP server.
+        #
+        # @option options [String] :user
+        #   The user to authenticate with when connecting to the HTTP server.
+        #
+        # @option options [String] :password
+        #   The password to authenticate with when connecting to the HTTP
+        #   server.
         #
-        # @yield [session]
-        #   If a block is given, it will be passes the new HTTP session
-        #   object.
+        # @option options [Boolean, Hash] :ssl
+        #   Enables SSL for the HTTP connection.
         #
-        # @yieldparam [Net::HTTP] session
+        # @option :ssl [Symbol] :verify
+        #   Specifies the SSL certificate verification mode.
+        #   
+        # @yield [http]
+        #   If a block is given, it will be passed the newly created HTTP
+        #   session object.
+        #
+        # @yieldparam [Net::HTTP] http
         #   The newly created HTTP session.
         #
         # @return [Net::HTTP]
         #   The HTTP session object.
         #
-        # @see Network::HTTP#http_session
+        # @see Network::HTTP#http_connect
         #
         # @api public
         #
-        def http_session(options={},&block)
-          options = http_merge_options(options)
-
-          super(options) do |http,expanded_options|
-            print_info "Starting HTTP Session with #{host_port}"
-
-            if block.arity == 2
-              block.call(http,expanded_options)
-            else
-              block.call(http)
-            end
-
-            print_info "Closing HTTP Session with #{host_port}"
-          end
+        # @since 0.5.0
+        #
+        def http_connect(options={},&block)
+          super(http_merge_options(options),&block)
         end
 
         #
@@ -163,523 +158,32 @@ module Ronin
         # @api public
         #
         def http_request(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP #{options[:method]} #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Returns the Status Code of the Response.
-        #
-        # @param [Hash] options
-        #   Additional options.
-        #
-        # @option options [Symbol, String] :method (:head)
-        #   The method to use for the request.
-        #
-        # @return [Integer]
-        #   The HTTP Response Status.
-        #
-        # @see Network::HTTP#http_status
-        #
-        # @since 1.1.0
-        #
-        # @api public
-        #
-        def http_status(options={})
-          options = http_merge_options(options)
-
-          if (result = super(options))
-            print_debug "HTTP #{result} #{http_options_to_s(options)}"
-          end
-
-          return result
-        end
-
-        #
-        # Checks if the response has an HTTP OK status code.
-        #
-        # @param [Hash] options
-        #   Additional options.
-        #
-        # @option options [Symbol, String] :method (:head)
-        #   The method to use for the request.
-        #
-        # @return [Boolean]
-        #   Specifies whether the response had an HTTP OK status code or not.
-        #
-        # @see Network::HTTP#http_ok?
-        #
-        # @since 1.1.0
-        #
-        # @api public
-        #
-        def http_ok?(options={})
-          options = http_merge_options(options)
+          response = super(options) do |request,expanded_options|
+            if block
+              if block.arity == 2
+                block.call(request,expanded_options)
+              else
+                block.call(request)
+              end
+            end
 
-          if (result = super(options))
-            print_debug "HTTP 200 OK #{http_options_to_s(options)}"
-          end
+            host = options.fetch(:host,self.host)
+            port = options.fetch(:port,self.port)
 
-          return result
-        end
+            print_info "HTTP #{request.method} #{host}:#{port} #{request.path}"
 
-        #
-        # Sends a HTTP Head request and returns the HTTP Server header.
-        #
-        # @param [Hash] options
-        #   Additional options.
-        #
-        # @option options [Symbol, String] :method (:head)
-        #   The method to use for the request.
-        #
-        # @return [String]
-        #   The HTTP `Server` header.
-        #
-        # @see Network::HTTP#http_server
-        #
-        # @since 1.1.0
-        #
-        # @api public
-        #
-        def http_server(options={})
-          options = http_merge_options(options)
-
-          if (result = super(options))
-            print_debug "HTTP Server: #{result}"
+            request.each_capitalized do |name,value|
+              print_debug "  #{name}: #{value}"
+            end
           end
 
-          return result
-        end
-
-        #
-        # Sends an HTTP Head request and returns the HTTP X-Powered-By header.
-        #
-        # @param [Hash] options
-        #   Additional options.
-        #
-        # @option options [Symbol, String] :method (:get)
-        #   The method to use for the request.
-        #
-        # @return [String]
-        #   The HTTP `X-Powered-By` header.
-        #
-        # @see Network::HTTP#http_powered_by
-        #
-        # @since 1.1.0
-        #
-        # @api public
-        #
-        def http_powered_by(options={})
-          options = http_merge_options(options)
+          print_info "HTTP #{response.code} #{response.message}"
 
-          if (result = super(options))
-            print_debug "HTTP X-Powered-By: #{result}"
+          response.each_capitalized do |name,value|
+            print_debug "  #{name}: #{value}"
           end
 
-          return result
-        end
-
-        #
-        # Performs an HTTP Copy request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [Net::HTTP::Response]
-        #   The response of the HTTP request.
-        #
-        # @see Network::HTTP#http_copy
-        #
-        # @api public
-        #
-        def http_copy(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP COPY #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Performs an HTTP Delete request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [Net::HTTP::Response]
-        #   The response of the HTTP request.
-        #
-        # @see Network::HTTP#http_delete
-        #
-        # @api public
-        #
-        def http_delete(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP DELETE #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Performs an HTTP Get request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [Net::HTTP::Response]
-        #   The response of the HTTP request.
-        #
-        # @see Network::HTTP#http_get
-        #
-        # @api public
-        #
-        def http_get(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP GET #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Performs an HTTP Get request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [String]
-        #   The body of the HTTP Get request.
-        #
-        # @see Network::HTTP#http_get_body
-        #
-        # @api public
-        #
-        def http_get_body(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP GET #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Performs an HTTP Head request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [Net::HTTP::Response]
-        #   The response of the HTTP request.
-        #
-        # @see Network::HTTP#http_head
-        #
-        # @api public
-        #
-        def http_head(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP HEAD #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Performs an HTTP Lock request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [Net::HTTP::Response]
-        #   The response of the HTTP request.
-        #
-        # @see Network::HTTP#http_lock
-        #
-        # @api public
-        #
-        def http_lock(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP LOCK #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Performs an HTTP Mkcol request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [Net::HTTP::Response]
-        #   The response of the HTTP request.
-        #
-        # @see Network::HTTP#http_mkcol
-        #
-        # @api public
-        #
-        def http_mkcol(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP MKCOL #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Performs an HTTP Move request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [Net::HTTP::Response]
-        #   The response of the HTTP request.
-        #
-        # @see Network::HTTP#http_move
-        #
-        # @api public
-        #
-        def http_move(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP MOVE #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Performs an HTTP Options request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [Net::HTTP::Response]
-        #   The response of the HTTP request.
-        #
-        # @see Network::HTTP#http_options
-        #
-        # @api public
-        #
-        def http_options(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP OPTIONS #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Performs an HTTP Post request.
-        #
-        # @param [Hash] options
-        #   Additional options.
-        #
-        # @option options [String] :post_data
-        #   The `POSTDATA` to send with the HTTP Post request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [Net::HTTP::Response]
-        #   The response of the HTTP request.
-        #
-        # @see Network::HTTP#http_post
-        #
-        # @api public
-        #
-        def http_post(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP POST #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Performs an HTTP Post request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [String]
-        #   The body of the Post request.
-        #
-        # @see Network::HTTP#http_post_body
-        #
-        # @api public
-        #
-        def http_post_body(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP POST #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Performs an HTTP PUT request.
-        #
-        # @param [Hash] options
-        #   Additional options.
-        #
-        # @option options [String] :body
-        #   The body for the request.
-        #
-        # @option options [String] :post_data
-        #   The `POSTDATA` to send with the HTTP PUT request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [Net::HTTP::Response]
-        #   The response of the HTTP request.
-        #
-        # @see Network::HTTP#http_put
-        #
-        # @since 0.4.0
-        #
-        # @api public
-        #
-        def http_put(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP PUT #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Performs an HTTP Propfind request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [Net::HTTP::Response]
-        #   The response of the HTTP request.
-        #
-        # @see Network::HTTP#http_prop_find
-        #
-        # @api public
-        #
-        def http_prop_find(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP PROPFIND #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Performs an HTTP Proppatch request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [Net::HTTP::Response]
-        #   The response of the HTTP request.
-        #
-        # @see Network::HTTP#http_prop_patch
-        #
-        # @api public
-        #
-        def http_prop_patch(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP PROPPATCH #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Performs an HTTP Trace request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [Net::HTTP::Response]
-        #   The response of the HTTP request.
-        #
-        # @see Network::HTTP#http_trace
-        #
-        # @api public
-        #
-        def http_trace(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP TRACE #{http_options_to_s(options)}"
-
-          return super(options,&block)
-        end
-
-        #
-        # Performs an HTTP Unlock request.
-        #
-        # @yield [response]
-        #   If a block is given, it will be passed the response received
-        #   from the request.
-        #
-        # @yieldparam [Net::HTTP::Response] response
-        #   The HTTP response object.
-        #
-        # @return [Net::HTTP::Response]
-        #   The response of the HTTP request.
-        #
-        # @see Network::HTTP#http_unlock
-        #
-        # @api public
-        #
-        def http_unlock(options={},&block)
-          options = http_merge_options(options)
-          print_info "HTTP UNLOCK #{http_options_to_s(options)}"
-
-          return super(options,&block)
+          return response
         end
 
         private
@@ -698,58 +202,25 @@ module Ronin
         # @api private
         #
         def http_merge_options(options={})
-          options[:host] ||= self.host if self.host
-          options[:port] ||= self.port if self.port
+          options = options.dup
+
+          options[:proxy] ||= self.http_proxy if self.http_proxy
+          options[:host]  ||= self.host       if self.host
+          options[:port]  ||= self.port       if self.port
 
           if (self.http_vhost || self.http_user_agent)
-            headers = options.fetch(:headers,{})
+            headers = (options[:headers] ||= {})
 
-            headers[:host] ||= self.http_vhost if self.http_vhost
+            headers[:host]       ||= self.http_vhost      if self.http_vhost
             headers[:user_agent] ||= self.http_user_agent if self.http_user_agent
-
-            options[:headers] = headers
           end
 
-          options[:user] ||= self.http_user if self.http_user
+          options[:user]     ||= self.http_user     if self.http_user
           options[:password] ||= self.http_password if self.http_password
 
-          options[:proxy] ||= self.http_proxy if self.http_proxy
-
           return options
         end
 
-        #
-        # Converts the HTTP options to a printable String.
-        #
-        # @param [Hash] options
-        #   HTTP options.
-        #
-        # @return [String]
-        #   The printable String.
-        #
-        # @since 1.1.0
-        #
-        # @api private
-        #
-        def http_options_to_s(options)
-          fields = ["#{options[:host]}:#{options[:port]}"]
-
-          if (options[:user] || options[:password])
-            fields << "#{options[:user]}:#{options[:password]}"
-          end
-
-          path = options[:path]
-          path += "?#{options[:query]}" if options[:query]
-
-          fields << path
-
-          if options[:headers]
-            fields << ("%p" % options[:headers])
-          end
-
-          return fields.join(' ')
-        end
-
       end
     end
   end