1498 lines
42 KiB
Go
1498 lines
42 KiB
Go
// Copyright 2020 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// Package netip defines an IP address type that's a small value type.
|
|
// Building on that Addr type, the package also defines AddrPort (an
|
|
// IP address and a port), and Prefix (an IP address and a bit length
|
|
// prefix).
|
|
//
|
|
// Compared to the net.IP type, this package's Addr type takes less
|
|
// memory, is immutable, and is comparable (supports == and being a
|
|
// map key).
|
|
package netip
|
|
|
|
import (
|
|
"errors"
|
|
"math"
|
|
"strconv"
|
|
|
|
"internal/bytealg"
|
|
"internal/intern"
|
|
"internal/itoa"
|
|
)
|
|
|
|
// Sizes: (64-bit)
|
|
// net.IP: 24 byte slice header + {4, 16} = 28 to 40 bytes
|
|
// net.IPAddr: 40 byte slice header + {4, 16} = 44 to 56 bytes + zone length
|
|
// netip.Addr: 24 bytes (zone is per-name singleton, shared across all users)
|
|
|
|
// Addr represents an IPv4 or IPv6 address (with or without a scoped
|
|
// addressing zone), similar to net.IP or net.IPAddr.
|
|
//
|
|
// Unlike net.IP or net.IPAddr, Addr is a comparable value
|
|
// type (it supports == and can be a map key) and is immutable.
|
|
//
|
|
// The zero Addr is not a valid IP address.
|
|
// Addr{} is distinct from both 0.0.0.0 and ::.
|
|
type Addr struct {
|
|
// addr is the hi and lo bits of an IPv6 address. If z==z4,
|
|
// hi and lo contain the IPv4-mapped IPv6 address.
|
|
//
|
|
// hi and lo are constructed by interpreting a 16-byte IPv6
|
|
// address as a big-endian 128-bit number. The most significant
|
|
// bits of that number go into hi, the rest into lo.
|
|
//
|
|
// For example, 0011:2233:4455:6677:8899:aabb:ccdd:eeff is stored as:
|
|
// addr.hi = 0x0011223344556677
|
|
// addr.lo = 0x8899aabbccddeeff
|
|
//
|
|
// We store IPs like this, rather than as [16]byte, because it
|
|
// turns most operations on IPs into arithmetic and bit-twiddling
|
|
// operations on 64-bit registers, which is much faster than
|
|
// bytewise processing.
|
|
addr uint128
|
|
|
|
// z is a combination of the address family and the IPv6 zone.
|
|
//
|
|
// nil means invalid IP address (for a zero Addr).
|
|
// z4 means an IPv4 address.
|
|
// z6noz means an IPv6 address without a zone.
|
|
//
|
|
// Otherwise it's the interned zone name string.
|
|
z *intern.Value
|
|
}
|
|
|
|
// z0, z4, and z6noz are sentinel IP.z values.
|
|
// See the IP type's field docs.
|
|
var (
|
|
z0 = (*intern.Value)(nil)
|
|
z4 = new(intern.Value)
|
|
z6noz = new(intern.Value)
|
|
)
|
|
|
|
// IPv6LinkLocalAllNodes returns the IPv6 link-local all nodes multicast
|
|
// address ff02::1.
|
|
func IPv6LinkLocalAllNodes() Addr { return AddrFrom16([16]byte{0: 0xff, 1: 0x02, 15: 0x01}) }
|
|
|
|
// IPv6Unspecified returns the IPv6 unspecified address "::".
|
|
func IPv6Unspecified() Addr { return Addr{z: z6noz} }
|
|
|
|
// IPv4Unspecified returns the IPv4 unspecified address "0.0.0.0".
|
|
func IPv4Unspecified() Addr { return AddrFrom4([4]byte{}) }
|
|
|
|
// AddrFrom4 returns the address of the IPv4 address given by the bytes in addr.
|
|
func AddrFrom4(addr [4]byte) Addr {
|
|
return Addr{
|
|
addr: uint128{0, 0xffff00000000 | uint64(addr[0])<<24 | uint64(addr[1])<<16 | uint64(addr[2])<<8 | uint64(addr[3])},
|
|
z: z4,
|
|
}
|
|
}
|
|
|
|
// AddrFrom16 returns the IPv6 address given by the bytes in addr.
|
|
// An IPv6-mapped IPv4 address is left as an IPv6 address.
|
|
// (Use Unmap to convert them if needed.)
|
|
func AddrFrom16(addr [16]byte) Addr {
|
|
return Addr{
|
|
addr: uint128{
|
|
beUint64(addr[:8]),
|
|
beUint64(addr[8:]),
|
|
},
|
|
z: z6noz,
|
|
}
|
|
}
|
|
|
|
// ipv6Slice is like IPv6Raw, but operates on a 16-byte slice. Assumes
|
|
// slice is 16 bytes, caller must enforce this.
|
|
func ipv6Slice(addr []byte) Addr {
|
|
return Addr{
|
|
addr: uint128{
|
|
beUint64(addr[:8]),
|
|
beUint64(addr[8:]),
|
|
},
|
|
z: z6noz,
|
|
}
|
|
}
|
|
|
|
// ParseAddr parses s as an IP address, returning the result. The string
|
|
// s can be in dotted decimal ("192.0.2.1"), IPv6 ("2001:db8::68"),
|
|
// or IPv6 with a scoped addressing zone ("fe80::1cc0:3e8c:119f:c2e1%ens18").
|
|
func ParseAddr(s string) (Addr, error) {
|
|
for i := 0; i < len(s); i++ {
|
|
switch s[i] {
|
|
case '.':
|
|
return parseIPv4(s)
|
|
case ':':
|
|
return parseIPv6(s)
|
|
case '%':
|
|
// Assume that this was trying to be an IPv6 address with
|
|
// a zone specifier, but the address is missing.
|
|
return Addr{}, parseAddrError{in: s, msg: "missing IPv6 address"}
|
|
}
|
|
}
|
|
return Addr{}, parseAddrError{in: s, msg: "unable to parse IP"}
|
|
}
|
|
|
|
// MustParseAddr calls ParseAddr(s) and panics on error.
|
|
// It is intended for use in tests with hard-coded strings.
|
|
func MustParseAddr(s string) Addr {
|
|
ip, err := ParseAddr(s)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return ip
|
|
}
|
|
|
|
type parseAddrError struct {
|
|
in string // the string given to ParseAddr
|
|
msg string // an explanation of the parse failure
|
|
at string // optionally, the unparsed portion of in at which the error occurred.
|
|
}
|
|
|
|
func (err parseAddrError) Error() string {
|
|
q := strconv.Quote
|
|
if err.at != "" {
|
|
return "ParseAddr(" + q(err.in) + "): " + err.msg + " (at " + q(err.at) + ")"
|
|
}
|
|
return "ParseAddr(" + q(err.in) + "): " + err.msg
|
|
}
|
|
|
|
// parseIPv4 parses s as an IPv4 address (in form "192.168.0.1").
|
|
func parseIPv4(s string) (ip Addr, err error) {
|
|
var fields [4]uint8
|
|
var val, pos int
|
|
var digLen int // number of digits in current octet
|
|
for i := 0; i < len(s); i++ {
|
|
if s[i] >= '0' && s[i] <= '9' {
|
|
if digLen == 1 && val == 0 {
|
|
return Addr{}, parseAddrError{in: s, msg: "IPv4 field has octet with leading zero"}
|
|
}
|
|
val = val*10 + int(s[i]) - '0'
|
|
digLen++
|
|
if val > 255 {
|
|
return Addr{}, parseAddrError{in: s, msg: "IPv4 field has value >255"}
|
|
}
|
|
} else if s[i] == '.' {
|
|
// .1.2.3
|
|
// 1.2.3.
|
|
// 1..2.3
|
|
if i == 0 || i == len(s)-1 || s[i-1] == '.' {
|
|
return Addr{}, parseAddrError{in: s, msg: "IPv4 field must have at least one digit", at: s[i:]}
|
|
}
|
|
// 1.2.3.4.5
|
|
if pos == 3 {
|
|
return Addr{}, parseAddrError{in: s, msg: "IPv4 address too long"}
|
|
}
|
|
fields[pos] = uint8(val)
|
|
pos++
|
|
val = 0
|
|
digLen = 0
|
|
} else {
|
|
return Addr{}, parseAddrError{in: s, msg: "unexpected character", at: s[i:]}
|
|
}
|
|
}
|
|
if pos < 3 {
|
|
return Addr{}, parseAddrError{in: s, msg: "IPv4 address too short"}
|
|
}
|
|
fields[3] = uint8(val)
|
|
return AddrFrom4(fields), nil
|
|
}
|
|
|
|
// parseIPv6 parses s as an IPv6 address (in form "2001:db8::68").
|
|
func parseIPv6(in string) (Addr, error) {
|
|
s := in
|
|
|
|
// Split off the zone right from the start. Yes it's a second scan
|
|
// of the string, but trying to handle it inline makes a bunch of
|
|
// other inner loop conditionals more expensive, and it ends up
|
|
// being slower.
|
|
zone := ""
|
|
i := bytealg.IndexByteString(s, '%')
|
|
if i != -1 {
|
|
s, zone = s[:i], s[i+1:]
|
|
if zone == "" {
|
|
// Not allowed to have an empty zone if explicitly specified.
|
|
return Addr{}, parseAddrError{in: in, msg: "zone must be a non-empty string"}
|
|
}
|
|
}
|
|
|
|
var ip [16]byte
|
|
ellipsis := -1 // position of ellipsis in ip
|
|
|
|
// Might have leading ellipsis
|
|
if len(s) >= 2 && s[0] == ':' && s[1] == ':' {
|
|
ellipsis = 0
|
|
s = s[2:]
|
|
// Might be only ellipsis
|
|
if len(s) == 0 {
|
|
return IPv6Unspecified().WithZone(zone), nil
|
|
}
|
|
}
|
|
|
|
// Loop, parsing hex numbers followed by colon.
|
|
i = 0
|
|
for i < 16 {
|
|
// Hex number. Similar to parseIPv4, inlining the hex number
|
|
// parsing yields a significant performance increase.
|
|
off := 0
|
|
acc := uint32(0)
|
|
for ; off < len(s); off++ {
|
|
c := s[off]
|
|
if c >= '0' && c <= '9' {
|
|
acc = (acc << 4) + uint32(c-'0')
|
|
} else if c >= 'a' && c <= 'f' {
|
|
acc = (acc << 4) + uint32(c-'a'+10)
|
|
} else if c >= 'A' && c <= 'F' {
|
|
acc = (acc << 4) + uint32(c-'A'+10)
|
|
} else {
|
|
break
|
|
}
|
|
if acc > math.MaxUint16 {
|
|
// Overflow, fail.
|
|
return Addr{}, parseAddrError{in: in, msg: "IPv6 field has value >=2^16", at: s}
|
|
}
|
|
}
|
|
if off == 0 {
|
|
// No digits found, fail.
|
|
return Addr{}, parseAddrError{in: in, msg: "each colon-separated field must have at least one digit", at: s}
|
|
}
|
|
|
|
// If followed by dot, might be in trailing IPv4.
|
|
if off < len(s) && s[off] == '.' {
|
|
if ellipsis < 0 && i != 12 {
|
|
// Not the right place.
|
|
return Addr{}, parseAddrError{in: in, msg: "embedded IPv4 address must replace the final 2 fields of the address", at: s}
|
|
}
|
|
if i+4 > 16 {
|
|
// Not enough room.
|
|
return Addr{}, parseAddrError{in: in, msg: "too many hex fields to fit an embedded IPv4 at the end of the address", at: s}
|
|
}
|
|
// TODO: could make this a bit faster by having a helper
|
|
// that parses to a [4]byte, and have both parseIPv4 and
|
|
// parseIPv6 use it.
|
|
ip4, err := parseIPv4(s)
|
|
if err != nil {
|
|
return Addr{}, parseAddrError{in: in, msg: err.Error(), at: s}
|
|
}
|
|
ip[i] = ip4.v4(0)
|
|
ip[i+1] = ip4.v4(1)
|
|
ip[i+2] = ip4.v4(2)
|
|
ip[i+3] = ip4.v4(3)
|
|
s = ""
|
|
i += 4
|
|
break
|
|
}
|
|
|
|
// Save this 16-bit chunk.
|
|
ip[i] = byte(acc >> 8)
|
|
ip[i+1] = byte(acc)
|
|
i += 2
|
|
|
|
// Stop at end of string.
|
|
s = s[off:]
|
|
if len(s) == 0 {
|
|
break
|
|
}
|
|
|
|
// Otherwise must be followed by colon and more.
|
|
if s[0] != ':' {
|
|
return Addr{}, parseAddrError{in: in, msg: "unexpected character, want colon", at: s}
|
|
} else if len(s) == 1 {
|
|
return Addr{}, parseAddrError{in: in, msg: "colon must be followed by more characters", at: s}
|
|
}
|
|
s = s[1:]
|
|
|
|
// Look for ellipsis.
|
|
if s[0] == ':' {
|
|
if ellipsis >= 0 { // already have one
|
|
return Addr{}, parseAddrError{in: in, msg: "multiple :: in address", at: s}
|
|
}
|
|
ellipsis = i
|
|
s = s[1:]
|
|
if len(s) == 0 { // can be at end
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
// Must have used entire string.
|
|
if len(s) != 0 {
|
|
return Addr{}, parseAddrError{in: in, msg: "trailing garbage after address", at: s}
|
|
}
|
|
|
|
// If didn't parse enough, expand ellipsis.
|
|
if i < 16 {
|
|
if ellipsis < 0 {
|
|
return Addr{}, parseAddrError{in: in, msg: "address string too short"}
|
|
}
|
|
n := 16 - i
|
|
for j := i - 1; j >= ellipsis; j-- {
|
|
ip[j+n] = ip[j]
|
|
}
|
|
for j := ellipsis + n - 1; j >= ellipsis; j-- {
|
|
ip[j] = 0
|
|
}
|
|
} else if ellipsis >= 0 {
|
|
// Ellipsis must represent at least one 0 group.
|
|
return Addr{}, parseAddrError{in: in, msg: "the :: must expand to at least one field of zeros"}
|
|
}
|
|
return AddrFrom16(ip).WithZone(zone), nil
|
|
}
|
|
|
|
// AddrFromSlice parses the 4- or 16-byte byte slice as an IPv4 or IPv6 address.
|
|
// Note that a net.IP can be passed directly as the []byte argument.
|
|
// If slice's length is not 4 or 16, AddrFromSlice returns Addr{}, false.
|
|
func AddrFromSlice(slice []byte) (ip Addr, ok bool) {
|
|
switch len(slice) {
|
|
case 4:
|
|
return AddrFrom4(*(*[4]byte)(slice)), true
|
|
case 16:
|
|
return ipv6Slice(slice), true
|
|
}
|
|
return Addr{}, false
|
|
}
|
|
|
|
// v4 returns the i'th byte of ip. If ip is not an IPv4, v4 returns
|
|
// unspecified garbage.
|
|
func (ip Addr) v4(i uint8) uint8 {
|
|
return uint8(ip.addr.lo >> ((3 - i) * 8))
|
|
}
|
|
|
|
// v6 returns the i'th byte of ip. If ip is an IPv4 address, this
|
|
// accesses the IPv4-mapped IPv6 address form of the IP.
|
|
func (ip Addr) v6(i uint8) uint8 {
|
|
return uint8(*(ip.addr.halves()[(i/8)%2]) >> ((7 - i%8) * 8))
|
|
}
|
|
|
|
// v6u16 returns the i'th 16-bit word of ip. If ip is an IPv4 address,
|
|
// this accesses the IPv4-mapped IPv6 address form of the IP.
|
|
func (ip Addr) v6u16(i uint8) uint16 {
|
|
return uint16(*(ip.addr.halves()[(i/4)%2]) >> ((3 - i%4) * 16))
|
|
}
|
|
|
|
// isZero reports whether ip is the zero value of the IP type.
|
|
// The zero value is not a valid IP address of any type.
|
|
//
|
|
// Note that "0.0.0.0" and "::" are not the zero value. Use IsUnspecified to
|
|
// check for these values instead.
|
|
func (ip Addr) isZero() bool {
|
|
// Faster than comparing ip == Addr{}, but effectively equivalent,
|
|
// as there's no way to make an IP with a nil z from this package.
|
|
return ip.z == z0
|
|
}
|
|
|
|
// IsValid reports whether the Addr is an initialized address (not the zero Addr).
|
|
//
|
|
// Note that "0.0.0.0" and "::" are both valid values.
|
|
func (ip Addr) IsValid() bool { return ip.z != z0 }
|
|
|
|
// BitLen returns the number of bits in the IP address:
|
|
// 128 for IPv6, 32 for IPv4, and 0 for the zero Addr.
|
|
//
|
|
// Note that IPv4-mapped IPv6 addresses are considered IPv6 addresses
|
|
// and therefore have bit length 128.
|
|
func (ip Addr) BitLen() int {
|
|
switch ip.z {
|
|
case z0:
|
|
return 0
|
|
case z4:
|
|
return 32
|
|
}
|
|
return 128
|
|
}
|
|
|
|
// Zone returns ip's IPv6 scoped addressing zone, if any.
|
|
func (ip Addr) Zone() string {
|
|
if ip.z == nil {
|
|
return ""
|
|
}
|
|
zone, _ := ip.z.Get().(string)
|
|
return zone
|
|
}
|
|
|
|
// Compare returns an integer comparing two IPs.
|
|
// The result will be 0 if ip == ip2, -1 if ip < ip2, and +1 if ip > ip2.
|
|
// The definition of "less than" is the same as the Less method.
|
|
func (ip Addr) Compare(ip2 Addr) int {
|
|
f1, f2 := ip.BitLen(), ip2.BitLen()
|
|
if f1 < f2 {
|
|
return -1
|
|
}
|
|
if f1 > f2 {
|
|
return 1
|
|
}
|
|
hi1, hi2 := ip.addr.hi, ip2.addr.hi
|
|
if hi1 < hi2 {
|
|
return -1
|
|
}
|
|
if hi1 > hi2 {
|
|
return 1
|
|
}
|
|
lo1, lo2 := ip.addr.lo, ip2.addr.lo
|
|
if lo1 < lo2 {
|
|
return -1
|
|
}
|
|
if lo1 > lo2 {
|
|
return 1
|
|
}
|
|
if ip.Is6() {
|
|
za, zb := ip.Zone(), ip2.Zone()
|
|
if za < zb {
|
|
return -1
|
|
}
|
|
if za > zb {
|
|
return 1
|
|
}
|
|
}
|
|
return 0
|
|
}
|
|
|
|
// Less reports whether ip sorts before ip2.
|
|
// IP addresses sort first by length, then their address.
|
|
// IPv6 addresses with zones sort just after the same address without a zone.
|
|
func (ip Addr) Less(ip2 Addr) bool { return ip.Compare(ip2) == -1 }
|
|
|
|
func (ip Addr) lessOrEq(ip2 Addr) bool { return ip.Compare(ip2) <= 0 }
|
|
|
|
// Is4 reports whether ip is an IPv4 address.
|
|
//
|
|
// It returns false for IP4-mapped IPv6 addresses. See IP.Unmap.
|
|
func (ip Addr) Is4() bool {
|
|
return ip.z == z4
|
|
}
|
|
|
|
// Is4In6 reports whether ip is an IPv4-mapped IPv6 address.
|
|
func (ip Addr) Is4In6() bool {
|
|
return ip.Is6() && ip.addr.hi == 0 && ip.addr.lo>>32 == 0xffff
|
|
}
|
|
|
|
// Is6 reports whether ip is an IPv6 address, including IPv4-mapped
|
|
// IPv6 addresses.
|
|
func (ip Addr) Is6() bool {
|
|
return ip.z != z0 && ip.z != z4
|
|
}
|
|
|
|
// Unmap returns ip with any IPv4-mapped IPv6 address prefix removed.
|
|
//
|
|
// That is, if ip is an IPv6 address wrapping an IPv4 adddress, it
|
|
// returns the wrapped IPv4 address. Otherwise it returns ip unmodified.
|
|
func (ip Addr) Unmap() Addr {
|
|
if ip.Is4In6() {
|
|
ip.z = z4
|
|
}
|
|
return ip
|
|
}
|
|
|
|
// WithZone returns an IP that's the same as ip but with the provided
|
|
// zone. If zone is empty, the zone is removed. If ip is an IPv4
|
|
// address, WithZone is a no-op and returns ip unchanged.
|
|
func (ip Addr) WithZone(zone string) Addr {
|
|
if !ip.Is6() {
|
|
return ip
|
|
}
|
|
if zone == "" {
|
|
ip.z = z6noz
|
|
return ip
|
|
}
|
|
ip.z = intern.GetByString(zone)
|
|
return ip
|
|
}
|
|
|
|
// withoutZone unconditionally strips the zone from IP.
|
|
// It's similar to WithZone, but small enough to be inlinable.
|
|
func (ip Addr) withoutZone() Addr {
|
|
if !ip.Is6() {
|
|
return ip
|
|
}
|
|
ip.z = z6noz
|
|
return ip
|
|
}
|
|
|
|
// hasZone reports whether IP has an IPv6 zone.
|
|
func (ip Addr) hasZone() bool {
|
|
return ip.z != z0 && ip.z != z4 && ip.z != z6noz
|
|
}
|
|
|
|
// IsLinkLocalUnicast reports whether ip is a link-local unicast address.
|
|
func (ip Addr) IsLinkLocalUnicast() bool {
|
|
// Dynamic Configuration of IPv4 Link-Local Addresses
|
|
// https://datatracker.ietf.org/doc/html/rfc3927#section-2.1
|
|
if ip.Is4() {
|
|
return ip.v4(0) == 169 && ip.v4(1) == 254
|
|
}
|
|
// IP Version 6 Addressing Architecture (2.4 Address Type Identification)
|
|
// https://datatracker.ietf.org/doc/html/rfc4291#section-2.4
|
|
if ip.Is6() {
|
|
return ip.v6u16(0)&0xffc0 == 0xfe80
|
|
}
|
|
return false // zero value
|
|
}
|
|
|
|
// IsLoopback reports whether ip is a loopback address.
|
|
func (ip Addr) IsLoopback() bool {
|
|
// Requirements for Internet Hosts -- Communication Layers (3.2.1.3 Addressing)
|
|
// https://datatracker.ietf.org/doc/html/rfc1122#section-3.2.1.3
|
|
if ip.Is4() {
|
|
return ip.v4(0) == 127
|
|
}
|
|
// IP Version 6 Addressing Architecture (2.4 Address Type Identification)
|
|
// https://datatracker.ietf.org/doc/html/rfc4291#section-2.4
|
|
if ip.Is6() {
|
|
return ip.addr.hi == 0 && ip.addr.lo == 1
|
|
}
|
|
return false // zero value
|
|
}
|
|
|
|
// IsMulticast reports whether ip is a multicast address.
|
|
func (ip Addr) IsMulticast() bool {
|
|
// Host Extensions for IP Multicasting (4. HOST GROUP ADDRESSES)
|
|
// https://datatracker.ietf.org/doc/html/rfc1112#section-4
|
|
if ip.Is4() {
|
|
return ip.v4(0)&0xf0 == 0xe0
|
|
}
|
|
// IP Version 6 Addressing Architecture (2.4 Address Type Identification)
|
|
// https://datatracker.ietf.org/doc/html/rfc4291#section-2.4
|
|
if ip.Is6() {
|
|
return ip.addr.hi>>(64-8) == 0xff // ip.v6(0) == 0xff
|
|
}
|
|
return false // zero value
|
|
}
|
|
|
|
// IsInterfaceLocalMulticast reports whether ip is an IPv6 interface-local
|
|
// multicast address.
|
|
func (ip Addr) IsInterfaceLocalMulticast() bool {
|
|
// IPv6 Addressing Architecture (2.7.1. Pre-Defined Multicast Addresses)
|
|
// https://datatracker.ietf.org/doc/html/rfc4291#section-2.7.1
|
|
if ip.Is6() {
|
|
return ip.v6u16(0)&0xff0f == 0xff01
|
|
}
|
|
return false // zero value
|
|
}
|
|
|
|
// IsLinkLocalMulticast reports whether ip is a link-local multicast address.
|
|
func (ip Addr) IsLinkLocalMulticast() bool {
|
|
// IPv4 Multicast Guidelines (4. Local Network Control Block (224.0.0/24))
|
|
// https://datatracker.ietf.org/doc/html/rfc5771#section-4
|
|
if ip.Is4() {
|
|
return ip.v4(0) == 224 && ip.v4(1) == 0 && ip.v4(2) == 0
|
|
}
|
|
// IPv6 Addressing Architecture (2.7.1. Pre-Defined Multicast Addresses)
|
|
// https://datatracker.ietf.org/doc/html/rfc4291#section-2.7.1
|
|
if ip.Is6() {
|
|
return ip.v6u16(0)&0xff0f == 0xff02
|
|
}
|
|
return false // zero value
|
|
}
|
|
|
|
// IsGlobalUnicast reports whether ip is a global unicast address.
|
|
//
|
|
// It returns true for IPv6 addresses which fall outside of the current
|
|
// IANA-allocated 2000::/3 global unicast space, with the exception of the
|
|
// link-local address space. It also returns true even if ip is in the IPv4
|
|
// private address space or IPv6 unique local address space.
|
|
// It returns false for the zero Addr.
|
|
//
|
|
// For reference, see RFC 1122, RFC 4291, and RFC 4632.
|
|
func (ip Addr) IsGlobalUnicast() bool {
|
|
if ip.z == z0 {
|
|
// Invalid or zero-value.
|
|
return false
|
|
}
|
|
|
|
// Match package net's IsGlobalUnicast logic. Notably private IPv4 addresses
|
|
// and ULA IPv6 addresses are still considered "global unicast".
|
|
if ip.Is4() && (ip == IPv4Unspecified() || ip == AddrFrom4([4]byte{255, 255, 255, 255})) {
|
|
return false
|
|
}
|
|
|
|
return ip != IPv6Unspecified() &&
|
|
!ip.IsLoopback() &&
|
|
!ip.IsMulticast() &&
|
|
!ip.IsLinkLocalUnicast()
|
|
}
|
|
|
|
// IsPrivate reports whether ip is a private address, according to RFC 1918
|
|
// (IPv4 addresses) and RFC 4193 (IPv6 addresses). That is, it reports whether
|
|
// ip is in 10.0.0.0/8, 172.16.0.0/12, 192.168.0.0/16, or fc00::/7. This is the
|
|
// same as net.IP.IsPrivate.
|
|
func (ip Addr) IsPrivate() bool {
|
|
// Match the stdlib's IsPrivate logic.
|
|
if ip.Is4() {
|
|
// RFC 1918 allocates 10.0.0.0/8, 172.16.0.0/12, and 192.168.0.0/16 as
|
|
// private IPv4 address subnets.
|
|
return ip.v4(0) == 10 ||
|
|
(ip.v4(0) == 172 && ip.v4(1)&0xf0 == 16) ||
|
|
(ip.v4(0) == 192 && ip.v4(1) == 168)
|
|
}
|
|
|
|
if ip.Is6() {
|
|
// RFC 4193 allocates fc00::/7 as the unique local unicast IPv6 address
|
|
// subnet.
|
|
return ip.v6(0)&0xfe == 0xfc
|
|
}
|
|
|
|
return false // zero value
|
|
}
|
|
|
|
// IsUnspecified reports whether ip is an unspecified address, either the IPv4
|
|
// address "0.0.0.0" or the IPv6 address "::".
|
|
//
|
|
// Note that the zero Addr is not an unspecified address.
|
|
func (ip Addr) IsUnspecified() bool {
|
|
return ip == IPv4Unspecified() || ip == IPv6Unspecified()
|
|
}
|
|
|
|
// Prefix keeps only the top b bits of IP, producing a Prefix
|
|
// of the specified length.
|
|
// If ip is a zero Addr, Prefix always returns a zero Prefix and a nil error.
|
|
// Otherwise, if bits is less than zero or greater than ip.BitLen(),
|
|
// Prefix returns an error.
|
|
func (ip Addr) Prefix(b int) (Prefix, error) {
|
|
if b < 0 {
|
|
return Prefix{}, errors.New("negative Prefix bits")
|
|
}
|
|
effectiveBits := b
|
|
switch ip.z {
|
|
case z0:
|
|
return Prefix{}, nil
|
|
case z4:
|
|
if b > 32 {
|
|
return Prefix{}, errors.New("prefix length " + itoa.Itoa(b) + " too large for IPv4")
|
|
}
|
|
effectiveBits += 96
|
|
default:
|
|
if b > 128 {
|
|
return Prefix{}, errors.New("prefix length " + itoa.Itoa(b) + " too large for IPv6")
|
|
}
|
|
}
|
|
ip.addr = ip.addr.and(mask6(effectiveBits))
|
|
return PrefixFrom(ip, b), nil
|
|
}
|
|
|
|
const (
|
|
netIPv4len = 4
|
|
netIPv6len = 16
|
|
)
|
|
|
|
// As16 returns the IP address in its 16-byte representation.
|
|
// IPv4 addresses are returned in their v6-mapped form.
|
|
// IPv6 addresses with zones are returned without their zone (use the
|
|
// Zone method to get it).
|
|
// The ip zero value returns all zeroes.
|
|
func (ip Addr) As16() (a16 [16]byte) {
|
|
bePutUint64(a16[:8], ip.addr.hi)
|
|
bePutUint64(a16[8:], ip.addr.lo)
|
|
return a16
|
|
}
|
|
|
|
// As4 returns an IPv4 or IPv4-in-IPv6 address in its 4-byte representation.
|
|
// If ip is the zero Addr or an IPv6 address, As4 panics.
|
|
// Note that 0.0.0.0 is not the zero Addr.
|
|
func (ip Addr) As4() (a4 [4]byte) {
|
|
if ip.z == z4 || ip.Is4In6() {
|
|
bePutUint32(a4[:], uint32(ip.addr.lo))
|
|
return a4
|
|
}
|
|
if ip.z == z0 {
|
|
panic("As4 called on IP zero value")
|
|
}
|
|
panic("As4 called on IPv6 address")
|
|
}
|
|
|
|
// AsSlice returns an IPv4 or IPv6 address in its respective 4-byte or 16-byte representation.
|
|
func (ip Addr) AsSlice() []byte {
|
|
switch ip.z {
|
|
case z0:
|
|
return nil
|
|
case z4:
|
|
var ret [4]byte
|
|
bePutUint32(ret[:], uint32(ip.addr.lo))
|
|
return ret[:]
|
|
default:
|
|
var ret [16]byte
|
|
bePutUint64(ret[:8], ip.addr.hi)
|
|
bePutUint64(ret[8:], ip.addr.lo)
|
|
return ret[:]
|
|
}
|
|
}
|
|
|
|
// Next returns the address following ip.
|
|
// If there is none, it returns the zero Addr.
|
|
func (ip Addr) Next() Addr {
|
|
ip.addr = ip.addr.addOne()
|
|
if ip.Is4() {
|
|
if uint32(ip.addr.lo) == 0 {
|
|
// Overflowed.
|
|
return Addr{}
|
|
}
|
|
} else {
|
|
if ip.addr.isZero() {
|
|
// Overflowed
|
|
return Addr{}
|
|
}
|
|
}
|
|
return ip
|
|
}
|
|
|
|
// Prev returns the IP before ip.
|
|
// If there is none, it returns the IP zero value.
|
|
func (ip Addr) Prev() Addr {
|
|
if ip.Is4() {
|
|
if uint32(ip.addr.lo) == 0 {
|
|
return Addr{}
|
|
}
|
|
} else if ip.addr.isZero() {
|
|
return Addr{}
|
|
}
|
|
ip.addr = ip.addr.subOne()
|
|
return ip
|
|
}
|
|
|
|
// String returns the string form of the IP address ip.
|
|
// It returns one of 5 forms:
|
|
//
|
|
// - "invalid IP", if ip is the zero Addr
|
|
// - IPv4 dotted decimal ("192.0.2.1")
|
|
// - IPv6 ("2001:db8::1")
|
|
// - "::ffff:1.2.3.4" (if Is4In6)
|
|
// - IPv6 with zone ("fe80:db8::1%eth0")
|
|
//
|
|
// Note that unlike package net's IP.String method,
|
|
// IP4-mapped IPv6 addresses format with a "::ffff:"
|
|
// prefix before the dotted quad.
|
|
func (ip Addr) String() string {
|
|
switch ip.z {
|
|
case z0:
|
|
return "invalid IP"
|
|
case z4:
|
|
return ip.string4()
|
|
default:
|
|
if ip.Is4In6() {
|
|
// TODO(bradfitz): this could alloc less.
|
|
if z := ip.Zone(); z != "" {
|
|
return "::ffff:" + ip.Unmap().String() + "%" + z
|
|
} else {
|
|
return "::ffff:" + ip.Unmap().String()
|
|
}
|
|
}
|
|
return ip.string6()
|
|
}
|
|
}
|
|
|
|
// AppendTo appends a text encoding of ip,
|
|
// as generated by MarshalText,
|
|
// to b and returns the extended buffer.
|
|
func (ip Addr) AppendTo(b []byte) []byte {
|
|
switch ip.z {
|
|
case z0:
|
|
return b
|
|
case z4:
|
|
return ip.appendTo4(b)
|
|
default:
|
|
if ip.Is4In6() {
|
|
b = append(b, "::ffff:"...)
|
|
b = ip.Unmap().appendTo4(b)
|
|
if z := ip.Zone(); z != "" {
|
|
b = append(b, '%')
|
|
b = append(b, z...)
|
|
}
|
|
return b
|
|
}
|
|
return ip.appendTo6(b)
|
|
}
|
|
}
|
|
|
|
// digits is a string of the hex digits from 0 to f. It's used in
|
|
// appendDecimal and appendHex to format IP addresses.
|
|
const digits = "0123456789abcdef"
|
|
|
|
// appendDecimal appends the decimal string representation of x to b.
|
|
func appendDecimal(b []byte, x uint8) []byte {
|
|
// Using this function rather than strconv.AppendUint makes IPv4
|
|
// string building 2x faster.
|
|
|
|
if x >= 100 {
|
|
b = append(b, digits[x/100])
|
|
}
|
|
if x >= 10 {
|
|
b = append(b, digits[x/10%10])
|
|
}
|
|
return append(b, digits[x%10])
|
|
}
|
|
|
|
// appendHex appends the hex string representation of x to b.
|
|
func appendHex(b []byte, x uint16) []byte {
|
|
// Using this function rather than strconv.AppendUint makes IPv6
|
|
// string building 2x faster.
|
|
|
|
if x >= 0x1000 {
|
|
b = append(b, digits[x>>12])
|
|
}
|
|
if x >= 0x100 {
|
|
b = append(b, digits[x>>8&0xf])
|
|
}
|
|
if x >= 0x10 {
|
|
b = append(b, digits[x>>4&0xf])
|
|
}
|
|
return append(b, digits[x&0xf])
|
|
}
|
|
|
|
// appendHexPad appends the fully padded hex string representation of x to b.
|
|
func appendHexPad(b []byte, x uint16) []byte {
|
|
return append(b, digits[x>>12], digits[x>>8&0xf], digits[x>>4&0xf], digits[x&0xf])
|
|
}
|
|
|
|
func (ip Addr) string4() string {
|
|
const max = len("255.255.255.255")
|
|
ret := make([]byte, 0, max)
|
|
ret = ip.appendTo4(ret)
|
|
return string(ret)
|
|
}
|
|
|
|
func (ip Addr) appendTo4(ret []byte) []byte {
|
|
ret = appendDecimal(ret, ip.v4(0))
|
|
ret = append(ret, '.')
|
|
ret = appendDecimal(ret, ip.v4(1))
|
|
ret = append(ret, '.')
|
|
ret = appendDecimal(ret, ip.v4(2))
|
|
ret = append(ret, '.')
|
|
ret = appendDecimal(ret, ip.v4(3))
|
|
return ret
|
|
}
|
|
|
|
// string6 formats ip in IPv6 textual representation. It follows the
|
|
// guidelines in section 4 of RFC 5952
|
|
// (https://tools.ietf.org/html/rfc5952#section-4): no unnecessary
|
|
// zeros, use :: to elide the longest run of zeros, and don't use ::
|
|
// to compact a single zero field.
|
|
func (ip Addr) string6() string {
|
|
// Use a zone with a "plausibly long" name, so that most zone-ful
|
|
// IP addresses won't require additional allocation.
|
|
//
|
|
// The compiler does a cool optimization here, where ret ends up
|
|
// stack-allocated and so the only allocation this function does
|
|
// is to construct the returned string. As such, it's okay to be a
|
|
// bit greedy here, size-wise.
|
|
const max = len("ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff%enp5s0")
|
|
ret := make([]byte, 0, max)
|
|
ret = ip.appendTo6(ret)
|
|
return string(ret)
|
|
}
|
|
|
|
func (ip Addr) appendTo6(ret []byte) []byte {
|
|
zeroStart, zeroEnd := uint8(255), uint8(255)
|
|
for i := uint8(0); i < 8; i++ {
|
|
j := i
|
|
for j < 8 && ip.v6u16(j) == 0 {
|
|
j++
|
|
}
|
|
if l := j - i; l >= 2 && l > zeroEnd-zeroStart {
|
|
zeroStart, zeroEnd = i, j
|
|
}
|
|
}
|
|
|
|
for i := uint8(0); i < 8; i++ {
|
|
if i == zeroStart {
|
|
ret = append(ret, ':', ':')
|
|
i = zeroEnd
|
|
if i >= 8 {
|
|
break
|
|
}
|
|
} else if i > 0 {
|
|
ret = append(ret, ':')
|
|
}
|
|
|
|
ret = appendHex(ret, ip.v6u16(i))
|
|
}
|
|
|
|
if ip.z != z6noz {
|
|
ret = append(ret, '%')
|
|
ret = append(ret, ip.Zone()...)
|
|
}
|
|
return ret
|
|
}
|
|
|
|
// StringExpanded is like String but IPv6 addresses are expanded with leading
|
|
// zeroes and no "::" compression. For example, "2001:db8::1" becomes
|
|
// "2001:0db8:0000:0000:0000:0000:0000:0001".
|
|
func (ip Addr) StringExpanded() string {
|
|
switch ip.z {
|
|
case z0, z4:
|
|
return ip.String()
|
|
}
|
|
|
|
const size = len("ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff")
|
|
ret := make([]byte, 0, size)
|
|
for i := uint8(0); i < 8; i++ {
|
|
if i > 0 {
|
|
ret = append(ret, ':')
|
|
}
|
|
|
|
ret = appendHexPad(ret, ip.v6u16(i))
|
|
}
|
|
|
|
if ip.z != z6noz {
|
|
// The addition of a zone will cause a second allocation, but when there
|
|
// is no zone the ret slice will be stack allocated.
|
|
ret = append(ret, '%')
|
|
ret = append(ret, ip.Zone()...)
|
|
}
|
|
return string(ret)
|
|
}
|
|
|
|
// MarshalText implements the encoding.TextMarshaler interface,
|
|
// The encoding is the same as returned by String, with one exception:
|
|
// If ip is the zero Addr, the encoding is the empty string.
|
|
func (ip Addr) MarshalText() ([]byte, error) {
|
|
switch ip.z {
|
|
case z0:
|
|
return []byte(""), nil
|
|
case z4:
|
|
max := len("255.255.255.255")
|
|
b := make([]byte, 0, max)
|
|
return ip.appendTo4(b), nil
|
|
default:
|
|
max := len("ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff%enp5s0")
|
|
b := make([]byte, 0, max)
|
|
if ip.Is4In6() {
|
|
b = append(b, "::ffff:"...)
|
|
b = ip.Unmap().appendTo4(b)
|
|
if z := ip.Zone(); z != "" {
|
|
b = append(b, '%')
|
|
b = append(b, z...)
|
|
}
|
|
return b, nil
|
|
}
|
|
return ip.appendTo6(b), nil
|
|
}
|
|
|
|
}
|
|
|
|
// UnmarshalText implements the encoding.TextUnmarshaler interface.
|
|
// The IP address is expected in a form accepted by ParseAddr.
|
|
//
|
|
// If text is empty, UnmarshalText sets *ip to the zero Addr and
|
|
// returns no error.
|
|
func (ip *Addr) UnmarshalText(text []byte) error {
|
|
if len(text) == 0 {
|
|
*ip = Addr{}
|
|
return nil
|
|
}
|
|
var err error
|
|
*ip, err = ParseAddr(string(text))
|
|
return err
|
|
}
|
|
|
|
func (ip Addr) marshalBinaryWithTrailingBytes(trailingBytes int) []byte {
|
|
var b []byte
|
|
switch ip.z {
|
|
case z0:
|
|
b = make([]byte, trailingBytes)
|
|
case z4:
|
|
b = make([]byte, 4+trailingBytes)
|
|
bePutUint32(b, uint32(ip.addr.lo))
|
|
default:
|
|
z := ip.Zone()
|
|
b = make([]byte, 16+len(z)+trailingBytes)
|
|
bePutUint64(b[:8], ip.addr.hi)
|
|
bePutUint64(b[8:], ip.addr.lo)
|
|
copy(b[16:], z)
|
|
}
|
|
return b
|
|
}
|
|
|
|
// MarshalBinary implements the encoding.BinaryMarshaler interface.
|
|
// It returns a zero-length slice for the zero Addr,
|
|
// the 4-byte form for an IPv4 address,
|
|
// and the 16-byte form with zone appended for an IPv6 address.
|
|
func (ip Addr) MarshalBinary() ([]byte, error) {
|
|
return ip.marshalBinaryWithTrailingBytes(0), nil
|
|
}
|
|
|
|
// UnmarshalBinary implements the encoding.BinaryUnmarshaler interface.
|
|
// It expects data in the form generated by MarshalBinary.
|
|
func (ip *Addr) UnmarshalBinary(b []byte) error {
|
|
n := len(b)
|
|
switch {
|
|
case n == 0:
|
|
*ip = Addr{}
|
|
return nil
|
|
case n == 4:
|
|
*ip = AddrFrom4(*(*[4]byte)(b))
|
|
return nil
|
|
case n == 16:
|
|
*ip = ipv6Slice(b)
|
|
return nil
|
|
case n > 16:
|
|
*ip = ipv6Slice(b[:16]).WithZone(string(b[16:]))
|
|
return nil
|
|
}
|
|
return errors.New("unexpected slice size")
|
|
}
|
|
|
|
// AddrPort is an IP and a port number.
|
|
type AddrPort struct {
|
|
ip Addr
|
|
port uint16
|
|
}
|
|
|
|
// AddrPortFrom returns an AddrPort with the provided IP and port.
|
|
// It does not allocate.
|
|
func AddrPortFrom(ip Addr, port uint16) AddrPort { return AddrPort{ip: ip, port: port} }
|
|
|
|
// Addr returns p's IP address.
|
|
func (p AddrPort) Addr() Addr { return p.ip }
|
|
|
|
// Port returns p's port.
|
|
func (p AddrPort) Port() uint16 { return p.port }
|
|
|
|
// splitAddrPort splits s into an IP address string and a port
|
|
// string. It splits strings shaped like "foo:bar" or "[foo]:bar",
|
|
// without further validating the substrings. v6 indicates whether the
|
|
// ip string should parse as an IPv6 address or an IPv4 address, in
|
|
// order for s to be a valid ip:port string.
|
|
func splitAddrPort(s string) (ip, port string, v6 bool, err error) {
|
|
i := stringsLastIndexByte(s, ':')
|
|
if i == -1 {
|
|
return "", "", false, errors.New("not an ip:port")
|
|
}
|
|
|
|
ip, port = s[:i], s[i+1:]
|
|
if len(ip) == 0 {
|
|
return "", "", false, errors.New("no IP")
|
|
}
|
|
if len(port) == 0 {
|
|
return "", "", false, errors.New("no port")
|
|
}
|
|
if ip[0] == '[' {
|
|
if len(ip) < 2 || ip[len(ip)-1] != ']' {
|
|
return "", "", false, errors.New("missing ]")
|
|
}
|
|
ip = ip[1 : len(ip)-1]
|
|
v6 = true
|
|
}
|
|
|
|
return ip, port, v6, nil
|
|
}
|
|
|
|
// ParseAddrPort parses s as an AddrPort.
|
|
//
|
|
// It doesn't do any name resolution: both the address and the port
|
|
// must be numeric.
|
|
func ParseAddrPort(s string) (AddrPort, error) {
|
|
var ipp AddrPort
|
|
ip, port, v6, err := splitAddrPort(s)
|
|
if err != nil {
|
|
return ipp, err
|
|
}
|
|
port16, err := strconv.ParseUint(port, 10, 16)
|
|
if err != nil {
|
|
return ipp, errors.New("invalid port " + strconv.Quote(port) + " parsing " + strconv.Quote(s))
|
|
}
|
|
ipp.port = uint16(port16)
|
|
ipp.ip, err = ParseAddr(ip)
|
|
if err != nil {
|
|
return AddrPort{}, err
|
|
}
|
|
if v6 && ipp.ip.Is4() {
|
|
return AddrPort{}, errors.New("invalid ip:port " + strconv.Quote(s) + ", square brackets can only be used with IPv6 addresses")
|
|
} else if !v6 && ipp.ip.Is6() {
|
|
return AddrPort{}, errors.New("invalid ip:port " + strconv.Quote(s) + ", IPv6 addresses must be surrounded by square brackets")
|
|
}
|
|
return ipp, nil
|
|
}
|
|
|
|
// MustParseAddrPort calls ParseAddrPort(s) and panics on error.
|
|
// It is intended for use in tests with hard-coded strings.
|
|
func MustParseAddrPort(s string) AddrPort {
|
|
ip, err := ParseAddrPort(s)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return ip
|
|
}
|
|
|
|
// isZero reports whether p is the zero AddrPort.
|
|
func (p AddrPort) isZero() bool { return p == AddrPort{} }
|
|
|
|
// IsValid reports whether p.IP() is valid.
|
|
// All ports are valid, including zero.
|
|
func (p AddrPort) IsValid() bool { return p.ip.IsValid() }
|
|
|
|
func (p AddrPort) String() string {
|
|
switch p.ip.z {
|
|
case z0:
|
|
return "invalid AddrPort"
|
|
case z4:
|
|
a := p.ip.As4()
|
|
buf := make([]byte, 0, 21)
|
|
for i := range a {
|
|
buf = strconv.AppendUint(buf, uint64(a[i]), 10)
|
|
buf = append(buf, "...:"[i])
|
|
}
|
|
buf = strconv.AppendUint(buf, uint64(p.port), 10)
|
|
return string(buf)
|
|
default:
|
|
// TODO: this could be more efficient allocation-wise:
|
|
return joinHostPort(p.ip.String(), itoa.Itoa(int(p.port)))
|
|
}
|
|
}
|
|
|
|
func joinHostPort(host, port string) string {
|
|
// We assume that host is a literal IPv6 address if host has
|
|
// colons.
|
|
if bytealg.IndexByteString(host, ':') >= 0 {
|
|
return "[" + host + "]:" + port
|
|
}
|
|
return host + ":" + port
|
|
}
|
|
|
|
// AppendTo appends a text encoding of p,
|
|
// as generated by MarshalText,
|
|
// to b and returns the extended buffer.
|
|
func (p AddrPort) AppendTo(b []byte) []byte {
|
|
switch p.ip.z {
|
|
case z0:
|
|
return b
|
|
case z4:
|
|
b = p.ip.appendTo4(b)
|
|
default:
|
|
if p.ip.Is4In6() {
|
|
b = append(b, "[::ffff:"...)
|
|
b = p.ip.Unmap().appendTo4(b)
|
|
if z := p.ip.Zone(); z != "" {
|
|
b = append(b, '%')
|
|
b = append(b, z...)
|
|
}
|
|
} else {
|
|
b = append(b, '[')
|
|
b = p.ip.appendTo6(b)
|
|
}
|
|
b = append(b, ']')
|
|
}
|
|
b = append(b, ':')
|
|
b = strconv.AppendInt(b, int64(p.port), 10)
|
|
return b
|
|
}
|
|
|
|
// MarshalText implements the encoding.TextMarshaler interface. The
|
|
// encoding is the same as returned by String, with one exception: if
|
|
// p.Addr() is the zero Addr, the encoding is the empty string.
|
|
func (p AddrPort) MarshalText() ([]byte, error) {
|
|
var max int
|
|
switch p.ip.z {
|
|
case z0:
|
|
case z4:
|
|
max = len("255.255.255.255:65535")
|
|
default:
|
|
max = len("[ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff%enp5s0]:65535")
|
|
}
|
|
b := make([]byte, 0, max)
|
|
b = p.AppendTo(b)
|
|
return b, nil
|
|
}
|
|
|
|
// UnmarshalText implements the encoding.TextUnmarshaler
|
|
// interface. The AddrPort is expected in a form
|
|
// generated by MarshalText or accepted by ParseAddrPort.
|
|
func (p *AddrPort) UnmarshalText(text []byte) error {
|
|
if len(text) == 0 {
|
|
*p = AddrPort{}
|
|
return nil
|
|
}
|
|
var err error
|
|
*p, err = ParseAddrPort(string(text))
|
|
return err
|
|
}
|
|
|
|
// MarshalBinary implements the encoding.BinaryMarshaler interface.
|
|
// It returns Addr.MarshalBinary with an additional two bytes appended
|
|
// containing the port in little-endian.
|
|
func (p AddrPort) MarshalBinary() ([]byte, error) {
|
|
b := p.Addr().marshalBinaryWithTrailingBytes(2)
|
|
lePutUint16(b[len(b)-2:], p.Port())
|
|
return b, nil
|
|
}
|
|
|
|
// UnmarshalBinary implements the encoding.BinaryUnmarshaler interface.
|
|
// It expects data in the form generated by MarshalBinary.
|
|
func (p *AddrPort) UnmarshalBinary(b []byte) error {
|
|
if len(b) < 2 {
|
|
return errors.New("unexpected slice size")
|
|
}
|
|
var addr Addr
|
|
err := addr.UnmarshalBinary(b[:len(b)-2])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*p = AddrPortFrom(addr, leUint16(b[len(b)-2:]))
|
|
return nil
|
|
}
|
|
|
|
// Prefix is an IP address prefix (CIDR) representing an IP network.
|
|
//
|
|
// The first Bits() of Addr() are specified. The remaining bits match any address.
|
|
// The range of Bits() is [0,32] for IPv4 or [0,128] for IPv6.
|
|
type Prefix struct {
|
|
ip Addr
|
|
|
|
// bits is logically a uint8 (storing [0,128]) but also
|
|
// encodes an "invalid" bit, currently represented by the
|
|
// invalidPrefixBits sentinel value. It could be packed into
|
|
// the uint8 more with more complicated expressions in the
|
|
// accessors, but the extra byte (in padding anyway) doesn't
|
|
// hurt and simplifies code below.
|
|
bits int16
|
|
}
|
|
|
|
// invalidPrefixBits is the Prefix.bits value used when PrefixFrom is
|
|
// outside the range of a uint8. It's returned as the int -1 in the
|
|
// public API.
|
|
const invalidPrefixBits = -1
|
|
|
|
// PrefixFrom returns a Prefix with the provided IP address and bit
|
|
// prefix length.
|
|
//
|
|
// It does not allocate. Unlike Addr.Prefix, PrefixFrom does not mask
|
|
// off the host bits of ip.
|
|
//
|
|
// If bits is less than zero or greater than ip.BitLen, Prefix.Bits
|
|
// will return an invalid value -1.
|
|
func PrefixFrom(ip Addr, bits int) Prefix {
|
|
if bits < 0 || bits > ip.BitLen() {
|
|
bits = invalidPrefixBits
|
|
}
|
|
b16 := int16(bits)
|
|
return Prefix{
|
|
ip: ip.withoutZone(),
|
|
bits: b16,
|
|
}
|
|
}
|
|
|
|
// Addr returns p's IP address.
|
|
func (p Prefix) Addr() Addr { return p.ip }
|
|
|
|
// Bits returns p's prefix length.
|
|
//
|
|
// It reports -1 if invalid.
|
|
func (p Prefix) Bits() int { return int(p.bits) }
|
|
|
|
// IsValid reports whether p.Bits() has a valid range for p.IP().
|
|
// If p.Addr() is the zero Addr, IsValid returns false.
|
|
// Note that if p is the zero Prefix, then p.IsValid() == false.
|
|
func (p Prefix) IsValid() bool { return !p.ip.isZero() && p.bits >= 0 && int(p.bits) <= p.ip.BitLen() }
|
|
|
|
func (p Prefix) isZero() bool { return p == Prefix{} }
|
|
|
|
// IsSingleIP reports whether p contains exactly one IP.
|
|
func (p Prefix) IsSingleIP() bool { return p.bits != 0 && int(p.bits) == p.ip.BitLen() }
|
|
|
|
// ParsePrefix parses s as an IP address prefix.
|
|
// The string can be in the form "192.168.1.0/24" or "2001:db8::/32",
|
|
// the CIDR notation defined in RFC 4632 and RFC 4291.
|
|
//
|
|
// Note that masked address bits are not zeroed. Use Masked for that.
|
|
func ParsePrefix(s string) (Prefix, error) {
|
|
i := stringsLastIndexByte(s, '/')
|
|
if i < 0 {
|
|
return Prefix{}, errors.New("netip.ParsePrefix(" + strconv.Quote(s) + "): no '/'")
|
|
}
|
|
ip, err := ParseAddr(s[:i])
|
|
if err != nil {
|
|
return Prefix{}, errors.New("netip.ParsePrefix(" + strconv.Quote(s) + "): " + err.Error())
|
|
}
|
|
bitsStr := s[i+1:]
|
|
bits, err := strconv.Atoi(bitsStr)
|
|
if err != nil {
|
|
return Prefix{}, errors.New("netip.ParsePrefix(" + strconv.Quote(s) + ": bad bits after slash: " + strconv.Quote(bitsStr))
|
|
}
|
|
maxBits := 32
|
|
if ip.Is6() {
|
|
maxBits = 128
|
|
}
|
|
if bits < 0 || bits > maxBits {
|
|
return Prefix{}, errors.New("netip.ParsePrefix(" + strconv.Quote(s) + ": prefix length out of range")
|
|
}
|
|
return PrefixFrom(ip, bits), nil
|
|
}
|
|
|
|
// MustParsePrefix calls ParsePrefix(s) and panics on error.
|
|
// It is intended for use in tests with hard-coded strings.
|
|
func MustParsePrefix(s string) Prefix {
|
|
ip, err := ParsePrefix(s)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return ip
|
|
}
|
|
|
|
// Masked returns p in its canonical form, with all but the high
|
|
// p.Bits() bits of p.Addr() masked off.
|
|
//
|
|
// If p is zero or otherwise invalid, Masked returns the zero Prefix.
|
|
func (p Prefix) Masked() Prefix {
|
|
if m, err := p.ip.Prefix(int(p.bits)); err == nil {
|
|
return m
|
|
}
|
|
return Prefix{}
|
|
}
|
|
|
|
// Contains reports whether the network p includes ip.
|
|
//
|
|
// An IPv4 address will not match an IPv6 prefix.
|
|
// A v6-mapped IPv6 address will not match an IPv4 prefix.
|
|
// A zero-value IP will not match any prefix.
|
|
// If ip has an IPv6 zone, Contains returns false,
|
|
// because Prefixes strip zones.
|
|
func (p Prefix) Contains(ip Addr) bool {
|
|
if !p.IsValid() || ip.hasZone() {
|
|
return false
|
|
}
|
|
if f1, f2 := p.ip.BitLen(), ip.BitLen(); f1 == 0 || f2 == 0 || f1 != f2 {
|
|
return false
|
|
}
|
|
if ip.Is4() {
|
|
// xor the IP addresses together; mismatched bits are now ones.
|
|
// Shift away the number of bits we don't care about.
|
|
// Shifts in Go are more efficient if the compiler can prove
|
|
// that the shift amount is smaller than the width of the shifted type (64 here).
|
|
// We know that p.bits is in the range 0..32 because p is Valid;
|
|
// the compiler doesn't know that, so mask with 63 to help it.
|
|
// Now truncate to 32 bits, because this is IPv4.
|
|
// If all the bits we care about are equal, the result will be zero.
|
|
return uint32((ip.addr.lo^p.ip.addr.lo)>>((32-p.bits)&63)) == 0
|
|
} else {
|
|
// xor the IP addresses together.
|
|
// Mask away the bits we don't care about.
|
|
// If all the bits we care about are equal, the result will be zero.
|
|
return ip.addr.xor(p.ip.addr).and(mask6(int(p.bits))).isZero()
|
|
}
|
|
}
|
|
|
|
// Overlaps reports whether p and o contain any IP addresses in common.
|
|
//
|
|
// If p and o are of different address families or either have a zero
|
|
// IP, it reports false. Like the Contains method, a prefix with a
|
|
// v6-mapped IPv4 IP is still treated as an IPv6 mask.
|
|
func (p Prefix) Overlaps(o Prefix) bool {
|
|
if !p.IsValid() || !o.IsValid() {
|
|
return false
|
|
}
|
|
if p == o {
|
|
return true
|
|
}
|
|
if p.ip.Is4() != o.ip.Is4() {
|
|
return false
|
|
}
|
|
var minBits int16
|
|
if p.bits < o.bits {
|
|
minBits = p.bits
|
|
} else {
|
|
minBits = o.bits
|
|
}
|
|
if minBits == 0 {
|
|
return true
|
|
}
|
|
// One of these Prefix calls might look redundant, but we don't require
|
|
// that p and o values are normalized (via Prefix.Masked) first,
|
|
// so the Prefix call on the one that's already minBits serves to zero
|
|
// out any remaining bits in IP.
|
|
var err error
|
|
if p, err = p.ip.Prefix(int(minBits)); err != nil {
|
|
return false
|
|
}
|
|
if o, err = o.ip.Prefix(int(minBits)); err != nil {
|
|
return false
|
|
}
|
|
return p.ip == o.ip
|
|
}
|
|
|
|
// AppendTo appends a text encoding of p,
|
|
// as generated by MarshalText,
|
|
// to b and returns the extended buffer.
|
|
func (p Prefix) AppendTo(b []byte) []byte {
|
|
if p.isZero() {
|
|
return b
|
|
}
|
|
if !p.IsValid() {
|
|
return append(b, "invalid Prefix"...)
|
|
}
|
|
|
|
// p.ip is non-nil, because p is valid.
|
|
if p.ip.z == z4 {
|
|
b = p.ip.appendTo4(b)
|
|
} else {
|
|
if p.ip.Is4In6() {
|
|
b = append(b, "::ffff:"...)
|
|
b = p.ip.Unmap().appendTo4(b)
|
|
} else {
|
|
b = p.ip.appendTo6(b)
|
|
}
|
|
}
|
|
|
|
b = append(b, '/')
|
|
b = appendDecimal(b, uint8(p.bits))
|
|
return b
|
|
}
|
|
|
|
// MarshalText implements the encoding.TextMarshaler interface,
|
|
// The encoding is the same as returned by String, with one exception:
|
|
// If p is the zero value, the encoding is the empty string.
|
|
func (p Prefix) MarshalText() ([]byte, error) {
|
|
var max int
|
|
switch p.ip.z {
|
|
case z0:
|
|
case z4:
|
|
max = len("255.255.255.255/32")
|
|
default:
|
|
max = len("ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff%enp5s0/128")
|
|
}
|
|
b := make([]byte, 0, max)
|
|
b = p.AppendTo(b)
|
|
return b, nil
|
|
}
|
|
|
|
// UnmarshalText implements the encoding.TextUnmarshaler interface.
|
|
// The IP address is expected in a form accepted by ParsePrefix
|
|
// or generated by MarshalText.
|
|
func (p *Prefix) UnmarshalText(text []byte) error {
|
|
if len(text) == 0 {
|
|
*p = Prefix{}
|
|
return nil
|
|
}
|
|
var err error
|
|
*p, err = ParsePrefix(string(text))
|
|
return err
|
|
}
|
|
|
|
// MarshalBinary implements the encoding.BinaryMarshaler interface.
|
|
// It returns Addr.MarshalBinary with an additional byte appended
|
|
// containing the prefix bits.
|
|
func (p Prefix) MarshalBinary() ([]byte, error) {
|
|
b := p.Addr().withoutZone().marshalBinaryWithTrailingBytes(1)
|
|
b[len(b)-1] = uint8(p.Bits())
|
|
return b, nil
|
|
}
|
|
|
|
// UnmarshalBinary implements the encoding.BinaryUnmarshaler interface.
|
|
// It expects data in the form generated by MarshalBinary.
|
|
func (p *Prefix) UnmarshalBinary(b []byte) error {
|
|
if len(b) < 1 {
|
|
return errors.New("unexpected slice size")
|
|
}
|
|
var addr Addr
|
|
err := addr.UnmarshalBinary(b[:len(b)-1])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*p = PrefixFrom(addr, int(b[len(b)-1]))
|
|
return nil
|
|
}
|
|
|
|
// String returns the CIDR notation of p: "<ip>/<bits>".
|
|
func (p Prefix) String() string {
|
|
if !p.IsValid() {
|
|
return "invalid Prefix"
|
|
}
|
|
return p.ip.String() + "/" + itoa.Itoa(int(p.bits))
|
|
}
|