ronin-support 0.5.2 → 1.0.0.beta2

data/lib/ronin/support/network/ip_range/cidr.rb

@@ -0,0 +1,252 @@
+# frozen_string_literal: true
+#
+# Copyright (c) 2006-2022 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# 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 'ipaddr'
+
+module Ronin
+  module Support
+    module Network
+      class IPRange
+        #
+        # Represents CIDR notation IP ranges.
+        #
+        # ## Examples
+        #
+        #     cidr = Network::IP::CIDR.new('10.0.0.1/24')
+        #     cidr.each { |ip puts }
+        #     # 10.0.0.0
+        #     # 10.0.0.1
+        #     # 10.0.0.2
+        #     # ...
+        #     # 10.0.0.254
+        #     # 10.0.0.255
+        #
+        # @api public
+        #
+        # @since 1.0.0
+        #
+        class CIDR < IPAddr
+
+          include Enumerable
+
+          # The CIDR IP range string.
+          #
+          # @return [String]
+          attr_reader :string
+
+          #
+          # Initializes and parses the CIDR range.
+          #
+          # @param [String] string
+          #   The CIDR range string to parse.
+          #
+          # @param [Integer] family
+          #   The address family for the CIDR range. This is mainly for
+          #   backwards compatibility with `IPAddr#initialize`.
+          #
+          def initialize(string,family=Socket::AF_UNSPEC)
+            super(string,family)
+
+            @string = string
+          end
+
+          #
+          # Alias for {#initialize new}.
+          #
+          # @param [String] string
+          #   The CIDR range string to parse.
+          #
+          # @return [CIDR]
+          #   The parsed CIDR range.
+          #
+          # @see #initialize
+          #
+          def self.parse(string)
+            new(string)
+          end
+
+          # Socket families and IP address sizes
+          SIZES = {
+            Socket::AF_INET  => 32,
+            Socket::AF_INET6 => 128
+          }
+
+          # Socket families and IP address masks
+          MASKS = {
+            Socket::AF_INET  => IN4MASK,
+            Socket::AF_INET6 => IN6MASK
+          }
+
+          #
+          # Calcualtes the CIDR range between two IP addresses.
+          #
+          # @param [String, IPAddr] first
+          #   The first IP address in the CIDR range.
+          #
+          # @param [String, IPAddr] last
+          #   The last IP Address in the CIDR range.
+          #
+          # @return [CIDR]
+          #   The CIDR range between the two IP addresses.
+          #
+          # @example
+          #   IPRange::CIDR.range("1.1.1.1","1.1.1.255")
+          #   # => #<Ronin::Support::Network::IPRange::CIDR: 1.1.1.1/24>
+          #
+          def self.range(first,last)
+            first_ip = case first
+                       when IPAddr then first
+                       else             IPAddr.new(first)
+                       end
+
+            last_ip = case last
+                      when IPAddr then last
+                      else             IPAddr.new(last)
+                      end
+
+            unless (first_ip.family == last_ip.family)
+              raise(ArgumentError,"must specify two IPv4 or IPv6 addresses: #{first.inspect} #{last.inspect}")
+            end
+
+            num_bits  = SIZES.fetch(first_ip.family)
+            mask      = MASKS.fetch(first_ip.family)
+            diff_bits = first_ip.to_i ^ last_ip.to_i
+
+            if diff_bits != 0
+              prefix_length = num_bits - Math.log2(diff_bits).ceil
+
+              return new("#{first_ip}/#{prefix_length}")
+            else
+              return new(first_ip.to_s)
+            end
+          end
+
+          #
+          # Enumerates over each IP address that is included in the addresses
+          # netmask. Supports both IPv4 and IPv6 addresses.
+          #
+          # @param [String] string
+          #   The CIDR range string to parse and enumerate over.
+          #
+          # @yield [ip]
+          #   The block which will be passed every IP address covered be the
+          #   netmask of the IPAddr object.
+          #
+          # @yieldparam [String] ip
+          #   An IP address.
+          #
+          # @example
+          #   IPRange::CIDR.each('10.0.0.1/24') { |ip| puts ip }
+          #   # 10.0.0.1
+          #   # 10.0.0.2
+          #   # ...
+          #   # 10.0.0.254
+          #   # 10.0.0.255
+          #
+          def self.each(string,&block)
+            new(string).each(&block)
+          end
+
+          #
+          # Iterates over each IP address that is included in the addresses
+          # netmask. Supports both IPv4 and IPv6 addresses.
+          #
+          # @yield [ip]
+          #   The block which will be passed every IP address covered be the
+          #   netmask of the IPAddr object.
+          #
+          # @yieldparam [String] ip
+          #   An IP address.
+          #
+          # @return [self]
+          #
+          # @note
+          #   This method will skip IPv4 addresses ending in `.0` or `.255`.
+          #
+          # @example
+          #   cidr = IPAddr.new('10.1.1.1/24')
+          #   cidr.each { |ip| puts ip }
+          #   # 10.0.0.1
+          #   # 10.0.0.2
+          #   # ...
+          #   # 10.0.0.253
+          #   # 10.0.0.254
+          #
+          def each
+            return enum_for(__method__) unless block_given?
+
+            family_mask = MASKS[@family]
+            octet_mask  = if ipv6? then 0xffff
+                          else          0xff
+                          end
+
+            (0..((~@mask_addr) & family_mask)).each do |i|
+              ip_uint = (@addr | i)
+
+              # skip IPv4 addresses ending in .0 or .255
+              if (ipv4? && ((ip_uint & 0xff) == 0 || (ip_uint & 0xff) == 0xff))
+                next
+              end
+
+              yield _to_string(ip_uint)
+            end
+
+            return self
+          end
+
+          #
+          # The first IP address of the CIDR Range.
+          #
+          # @return [String]
+          #
+          def first
+            _to_string(@addr)
+          end
+
+          #
+          # The last IP address of the CIDR range.
+          #
+          # @return [String]
+          #
+          def last
+            _to_string(@addr | ~@mask_addr)
+          end
+
+          #
+          # Converts the CIDR range back into a String.
+          #
+          # @return [String]
+          #
+          def to_s
+            @string
+          end
+
+          #
+          # Inspects the CIDR range.
+          #
+          # @return [String]
+          #
+          def inspect
+            "#<#{self.class}: #{@string}>"
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/ronin/support/network/ip_range/glob.rb

@@ -0,0 +1,309 @@
+# frozen_string_literal: true
+#
+# Copyright (c) 2006-2022 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# 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 'combinatorics/list_comprehension'
+
+module Ronin
+  module Support
+    module Network
+      class IPRange
+        #
+        # Represents an IP-glob range.
+        #
+        # ## Examples
+        #
+        #     ip_range = IPRange::Glob.new('10.0.1-3.*/24')
+        #     ip_range.each { |ip| puts ip }
+        #     # 10.0.1.1
+        #     # 10.0.1.2
+        #     # ...
+        #     # 10.0.1.253
+        #     # 10.0.1.254
+        #     # ...
+        #     # 10.0.2.1
+        #     # 10.0.2.2
+        #     # ...
+        #     # 10.0.2.253
+        #     # 10.0.2.254
+        #     # ...
+        #     # 10.0.3.1
+        #     # 10.0.3.2
+        #     # ...
+        #     # 10.0.3.253
+        #     # 10.0.3.254
+        #
+        # @api public
+        #
+        # @since 1.0.0
+        #
+        class Glob
+
+          include Enumerable
+
+          # The IP glob string.
+          #
+          # @return [String]
+          attr_reader :string
+
+          #
+          # Initializes and parses the IP-glob range.
+          #
+          # @param [String] string
+          #   The IP-glob string to parse.
+          #
+          def initialize(string)
+            @string = string
+
+            if @string.include?(':') # IPv6
+              @version   = 6
+              @separator = ':'
+              @base      = 16
+              @formatter = method(:format_ipv6_address)
+            else # IPv4
+              @version   = 4
+              @separator = '.'
+              @base      = 10
+              @formatter = method(:format_ipv4_address)
+            end
+
+            @ranges = @string.split(@separator).map do |segment|
+              case segment
+              when '*' then (1..254)
+              when /,/ then parse_list(segment)
+              when /-/ then parse_range(segment)
+              else          [segment]
+              end
+            end
+          end
+
+          #
+          # Alias for {#initialize new}.
+          #
+          # @param [String] string
+          #   The IP-glob string to parse.
+          #
+          # @return [Glob]
+          #   The parsed IP-glob range.
+          #
+          # @see #initialize
+          #
+          def self.parse(string)
+            new(string)
+          end
+
+          #
+          # Enumerates over the IP-glob range.
+          #
+          # @param [String] string
+          #   The IP-glob string to parse and enumerate over.
+          #
+          # @yield [ip]
+          #   The block which will be passed each IP address contained within
+          #   the IP address range.
+          #
+          # @yieldparam [String] ip
+          #   An IP address within the IP address range.
+          #
+          # @return [self]
+          #
+          # @example Enumerate through a IPv4 glob range:
+          #   IPRange::Glob.each('10.0.1-3.*') { |ip| puts ip }
+          #   # 10.0.1.1
+          #   # 10.0.1.2
+          #   # ...
+          #   # 10.0.1.253
+          #   # 10.0.1.254
+          #   # ...
+          #   # 10.0.2.1
+          #   # 10.0.2.2
+          #   # ...
+          #   # 10.0.2.253
+          #   # 10.0.2.254
+          #   # ...
+          #   # 10.0.3.1
+          #   # 10.0.3.2
+          #   # ...
+          #   # 10.0.3.253
+          #   # 10.0.3.254
+          #
+          # @example Enumerate through a globbed IPv6 range:
+          #   IPRange::Glob.each('::ff::02-0a::c3') { |ip| puts ip }
+          #
+          def self.each(string,&block)
+            new(string).each(&block)
+          end
+
+          #
+          # Determines if the IP-glob range is IPv4.
+          #
+          # @return [Boolean]
+          #
+          def ipv4?
+            @version == 4
+          end
+
+          #
+          # Determines if the IP-glob range is IPv6.
+          #
+          # @return [Boolean]
+          #
+          def ipv6?
+            @version == 6
+          end
+
+          #
+          # Enumerates over the IP-glob range.
+          #
+          # @yield [ip]
+          #   The block which will be passed each IP address contained within
+          #   the IP address range.
+          #
+          # @yieldparam [String] ip
+          #   An IP address within the IP address range.
+          #
+          # @return [self]
+          #
+          # @example Enumerate through a IPv4 glob range:
+          #   ip_range = IPRange::Glob.new('10.0.1-3.*')
+          #   ip_range.each { |ip| puts ip }
+          #   # 10.0.1.1
+          #   # 10.0.1.2
+          #   # ...
+          #   # 10.0.1.253
+          #   # 10.0.1.254
+          #   # ...
+          #   # 10.0.2.1
+          #   # 10.0.2.2
+          #   # ...
+          #   # 10.0.2.253
+          #   # 10.0.2.254
+          #   # ...
+          #   # 10.0.3.1
+          #   # 10.0.3.2
+          #   # ...
+          #   # 10.0.3.253
+          #   # 10.0.3.254
+          #
+          # @example Enumerate through a globbed IPv6 range:
+          #   ip_range = IPRange::Glob.new('::ff::02-0a::c3')
+          #   ip_range.each { |ip| puts ip }
+          #
+          def each
+            return enum_for(__method__) unless block_given?
+
+            # cycle through the address ranges
+            @ranges.comprehension do |address|
+              yield @formatter.call(address)
+            end
+
+            return self
+          end
+
+          #
+          # Converts the IP-glob range back into a String.
+          #
+          # @return [String]
+          #
+          def to_s
+            @string
+          end
+
+          #
+          # Inspects the IP-glob range.
+          #
+          # @return [String]
+          #
+          def inspect
+            "#<#{self.class}: #{@string}>"
+          end
+
+          private
+
+          #
+          # Parses a comma separated list of numbers or ranges (ex: `i,j,k-l`).
+          #
+          # @param [String] list
+          #   The string to parse.
+          #
+          # @return [Array<Integer, Range>]
+          #   The parsed list.
+          #
+          def parse_list(list)
+            list.split(',').flat_map do |octet|
+              if octet.include?('-')
+                # i-j range
+                parse_range(octet)
+              else
+                octet.to_i(@base)
+              end
+            end
+          end
+
+          #
+          # Parses a range of numbers. (ex: `i-j`).
+          #
+          # @param [String] range
+          #   The string to parse.
+          #
+          # @return [Range<Integer,Integer>]
+          #   The parsed range.
+          #
+          def parse_range(range)
+            start, stop = range.split('-',2)
+            start = start.to_i(@base)
+            stop  = stop.to_i(@base)
+
+            (start..stop).to_a
+          end
+
+          #
+          # Formats an IPv4 address.
+          #
+          # @param [Array<String, Integer>] parts
+          #   The address parts to format.
+          #
+          # @return [String]
+          #   The formatted IPv4 address.
+          #
+          def format_ipv4_address(parts)
+            parts.join('.')
+          end
+
+          #
+          # Formats an IPv6 address.
+          #
+          # @param [Array<String, Integer>] parts
+          #   The address parts to format.
+          #
+          # @return [String]
+          #   The formatted IPv6 address.
+          #
+          def format_ipv6_address(parts)
+            parts.map { |part|
+              case part
+              when Integer then part.to_s(16)
+              else              part
+              end
+            }.join(':')
+          end
+
+        end
+      end
+    end
+  end
+end