ronin-support 0.1.0.pre1

data/lib/ronin/network/extensions/udp.rb

@@ -0,0 +1,22 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This library 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 2.1 of the License, or (at your option) any later version.
+#
+# This library 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 this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor,
+# Boston, MA  02110-1301  USA
+#
+
+require 'ronin/network/extensions/udp/net'

data/lib/ronin/network/extensions/udp/net.rb

@@ -0,0 +1,226 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This library 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 2.1 of the License, or (at your option) any later version.
+#
+# This library 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 this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor,
+# Boston, MA  02110-1301  USA
+#
+
+require 'socket'
+
+module Net
+  #
+  # Creates a new UDPSocket object connected to a given host and port.
+  #
+  # @param [String] host
+  #   The host to connect to.
+  #
+  # @param [Integer] port
+  #   The port to connect to.
+  #
+  # @param [String] local_host (nil)
+  #   The local host to bind to.
+  #
+  # @param [Integer] local_port (nil)
+  #   The local port to bind to.
+  #
+  # @yield [socket]
+  #   If a block is given, it will be passed the newly created socket.
+  #
+  # @yieldparam [UDPsocket] socket
+  #   The newly created UDPSocket object.
+  #
+  # @return [UDPSocket]
+  #   The newly created UDPSocket object.
+  #
+  # @example
+  #   Net.udp_connect('www.hackety.org',80)
+  #   # => UDPSocket
+  #
+  # @example
+  #   Net.udp_connect('www.wired.com',80) do |sock|
+  #     puts sock.readlines
+  #   end
+  #
+  def Net.udp_connect(host,port,local_host=nil,local_port=nil)
+    host = host.to_s
+    local_host = if local_host
+                   local_host.to_s
+                 end
+
+    sock = UDPSocket.new(host,port,local_host,local_port)
+
+    yield sock if block_given?
+    return sock
+  end
+
+  #
+  # Creates a new UDPSocket object, connected to a given host and port.
+  # The given data will then be written to the newly created UDPSocket.
+  #
+  # @param [String] data
+  #   The data to send through the connection.
+  #
+  # @param [String] host
+  #   The host to connect to.
+  #
+  # @param [Integer] port
+  #   The port to connect to.
+  #
+  # @param [String] local_host (nil)
+  #   The local host to bind to.
+  #
+  # @param [Integer] local_port (nil)
+  #   The local port to bind to.
+  #
+  # @yield [socket]
+  #   If a block is given, it will be passed the newly created socket.
+  #
+  # @yieldparam [UDPsocket] socket
+  #   The newly created UDPSocket object.
+  #
+  # @return [UDPSocket]
+  #   The newly created UDPSocket object.
+  #
+  def Net.udp_connect_and_send(data,host,port,local_host=nil,local_port=nil)
+    sock = Net.udp_connect(host,port,local_host,local_port)
+    sock.write(data)
+
+    yield sock if block_given?
+    return sock
+  end
+
+  #
+  # Creates a new temporary UDPSocket object, connected to the given host
+  # and port.
+  #
+  # @param [String] host
+  #   The host to connect to.
+  #
+  # @param [Integer] port
+  #   The port to connect to.
+  #
+  # @param [String] local_host (nil)
+  #   The local host to bind to.
+  #
+  # @param [Integer] local_port (nil)
+  #   The local port to bind to.
+  #
+  # @yield [socket]
+  #   If a block is given, it will be passed the newly created socket.
+  #   After the block has returned, the socket will then be closed.
+  #
+  # @yieldparam [UDPsocket] socket
+  #   The newly created UDPSocket object.
+  #
+  # @return [nil]
+  #
+  def Net.udp_session(host,port,local_host=nil,local_port=nil)
+    sock = Net.udp_connect(host,port,local_host,local_port)
+
+    yield sock if block_given?
+
+    sock.close
+    return nil
+  end
+
+  #
+  # Reads the banner from the service running on the given host and port.
+  #
+  # @param [String] host
+  #   The host to connect to.
+  #
+  # @param [Integer] port
+  #   The port to connect to.
+  #
+  # @param [String] local_host (nil)
+  #   The local host to bind to.
+  #
+  # @param [Integer] local_port (nil)
+  #   The local port to bind to.
+  #
+  # @yield [banner]
+  #   If a block is given, it will be passed the grabbed banner.
+  #
+  # @yieldparam [String] banner
+  #   The grabbed banner.
+  #
+  # @return [String]
+  #   The grabbed banner.
+  #
+  def Net.udp_banner(host,port,local_host=nil,local_port=nil)
+    banner = nil
+
+    Net.udp_session(host,port,local_host,local_port) do |sock|
+      banner = sock.readline
+    end
+
+    yield banner if block_given?
+    return banner
+  end
+
+  #
+  # Creates a new UDPServer listening on a given host and port.
+  #
+  # @param [Integer] port
+  #   The local port to listen on.
+  #
+  # @param [String] host ('0.0.0.0')
+  #   The host to bind to.
+  #
+  # @return [UDPServer]
+  #   The new UDP server.
+  #
+  # @example
+  #   Net.udp_server(1337)
+  #
+  def Net.udp_server(port,host='0.0.0.0')
+    host = host.to_s
+    server = UDPServer.new(host,port)
+
+    yield server if block_given?
+    return server
+  end
+
+  #
+  # Creates a new temporary UDPServer listening on a given host and port.
+  #
+  # @param [Integer] port
+  #   The local port to bind to.
+  #
+  # @param [String] host ('0.0.0.0')
+  #   The host to bind to.
+  #
+  # @yield [server]
+  #   The block which will be called after the _server_ has been created.
+  #   After the block has finished, the _server_ will be closed.
+  #
+  # @yieldparam [UDPServer] server
+  #   The newly created UDP server.
+  #
+  # @return [nil]
+  #
+  # @example
+  #   Net.udp_server_session(1337) do |server|
+  #     data, sender = server.recvfrom(1024)
+  #   end
+  #
+  def Net.udp_server_session(port,host='0.0.0.0',&block)
+    server = Net.udp_server(port,host,&block)
+    server.close()
+    return nil
+  end
+end

data/lib/ronin/network/http.rb

@@ -0,0 +1,24 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This library 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 2.1 of the License, or (at your option) any later version.
+#
+# This library 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 this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor,
+# Boston, MA  02110-1301  USA
+#
+
+require 'ronin/network/http/exceptions'
+require 'ronin/network/http/http'
+require 'ronin/network/extensions/http'

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

@@ -0,0 +1,22 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This library 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 2.1 of the License, or (at your option) any later version.
+#
+# This library 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 this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor,
+# Boston, MA  02110-1301  USA
+#
+
+require 'ronin/network/http/exceptions/unknown_request'

data/lib/ronin/network/http/exceptions/unknown_request.rb

@@ -0,0 +1,29 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This library 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 2.1 of the License, or (at your option) any later version.
+#
+# This library 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 this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor,
+# Boston, MA  02110-1301  USA
+#
+
+module Ronin
+  module Network
+    module HTTP
+      class UnknownRequest < StandardError
+      end
+    end
+  end
+end

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

@@ -0,0 +1,290 @@
+#
+# Ronin - A Ruby platform for exploit development and security research.
+#
+# Copyright (c) 2006-2010 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This library 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 2.1 of the License, or (at your option) any later version.
+#
+# This library 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 this library; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor,
+# Boston, MA  02110-1301  USA
+#
+
+require 'ronin/network/http/exceptions/unknown_request'
+require 'ronin/network/http/proxy'
+require 'ronin/network/extensions/http'
+
+module Ronin
+  module Network
+    module HTTP
+      #
+      # The Ronin HTTP proxy to use. Parses the value of the `HTTP_PROXY`
+      # environment variable if set.
+      #
+      # @return [Proxy]
+      #   The Ronin HTTP proxy.
+      #
+      # @see Proxy.new
+      # @see Proxy.parse
+      #
+      def HTTP.proxy
+        @proxy ||= if ENV['HTTP_PROXY']
+                     Proxy.parse(ENV['HTTP_PROXY'])
+                   else
+                     Proxy.new
+                   end
+      end
+
+      #
+      # Sets the Ronin HTTP proxy to use.
+      #
+      # @param [Proxy, URI::HTTP, Hash, String] new_proxy
+      #   The new proxy information to use.
+      #
+      # @return [Proxy]
+      #   The new proxy.
+      #
+      # @raise [ArgumentError]
+      #   The given proxy information was not a {Proxy}, `URI::HTTP`,
+      #   `Hash` or {String}.
+      #
+      def HTTP.proxy=(new_proxy)
+        @proxy = Proxy.create(new_proxy)
+      end
+
+      #
+      # The default Ronin HTTP User-Agent string.
+      #
+      # @return [String, nil]
+      #   The default Ronin HTTP User-Agent.
+      #
+      def HTTP.user_agent
+        @user_agent ||= nil
+      end
+
+      #
+      # Sets the default Ronin HTTP User-Agent string.
+      #
+      # @param [String] agent
+      #   The new User-Agent string to use.
+      #
+      def HTTP.user_agent=(agent)
+        @user_agent = agent
+      end
+
+      #
+      # Expands the URL into options.
+      #
+      # @param [URI::HTTP, String, nil] url
+      #   The URL to expand.
+      #
+      # @return [Hash{Symbol => Object}]
+      #   The options for the URL.
+      #
+      def HTTP.expand_url(url)
+        new_options = {
+          :port => Net::HTTP.default_port,
+          :path => '/'
+        }
+
+        if url
+          url = case url
+                when URI
+                  url
+                when Hash
+                  URI::HTTP.build(url)
+                else
+                  URI(url.to_s)
+                end
+
+          new_options[:ssl] = {} if url.scheme == 'https'
+
+          new_options[:host] = url.host
+          new_options[:port] = url.port
+
+          new_options[:user] = url.user if url.user
+          new_options[:password] = url.password if url.password
+
+          new_options[:path] = url.path unless url.path.empty?
+          new_options[:path] += "?#{url.query}" if url.query
+        end
+
+        return new_options
+      end
+
+      #
+      # Expands the given HTTP options.
+      #
+      # @param [Hash] options
+      #   HTTP options.
+      #
+      # @option options [String, URI::HTTP, URI::HTTPS] :url
+      #   The URL to request.
+      #
+      # @option options [String] :host
+      #   The host to connect to.
+      #
+      # @option options [String] :port (::Net::HTTP.default_port)
+      #   The port to connect to.
+      #
+      # @option options [String] :user
+      #   The user to authenticate as.
+      #
+      # @option options [String] :password
+      #   The password to authenticate with.
+      #
+      # @option options [String] :path ('/')
+      #   The path to request.
+      #
+      # @option options [String, Hash] :proxy (Ronin::Network::HTTP.proxy)
+      #   The Proxy information.
+      #
+      # @return [Hash]
+      #   The expanded version of options.
+      #
+      def HTTP.expand_options(options={})
+        new_options = options.dup
+
+        if new_options[:ssl] == true
+          new_options[:ssl] = {}
+        end
+
+        url = new_options.delete(:url)
+        new_options.merge!(HTTP.expand_url(url))
+
+        new_options[:proxy] = if new_options.has_key?(:proxy)
+                                HTTP::Proxy.create(new_options[:proxy])
+                              else
+                                HTTP.proxy
+                              end
+
+        return new_options
+      end
+
+      #
+      # Converts an underscored, dashed, lowercase or uppercase HTTP header
+      # name to the standard camel-case HTTP header name.
+      #
+      # @param [Symbol, String] name
+      #   The unformatted HTTP header name.
+      #
+      # @return [String]
+      #   The camel-case HTTP header name.
+      #
+      def HTTP.header_name(name)
+        name.to_s.split(/[\s+_-]/).map { |word| word.capitalize }.join('-')
+      end
+
+      #
+      # Converts underscored, dashed, lowercase and uppercase HTTP headers
+      # to standard camel-cased HTTP headers.
+      #
+      # @param [Hash{Symbol,String => String}] options
+      #   Ronin HTTP headers.
+      #
+      # @return [Hash]
+      #   The camel-cased HTTP headers created from the given options.
+      #
+      def HTTP.headers(options={})
+        headers = {}
+
+        if HTTP.user_agent
+          headers['User-Agent'] = HTTP.user_agent
+        end
+
+        if options
+          options.each do |name,value|
+            headers[HTTP.header_name(name)] = value.to_s
+          end
+        end
+
+        return headers
+      end
+
+      #
+      # Creates a specific type of HTTP request object.
+      #
+      # @param [Hash] options
+      #   The HTTP options for the request.
+      #
+      # @option options [Symbol, String] :method
+      #   The HTTP method to use for the request.
+      #
+      # @option options [String] :path ('/')
+      #   The path to request.
+      #
+      # @option options [String] :body
+      #   The body of the request.
+      #
+      # @option options [Hash, String] :form_data
+      #   The form data that may be sent in the body of the request.
+      #
+      # @option options [String] :user
+      #   The user to authenticate as.
+      #
+      # @option options [String] :password
+      #   The password to authenticate with.
+      #
+      # @option options [Hash{Symbol,String => String}] :headers
+      #   Additional HTTP headers to use for the request.
+      #
+      # @return [HTTP::Request]
+      #   The new HTTP Request object.
+      #
+      # @raise [ArgumentError]
+      #   The `:method` option must be specified.
+      #
+      # @raise [UnknownRequest]
+      #   The `:method` option did not match a known Net::HTTP request
+      #   class.
+      #
+      # @see HTTP.expand_options
+      #
+      def HTTP.request(options={})
+        unless options[:method]
+          raise(ArgumentError,"the :method option must be specified")
+        end
+
+        name = options[:method].to_s.capitalize
+
+        unless Net::HTTP.const_defined?(name)
+          raise(UnknownRequest,"unknown HTTP request type #{name.dump}")
+        end
+
+        headers = HTTP.headers(options[:headers])
+        path = (options[:path] || '/').to_s
+
+        request = Net::HTTP.const_get(name).new(path,headers)
+
+        if request.request_body_permitted?
+          if options[:form_data]
+            request.set_form_data(options[:form_data])
+          elsif options[:body]
+            request.body = options[:body]
+          end
+        end
+
+        if (user = options.delete(:user))
+          user = user.to_s
+
+          if (password = options.delete(:password))
+            password = password.to_s
+          end
+
+          request.basic_auth(user,password)
+        end
+
+        return request
+      end
+    end
+  end
+end