2021-11-10 12:25:10 +08:00
|
|
|
|
package task
|
|
|
|
|
|
2021-11-10 18:47:21 +08:00
|
|
|
|
import (
|
|
|
|
|
"bufio"
|
|
|
|
|
"log"
|
|
|
|
|
"math/rand"
|
|
|
|
|
"net"
|
|
|
|
|
"os"
|
|
|
|
|
"strconv"
|
|
|
|
|
"strings"
|
|
|
|
|
"time"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
const defaultInputFile = "ip.txt"
|
|
|
|
|
|
2021-11-10 12:25:10 +08:00
|
|
|
|
var (
|
2021-11-10 17:12:12 +08:00
|
|
|
|
// TestAll test all ip
|
|
|
|
|
TestAll = false
|
|
|
|
|
// IPFile is the filename of IP Rangs
|
2021-11-10 18:47:21 +08:00
|
|
|
|
IPFile = defaultInputFile
|
2023-01-31 20:11:13 +08:00
|
|
|
|
IPText string
|
2021-11-10 12:25:10 +08:00
|
|
|
|
)
|
2021-11-10 18:47:21 +08:00
|
|
|
|
|
2021-11-17 15:40:53 +08:00
|
|
|
|
func InitRandSeed() {
|
2021-11-10 18:47:21 +08:00
|
|
|
|
rand.Seed(time.Now().UnixNano())
|
2021-11-17 15:40:53 +08:00
|
|
|
|
}
|
|
|
|
|
|
2022-11-08 15:47:19 +08:00
|
|
|
|
func isIPv4(ip string) bool {
|
|
|
|
|
return strings.Contains(ip, ".")
|
|
|
|
|
}
|
|
|
|
|
|
2021-11-17 15:40:53 +08:00
|
|
|
|
func randIPEndWith(num byte) byte {
|
2022-02-19 17:36:13 +08:00
|
|
|
|
if num == 0 { // 对于 /32 这种单独的 IP
|
|
|
|
|
return byte(0)
|
|
|
|
|
}
|
2021-11-12 00:14:10 +08:00
|
|
|
|
return byte(rand.Intn(int(num)))
|
2021-11-10 18:47:21 +08:00
|
|
|
|
}
|
|
|
|
|
|
2021-11-12 00:14:10 +08:00
|
|
|
|
type IPRanges struct {
|
|
|
|
|
ips []*net.IPAddr
|
|
|
|
|
mask string
|
|
|
|
|
firstIP net.IP
|
|
|
|
|
ipNet *net.IPNet
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func newIPRanges() *IPRanges {
|
|
|
|
|
return &IPRanges{
|
|
|
|
|
ips: make([]*net.IPAddr, 0),
|
2021-11-10 18:47:21 +08:00
|
|
|
|
}
|
2021-11-12 00:14:10 +08:00
|
|
|
|
}
|
|
|
|
|
|
2023-04-28 14:52:59 +08:00
|
|
|
|
// 如果是单独 IP 则加上子网掩码,反之则获取子网掩码(r.mask)
|
2021-11-12 00:14:10 +08:00
|
|
|
|
func (r *IPRanges) fixIP(ip string) string {
|
|
|
|
|
// 如果不含有 '/' 则代表不是 IP 段,而是一个单独的 IP,因此需要加上 /32 /128 子网掩码
|
|
|
|
|
if i := strings.IndexByte(ip, '/'); i < 0 {
|
2022-11-08 15:47:19 +08:00
|
|
|
|
if isIPv4(ip) {
|
|
|
|
|
r.mask = "/32"
|
|
|
|
|
} else {
|
2021-11-12 00:14:10 +08:00
|
|
|
|
r.mask = "/128"
|
2021-11-10 18:47:21 +08:00
|
|
|
|
}
|
2021-11-12 00:14:10 +08:00
|
|
|
|
ip += r.mask
|
|
|
|
|
} else {
|
|
|
|
|
r.mask = ip[i:]
|
2021-11-10 23:58:40 +08:00
|
|
|
|
}
|
2021-11-12 00:14:10 +08:00
|
|
|
|
return ip
|
2021-11-10 23:58:40 +08:00
|
|
|
|
}
|
|
|
|
|
|
2023-04-29 23:32:29 +08:00
|
|
|
|
// 解析 IP 段,获得 IP、IP 范围、子网掩码
|
2021-11-12 00:14:10 +08:00
|
|
|
|
func (r *IPRanges) parseCIDR(ip string) {
|
|
|
|
|
var err error
|
|
|
|
|
if r.firstIP, r.ipNet, err = net.ParseCIDR(r.fixIP(ip)); err != nil {
|
|
|
|
|
log.Fatalln("ParseCIDR err", err)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (r *IPRanges) appendIPv4(d byte) {
|
|
|
|
|
r.appendIP(net.IPv4(r.firstIP[12], r.firstIP[13], r.firstIP[14], d))
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (r *IPRanges) appendIP(ip net.IP) {
|
|
|
|
|
r.ips = append(r.ips, &net.IPAddr{IP: ip})
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 返回第四段 ip 的最小值及可用数目
|
|
|
|
|
func (r *IPRanges) getIPRange() (minIP, hosts byte) {
|
|
|
|
|
minIP = r.firstIP[15] & r.ipNet.Mask[3] // IP 第四段最小值
|
|
|
|
|
|
|
|
|
|
// 根据子网掩码获取主机数量
|
|
|
|
|
m := net.IPv4Mask(255, 255, 255, 255)
|
|
|
|
|
for i, v := range r.ipNet.Mask {
|
|
|
|
|
m[i] ^= v
|
|
|
|
|
}
|
|
|
|
|
total, _ := strconv.ParseInt(m.String(), 16, 32) // 总可用 IP 数
|
|
|
|
|
if total > 255 { // 矫正 第四段 可用 IP 数
|
|
|
|
|
hosts = 255
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
hosts = byte(total)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (r *IPRanges) chooseIPv4() {
|
2023-04-29 23:32:29 +08:00
|
|
|
|
if r.mask == "/32" { // 单个 IP 则无需随机,直接加入自身即可
|
|
|
|
|
r.appendIP(r.firstIP)
|
|
|
|
|
} else {
|
|
|
|
|
minIP, hosts := r.getIPRange() // 返回第四段 IP 的最小值及可用数目
|
|
|
|
|
for r.ipNet.Contains(r.firstIP) { // 只要该 IP 没有超出 IP 网段范围,就继续循环随机
|
|
|
|
|
if TestAll { // 如果是测速全部 IP
|
|
|
|
|
for i := 0; i <= int(hosts); i++ { // 遍历 IP 最后一段最小值到最大值
|
|
|
|
|
r.appendIPv4(byte(i) + minIP)
|
|
|
|
|
}
|
|
|
|
|
} else { // 随机 IP 的最后一段 0.0.0.X
|
|
|
|
|
r.appendIPv4(minIP + randIPEndWith(hosts))
|
2021-11-10 18:47:21 +08:00
|
|
|
|
}
|
2023-04-29 23:32:29 +08:00
|
|
|
|
r.firstIP[14]++ // 0.0.(X+1).X
|
|
|
|
|
if r.firstIP[14] == 0 {
|
|
|
|
|
r.firstIP[13]++ // 0.(X+1).X.X
|
|
|
|
|
if r.firstIP[13] == 0 {
|
|
|
|
|
r.firstIP[12]++ // (X+1).X.X.X
|
|
|
|
|
}
|
2021-11-10 18:47:21 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2021-11-12 00:14:10 +08:00
|
|
|
|
func (r *IPRanges) chooseIPv6() {
|
2023-04-29 23:32:29 +08:00
|
|
|
|
if r.mask == "/128" { // 单个 IP 则无需随机,直接加入自身即可
|
|
|
|
|
r.appendIP(r.firstIP)
|
|
|
|
|
} else {
|
|
|
|
|
var tempIP uint8 // 临时变量,用于记录前一位的值
|
|
|
|
|
for r.ipNet.Contains(r.firstIP) { // 只要该 IP 没有超出 IP 网段范围,就继续循环随机
|
2021-11-12 00:14:10 +08:00
|
|
|
|
r.firstIP[15] = randIPEndWith(255) // 随机 IP 的最后一段
|
|
|
|
|
r.firstIP[14] = randIPEndWith(255) // 随机 IP 的最后一段
|
2023-04-29 23:32:29 +08:00
|
|
|
|
|
|
|
|
|
targetIP := make([]byte, len(r.firstIP))
|
|
|
|
|
copy(targetIP, r.firstIP)
|
|
|
|
|
r.appendIP(targetIP) // 加入 IP 地址池
|
|
|
|
|
|
|
|
|
|
for i := 13; i >= 0; i-- { // 从倒数第三位开始往前随机
|
|
|
|
|
tempIP = r.firstIP[i] // 保存前一位的值
|
|
|
|
|
r.firstIP[i] += randIPEndWith(255) // 随机 0~255,加到当前位上
|
|
|
|
|
if r.firstIP[i] >= tempIP { // 如果当前位的值大于等于前一位的值,说明随机成功了,可以退出该循环
|
|
|
|
|
break
|
|
|
|
|
}
|
2021-11-12 17:58:13 +08:00
|
|
|
|
}
|
2021-11-10 18:47:21 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2021-11-12 00:14:10 +08:00
|
|
|
|
func loadIPRanges() []*net.IPAddr {
|
|
|
|
|
ranges := newIPRanges()
|
2023-01-31 20:11:13 +08:00
|
|
|
|
if IPText != "" { // 从参数中获取 IP 段数据
|
2023-04-28 14:52:59 +08:00
|
|
|
|
IPs := strings.Split(IPText, ",") // 以逗号分隔为数组并循环遍历
|
2023-01-31 20:11:13 +08:00
|
|
|
|
for _, IP := range IPs {
|
2023-04-28 14:52:59 +08:00
|
|
|
|
IP = strings.TrimSpace(IP) // 去除首尾的空白字符(空格、制表符、换行符等)
|
|
|
|
|
if IP == "" { // 跳过空的(即开头、结尾或连续多个 ,, 的情况)
|
|
|
|
|
continue
|
|
|
|
|
}
|
2023-04-29 23:32:29 +08:00
|
|
|
|
ranges.parseCIDR(IP) // 解析 IP 段,获得 IP、IP 范围、子网掩码
|
|
|
|
|
if isIPv4(IP) { // 生成要测速的所有 IPv4 / IPv6 地址(单个/随机/全部)
|
2023-01-31 20:11:13 +08:00
|
|
|
|
ranges.chooseIPv4()
|
|
|
|
|
} else {
|
|
|
|
|
ranges.chooseIPv6()
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
} else { // 从文件中获取 IP 段数据
|
|
|
|
|
if IPFile == "" {
|
|
|
|
|
IPFile = defaultInputFile
|
|
|
|
|
}
|
|
|
|
|
file, err := os.Open(IPFile)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
defer file.Close()
|
|
|
|
|
scanner := bufio.NewScanner(file)
|
2023-04-28 14:52:59 +08:00
|
|
|
|
for scanner.Scan() { // 循环遍历文件每一行
|
|
|
|
|
line := strings.TrimSpace(scanner.Text()) // 去除首尾的空白字符(空格、制表符、换行符等)
|
|
|
|
|
if line == "" { // 跳过空行
|
|
|
|
|
continue
|
|
|
|
|
}
|
2023-04-29 23:32:29 +08:00
|
|
|
|
ranges.parseCIDR(line) // 解析 IP 段,获得 IP、IP 范围、子网掩码
|
|
|
|
|
if isIPv4(line) { // 生成要测速的所有 IPv4 / IPv6 地址(单个/随机/全部)
|
2023-01-31 20:11:13 +08:00
|
|
|
|
ranges.chooseIPv4()
|
|
|
|
|
} else {
|
|
|
|
|
ranges.chooseIPv6()
|
|
|
|
|
}
|
2021-11-12 00:14:10 +08:00
|
|
|
|
}
|
2021-11-10 23:58:40 +08:00
|
|
|
|
}
|
2021-11-12 00:14:10 +08:00
|
|
|
|
return ranges.ips
|
2021-11-10 18:47:21 +08:00
|
|
|
|
}
|