ronin-support 0.4.1 → 0.5.0.rc1

data/lib/ronin/fuzzing/fuzzing.rb

@@ -66,6 +66,8 @@ module Ronin
       end
     end
 
+    module_function
+
     #
     # Various bad-strings.
     #
@@ -77,7 +79,7 @@ module Ronin
     #   or null-bytes (see {NULL_BYTES}), of varying length
     #   (see {SHORT_LENGTHS} and {LONG_LENGTHS}).
     #
-    def self.bad_strings(&block)
+    def bad_strings(&block)
       yield ''
 
       chars = [
@@ -118,7 +120,7 @@ module Ronin
     # @yieldparam [String] fmt_string
     #   A format-string containing format operators (see {FORMAT_STRINGS}).
     #
-    def self.format_strings(&block)
+    def format_strings(&block)
       FORMAT_STRINGS.each do |fmt|
         yield fmt
         yield fmt * 100
@@ -136,7 +138,7 @@ module Ronin
     # @yieldparam [String] path
     #   A known bad path.
     #
-    def self.bad_paths(&block)
+    def bad_paths(&block)
       padding = 'A' * 5_000
 
       yield "/.:/#{padding}\x00\x00"
@@ -166,7 +168,7 @@ module Ronin
     # @yieldparam [String] bitfield
     #   A bit-field (8bit - 64bit).
     #
-    def self.bit_fields(&block)
+    def bit_fields(&block)
       ("\x00".."\xff").each do |c|
         yield c
         yield c << c # x2
@@ -184,7 +186,7 @@ module Ronin
     # @yieldparam [String] bitfield
     #   A signed bit-field (8bit - 64bit).
     #
-    def self.signed_bit_fields(&block)
+    def signed_bit_fields(&block)
       ("\x80".."\xff").each do |c|
         yield c
         yield c << c # x2
@@ -202,7 +204,7 @@ module Ronin
     # @yieldparam [String] int
     #   A unsigned 8bit integer.
     #
-    def self.uint8(&block)
+    def uint8(&block)
       ("\x00".."\xff").each(&block)
     end
 
@@ -215,7 +217,7 @@ module Ronin
     # @yieldparam [String] int
     #   A unsigned 16bit integer.
     #
-    def self.uint16
+    def uint16
       uint8 { |c| yield c * 2 }
     end
 
@@ -228,7 +230,7 @@ module Ronin
     # @yieldparam [String] int
     #   A unsigned 32bit integer.
     #
-    def self.uint32
+    def uint32
       uint8 { |c| yield c * 4 }
     end
 
@@ -241,7 +243,7 @@ module Ronin
     # @yieldparam [String] int
     #   A unsigned 64bit integer.
     #
-    def self.uint64
+    def uint64
       uint8 { |c| yield c * 8 }
     end
 
@@ -254,7 +256,7 @@ module Ronin
     # @yieldparam [String] int
     #   A signed 8bit integer.
     #
-    def self.int8(&block)
+    def int8(&block)
       ("\x00".."\x70").each(&block)
     end
 
@@ -267,7 +269,7 @@ module Ronin
     # @yieldparam [String] int
     #   A signed 16bit integer.
     #
-    def self.int16
+    def int16
       int8 { |c| yield c * 2 }
     end
 
@@ -280,7 +282,7 @@ module Ronin
     # @yieldparam [String] int
     #   A signed 32bit integer.
     #
-    def self.int32
+    def int32
       int8 { |c| yield c * 4 }
     end
 
@@ -293,7 +295,7 @@ module Ronin
     # @yieldparam [String] int
     #   A signed 64bit integer.
     #
-    def self.int64
+    def int64
       int8 { |c| yield c * 8 }
     end
 
@@ -306,7 +308,7 @@ module Ronin
     # @yieldparam [String] int
     #   A negative-signed 8bit integer.
     #
-    def self.sint8(&block)
+    def sint8(&block)
       ("\x80".."\xff").each(&block)
     end
 
@@ -319,7 +321,7 @@ module Ronin
     # @yieldparam [String] int
     #   A negative-signed 16bit integer.
     #
-    def self.sint16
+    def sint16
       sint8 { |c| yield c * 2 }
     end
 
@@ -332,7 +334,7 @@ module Ronin
     # @yieldparam [String] int
     #   A negative-signed 32bit integer.
     #
-    def self.sint32
+    def sint32
       sint8 { |c| yield c * 4 }
     end
 
@@ -345,7 +347,7 @@ module Ronin
     # @yieldparam [String] int
     #   A negative-signed 64bit integer.
     #
-    def self.sint64
+    def sint64
       sint8 { |c| yield c * 8 }
     end
 

data/lib/ronin/network.rb

@@ -20,10 +20,11 @@
 require 'ronin/network/dns'
 require 'ronin/network/tcp'
 require 'ronin/network/udp'
-require 'ronin/network/ssl'
+require 'ronin/network/unix'
 require 'ronin/network/smtp'
 require 'ronin/network/esmtp'
 require 'ronin/network/pop3'
 require 'ronin/network/imap'
 require 'ronin/network/telnet'
+require 'ronin/network/ftp'
 require 'ronin/network/http'

data/lib/ronin/network/dns.rb

@@ -42,7 +42,7 @@ module Ronin
       #
       # Sets the DNS nameserver to be queried.
       #
-      # @param [IPAddr, String, nil]
+      # @param [IPAddr, String, nil] address
       #   The address of the nameserver.
       #
       # @return [String, nil]
@@ -84,18 +84,28 @@ module Ronin
       # @param [String, nil] nameserver
       #   Optional DNS nameserver to query.
       #
+      # @yield [address]
+      #   If a block is given and the hostname was resolved, the address will
+      #   be passed to the block.
+      #
+      # @yieldparam [String] address
+      #   The address of the hostname.
+      #
       # @return [String, nil]
       #   The address of the hostname.
       #
       # @api public
       #
       def dns_lookup(hostname,nameserver=DNS.nameserver)
-        resolv = dns_resolver(nameserver)
+        hostname = hostname.to_s
+        resolv   = dns_resolver(nameserver)
+        address  = begin
+                     resolv.getaddress(hostname).to_s
+                   rescue Resolv::ResolvError
+                   end
 
-        begin
-          resolv.getaddress(hostname.to_s).to_s
-        rescue Resolv::ResolvError
-        end
+        yield(address) if (block_given? && address)
+        return address
       end
 
       #
@@ -107,13 +117,24 @@ module Ronin
       # @param [String, nil] nameserver
       #   Optional DNS nameserver to query.
       #
+      # @yield [address]
+      #   If a block is given, each resolved address will be passed to the
+      #   block.
+      #
+      # @yieldparam [String] address
+      #   A address of the hostname.
+      #
       # @return [Array<String>]
       #   The addresses of the hostname.
       #
       # @api public
       #
-      def dns_lookup_all(hostname,nameserver=DNS.nameserver)
-        dns_resolver(nameserver).getaddresses(hostname.to_s).map(&:to_s)
+      def dns_lookup_all(hostname,nameserver=DNS.nameserver,&block)
+        hostname  = hostname.to_s
+        addresses = dns_resolver(nameserver).getaddresses(hostname).map(&:to_s)
+
+        addresses.each(&block) if block
+        return addresses
       end
 
       #
@@ -125,18 +146,28 @@ module Ronin
       # @param [String, nil] nameserver
       #   Optional DNS nameserver to query.
       #
+      # @yield [hostname]
+      #   If a block is given and a hostname was found for the address,
+      #   the resolved hostname will be passed to the block.
+      #
+      # @yieldparam [String] hostname
+      #   The hostname of the address.
+      #
       # @return [String, nil]
       #   The hostname of the address.
       #
       # @api public
       #
       def dns_reverse_lookup(address,nameserver=DNS.nameserver)
-        resolv = dns_resolver(nameserver)
+        address  = address.to_s
+        resolv   = dns_resolver(nameserver)
+        hostname = begin
+                     resolv.getname(address).to_s
+                   rescue Resolv::ResolvError
+                   end
 
-        begin
-          resolv.getname(address.to_s).to_s
-        rescue Resolv::ResolvError
-        end
+        yield(hostname) if (block_given? && hostname)
+        return hostname
       end
 
       #
@@ -148,13 +179,24 @@ module Ronin
       # @param [String, nil] nameserver
       #   Optional DNS nameserver to query.
       #
+      # @yield [hostname]
+      #   If a block is given and hostnames were found for the address,
+      #   each hostname will be passed to the block.
+      #
+      # @yieldparam [String] hostname
+      #   A hostname of the address.
+      #
       # @return [Array<String>]
       #   The hostnames of the address.
       #
       # @api public
       #
-      def dns_reverse_lookup_all(address,nameserver=DNS.nameserver)
-        dns_resolver(nameserver).getnames(address.to_s).map(&:to_s)
+      def dns_reverse_lookup_all(address,nameserver=DNS.nameserver,&block)
+        address   = address.to_s
+        hostnames = dns_resolver(nameserver).getnames(address).map(&:to_s)
+
+        hostnames.each(&block) if block
+        return hostnames
       end
     end
   end

data/lib/ronin/network/extensions.rb

@@ -20,7 +20,6 @@
 require 'ronin/network/extensions/dns'
 require 'ronin/network/extensions/tcp'
 require 'ronin/network/extensions/udp'
-require 'ronin/network/extensions/ssl'
 require 'ronin/network/extensions/smtp'
 require 'ronin/network/extensions/esmtp'
 require 'ronin/network/extensions/pop3'

data/lib/ronin/network/ftp.rb

@@ -0,0 +1,145 @@
+#
+# 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 'net/ftp'
+
+module Ronin
+  module Network
+    #
+    # Provides helper methods for communicating with FTP services.
+    #
+    # @since 0.5.0
+    #
+    module FTP
+      # Default FTP port
+      DEFAULT_PORT = 21
+
+      # Default FTP user
+      DEFAULT_USER = 'anonymous'
+
+      #
+      # Default port used by {FTP}.
+      #
+      # @return [Integer]
+      #   The default Ronin FTP port.
+      #
+      # @api public
+      #
+      def self.default_port
+        @default_port ||= DEFAULT_PORT
+      end
+
+      #
+      # Sets the default port used by {FTP}.
+      #
+      # @param [Integer] new_port
+      #   The new default Ronin FTP port.
+      #
+      # @api public
+      #
+      def self.default_port=(new_port)
+        @default_port = new_port
+      end
+
+      #
+      # Creates a connection to the FTP server.
+      #
+      # @param [String] host
+      #   The host to connect to.
+      #
+      # @param [Hash] options
+      #   Additional options.
+      #
+      # @option options [Integer] :port (FTP.default_port)
+      #   The port to connect to.
+      #
+      # @option options [String] :user (DEFAULT_USER)
+      #   The user to authenticate with.
+      #
+      # @option options [String] :password
+      #   The password to authenticate with.
+      #
+      # @option options [String] :account
+      #   The FTP account information to send via the `ACCT` command.
+      #
+      # @yield [session]
+      #   If a block is given, it will be passed an FTP session object.
+      #
+      # @yieldparam [Net::FTP] session
+      #   The FTP session.
+      #
+      # @return [Net::FTP]
+      #   The FTP session.
+      #
+      # @example
+      #   ftp_connect('www.example.com', :user => 'joe', :password => 'secret')
+      #
+      # @api public
+      #
+      def ftp_connect(host,options={})
+        host     = host.to_s
+        port     = (options[:port] || FTP.default_port)
+        user     = (options[:user] || DEFAULT_USER)
+        password = options[:password]
+        acct     = options[:account]
+
+        session = Net::FTP.new
+        session.connect(host,port)
+        session.login(user,password,acct)
+
+        yield session if block_given?
+        return session
+      end
+
+      #
+      # Starts a session with the FTP server.
+      #
+      # @param [String] host
+      #   The host to connect to.
+      #
+      # @param [Hash] options
+      #   Additional options.
+      #
+      # @yield [session]
+      #   If a block is given, it will be passed an FTP session object.
+      #   After the block has returned, the session will be closed.
+      #
+      # @yieldparam [Net::FTP] session
+      #   The FTP session.
+      #
+      # @example
+      #   ftp_session('www.example.com', :user => 'joe') do |ftp|
+      #     # ...
+      #   end
+      #
+      # @see #ftp_connect
+      #
+      # @api public
+      #
+      def ftp_session(host,options={})
+        ftp = ftp_connect(host,options)
+
+        yield ftp if block_given?
+
+        ftp.close
+        return nil
+      end
+    end
+  end
+end