class IPAddr

# File rake/lib/ipaddr.rb, line 452
def initialize(addr = '::', family = Socket::AF_UNSPEC)
  if !addr.kind_of?(String)
    case family
    when Socket::AF_INET, Socket::AF_INET6
      set(addr.to_i, family)
      @mask_addr = (family == Socket::AF_INET) ? IN4MASK : IN6MASK
      return
    when Socket::AF_UNSPEC
      raise ArgumentError, "address family must be specified"
    else
      raise ArgumentError, "unsupported address family: #{family}"
    end
  end
  prefix, prefixlen = addr.split('/')
  if prefix =~ /^\[(.*)\]$/
    prefix = $1
    family = Socket::AF_INET6
  end
  # It seems AI_NUMERICHOST doesn't do the job.
  #Socket.getaddrinfo(left, nil, Socket::AF_INET6, Socket::SOCK_STREAM, nil,
  #                  Socket::AI_NUMERICHOST)
  @addr = @family = nil
  if family == Socket::AF_UNSPEC || family == Socket::AF_INET
    @addr = in_addr(prefix)
    if @addr
      @family = Socket::AF_INET
    end
  end
  if !@addr && (family == Socket::AF_UNSPEC || family == Socket::AF_INET6)
    @addr = in6_addr(prefix)
    @family = Socket::AF_INET6
  end
  if family != Socket::AF_UNSPEC && @family != family
    raise ArgumentError, "address family mismatch"
  end
  if prefixlen
    mask!(prefixlen)
  else
    @mask_addr = (@family == Socket::AF_INET) ? IN4MASK : IN6MASK
  end
end

# File rake/lib/ipaddr.rb, line 90
def IPAddr::new_ntoh(addr)
  return IPAddr.new(IPAddr::ntop(addr))
end

# File rake/lib/ipaddr.rb, line 96
def IPAddr::ntop(addr)
  case addr.size
  when 4
    s = addr.unpack('C4').join('.')
  when 16
    s = IN6FORMAT % addr.unpack('n8')
  else
    raise ArgumentError, "unsupported address family"
  end
  return s
end

# File rake/lib/ipaddr.rb, line 109
def &(other)
  return self.clone.set(@addr & coerce_other(other).to_i)
end

# File rake/lib/ipaddr.rb, line 124
def <<(num)
  return self.clone.set(addr_mask(@addr << num))
end

# File rake/lib/ipaddr.rb, line 320
def <=>(other)
  other = coerce_other(other)

  return nil if other.family != @family

  return @addr <=> other.to_i
end

# File rake/lib/ipaddr.rb, line 134
def ==(other)
  other = coerce_other(other)
  return @family == other.family && @addr == other.to_i
end

# File rake/lib/ipaddr.rb, line 119
def >>(num)
  return self.clone.set(@addr >> num)
end

# File rake/lib/ipaddr.rb, line 330
def eql?(other)
  return self.class == other.class && self.hash == other.hash && self == other
end

# File rake/lib/ipaddr.rb, line 335
def hash
  return ([@addr, @mask_addr].hash << 1) | (ipv4? ? 0 : 1)
end

# File rake/lib/ipaddr.rb, line 220
def hton
  case @family
  when Socket::AF_INET
    return [@addr].pack('N')
  when Socket::AF_INET6
    return (0..7).map { |i|
      (@addr >> (112 - 16 * i)) & 0xffff
    }.pack('n8')
  else
    raise "unsupported address family"
  end
end

# File rake/lib/ipaddr.rb, line 154
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))
end

# File rake/lib/ipaddr.rb, line 357
def inspect
  case @family
  when Socket::AF_INET
    af = "IPv4"
  when Socket::AF_INET6
    af = "IPv6"
  else
    raise "unsupported address family"
  end
  return sprintf("#<%s: %s:%s/%s>", self.class.name,
                 af, _to_string(@addr), _to_string(@mask_addr))
end

# File rake/lib/ipaddr.rb, line 299
def ip6_arpa
  if !ipv6?
    raise ArgumentError, "not an IPv6 address"
  end
  return _reverse + ".ip6.arpa"
end

# File rake/lib/ipaddr.rb, line 307
def ip6_int
  if !ipv6?
    raise ArgumentError, "not an IPv6 address"
  end
  return _reverse + ".ip6.int"
end

# File rake/lib/ipaddr.rb, line 234
def ipv4?
  return @family == Socket::AF_INET
end

# File rake/lib/ipaddr.rb, line 268
def ipv4_compat
  if !ipv4?
    raise ArgumentError, "not an IPv4 address"
  end
  return self.clone.set(@addr, Socket::AF_INET6)
end

# File rake/lib/ipaddr.rb, line 249
def ipv4_compat?
  if !ipv6? || (@addr >> 32) != 0
    return false
  end
  a = (@addr & IN4MASK)
  return a != 0 && a != 1
end

# File rake/lib/ipaddr.rb, line 259
def ipv4_mapped
  if !ipv4?
    raise ArgumentError, "not an IPv4 address"
  end
  return self.clone.set(@addr | 0xffff00000000, Socket::AF_INET6)
end

# File rake/lib/ipaddr.rb, line 244
def ipv4_mapped?
  return ipv6? && (@addr >> 32) == 0xffff
end

# File rake/lib/ipaddr.rb, line 239
def ipv6?
  return @family == Socket::AF_INET6
end

# File rake/lib/ipaddr.rb, line 141
def mask(prefixlen)
  return self.clone.mask!(prefixlen)
end

# File rake/lib/ipaddr.rb, line 278
def native
  if !ipv4_mapped? && !ipv4_compat?
    return self
  end
  return self.clone.set(@addr & IN4MASK, Socket::AF_INET)
end

# File rake/lib/ipaddr.rb, line 287
def reverse
  case @family
  when Socket::AF_INET
    return _reverse + ".in-addr.arpa"
  when Socket::AF_INET6
    return ip6_arpa
  else
    raise "unsupported address family"
  end
end

# File rake/lib/ipaddr.rb, line 315
def succ
  return self.clone.set(@addr + 1, @family)
end

# File rake/lib/ipaddr.rb, line 184
def to_i
  return @addr
end

# File rake/lib/ipaddr.rb, line 340
def to_range
  begin_addr = (@addr & @mask_addr)

  case @family
  when Socket::AF_INET
    end_addr = (@addr | (IN4MASK ^ @mask_addr))
  when Socket::AF_INET6
    end_addr = (@addr | (IN6MASK ^ @mask_addr))
  else
    raise "unsupported address family"
  end

  return clone.set(begin_addr, @family)..clone.set(end_addr, @family)
end

# File rake/lib/ipaddr.rb, line 189
def to_s
  str = to_string
  return str if ipv4?

  str.gsub!(/\b0{1,3}([\da-f]+)\b/, '\1')
  loop do
    break if str.sub!(/\A0:0:0:0:0:0:0:0\z/, '::')
    break if str.sub!(/\b0:0:0:0:0:0:0\b/, ':')
    break if str.sub!(/\b0:0:0:0:0:0\b/, ':')
    break if str.sub!(/\b0:0:0:0:0\b/, ':')
    break if str.sub!(/\b0:0:0:0\b/, ':')
    break if str.sub!(/\b0:0:0\b/, ':')
    break if str.sub!(/\b0:0\b/, ':')
    break
  end
  str.sub!(/:{3,}/, '::')

  if /\A::(ffff:)?([\da-f]{1,4}):([\da-f]{1,4})\z/ =~ str
    str = sprintf('::%s%d.%d.%d.%d', $1, $2.hex / 256, $2.hex % 256, $3.hex / 256, $3.hex % 256)
  end

  str
end

# File rake/lib/ipaddr.rb, line 215
def to_string
  return _to_string(@addr)
end

# File rake/lib/ipaddr.rb, line 114
def |(other)
  return self.clone.set(@addr | coerce_other(other).to_i)
end

# File rake/lib/ipaddr.rb, line 129
def ~
  return self.clone.set(addr_mask(~@addr))
end

# File rake/lib/ipaddr.rb, line 396
def mask!(mask)
  if mask.kind_of?(String)
    if mask =~ /^\d+$/
      prefixlen = mask.to_i
    else
      m = IPAddr.new(mask)
      if m.family != @family
        raise ArgumentError, "address family is not same"
      end
      @mask_addr = m.to_i
      @addr &= @mask_addr
      return self
    end
  else
    prefixlen = mask
  end
  case @family
  when Socket::AF_INET
    if prefixlen < 0 || prefixlen > 32
      raise ArgumentError, "invalid length"
    end
    masklen = 32 - prefixlen
    @mask_addr = ((IN4MASK >> masklen) << masklen)
  when Socket::AF_INET6
    if prefixlen < 0 || prefixlen > 128
      raise ArgumentError, "invalid length"
    end
    masklen = 128 - prefixlen
    @mask_addr = ((IN6MASK >> masklen) << masklen)
  else
    raise "unsupported address family"
  end
  @addr = ((@addr >> masklen) << masklen)
  return self
end

# File rake/lib/ipaddr.rb, line 375
def set(addr, *family)
  case family[0] ? family[0] : @family
  when Socket::AF_INET
    if addr < 0 || addr > IN4MASK
      raise ArgumentError, "invalid address"
    end
  when Socket::AF_INET6
    if addr < 0 || addr > IN6MASK
      raise ArgumentError, "invalid address"
    end
  else
    raise ArgumentError, "unsupported address family"
  end
  @addr = addr
  if family[0]
    @family = family[0]
  end
  return self
end