优化 指定测速条件时的进度条显示内容

Merge pull request #13 from zhangsean/master

IPRangeLoader.go

@@ -1,39 +1,167 @@
-package main
+package main
-
+
-import (
+import (
-	"bufio"
+	"bufio"
-	"log"
+	"log"
-	"net"
+	"net"
-	"os"
+	"os"
-)
+	"strconv"
-
+	"strings"
-func loadFirstIPOfRangeFromFile() []net.IPAddr {
+)
-	file, err := os.Open("ip.txt")
+
-	if err != nil {
+// 根据子网掩码获取主机数量
-		log.Fatal(err)
+func getCidrHostNum(maskLen int) int {
-	}
+	cidrIPNum := int(0)
-	firstIPs := make([]net.IPAddr, 0)
+	if maskLen < 32 {
-	scanner := bufio.NewScanner(file)
+		var i int = int(32 - maskLen - 1)
-	scanner.Split(bufio.ScanLines)
+		for ; i >= 1; i-- {
-	for scanner.Scan() {
+			cidrIPNum += 1 << i
-		IPString := scanner.Text()
+		}
-		firstIP, IPRange, err := net.ParseCIDR(IPString)
+		cidrIPNum += 2
-		if err != nil {
+	} else {
-			log.Fatal(err)
+		cidrIPNum = 1
-		}
+	}
-		firstIP[15] = ipEndWith
+	if cidrIPNum > 255 {
-		for IPRange.Contains(firstIP) {
+		cidrIPNum = 255
-			firstIPCopy := make([]byte, len(firstIP))
+	}
-			copy(firstIPCopy, firstIP)
+	return cidrIPNum
-			firstIPs = append(firstIPs, net.IPAddr{IP: firstIPCopy})
+}
-			firstIP[14]++
+
-			if firstIP[14] == 0 {
+// 获取 IP 最后一段最小值和最大值
-				firstIP[13]++
+func getCidrIPRange(cidr string) (uint8, uint8) {
-				if firstIP[13] == 0 {
+	ip := strings.Split(cidr, "/")[0]
-					firstIP[12]++
+	ipSegs := strings.Split(ip, ".")
-				}
+	maskLen, _ := strconv.Atoi(strings.Split(cidr, "/")[1])
-			}
+	seg4MinIP, seg4MaxIP := getIPSeg4Range(ipSegs, maskLen)
-		}
+	//ipPrefix := ipSegs[0] + "." + ipSegs[1] + "." + ipSegs[2] + "."
-	}
+
-	return firstIPs
+	return seg4MinIP,
-}
+		seg4MaxIP
+}
+
+// 获取 IP 最后一段的区间
+func getIPSeg4Range(ipSegs []string, maskLen int) (uint8, uint8) {
+	ipSeg, _ := strconv.Atoi(ipSegs[3])
+	return getIPSegRange(uint8(ipSeg), uint8(32-maskLen))
+}
+
+// 根据输入的基础IP地址和CIDR掩码计算一个IP片段的区间
+func getIPSegRange(userSegIP, offset uint8) (uint8, uint8) {
+	var ipSegMax uint8 = 255
+	netSegIP := ipSegMax << offset
+	segMinIP := netSegIP & userSegIP
+	segMaxIP := userSegIP&(255<<offset) | ^(255 << offset)
+	return uint8(segMinIP), uint8(segMaxIP)
+}
+
+func loadFirstIPOfRangeFromFile(ipFile string) []net.IPAddr {
+	file, err := os.Open(ipFile)
+	if err != nil {
+		log.Fatal(err)
+	}
+	firstIPs := make([]net.IPAddr, 0)
+	scanner := bufio.NewScanner(file)
+	scanner.Split(bufio.ScanLines)
+	for scanner.Scan() {
+		IPString := scanner.Text()
+		firstIP, IPRange, err := net.ParseCIDR(IPString)
+		//fmt.Println(firstIP)
+		//fmt.Println(IPRange)
+		if err != nil {
+			log.Fatal(err)
+		}
+		if !ipv6Mode { // IPv4
+			minIP, maxIP := getCidrIPRange(scanner.Text())                 // 获取 IP 最后一段最小值和最大值
+			Mask, _ := strconv.Atoi(strings.Split(scanner.Text(), "/")[1]) // 获取子网掩码
+			MaxIPNum := getCidrHostNum(Mask)                               // 根据子网掩码获取主机数量
+			for IPRange.Contains(firstIP) {
+				if allip { // 如果是测速全部 IP
+					for i := int(minIP); i <= int(maxIP); i++ { // 遍历 IP 最后一段最小值到最大值
+						firstIP[15] = uint8(i)
+						firstIPCopy := make([]byte, len(firstIP))
+						copy(firstIPCopy, firstIP)
+						firstIPs = append(firstIPs, net.IPAddr{IP: firstIPCopy})
+					}
+				} else { // 随机 IP 的最后一段 0.0.0.X
+					firstIP[15] = minIP + randipEndWith(MaxIPNum)
+					firstIPCopy := make([]byte, len(firstIP))
+					copy(firstIPCopy, firstIP)
+					firstIPs = append(firstIPs, net.IPAddr{IP: firstIPCopy})
+				}
+				firstIP[14]++ // 0.0.(X+1).X
+				if firstIP[14] == 0 {
+					firstIP[13]++ // 0.(X+1).X.X
+					if firstIP[13] == 0 {
+						firstIP[12]++ // (X+1).X.X.X
+					}
+				}
+			}
+		} else { //IPv6
+			var tempIP uint8
+			for IPRange.Contains(firstIP) {
+				//fmt.Println(firstIP)
+				//fmt.Println(firstIP[0], firstIP[1], firstIP[2], firstIP[3], firstIP[4], firstIP[5], firstIP[6], firstIP[7], firstIP[8], firstIP[9], firstIP[10], firstIP[11], firstIP[12], firstIP[13], firstIP[14], firstIP[15])
+				firstIP[15] = randipEndWith(255) // 随机 IP 的最后一段
+				firstIP[14] = randipEndWith(255) // 随机 IP 的最后一段
+				firstIPCopy := make([]byte, len(firstIP))
+				copy(firstIPCopy, firstIP)
+				firstIPs = append(firstIPs, net.IPAddr{IP: firstIPCopy})
+				tempIP = firstIP[13]
+				firstIP[13] += randipEndWith(255)
+				if firstIP[13] < tempIP {
+					tempIP = firstIP[12]
+					firstIP[12] += randipEndWith(255)
+					if firstIP[12] < tempIP {
+						tempIP = firstIP[11]
+						firstIP[11] += randipEndWith(255)
+						if firstIP[11] < tempIP {
+							tempIP = firstIP[10]
+							firstIP[10] += randipEndWith(255)
+							if firstIP[10] < tempIP {
+								tempIP = firstIP[9]
+								firstIP[9] += randipEndWith(255)
+								if firstIP[9] < tempIP {
+									tempIP = firstIP[8]
+									firstIP[8] += randipEndWith(255)
+									if firstIP[8] < tempIP {
+										tempIP = firstIP[7]
+										firstIP[7] += randipEndWith(255)
+										if firstIP[7] < tempIP {
+											tempIP = firstIP[6]
+											firstIP[6] += randipEndWith(255)
+											if firstIP[6] < tempIP {
+												tempIP = firstIP[5]
+												firstIP[5] += randipEndWith(255)
+												if firstIP[5] < tempIP {
+													tempIP = firstIP[4]
+													firstIP[4] += randipEndWith(255)
+													if firstIP[4] < tempIP {
+														tempIP = firstIP[3]
+														firstIP[3] += randipEndWith(255)
+														if firstIP[3] < tempIP {
+															tempIP = firstIP[2]
+															firstIP[2] += randipEndWith(255)
+															if firstIP[2] < tempIP {
+																tempIP = firstIP[1]
+																firstIP[1] += randipEndWith(255)
+																if firstIP[1] < tempIP {
+																	tempIP = firstIP[0]
+																	firstIP[0] += randipEndWith(255)
+																}
+															}
+														}
+													}
+												}
+											}
+										}
+									}
+								}
+							}
+						}
+					}
+				}
+			}
+		}
+	}
+	return firstIPs
+}

README.md

@@ -1,76 +1,180 @@
-# CloudflareSpeedTest
+# XIU2/CloudflareSpeedTest
-
+
-国外很多网站都在使用 Cloudflare CDN，但分配给中国访客的 IP 并不友好。  
+[![Go Version](https://img.shields.io/github/go-mod/go-version/XIU2/CloudflareSpeedTest.svg?style=flat-square&label=Go&color=00ADD8)](https://github.com/XIU2/CloudflareSpeedTest/blob/master/go.mod)
-虽然 Cloudflare 公开了所有 [IP 段](https://www.cloudflare.com/ips/) ，但想要在这么多 IP 中找到适合自己的，怕是要累死，所以就有了这个软件。  
+[![Release Version](https://img.shields.io/github/v/release/XIU2/CloudflareSpeedTest.svg?style=flat-square&label=Release&color=1784ff)](https://github.com/XIU2/CloudflareSpeedTest/releases/latest)
-
+[![GitHub license](https://img.shields.io/github/license/XIU2/CloudflareSpeedTest.svg?style=flat-square&label=License&color=f38020)](https://github.com/XIU2/CloudflareSpeedTest/blob/master/LICENSE)
-该软件可以**测试 Cloudflare CDN 所有 IP 的延迟和速度，获得最快 IP**！  
+[![GitHub Star](https://img.shields.io/github/stars/XIU2/CloudflareSpeedTest.svg?style=flat-square&label=Star&color=f38020)](https://github.com/XIU2/CloudflareSpeedTest/stargazers)
-你可以将 IP 添加到 `Hosts` 文件中，以提高访问使用 Cloudflare CDN 服务的国外网站速度！  
+[![GitHub Fork](https://img.shields.io/github/forks/XIU2/CloudflareSpeedTest.svg?style=flat-square&label=Fork&color=f38020)](https://github.com/XIU2/CloudflareSpeedTest/network/members)
-
+
-****
+国外很多网站都在使用 Cloudflare CDN，但分配给中国访客的 IP 并不友好。  
-### 快速使用
+虽然 Cloudflare 公开了所有 [IP 段](https://www.cloudflare.com/ips/) ，但想要在这么多 IP 中找到适合自己的，怕是要累死，所以就有了这个软件。  
-
+
-1. 下载编译好的可执行文件 [蓝奏云](https://www.lanzoux.com/b0742hkxe) / [Github](https://github.com/XIU2/CloudflareSpeedTest/releases) 并解压。  
+该软件可以**测试 Cloudflare CDN 延迟和速度，获取最快 IP (IPv4+IPv6)**！觉得好用请**点个⭐鼓励一下下~**  
-2. 双击运行 `CloudflareST.exe`文件（Windows系统），等待测速...  
+
-
+> _本项目也**适用于其他 CDN**，但是需要自行寻找 **CDN IP 段及下载测速地址**（否则只能延迟测速）。_
-测速完毕后，会把结果保存在当前目录下的 `result.csv` 文件中（只输出丢包率 50% 以下的），用记事本打开，排序为**延迟由低到高**，每一列用逗号分隔，分别是：  
+
-```
+> _我另一个开源项目： **[一个 \[油猴脚本\] 轻松解决「Github」文件下载速度慢的问题！](https://github.com/XIU2/UserScript)**_   
-IP 地址, Ping 发送次数, Ping 接收次数, Ping 接收率, 平均延迟, 下载速度 (MB/s)
+
-104.27.70.18, 4, 4, 1.0000, 150.7948, 12.8951
+****
-```
+## 快速使用
-选择一个平均延迟与下载速度都不错的 IP 放到 `Hosts` 文件中（指向域名）。  
+
-
+### 下载运行
-****
+
-### 进阶使用
+1. 下载编译好的可执行文件 [蓝奏云](https://xiu.lanzoux.com/b0742hkxe) / [Github](https://github.com/XIU2/CloudflareSpeedTest/releases) 并解压。  
-
+2. 双击运行 `CloudflareST.exe`文件（Windows），等待测速完成...  
-直接双击运行使用的是默认参数，如果想要测试速度更快、测试结果更全面，可以自定义参数。  
+
-``` cmd
+>  _**提示：Linux 系统**请先赋予权限 `chmod +x CloudflareST` ，然后再 `./CloudflareST` 运行。_   
-C:\>CloudflareST.exe -h
+
-
+> _**注意：建议测速时避开晚上高峰期（20:00~24:00）**，否则测速结果会与其他时间**差距很大...**_  
-CloudflareSpeedTest
+
-测试 Cloudflare CDN 所有 IP 的延迟和速度，获取最快 IP！
+### 结果示例
-https://github.com/XIU2/CloudflareSpeedTest
+
-
+测速完毕后，默认会显示**最快的 20 个 IP**，示例（我的白天测速结果）：  
-参数：
+
-    -n 500
+```
-        测速线程数量；请勿超过1000 (默认 500)
+IP 地址           已发送  已接收  丢包率  平均延迟  下载速度 (MB/s)
-    -t 4
+104.27.200.69     4       4       0.00    146.23    28.64
-        延迟测速次数；单个 IP 测速次数，TCP协议 (默认 4)
+172.67.60.78      4       4       0.00    139.82    15.02
-    -dn 20
+104.25.140.153    4       4       0.00    146.49    14.90
-        下载测速数量；延迟测速后，从最低延迟起测试下载速度的数量，请勿太多 (默认 20)
+104.27.192.65     4       4       0.00    140.28    14.07
-    -dt 10
+172.67.62.214     4       4       0.00    139.29    12.71
-        下载测速时间；单个 IP 测速最长时间，单位：秒 (默认 10)
+104.27.207.5      4       4       0.00    145.92    11.95
-    -v
+172.67.54.193     4       4       0.00    146.71    11.55
-        打印程序版本
+104.22.66.8       4       4       0.00    147.42    11.11
-    -h
+104.27.197.63     4       4       0.00    131.29    10.26
-        打印帮助说明
+172.67.58.91      4       4       0.00    140.19    9.14
-
+...
-示例：
+```
-    Windows：CloudflareST.exe -n 800 -t 4 -dn 20 -dt 10
+
-    Linux：CloudflareST -n 800 -t 4 -dn 20 -dt 10
+> _软件是先**延迟测速并按从低到高排序**后，再**从最低延迟的 IP 开始下载测速**的，所以：_  
-```
+
-
+测速结果第一行就是**既下载速度最快、又平均延迟最低的最快 IP**！至于拿来干嘛？取决于你~  
-#### 使用示例
+
-
+完整结果保存在当前目录下的 `result.csv` 文件中，用**记事本/表格软件**打开，格式如下：  
-在 CMD 中运行，或者把启动参数添加到快捷方式中。  
+
-> **注意：** 不需要四个参数都加上，如果你认为某个参数默认就很合适，那就跳过。  
+```
-
+IP 地址, 已发送, 已接收, 丢包率, 平均延迟, 下载速度 (MB/s)
-``` cmd
+104.27.200.69, 4, 4, 0.00, 146.23, 28.64
-# CMD 示例
+```
-CloudflareST.exe -n 800 -t 4 -dn 20 -dt 10
+
-```
+> _大家可以按自己需求，对完整结果**进一步筛选处理**，或者去看一看进阶使用**指定过滤条件**！_
-
+
-``` cmd
+****
-# 快捷方式示例（右键快捷方式 - 目标）
+## 进阶使用
-## 如果有引号就放在引号外面，记得引号和 - 之间有空格。
+
-"D:\Program Files\CloudflareST\CloudflareST.exe" -n 800 -t 4 -dn 20 -dt 10
+直接运行使用的是默认参数，如果想要测速结果更全面、更符合自己的要求，可以自定义参数。  
-```
+
-
+``` cmd
-****
+C:\>CloudflareST.exe -h
-### 感谢项目
+
-* https://github.com/Spedoske/CloudflareScanner
+CloudflareSpeedTest vX.X.X
-
+测试 Cloudflare CDN 所有 IP 的延迟和速度，获取最快 IP (IPv4+IPv6)！
-意外发现了这个项目，看了之后发现正好解决了我的问题，但是我更喜欢用户命令行方式运行，这样会更方便、有更多使用姿势，于是我临时学了下 Golang 并 Fork 修改了一份命令行方式交互的版本，如果有什么问题可以告诉我，虽然我不一定会~
+https://github.com/XIU2/CloudflareSpeedTest
-
+
-****
+参数：
-### 许可证
+    -n 500
-The GPL-3.0 License.
+        测速线程数量；越多测速越快，性能弱的设备 (如路由器) 请适当调低；(默认 500 最多 1000)
+    -t 4
+        延迟测速次数；单个 IP 延迟测速次数，为 1 时将过滤丢包的IP，TCP协议；(默认 4)
+    -tp 443
+        延迟测速端口；延迟测速 TCP 协议的端口；(默认 443)
+    -dn 20
+        下载测速数量；延迟测速并排序后，从最低延迟起下载测速的数量；(默认 20)
+    -dt 10
+        下载测速时间；单个 IP 下载测速最长时间，单位：秒；(默认 10)
+    -url https://cf.xiu2.xyz/Github/CloudflareSpeedTest.png
+        下载测速地址；用来下载测速的 Cloudflare CDN 文件地址，如地址含有空格请加上引号；
+    -tl 200
+        平均延迟上限；只输出低于指定平均延迟的 IP，与下载速度下限搭配使用；(默认 9999 ms)
+    -sl 5
+        下载速度下限；只输出高于指定下载速度的 IP，凑够指定数量 [-dn] 才会停止测速；(默认 0 MB/s)
+    -p 20
+        显示结果数量；测速后直接显示指定数量的结果，为 0 时不显示结果直接退出；(默认 20)
+    -f ip.txt
+        IP段数据文件；如路径含有空格请加上引号；支持其他 CDN IP段；(默认 ip.txt)
+    -o result.csv
+        输出结果文件；如路径含有空格请加上引号；值为空格时不输出 [-o " "]；(默认 result.csv)
+    -dd
+        禁用下载测速；禁用后测速结果会按延迟排序 (默认按下载速度排序)；(默认 启用)
+    -ipv6
+        IPv6测速模式；确保 IP 段数据文件内只包含 IPv6 IP段，软件不支持同时测速 IPv4+IPv6；(默认 IPv4)
+    -allip
+        测速全部的IP；对 IP 段中的每个 IP (仅支持 IPv4) 进行测速；(默认 每个 IP 段随机测速一个 IP)
+    -v
+        打印程序版本+检查版本更新
+    -h
+        打印帮助说明
+```
+
+> _如果**下载速度都是 0.00**，那可能默认的**下载测速地址**用的人太多到上限了，**请去这个 [Issues](https://github.com/XIU2/CloudflareSpeedTest/issues/6) 获得解决方法！**_  
+
+### 使用示例
+
+在 CMD 中运行，或者把启动参数添加到快捷方式中。  
+
+``` bash
+# 命令行示例
+# 注意：各参数均有默认值，只有不使用默认值时，才需要手动指定参数的值（按需选择），参数不分前后顺序。  
+# 提示： Linux 系统只需要把下面命令中的 CloudflareST.exe 改为 ./CloudflareST 即可。  
+
+# 指定 IPv4 数据文件，不显示结果直接退出（-p 值为 0）
+CloudflareST.exe -p 0 -f ip.txt -dd
+
+# 指定 IPv6 数据文件( ipv6.txt )，不显示结果直接退出（-p 值为 0）
+CloudflareST.exe -p 0 -f ipv6.txt -dd -ipv6
+
+# ——————————————————————
+
+# 指定 IPv4 数据文件，不输出结果到文件，直接显示结果（-p 值为 10 条）
+CloudflareST.exe -p 10 -f ip.txt -o " " -dd
+
+# 指定 IPv4 数据文件 及 输出结果到文件（相对路径，即当前目录下，如含空格请加上引号）
+CloudflareST.exe -f ip.txt -o result.csv -dd
+
+# 指定 IPv4 数据文件 及 输出结果到文件（绝对路径，即 C:\abc\ 目录下，如含空格请加上引号）
+CloudflareST.exe -f C:\abc\ip.txt -o C:\abc\result.csv -dd
+
+# ——————————————————————
+
+# 指定下载测速地址（要求：可以直接下载、文件大小超过 200MB、用的是 Cloudflare CDN），如含空格请加上引号
+CloudflareST.exe -url https://cf.xiu2.xyz/Github/CloudflareSpeedTest.png
+
+# ——————————————————————
+
+# 指定测速条件（只有同时满足三个条件时才会停止测速）：
+
+# 平均延迟上限：9999 ms，下载速度下限：5 MB/s，数量：10 个
+# 即需要找到 10 个平均延迟低于 9999 ms 且 下载速度高于 5 MB/s 的 IP 才会停止测速。
+CloudflareST.exe -sl 5 -dn 10
+
+# 没有指定平均延迟上限时，如果一直凑不够满足条件的 IP 数量，会一直测速下去。  
+# 所以建议同时指定 下载速度下限 和 平均延迟上限，这样测试到指定延迟还没凑够数量，就会终止测速。
+
+# 平均延迟上限：200 ms，下载速度下限：5 MB/s，数量：10 个
+# 即需要找到 10 个平均延迟低于 200 ms 且 下载速度高于 5 MB/s 的 IP 才会停止测速。
+CloudflareST.exe -tl 200 -sl 5 -dn 10
+
+# 如果一个满足条件的 IP 都没找到，那么就会和不指定条件一样输出完整结果。
+# 所以建议先不指定条件测速一遍，看看平均延迟和下载速度大概在什么范围，避免指定条件过低/过高！
+```
+
+``` bash
+# Windows 快捷方式示例（右键快捷方式 - 目标）
+# 如果要不输出结果文件，那么请加上 -o " "，引号里的是空格（没有空格会导致该参数被省略）。
+
+D:\ABC\CloudflareST\CloudflareST.exe -n 500 -t 4 -dn 20 -dt 5 -o " "
+
+# 如果文件路径包含引号，则需要把启动参数放在引号外面，记得引号和 - 之间有空格。
+"D:\Program Files\CloudflareST\CloudflareST.exe" -n 500 -t 4 -dn 20 -dt 5 -o " "
+```
+
+****
+## 感谢项目
+* https://github.com/Spedoske/CloudflareScanner
+
+> _因为该项目已经很长时间没更新了，而我又产生了很多功能需求，所以我干脆临时学了下 Go 语言就上手了..._  
+> _本软件基于该项目制作，但**已添加大量功能及修复 BUG**，并根据大家的使用反馈去添加、优化功能（闲）..._
+
+****
+## 许可证
+The GPL-3.0 License.

go.mod

@@ -1,4 +1,4 @@
-module CloudflareIPScanner
+module CloudflareSpeedTest
 
 go 1.14
 

ip.txt

@@ -1,3 +1,7 @@
+1.1.1.0/24
+1.0.0.0/24
+1.1.1.1/32
+1.0.0.1/32
 173.245.48.0/20
 103.21.244.0/22
 103.22.200.0/22

ipv6.txt

@@ -0,0 +1,42 @@
+2606:4700:10::6814:0/112
+2606:4700:10::ac43:0/112
+2606:4700:3000::/48
+2606:4700:3001::/48
+2606:4700:3002::/48
+2606:4700:3003::/48
+2606:4700:3004::/48
+2606:4700:3005::/48
+2606:4700:3006::/48
+2606:4700:3007::/48
+2606:4700:3008::/48
+2606:4700:3009::/48
+2606:4700:3010::/48
+2606:4700:3011::/48
+2606:4700:3012::/48
+2606:4700:3013::/48
+2606:4700:3014::/48
+2606:4700:3015::/48
+2606:4700:3016::/48
+2606:4700:3017::/48
+2606:4700:3018::/48
+2606:4700:3019::/48
+2606:4700:3020::/48
+2606:4700:3021::/48
+2606:4700:3022::/48
+2606:4700:3023::/48
+2606:4700:3024::/48
+2606:4700:3025::/48
+2606:4700:3026::/48
+2606:4700:3027::/48
+2606:4700:3028::/48
+2606:4700:3029::/48
+2606:4700:3030::/48
+2606:4700:3031::/48
+2606:4700:3032::/48
+2606:4700:3033::/48
+2606:4700:3034::/48
+2606:4700:3035::/48
+2606:4700:3036::/48
+2606:4700:3037::/48
+2606:4700:3038::/48
+2606:4700:3039::/48

main.go

@@ -1,89 +1,271 @@
-package main
+package main
-
+
-import (
+import (
-	"flag"
+	"flag"
-	"fmt"
+	"fmt"
-	"os"
+	"io/ioutil"
-	"sort"
+	"net/http"
-	"sync"
+	"os"
-	"time"
+	"runtime"
-
+	"sort"
-	"github.com/cheggaaa/pb/v3"
+	"strconv"
-)
+	"sync"
-
+	"time"
-var version string
+
-
+	"github.com/cheggaaa/pb/v3"
-func init() {
+)
-	var downloadSecond int64
+
-	var printVersion bool
+var version, ipFile, outputFile, versionNew string
-	const help = `
+var disableDownload, ipv6Mode, allip bool
-CloudflareSpeedTest
+var tcpPort, printResultNum, timeLimit, speedLimit, downloadSecond int
-测试 Cloudflare CDN 所有 IP 的延迟和速度，获取最快 IP！
+
-https://github.com/XIU2/CloudflareSpeedTest
+func init() {
-
+	var printVersion bool
-参数：
+	var help = `
-    -n 500
+CloudflareSpeedTest ` + version + `
-        测速线程数量；请勿超过1000 (默认 500)
+测试 Cloudflare CDN 所有 IP 的延迟和速度，获取最快 IP (IPv4+IPv6)！
-    -t 4
+https://github.com/XIU2/CloudflareSpeedTest
-        延迟测速次数；单个 IP 测速次数，TCP协议 (默认 4)
+
-    -dn 20
+参数：
-        下载测速数量；延迟测速后，从最低延迟起测试下载速度的数量，请勿太多 (默认 20)
+    -n 500
-    -dt 10
+        测速线程数量；越多测速越快，性能弱的设备 (如路由器) 请适当调低；(默认 500 最多 1000)
-        下载测速时间；单个 IP 测速最长时间，单位：秒 (默认 10)
+    -t 4
-    -v
+        延迟测速次数；单个 IP 延迟测速次数，为 1 时将过滤丢包的IP，TCP协议；(默认 4)
-        打印程序版本
+    -tp 443
-    -h
+        延迟测速端口；延迟测速 TCP 协议的端口；(默认 443)
-        打印帮助说明
+    -dn 20
-
+        下载测速数量；延迟测速并排序后，从最低延迟起下载测速的数量；(默认 20)
-示例：
+    -dt 10
-    Windows：CloudflareST.exe -n 800 -t 4 -dn 20 -dt 10
+        下载测速时间；单个 IP 下载测速最长时间，单位：秒；(默认 10)
-    Linux：CloudflareST -n 800 -t 4 -dn 20 -dt 10
+    -url https://cf.xiu2.xyz/Github/CloudflareSpeedTest.png
-`
+        下载测速地址；用来下载测速的 Cloudflare CDN 文件地址，如地址含有空格请加上引号；
-
+    -tl 200
-	pingRoutine = *flag.Int("n", 500, "测速线程数量")
+        平均延迟上限；只输出低于指定平均延迟的 IP，与下载速度下限搭配使用；(默认 9999 ms)
-	pingTime = *flag.Int("t", 4, "延迟测速次数")
+    -sl 5
-	downloadTestCount = *flag.Int("dn", 20, "下载测速数量")
+        下载速度下限；只输出高于指定下载速度的 IP，凑够指定数量 [-dn] 才会停止测速；(默认 0 MB/s)
-	flag.Int64Var(&downloadSecond, "dt", 10, "下载测速时间")
+    -p 20
-	flag.BoolVar(&printVersion, "v", false, "打印程序版本")
+        显示结果数量；测速后直接显示指定数量的结果，为 0 时不显示结果直接退出；(默认 20)
-
+    -f ip.txt
-	downloadTestTime = time.Duration(downloadSecond) * time.Second
+        IP段数据文件；如路径含有空格请加上引号；支持其他 CDN IP段；(默认 ip.txt)
-
+    -o result.csv
-	flag.Usage = func() { fmt.Print(help) }
+        输出结果文件；如路径含有空格请加上引号；值为空格时不输出 [-o " "]；(默认 result.csv)
-	flag.Parse()
+    -dd
-	if printVersion {
+        禁用下载测速；禁用后测速结果会按延迟排序 (默认按下载速度排序)；(默认 启用)
-		println(version)
+    -ipv6
-		os.Exit(0)
+        IPv6测速模式；确保 IP 段数据文件内只包含 IPv6 IP段，软件不支持同时测速 IPv4+IPv6；(默认 IPv4)
-	}
+    -allip
-}
+        测速全部的IP；对 IP 段中的每个 IP (仅支持 IPv4) 进行测速；(默认 每个 IP 段随机测速一个 IP)
-
+    -v
-func main() {
+        打印程序版本+检查版本更新
-	initipEndWith()
+    -h
-	ips := loadFirstIPOfRangeFromFile()
+        打印帮助说明
-	pingCount := len(ips) * pingTime
+`
-	bar := pb.StartNew(pingCount)
+
-	var wg sync.WaitGroup
+	flag.IntVar(&pingRoutine, "n", 500, "测速线程数量")
-	var mu sync.Mutex
+	flag.IntVar(&pingTime, "t", 4, "延迟测速次数")
-	var data = make([]CloudflareIPData, 0)
+	flag.IntVar(&tcpPort, "tp", 443, "延迟测速端口")
-
+	flag.IntVar(&downloadTestCount, "dn", 20, "下载测速数量")
-	fmt.Println("开始延迟测速(TCP)：")
+	flag.IntVar(&downloadSecond, "dt", 10, "下载测速时间")
-
+	flag.StringVar(&url, "url", "https://cf.xiu2.xyz/Github/CloudflareSpeedTest.png", "下载测速地址")
-	control := make(chan bool, pingRoutine)
+	flag.IntVar(&timeLimit, "tl", 9999, "延迟时间上限")
-	for _, ip := range ips {
+	flag.IntVar(&speedLimit, "sl", 0, "下载速度下限")
-		wg.Add(1)
+	flag.IntVar(&printResultNum, "p", 20, "显示结果数量")
-		control <- false
+	flag.BoolVar(&disableDownload, "dd", false, "禁用下载测速")
-		handleProgress := handleProgressGenerator(bar)
+	flag.BoolVar(&ipv6Mode, "ipv6", false, "禁用下载测速")
-		go tcpingGoroutine(&wg, &mu, ip, pingTime, &data, control, handleProgress)
+	flag.BoolVar(&allip, "allip", false, "测速全部 IP")
-	}
+	flag.StringVar(&ipFile, "f", "ip.txt", "IP 数据文件")
-	wg.Wait()
+	flag.StringVar(&outputFile, "o", "result.csv", "输出结果文件")
-	bar.Finish()
+	flag.BoolVar(&printVersion, "v", false, "打印程序版本")
-	bar = pb.StartNew(downloadTestCount)
+
-	sort.Sort(CloudflareIPDataSet(data))
+	flag.Usage = func() { fmt.Print(help) }
-	fmt.Println("开始下载测速：")
+	flag.Parse()
-	for i := 0; i < downloadTestCount; i++ {
+	if printVersion {
-		_, speed := DownloadSpeedHandler(data[i].ip)
+		println(version)
-		data[i].downloadSpeed = speed
+		fmt.Println("检查版本更新中...")
-		bar.Add(1)
+		checkUpdate()
-	}
+		if versionNew != "" {
-	bar.Finish()
+			fmt.Println("发现新版本 [" + versionNew + "]！请前往 [https://github.com/XIU2/CloudflareSpeedTest] 更新！")
-	ExportCsv("./result.csv", data)
+		} else {
-}
+			fmt.Println("当前为最新版本 [" + version + "]！")
+		}
+		os.Exit(0)
+	}
+	if pingRoutine <= 0 {
+		pingRoutine = 500
+	}
+	if pingTime <= 0 {
+		pingTime = 4
+	}
+	if tcpPort < 1 || tcpPort > 65535 {
+		tcpPort = 443
+	}
+	if downloadTestCount <= 0 {
+		downloadTestCount = 20
+	}
+	if downloadSecond <= 0 {
+		downloadSecond = 10
+	}
+	if url == "" {
+		url = "https://cf.xiu2.xyz/Github/CloudflareSpeedTest.png"
+	}
+	if timeLimit <= 0 {
+		timeLimit = 9999
+	}
+	if speedLimit < 0 {
+		speedLimit = 0
+	}
+	if printResultNum < 0 {
+		printResultNum = 20
+	}
+	if ipFile == "" {
+		ipFile = "ip.txt"
+	}
+	if outputFile == " " {
+		outputFile = ""
+	}
+}
+
+func main() {
+	go checkUpdate()                          // 检查版本更新
+	initRandSeed()                            // 置随机数种子
+	failTime = pingTime                       // 设置接收次数
+	ips := loadFirstIPOfRangeFromFile(ipFile) // 读入IP
+	pingCount := len(ips) * pingTime          // 计算进度条总数（IP*测试次数）
+	bar := pb.Simple.Start(pingCount)         // 进度条总数
+	var wg sync.WaitGroup
+	var mu sync.Mutex
+	var data = make([]CloudflareIPData, 0)
+	var data_2 = make([]CloudflareIPData, 0)
+	downloadTestTime = time.Duration(downloadSecond) * time.Second
+
+	fmt.Println("# XIU2/CloudflareSpeedTest " + version + "\n")
+	if ipv6Mode {
+		fmt.Println("开始延迟测速（模式：TCP IPv6，端口：" + strconv.Itoa(tcpPort) + "）：")
+	} else {
+		fmt.Println("开始延迟测速（模式：TCP IPv4，端口：" + strconv.Itoa(tcpPort) + "）：")
+	}
+	control := make(chan bool, pingRoutine)
+	for _, ip := range ips {
+		wg.Add(1)
+		control <- false
+		handleProgress := handleProgressGenerator(bar) // 多线程进度条
+		go tcpingGoroutine(&wg, &mu, ip, tcpPort, pingTime, &data, control, handleProgress)
+	}
+	wg.Wait()
+	bar.Finish()
+
+	sort.Sort(CloudflareIPDataSet(data)) // 排序
+
+	// 下载测速
+	if !disableDownload { // 如果禁用下载测速就跳过
+		if len(data) > 0 { // IP数组长度(IP数量) 大于 0 时继续
+			if len(data) < downloadTestCount { // 如果IP数组长度(IP数量) 小于 下载测速次数，则次数改为IP数
+				//fmt.Println("\n[信息] IP 数量小于下载测速次数（" + strconv.Itoa(downloadTestCount) + " < " + strconv.Itoa(len(data)) + "），下载测速次数改为IP数。\n")
+				downloadTestCount = len(data)
+			}
+			var downloadTestCount_2 int // 临时的下载测速次数
+			if timeLimit == 9999 && speedLimit == 0 {
+				downloadTestCount_2 = downloadTestCount // 如果没有指定条件，则临时变量为下载测速次数
+				fmt.Println("开始下载测速：")
+			} else if timeLimit > 0 || speedLimit >= 0 {
+				downloadTestCount_2 = len(data) // 如果指定了任意一个条件，则临时变量改为总数量
+				fmt.Println("开始下载测速（延迟时间上限：" + strconv.Itoa(timeLimit) + " ms，下载速度下限：" + strconv.Itoa(speedLimit) + " MB/s）：")
+			}
+			bar = pb.Simple.Start(downloadTestCount)
+			for i := 0; i < downloadTestCount_2; i++ {
+				_, speed := DownloadSpeedHandler(data[i].ip)
+				data[i].downloadSpeed = speed
+				if int(data[i].pingTime) <= timeLimit && int(float64(speed)/1024/1024) >= speedLimit {
+					data_2 = append(data_2, data[i]) // 延迟和速度均满足条件时，添加到新数组中
+					bar.Add(1)
+					if len(data_2) == downloadTestCount { // 满足条件的 IP =下载测速次数，则跳出循环
+						break
+					}
+				} else if int(data[i].pingTime) > timeLimit {
+					break
+				}
+			}
+			bar.Finish()
+		} else {
+			fmt.Println("\n[信息] IP数量为 0，跳过下载测速。")
+		}
+	}
+
+	if len(data_2) > 0 { // 如果该数字有内容，说明进行过指定条件的下载测速
+		sort.Sort(CloudflareIPDataSetD(data_2)) // 排序
+		if outputFile != "" {
+			ExportCsv(outputFile, data_2) // 输出结果到文件（指定延迟时间或下载速度的）
+		}
+		printResult(data_2) // 显示最快结果（指定延迟时间或下载速度的）
+	} else {
+		if outputFile != "" {
+			ExportCsv(outputFile, data) // 输出结果到文件
+		}
+		printResult(data) // 显示最快结果
+	}
+}
+
+// 显示最快结果
+func printResult(data []CloudflareIPData) {
+	sysType := runtime.GOOS
+	if printResultNum > 0 { // 如果禁止直接输出结果就跳过
+		dateString := convertToString(data) // 转为多维数组 [][]String
+		if len(dateString) > 0 {            // IP数组长度(IP数量) 大于 0 时继续
+			if len(dateString) < printResultNum { // 如果IP数组长度(IP数量) 小于  打印次数，则次数改为IP数量
+				//fmt.Println("\n[信息] IP 数量小于显示结果数量（" + strconv.Itoa(printResultNum) + " < " + strconv.Itoa(len(dateString)) + "），显示结果数量改为IP数量。\n")
+				printResultNum = len(dateString)
+			}
+			if ipv6Mode { // IPv6 太长了，所以需要调整一下间隔
+				fmt.Printf("%-40s%-5s%-5s%-5s%-6s%-11s\n", "IP 地址", "已发送", "已接收", "丢包率", "平均延迟", "下载速度 (MB/s)")
+				for i := 0; i < printResultNum; i++ {
+					fmt.Printf("%-42s%-8s%-8s%-8s%-10s%-15s\n", ipPadding(dateString[i][0]), dateString[i][1], dateString[i][2], dateString[i][3], dateString[i][4], dateString[i][5])
+				}
+			} else {
+				fmt.Printf("%-16s%-5s%-5s%-5s%-6s%-11s\n", "IP 地址", "已发送", "已接收", "丢包率", "平均延迟", "下载速度 (MB/s)")
+				for i := 0; i < printResultNum; i++ {
+					fmt.Printf("%-18s%-8s%-8s%-8s%-10s%-15s\n", ipPadding(dateString[i][0]), dateString[i][1], dateString[i][2], dateString[i][3], dateString[i][4], dateString[i][5])
+				}
+			}
+
+			if versionNew != "" {
+				fmt.Println("\n发现新版本 [" + versionNew + "]！请前往 [https://github.com/XIU2/CloudflareSpeedTest] 更新！")
+			}
+
+			if sysType == "windows" { // 如果是 Windows 系统，则需要按下 回车键 或 Ctrl+C 退出
+				if outputFile != "" {
+					fmt.Printf("\n完整测速结果已写入 %v 文件，请使用记事本/表格软件查看。\n按下 回车键 或 Ctrl+C 退出。", outputFile)
+				} else {
+					fmt.Printf("\n按下 回车键 或 Ctrl+C 退出。")
+				}
+				var pause int
+				fmt.Scanln(&pause)
+			} else { // 其它系统直接退出
+				if outputFile != "" {
+					fmt.Println("\n完整测速结果已写入 " + outputFile + " 文件，请使用记事本/表格软件查看。")
+				}
+			}
+		} else {
+			fmt.Println("\n[信息] IP数量为 0，跳过输出结果。")
+		}
+	} else {
+		fmt.Println("\n完整测速结果已写入 " + outputFile + " 文件，请使用记事本/表格软件查看。")
+	}
+}
+
+// 检查更新
+func checkUpdate() {
+	timeout := time.Duration(10 * time.Second)
+	client := http.Client{Timeout: timeout}
+	res, err := client.Get("https://api.xiuer.pw/ver/cloudflarespeedtest.txt")
+	if err == nil {
+		// 读取资源数据 body: []byte
+		body, err := ioutil.ReadAll(res.Body)
+		// 关闭资源流
+		res.Body.Close()
+		if err == nil {
+			if string(body) != version {
+				versionNew = string(body)
+			}
+		}
+	}
+}

tcping.go

@@ -1,159 +1,172 @@
-package main
+package main
-
+
-import (
+import (
-	"context"
+	"context"
-	"github.com/VividCortex/ewma"
+	"io"
-	"io"
+	"net"
-	"net"
+	"net/http"
-	"net/http"
+	"strconv"
-	"strconv"
+	"sync"
-	"sync"
+	"time"
-	"time"
+
-)
+	"github.com/VividCortex/ewma"
-
+)
-//bool connectionSucceed float32 time
+
-func tcping(ip net.IPAddr) (bool, float32) {
+//bool connectionSucceed float32 time
-	startTime := time.Now()
+func tcping(ip net.IPAddr, tcpPort int) (bool, float32) {
-	conn, err := net.DialTimeout("tcp", ip.String()+":"+strconv.Itoa(defaultTcpPort), tcpConnectTimeout)
+	startTime := time.Now()
-	if err != nil {
+	var fullAddress string
-		return false, 0
+	//fmt.Println(ip.String())
-	} else {
+	if ipv6Mode { // IPv6 需要加上 []
-		var endTime = time.Since(startTime)
+		fullAddress = "[" + ip.String() + "]:" + strconv.Itoa(tcpPort)
-		var duration = float32(endTime.Microseconds()) / 1000.0
+	} else {
-		_ = conn.Close()
+		fullAddress = ip.String() + ":" + strconv.Itoa(tcpPort)
-		return true, duration
+	}
-	}
+	conn, err := net.DialTimeout("tcp", fullAddress, tcpConnectTimeout)
-}
+	if err != nil {
-
+		return false, 0
-//pingReceived pingTotalTime
+	} else {
-func checkConnection(ip net.IPAddr) (int, float32) {
+		var endTime = time.Since(startTime)
-	pingRecv := 0
+		var duration = float32(endTime.Microseconds()) / 1000.0
-	var pingTime float32 = 0.0
+		_ = conn.Close()
-	for i := 1; i <= failTime; i++ {
+		return true, duration
-		pingSucceed, pingTimeCurrent := tcping(ip)
+	}
-		if pingSucceed {
+}
-			pingRecv++
+
-			pingTime += pingTimeCurrent
+//pingReceived pingTotalTime
-		}
+func checkConnection(ip net.IPAddr, tcpPort int) (int, float32) {
-	}
+	pingRecv := 0
-	return pingRecv, pingTime
+	var pingTime float32 = 0.0
-}
+	for i := 1; i <= failTime; i++ {
-
+		pingSucceed, pingTimeCurrent := tcping(ip, tcpPort)
-//return Success packetRecv averagePingTime specificIPAddr
+		if pingSucceed {
-func tcpingHandler(ip net.IPAddr, pingCount int, progressHandler func(e progressEvent)) (bool, int, float32, net.IPAddr) {
+			pingRecv++
-	ipCanConnect := false
+			pingTime += pingTimeCurrent
-	pingRecv := 0
+		}
-	var pingTime float32 = 0.0
+	}
-	for !ipCanConnect {
+	return pingRecv, pingTime
-		pingRecvCurrent, pingTimeCurrent := checkConnection(ip)
+}
-		if pingRecvCurrent != 0 {
+
-			ipCanConnect = true
+//return Success packetRecv averagePingTime specificIPAddr
-			pingRecv = pingRecvCurrent
+func tcpingHandler(ip net.IPAddr, tcpPort int, pingCount int, progressHandler func(e progressEvent)) (bool, int, float32, net.IPAddr) {
-			pingTime = pingTimeCurrent
+	ipCanConnect := false
-		} else {
+	pingRecv := 0
-			ip.IP[15]++
+	var pingTime float32 = 0.0
-			if ip.IP[15] == 0 {
+	for !ipCanConnect {
-				break
+		pingRecvCurrent, pingTimeCurrent := checkConnection(ip, tcpPort)
-			}
+		if pingRecvCurrent != 0 {
-			break
+			ipCanConnect = true
-		}
+			pingRecv = pingRecvCurrent
-	}
+			pingTime = pingTimeCurrent
-	if ipCanConnect {
+		} else {
-		progressHandler(AvailableIPFound)
+			ip.IP[15]++
-		for i := failTime; i < pingCount; i++ {
+			if ip.IP[15] == 0 {
-			pingSuccess, pingTimeCurrent := tcping(ip)
+				break
-			progressHandler(NormalPing)
+			}
-			if pingSuccess {
+			break
-				pingRecv++
+		}
-				pingTime += pingTimeCurrent
+	}
-			}
+	if ipCanConnect {
-		}
+		progressHandler(AvailableIPFound)
-		return true, pingRecv, pingTime / float32(pingRecv), ip
+		for i := failTime; i < pingCount; i++ {
-	} else {
+			pingSuccess, pingTimeCurrent := tcping(ip, tcpPort)
-		progressHandler(NoAvailableIPFound)
+			progressHandler(NormalPing)
-		return false, 0, 0, net.IPAddr{}
+			if pingSuccess {
-	}
+				pingRecv++
-}
+				pingTime += pingTimeCurrent
-
+			}
-func tcpingGoroutine(wg *sync.WaitGroup, mutex *sync.Mutex, ip net.IPAddr, pingCount int, csv *[]CloudflareIPData, control chan bool, progressHandler func(e progressEvent)) {
+		}
-	defer wg.Done()
+		return true, pingRecv, pingTime / float32(pingRecv), ip
-	success, pingRecv, pingTimeAvg, currentIP := tcpingHandler(ip, pingCount, progressHandler)
+	} else {
-	if success {
+		progressHandler(NoAvailableIPFound)
-		mutex.Lock()
+		return false, 0, 0, net.IPAddr{}
-		var cfdata CloudflareIPData
+	}
-		cfdata.ip = currentIP
+}
-		cfdata.pingReceived = pingRecv
+
-		cfdata.pingTime = pingTimeAvg
+func tcpingGoroutine(wg *sync.WaitGroup, mutex *sync.Mutex, ip net.IPAddr, tcpPort int, pingCount int, csv *[]CloudflareIPData, control chan bool, progressHandler func(e progressEvent)) {
-		cfdata.pingCount = pingCount
+	defer wg.Done()
-		*csv = append(*csv, cfdata)
+	success, pingRecv, pingTimeAvg, currentIP := tcpingHandler(ip, tcpPort, pingCount, progressHandler)
-		mutex.Unlock()
+	if success {
-	}
+		mutex.Lock()
-	<-control
+		var cfdata CloudflareIPData
-}
+		cfdata.ip = currentIP
-
+		cfdata.pingReceived = pingRecv
-func GetDialContextByAddr(fakeSourceAddr string) func(ctx context.Context, network, address string) (net.Conn, error) {
+		cfdata.pingTime = pingTimeAvg
-	return func(ctx context.Context, network, address string) (net.Conn, error) {
+		cfdata.pingCount = pingCount
-		c, e := (&net.Dialer{}).DialContext(ctx, network, fakeSourceAddr)
+		*csv = append(*csv, cfdata)
-		return c, e
+		mutex.Unlock()
-	}
+	}
-}
+	<-control
-
+}
-//bool : can download,float32 downloadSpeed
+
-func DownloadSpeedHandler(ip net.IPAddr) (bool, float32) {
+func GetDialContextByAddr(fakeSourceAddr string) func(ctx context.Context, network, address string) (net.Conn, error) {
-	var client = http.Client{
+	return func(ctx context.Context, network, address string) (net.Conn, error) {
-		Transport:     nil,
+		c, e := (&net.Dialer{}).DialContext(ctx, network, fakeSourceAddr)
-		CheckRedirect: nil,
+		return c, e
-		Jar:           nil,
+	}
-		Timeout:       0,
+}
-	}
+
-	client.Transport = &http.Transport{
+//bool : can download,float32 downloadSpeed
-		DialContext: GetDialContextByAddr(ip.String() + ":443"),
+func DownloadSpeedHandler(ip net.IPAddr) (bool, float32) {
-	}
+	var client = http.Client{
-	response, err := client.Get(url)
+		Transport:     nil,
-
+		CheckRedirect: nil,
-	if err != nil {
+		Jar:           nil,
-		return false, 0
+		Timeout:       downloadTestTime,
-	} else {
+	}
-		defer func() { _ = response.Body.Close() }()
+	var fullAddress string
-		if response.StatusCode == 200 {
+	if ipv6Mode { // IPv6 需要加上 []
-			timeStart := time.Now()
+		fullAddress = "[" + ip.String() + "]:443"
-			timeEnd := timeStart.Add(downloadTestTime)
+	} else {
-
+		fullAddress = ip.String() + ":443"
-			contentLength := response.ContentLength
+	}
-			buffer := make([]byte, downloadBufferSize)
+	client.Transport = &http.Transport{
-
+		DialContext: GetDialContextByAddr(fullAddress),
-			var contentRead int64 = 0
+	}
-			var timeSlice = downloadTestTime / 100
+	response, err := client.Get(url)
-			var timeCounter = 1
+	if err != nil {
-			var lastContentRead int64 = 0
+		return false, 0
-
+	} else {
-			var nextTime = timeStart.Add(timeSlice * time.Duration(timeCounter))
+		defer func() { _ = response.Body.Close() }()
-			e := ewma.NewMovingAverage()
+		if response.StatusCode == 200 {
-
+			timeStart := time.Now()
-			for ; contentLength != contentRead; {
+			timeEnd := timeStart.Add(downloadTestTime)
-				var currentTime = time.Now()
+
-				if currentTime.After(nextTime) {
+			contentLength := response.ContentLength
-					timeCounter += 1
+			buffer := make([]byte, downloadBufferSize)
-					nextTime = timeStart.Add(timeSlice * time.Duration(timeCounter))
+
-					e.Add(float64(contentRead - lastContentRead))
+			var contentRead int64 = 0
-					lastContentRead = contentRead
+			var timeSlice = downloadTestTime / 100
-				}
+			var timeCounter = 1
-				if currentTime.After(timeEnd) {
+			var lastContentRead int64 = 0
-					break
+
-				}
+			var nextTime = timeStart.Add(timeSlice * time.Duration(timeCounter))
-				bufferRead, err := response.Body.Read(buffer)
+			e := ewma.NewMovingAverage()
-				contentRead += int64(bufferRead)
+
-				if err != nil {
+			for contentLength != contentRead {
-					if err != io.EOF {
+				var currentTime = time.Now()
-						break
+				if currentTime.After(nextTime) {
-					} else {
+					timeCounter += 1
-						e.Add(float64(contentRead-lastContentRead) / (float64(nextTime.Sub(currentTime)) / float64(timeSlice)))
+					nextTime = timeStart.Add(timeSlice * time.Duration(timeCounter))
-					}
+					e.Add(float64(contentRead - lastContentRead))
-				}
+					lastContentRead = contentRead
-			}
+				}
-			return true, float32(e.Value()) / (float32(downloadTestTime.Seconds()) / 100)
+				if currentTime.After(timeEnd) {
-		} else {
+					break
-			return false, 0
+				}
-		}
+				bufferRead, err := response.Body.Read(buffer)
-	}
+				contentRead += int64(bufferRead)
-}
+				if err != nil {
+					if err != io.EOF {
+						break
+					} else {
+						e.Add(float64(contentRead-lastContentRead) / (float64(nextTime.Sub(currentTime)) / float64(timeSlice)))
+					}
+				}
+			}
+			return true, float32(e.Value()) / (float32(downloadTestTime.Seconds()) / 120)
+		} else {
+			return false, 0
+		}
+	}
+}

util.go

@@ -1,123 +1,164 @@
-package main
+package main
-
+
-import (
+import (
-	"encoding/csv"
+	"encoding/csv"
-	"log"
+	"log"
-	"math/rand"
+	"math/rand"
-	"net"
+	"net"
-	"os"
+	"os"
-	"strconv"
+	"strconv"
-	"time"
+	"time"
-
+
-	"github.com/cheggaaa/pb/v3"
+	"github.com/cheggaaa/pb/v3"
-)
+)
-
+
-type CloudflareIPData struct {
+type CloudflareIPData struct {
-	ip            net.IPAddr
+	ip            net.IPAddr
-	pingCount     int
+	pingCount     int
-	pingReceived  int
+	pingReceived  int
-	recvRate      float32
+	recvRate      float32
-	downloadSpeed float32
+	downloadSpeed float32
-	pingTime      float32
+	pingTime      float32
-}
+}
-
+
-func (cf *CloudflareIPData) getRecvRate() float32 {
+func (cf *CloudflareIPData) getRecvRate() float32 {
-	if cf.recvRate == 0 {
+	if cf.recvRate == 0 {
-		cf.recvRate = float32(cf.pingReceived) / float32(cf.pingCount)
+		pingLost := cf.pingCount - cf.pingReceived
-	}
+		cf.recvRate = float32(pingLost) / float32(cf.pingCount)
-	return cf.recvRate
+	}
-}
+	return cf.recvRate
-
+}
-func ExportCsv(filePath string, data []CloudflareIPData) {
+
-	fp, err := os.Create(filePath)
+func ExportCsv(filePath string, data []CloudflareIPData) {
-	if err != nil {
+	fp, err := os.Create(filePath)
-		log.Fatalf("创建文件["+filePath+"]句柄失败,%v", err)
+	if err != nil {
-		return
+		log.Fatalf("创建文件["+filePath+"]句柄失败,%v", err)
-	}
+		return
-	defer fp.Close()
+	}
-	w := csv.NewWriter(fp) //创建一个新的写入文件流
+	defer fp.Close()
-	w.Write([]string{"IP 地址", "Ping 发送次数", "Ping 接收次数", "Ping 接收率", "平均延迟", "下载速度 (MB/s)"})
+	w := csv.NewWriter(fp) //创建一个新的写入文件流
-	w.WriteAll(convertToString(data))
+	w.Write([]string{"IP 地址", "已发送", "已接收", "丢包率", "平均延迟", "下载速度 (MB/s)"})
-	w.Flush()
+	w.WriteAll(convertToString(data))
-}
+	w.Flush()
-
+}
-//"IP Address","Ping Count","Ping received","Ping received rate","Ping time","Download speed"
+
-
+func (cf *CloudflareIPData) toString() []string {
-func (cf *CloudflareIPData) toString() []string {
+	result := make([]string, 6)
-	result := make([]string, 6)
+	result[0] = cf.ip.String()
-	result[0] = cf.ip.String()
+	result[1] = strconv.Itoa(cf.pingCount)
-	result[1] = strconv.Itoa(cf.pingCount)
+	result[2] = strconv.Itoa(cf.pingReceived)
-	result[2] = strconv.Itoa(cf.pingReceived)
+	result[3] = strconv.FormatFloat(float64(cf.getRecvRate()), 'f', 2, 32)
-	result[3] = strconv.FormatFloat(float64(cf.getRecvRate()), 'f', 4, 32)
+	result[4] = strconv.FormatFloat(float64(cf.pingTime), 'f', 2, 32)
-	result[4] = strconv.FormatFloat(float64(cf.pingTime), 'f', 4, 32)
+	result[5] = strconv.FormatFloat(float64(cf.downloadSpeed)/1024/1024, 'f', 2, 32)
-	result[5] = strconv.FormatFloat(float64(cf.downloadSpeed)/1024/1024, 'f', 4, 32)
+	return result
-	return result
+}
-}
+
-
+func convertToString(data []CloudflareIPData) [][]string {
-func convertToString(data []CloudflareIPData) [][]string {
+	result := make([][]string, 0)
-	result := make([][]string, 0)
+	for _, v := range data {
-	for _, v := range data {
+		result = append(result, v.toString())
-		result = append(result, v.toString())
+	}
-	}
+	return result
-	return result
+}
-}
+
-
+var pingTime int
-var pingTime int
+var pingRoutine int
-var pingRoutine int
+
-
+type progressEvent int
-var ipEndWith uint8 = 0
+
-
+const (
-type progressEvent int
+	NoAvailableIPFound progressEvent = iota
-
+	AvailableIPFound
-const (
+	NormalPing
-	NoAvailableIPFound progressEvent = iota
+)
-	AvailableIPFound
+
-	NormalPing
+var url string
-)
+
-
+var downloadTestTime time.Duration
-const url string = "https://apple.freecdn.workers.dev/105/media/us/iphone-11-pro/2019/3bd902e4-0752-4ac1-95f8-6225c32aec6d/films/product/iphone-11-pro-product-tpl-cc-us-2019_1280x720h.mp4"
+
-
+const downloadBufferSize = 1024
-var downloadTestTime time.Duration
+
-
+var downloadTestCount int
-const downloadBufferSize = 1024
+
-
+//const defaultTcpPort = 443
-var downloadTestCount int
+const tcpConnectTimeout = time.Second * 1
-
+
-const defaultTcpPort = 443
+var failTime int
-const tcpConnectTimeout = time.Second * 1
+
-const failTime = 4
+// 平均延迟排序（丢包另算）
-
+type CloudflareIPDataSet []CloudflareIPData
-type CloudflareIPDataSet []CloudflareIPData
+
-
+// 下载速度排序
-func initipEndWith() {
+type CloudflareIPDataSetD []CloudflareIPData
-	rand.Seed(time.Now().UnixNano())
+
-	ipEndWith = uint8(rand.Intn(254) + 1)
+func initRandSeed() {
-}
+	rand.Seed(time.Now().UnixNano())
-
+}
-func handleProgressGenerator(pb *pb.ProgressBar) func(e progressEvent) {
+
-	return func(e progressEvent) {
+func randipEndWith(num int) uint8 {
-		switch e {
+	return uint8(rand.Intn(num))
-		case NoAvailableIPFound:
+}
-			pb.Add(pingTime)
+
-		case AvailableIPFound:
+func GetRandomString() string {
-			pb.Add(failTime)
+	str := "0123456789abcdef"
-		case NormalPing:
+	bytes := []byte(str)
-			pb.Increment()
+	result := []byte{}
-		}
+	r := rand.New(rand.NewSource(time.Now().UnixNano()))
-	}
+	for i := 0; i < 4; i++ {
-}
+		result = append(result, bytes[r.Intn(len(bytes))])
-
+	}
-func (cfs CloudflareIPDataSet) Len() int {
+	return string(result)
-	return len(cfs)
+}
-}
+
-
+func ipPadding(ip string) string {
-func (cfs CloudflareIPDataSet) Less(i, j int) bool {
+	var ipLength int
-	if (cfs)[i].getRecvRate() != cfs[j].getRecvRate() {
+	var ipPrint string
-		return cfs[i].getRecvRate() > cfs[j].getRecvRate()
+	ipPrint = ip
-	}
+	ipLength = len(ipPrint)
-	return cfs[i].pingTime < cfs[j].pingTime
+	if ipLength < 15 {
-}
+		for i := 0; i <= 15-ipLength; i++ {
-
+			ipPrint += " "
-func (cfs CloudflareIPDataSet) Swap(i, j int) {
+		}
-	cfs[i], cfs[j] = cfs[j], cfs[i]
+	}
-}
+	return ipPrint
+}
+
+func handleProgressGenerator(pb *pb.ProgressBar) func(e progressEvent) {
+	return func(e progressEvent) {
+		switch e {
+		case NoAvailableIPFound:
+			pb.Add(pingTime)
+		case AvailableIPFound:
+			pb.Add(failTime)
+		case NormalPing:
+			pb.Increment()
+		}
+	}
+}
+
+func (cfs CloudflareIPDataSet) Len() int {
+	return len(cfs)
+}
+
+func (cfs CloudflareIPDataSet) Less(i, j int) bool {
+	if (cfs)[i].getRecvRate() != cfs[j].getRecvRate() {
+		return cfs[i].getRecvRate() < cfs[j].getRecvRate()
+	}
+	return cfs[i].pingTime < cfs[j].pingTime
+}
+
+func (cfs CloudflareIPDataSet) Swap(i, j int) {
+	cfs[i], cfs[j] = cfs[j], cfs[i]
+}
+
+func (cfs CloudflareIPDataSetD) Len() int {
+	return len(cfs)
+}
+
+func (cfs CloudflareIPDataSetD) Less(i, j int) bool {
+	return cfs[i].downloadSpeed > cfs[j].downloadSpeed
+}
+
+func (cfs CloudflareIPDataSetD) Swap(i, j int) {
+	cfs[i], cfs[j] = cfs[j], cfs[i]
+}