go/src/netstack/util/parse.go

// Copyright 2018 The Fuchsia 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 util

import (
	"net"

	"github.com/google/netstack/tcpip"
)

func ApplyMask(addr tcpip.Address, mask tcpip.AddressMask) tcpip.Address {
	return tcpip.Address(net.IP(addr).Mask(net.IPMask(mask)))
}

func CIDRMask(ones, bits int) tcpip.AddressMask {
	return tcpip.AddressMask(net.CIDRMask(ones, bits))
}

func ipToAddress(ip net.IP) tcpip.Address {
	if v4 := ip.To4(); v4 != nil {
		return tcpip.Address(v4)
	}
	return tcpip.Address(ip)
}

func ParseCIDR(s string) (tcpip.Address, tcpip.Subnet, error) {
	ip, subnet, err := net.ParseCIDR(s)
	if err != nil {
		return "", tcpip.Subnet{}, err
	}
	sn, err := tcpip.NewSubnet(tcpip.Address(subnet.IP), tcpip.AddressMask(subnet.Mask))
	if err != nil {
		return "", tcpip.Subnet{}, err
	}
	return ipToAddress(ip), sn, nil
}

// Parse parses the string representation of an IPv4 or IPv6 address.
func Parse(src string) tcpip.Address {
	return ipToAddress(net.ParseIP(src))
}