You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
901 B
32 lines
901 B
# frozen_string_literal: true |
|
|
|
class FastIpMap |
|
MAX_IPV4_PREFIX = 32 |
|
MAX_IPV6_PREFIX = 128 |
|
|
|
# @param [Enumerable<IPAddr>] addresses |
|
def initialize(addresses) |
|
@fast_lookup = {} |
|
@ranges = [] |
|
|
|
# Hash look-up is faster but only works for exact matches, so we split |
|
# exact addresses from non-exact ones |
|
addresses.each do |address| |
|
if (address.ipv4? && address.prefix == MAX_IPV4_PREFIX) || (address.ipv6? && address.prefix == MAX_IPV6_PREFIX) |
|
@fast_lookup[address.to_s] = true |
|
else |
|
@ranges << address |
|
end |
|
end |
|
|
|
# We're more likely to hit wider-reaching ranges when checking for |
|
# inclusion, so make sure they're sorted first |
|
@ranges.sort_by!(&:prefix) |
|
end |
|
|
|
# @param [IPAddr] address |
|
# @return [Boolean] |
|
def include?(address) |
|
@fast_lookup[address.to_s] || @ranges.any? { |cidr| cidr.include?(address) } |
|
end |
|
end
|
|
|