函数Hosts接受一个字符串参数scope和一个整数参数size,返回一个字符串切片hosts和一个错误值。该函数生成多个IP地址,并通过去重和排序来确保这些地址都是唯一的且按升序排列。
函数Simulate接受一个上下文对象ctx和一个字符串参数dst,返回一个错误值。该函数首先计算每个端口应有的超时时间callTimeout,然后循环遍历所有扫描端口,并在每个端口上启动tcp模拟扫描。在循环内部,使用context.WithTimeout方法创建具有超时时间的新上下文对象ctx,并在操作完成或超时时调用cancelFn取消该操作。
伪代码如下:
// Hosts方法 hosts := 空的字符串切片 dedup := 新建空map[string]bool numOfNets = size/20 + 1 如果numOfNets大于scanIPRanges长度,则将其设置为scanIPRanges长度 netIdx = 随机从scanIPRanges中选取numOfNets个索引 k = 0 for len(hosts) < size && k < 2*size { ip = 在scanIPRanges[netIdx[len(hosts)%len(netIdx)]]范围内随机生成一个IP地址 如果ip已经出现过了(即存在于dedup中),则跳过本次循环 将ip添加到hosts末尾,并将其标记为已出现过(即添加到dedup中) k++ } 按升序对hosts排序 返回hosts和nil
// Simulate方法 callTimeout = 200ms / len(scanPorts) 如果ctx有截止时间,则将callTimeout设置为剩余时间除以len(scanPorts) for _, port := range scanPorts { ctx, cancelFn = 新建一个超时上下文对象ctx,超时时间为callTimeout defer 调用cancelFn函数释放资源 启动tcp扫描操作,并传入dst和port参数 } 返回nil或最后一个操作的错误值