Function IterateCIDR

utils.go:255–312

IterateCIDR 迭代CIDR网段中的所有IP地址

(cidr string)

// IterateCIDR 迭代CIDR网段中的所有IP地址
func IterateCIDR(cidr string) <-chan Host {
	hostChan := make(chan Host, 100)
	
	go func() {
		defer close(hostChan)
		
		// 解析CIDR
		_, ipNet, err := net.ParseCIDR(cidr)
		if err != nil {
			printError(fmt.Sprintf("解析CIDR失败: %s - %v", cidr, err))
			return
		}
		
		count := 0
		maxHosts := 65536 // 限制最大主机数，防止内存溢出
		
		// 获取网络地址和掩码
		ip := make(net.IP, len(ipNet.IP))
		copy(ip, ipNet.IP)
		
		// 计算网络中的主机数
		ones, bits := ipNet.Mask.Size()
		if bits-ones > 16 { // 如果主机位超过16位，限制扫描范围
			printError(fmt.Sprintf("CIDR %s 包含的主机数过多，已限制为前%d个", cidr, maxHosts))
		}
		
		// 遍历网络中的所有IP
		for {
			if !ipNet.Contains(ip) {
				break
			}
			
			if count >= maxHosts {
				printError(fmt.Sprintf("CIDR %s 包含的主机数超过限制(%d)，已截断", cidr, maxHosts))
				break
			}
			
			// 创建新的Host并发送到channel
			newHost := Host{
				IP:     make(net.IP, len(ip)),
				Origin: cidr,
				Type:   HostTypeIP,
			}
			copy(newHost.IP, ip)
			hostChan <- newHost
			
			// 递增IP地址
			ip = NextIP(ip, true)
			count++
		}
		
		if config.Verbose {
			printInfo(fmt.Sprintf("CIDR %s 展开为 %d 个IP地址", cidr, count))
		}
	}()
	
	return hostChan
}