ronin-support 0.4.0.rc1 → 0.4.0.rc2

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/network/imap.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/network/mixins.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #
@@ -122,9 +122,9 @@ module Ronin
             print_info "Starting HTTP Session with #{host_port}"
 
             if block.arity == 2
-              block.call(sess,expanded_options)
+              block.call(http,expanded_options)
             else
-              block.call(sess)
+              block.call(http)
             end
 
             print_info "Closing HTTP Session with #{host_port}"

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

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

@@ -0,0 +1,144 @@
+#
+# 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 'ronin/network/mixins/mixin'
+require 'ronin/network/ssl'
+
+module Ronin
+  module Network
+    module Mixins
+      #
+      # Adds SSL convenience methods and connection parameters to a class.
+      #
+      # Defines the following parameters:
+      #
+      # * `host` (`String`) - SSL host.
+      # * `port` (`Integer`) - SSL port.
+      # * `local_host` (`String`) - SSL local host.
+      # * `local_port` (`Integer`) - SSL local port.
+      #
+      # @since 0.4.0
+      #
+      module SSL
+        include Mixin, Network::SSL
+
+        # SSL host
+        parameter :host, :type => String,
+                         :description => 'SSL host'
+
+        # SSL port
+        parameter :port, :type => Integer,
+                         :description => 'SSL port'
+
+        # SSL local host
+        parameter :local_host, :type => String,
+                               :description => 'SSL local host'
+
+        # SSL local port
+        parameter :local_port, :type => Integer,
+                               :description => 'SSL local port'
+
+        protected
+
+        #
+        # Establishes a SSL connection.
+        #
+        # @option options [Symbol] :verify
+        #   Specifies whether to verify the SSL certificate.
+        #   May be one of the following:
+        #
+        #   * `:none`
+        #   * `:peer`
+        #   * `:client_once`
+        #   * `:fail_if_no_peer_cert`
+        #
+        # @option options [String] :cert
+        #   The path to the SSL certificate.
+        #
+        # @option options [String] :key
+        #   The path to the SSL key.
+        #
+        # @yield [ssl_socket]
+        #   The given block will be passed the new SSL Socket.
+        #
+        # @yieldparam [OpenSSL::SSL::SSLSocket] ssl_socket
+        #   The new SSL Socket.
+        #
+        # @return [OpenSSL::SSL::SSLSocket]
+        #   the new SSL Socket.
+        #
+        # @example
+        #   socket = ssl_connect
+        #
+        # @api public
+        #
+        def ssl_connect(host,port,options={},&block)
+          options = options.merge(
+            :local_host => self.local_host,
+            :local_port => self.local_port
+          )
+
+          super(self.host,self.port,options,&block)
+        end
+
+        #
+        # Creates a new temporary SSL connection.
+        #
+        # @param [Hash] options
+        #   Additional options.
+        #
+        # @option options [String] :local_host
+        #   The local host to bind to.
+        #
+        # @option options [Integer] :local_port
+        #   The local port to bind to.
+        #
+        # @option options [Symbol] :verify
+        #   Specifies whether to verify the SSL certificate.
+        #
+        # @option options [String] :cert
+        #   The path to the SSL certificate.
+        #
+        # @option options [String] :key
+        #   The path to the SSL key.
+        #
+        # @yield [ssl_socket]
+        #   The given block will be passed the temporary SSL Socket.
+        #
+        # @yieldparam [OpenSSL::SSL::SSLSocket] ssl_socket
+        #   The temporary SSL Socket.
+        #
+        # @return [nil]
+        #
+        # @example
+        #   ssl_session do |socket|
+        #     socket.write("GET /\n\n")
+        #
+        #     socket.each_line { |line| puts line }
+        #   end
+        #
+        # @api public
+        #
+        def ssl_session(options={},&block)
+          super(self.host,self.port,options,&block)
+        end
+      end
+    end
+  end
+end

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #
@@ -83,10 +83,11 @@ module Ronin
         #   tcp_connect # => TCPSocket
         #
         # @example
-        #   tcp_connect do |sock|
-        #     sock.write("GET /\n\n")
-        #     puts sock.readlines
-        #     sock.close
+        #   tcp_connect do |socket|
+        #     socket.write("GET /\n\n")
+        #
+        #     puts socket.readlines
+        #     socket.close
         #   end
         #
         # @api public

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

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

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #
@@ -84,8 +84,8 @@ module Ronin
         #   # => UDPSocket
         #
         # @example
-        #   udp_connect do |sock|
-        #     puts sock.readlines
+        #   udp_connect do |socket|
+        #     puts socket.readlines
         #   end
         #
         # @api public
@@ -149,6 +149,32 @@ module Ronin
           return nil
         end
 
+        #
+        # Connects to the host and port specified by the `host` and `port`
+        # parameters, sends the given data and then disconnects.
+        #
+        # @return [true]
+        #   The data was successfully sent.
+        #
+        # @example
+        #   buffer = "GET /" + ('A' * 4096) + "\n\r"
+        #   Net.udp_send(buffer)
+        #   # => true
+        #
+        # @api public
+        #
+        # @since 0.4.0
+        #
+        def udp_send(data)
+          print_info "Connecting to #{host_port} ..."
+          print_debug "Sending data: #{data.inspect}"
+
+          super(data,self.host,self.port,self.local_host,self.local_port)
+
+          print_info "Disconnected from #{host_port}"
+          return true
+        end
+
         #
         # Creates a new UDPServer object listening on `server_host` and
         # `server_port` parameters.

data/lib/ronin/network/network.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/network/pop3.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/network/smtp.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/network/smtp/email.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/network/smtp/smtp.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #

data/lib/ronin/network/ssl.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #
@@ -173,11 +173,8 @@ module Ronin
     #
     # @api public
     #
-    def ssl_session(host,port)
-      ssl_socket = ssl_connect(host,port)
-
-      yield ssl_socket if block_given?
-
+    def ssl_session(host,port,options={},&block)
+      ssl_socket = ssl_connect(host,port,options,&block)
       ssl_socket.close
       return nil
     end

data/lib/ronin/network/tcp.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2011 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin Support.
 #
@@ -34,7 +34,7 @@ module Ronin
       # @param [Integer] port
       #   The port to connect to.
       #
-      # @param [String] local_host (nil)
+      # @param [String] local_host ('0.0.0.0')
       #   The local host to bind to.
       #
       # @param [Integer] local_port (nil)
@@ -53,24 +53,23 @@ module Ronin
       #   tcp_connect('www.hackety.org',80) # => TCPSocket
       #
       # @example
-      #   tcp_connect('www.wired.com',80) do |sock|
-      #     sock.write("GET /\n\n")
-      #     puts sock.readlines
-      #     sock.close
+      #   tcp_connect('www.wired.com',80) do |socket|
+      #     socket.write("GET /\n\n")
+      #
+      #     puts socket.readlines
+      #     socket.close
       #   end
       #
       # @api public
       #
       def tcp_connect(host,port,local_host=nil,local_port=nil)
-        host = host.to_s
-        local_host = if local_host
-                       local_host.to_s
-                     end
+        host       = host.to_s
+        local_host = (local_host || '0.0.0.0').to_s
 
-        sock = TCPSocket.new(host,port,local_host,local_port)
+        socket = TCPSocket.new(host,port,local_host,local_port)
 
-        yield sock if block_given?
-        return sock
+        yield socket if block_given?
+        return socket
       end
 
       #
@@ -101,11 +100,11 @@ module Ronin
       # @api public
       #
       def tcp_connect_and_send(data,host,port,local_host=nil,local_port=nil)
-        sock = tcp_connect(host,port,local_host,local_port)
-        sock.write(data)
+        socket = tcp_connect(host,port,local_host,local_port)
+        socket.write(data)
 
-        yield sock if block_given?
-        return sock
+        yield socket if block_given?
+        return socket
       end
 
       #
@@ -136,11 +135,11 @@ module Ronin
       # @api public
       #
       def tcp_session(host,port,local_host=nil,local_port=nil)
-        sock = tcp_connect(host,port,local_host,local_port)
+        socket = tcp_connect(host,port,local_host,local_port)
 
-        yield sock if block_given?
+        yield socket if block_given?
 
-        sock.close
+        socket.close
         return nil
       end
 
@@ -177,8 +176,8 @@ module Ronin
       def tcp_banner(host,port,local_host=nil,local_port=nil)
         banner = nil
 
-        tcp_session(host,port,local_host,local_port) do |sock|
-          banner = sock.readline.strip
+        tcp_session(host,port,local_host,local_port) do |socket|
+          banner = socket.readline.strip
         end
 
         yield banner if block_given?
@@ -215,8 +214,8 @@ module Ronin
       # @api public
       #
       def tcp_send(data,host,port,local_host=nil,local_port=nil)
-        tcp_session(host,port,local_host,local_port) do |sock|
-          sock.write(data)
+        tcp_session(host,port,local_host,local_port) do |socket|
+          socket.write(data)
         end
 
         return true
@@ -239,8 +238,8 @@ module Ronin
       #
       # @api public
       #
-      def tcp_server(port,host='0.0.0.0')
-        host = host.to_s
+      def tcp_server(port=nil,host=nil)
+        host = (host || '0.0.0.0').to_s
 
         server = TCPServer.new(host,port)
         server.listen(3)
@@ -280,7 +279,7 @@ module Ronin
       #
       # @api public
       #
-      def tcp_server_session(port,host='0.0.0.0',&block)
+      def tcp_server_session(port=nil,host=nil,&block)
         server = tcp_server(port,host,&block)
         server.close()
         return nil
@@ -306,7 +305,7 @@ module Ronin
       #
       # @api public
       #
-      def tcp_single_server(port,host='0.0.0.0')
+      def tcp_single_server(port=nil,host=nil)
         host = host.to_s
 
         server = TCPServer.new(host,port)