Contents

require 'ipaddress'

module ForestLiana
  class IpWhitelistChecker
    module RuleType
      IP = 0
      RANGE = 1
      SUBNET = 2
    end

    def self.is_ip_matches_any_rule(ip, rules)
      rules.any? { |rule| IpWhitelistChecker.is_ip_matches_rule(ip, rule) }
    end

    def self.is_ip_matches_rule(ip, rule)
      if rule['type'] == RuleType::IP
        return IpWhitelistChecker.is_ip_match_ip(ip, rule['ip'])
      elsif rule['type'] == RuleType::RANGE
        return IpWhitelistChecker.is_ip_match_range(ip, rule)
      elsif rule['type'] == RuleType::SUBNET
        return IpWhitelistChecker.is_ip_match_subnet(ip, rule['range'])
      end

      raise 'Invalid rule type'
    end

    def self.ip_version(ip)
      (IPAddress ip).is_a?(IPAddress::IPv4) ? :ip_v4 : :ip_v6
    end

    def self.is_same_ip_version(ip1, ip2)
      ip1_version = IpWhitelistChecker.ip_version(ip1)
      ip2_version = IpWhitelistChecker.ip_version(ip2)

      ip1_version == ip2_version
    end

    def self.is_both_loopback(ip1, ip2)
      IPAddress(ip1).loopback? && IPAddress(ip2).loopback?
    end

    def self.is_ip_match_ip(ip1, ip2)
      if !IpWhitelistChecker.is_same_ip_version(ip1, ip2)
        return IpWhitelistChecker.is_both_loopback(ip1, ip2)
      end

      if IPAddress(ip1) == IPAddress(ip2)
        true
      else
        IpWhitelistChecker.is_both_loopback(ip1, ip2)
      end
    end

    def self.is_ip_match_range(ip, rule)
      return false if !IpWhitelistChecker.is_same_ip_version(ip, rule['ip_minimum'])

      ip_range_minimum = (IPAddress rule['ip_minimum']).to_i
      ip_range_maximum = (IPAddress rule['ip_maximum']).to_i
      ip_value = (IPAddress ip).to_i

      return ip_value >= ip_range_minimum && ip_value <= ip_range_maximum
    end

    def self.is_ip_match_subnet(ip, subnet)
      return false if !IpWhitelistChecker.is_same_ip_version(ip, subnet)

      IPAddress(subnet).include?(IPAddress(ip))
    end
  end

end

Version data entries

155 entries across 155 versions & 1 rubygems

Version	Path
forest_liana-9.10.0	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.9.1	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.9.0	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.8.0	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.7.0	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.6.4	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.6.3	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.6.0	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.5.7	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.5.6	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.5.5	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.5.4	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.5.3	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.5.2	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.5.1	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.5.0	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.4.3	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.4.2	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.4.1	app/services/forest_liana/ip_whitelist_checker.rb
forest_liana-9.4.0	app/services/forest_liana/ip_whitelist_checker.rb