在go语言中实现在IP/CIDR中IP判断及IP/CIDR与ip和netmask互相转换示例
在go语言中需要借助net包来实现网卡项目中IP/CIDR转换功能,比如
IP/CIDR 与 ip 和 netmask 的相互转换,IP/CIDR 转换为 IP 段形式以及转换回来,
判断某 IP 是否在一个 IP/CIDR内等等。
示例代码
IP/CIDR 与 IP netmask 的转换
IP/CIDR 转换为 ip 与 netmask
type IpNetmask struct {
IP string
Netmask string
}
func CIDRToIPNetmase(cidr string) (*IpNetmask, error) {
_, ipNet, err := net.ParseCIDR(cidr)
if err != nil {
return nil, fmt.Errorf("parse CIDR failed: err= %v", err)
}
ones, _ := ipNet.Mask.Size()
mask := net.CIDRMask(ones, 32)
return &IpNetmask{
IP: ipNet.IP.String(),
Netmask: net.IP(mask).String(),
}, nil
}
上文中为了得到 perfix 或者说 子网掩码我们做了多次操作,
但是在 go 1.18 之后,net/netip 包提出了一个新的方法可以快速得到
netip.ParsePrefix(t)
该方法会在返回一个 address 与 int 类型的 perfix
netmask 反向转换
func netmaskToSub(mask string) (int, error) {
masks := strings.Split(mask, ".")
if len(masks) != 4 {
return 0, fmt.Errorf("invalid mask, mask: %s", mask)
}
var subs []int
for _, i := range masks {
one, err := strconv.Atoi(i)
if err != nil {
return 0, fmt.Errorf("invalid mask, mask: %s", mask)
}
subs = append(subs, one)
}
ones, _ := net.IPv4Mask(byte(subs[0]), byte(subs[1]), byte(subs[2]), byte(subs[3])).Size()
return ones, nil
}
netmask, bits := net.IPv4Mask(255, 255, 255, 248).Size()
CIDR 转换为地址段
func CIDRtoIPRange(address string) (string, error) {
ip, ipnet, err := net.ParseCIDR(address)
if err != nil {
return "", fmt.Errorf("parse cidr failedL err= %v", err)
}
var ips []string
for ip := ip.Mask(ipnet.Mask); ipnet.Contains(ip); inc(ip) {
ips = append(ips, ip.String())
}
return fmt.Sprintf("%s-%s", ips[0], ips[len(ips)-1]), nil
}
func inc(ip net.IP) {
for j := len(ip) - 1; j > 0; j-- {
ip[j]++
if ip[j] > 0 {
break
}
}
}
判断某 IP 是否在 CIDR 内
这个没什么做的,前文通过 net.ParseCIDR 方法得出的
ipNet有一个方法就是 contains判断输入ip是否在其内,
在的话返回true
ipnet.Contains()
总结
本文介绍了 ipCIDR 形式转换为 ip ,网关,ip 段以及判断一个 ip 是否在一个 ipcidr 内的方法。
相关文章