ronin-support 0.4.1 → 0.5.2

data/lib/ronin/fuzzing/repeater.rb

@@ -0,0 +1,81 @@
+#
+# Copyright (c) 2006-2021 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 <https://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-2021 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 <https://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/mixin.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2021 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This file is part of Ronin.
 #
@@ -14,7 +14,7 @@
 # GNU General Public License for more details.
 #
 # You should have received a copy of the GNU General Public License
-# along with Ronin.  If not, see <http://www.gnu.org/licenses/>.
+# along with Ronin.  If not, see <https://www.gnu.org/licenses/>.
 #
 
 module Ronin

data/lib/ronin/network.rb

@@ -1,29 +1,31 @@
 #
-# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2021 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
-# This file is part of Ronin Support.
+# This file is part of ronin-support.
 #
-# Ronin Support is free software: you can redistribute it and/or modify
+# 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,
+# 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/>.
+# along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
 #
 
 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

@@ -1,20 +1,20 @@
 #
-# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2021 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
-# This file is part of Ronin Support.
+# This file is part of ronin-support.
 #
-# Ronin Support is free software: you can redistribute it and/or modify
+# 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,
+# 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/>.
+# along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
 #
 
 require 'resolv'
@@ -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]
@@ -68,10 +68,8 @@ module Ronin
       # @api public
       #
       def dns_resolver(nameserver=DNS.nameserver)
-        if nameserver
-          Resolv::DNS.new(:nameserver => nameserver)
-        else
-          Resolv
+        if nameserver then Resolv::DNS.new(:nameserver => nameserver)
+        else               Resolv
         end
       end
 
@@ -84,18 +82,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 +115,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 +144,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 +177,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/esmtp.rb

@@ -1,20 +1,20 @@
 #
-# Copyright (c) 2006-2012 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2006-2021 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
-# This file is part of Ronin Support.
+# This file is part of ronin-support.
 #
-# Ronin Support is free software: you can redistribute it and/or modify
+# 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,
+# 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/>.
+# along with ronin-support.  If not, see <https://www.gnu.org/licenses/>.
 #
 
 require 'ronin/network/smtp'