diff --git a/lib/ipaddr.rb b/lib/ipaddr.rb index 56a2e77..cd40d35 100644 --- a/lib/ipaddr.rb +++ b/lib/ipaddr.rb @@ -168,34 +168,15 @@ def mask(prefixlen) # net1 = IPAddr.new("192.168.2.0/24") # net2 = IPAddr.new("192.168.2.100") # net3 = IPAddr.new("192.168.3.0") + # net4 = IPAddr.new("192.168.2.0/16") # p net1.include?(net2) #=> true # p net1.include?(net3) #=> false + # p net1.include?(net4) #=> false + # p net4.include?(net1) #=> true def include?(other) - other = coerce_other(other) - if ipv4_mapped? - if (@mask_addr >> 32) != 0xffffffffffffffffffffffff - return false - end - mask_addr = (@mask_addr & IN4MASK) - addr = (@addr & IN4MASK) - family = Socket::AF_INET - else - mask_addr = @mask_addr - addr = @addr - family = @family - end - if other.ipv4_mapped? - other_addr = (other.to_i & IN4MASK) - other_family = Socket::AF_INET - else - other_addr = other.to_i - other_family = other.family - end - - if family != other_family - return false - end - return ((addr & mask_addr) == (other_addr & mask_addr)) + range = to_range + other = coerce_other(other).to_range + range.begin <= other.begin && range.end >= other.end end alias === include? diff --git a/test/test_ipaddr.rb b/test/test_ipaddr.rb index d894a17..42b5214 100644 --- a/test/test_ipaddr.rb +++ b/test/test_ipaddr.rb @@ -303,6 +303,8 @@ def test_include? assert_equal(true, net1.include?(IPAddr.new("192.168.2.0"))) assert_equal(true, net1.include?(IPAddr.new("192.168.2.255"))) assert_equal(false, net1.include?(IPAddr.new("192.168.3.0"))) + assert_equal(true, net1.include?(IPAddr.new("192.168.2.0/28"))) + assert_equal(false, net1.include?(IPAddr.new("192.168.2.0/16"))) # test with integer parameter int = (192 << 24) + (168 << 16) + (2 << 8) + 13