netbird/client/iface/wgaddr/address.go

package wgaddr

import (
	"net/netip"
)

// Address WireGuard parsed address
type Address struct {
	IP      netip.Addr
	Network netip.Prefix

	// IPv6 overlay address, if assigned.
	IPv6    netip.Addr
	IPv6Net netip.Prefix
}

// ParseWGAddress parse a string ("1.2.3.4/24") address to WG Address
func ParseWGAddress(address string) (Address, error) {
	prefix, err := netip.ParsePrefix(address)
	if err != nil {
		return Address{}, err
	}
	return Address{
		IP:      prefix.Addr().Unmap(),
		Network: prefix.Masked(),
	}, nil
}

// HasIPv6 reports whether a v6 overlay address is assigned.
func (addr Address) HasIPv6() bool {
	return addr.IPv6.IsValid()
}

func (addr Address) String() string {
	return addr.Prefix().String()
}

// IPv6String returns the v6 address in CIDR notation, or empty string if none.
func (addr Address) IPv6String() string {
	if !addr.HasIPv6() {
		return ""
	}
	return addr.IPv6Prefix().String()
}

// Prefix returns the v4 host address with its network prefix length (e.g. 100.64.0.1/16).
func (addr Address) Prefix() netip.Prefix {
	return netip.PrefixFrom(addr.IP, addr.Network.Bits())
}

// IPv6Prefix returns the v6 host address with its network prefix length, or a zero prefix if none.
func (addr Address) IPv6Prefix() netip.Prefix {
	if !addr.HasIPv6() {
		return netip.Prefix{}
	}
	return netip.PrefixFrom(addr.IPv6, addr.IPv6Net.Bits())
}

// ClearIPv6 removes the IPv6 overlay address, leaving only v4.
//
//nolint:recvcheck // ClearIPv6 is the only mutating method on this otherwise value-type struct.
func (addr *Address) ClearIPv6() {
	addr.IPv6 = netip.Addr{}
	addr.IPv6Net = netip.Prefix{}
}