在go语言中实现在IP/CIDR中IP判断及IP/CIDR与ip和netmask互相转换示例

Go  /  管理员 发布于 2年前   754

在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 内的方法。